84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
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
|
|
}
|
|
}
|