aboutsummaryrefslogtreecommitdiffstats
path: root/vendor/github.com/shurcooL/vfsgen/stringwriter.go
blob: a781efdc6dd8672aa632c46c7d0c3f41c74c4f83 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
package vfsgen

import (
	"io"
)

// stringWriter writes given bytes to underlying io.Writer as a Go interpreted string literal value,
// not including double quotes. It tracks the total number of bytes written.
type stringWriter struct {
	io.Writer
	N int64 // Total bytes written.
}

func (sw *stringWriter) Write(p []byte) (n int, err error) {
	const hex = "0123456789abcdef"
	buf := []byte{'\\', 'x', 0, 0}
	for _, b := range p {
		buf[2], buf[3] = hex[b/16], hex[b%16]
		_, err = sw.Writer.Write(buf)
		if err != nil {
			return n, err
		}
		n++
		sw.N++
	}
	return n, nil
}