package version

import (
	"runtime"

	"gorm.io/gorm"

	"git.ilapage.cn/ila/yovision/Sense/server/app/sense/integration/brain_control"
	"git.ilapage.cn/ila/yovision/Sense/server/cmd/migrate/migration"
	common "git.ilapage.cn/ila/yovision/Sense/server/common/models"
)

func init() {
	_, fileName, _, _ := runtime.Caller(0)
	migration.Migrate.SetVersion(migration.GetFilename(fileName), migrateBrainRuntime)
}

func migrateBrainRuntime(db *gorm.DB, version string) error {
	return db.Transaction(func(tx *gorm.DB) error {
		if err := tx.AutoMigrate(&brain_control.ReplayToken{}, &brain_control.RuntimeProjection{}, &brain_control.SourceRevision{}); err != nil {
			return err
		}
		return tx.Create(&common.Migration{Version: version}).Error
	})
}