feat: i18n with default lang zh-Hans (translating)

cmd/geni18n/main.go

@@ -0,0 +1,84 @@
+// cmd/gen_i18n/main.go
+package main
+
+import (
+	"bufio"
+	"flag"
+	"fmt"
+	"io/fs"
+	"os"
+	"path/filepath"
+	"regexp"
+	"sort"
+	"strings"
+)
+
+func main() {
+	dir := flag.String("dir", "./i18n/locale", "Locales directory path")
+	out := flag.String("out", "i18n/i18nk/keys.go", "Output file path")
+	pkg := flag.String("pkg", "i18nk", "Package name for generated file")
+	flag.Parse()
+
+	keys := make(map[string]struct{})
+	re := regexp.MustCompile(`^\s*\[+\s*([^\]\[]+)\s*\]+`)
+
+	err := filepath.WalkDir(*dir, func(path string, d fs.DirEntry, err error) error {
+		if err != nil {
+			return err
+		}
+		if d.IsDir() || !strings.HasSuffix(d.Name(), ".toml") {
+			return nil
+		}
+		f, err := os.Open(path)
+		if err != nil {
+			return err
+		}
+		defer f.Close()
+
+		s := bufio.NewScanner(f)
+		for s.Scan() {
+			if m := re.FindStringSubmatch(s.Text()); m != nil {
+				keys[m[1]] = struct{}{}
+			}
+		}
+		return s.Err()
+	})
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error walking directory: %v\n", err)
+		os.Exit(1)
+	}
+
+	var list []string
+	for k := range keys {
+		list = append(list, k)
+	}
+	sort.Strings(list)
+
+	f, err := os.Create(*out)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error creating output file: %v\n", err)
+		os.Exit(1)
+	}
+	defer f.Close()
+
+	w := bufio.NewWriter(f)
+	fmt.Fprintf(w, "// Code generated by cmd/gen_i18n. DO NOT EDIT.\n")
+	fmt.Fprintf(w, "package %s\n\n", *pkg)
+	fmt.Fprintf(w, "const (\n")
+	for _, key := range list {
+		name := toPascal(key)
+		fmt.Fprintf(w, "\t%s = %q\n", name, key)
+	}
+	fmt.Fprintf(w, ")\n")
+	w.Flush()
+}
+
+func toPascal(key string) string {
+	parts := strings.Split(key, ".")
+	for i, p := range parts {
+		if len(p) > 0 {
+			parts[i] = strings.ToUpper(string(p[0])) + p[1:]
+		}
+	}
+	return strings.Join(parts, "")
+}

cmd/run.go

@@ -14,6 +14,8 @@ import (
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/core"
 	"github.com/krau/SaveAny-Bot/dao"
+	"github.com/krau/SaveAny-Bot/i18n"
+	"github.com/krau/SaveAny-Bot/i18n/i18nk"
 	"github.com/krau/SaveAny-Bot/storage"
 	"github.com/spf13/cobra"
 )
@@ -25,41 +27,52 @@ func Run(_ *cobra.Command, _ []string) {
 	quit := make(chan os.Signal, 1)
 	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
 	sig := <-quit
-	common.Log.Info(sig, ", exitting...")
-	defer common.Log.Info("Bye!")
+	common.Log.Info(sig, i18n.T(i18nk.Exiting))
+	defer common.Log.Info(i18n.T(i18nk.Bye))
 	if config.Cfg.NoCleanCache {
 		return
 	}
 	if config.Cfg.Temp.BasePath != "" && !config.Cfg.Stream {
 		if slices.Contains([]string{"/", ".", "\\", ".."}, filepath.Clean(config.Cfg.Temp.BasePath)) {
-			common.Log.Error("无效的缓存文件夹: ", config.Cfg.Temp.BasePath)
+			common.Log.Error(i18n.T(i18nk.InvalidCacheDir, map[string]any{
+				"Path": config.Cfg.Temp.BasePath,
+			}))
 			return
 		}
 		currentDir, err := os.Getwd()
 		if err != nil {
-			common.Log.Error("获取工作目录失败: ", err)
+			common.Log.Error(i18n.T(i18nk.GetWorkdirFailed, map[string]any{
+				"Error": err,
+			}))
 			return
 		}
 		cachePath := filepath.Join(currentDir, config.Cfg.Temp.BasePath)
 		cachePath, err = filepath.Abs(cachePath)
 		if err != nil {
-			common.Log.Error("获取缓存绝对路径失败: ", err)
+			common.Log.Error(i18n.T(i18nk.GetCacheAbsPathFailed, map[string]any{
+				"Error": err,
+			}))
 			return
 		}
-		common.Log.Info("正在清理缓存文件夹: ", cachePath)
+		common.Log.Info(i18n.T(i18nk.CleaningCache, map[string]any{
+			"Path": cachePath,
+		}))
 		if err := common.RemoveAllInDir(cachePath); err != nil {
-			common.Log.Error("清理缓存失败: ", err)
+			common.Log.Error(i18n.T(i18nk.CleanCacheFailed, map[string]any{
+				"Error": err,
+			}))
 		}
 	}
 }
 
 func InitAll() {
 	if err := config.Init(); err != nil {
-		fmt.Println("加载配置文件失败: ", err)
+		fmt.Println("Failed to load config:", err)
 		os.Exit(1)
 	}
 	common.InitLogger()
-	common.Log.Info("正在启动 SaveAny-Bot...")
+	i18n.Init(config.Cfg.Lang)
+	common.Log.Info(i18n.T(i18nk.Initing))
 	dao.Init()
 	storage.LoadStorages()
 	common.Init()