feat: 添加登录功能和路由注册

internal/handler/handler.go

@@ -26,44 +26,49 @@ func NewServer(db *gorm.DB, config types.ConfigManager) *Server {
 	}
 }
 
-// RegisterAPIRoutes registers all API routes under a given router group
+// LoginRequest represents the login request payload
-func (s *Server) RegisterAPIRoutes(api *gin.RouterGroup) {
+type LoginRequest struct {
-	// Group management routes
+	AuthKey string `json:"auth_key" binding:"required"`
-	groups := api.Group("/groups")
+}
-	{
-		groups.POST("", s.CreateGroup)
-		groups.GET("", s.ListGroups)
-		groups.GET("/:id", s.GetGroup)
-		groups.PUT("/:id", s.UpdateGroup)
-		groups.DELETE("/:id", s.DeleteGroup)
 
-		// Key management routes within a group
+// LoginResponse represents the login response
-		keys := groups.Group("/:id/keys")
+type LoginResponse struct {
-		{
+	Success bool   `json:"success"`
-			keys.POST("", s.CreateKeysInGroup)
+	Message string `json:"message"`
-			keys.GET("", s.ListKeysInGroup)
+}
-			keys.PUT("/:key_id", s.UpdateKey)
+
-			keys.DELETE("", s.DeleteKeys)
+// Login handles authentication verification
-		}
+func (s *Server) Login(c *gin.Context) {
+	var req LoginRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"success": false,
+			"message": "Invalid request format",
+		})
+		return
 	}
 
-	// Dashboard and logs routes
+	authConfig := s.config.GetAuthConfig()
-	dashboard := api.Group("/dashboard")
+	
-	{
+	if !authConfig.Enabled {
-		dashboard.GET("/stats", s.Stats)
+		c.JSON(http.StatusOK, LoginResponse{
+			Success: true,
+			Message: "Authentication disabled",
+		})
+		return
 	}
 
-	api.GET("/logs", GetLogs)
+	if req.AuthKey == authConfig.Key {
-
+		c.JSON(http.StatusOK, LoginResponse{
-	// Settings routes
+			Success: true,
-	settings := api.Group("/settings")
+			Message: "Authentication successful",
-	{
+		})
-		settings.GET("", GetSettings)
+	} else {
-		settings.PUT("", UpdateSettings)
+		c.JSON(http.StatusUnauthorized, LoginResponse{
+			Success: false,
+			Message: "Invalid authentication key",
+		})
 	}
-
-	// Reload route
-	api.POST("/reload", s.ReloadConfig)
 }
 
 // Health handles health check requests

internal/proxy/server.go

@@ -32,11 +32,6 @@ func NewProxyServer(db *gorm.DB, requestLogChan chan models.RequestLog) (*ProxyS
 	}, nil
 }
 
-// RegisterProxyRoutes registers the main proxy route under a given router group
-func (ps *ProxyServer) RegisterProxyRoutes(proxy *gin.RouterGroup) {
-	proxy.Any("/:group_name/*path", ps.HandleProxy)
-}
-
 // HandleProxy handles the main proxy logic
 func (ps *ProxyServer) HandleProxy(c *gin.Context) {
 	startTime := time.Now()

internal/router/router.go

@@ -13,14 +13,12 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
-// New 创建并配置一个完整的 gin.Engine 实例
 func New(
 	serverHandler *handler.Server,
 	proxyServer *proxy.ProxyServer,
 	configManager types.ConfigManager,
 	webUI fs.FS,
 ) *gin.Engine {
-	// 设置 Gin 模式
 	gin.SetMode(gin.ReleaseMode)
 
 	router := gin.New()
@@ -31,43 +29,108 @@ func New(
 	router.Use(middleware.Logger(configManager.GetLogConfig()))
 	router.Use(middleware.CORS(configManager.GetCORSConfig()))
 	router.Use(middleware.RateLimiter(configManager.GetPerformanceConfig()))
-
-	// 添加服务器启动时间中间件
-	startTime := time.Now()
 	router.Use(func(c *gin.Context) {
-		c.Set("serverStartTime", startTime)
+		c.Set("serverStartTime", time.Now())
 		c.Next()
 	})
 
-	// 注册 Web UI 和通用端点
+	// 注册路由
+	registerSystemRoutes(router, serverHandler)
+	registerAPIRoutes(router, serverHandler, configManager)
+	registerProxyRoutes(router, proxyServer, configManager)
+	registerFrontendRoutes(router, webUI)
+
+	return router
+}
+
+// registerSystemRoutes 注册系统级路由
+func registerSystemRoutes(router *gin.Engine, serverHandler *handler.Server) {
 	router.GET("/health", serverHandler.Health)
 	router.GET("/stats", serverHandler.Stats)
-	router.GET("/config", serverHandler.GetConfig) // Debug endpoint
+	// router.GET("/config", serverHandler.GetConfig)
+}
 
-	// 注册管理 API 路由
+// registerAPIRoutes 注册API路由
+func registerAPIRoutes(router *gin.Engine, serverHandler *handler.Server, configManager types.ConfigManager) {
 	api := router.Group("/api")
 	authConfig := configManager.GetAuthConfig()
-	if authConfig.Enabled {
-		api.Use(middleware.Auth(authConfig))
-	}
-	serverHandler.RegisterAPIRoutes(api)
 
-	// 注册代理路由
+	// 公开
+	registerPublicAPIRoutes(api, serverHandler)
+
+	// 认证
+	if authConfig.Enabled {
+		protectedAPI := api.Group("")
+		protectedAPI.Use(middleware.Auth(authConfig))
+		registerProtectedAPIRoutes(protectedAPI, serverHandler)
+	} else {
+		registerProtectedAPIRoutes(api, serverHandler)
+	}
+}
+
+// registerPublicAPIRoutes 公开API路由
+func registerPublicAPIRoutes(api *gin.RouterGroup, serverHandler *handler.Server) {
+	api.POST("/auth/login", serverHandler.Login)
+}
+
+// registerProtectedAPIRoutes 认证API路由
+func registerProtectedAPIRoutes(api *gin.RouterGroup, serverHandler *handler.Server) {
+	groups := api.Group("/groups")
+	{
+		groups.POST("", serverHandler.CreateGroup)
+		groups.GET("", serverHandler.ListGroups)
+		groups.GET("/:id", serverHandler.GetGroup)
+		groups.PUT("/:id", serverHandler.UpdateGroup)
+		groups.DELETE("/:id", serverHandler.DeleteGroup)
+
+		keys := groups.Group("/:id/keys")
+		{
+			keys.POST("", serverHandler.CreateKeysInGroup)
+			keys.GET("", serverHandler.ListKeysInGroup)
+			keys.PUT("/:key_id", serverHandler.UpdateKey)
+			keys.DELETE("", serverHandler.DeleteKeys)
+		}
+	}
+
+	// 仪表板和日志
+	dashboard := api.Group("/dashboard")
+	{
+		dashboard.GET("/stats", serverHandler.Stats)
+	}
+
+	// 日志
+	api.GET("/logs", handler.GetLogs)
+
+	// 设置
+	settings := api.Group("/settings")
+	{
+		settings.GET("", handler.GetSettings)
+		settings.PUT("", handler.UpdateSettings)
+	}
+
+	// 重载配置
+	api.POST("/reload", serverHandler.ReloadConfig)
+}
+
+// registerProxyRoutes 注册代理路由
+func registerProxyRoutes(router *gin.Engine, proxyServer *proxy.ProxyServer, configManager types.ConfigManager) {
 	proxyGroup := router.Group("/proxy")
+	authConfig := configManager.GetAuthConfig()
+
 	if authConfig.Enabled {
 		proxyGroup.Use(middleware.Auth(authConfig))
 	}
-	proxyServer.RegisterProxyRoutes(proxyGroup)
 
-	// 处理 405 Method Not Allowed
+	proxyGroup.Any("/:group_name/*path", proxyServer.HandleProxy)
+}
+
+// registerFrontendRoutes 注册前端路由
+func registerFrontendRoutes(router *gin.Engine, webUI fs.FS) {
 	router.NoMethod(func(c *gin.Context) {
 		c.JSON(http.StatusMethodNotAllowed, gin.H{"error": "Method not allowed"})
 	})
 
-	// 其他所有路由都交给前端 UI 处理
 	router.NoRoute(ServeUI(webUI))
-
-	return router
 }
 
 // ServeUI 返回一个 gin.HandlerFunc 来服务嵌入式前端 UI
@@ -75,10 +138,7 @@ func ServeUI(webUI fs.FS) gin.HandlerFunc {
 	fileServer := http.FileServer(http.FS(webUI))
 
 	return func(c *gin.Context) {
-		// 检查文件是否存在于嵌入的文件系统中
 		if _, err := webUI.Open(strings.TrimPrefix(c.Request.URL.Path, "/")); err != nil {
-			// 如果文件不存在，并且不是API或代理请求，则将请求重写为 /
-			// 这将提供 index.html，以支持 SPA 的前端路由
 			if !strings.HasPrefix(c.Request.URL.Path, "/api/") && !strings.HasPrefix(c.Request.URL.Path, "/proxy/") {
 				c.Request.URL.Path = "/"
 			}