migrations.go

package gen

import (
	"github.com/jinzhu/gorm"
	"github.com/rs/zerolog/log"
	"gopkg.in/gormigrate.v1"
)

// Migrate runs migrations
func Migrate(db *gorm.DB, options *gormigrate.Options, migrations []*gormigrate.Migration) error {
	m := gormigrate.New(db, options, migrations)

	// // it's possible to use this, but in case of any specific keys or columns are created in migrations, they will not be generated by automigrate
	// m.InitSchema(func(tx *gorm.DB) error {
	// 	return AutoMigrate(db)
	// })

	return m.Migrate()
}

func AutoMigrate(db *gorm.DB) (err error) {
	_db := db.AutoMigrate(
		User{},
		UserAPIKey{},
		Profile{},
		Role{},
		Permission{},
	)
	if _db.Error != nil {
		log.Err(_db.Error).Send()
		return _db.Error
	}
	if _db.Dialect().GetName() != "sqlite3" {

		err = _db.Model(RoleUsers{}).RemoveForeignKey("userId", TableName("users")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(RoleUsers{}).AddForeignKey("userId", TableName("users")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(ProfileUsers{}).RemoveForeignKey("userId", TableName("users")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(ProfileUsers{}).AddForeignKey("userId", TableName("users")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(PermissionUsers{}).RemoveForeignKey("userId", TableName("users")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(PermissionUsers{}).AddForeignKey("userId", TableName("users")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(UserAPIKey{}).RemoveForeignKey("userId", TableName("users")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(UserAPIKey{}).AddForeignKey("userId", TableName("users")+"(id)", "SET NULL", "SET NULL").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(UserAPIKeyPermissions{}).RemoveForeignKey("apikeyId", TableName("user_api_keys")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(UserAPIKeyPermissions{}).AddForeignKey("apikeyId", TableName("user_api_keys")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(ProfileUsers{}).RemoveForeignKey("profileId", TableName("profiles")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(ProfileUsers{}).AddForeignKey("profileId", TableName("profiles")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(RoleUsers{}).RemoveForeignKey("roleId", TableName("roles")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(RoleUsers{}).AddForeignKey("roleId", TableName("roles")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(RoleParents{}).RemoveForeignKey("childId", TableName("roles")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(RoleParents{}).AddForeignKey("childId", TableName("roles")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(RoleParents{}).RemoveForeignKey("parentId", TableName("roles")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(RoleParents{}).AddForeignKey("parentId", TableName("roles")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(PermissionRoles{}).RemoveForeignKey("roleId", TableName("roles")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(PermissionRoles{}).AddForeignKey("roleId", TableName("roles")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(PermissionUsers{}).RemoveForeignKey("permissionId", TableName("permissions")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(PermissionUsers{}).AddForeignKey("permissionId", TableName("permissions")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(PermissionRoles{}).RemoveForeignKey("permissionId", TableName("permissions")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(PermissionRoles{}).AddForeignKey("permissionId", TableName("permissions")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		err = _db.Model(UserAPIKeyPermissions{}).RemoveForeignKey("permissionId", TableName("permissions")+"(id)").Error
		if err != nil {
			log.Err(err).Send()
		}
		err = _db.Model(UserAPIKeyPermissions{}).AddForeignKey("permissionId", TableName("permissions")+"(id)", "CASCADE", "CASCADE").Error
		if err != nil {
			log.Err(err).Send()
		}

		if _db.Error != nil {
			log.Err(_db.Error).Send()
		}
	}
	return nil // _db.Error
}