init
This commit is contained in:
Connor
212
api/routes/routes.go
Normal file
212
api/routes/routes.go
Normal file
@@ -0,0 +1,212 @@
|
||||
package routes
|
||||
|
||||
import (
|
||||
handlers2 "github.com/drama-generator/backend/api/handlers"
|
||||
middlewares2 "github.com/drama-generator/backend/api/middlewares"
|
||||
services2 "github.com/drama-generator/backend/application/services"
|
||||
storage2 "github.com/drama-generator/backend/infrastructure/storage"
|
||||
"github.com/drama-generator/backend/pkg/config"
|
||||
"github.com/drama-generator/backend/pkg/logger"
|
||||
"github.com/gin-gonic/gin"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func SetupRouter(cfg *config.Config, db *gorm.DB, log *logger.Logger, localStorage interface{}) *gin.Engine {
|
||||
r := gin.New()
|
||||
|
||||
r.Use(gin.Recovery())
|
||||
r.Use(middlewares2.LoggerMiddleware(log))
|
||||
r.Use(middlewares2.CORSMiddleware(cfg.Server.CORSOrigins))
|
||||
|
||||
// 静态文件服务（用户上传的文件）
|
||||
r.Static("/static", cfg.Storage.LocalPath)
|
||||
|
||||
r.GET("/health", func(c *gin.Context) {
|
||||
c.JSON(200, gin.H{
|
||||
"status": "ok",
|
||||
"app": cfg.App.Name,
|
||||
"version": cfg.App.Version,
|
||||
})
|
||||
})
|
||||
|
||||
aiService := services2.NewAIService(db, log)
|
||||
dramaHandler := handlers2.NewDramaHandler(db, cfg, log, nil)
|
||||
aiConfigHandler := handlers2.NewAIConfigHandler(db, cfg, log)
|
||||
scriptGenHandler := handlers2.NewScriptGenerationHandler(db, cfg, log)
|
||||
imageGenService := services2.NewImageGenerationService(db, nil, log)
|
||||
imageGenHandler := handlers2.NewImageGenerationHandler(db, cfg, log, nil)
|
||||
localStoragePtr := localStorage.(*storage2.LocalStorage)
|
||||
transferService := services2.NewResourceTransferService(db, log)
|
||||
videoGenHandler := handlers2.NewVideoGenerationHandler(db, transferService, localStoragePtr, aiService, log)
|
||||
videoMergeHandler := handlers2.NewVideoMergeHandler(db, nil, cfg.Storage.LocalPath, cfg.Storage.BaseURL, log)
|
||||
assetHandler := handlers2.NewAssetHandler(db, cfg, log)
|
||||
characterLibraryService := services2.NewCharacterLibraryService(db, log)
|
||||
characterLibraryHandler := handlers2.NewCharacterLibraryHandler(db, cfg, log, nil)
|
||||
uploadHandler, err := handlers2.NewUploadHandler(cfg, log, characterLibraryService)
|
||||
if err != nil {
|
||||
log.Fatalw("Failed to create upload handler", "error", err)
|
||||
}
|
||||
storyboardHandler := handlers2.NewStoryboardHandler(db, cfg, log)
|
||||
sceneHandler := handlers2.NewSceneHandler(db, log, imageGenService)
|
||||
taskHandler := handlers2.NewTaskHandler(db, log)
|
||||
framePromptService := services2.NewFramePromptService(db, log)
|
||||
framePromptHandler := handlers2.NewFramePromptHandler(framePromptService, log)
|
||||
|
||||
api := r.Group("/api/v1")
|
||||
{
|
||||
api.Use(middlewares2.RateLimitMiddleware())
|
||||
|
||||
dramas := api.Group("/dramas")
|
||||
{
|
||||
dramas.GET("", dramaHandler.ListDramas)
|
||||
dramas.POST("", dramaHandler.CreateDrama)
|
||||
dramas.GET("/stats", dramaHandler.GetDramaStats)
|
||||
dramas.GET("/:id/characters", dramaHandler.GetCharacters)
|
||||
dramas.PUT("/:id/characters", dramaHandler.SaveCharacters)
|
||||
dramas.PUT("/:id/outline", dramaHandler.SaveOutline)
|
||||
dramas.PUT("/:id/episodes", dramaHandler.SaveEpisodes)
|
||||
dramas.PUT("/:id/progress", dramaHandler.SaveProgress)
|
||||
dramas.GET("/:id", dramaHandler.GetDrama)
|
||||
dramas.PUT("/:id", dramaHandler.UpdateDrama)
|
||||
dramas.DELETE("/:id", dramaHandler.DeleteDrama)
|
||||
}
|
||||
|
||||
aiConfigs := api.Group("/ai-configs")
|
||||
{
|
||||
aiConfigs.GET("", aiConfigHandler.ListConfigs)
|
||||
aiConfigs.POST("", aiConfigHandler.CreateConfig)
|
||||
aiConfigs.POST("/test", aiConfigHandler.TestConnection)
|
||||
aiConfigs.GET("/:id", aiConfigHandler.GetConfig)
|
||||
aiConfigs.PUT("/:id", aiConfigHandler.UpdateConfig)
|
||||
aiConfigs.DELETE("/:id", aiConfigHandler.DeleteConfig)
|
||||
}
|
||||
|
||||
generation := api.Group("/generation")
|
||||
{
|
||||
generation.POST("/outline", scriptGenHandler.GenerateOutline)
|
||||
generation.POST("/characters", scriptGenHandler.GenerateCharacters)
|
||||
generation.POST("/episodes", scriptGenHandler.GenerateEpisodes)
|
||||
}
|
||||
|
||||
// 角色库路由
|
||||
characterLibrary := api.Group("/character-library")
|
||||
{
|
||||
characterLibrary.GET("", characterLibraryHandler.ListLibraryItems)
|
||||
characterLibrary.POST("", characterLibraryHandler.CreateLibraryItem)
|
||||
characterLibrary.GET("/:id", characterLibraryHandler.GetLibraryItem)
|
||||
characterLibrary.DELETE("/:id", characterLibraryHandler.DeleteLibraryItem)
|
||||
}
|
||||
|
||||
// 角色图片相关路由
|
||||
characters := api.Group("/characters")
|
||||
{
|
||||
characters.PUT("/:id", characterLibraryHandler.UpdateCharacter)
|
||||
characters.DELETE("/:id", characterLibraryHandler.DeleteCharacter)
|
||||
characters.POST("/batch-generate-images", characterLibraryHandler.BatchGenerateCharacterImages)
|
||||
characters.POST("/:id/generate-image", characterLibraryHandler.GenerateCharacterImage)
|
||||
characters.POST("/:id/upload-image", uploadHandler.UploadCharacterImage)
|
||||
characters.PUT("/:id/image", characterLibraryHandler.UploadCharacterImage)
|
||||
characters.PUT("/:id/image-from-library", characterLibraryHandler.ApplyLibraryItemToCharacter)
|
||||
characters.POST("/:id/add-to-library", characterLibraryHandler.AddCharacterToLibrary)
|
||||
}
|
||||
|
||||
// 文件上传路由
|
||||
upload := api.Group("/upload")
|
||||
{
|
||||
upload.POST("/image", uploadHandler.UploadImage)
|
||||
}
|
||||
|
||||
// 分镜头路由
|
||||
episodes := api.Group("/episodes")
|
||||
{
|
||||
// 分镜头
|
||||
episodes.POST("/:episode_id/storyboards", storyboardHandler.GenerateStoryboard)
|
||||
episodes.GET("/:episode_id/storyboards", sceneHandler.GetStoryboardsForEpisode)
|
||||
episodes.POST("/:episode_id/finalize", dramaHandler.FinalizeEpisode)
|
||||
episodes.GET("/:episode_id/download", dramaHandler.DownloadEpisodeVideo)
|
||||
}
|
||||
|
||||
// 任务路由
|
||||
tasks := api.Group("/tasks")
|
||||
{
|
||||
tasks.GET("/:task_id", taskHandler.GetTaskStatus)
|
||||
tasks.GET("", taskHandler.GetResourceTasks)
|
||||
}
|
||||
|
||||
// 场景路由
|
||||
scenes := api.Group("/scenes")
|
||||
{
|
||||
scenes.PUT("/:scene_id", sceneHandler.UpdateScene)
|
||||
scenes.POST("/generate-image", sceneHandler.GenerateSceneImage)
|
||||
}
|
||||
|
||||
images := api.Group("/images")
|
||||
{
|
||||
images.GET("", imageGenHandler.ListImageGenerations)
|
||||
images.POST("", imageGenHandler.GenerateImage)
|
||||
images.GET("/:id", imageGenHandler.GetImageGeneration)
|
||||
images.DELETE("/:id", imageGenHandler.DeleteImageGeneration)
|
||||
images.POST("/scene/:scene_id", imageGenHandler.GenerateImagesForScene)
|
||||
images.GET("/episode/:episode_id/backgrounds", imageGenHandler.GetBackgroundsForEpisode)
|
||||
images.POST("/episode/:episode_id/backgrounds/extract", imageGenHandler.ExtractBackgroundsForEpisode)
|
||||
images.POST("/episode/:episode_id/batch", imageGenHandler.BatchGenerateForEpisode)
|
||||
}
|
||||
|
||||
videos := api.Group("/videos")
|
||||
{
|
||||
videos.GET("", videoGenHandler.ListVideoGenerations)
|
||||
videos.POST("", videoGenHandler.GenerateVideo)
|
||||
videos.GET("/:id", videoGenHandler.GetVideoGeneration)
|
||||
videos.DELETE("/:id", videoGenHandler.DeleteVideoGeneration)
|
||||
videos.POST("/image/:image_gen_id", videoGenHandler.GenerateVideoFromImage)
|
||||
videos.POST("/episode/:episode_id/batch", videoGenHandler.BatchGenerateForEpisode)
|
||||
}
|
||||
|
||||
videoMerges := api.Group("/video-merges")
|
||||
{
|
||||
videoMerges.GET("", videoMergeHandler.ListMerges)
|
||||
videoMerges.POST("", videoMergeHandler.MergeVideos)
|
||||
videoMerges.GET("/:merge_id", videoMergeHandler.GetMerge)
|
||||
videoMerges.DELETE("/:merge_id", videoMergeHandler.DeleteMerge)
|
||||
}
|
||||
|
||||
assets := api.Group("/assets")
|
||||
{
|
||||
assets.GET("", assetHandler.ListAssets)
|
||||
assets.POST("", assetHandler.CreateAsset)
|
||||
assets.GET("/:id", assetHandler.GetAsset)
|
||||
assets.PUT("/:id", assetHandler.UpdateAsset)
|
||||
assets.DELETE("/:id", assetHandler.DeleteAsset)
|
||||
assets.POST("/import/image/:image_gen_id", assetHandler.ImportFromImageGen)
|
||||
assets.POST("/import/video/:video_gen_id", assetHandler.ImportFromVideoGen)
|
||||
}
|
||||
|
||||
storyboards := api.Group("/storyboards")
|
||||
{
|
||||
storyboards.PUT("/:id", storyboardHandler.UpdateStoryboard)
|
||||
storyboards.POST("/:id/frame-prompt", framePromptHandler.GenerateFramePrompt)
|
||||
storyboards.GET("/:id/frame-prompts", handlers2.GetStoryboardFramePrompts(db, log))
|
||||
}
|
||||
}
|
||||
|
||||
// 前端静态文件服务（放在API路由之后，避免冲突）
|
||||
// 服务前端构建产物
|
||||
r.Static("/assets", "./web/dist/assets")
|
||||
r.StaticFile("/favicon.ico", "./web/dist/favicon.ico")
|
||||
|
||||
// NoRoute处理：对于所有未匹配的路由
|
||||
r.NoRoute(func(c *gin.Context) {
|
||||
path := c.Request.URL.Path
|
||||
|
||||
// 如果是API路径，返回404
|
||||
if len(path) >= 4 && path[:4] == "/api" {
|
||||
c.JSON(404, gin.H{"error": "API endpoint not found"})
|
||||
return
|
||||
}
|
||||
|
||||
// SPA fallback - 返回index.html
|
||||
c.File("./web/dist/index.html")
|
||||
})
|
||||
|
||||
return r
|
||||
}
|
||||
Reference in New Issue
Block a user