142 lines
3.0 KiB
Go
142 lines
3.0 KiB
Go
package handler
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/gin-gonic/gin"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"rear/internal/container"
|
|
"rear/internal/model"
|
|
)
|
|
|
|
type Handler struct {
|
|
container *container.Container
|
|
}
|
|
|
|
func NewLibraryHandler(container *container.Container) *Handler {
|
|
return &Handler{container: container}
|
|
}
|
|
|
|
// GetLibrary 获取存储哭
|
|
func (h *Handler) GetLibrary(c *gin.Context) {
|
|
libraries, err := h.container.LibraryRepo.GetAllLibrary()
|
|
if err != nil {
|
|
log.Printf("Failed to get all libraries: %v", err)
|
|
c.JSON(http.StatusInternalServerError, model.Response{
|
|
Code: http.StatusInternalServerError,
|
|
Message: "Internal server error",
|
|
Data: nil,
|
|
})
|
|
return
|
|
}
|
|
|
|
log.Printf("Total libraries: %d", len(libraries))
|
|
c.JSON(http.StatusOK, model.Response{
|
|
Code: http.StatusOK,
|
|
Message: "Success",
|
|
Data: libraries,
|
|
})
|
|
}
|
|
|
|
func (h *Handler) AddLibrary(c *gin.Context) {
|
|
// 定义请求体结构
|
|
type AddLibraryRequest struct {
|
|
Path string `json:"path"`
|
|
}
|
|
|
|
var req AddLibraryRequest
|
|
|
|
// 读取原始JSON数据（用于调试）
|
|
rawData, _ := c.GetRawData()
|
|
// 重新设置请求体，因为GetRawData()会消耗掉body
|
|
c.Request.Body = io.NopCloser(bytes.NewBuffer(rawData))
|
|
|
|
// 绑定JSON到结构体
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, model.Response{
|
|
Code: http.StatusBadRequest,
|
|
Message: "Invalid request body",
|
|
})
|
|
return
|
|
}
|
|
|
|
// 现在可以使用 req.Path
|
|
path := req.Path
|
|
|
|
library := &model.LibraryTable{
|
|
ImgPath: path,
|
|
IsEnable: true,
|
|
}
|
|
err := h.container.LibraryRepo.AddLibrary(library)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, model.Response{
|
|
Code: http.StatusInternalServerError,
|
|
Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.Response{
|
|
Code: http.StatusOK,
|
|
Message: "Success",
|
|
Data: nil,
|
|
})
|
|
}
|
|
func (h *Handler) UpdateLibrary(c *gin.Context) {
|
|
var path string
|
|
if err := c.ShouldBindJSON(&path); err != nil {
|
|
c.JSON(http.StatusBadRequest, model.Response{
|
|
Code: http.StatusBadRequest,
|
|
Message: "Invalid request body",
|
|
})
|
|
return
|
|
}
|
|
library := &model.LibraryTable{
|
|
ImgPath: path,
|
|
IsEnable: true,
|
|
}
|
|
err := h.container.LibraryRepo.AddLibrary(library)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, model.Response{
|
|
Code: http.StatusInternalServerError,
|
|
Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.Response{
|
|
Code: http.StatusOK,
|
|
Message: "Success",
|
|
Data: nil,
|
|
})
|
|
}
|
|
func (h *Handler) DeleteLibrary(c *gin.Context) {
|
|
var path string
|
|
if err := c.ShouldBindJSON(&path); err != nil {
|
|
c.JSON(http.StatusBadRequest, model.Response{
|
|
Code: http.StatusBadRequest,
|
|
Message: "Invalid request body",
|
|
})
|
|
return
|
|
}
|
|
library := &model.LibraryTable{
|
|
ImgPath: path,
|
|
IsEnable: true,
|
|
}
|
|
err := h.container.LibraryRepo.AddLibrary(library)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, model.Response{
|
|
Code: http.StatusInternalServerError,
|
|
Message: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.Response{
|
|
Code: http.StatusOK,
|
|
Message: "Success",
|
|
Data: nil,
|
|
})
|
|
}
|