engigu
107 lines
2.3 KiB
Go
107 lines
2.3 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/engigu/baihu-panel/internal/constant"
|
|
"github.com/engigu/baihu-panel/internal/database"
|
|
"github.com/engigu/baihu-panel/internal/logger"
|
|
"github.com/engigu/baihu-panel/internal/router"
|
|
"github.com/engigu/baihu-panel/internal/services"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type App struct {
|
|
Config *services.AppConfig
|
|
Router *gin.Engine
|
|
}
|
|
|
|
func New() *App {
|
|
app := InitBasic()
|
|
app.initRouter()
|
|
return app
|
|
}
|
|
|
|
// InitBasic 初始化基础环境（配置和数据库），不启动后台服务和路由
|
|
func InitBasic() *App {
|
|
app := &App{}
|
|
app.initConfig()
|
|
app.initDatabase()
|
|
return app
|
|
}
|
|
|
|
func (a *App) initConfig() {
|
|
cfg, err := services.LoadConfig(constant.ConfigPath)
|
|
if err != nil {
|
|
logger.Fatalf("Failed to load config: %v", err)
|
|
}
|
|
a.Config = cfg
|
|
|
|
// Ensure directories exist
|
|
err = os.MkdirAll(constant.DataDir, 0755)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = os.MkdirAll(constant.ScriptsWorkDir, 0755)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
a.setupBaihuBin()
|
|
}
|
|
|
|
func (a *App) setupBaihuBin() {
|
|
binDir := filepath.Join(constant.DataDir, "bin")
|
|
_ = os.MkdirAll(binDir, 0755)
|
|
|
|
exe, err := os.Executable()
|
|
if err == nil {
|
|
linkPath := filepath.Join(binDir, "baihu")
|
|
if runtime.GOOS == "windows" {
|
|
linkPath += ".exe"
|
|
}
|
|
os.Remove(linkPath)
|
|
_ = os.Symlink(exe, linkPath)
|
|
}
|
|
}
|
|
|
|
func (a *App) initDatabase() {
|
|
dbCfg := &database.Config{
|
|
Type: a.Config.Database.Type,
|
|
Host: a.Config.Database.Host,
|
|
Port: a.Config.Database.Port,
|
|
User: a.Config.Database.User,
|
|
Password: a.Config.Database.Password,
|
|
DBName: a.Config.Database.DBName,
|
|
Path: a.Config.Database.Path,
|
|
}
|
|
|
|
if err := database.Init(dbCfg); err != nil {
|
|
logger.Fatalf("Failed to init database: %v", err)
|
|
}
|
|
|
|
// 执行 V3 迁移（ID 变更迁移）
|
|
if err := services.RunMigrationV3(); err != nil {
|
|
logger.Fatalf("Failed to run V3 migration: %v", err)
|
|
}
|
|
|
|
if err := database.Migrate(); err != nil {
|
|
logger.Fatalf("Failed to migrate database: %v", err)
|
|
}
|
|
}
|
|
|
|
func (a *App) initRouter() {
|
|
ctrls := router.RegisterControllers()
|
|
a.Router = router.Setup(ctrls)
|
|
}
|
|
|
|
func (a *App) Run() {
|
|
addr := fmt.Sprintf("%s:%d", a.Config.Server.Host, a.Config.Server.Port)
|
|
logger.Infof("Starting server on %s", addr)
|
|
a.Router.Run(addr)
|
|
}
|