// Tool Url: https://github.com/go-gorm/gorm
// Tool Guide: https://gorm.io/docs/

package database

import (
	"context"
	"fmt"
	"os"
	"time"

	"github.com/tidbcloud/auth-server/commons/config"

	"github.com/tidbcloud/tiinf/monitor/tilog"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
)

// Mysql handler infomation
type MysqlInfo struct {
	DB *gorm.DB
}

var DBConn = &MysqlInfo{}

func Connect(ctx context.Context, centralConfig *config.CentralConfig) {
	password := os.Getenv("DB_PASSWORD")
	if password == "" {
		panic("DB_PASSWORD is blank")
	}

	// Params
	url := fmt.Sprintf("%s:%s@tcp(%s:%v)/%s?charset=%s&parseTime=True&loc=%s&collation=%s",
		centralConfig.MysqlConfig.User,
		password,
		centralConfig.MysqlConfig.Host,
		centralConfig.MysqlConfig.Port,
		centralConfig.MysqlConfig.Database,
		centralConfig.MysqlConfig.CharSet,
		centralConfig.MysqlConfig.TimeZone,
		centralConfig.MysqlConfig.Collation,
	)
	tilog.Info(ctx, "db connection begin!")

	var logLevel logger.LogLevel

	if config.IsLocalEnv() || config.IsDevEnv() {
		// will print sql int console
		logLevel = logger.Info
	} else {
		logLevel = logger.Error
	}

	logger := logger.Default.LogMode(logLevel)
	// Connect
	conn, err := gorm.Open(mysql.Open(url), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			TablePrefix: "auth_server_",
		},
		Logger: logger,
	})
	if err != nil {
		panic(err.Error())
	}
	sqlDB, err := conn.DB()
	if err != nil {
		panic(err.Error())
	}
	sqlDB.SetMaxIdleConns(20)
	sqlDB.SetMaxOpenConns(100)
	sqlDB.SetConnMaxLifetime(time.Second * 600)

	DBConn.DB = conn
}