package migrate

import (
	"github.com/pingcap/errors"
	"github.com/pingcap/log"
	"go.uber.org/zap"
	"gorm.io/gorm"

	"github.com/tidbcloud/auth-server/internal/entity"
)

var _ = AddAuthServerDBMigrateRecord(
	"202403181800",
	func(db *gorm.DB) error {
		if err := db.AutoMigrate(&entity.Oauth2AccessToken{}, &entity.Oauth2Application{}, &entity.Oauth2CodeGrant{}); err != nil {
			errMsg := "migrate oauth2 schema table failed"
			log.Error(errMsg, zap.Error(err))
			return errors.Annotate(err, errMsg)
		}
		return nil
	},
)