package mysql

import (
	"strings"

	"gorm.io/gorm"
	"zestack.dev/cast"
)

// func init() {
// 	db.RegisterDriver("mysql", func(opts *db.DriverOptions) db.Driver {
// 		return &Driver{
// 			Migrator: Migrator{
// 				DB:    opts.DB,
// 				Log:   opts.Log,
// 				Retry: opts.Retry,
// 			},
// 		}
// 	})
// }

type Driver struct {
	Migrator
}

func (*Driver) UseTenant(tx *gorm.DB, tenant uint) (reset func() error, err error) {
	return UseDatabase(tx, TenantDatabase(tenant))
}

func (*Driver) CurrentTenant(tx *gorm.DB) (tenant uint, ok bool) {
	databse := CurrentDatabase(tx)
	basename := TenantBasename()

	if databse != "" && strings.HasPrefix(databse, basename) {
		var err error
		tenant, err = cast.Uint(databse[len(basename):])
		ok = err == nil && tenant > 0
	}

	return
}

func createSharedSchema(tx *gorm.DB) error {
	database := PublicDatabase()
	sql := "CREATE DATABASE IF NOT EXISTS " + tx.Statement.Quote(database)
	return tx.Exec(sql).Error
}

func (d *Driver) CreateTenantSchema(tx *gorm.DB, tenantId uint) error {
	database := TenantDatabase(tenantId)
	sql := "CREATE DATABASE IF NOT EXISTS " + tx.Statement.Quote(database)
	return tx.Exec(sql).Error
}

func (d *Driver) DropSchemaForTenant(tenant uint) error {
	return d.DropDatabaseForTenant(tenant)
}