🎨 优化项目结构 && 完善ai配置

server/api/v1/app/ai_preset.go

@@ -1,11 +1,9 @@
 package app
 
 import (
-	"strconv"
-
 	"git.echol.cn/loser/ai_proxy/server/global"
-	"git.echol.cn/loser/ai_proxy/server/model/app/request"
-	commonRequest "git.echol.cn/loser/ai_proxy/server/model/common/request"
+	"git.echol.cn/loser/ai_proxy/server/model/app"
+	"git.echol.cn/loser/ai_proxy/server/model/common/request"
 	"git.echol.cn/loser/ai_proxy/server/model/common/response"
 	"git.echol.cn/loser/ai_proxy/server/service"
 	"git.echol.cn/loser/ai_proxy/server/utils"
@@ -17,216 +15,163 @@ type AiPresetApi struct{}
 
 var aiPresetService = service.ServiceGroupApp.AppServiceGroup.AiPresetService
 
-// CreateAiPreset 创建AI预设
+// CreateAiPreset 创建预设
 // @Tags AiPreset
-// @Summary 创建AI预设
+// @Summary 创建预设
 // @Security ApiKeyAuth
 // @accept application/json
 // @Produce application/json
-// @Param data body request.CreateAiPresetRequest true "预设信息"
-// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "创建成功"
-// @Router /app/preset [post]
+// @Param data body app.AiPreset true "预设信息"
+// @Success 200 {object} response.Response{msg=string} "创建成功"
+// @Router /aiPreset/createAiPreset [post]
 func (a *AiPresetApi) CreateAiPreset(c *gin.Context) {
-	var req request.CreateAiPresetRequest
-	err := c.ShouldBindJSON(&req)
+	var preset app.AiPreset
+	err := c.ShouldBindJSON(&preset)
 	if err != nil {
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
+	preset.UserID = utils.GetUserID(c)
 
-	// 尝试获取用户ID，如果没有则使用0（公开访问）
-	userId := uint(0)
-	if id := utils.GetUserID(c); id > 0 {
-		userId = id
+	if err := aiPresetService.CreateAiPreset(&preset); err != nil {
+		global.GVA_LOG.Error("创建失败!", zap.Error(err))
+		response.FailWithMessage("创建失败", c)
+	} else {
+		response.OkWithMessage("创建成功", c)
 	}
-	preset, err := aiPresetService.CreateAiPreset(userId, &req)
-	if err != nil {
-		global.GVA_LOG.Error("创建预设失败!", zap.Error(err))
-		response.FailWithMessage("创建预设失败", c)
-		return
-	}
-
-	response.OkWithData(preset, c)
 }
 
-// DeleteAiPreset 删除AI预设
+// DeleteAiPreset 删除预设
 // @Tags AiPreset
-// @Summary 删除AI预设
+// @Summary 删除预设
 // @Security ApiKeyAuth
 // @accept application/json
 // @Produce application/json
-// @Param id path uint true "预设ID"
+// @Param data body request.GetById true "ID"
 // @Success 200 {object} response.Response{msg=string} "删除成功"
-// @Router /app/preset/:id [delete]
+// @Router /aiPreset/deleteAiPreset [delete]
 func (a *AiPresetApi) DeleteAiPreset(c *gin.Context) {
-	id, _ := strconv.ParseUint(c.Param("id"), 10, 32)
-	// 尝试获取用户ID，如果没有则使用0（公开访问）
-	userId := uint(0)
-	if uid := utils.GetUserID(c); uid > 0 {
-		userId = uid
-	}
-
-	err := aiPresetService.DeleteAiPreset(uint(id), userId)
-	if err != nil {
-		global.GVA_LOG.Error("删除预设失败!", zap.Error(err))
-		response.FailWithMessage("删除预设失败", c)
-		return
-	}
-
-	response.OkWithMessage("删除成功", c)
-}
-
-// UpdateAiPreset 更新AI预设
-// @Tags AiPreset
-// @Summary 更新AI预设
-// @Security ApiKeyAuth
-// @accept application/json
-// @Produce application/json
-// @Param data body request.UpdateAiPresetRequest true "预设信息"
-// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "更新成功"
-// @Router /app/preset [put]
-func (a *AiPresetApi) UpdateAiPreset(c *gin.Context) {
-	var req request.UpdateAiPresetRequest
-	err := c.ShouldBindJSON(&req)
+	var reqId request.GetById
+	err := c.ShouldBindJSON(&reqId)
 	if err != nil {
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
+	userID := utils.GetUserID(c)
 
-	// 尝试获取用户ID，如果没有则使用0（公开访问）
-	userId := uint(0)
-	if id := utils.GetUserID(c); id > 0 {
-		userId = id
+	if err := aiPresetService.DeleteAiPreset(reqId.Uint(), userID); err != nil {
+		global.GVA_LOG.Error("删除失败!", zap.Error(err))
+		response.FailWithMessage("删除失败", c)
+	} else {
+		response.OkWithMessage("删除成功", c)
 	}
-	preset, err := aiPresetService.UpdateAiPreset(userId, &req)
-	if err != nil {
-		global.GVA_LOG.Error("更新预设失败!", zap.Error(err))
-		response.FailWithMessage("更新预设失败", c)
-		return
-	}
-
-	response.OkWithData(preset, c)
 }
 
-// GetAiPreset 获取AI预设详情
+// UpdateAiPreset 更新预设
 // @Tags AiPreset
-// @Summary 获取AI预设详情
+// @Summary 更新预设
 // @Security ApiKeyAuth
 // @accept application/json
 // @Produce application/json
-// @Param id path uint true "预设ID"
-// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "获取成功"
-// @Router /app/preset/:id [get]
-func (a *AiPresetApi) GetAiPreset(c *gin.Context) {
-	id, _ := strconv.ParseUint(c.Param("id"), 10, 32)
-	// 尝试获取用户ID，如果没有则使用0（公开访问）
-	userId := uint(0)
-	if uid := utils.GetUserID(c); uid > 0 {
-		userId = uid
-	}
-
-	preset, err := aiPresetService.GetAiPreset(uint(id), userId)
+// @Param data body app.AiPreset true "预设信息"
+// @Success 200 {object} response.Response{msg=string} "更新成功"
+// @Router /aiPreset/updateAiPreset [put]
+func (a *AiPresetApi) UpdateAiPreset(c *gin.Context) {
+	var preset app.AiPreset
+	err := c.ShouldBindJSON(&preset)
 	if err != nil {
-		global.GVA_LOG.Error("获取预设失败!", zap.Error(err))
-		response.FailWithMessage("获取预设失败", c)
+		response.FailWithMessage(err.Error(), c)
 		return
 	}
+	userID := utils.GetUserID(c)
 
-	response.OkWithData(preset, c)
+	if err := aiPresetService.UpdateAiPreset(&preset, userID); err != nil {
+		global.GVA_LOG.Error("更新失败!", zap.Error(err))
+		response.FailWithMessage("更新失败", c)
+	} else {
+		response.OkWithMessage("更新成功", c)
+	}
 }
 
-// GetAiPresetList 获取AI预设列表
+// FindAiPreset 查询预设
 // @Tags AiPreset
-// @Summary 获取AI预设列表
+// @Summary 查询预设
 // @Security ApiKeyAuth
 // @accept application/json
 // @Produce application/json
-// @Param data query request.PageInfo true "分页信息"
+// @Param data query request.GetById true "ID"
+// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "查询成功"
+// @Router /aiPreset/findAiPreset [get]
+func (a *AiPresetApi) FindAiPreset(c *gin.Context) {
+	var reqId request.GetById
+	err := c.ShouldBindQuery(&reqId)
+	if err != nil {
+		response.FailWithMessage(err.Error(), c)
+		return
+	}
+	userID := utils.GetUserID(c)
+
+	if preset, err := aiPresetService.GetAiPreset(reqId.Uint(), userID); err != nil {
+		global.GVA_LOG.Error("查询失败!", zap.Error(err))
+		response.FailWithMessage("查询失败", c)
+	} else {
+		response.OkWithData(preset, c)
+	}
+}
+
+// GetAiPresetList 获取预设列表
+// @Tags AiPreset
+// @Summary 获取预设列表
+// @Security ApiKeyAuth
+// @accept application/json
+// @Produce application/json
+// @Param data query request.PageInfo true "分页参数"
 // @Success 200 {object} response.Response{data=response.PageResult,msg=string} "获取成功"
-// @Router /app/preset/list [get]
+// @Router /aiPreset/getAiPresetList [get]
 func (a *AiPresetApi) GetAiPresetList(c *gin.Context) {
-	var pageInfo commonRequest.PageInfo
+	var pageInfo request.PageInfo
 	err := c.ShouldBindQuery(&pageInfo)
 	if err != nil {
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
+	userID := utils.GetUserID(c)
 
-	// 尝试获取用户ID，如果没有则使用0（公开访问）
-	userId := uint(0)
-	if id := utils.GetUserID(c); id > 0 {
-		userId = id
+	if list, total, err := aiPresetService.GetAiPresetList(pageInfo, userID); err != nil {
+		global.GVA_LOG.Error("获取失败!", zap.Error(err))
+		response.FailWithMessage("获取失败", c)
+	} else {
+		response.OkWithDetailed(response.PageResult{
+			List:     list,
+			Total:    total,
+			Page:     pageInfo.Page,
+			PageSize: pageInfo.PageSize,
+		}, "获取成功", c)
 	}
-	list, total, err := aiPresetService.GetAiPresetList(userId, pageInfo)
-	if err != nil {
-		global.GVA_LOG.Error("获取预设列表失败!", zap.Error(err))
-		response.FailWithMessage("获取预设列表失败", c)
-		return
-	}
-
-	response.OkWithDetailed(response.PageResult{
-		List:     list,
-		Total:    total,
-		Page:     pageInfo.Page,
-		PageSize: pageInfo.PageSize,
-	}, "获取成功", c)
 }
 
-// ImportAiPreset 导入AI预设（支持SillyTavern格式）
+// ImportAiPreset 导入预设
 // @Tags AiPreset
-// @Summary 导入AI预设
+// @Summary 导入预设(支持SillyTavern格式)
 // @Security ApiKeyAuth
 // @accept application/json
 // @Produce application/json
-// @Param data body request.ImportAiPresetRequest true "导入数据"
-// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "导入成功"
-// @Router /app/preset/import [post]
+// @Param data body app.AiPreset true "预设JSON"
+// @Success 200 {object} response.Response{msg=string} "导入成功"
+// @Router /aiPreset/importAiPreset [post]
 func (a *AiPresetApi) ImportAiPreset(c *gin.Context) {
-	var req request.ImportAiPresetRequest
-	err := c.ShouldBindJSON(&req)
+	var preset app.AiPreset
+	err := c.ShouldBindJSON(&preset)
 	if err != nil {
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
+	preset.UserID = utils.GetUserID(c)
 
-	// 尝试获取用户ID，如果没有则使用0（公开访问）
-	userId := uint(0)
-	if id := utils.GetUserID(c); id > 0 {
-		userId = id
+	if err := aiPresetService.CreateAiPreset(&preset); err != nil {
+		global.GVA_LOG.Error("导入失败!", zap.Error(err))
+		response.FailWithMessage("导入失败", c)
+	} else {
+		response.OkWithMessage("导入成功", c)
 	}
-	preset, err := aiPresetService.ImportAiPreset(userId, &req)
-	if err != nil {
-		global.GVA_LOG.Error("导入预设失败!", zap.Error(err))
-		response.FailWithMessage("导入预设失败", c)
-		return
-	}
-
-	response.OkWithData(preset, c)
-}
-
-// ExportAiPreset 导出AI预设
-// @Tags AiPreset
-// @Summary 导出AI预设
-// @Security ApiKeyAuth
-// @accept application/json
-// @Produce application/json
-// @Param id path uint true "预设ID"
-// @Success 200 {object} map[string]interface{} "导出数据"
-// @Router /app/preset/:id/export [get]
-func (a *AiPresetApi) ExportAiPreset(c *gin.Context) {
-	id, _ := strconv.ParseUint(c.Param("id"), 10, 32)
-	// 尝试获取用户ID，如果没有则使用0（公开访问）
-	userId := uint(0)
-	if uid := utils.GetUserID(c); uid > 0 {
-		userId = uid
-	}
-
-	data, err := aiPresetService.ExportAiPreset(uint(id), userId)
-	if err != nil {
-		global.GVA_LOG.Error("导出预设失败!", zap.Error(err))
-		response.FailWithMessage("导出预设失败", c)
-		return
-	}
-
-	c.JSON(200, data)
 }