bnbnet/lang.go at main · zhyonc/bnbnet · GitHub

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
51
52
53
54
55
56
57
58
59
60
61
62
63
package bnbnet

import (
	"bytes"
	"io"
	"log/slog"
	"strings"

	"github.com/zhyonc/bnbnet/enum"
	"golang.org/x/text/encoding"
	"golang.org/x/text/encoding/japanese"
	"golang.org/x/text/encoding/korean"
	"golang.org/x/text/encoding/simplifiedchinese"
	"golang.org/x/text/encoding/traditionalchinese"
	"golang.org/x/text/transform"
)

var (
	langEncoder *encoding.Encoder
	langDecoder *encoding.Decoder
)

func SetCharset(region enum.Region) {
	switch region {
	case enum.RegionKR:
		langEncoder = korean.EUCKR.NewEncoder()
		langDecoder = korean.EUCKR.NewDecoder()
	case enum.RegionCN:
		langEncoder = simplifiedchinese.GBK.NewEncoder()
		langDecoder = simplifiedchinese.GBK.NewDecoder()
	case enum.RegionTW:
		langEncoder = traditionalchinese.Big5.NewEncoder()
		langDecoder = traditionalchinese.Big5.NewDecoder()
	case enum.RegionJP:
		langEncoder = japanese.ShiftJIS.NewEncoder()
		langDecoder = japanese.ShiftJIS.NewDecoder()
	default:
		langEncoder = encoding.Nop.NewEncoder()
		langDecoder = encoding.Nop.NewDecoder()
	}
}

func GetLangBuf(s string) []byte {
	reader := strings.NewReader(s)
	transformer := transform.NewReader(reader, langEncoder)
	buf, err := io.ReadAll(transformer)
	if err != nil {
		slog.Error("Failed to get local buf", "str", s)
		return nil
	}
	return buf
}

func GetLangStr(buf []byte) string {
	reader := bytes.NewReader(buf)
	transformer := transform.NewReader(reader, langDecoder)
	decodedBytes, err := io.ReadAll(transformer)
	if err != nil {
		slog.Error("Failed to get local str", "buf", buf)
		return ""
	}
	return string(decodedBytes)
}