feat: 初始化仓库

main.go

@@ -0,0 +1,83 @@
+package main
+
+import (
+	"fmt"
+	"github.com/fatih/color"
+	"github.com/gin-gonic/gin"
+	"log"
+	"os"
+	"runtime"
+	"syscall"
+	"unsafe"
+	"zhixiangtong/utils"
+)
+
+var (
+	kernel32        = syscall.NewLazyDLL("kernel32.dll")
+	setConsoleTitle = kernel32.NewProc("SetConsoleTitleW")
+	setConsoleMode  = kernel32.NewProc("SetConsoleMode")
+)
+
+const (
+	ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004
+)
+
+// enableVirtualTerminalProcessing 启用 Windows 的虚拟终端处理
+func enableVirtualTerminalProcessing(fd uintptr) {
+	var mode uint32
+	handle := syscall.Handle(fd)
+	syscall.GetConsoleMode(handle, &mode)
+	mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING
+	setConsoleMode.Call(uintptr(handle), uintptr(mode))
+}
+
+// 设置终端标题
+func setTerminalTitle(title string) {
+	utf16Title, _ := syscall.UTF16PtrFromString(title)
+	setConsoleTitle.Call(uintptr(unsafe.Pointer(utf16Title)))
+}
+
+func main() {
+	// 设置终端标题
+	setTerminalTitle("智享通")
+	if os.Getenv("TERM") == "" {
+		// 在 Windows 平台上启用虚拟终端处理，以支持颜色输出
+		enableVirtualTerminalProcessing(os.Stdout.Fd())
+	}
+	// 初始化配置
+	dataDir := "data"
+	// 确保 data 目录存在
+	if _, err := os.Stat(dataDir); os.IsNotExist(err) {
+		err := os.Mkdir(dataDir, os.ModePerm)
+		if err != nil {
+			log.Fatalf("无法创建 data 目录：%v", err)
+		}
+	}
+	config, err := utils.InitConfig(dataDir)
+	if err != nil {
+		log.Fatalf("初始化配置失败：%v", err)
+	}
+
+	// 初始化数据库
+	db, err := utils.InitDatabase(dataDir)
+	if err != nil {
+		log.Fatalf("初始化数据库失败：%v", err)
+	}
+	// 如果是 Windows 平台，强制启用颜色支持
+	if runtime.GOOS == "windows" {
+		color.NoColor = false
+	}
+	// 设置为生产模式
+	gin.SetMode(gin.ReleaseMode)
+	// 配置路由
+	router := utils.SetupRouter(db, config)
+
+	// 启动服务
+	addr := fmt.Sprintf("%s:%s", config.Host, config.Port)
+	log.Printf("服务正在运行在 %s", addr)
+	err = router.Run(addr)
+	if err != nil {
+		log.Fatalf("服务框架启动失败 %v", err)
+		return
+	}
+}