-
Notifications
You must be signed in to change notification settings - Fork 0
/
util.go
37 lines (34 loc) · 873 Bytes
/
util.go
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
package nesbconvertpin
import (
"bufio"
"bytes"
"encoding/xml"
"golang.org/x/text/encoding/simplifiedchinese"
"golang.org/x/text/transform"
"io"
"os"
"path"
)
func getRootDir() string {
return os.Getenv("FAPWORKDIR")
}
func fileScanner(filename string) (file *os.File, scanner *bufio.Scanner) {
var err error
file, err = os.Open(path.Join(getRootDir(), filename))
if err != nil {
panic(err)
}
scanner = bufio.NewScanner(file)
return
}
func getGbFileDecoder(fileName string) *xml.Decoder {
buf, err := os.ReadFile(fileName)
if err != nil {
panic(err)
}
decoder := xml.NewDecoder(transform.NewReader(bytes.NewReader(buf), simplifiedchinese.GB18030.NewDecoder()))
decoder.CharsetReader = func(charset string, input io.Reader) (io.Reader, error) {
return transform.NewReader(input, simplifiedchinese.GB18030.NewDecoder()), nil
}
return decoder
}