create project

api/handlers/upload.go

@@ -0,0 +1,142 @@
+package handlers
+
+import (
+	services2 "github.com/drama-generator/backend/application/services"
+	"github.com/drama-generator/backend/pkg/config"
+	"github.com/drama-generator/backend/pkg/logger"
+	"github.com/drama-generator/backend/pkg/response"
+	"github.com/gin-gonic/gin"
+)
+
+type UploadHandler struct {
+	uploadService           *services2.UploadService
+	characterLibraryService *services2.CharacterLibraryService
+	log                     *logger.Logger
+}
+
+func NewUploadHandler(cfg *config.Config, log *logger.Logger, characterLibraryService *services2.CharacterLibraryService) (*UploadHandler, error) {
+	uploadService, err := services2.NewUploadService(cfg, log)
+	if err != nil {
+		return nil, err
+	}
+
+	return &UploadHandler{
+		uploadService:           uploadService,
+		characterLibraryService: characterLibraryService,
+		log:                     log,
+	}, nil
+}
+
+// UploadImage 上传图片
+func (h *UploadHandler) UploadImage(c *gin.Context) {
+	// 获取上传的文件
+	file, header, err := c.Request.FormFile("file")
+	if err != nil {
+		response.BadRequest(c, "请选择文件")
+		return
+	}
+	defer file.Close()
+
+	// 检查文件类型
+	contentType := header.Header.Get("Content-Type")
+	if contentType == "" {
+		contentType = "application/octet-stream"
+	}
+
+	// 验证是图片类型
+	allowedTypes := map[string]bool{
+		"image/jpeg": true,
+		"image/jpg":  true,
+		"image/png":  true,
+		"image/gif":  true,
+		"image/webp": true,
+	}
+
+	if !allowedTypes[contentType] {
+		response.BadRequest(c, "只支持图片格式 (jpg, png, gif, webp)")
+		return
+	}
+
+	// 检查文件大小 (10MB)
+	if header.Size > 10*1024*1024 {
+		response.BadRequest(c, "文件大小不能超过10MB")
+		return
+	}
+
+	// 上传到MinIO
+	fileURL, err := h.uploadService.UploadCharacterImage(file, header.Filename, contentType)
+	if err != nil {
+		h.log.Errorw("Failed to upload image", "error", err)
+		response.InternalError(c, "上传失败")
+		return
+	}
+
+	response.Success(c, gin.H{
+		"url":      fileURL,
+		"filename": header.Filename,
+		"size":     header.Size,
+	})
+}
+
+// UploadCharacterImage 上传角色图片（带角色ID）
+func (h *UploadHandler) UploadCharacterImage(c *gin.Context) {
+	characterID := c.Param("id")
+
+	// 获取上传的文件
+	file, header, err := c.Request.FormFile("file")
+	if err != nil {
+		response.BadRequest(c, "请选择文件")
+		return
+	}
+	defer file.Close()
+
+	// 检查文件类型
+	contentType := header.Header.Get("Content-Type")
+	if contentType == "" {
+		contentType = "application/octet-stream"
+	}
+
+	// 验证是图片类型
+	allowedTypes := map[string]bool{
+		"image/jpeg": true,
+		"image/jpg":  true,
+		"image/png":  true,
+		"image/gif":  true,
+		"image/webp": true,
+	}
+
+	if !allowedTypes[contentType] {
+		response.BadRequest(c, "只支持图片格式 (jpg, png, gif, webp)")
+		return
+	}
+
+	// 检查文件大小 (10MB)
+	if header.Size > 10*1024*1024 {
+		response.BadRequest(c, "文件大小不能超过10MB")
+		return
+	}
+
+	// 上传到MinIO
+	fileURL, err := h.uploadService.UploadCharacterImage(file, header.Filename, contentType)
+	if err != nil {
+		h.log.Errorw("Failed to upload character image", "error", err)
+		response.InternalError(c, "上传失败")
+		return
+	}
+
+	// 更新角色的image_url字段到数据库
+	err = h.characterLibraryService.UploadCharacterImage(characterID, fileURL)
+	if err != nil {
+		h.log.Errorw("Failed to update character image_url", "error", err, "character_id", characterID)
+		response.InternalError(c, "更新角色图片失败")
+		return
+	}
+
+	h.log.Infow("Character image uploaded and saved", "character_id", characterID, "url", fileURL)
+
+	response.Success(c, gin.H{
+		"url":      fileURL,
+		"filename": header.Filename,
+		"size":     header.Size,
+	})
+}