package db_docker

import (
	"sync"

	xormadapter "github.com/casbin/xorm-adapter/v2"
	_ "github.com/go-sql-driver/mysql"
	"github.com/pingcap/log"
	"go.uber.org/zap"

	"github.com/pingcap-inc/clinic/pkg/config"
	"github.com/pingcap-inc/clinic/pkg/db"
	"github.com/pingcap-inc/clinic/pkg/db_docker/db/iface"
	"github.com/pingcap-inc/clinic/pkg/db_docker/db/impl"
)

var (
	DockerDB *iface.DockerDB
	// clinicDB       *gorm.DB
	centralInited = false
	lck           sync.Mutex
)

// TryInitCentralDB will start a docker DB and init that database
func TryInitDB() error {
	lck.Lock()
	defer lck.Unlock()

	if centralInited {
		return nil
	}

	if err := tryStartDB(); err != nil {
		return err
	}

	dbcfg := &config.DBConfig{
		User:     "root", // centralDockerDB.User,
		Host:     DockerDB.Host,
		Port:     DockerDB.Port,
		Password: DockerDB.Password,
		Database: "unittest",
	}

	err := db.CreateDBIfAbsent(dbcfg)
	if err != nil {
		return err
	}

	_, err = xormadapter.NewAdapter("mysql", dbcfg.DSN(), true)
	if err != nil {
		return err
	}

	err = db.Initialize(dbcfg, false)
	if err != nil {
		return err
	}

	if err := db.GetMigrationRecords()[0].Migrate(db.GlobalDB()); err != nil {
		return err
	}

	centralInited = true
	return nil
}

func tryStartDB() error {
	if DockerDB == nil {
		d, err := impl.BorrowDB()
		if err != nil {
			log.Error("failed to start DB", zap.Error(err))
			return err
		}
		DockerDB = d
	}
	return nil
}