150 lines
3.9 KiB
Go
150 lines
3.9 KiB
Go
package handlers
|
||
|
||
import (
|
||
"strconv"
|
||
|
||
"github.com/drama-generator/backend/application/services"
|
||
"github.com/drama-generator/backend/infrastructure/storage"
|
||
"github.com/drama-generator/backend/pkg/logger"
|
||
"github.com/drama-generator/backend/pkg/response"
|
||
"github.com/gin-gonic/gin"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type VideoGenerationHandler struct {
|
||
videoService *services.VideoGenerationService
|
||
log *logger.Logger
|
||
}
|
||
|
||
func NewVideoGenerationHandler(db *gorm.DB, transferService *services.ResourceTransferService, localStorage *storage.LocalStorage, aiService *services.AIService, log *logger.Logger) *VideoGenerationHandler {
|
||
return &VideoGenerationHandler{
|
||
videoService: services.NewVideoGenerationService(db, transferService, localStorage, aiService, log),
|
||
log: log,
|
||
}
|
||
}
|
||
|
||
func (h *VideoGenerationHandler) GenerateVideo(c *gin.Context) {
|
||
|
||
var req services.GenerateVideoRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
response.BadRequest(c, err.Error())
|
||
return
|
||
}
|
||
|
||
videoGen, err := h.videoService.GenerateVideo(&req)
|
||
if err != nil {
|
||
h.log.Errorw("Failed to generate video", "error", err)
|
||
response.InternalError(c, err.Error())
|
||
return
|
||
}
|
||
|
||
response.Success(c, videoGen)
|
||
}
|
||
|
||
func (h *VideoGenerationHandler) GenerateVideoFromImage(c *gin.Context) {
|
||
|
||
imageGenID, err := strconv.ParseUint(c.Param("image_gen_id"), 10, 32)
|
||
if err != nil {
|
||
response.BadRequest(c, "无效的图片ID")
|
||
return
|
||
}
|
||
|
||
videoGen, err := h.videoService.GenerateVideoFromImage(uint(imageGenID))
|
||
if err != nil {
|
||
h.log.Errorw("Failed to generate video from image", "error", err)
|
||
response.InternalError(c, err.Error())
|
||
return
|
||
}
|
||
|
||
response.Success(c, videoGen)
|
||
}
|
||
|
||
func (h *VideoGenerationHandler) BatchGenerateForEpisode(c *gin.Context) {
|
||
|
||
episodeID := c.Param("episode_id")
|
||
|
||
videos, err := h.videoService.BatchGenerateVideosForEpisode(episodeID)
|
||
if err != nil {
|
||
h.log.Errorw("Failed to batch generate videos", "error", err)
|
||
response.InternalError(c, err.Error())
|
||
return
|
||
}
|
||
|
||
response.Success(c, videos)
|
||
}
|
||
|
||
func (h *VideoGenerationHandler) GetVideoGeneration(c *gin.Context) {
|
||
|
||
videoGenID, err := strconv.ParseUint(c.Param("id"), 10, 32)
|
||
if err != nil {
|
||
response.BadRequest(c, "无效的ID")
|
||
return
|
||
}
|
||
|
||
videoGen, err := h.videoService.GetVideoGeneration(uint(videoGenID))
|
||
if err != nil {
|
||
response.NotFound(c, "视频生成记录不存在")
|
||
return
|
||
}
|
||
|
||
response.Success(c, videoGen)
|
||
}
|
||
|
||
func (h *VideoGenerationHandler) ListVideoGenerations(c *gin.Context) {
|
||
var storyboardID *uint
|
||
// 优先使用storyboard_id参数
|
||
if storyboardIDStr := c.Query("storyboard_id"); storyboardIDStr != "" {
|
||
id, err := strconv.ParseUint(storyboardIDStr, 10, 32)
|
||
if err == nil {
|
||
uid := uint(id)
|
||
storyboardID = &uid
|
||
}
|
||
}
|
||
status := c.Query("status")
|
||
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
||
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
|
||
|
||
if page < 1 {
|
||
page = 1
|
||
}
|
||
if pageSize < 1 || pageSize > 100 {
|
||
pageSize = 20
|
||
}
|
||
|
||
var dramaIDUint *uint
|
||
if dramaIDStr := c.Query("drama_id"); dramaIDStr != "" {
|
||
did, _ := strconv.ParseUint(dramaIDStr, 10, 32)
|
||
didUint := uint(did)
|
||
dramaIDUint = &didUint
|
||
}
|
||
|
||
// 计算offset：(page - 1) * pageSize
|
||
offset := (page - 1) * pageSize
|
||
videos, total, err := h.videoService.ListVideoGenerations(dramaIDUint, storyboardID, status, pageSize, offset)
|
||
|
||
if err != nil {
|
||
h.log.Errorw("Failed to list videos", "error", err)
|
||
response.InternalError(c, err.Error())
|
||
return
|
||
}
|
||
|
||
response.SuccessWithPagination(c, videos, total, page, pageSize)
|
||
}
|
||
|
||
func (h *VideoGenerationHandler) DeleteVideoGeneration(c *gin.Context) {
|
||
|
||
videoGenID, err := strconv.ParseUint(c.Param("id"), 10, 32)
|
||
if err != nil {
|
||
response.BadRequest(c, "无效的ID")
|
||
return
|
||
}
|
||
|
||
if err := h.videoService.DeleteVideoGeneration(uint(videoGenID)); err != nil {
|
||
h.log.Errorw("Failed to delete video", "error", err)
|
||
response.InternalError(c, err.Error())
|
||
return
|
||
}
|
||
|
||
response.Success(c, nil)
|
||
}
|