138 lines
2.9 KiB
Go
138 lines
2.9 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"yingce/backend/internal/service"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// RegisterToolRoutes 注册画布工具 API：查询、收藏、新增、删除。
|
|
func RegisterToolRoutes(r *gin.RouterGroup, svc *service.Service) {
|
|
r.GET("/tools", func(c *gin.Context) {
|
|
user, err := currentUser(c, svc)
|
|
if err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
page, pageSize, err := parsePaginationQuery(c, 20)
|
|
if err != nil {
|
|
fail(c, http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
result, err := svc.Tools(user.ID, service.ToolListRequest{
|
|
Page: page,
|
|
PageSize: pageSize,
|
|
Scope: c.DefaultQuery("scope", "public"),
|
|
Type: c.Query("type"),
|
|
Tag: c.Query("tag"),
|
|
Search: c.Query("search"),
|
|
})
|
|
if err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
ok(c, result)
|
|
})
|
|
|
|
r.GET("/tools/:id", func(c *gin.Context) {
|
|
user, err := currentUser(c, svc)
|
|
if err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
toolID, err := parseToolID(c)
|
|
if err != nil {
|
|
fail(c, http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
item, err := svc.ToolDetail(user.ID, toolID)
|
|
if err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
ok(c, item)
|
|
})
|
|
|
|
r.POST("/tools", func(c *gin.Context) {
|
|
user, err := currentUser(c, svc)
|
|
if err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, 64<<10)
|
|
var req service.ToolMutationRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
failService(c, service.BadAuthRequest("请求体格式无效"))
|
|
return
|
|
}
|
|
item, err := svc.CreateTool(user.ID, req)
|
|
if err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
ok(c, item)
|
|
})
|
|
|
|
r.POST("/tools/:id/favorite", func(c *gin.Context) {
|
|
setToolFavorite(c, svc, true)
|
|
})
|
|
|
|
r.DELETE("/tools/:id/favorite", func(c *gin.Context) {
|
|
setToolFavorite(c, svc, false)
|
|
})
|
|
|
|
r.DELETE("/tools/:id", func(c *gin.Context) {
|
|
user, err := currentUser(c, svc)
|
|
if err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
toolID, err := parseToolID(c)
|
|
if err != nil {
|
|
fail(c, http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
if err := svc.DeleteTool(user.ID, toolID); err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
ok(c, gin.H{"deleted": true})
|
|
})
|
|
}
|
|
|
|
func setToolFavorite(c *gin.Context, svc *service.Service, favorite bool) {
|
|
user, err := currentUser(c, svc)
|
|
if err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
toolID, err := parseToolID(c)
|
|
if err != nil {
|
|
fail(c, http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
item, err := svc.SetToolFavorite(user.ID, toolID, favorite)
|
|
if err != nil {
|
|
failService(c, err)
|
|
return
|
|
}
|
|
ok(c, item)
|
|
}
|
|
|
|
func parseToolID(c *gin.Context) (int64, error) {
|
|
id, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
|
if err != nil || id <= 0 {
|
|
return 0, errInvalidToolID
|
|
}
|
|
return id, nil
|
|
}
|
|
|
|
type toolIDError string
|
|
|
|
func (e toolIDError) Error() string { return string(e) }
|
|
|
|
const errInvalidToolID = toolIDError("工具 ID 无效")
|