File: translate.go

package info (click to toggle)
golang-github-axgle-mahonia 0.0~git20180208.3358181-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 1,856 kB
  • sloc: makefile: 2
file content (50 lines) | stat: -rw-r--r-- 922 bytes parent folder | download | duplicates (2)
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package mahonia

import "unicode/utf8"

// Translate enables a Decoder to implement go-charset's Translator interface.
func (d Decoder) Translate(data []byte, eof bool) (n int, cdata []byte, err error) {
	cdata = make([]byte, len(data)+1)
	destPos := 0

	for n < len(data) {
		rune, size, status := d(data[n:])

		switch status {
		case STATE_ONLY:
			n += size
			continue

		case NO_ROOM:
			if !eof {
				return n, cdata[:destPos], nil
			}
			rune = 0xfffd
			n = len(data)

		default:
			n += size
		}

		if rune < 128 {
			if destPos >= len(cdata) {
				cdata = doubleLength(cdata)
			}
			cdata[destPos] = byte(rune)
			destPos++
		} else {
			if destPos+utf8.RuneLen(rune) > len(cdata) {
				cdata = doubleLength(cdata)
			}
			destPos += utf8.EncodeRune(cdata[destPos:], rune)
		}
	}

	return n, cdata[:destPos], nil
}

func doubleLength(b []byte) []byte {
	b2 := make([]byte, 2*len(b))
	copy(b2, b)
	return b2
}