157 lines
2.9 KiB
Go
157 lines
2.9 KiB
Go
package langx
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type langx struct {
|
|
ops *options
|
|
codeMap map[string]int
|
|
transMap map[string]map[string]string
|
|
}
|
|
|
|
var l *langx = &langx{}
|
|
|
|
func init() {
|
|
|
|
l = &langx{
|
|
ops: defaultOptions(),
|
|
codeMap: make(map[string]int),
|
|
transMap: make(map[string]map[string]string),
|
|
}
|
|
}
|
|
|
|
// 设置
|
|
func InitLangx(ops ...Option) {
|
|
for _, opt := range ops {
|
|
opt(l.ops)
|
|
}
|
|
}
|
|
|
|
// 这是语言的Code
|
|
func RegisterCode(datas map[string]int) {
|
|
l.codeMap = datas
|
|
}
|
|
|
|
// 注册语言翻译
|
|
func RegisterTrans(langName string, trans map[string]string) {
|
|
l.transMap[langName] = trans
|
|
}
|
|
|
|
|
|
// 直接读取文件夹获取配置
|
|
// 要求：
|
|
// 1.json格式文件
|
|
// 2.code.json为自定义响应码 格式map[string]int{}
|
|
// 3.其他的json文件为对应语音 格式map[string]string{}
|
|
// 4.如果json解析错误将会panic
|
|
func RegisterDir(dir string) error {
|
|
// 遍历dir获取.json的文件
|
|
err := filepath.Walk(dir, func(path string, info fs.FileInfo, err error) error {
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
if strings.HasSuffix(info.Name(), ".json") {
|
|
// 读取文件
|
|
fileName := strings.Replace(info.Name(), ".json", "", 1)
|
|
by, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if fileName == "code" {
|
|
data := map[string]int{}
|
|
err = json.Unmarshal(by, &data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
RegisterCode(data)
|
|
} else {
|
|
data := map[string]string{}
|
|
err = json.Unmarshal(by, &data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
RegisterTrans(fileName, data)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// code文件为状态码,其他为对应的语言文件，文件名为语言名
|
|
return nil
|
|
}
|
|
|
|
// 获取翻译
|
|
// 包含Code和Message
|
|
func GetTrans(lang string, key string, format map[string]string) (code int, str string) {
|
|
code = GetCode(key)
|
|
str = GetFormat(lang, key, format)
|
|
return
|
|
}
|
|
|
|
// 根据Key获取code
|
|
func GetCode(key string) int {
|
|
code, ok := l.codeMap[key]
|
|
if !ok {
|
|
return l.ops.defaultCode
|
|
}
|
|
return code
|
|
}
|
|
|
|
// 获取翻译
|
|
func GetMsg(lang string, key string) string {
|
|
// 找指定语言
|
|
str, ok := l.transMap[lang]
|
|
if ok {
|
|
val, ok := str[key]
|
|
if ok {
|
|
return val
|
|
}
|
|
}
|
|
// 找默认语言
|
|
str, ok = l.transMap[l.ops.defaultLang]
|
|
if ok {
|
|
val, ok := str[key]
|
|
if ok {
|
|
return val
|
|
}
|
|
}
|
|
|
|
return key
|
|
}
|
|
|
|
// 从ctx里面获取语言
|
|
func GetMsgCtx(ctx context.Context, key string) string {
|
|
ctxVal := ctx.Value(l.ops.ctxLangKey)
|
|
lang := l.ops.defaultLang
|
|
if ctxVal != nil {
|
|
lang = ctxVal.(string)
|
|
}
|
|
return GetMsg(lang, key)
|
|
}
|
|
|
|
// 拼接回复
|
|
func GetFormat(lang string, key string, arr map[string]string) string {
|
|
str := GetMsg(lang, key)
|
|
for k, v := range arr {
|
|
str = strings.ReplaceAll(str, fmt.Sprintf(l.ops.replaceKey, k), v)
|
|
}
|
|
return str
|
|
}
|
|
|
|
// 获取默认Code
|
|
func GetDefaultCode() int {
|
|
return l.ops.defaultCode
|
|
}
|