131 lines
3.0 KiB
Go
131 lines
3.0 KiB
Go
package handlers
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/niangaodev/art-code/models"
|
|
"github.com/niangaodev/art-code/repositories"
|
|
)
|
|
|
|
// 获取作品列表
|
|
func GetWorks(c *gin.Context) {
|
|
// 从数据库获取所有作品
|
|
works, err := repositories.GetWorks()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch works"})
|
|
return
|
|
}
|
|
|
|
// 构建响应
|
|
var responses []interface{}
|
|
for _, work := range works {
|
|
response, err := repositories.BuildWorkResponse(&work)
|
|
if err != nil {
|
|
log.Printf("Error building work response: %v", err)
|
|
continue
|
|
}
|
|
responses = append(responses, response)
|
|
}
|
|
|
|
c.JSON(http.StatusOK, responses)
|
|
}
|
|
|
|
func GetWork(c *gin.Context) {
|
|
id := c.Param("id")
|
|
// 从数据库获取作品
|
|
work, err := repositories.GetWorkByID(id)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch work"})
|
|
return
|
|
}
|
|
|
|
if work == nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "Work not found"})
|
|
return
|
|
}
|
|
|
|
// 构建响应
|
|
response, err := repositories.BuildWorkResponse(work)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to build work response"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, response)
|
|
}
|
|
|
|
// 获取作品列表 (Admin)
|
|
func AdminGetWorks(c *gin.Context) {
|
|
works, err := repositories.GetWorks()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get works"})
|
|
return
|
|
}
|
|
|
|
// 构建响应
|
|
var responses []interface{}
|
|
for _, work := range works {
|
|
response, err := repositories.BuildWorkResponse(&work)
|
|
if err != nil {
|
|
log.Printf("Error building work response: %v", err)
|
|
continue
|
|
}
|
|
responses = append(responses, response)
|
|
}
|
|
|
|
c.JSON(http.StatusOK, responses)
|
|
}
|
|
|
|
// 创建作品
|
|
func AdminCreateWork(c *gin.Context) {
|
|
var work models.Work
|
|
if err := c.ShouldBindJSON(&work); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request"})
|
|
return
|
|
}
|
|
|
|
// 创建作品
|
|
if err := repositories.CreateWork(&work); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create work"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "Work created successfully"})
|
|
}
|
|
|
|
// 更新作品
|
|
func AdminUpdateWork(c *gin.Context) {
|
|
workID := c.Param("id")
|
|
var work models.Work
|
|
if err := c.ShouldBindJSON(&work); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request"})
|
|
return
|
|
}
|
|
|
|
// 设置作品ID
|
|
work.ID = workID
|
|
|
|
// 更新作品
|
|
if err := repositories.UpdateWork(&work); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update work"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "Work updated successfully"})
|
|
}
|
|
|
|
// 删除作品
|
|
func AdminDeleteWork(c *gin.Context) {
|
|
workID := c.Param("id")
|
|
|
|
// 删除作品
|
|
if err := repositories.DeleteWork(workID); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to delete work"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "Work deleted successfully"})
|
|
}
|