辣评前后端服务统一与优化（二十二）

// cmd/server/main.go
package main

import (
    "log"
    "net/http"
    "path/filepath"

    "github.com/gin-gonic/gin"
    "github.com/yourusername/laping/cmd/server/config"
    "github.com/yourusername/laping/cmd/server/database"
    "github.com/yourusername/laping/cmd/server/router"
)

func main() {
    // 1. 加载配置
    cfg := config.LoadConfig()

    // 2. 初始化数据库
    if err := database.Initialize(); err != nil {
        log.Fatalf("数据库初始化失败: %v", err)
    }
    defer database.CloseDB()

    // 3. 设置 Gin 模式
    if cfg.Environment == "production" {
        gin.SetMode(gin.ReleaseMode)
    }

    // 4. 创建路由
    r := router.SetupRouter()

    // 5. 配置静态文件服务
    setupStaticFiles(r, cfg)

    // 6. 启动服务器
    addr := fmt.Sprintf(":%d", cfg.APIPort)
    log.Printf("服务器启动在端口 %d", cfg.APIPort)
    log.Printf("访问地址: http://localhost:%d", cfg.APIPort)

    if err := r.Run(addr); err != nil {
        log.Fatalf("服务器启动失败: %v", err)
    }
}

// setupStaticFiles 配置静态文件服务
func setupStaticFiles(r *gin.Engine, cfg *config.Config) {
    // 前端构建产物路径
    frontendPath := cfg.FrontendPath
    if frontendPath == "" {
        frontendPath = "./frontend"
    }

    // 检查前端目录是否存在
    if _, err := os.Stat(frontendPath); os.IsNotExist(err) {
        log.Printf("警告: 前端目录不存在: %s", frontendPath)
        return
    }

    // 静态资源路由（CSS、JS、图片等）
    r.Static("/assets", filepath.Join(frontendPath, "assets"))
    r.StaticFile("/favicon.ico", filepath.Join(frontendPath, "favicon.ico"))

    // SPA 路由处理
    r.NoRoute(func(c *gin.Context) {
        path := c.Request.URL.Path

        // API 路由返回 404
        if strings.HasPrefix(path, "/api/") {
            c.JSON(404, gin.H{
                "code":    404,
                "message": "接口不存在",
            })
            return
        }

        // 其他路由返回 index.html（SPA）
        c.File(filepath.Join(frontendPath, "index.html"))
    })

    log.Printf("静态文件服务已配置: %s", frontendPath)
}

// cmd/server/router/router.go
package router

import (
    "github.com/gin-gonic/gin"
    "github.com/yourusername/laping/cmd/server/handlers"
    "github.com/yourusername/laping/cmd/server/middleware"
)

func SetupRouter() *gin.Engine {
    r := gin.Default()

    // 中间件
    r.Use(middleware.CORS())
    r.Use(middleware.Logger())
    r.Use(middleware.Recovery())

    // API 路由组
    api := r.Group("/api")
    {
        // 公开路由
        public := api.Group("")
        {
            public.POST("/auth/login", handlers.Login)
            public.POST("/auth/register", handlers.Register)
            public.POST("/auth/refresh", handlers.RefreshToken)
        }

        // 认证路由
        auth := api.Group("")
        auth.Use(middleware.JWTAuth())
        {
            auth.GET("/user/profile", handlers.GetProfile)
            auth.PUT("/user/profile", handlers.UpdateProfile)

            auth.GET("/submissions", handlers.ListSubmissions)
            auth.POST("/submissions", handlers.CreateSubmission)
            auth.GET("/submissions/:id", handlers.GetSubmission)
            auth.PUT("/submissions/:id", handlers.UpdateSubmission)
            auth.DELETE("/submissions/:id", handlers.DeleteSubmission)

            auth.GET("/comments", handlers.ListComments)
            auth.POST("/comments", handlers.CreateComment)
            auth.PUT("/comments/:id", handlers.UpdateComment)
            auth.DELETE("/comments/:id", handlers.DeleteComment)
        }

        // 管理员路由
        admin := api.Group("/admin")
        admin.Use(middleware.JWTAuth(), middleware.RequireRole("admin"))
        {
            admin.GET("/users", handlers.ListUsers)
            admin.GET("/users/:id", handlers.GetUser)
            admin.PUT("/users/:id", handlers.UpdateUser)
            admin.POST("/users/:id/ban", handlers.BanUser)

            admin.GET("/dashboard/stats", handlers.GetDashboardStats)
            admin.GET("/statistics", handlers.GetStatistics)

            admin.POST("/submissions/proxy", handlers.CreateSubmissionForUser)
            admin.GET("/logs", handlers.GetOperationLogs)
        }
    }

    return r
}

// cmd/server/middleware/cors.go
package middleware

import (
    "github.com/gin-gonic/gin"
)

func CORS() gin.HandlerFunc {
    return func(c *gin.Context) {
        method := c.Request.Method
        origin := c.Request.Header.Get("Origin")

        // 允许的源
        if origin != "" {
            c.Header("Access-Control-Allow-Origin", origin)
            c.Header("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
            c.Header("Access-Control-Allow-Headers", "Content-Type, Authorization, X-Requested-With")
            c.Header("Access-Control-Expose-Headers", "Content-Length, Content-Type")
            c.Header("Access-Control-Allow-Credentials", "true")
        }

        // 处理 OPTIONS 请求
        if method == "OPTIONS" {
            c.AbortWithStatus(204)
            return
        }

        c.Next()
    }
}