250 lines
6.4 KiB
Go
250 lines
6.4 KiB
Go
/**
|
||
* package api
|
||
* 作用：AI 机器人相关 HTTP 接口。
|
||
*
|
||
* 权限模型：
|
||
* - AI 配置读写、机器人增删改：仅管理员（用户 id=1）；
|
||
* - 机器人列表查询：所有登录用户（好友列表要展示机器人）。
|
||
*/
|
||
package api
|
||
|
||
import (
|
||
"context"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"xk-websocket-v2/internal/service"
|
||
"xk-websocket-v2/internal/utils"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
/**
|
||
* requireAdmin
|
||
* 作用：校验当前登录用户是否为管理员（id=1）。
|
||
* 返回 false 时已写入 403 响应，调用方直接 return 即可
|
||
*/
|
||
func requireAdmin(c *gin.Context) bool {
|
||
userID, exists := c.Get("user_id")
|
||
if !exists || userID.(string) != service.AdminUserID {
|
||
utils.Error(c, utils.CodeForbidden, "仅管理员可操作")
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
/**
|
||
* maskAPIKey
|
||
* 作用：API 密钥脱敏（保留前4后4位），配置回显时避免明文泄露
|
||
*/
|
||
func maskAPIKey(key string) string {
|
||
if key == "" {
|
||
return ""
|
||
}
|
||
if len(key) <= 8 {
|
||
return "********"
|
||
}
|
||
return key[:4] + strings.Repeat("*", 8) + key[len(key)-4:]
|
||
}
|
||
|
||
/**
|
||
* GetAIConfigHandler
|
||
* 功能：获取全局 AI 配置（密钥脱敏返回）
|
||
* 路径：GET /api/ai/config （管理员）
|
||
*/
|
||
func GetAIConfigHandler(c *gin.Context) {
|
||
if !requireAdmin(c) {
|
||
return
|
||
}
|
||
cfg, err := service.AIBotSvc.GetAIConfig()
|
||
if err != nil {
|
||
utils.InternalError(c, "配置读取失败")
|
||
return
|
||
}
|
||
utils.SuccessWithData(c, gin.H{
|
||
"provider": cfg.Provider,
|
||
"base_url": cfg.BaseURL,
|
||
"api_key": maskAPIKey(cfg.APIKey),
|
||
"model": cfg.Model,
|
||
"enabled": cfg.Enabled,
|
||
// 前端据此判断"密钥是否已配置过"（脱敏值不能用于判断）
|
||
"has_key": cfg.APIKey != "",
|
||
}, "获取成功")
|
||
}
|
||
|
||
// saveAIConfigReq 保存 AI 配置请求体
|
||
type saveAIConfigReq struct {
|
||
Provider string `json:"provider" binding:"required"`
|
||
BaseURL string `json:"base_url"`
|
||
// 为空表示不修改已保存的密钥（前端回显的是脱敏值）
|
||
APIKey string `json:"api_key"`
|
||
Model string `json:"model" binding:"required"`
|
||
Enabled bool `json:"enabled"`
|
||
}
|
||
|
||
/**
|
||
* SaveAIConfigHandler
|
||
* 功能：保存全局 AI 配置
|
||
* 路径：POST /api/ai/config （管理员）
|
||
*/
|
||
func SaveAIConfigHandler(c *gin.Context) {
|
||
if !requireAdmin(c) {
|
||
return
|
||
}
|
||
var req saveAIConfigReq
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
utils.BadRequest(c, "参数错误: "+err.Error())
|
||
return
|
||
}
|
||
// 前端可能把脱敏值原样提交回来，含 **** 的一律视为"未修改"
|
||
if strings.Contains(req.APIKey, "****") {
|
||
req.APIKey = ""
|
||
}
|
||
cfg, err := service.AIBotSvc.SaveAIConfig(req.Provider, req.BaseURL, req.APIKey, req.Model, req.Enabled)
|
||
if err != nil {
|
||
utils.InternalError(c, "保存失败: "+err.Error())
|
||
return
|
||
}
|
||
utils.SuccessWithData(c, gin.H{
|
||
"provider": cfg.Provider,
|
||
"base_url": cfg.BaseURL,
|
||
"api_key": maskAPIKey(cfg.APIKey),
|
||
"model": cfg.Model,
|
||
"enabled": cfg.Enabled,
|
||
"has_key": cfg.APIKey != "",
|
||
}, "保存成功")
|
||
}
|
||
|
||
/**
|
||
* TestAIConfigHandler
|
||
* 功能：测试当前 AI 配置连通性（发一条固定问候语，成功返回模型回复）
|
||
* 路径：POST /api/ai/config/test （管理员）
|
||
*/
|
||
func TestAIConfigHandler(c *gin.Context) {
|
||
if !requireAdmin(c) {
|
||
return
|
||
}
|
||
cfg, err := service.AIBotSvc.GetAIConfig()
|
||
if err != nil {
|
||
utils.InternalError(c, "配置读取失败")
|
||
return
|
||
}
|
||
provider, err := service.NewAIProvider(cfg)
|
||
if err != nil {
|
||
utils.BadRequest(c, err.Error())
|
||
return
|
||
}
|
||
ctx, cancel := context.WithTimeout(c.Request.Context(), 30*time.Second)
|
||
defer cancel()
|
||
reply, err := provider.Chat(ctx, []service.AIMessage{
|
||
{Role: "user", Content: "你好，请用一句话介绍你自己"},
|
||
})
|
||
if err != nil {
|
||
utils.BadRequest(c, "测试失败: "+err.Error())
|
||
return
|
||
}
|
||
utils.SuccessWithData(c, gin.H{"reply": reply}, "测试成功")
|
||
}
|
||
|
||
/**
|
||
* ListAIBotsHandler
|
||
* 功能：机器人列表。管理员传 all=1 返回全部（含停用），普通用户只见启用中的
|
||
* 路径：GET /api/ai/bots （登录用户）
|
||
*/
|
||
func ListAIBotsHandler(c *gin.Context) {
|
||
userID, _ := c.Get("user_id")
|
||
onlyEnabled := true
|
||
if c.Query("all") == "1" && userID != nil && userID.(string) == service.AdminUserID {
|
||
onlyEnabled = false
|
||
}
|
||
bots, err := service.AIBotSvc.ListBots(onlyEnabled)
|
||
if err != nil {
|
||
utils.InternalError(c, "查询失败")
|
||
return
|
||
}
|
||
utils.SuccessWithData(c, bots, "获取成功")
|
||
}
|
||
|
||
// botReq 创建/更新机器人请求体
|
||
type botReq struct {
|
||
Name string `json:"name" binding:"required"`
|
||
Avatar string `json:"avatar"`
|
||
RolePrompt string `json:"role_prompt"`
|
||
Enabled *bool `json:"enabled"`
|
||
}
|
||
|
||
/**
|
||
* CreateAIBotHandler
|
||
* 功能：创建机器人（同时生成 users 虚拟用户）
|
||
* 路径：POST /api/ai/bots （管理员）
|
||
*/
|
||
func CreateAIBotHandler(c *gin.Context) {
|
||
if !requireAdmin(c) {
|
||
return
|
||
}
|
||
var req botReq
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
utils.BadRequest(c, "参数错误: "+err.Error())
|
||
return
|
||
}
|
||
bot, err := service.AIBotSvc.CreateBot(req.Name, req.Avatar, req.RolePrompt)
|
||
if err != nil {
|
||
utils.InternalError(c, "创建失败: "+err.Error())
|
||
return
|
||
}
|
||
utils.SuccessWithData(c, bot, "创建成功")
|
||
}
|
||
|
||
/**
|
||
* UpdateAIBotHandler
|
||
* 功能：更新机器人（名称/头像/角色设定/启停）
|
||
* 路径：POST /api/ai/bots/update/:id （管理员）
|
||
*/
|
||
func UpdateAIBotHandler(c *gin.Context) {
|
||
if !requireAdmin(c) {
|
||
return
|
||
}
|
||
id, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
||
if err != nil || id == 0 {
|
||
utils.BadRequest(c, "机器人ID不合法")
|
||
return
|
||
}
|
||
var req botReq
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
utils.BadRequest(c, "参数错误: "+err.Error())
|
||
return
|
||
}
|
||
enabled := true
|
||
if req.Enabled != nil {
|
||
enabled = *req.Enabled
|
||
}
|
||
bot, err := service.AIBotSvc.UpdateBot(uint(id), req.Name, req.Avatar, req.RolePrompt, enabled)
|
||
if err != nil {
|
||
utils.InternalError(c, "更新失败: "+err.Error())
|
||
return
|
||
}
|
||
utils.SuccessWithData(c, bot, "更新成功")
|
||
}
|
||
|
||
/**
|
||
* DeleteAIBotHandler
|
||
* 功能：删除机器人（连带删除虚拟用户与群成员关系）
|
||
* 路径：POST /api/ai/bots/delete/:id （管理员）
|
||
*/
|
||
func DeleteAIBotHandler(c *gin.Context) {
|
||
if !requireAdmin(c) {
|
||
return
|
||
}
|
||
id, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
||
if err != nil || id == 0 {
|
||
utils.BadRequest(c, "机器人ID不合法")
|
||
return
|
||
}
|
||
if err := service.AIBotSvc.DeleteBot(uint(id)); err != nil {
|
||
utils.InternalError(c, "删除失败: "+err.Error())
|
||
return
|
||
}
|
||
utils.Success(c, "删除成功")
|
||
}
|