feat: 前端搭建-未完成

internal/handler/key_handler.go

@@ -0,0 +1,119 @@
+// Package handler provides HTTP handlers for the application
+package handler
+
+import (
+	"gpt-load/internal/models"
+	"gpt-load/internal/response"
+	"net/http"
+	"strconv"
+
+	"github.com/gin-gonic/gin"
+)
+
+type CreateKeysRequest struct {
+	Keys []string `json:"keys" binding:"required"`
+}
+
+// CreateKeysInGroup handles creating new keys within a specific group.
+func (s *Server) CreateKeysInGroup(c *gin.Context) {
+	groupID, err := strconv.Atoi(c.Param("id"))
+	if err != nil {
+		response.Error(c, http.StatusBadRequest, "Invalid group ID")
+		return
+	}
+
+	var req CreateKeysRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.Error(c, http.StatusBadRequest, "Invalid request body")
+		return
+	}
+
+	var newKeys []models.APIKey
+	for _, keyVal := range req.Keys {
+		newKeys = append(newKeys, models.APIKey{
+			GroupID:  uint(groupID),
+			KeyValue: keyVal,
+			Status:   "active",
+		})
+	}
+
+	if err := s.DB.Create(&newKeys).Error; err != nil {
+		response.Error(c, http.StatusInternalServerError, "Failed to create keys")
+		return
+	}
+
+	response.Success(c, newKeys)
+}
+
+// ListKeysInGroup handles listing all keys within a specific group.
+func (s *Server) ListKeysInGroup(c *gin.Context) {
+	groupID, err := strconv.Atoi(c.Param("id"))
+	if err != nil {
+		response.Error(c, http.StatusBadRequest, "Invalid group ID")
+		return
+	}
+
+	var keys []models.APIKey
+	if err := s.DB.Where("group_id = ?", groupID).Find(&keys).Error; err != nil {
+		response.Error(c, http.StatusInternalServerError, "Failed to list keys")
+		return
+	}
+
+	response.Success(c, keys)
+}
+
+// UpdateKey handles updating a specific key.
+func (s *Server) UpdateKey(c *gin.Context) {
+	keyID, err := strconv.Atoi(c.Param("key_id"))
+	if err != nil {
+		response.Error(c, http.StatusBadRequest, "Invalid key ID")
+		return
+	}
+
+	var key models.APIKey
+	if err := s.DB.First(&key, keyID).Error; err != nil {
+		response.Error(c, http.StatusNotFound, "Key not found")
+		return
+	}
+
+	var updateData struct {
+		Status string `json:"status"`
+	}
+	if err := c.ShouldBindJSON(&updateData); err != nil {
+		response.Error(c, http.StatusBadRequest, "Invalid request body")
+		return
+	}
+
+	key.Status = updateData.Status
+	if err := s.DB.Save(&key).Error; err != nil {
+		response.Error(c, http.StatusInternalServerError, "Failed to update key")
+		return
+	}
+
+	response.Success(c, key)
+}
+
+type DeleteKeysRequest struct {
+	KeyIDs []uint `json:"key_ids" binding:"required"`
+}
+
+// DeleteKeys handles deleting one or more keys.
+func (s *Server) DeleteKeys(c *gin.Context) {
+	var req DeleteKeysRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		response.Error(c, http.StatusBadRequest, "Invalid request body")
+		return
+	}
+
+	if len(req.KeyIDs) == 0 {
+		response.Error(c, http.StatusBadRequest, "No key IDs provided")
+		return
+	}
+
+	if err := s.DB.Delete(&models.APIKey{}, req.KeyIDs).Error; err != nil {
+		response.Error(c, http.StatusInternalServerError, "Failed to delete keys")
+		return
+	}
+
+	response.Success(c, gin.H{"message": "Keys deleted successfully"})
+}