103 lines
2.4 KiB
Go
103 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"io/fs"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
|
|
"flnxi-agent/internal/api"
|
|
"flnxi-agent/internal/docker"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
//go:embed web/dist/*
|
|
var webFS embed.FS
|
|
|
|
func main() {
|
|
// 初始化 Docker 客户端
|
|
dockerClient, err := docker.NewClient()
|
|
if err != nil {
|
|
log.Fatalf("无法连接 Docker: %v", err)
|
|
}
|
|
defer dockerClient.Close()
|
|
|
|
// 设置 Gin 模式
|
|
if os.Getenv("GIN_MODE") == "" {
|
|
gin.SetMode(gin.ReleaseMode)
|
|
}
|
|
|
|
authConfig := api.LoadAgentAuthConfig()
|
|
api.LogAgentSecurityConfig(authConfig)
|
|
|
|
r := gin.New()
|
|
r.Use(gin.Recovery())
|
|
r.Use(gin.Logger())
|
|
r.Use(api.AgentSecurityMiddleware(authConfig))
|
|
|
|
// 禁用自动重定向
|
|
r.RedirectTrailingSlash = false
|
|
r.RedirectFixedPath = false
|
|
|
|
// API 路由
|
|
apiGroup := r.Group("/api")
|
|
{
|
|
// 系统信息
|
|
apiGroup.GET("/info", api.GetSystemInfo(dockerClient))
|
|
|
|
// 安装相关
|
|
apiGroup.POST("/install", api.Install(dockerClient))
|
|
apiGroup.GET("/install/status", api.GetInstallStatus(dockerClient))
|
|
|
|
// 更新相关
|
|
apiGroup.POST("/update", api.Update(dockerClient))
|
|
apiGroup.GET("/update/status", api.GetUpdateStatus(dockerClient))
|
|
apiGroup.GET("/versions", api.GetVersions(dockerClient))
|
|
|
|
// 付费模块运行
|
|
apiGroup.POST("/modules/sync", api.SyncModule(dockerClient))
|
|
apiGroup.GET("/modules/status", api.GetModuleStatuses())
|
|
|
|
// 容器管理
|
|
apiGroup.GET("/containers", api.ListContainers(dockerClient))
|
|
apiGroup.POST("/containers/:name/restart", api.RestartContainer(dockerClient))
|
|
apiGroup.POST("/containers/:name/stop", api.StopContainer(dockerClient))
|
|
apiGroup.POST("/containers/:name/start", api.StartContainer(dockerClient))
|
|
|
|
// 健康检查
|
|
apiGroup.GET("/health", api.HealthCheck(dockerClient))
|
|
}
|
|
|
|
// 静态文件服务
|
|
webDist, err := fs.Sub(webFS, "web/dist")
|
|
if err != nil {
|
|
log.Fatalf("无法加载静态文件: %v", err)
|
|
}
|
|
|
|
// 读取 index.html 内容
|
|
indexHTML, err := fs.ReadFile(webDist, "index.html")
|
|
if err != nil {
|
|
log.Fatalf("无法读取 index.html: %v", err)
|
|
}
|
|
|
|
// 根路径直接返回 index.html 内容
|
|
r.GET("/", func(c *gin.Context) {
|
|
c.Data(http.StatusOK, "text/html; charset=utf-8", indexHTML)
|
|
})
|
|
|
|
// 静态资源
|
|
r.StaticFS("/static", http.FS(webDist))
|
|
|
|
port := os.Getenv("PORT")
|
|
if port == "" {
|
|
port = "9090"
|
|
}
|
|
|
|
log.Printf("FLNXI Agent 启动在端口 %s", port)
|
|
if err := r.Run(":" + port); err != nil {
|
|
log.Fatalf("启动失败: %v", err)
|
|
}
|
|
}
|