114 lines
3.3 KiB
Go
114 lines
3.3 KiB
Go
// Package handler provides HTTP handlers for the application
|
|
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"gpt-load/internal/config"
|
|
"gpt-load/internal/services"
|
|
"gpt-load/internal/types"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/dig"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Server contains dependencies for HTTP handlers
|
|
type Server struct {
|
|
DB *gorm.DB
|
|
config types.ConfigManager
|
|
SettingsManager *config.SystemSettingsManager
|
|
GroupManager *services.GroupManager
|
|
KeyManualValidationService *services.KeyManualValidationService
|
|
TaskService *services.TaskService
|
|
KeyService *services.KeyService
|
|
KeyImportService *services.KeyImportService
|
|
LogService *services.LogService
|
|
CommonHandler *CommonHandler
|
|
}
|
|
|
|
// NewServerParams defines the dependencies for the NewServer constructor.
|
|
type NewServerParams struct {
|
|
dig.In
|
|
DB *gorm.DB
|
|
Config types.ConfigManager
|
|
SettingsManager *config.SystemSettingsManager
|
|
GroupManager *services.GroupManager
|
|
KeyManualValidationService *services.KeyManualValidationService
|
|
TaskService *services.TaskService
|
|
KeyService *services.KeyService
|
|
KeyImportService *services.KeyImportService
|
|
LogService *services.LogService
|
|
CommonHandler *CommonHandler
|
|
}
|
|
|
|
// NewServer creates a new handler instance with dependencies injected by dig.
|
|
func NewServer(params NewServerParams) *Server {
|
|
return &Server{
|
|
DB: params.DB,
|
|
config: params.Config,
|
|
SettingsManager: params.SettingsManager,
|
|
GroupManager: params.GroupManager,
|
|
KeyManualValidationService: params.KeyManualValidationService,
|
|
TaskService: params.TaskService,
|
|
KeyService: params.KeyService,
|
|
KeyImportService: params.KeyImportService,
|
|
LogService: params.LogService,
|
|
CommonHandler: params.CommonHandler,
|
|
}
|
|
}
|
|
|
|
// LoginRequest represents the login request payload
|
|
type LoginRequest struct {
|
|
AuthKey string `json:"auth_key" binding:"required"`
|
|
}
|
|
|
|
// LoginResponse represents the login response
|
|
type LoginResponse struct {
|
|
Success bool `json:"success"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
authConfig := s.config.GetAuthConfig()
|
|
|
|
if req.AuthKey == authConfig.Key {
|
|
c.JSON(http.StatusOK, LoginResponse{
|
|
Success: true,
|
|
Message: "Authentication successful",
|
|
})
|
|
} else {
|
|
c.JSON(http.StatusUnauthorized, LoginResponse{
|
|
Success: false,
|
|
Message: "Invalid authentication key",
|
|
})
|
|
}
|
|
}
|
|
|
|
// Health handles health check requests
|
|
func (s *Server) Health(c *gin.Context) {
|
|
uptime := "unknown"
|
|
if startTime, exists := c.Get("serverStartTime"); exists {
|
|
if st, ok := startTime.(time.Time); ok {
|
|
uptime = time.Since(st).String()
|
|
}
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"status": "healthy",
|
|
"timestamp": time.Now().UTC().Format(time.RFC3339),
|
|
"uptime": uptime,
|
|
})
|
|
}
|