143 lines
3.6 KiB
Go
143 lines
3.6 KiB
Go
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,
|
||
})
|
||
}
|