package docker_db

import (
	"context"
	"fmt"
	"sync"

	"gorm.io/driver/mysql"

	"github.com/tidbcloud/account/internel/svc/migrate"

	_ "github.com/go-sql-driver/mysql"
	"github.com/tidbcloud/account/commons/config"
	"github.com/tidbcloud/account/commons/database"
	ddb "github.com/tidbcloud/account/pkg/docker_db/db"
	"github.com/tidbcloud/account/pkg/docker_db/db/iface"
	"github.com/tidbcloud/tiinf/monitor/tilog"

	"go.uber.org/zap"
	"gorm.io/gorm"
	"k8s.io/utils/pointer"
)

var (
	centralDockerDB *iface.DockerDB
	centralDB       *gorm.DB
	centralInited   = false
	lck             sync.Mutex
)

// TryInitCentralDB will start a docker DB and init that database
func TryInitCentralDB(migrateFlag bool) (*gorm.DB, error) {
	config.Global().ServerConfig.ManagementPortalInitUserEmail = pointer.StringPtr("cloud@pingcap.com")
	lck.Lock()
	defer lck.Unlock()

	if centralInited {
		return centralDB, nil
	}

	if err := tryStartGlobalCentralDB(); err != nil {
		return nil, err
	}
	centralDB, err := database.Initialize(context.Background(), config.MysqlConfig{
		User:                     "root", // centralDockerDB.User,
		Host:                     centralDockerDB.Host,
		Port:                     centralDockerDB.Port,
		Password:                 centralDockerDB.Password,
		Database:                 "unittest",
		DisableNestedTransaction: true,
	})
	if err != nil {
		return nil, err
	}
	if migrateFlag {
		if err = migrate.RunCentralMigrate(centralDB); err != nil {
			if err != nil {
				return nil, err
			}
		}
	}
	centralInited = true
	return centralDB, nil
}

func tryStartGlobalCentralDB() error {
	if centralDockerDB == nil {
		d, err := ddb.BorrowDB()
		if err != nil {
			tilog.Error(context.Background(), "failed to start DB", zap.Error(err))
			return err
		}
		centralDockerDB = d
	}
	return nil
}

func InitDockerGormDB(migrateFlag bool) (string, *gorm.DB, error) {
	config.Global().ServerConfig.ManagementPortalInitUserEmail = pointer.StringPtr("cloud@pingcap.com")
	lck.Lock()
	defer lck.Unlock()

	dockerDb, err := ddb.BorrowDB()
	if err != nil {
		tilog.Error(context.Background(), "failed to start DB", zap.Error(err))
		return "", nil, err
	}

	dsn := fmt.Sprintf("%s:%s@(%s:%d)/%s?%s", "root", dockerDb.Password, dockerDb.Host, dockerDb.Port, "test", "charset=utf8mb4&parseTime=True&loc=UTC")
	db, err := gorm.Open(mysql.Open(dsn), nil)
	if err != nil {
		return "", nil, err
	}
	if migrateFlag {
		if err = migrate.RunCentralMigrate(db); err != nil {
			if err != nil {
				return "", nil, err
			}
		}
	}
	return dockerDb.ID, db, nil
}

func ReleaseDockerDB(id string) {
	ddb.ReleaseDB(id)
}