LokiStack/backend/routers/router.go

package routers

import (
	"net/http"
	"os"
	"path/filepath"
	"strings"
	"time"

	"lokideploy/controllers"
	"lokideploy/middleware"

	"github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
)

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

	// CORS
	cfg := cors.DefaultConfig()
	cfg.AllowAllOrigins = true
	cfg.AllowHeaders = []string{"Origin", "Content-Type", "Accept", "Authorization", "X-Requested-With"}
	cfg.AllowMethods = []string{"GET", "POST", "PUT", "DELETE", "OPTIONS", "PATCH"}
	r.Use(cors.New(cfg))

	// Controllers
	authCtrl := controllers.NewAuthController()
	clusterCtrl := controllers.NewClusterController()
	hostCtrl := controllers.NewHostController()
	storageCtrl := controllers.NewStorageController()
	deploymentCtrl := controllers.NewDeploymentController()
	collectorCtrl := controllers.NewCollectorController()

	// API v1
	api := r.Group("/api/v1")

	// 健康检查（公开访问，用于负载均衡和监控）
	api.GET("/health", func(c *gin.Context) {
		c.JSON(http.StatusOK, gin.H{
			"status":    "ok",
			"timestamp": time.Now().Unix(),
			"version":   "2.0.0",
		})
	})

	// Prometheus metrics 端点（公开访问，可被Prometheus抓取）
	api.GET("/metrics", func(c *gin.Context) {
		// 基础metrics输出（符合Prometheus文本格式）
		c.Header("Content-Type", "text/plain; version=0.0.4; charset=utf-8")
		c.String(http.StatusOK, "# LokiDeploy Metrics\n")
	})

	// Public
	api.POST("/auth/login", authCtrl.Login)
	api.POST("/auth/refresh", authCtrl.RefreshToken)

	// Auth required
	authorized := api.Group("")
	authorized.Use(middleware.JWTAuth())

	// Auth routes
	authorized.GET("/auth/user", authCtrl.GetUser)
	authorized.POST("/auth/password", authCtrl.ChangePassword)
	authorized.POST("/auth/register", middleware.AdminAuth(), authCtrl.Register)

	// Clusters
	authorized.GET("/clusters", clusterCtrl.List)
	authorized.GET("/clusters/:id", clusterCtrl.Get)
	authorized.POST("/clusters", clusterCtrl.Create)
	authorized.PUT("/clusters/:id", clusterCtrl.Update)
	authorized.DELETE("/clusters/:id", clusterCtrl.Delete)
	authorized.POST("/clusters/:id/validate", clusterCtrl.Validate)
	authorized.GET("/clusters/:id/info", clusterCtrl.GetInfo)

	// Hosts
	authorized.GET("/hosts", hostCtrl.List)
	authorized.GET("/hosts/:id", hostCtrl.Get)
	authorized.POST("/hosts", hostCtrl.Create)
	authorized.PUT("/hosts/:id", hostCtrl.Update)
	authorized.DELETE("/hosts/:id", hostCtrl.Delete)
	authorized.POST("/hosts/:id/validate", hostCtrl.Validate)
	authorized.GET("/hosts/:id/info", hostCtrl.GetInfo)
	authorized.POST("/hosts/:id/execute", hostCtrl.Execute)

	// Storage
	authorized.GET("/storages", storageCtrl.List)
	authorized.GET("/storages/:id", storageCtrl.Get)
	authorized.POST("/storages", storageCtrl.Create)
	authorized.PUT("/storages/:id", storageCtrl.Update)
	authorized.DELETE("/storages/:id", storageCtrl.Delete)
	authorized.GET("/storages/default", storageCtrl.GetDefault)

	// Deployments
	authorized.GET("/deployments", deploymentCtrl.List)
	authorized.GET("/deployments/:id", deploymentCtrl.Get)
	authorized.POST("/deployments", deploymentCtrl.Create)
	authorized.PUT("/deployments/:id", deploymentCtrl.Update)
	authorized.DELETE("/deployments/:id", deploymentCtrl.Delete)
	authorized.POST("/deployments/:id/deploy", deploymentCtrl.Deploy)
	authorized.POST("/deployments/:id/stop", deploymentCtrl.Stop)
	authorized.POST("/deployments/:id/restart", deploymentCtrl.Restart)
	authorized.GET("/deployments/:id/logs", deploymentCtrl.GetLogs)
	authorized.GET("/deployments/stats", deploymentCtrl.GetStats)

	// Collectors
	authorized.GET("/collectors", collectorCtrl.List)
	authorized.GET("/collectors/:id", collectorCtrl.Get)
	authorized.POST("/collectors", collectorCtrl.Create)
	authorized.PUT("/collectors/:id", collectorCtrl.Update)
	authorized.DELETE("/collectors/:id", collectorCtrl.Delete)
	authorized.POST("/collectors/:id/install", collectorCtrl.Install)
	authorized.POST("/collectors/:id/uninstall", collectorCtrl.Uninstall)

	// Static files - use executable directory as base
	distPath := findDistPath()

	if distPath != "" {
		r.Static("/assets", filepath.Join(distPath, "assets"))

		// favicon：NoRoute 会把所有非 /api/ 请求回退到 index.html，
		// 故此处显式注册 /favicon.svg 与 /favicon.ico，否则浏览器请求图标只会拿到 HTML。
		faviconSVG := filepath.Join(distPath, "favicon.svg")
		serveFavicon := func(c *gin.Context) {
			if _, err := os.Stat(faviconSVG); err != nil {
				c.Status(404)
				return
			}
			c.Header("Content-Type", "image/svg+xml")
			c.File(faviconSVG)
		}
		r.GET("/favicon.svg", serveFavicon)
		r.GET("/favicon.ico", serveFavicon)

		r.NoRoute(func(c *gin.Context) {
			if strings.HasPrefix(c.Request.URL.Path, "/api/") {
				c.JSON(404, gin.H{"code": 404, "message": "API endpoint not found"})
				return
			}
			c.File(filepath.Join(distPath, "index.html"))
		})
	}

	return r
}

// findDistPath finds the frontend dist directory
func findDistPath() string {
	// Try relative to executable
	if ex, err := os.Executable(); err == nil {
		execDir := filepath.Dir(ex)
		// If running via "go run", executable is in temp dir - fall through
		if !strings.Contains(execDir, "/tmp/go-build") {
			for _, rel := range []string{"dist", "../dist", "../../dist"} {
				p := filepath.Join(execDir, rel)
				if _, err := os.Stat(filepath.Join(p, "index.html")); err == nil {
					return p
				}
			}
		}
	}
	// Try working directory relative paths
	for _, p := range []string{"./dist", "../dist", "/app/dist"} {
		if abs, err := filepath.Abs(p); err == nil {
			if _, err := os.Stat(filepath.Join(abs, "index.html")); err == nil {
				return abs
			}
		}
	}
	return ""
}