package live

import (
	"gorm.io/gorm"
	"kra/pkg/database/migration"
)

func Migrations() []migration.Step {
	return []migration.Step{{ID: "202609190001_live_schema", Migrate: func(db *gorm.DB) error {
		if err := migration.CreateMissingTables(db, &categoryPO{}, &roomPO{}); err != nil {
			return err
		}
		var count int64
		if err := db.Model(&categoryPO{}).Count(&count).Error; err != nil {
			return err
		}
		if count > 0 {
			return nil
		}
		return db.Create(&[]categoryPO{{Name: "游戏竞技", Icon: "🎮", Sort: 1, Enabled: true}, {Name: "音乐区", Icon: "🎵", Sort: 2, Enabled: true}, {Name: "闲聊电台", Icon: "🌙", Sort: 3, Enabled: true}, {Name: "舞蹈才艺", Icon: "💃", Sort: 4, Enabled: true}, {Name: "娱乐直播", Icon: "🎪", Sort: 5, Enabled: true}}).Error
	}}, {ID: "202609220002_live_follow_schema", Migrate: func(db *gorm.DB) error {
		return migration.CreateMissingTables(db, &followPO{})
	}}}
}