feat: Initial commit

internal/controllers/script_controller.go

@@ -0,0 +1,99 @@
+package controllers
+
+import (
+	"strconv"
+
+	"baihu/internal/services"
+	"baihu/internal/utils"
+
+	"github.com/gin-gonic/gin"
+)
+
+type ScriptController struct {
+	scriptService *services.ScriptService
+}
+
+func NewScriptController(scriptService *services.ScriptService) *ScriptController {
+	return &ScriptController{scriptService: scriptService}
+}
+
+func (sc *ScriptController) CreateScript(c *gin.Context) {
+	userID := 1
+
+	var req struct {
+		Name    string `json:"name" binding:"required"`
+		Content string `json:"content" binding:"required"`
+	}
+
+	if err := c.ShouldBindJSON(&req); err != nil {
+		utils.BadRequest(c, err.Error())
+		return
+	}
+
+	script := sc.scriptService.CreateScript(req.Name, req.Content, userID)
+	utils.Success(c, script)
+}
+
+func (sc *ScriptController) GetScripts(c *gin.Context) {
+	userID := 1
+	scripts := sc.scriptService.GetScriptsByUserID(userID)
+	utils.Success(c, scripts)
+}
+
+func (sc *ScriptController) GetScript(c *gin.Context) {
+	id, err := strconv.Atoi(c.Param("id"))
+	if err != nil {
+		utils.BadRequest(c, "无效的脚本ID")
+		return
+	}
+
+	script := sc.scriptService.GetScriptByID(id)
+	if script == nil {
+		utils.NotFound(c, "脚本不存在")
+		return
+	}
+
+	utils.Success(c, script)
+}
+
+func (sc *ScriptController) UpdateScript(c *gin.Context) {
+	id, err := strconv.Atoi(c.Param("id"))
+	if err != nil {
+		utils.BadRequest(c, "无效的脚本ID")
+		return
+	}
+
+	var req struct {
+		Name    string `json:"name"`
+		Content string `json:"content"`
+	}
+
+	if err := c.ShouldBindJSON(&req); err != nil {
+		utils.BadRequest(c, err.Error())
+		return
+	}
+
+	script := sc.scriptService.UpdateScript(id, req.Name, req.Content)
+	if script == nil {
+		utils.NotFound(c, "脚本不存在")
+		return
+	}
+
+	utils.Success(c, script)
+}
+
+func (sc *ScriptController) DeleteScript(c *gin.Context) {
+	id, err := strconv.Atoi(c.Param("id"))
+	if err != nil {
+		utils.BadRequest(c, "无效的脚本ID")
+		return
+	}
+
+	success := sc.scriptService.DeleteScript(id)
+	if !success {
+		utils.NotFound(c, "脚本不存在")
+		return
+	}
+
+	utils.SuccessMsg(c, "删除成功")
+}