146 lines
3.7 KiB
Go
146 lines
3.7 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"sproutworkcollect-backend/internal/service"
|
|
)
|
|
|
|
// PublicHandler handles all publicly accessible API endpoints.
|
|
type PublicHandler struct {
|
|
workSvc *service.WorkService
|
|
settingsSvc *service.SettingsService
|
|
rateLimiter *service.RateLimiter
|
|
}
|
|
|
|
// NewPublicHandler wires up a PublicHandler with its dependencies.
|
|
func NewPublicHandler(
|
|
workSvc *service.WorkService,
|
|
settingsSvc *service.SettingsService,
|
|
rateLimiter *service.RateLimiter,
|
|
) *PublicHandler {
|
|
return &PublicHandler{
|
|
workSvc: workSvc,
|
|
settingsSvc: settingsSvc,
|
|
rateLimiter: rateLimiter,
|
|
}
|
|
}
|
|
|
|
// GetSettings handles GET /api/settings
|
|
func (h *PublicHandler) GetSettings(c *gin.Context) {
|
|
settings, err := h.settingsSvc.Load()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, settings)
|
|
}
|
|
|
|
// GetWorks handles GET /api/works
|
|
func (h *PublicHandler) GetWorks(c *gin.Context) {
|
|
works, err := h.workSvc.LoadAllWorks()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
|
|
return
|
|
}
|
|
|
|
responses := make([]any, len(works))
|
|
for i, w := range works {
|
|
responses[i] = h.workSvc.BuildResponse(w)
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": true,
|
|
"data": responses,
|
|
"total": len(responses),
|
|
})
|
|
}
|
|
|
|
// GetWorkDetail handles GET /api/works/:work_id
|
|
func (h *PublicHandler) GetWorkDetail(c *gin.Context) {
|
|
workID := c.Param("work_id")
|
|
|
|
work, err := h.workSvc.LoadWork(workID)
|
|
if err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"success": false, "message": "作品不存在"})
|
|
return
|
|
}
|
|
|
|
fp := service.Fingerprint(c.ClientIP(), c.GetHeader("User-Agent"))
|
|
if h.rateLimiter.CanPerform(fp, "view", workID) {
|
|
_ = h.workSvc.UpdateStats(workID, "view")
|
|
// Reload to return the updated view count.
|
|
if fresh, err2 := h.workSvc.LoadWork(workID); err2 == nil {
|
|
work = fresh
|
|
}
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": true,
|
|
"data": h.workSvc.BuildResponse(work),
|
|
})
|
|
}
|
|
|
|
// SearchWorks handles GET /api/search?q=...&category=...
|
|
func (h *PublicHandler) SearchWorks(c *gin.Context) {
|
|
works, err := h.workSvc.SearchWorks(c.Query("q"), c.Query("category"))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
|
|
return
|
|
}
|
|
|
|
responses := make([]any, len(works))
|
|
for i, w := range works {
|
|
responses[i] = h.workSvc.BuildResponse(w)
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": true,
|
|
"data": responses,
|
|
"total": len(responses),
|
|
})
|
|
}
|
|
|
|
// GetCategories handles GET /api/categories
|
|
func (h *PublicHandler) GetCategories(c *gin.Context) {
|
|
cats, err := h.workSvc.AllCategories()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"success": true, "data": cats})
|
|
}
|
|
|
|
// LikeWork handles POST /api/like/:work_id
|
|
func (h *PublicHandler) LikeWork(c *gin.Context) {
|
|
workID := c.Param("work_id")
|
|
|
|
if _, err := h.workSvc.LoadWork(workID); err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"success": false, "message": "作品不存在"})
|
|
return
|
|
}
|
|
|
|
fp := service.Fingerprint(c.ClientIP(), c.GetHeader("User-Agent"))
|
|
if !h.rateLimiter.CanPerform(fp, "like", workID) {
|
|
c.JSON(http.StatusTooManyRequests, gin.H{
|
|
"success": false,
|
|
"message": "操作太频繁，请稍后再试",
|
|
})
|
|
return
|
|
}
|
|
|
|
if err := h.workSvc.UpdateStats(workID, "like"); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": "点赞失败"})
|
|
return
|
|
}
|
|
|
|
work, _ := h.workSvc.LoadWork(workID)
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": true,
|
|
"message": "点赞成功",
|
|
"likes": work.Likes,
|
|
})
|
|
}
|