算了，后端我自己写吧

2026-04-01 12:09:02 +08:00
parent 4138340f53
commit 1a0a01a56d
69 changed files with 1949 additions and 102 deletions
@@ -0,0 +1,241 @@
+package handler
+
+import (
+	"ops/internal/service"
+	"ops/pkg/response"
+
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+type FileHandler struct {
+	service service.FileService
+}
+
+func NewFileHandler(db *gorm.DB) *FileHandler {
+	return &FileHandler{
+		service: service.NewFileService(db),
+	}
+}
+
+// UploadFile 上传文件
+// @Summary 上传文件
+// @Description 上传文件到服务器，支持图片、文档等多种类型
+// @Tags 文件管理
+// @Accept multipart/form-data
+// @Produce json
+// @Param userID header string false "用户ID" default("")
+// @Param file formData file true "文件内容"
+// @Param type formData string false "文件类型" default(image)
+// @Param description formData string false "文件描述"
+// @Success 200 {object} response.StandardResponse "成功"
+// @Failure 400 {object} response.StandardResponse "参数错误"
+// @Failure 401 {object} response.StandardResponse "未授权"
+// @Failure 413 {object} response.StandardResponse "文件过大"
+// @Failure 415 {object} response.StandardResponse "文件类型不支持"
+// @Failure 500 {object} response.StandardResponse "服务器错误"
+// @Router /api/v1/files/upload [post]
+func (h *FileHandler) UploadFile(c *gin.Context) {
+	// 从上下文中获取用户ID
+	userID, exists := c.Get("userID")
+	if !exists {
+		response.Unauthorized(c)
+		return
+	}
+
+	// 获取文件类型参数
+	fileType := c.PostForm("type")
+	if fileType == "" {
+		fileType = "image" // 默认类型为图片
+	}
+
+	// 获取文件描述
+	description := c.PostForm("description")
+
+	// 获取上传的文件
+	file, err := c.FormFile("file")
+	if err != nil {
+		response.BadRequest(c, "请选择要上传的文件")
+		return
+	}
+
+	// 调用Service上传文件
+	uploadResponse, success := h.service.UploadFile(c, userID.(uint), file, fileType, description)
+	if !success {
+		response.BadRequest(c, "文件上传失败，请检查文件格式和大小")
+		return
+	}
+
+	response.Success(c, uploadResponse)
+}
+
+// GetFileList 获取文件列表
+// @Summary 获取文件列表
+// @Description 获取当前用户上传的文件列表
+// @Tags 文件管理
+// @Accept json
+// @Produce json
+// @Param userID header string false "用户ID" default("")
+// @Param type query string false "文件类型过滤"
+// @Param page query int false "页码" default(1)
+// @Param entries query int false "每页数量" default(20)
+// @Success 200 {object} response.StandardResponse "成功"
+// @Failure 400 {object} response.StandardResponse "参数错误"
+// @Failure 401 {object} response.StandardResponse "未授权"
+// @Router /api/v1/files/list [get]
+func (h *FileHandler) GetFileList(c *gin.Context) {
+	// 从上下文中获取用户ID
+	userID, exists := c.Get("userID")
+	if !exists {
+		response.Unauthorized(c)
+		return
+	}
+
+	// 获取查询参数
+	fileType := c.Query("type")
+	page := GetIntParam(c, "page", 1)
+	entries := GetIntParam(c, "entries", 20)
+
+	// 调用Service获取文件列表
+	fileListResponse, success := h.service.GetFileList(userID.(uint), fileType, page, entries)
+	if !success {
+		response.BadRequest(c, "参数错误")
+		return
+	}
+
+	response.Success(c, fileListResponse)
+}
+
+// GetFileByID 获取文件信息
+// @Summary 获取文件信息
+// @Description 根据文件ID获取文件详细信息
+// @Tags 文件管理
+// @Accept json
+// @Produce json
+// @Param userID header string false "用户ID" default("")
+// @Param id path int true "文件ID"
+// @Success 200 {object} response.StandardResponse "成功"
+// @Failure 401 {object} response.StandardResponse "未授权"
+// @Failure 404 {object} response.StandardResponse "文件不存在"
+// @Router /api/v1/files/{id} [get]
+func (h *FileHandler) GetFileByID(c *gin.Context) {
+	// 从上下文中获取用户ID
+	userID, exists := c.Get("userID")
+	if !exists {
+		response.Unauthorized(c)
+		return
+	}
+
+	// 获取文件ID
+	fileID := GetUintParam(c, "id")
+	if fileID == 0 {
+		response.BadRequest(c, "文件ID无效")
+		return
+	}
+
+	// 调用Service获取文件信息
+	file, success := h.service.GetFileByID(fileID, userID.(uint))
+	if !success {
+		response.Error(c, "-100", "文件不存在或无权限访问")
+		return
+	}
+
+	response.Success(c, gin.H{
+		"file_id":    file.ID,
+		"name":       file.Name,
+		"sha256":     file.Sha256,
+		"mime":       file.Mime,
+		"type":       file.Type,
+		"size":       file.Const, // 注意：这里const字段实际上存储的是使用次数，需要确认实际字段
+		"created_at": file.Date.Format("2006-01-02T15:04:05Z"),
+		"path":       file.Path,
+	})
+}
+
+// DeleteFile 删除文件
+// @Summary 删除文件
+// @Description 删除用户上传的文件
+// @Tags 文件管理
+// @Accept json
+// @Produce json
+// @Param userID header string false "用户ID" default("")
+// @Param id path int true "文件ID"
+// @Success 200 {object} response.StandardResponse "成功"
+// @Failure 401 {object} response.StandardResponse "未授权"
+// @Failure 404 {object} response.StandardResponse "文件不存在"
+// @Router /api/v1/files/{id} [delete]
+func (h *FileHandler) DeleteFile(c *gin.Context) {
+	// 从上下文中获取用户ID
+	userID, exists := c.Get("userID")
+	if !exists {
+		response.Unauthorized(c)
+		return
+	}
+
+	// 获取文件ID
+	fileID := GetUintParam(c, "id")
+	if fileID == 0 {
+		response.BadRequest(c, "文件ID无效")
+		return
+	}
+
+	// 调用Service删除文件
+	success := h.service.DeleteFile(fileID, userID.(uint))
+	if !success {
+		response.Error(c, "-100", "文件删除失败，文件不存在或无权限")
+		return
+	}
+
+	response.Success(c, gin.H{"message": "文件删除成功"})
+}
+
+// DownloadFile 下载文件
+// @Summary 下载文件
+// @Description 下载文件内容（直接下载）
+// @Tags 文件管理
+// @Accept json
+// @Produce application/octet-stream
+// @Param hash path string true "文件SHA256哈希值"
+// @Success 200 {file} binary "文件内容"
+// @Failure 404 {object} response.StandardResponse "文件不存在"
+// @Router /api/v1/files/download/{hash} [get]
+func (h *FileHandler) DownloadFile(c *gin.Context) {
+	hash := c.Param("hash")
+	if hash == "" {
+		response.BadRequest(c, "文件哈希值无效")
+		return
+	}
+
+	// 调用Service下载文件
+	success := h.service.DownloadFile(c, hash, true)
+	if !success {
+		response.Error(c, "-100", "文件不存在")
+		return
+	}
+}
+
+// GetFile 获取文件（预览）
+// @Summary 获取文件（预览）
+// @Description 获取文件内容（浏览器预览）
+// @Tags 文件管理
+// @Accept json
+// @Produce *
+// @Param hash path string true "文件SHA256哈希值"
+// @Success 200 {file} binary "文件内容"
+// @Failure 404 {object} response.StandardResponse "文件不存在"
+// @Router /api/v1/files/get/{hash} [get]
+func (h *FileHandler) GetFile(c *gin.Context) {
+	hash := c.Param("hash")
+	if hash == "" {
+		response.BadRequest(c, "文件哈希值无效")
+		return
+	}
+
+	// 调用Service获取文件（预览模式）
+	success := h.service.DownloadFile(c, hash, false)
+	if !success {
+		response.Error(c, "-100", "文件不存在")
+		return
+	}
+}
+