add windows compatibility

Author: xtaci

charset_linux.go

@@ -0,0 +1,34 @@
+// +build !windows
+
+package main
+
+import (
+	"os"
+	"os/exec"
+	"strings"
+)
+
+var (
+	wideCharset = []string{"zh_", "jp_", "ko_", "ja_", "th_", "hi_"}
+)
+
+func fixCharset() {
+	locale := os.Getenv("LANG")
+
+	var asianCharset bool
+	for k := range wideCharset {
+		if strings.HasPrefix(locale, wideCharset[k]) {
+			asianCharset = true
+		}
+	}
+
+	if asianCharset {
+		os.Setenv("LANG", "C.UTF-8")
+		cmd := exec.Command(os.Args[0])
+		cmd.Stdin = os.Stdin
+		cmd.Stdout = os.Stdout
+		cmd.Stderr = os.Stderr
+		cmd.Run()
+		os.Exit(0)
+	}
+}

charset_windows.go

@@ -0,0 +1,23 @@
+// +build windows
+
+package main
+
+import (
+	"os"
+	"os/exec"
+)
+
+const codePageSet = "SAFEBOX_CODEPAGE_SET"
+
+func fixCharset() {
+	if os.Getenv(codePageSet) != "1" {
+		exec.Command("CHCP", "65001").Run()
+		os.Setenv(codePageSet, "1")
+		cmd := exec.Command(os.Args[0])
+		cmd.Stdin = os.Stdin
+		cmd.Stdout = os.Stdout
+		cmd.Stderr = os.Stderr
+		cmd.Run()
+		os.Exit(0)
+	}
+}

main.go

@@ -1,10 +1,6 @@
 package main
 
 import (
-	"os"
-	"os/exec"
-	"strings"
-
 	"github.com/gdamore/tcell/v2"
 	"github.com/rivo/tview"
 )
@@ -29,30 +25,8 @@ var theme = tview.Theme{
 	ContrastSecondaryTextColor:  tcell.ColorDarkCyan,
 }
 
-var (
-	wideCharset = []string{"zh_", "jp_", "ko_", "ja_", "th_", "hi_"}
-)
-
 func main() {
-	locale := os.Getenv("LANG")
-
-	var asianCharset bool
-	for k := range wideCharset {
-		if strings.HasPrefix(locale, wideCharset[k]) {
-			asianCharset = true
-		}
-	}
-
-	if asianCharset {
-		os.Setenv("LANG", "C.UTF-8")
-		cmd := exec.Command(os.Args[0])
-		cmd.Stdin = os.Stdin
-		cmd.Stdout = os.Stdout
-		cmd.Stderr = os.Stderr
-		cmd.Run()
-		return
-	}
-
+	fixCharset()
 	app = tview.NewApplication()
 	tview.Styles = theme
 	initLayouts()