package mysql

import (
	"database/sql"
	. "gitee.com/kristas/booting-go/framework/bean/bean_factory"
	appConfig "gitee.com/kristas/booting-go/framework/config"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	. "gorm.io/gorm/logger"
	"reflect"
	"strings"
)

func InitGormDatasource() {
	InitDatasource(handleFacade)
}

func handleFacade(config DataSource, db *sql.DB) {
	c := GetBean("app_config").(*appConfig.Config)
	gormDB, err := gorm.Open(mysql.New(mysql.Config{
		Conn: db,
	}), &gorm.Config{
		Logger: Default.LogMode(gormLogLevelAdapter(c.Logging.Level)),
	})
	if err != nil {
		log.Error("Failed to facade db to gormDB, caused by:", err)
	} else {
		log.Infof("Successfully connected to db %s", config.Tag)
	}
	gormBean := GetBean(config.Tag)
	reflect.ValueOf(gormBean).Elem().Set(
		reflect.ValueOf(gormDB).Elem())
}

func gormLogLevelAdapter(level string) LogLevel {
	switch strings.ToLower(level) {
	case "panic":
		return Silent
	case "fatal", "error":
		return Error
	case "warn", "warning":
		return Warn
	default:
		return Info
	}
}