create project

domain/models/drama.go

@@ -0,0 +1,148 @@
+package models
+
+import (
+	"time"
+
+	"gorm.io/datatypes"
+	"gorm.io/gorm"
+)
+
+type Drama struct {
+	ID            uint           `gorm:"primaryKey;autoIncrement" json:"id"`
+	Title         string         `gorm:"type:varchar(200);not null" json:"title"`
+	Description   *string        `gorm:"type:text" json:"description"`
+	Genre         *string        `gorm:"type:varchar(50)" json:"genre"`
+	Style         string         `gorm:"type:varchar(50);default:'realistic'" json:"style"`
+	TotalEpisodes int            `gorm:"default:1" json:"total_episodes"`
+	TotalDuration int            `gorm:"default:0" json:"total_duration"`
+	Status        string         `gorm:"type:varchar(20);default:'draft';not null" json:"status"`
+	Thumbnail     *string        `gorm:"type:varchar(500)" json:"thumbnail"`
+	Tags          datatypes.JSON `gorm:"type:json" json:"tags"`
+	Metadata      datatypes.JSON `gorm:"type:json" json:"metadata"`
+	CreatedAt     time.Time      `gorm:"not null;autoCreateTime" json:"created_at"`
+	UpdatedAt     time.Time      `gorm:"not null;autoUpdateTime" json:"updated_at"`
+	DeletedAt     gorm.DeletedAt `gorm:"index" json:"-"`
+
+	Episodes   []Episode   `gorm:"foreignKey:DramaID" json:"episodes,omitempty"`
+	Characters []Character `gorm:"foreignKey:DramaID" json:"characters,omitempty"`
+	Scenes     []Scene     `gorm:"foreignKey:DramaID" json:"scenes,omitempty"`
+}
+
+func (d *Drama) TableName() string {
+	return "dramas"
+}
+
+type Character struct {
+	ID              uint           `gorm:"primaryKey;autoIncrement" json:"id"`
+	DramaID         uint           `gorm:"not null;index" json:"drama_id"`
+	Name            string         `gorm:"type:varchar(100);not null" json:"name"`
+	Role            *string        `gorm:"type:varchar(50)" json:"role"`
+	Description     *string        `gorm:"type:text" json:"description"`
+	Appearance      *string        `gorm:"type:text" json:"appearance"`
+	Personality     *string        `gorm:"type:text" json:"personality"`
+	VoiceStyle      *string        `gorm:"type:varchar(200)" json:"voice_style"`
+	ImageURL        *string        `gorm:"type:varchar(500)" json:"image_url"`
+	ReferenceImages datatypes.JSON `gorm:"type:json" json:"reference_images"`
+	SeedValue       *string        `gorm:"type:varchar(100)" json:"seed_value"`
+	SortOrder       int            `gorm:"default:0" json:"sort_order"`
+	CreatedAt       time.Time      `gorm:"not null;autoCreateTime" json:"created_at"`
+	UpdatedAt       time.Time      `gorm:"not null;autoUpdateTime" json:"updated_at"`
+	DeletedAt       gorm.DeletedAt `gorm:"index" json:"-"`
+
+	// 多对多关系：角色可以属于多个章节
+	Episodes []Episode `gorm:"many2many:episode_characters;" json:"episodes,omitempty"`
+
+	// 运行时字段（不存储到数据库）
+	ImageGenerationStatus *string `gorm:"-" json:"image_generation_status,omitempty"`
+	ImageGenerationError  *string `gorm:"-" json:"image_generation_error,omitempty"`
+}
+
+func (c *Character) TableName() string {
+	return "characters"
+}
+
+type Episode struct {
+	ID            uint           `gorm:"primaryKey;autoIncrement" json:"id"`
+	DramaID       uint           `gorm:"not null;index" json:"drama_id"`
+	EpisodeNum    int            `gorm:"column:episode_number;not null" json:"episode_number"`
+	Title         string         `gorm:"type:varchar(200);not null" json:"title"`
+	ScriptContent *string        `gorm:"type:longtext" json:"script_content"`
+	Description   *string        `gorm:"type:text" json:"description"`
+	Duration      int            `gorm:"default:0" json:"duration"` // 总时长（秒）
+	Status        string         `gorm:"type:varchar(20);default:'draft'" json:"status"`
+	VideoURL      *string        `gorm:"type:varchar(500)" json:"video_url"`
+	Thumbnail     *string        `gorm:"type:varchar(500)" json:"thumbnail"`
+	CreatedAt     time.Time      `gorm:"not null;autoCreateTime" json:"created_at"`
+	UpdatedAt     time.Time      `gorm:"not null;autoUpdateTime" json:"updated_at"`
+	DeletedAt     gorm.DeletedAt `gorm:"index" json:"-"`
+
+	// 关联
+	Drama       Drama        `gorm:"foreignKey:DramaID" json:"drama,omitempty"`
+	Storyboards []Storyboard `gorm:"foreignKey:EpisodeID" json:"storyboards,omitempty"`
+	Characters  []Character  `gorm:"many2many:episode_characters;" json:"characters,omitempty"`
+	Scenes      []Scene      `gorm:"foreignKey:EpisodeID" json:"scenes,omitempty"`
+}
+
+func (e *Episode) TableName() string {
+	return "episodes"
+}
+
+type Storyboard struct {
+	ID               uint           `gorm:"primaryKey;autoIncrement" json:"id"`
+	EpisodeID        uint           `gorm:"not null;index:idx_storyboards_episode_id" json:"episode_id"`
+	SceneID          *uint          `gorm:"index:idx_storyboards_scene_id;column:scene_id" json:"scene_id"`
+	StoryboardNumber int            `gorm:"not null;column:storyboard_number" json:"storyboard_number"`
+	Title            *string        `gorm:"size:255" json:"title"`
+	Location         *string        `gorm:"size:255" json:"location"`
+	Time             *string        `gorm:"size:255" json:"time"`
+	ShotType         *string        `gorm:"size:100" json:"shot_type"`
+	Angle            *string        `gorm:"size:100" json:"angle"`
+	Movement         *string        `gorm:"size:100" json:"movement"`
+	Action           *string        `gorm:"type:text" json:"action"`
+	Result           *string        `gorm:"type:text" json:"result"`
+	Atmosphere       *string        `gorm:"type:text" json:"atmosphere"`
+	ImagePrompt      *string        `gorm:"type:text" json:"image_prompt"`
+	VideoPrompt      *string        `gorm:"type:text" json:"video_prompt"`
+	BgmPrompt        *string        `gorm:"type:text" json:"bgm_prompt"`
+	SoundEffect      *string        `gorm:"size:255" json:"sound_effect"`
+	Dialogue         *string        `gorm:"type:text" json:"dialogue"`
+	Description      *string        `gorm:"type:text" json:"description"`
+	Duration         int            `gorm:"default:5" json:"duration"`
+	ComposedImage    *string        `gorm:"type:text" json:"composed_image"`
+	VideoURL         *string        `gorm:"type:text" json:"video_url"`
+	Status           string         `gorm:"type:varchar(20);default:'pending'" json:"status"`
+	CreatedAt        time.Time      `gorm:"autoCreateTime" json:"created_at"`
+	UpdatedAt        time.Time      `gorm:"autoUpdateTime" json:"updated_at"`
+	DeletedAt        gorm.DeletedAt `gorm:"index" json:"-"`
+
+	Episode    Episode     `gorm:"foreignKey:EpisodeID;constraint:OnDelete:CASCADE" json:"episode,omitempty"`
+	Background *Scene      `gorm:"foreignKey:SceneID" json:"background,omitempty"`
+	Characters []Character `gorm:"many2many:storyboard_characters;" json:"characters,omitempty"`
+}
+
+func (s *Storyboard) TableName() string {
+	return "storyboards"
+}
+
+type Scene struct {
+	ID              uint           `gorm:"primaryKey;autoIncrement" json:"id"`
+	DramaID         uint           `gorm:"not null;index:idx_scenes_drama_id" json:"drama_id"`
+	EpisodeID       *uint          `gorm:"index:idx_scenes_episode_id" json:"episode_id"` // 场景所属章节
+	Location        string         `gorm:"type:varchar(200);not null" json:"location"`
+	Time            string         `gorm:"type:varchar(100);not null" json:"time"`
+	Prompt          string         `gorm:"type:text;not null" json:"prompt"`
+	StoryboardCount int            `gorm:"default:1" json:"storyboard_count"`
+	ImageURL        *string        `gorm:"type:varchar(500)" json:"image_url"`
+	Status          string         `gorm:"type:varchar(20);default:'pending'" json:"status"` // pending, generated, failed
+	CreatedAt       time.Time      `gorm:"not null;autoCreateTime" json:"created_at"`
+	UpdatedAt       time.Time      `gorm:"not null;autoUpdateTime" json:"updated_at"`
+	DeletedAt       gorm.DeletedAt `gorm:"index" json:"-"`
+
+	// 运行时字段（不存储到数据库）
+	ImageGenerationStatus *string `gorm:"-" json:"image_generation_status,omitempty"`
+	ImageGenerationError  *string `gorm:"-" json:"image_generation_error,omitempty"`
+}
+
+func (s *Scene) TableName() string {
+	return "scenes"
+}