package api import ( "net/http" "strconv" "github.com/gin-gonic/gin" "oci-portal/internal/service" ) // aiAdminHandler 处理面板侧 AI 网关管理接口(secured 组,自动进系统日志)。 type aiAdminHandler struct { gw *service.AiGatewayService } // ---- 密钥 ---- func (h *aiAdminHandler) listKeys(c *gin.Context) { keys, err := h.gw.Keys(c.Request.Context()) if err != nil { respondError(c, err) return } c.JSON(http.StatusOK, gin.H{"items": keys}) } // createKey 生成密钥;明文仅在本响应返回一次。 func (h *aiAdminHandler) createKey(c *gin.Context) { var req struct { Name string `json:"name" binding:"required"` Value string `json:"value"` Group string `json:"group"` } if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } plaintext, key, err := h.gw.CreateKey(c.Request.Context(), req.Name, req.Value, req.Group) if err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } c.JSON(http.StatusCreated, gin.H{"key": plaintext, "item": key}) } func (h *aiAdminHandler) updateKey(c *gin.Context) { id, ok := aiPathID(c) if !ok { return } var req struct { Name string `json:"name"` Enabled *bool `json:"enabled"` Group *string `json:"group"` } if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } if err := h.gw.UpdateKey(c.Request.Context(), id, req.Name, req.Enabled, req.Group); err != nil { respondError(c, err) return } c.Status(http.StatusNoContent) } func (h *aiAdminHandler) deleteKey(c *gin.Context) { id, ok := aiPathID(c) if !ok { return } if err := h.gw.DeleteKey(c.Request.Context(), id); err != nil { respondError(c, err) return } c.Status(http.StatusNoContent) } // updateKeyContentLog 设置密钥内容日志窗口(红线例外:显式限时开启,0 关闭)。 func (h *aiAdminHandler) updateKeyContentLog(c *gin.Context) { id, ok := aiPathID(c) if !ok { return } var req struct { Hours int `json:"hours"` } if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } key, err := h.gw.UpdateKeyContentLog(c.Request.Context(), id, req.Hours) if err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } c.JSON(http.StatusOK, key) } // listContentLogs 分页查询内容日志(可选 keyId / callLogId 过滤)。 func (h *aiAdminHandler) listContentLogs(c *gin.Context) { page, _ := strconv.Atoi(c.DefaultQuery("page", "1")) size, _ := strconv.Atoi(c.DefaultQuery("size", "20")) keyID, _ := strconv.Atoi(c.DefaultQuery("keyId", "0")) callLogID, _ := strconv.Atoi(c.DefaultQuery("callLogId", "0")) items, total, err := h.gw.ContentLogs(c.Request.Context(), uint(keyID), uint(callLogID), page, size) if err != nil { respondError(c, err) return } c.JSON(http.StatusOK, gin.H{"items": items, "total": total}) } // ---- 渠道 ---- func (h *aiAdminHandler) listChannels(c *gin.Context) { chs, err := h.gw.Channels(c.Request.Context()) if err != nil { respondError(c, err) return } c.JSON(http.StatusOK, gin.H{"items": chs}) } func (h *aiAdminHandler) createChannel(c *gin.Context) { var req service.ChannelInput if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } ch, err := h.gw.CreateChannel(c.Request.Context(), req) if err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } c.JSON(http.StatusCreated, ch) } func (h *aiAdminHandler) updateChannel(c *gin.Context) { id, ok := aiPathID(c) if !ok { return } var req service.ChannelInput if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } if err := h.gw.UpdateChannel(c.Request.Context(), id, req); err != nil { respondError(c, err) return } c.Status(http.StatusNoContent) } func (h *aiAdminHandler) deleteChannel(c *gin.Context) { id, ok := aiPathID(c) if !ok { return } if err := h.gw.DeleteChannel(c.Request.Context(), id); err != nil { respondError(c, err) return } c.Status(http.StatusNoContent) } // probeChannel 触发探测:服务可见性 + 模型同步 + 配额试调,返回更新后的渠道。 func (h *aiAdminHandler) probeChannel(c *gin.Context) { id, ok := aiPathID(c) if !ok { return } ch, err := h.gw.ProbeChannel(c.Request.Context(), id) if err != nil { respondError(c, err) return } c.JSON(http.StatusOK, ch) } func (h *aiAdminHandler) syncChannelModels(c *gin.Context) { id, ok := aiPathID(c) if !ok { return } models, err := h.gw.SyncModels(c.Request.Context(), id) if err != nil { c.JSON(http.StatusBadGateway, gin.H{"error": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"items": models}) } // ---- 聚合模型与调用日志 ---- func (h *aiAdminHandler) gatewayModels(c *gin.Context) { list, err := h.gw.GatewayModels(c.Request.Context(), "") if err != nil { respondError(c, err) return } c.JSON(http.StatusOK, gin.H{"items": list.Data}) } func (h *aiAdminHandler) listLogs(c *gin.Context) { page, _ := strconv.Atoi(c.DefaultQuery("page", "1")) size, _ := strconv.Atoi(c.DefaultQuery("size", "50")) items, total, err := h.gw.CallLogs(c.Request.Context(), page, size) if err != nil { respondError(c, err) return } c.JSON(http.StatusOK, gin.H{"items": items, "total": total}) } func aiPathID(c *gin.Context) (uint, bool) { id, err := strconv.ParseUint(c.Param("id"), 10, 64) if err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": "invalid id"}) return 0, false } return uint(id), true }