package xgorm

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

type DatabaseConfig struct {
	Host         string
	Port         int
	Dbname       string
	Charset      string
	User         string
	Password     string
	MaxOpenConn  int
	MaxIdleConn  int
	ReadTimeout  int
	WriteTimeout int
	Timeout      int
	Debug        bool
}

func (c *DatabaseConfig) dsn() string {
	dsn := "%s:%s@tcp(%s:%d)/%s?timeout=%ds&readTimeout=%ds&writeTimeout=%ds&parseTime=true&loc=Local&charset=%s"
	return fmt.Sprintf(dsn, c.User, c.Password, c.Host, c.Port, c.Dbname, c.Timeout, c.ReadTimeout, c.WriteTimeout, c.Charset)
}

func NewMysqlDB(cfg *DatabaseConfig) (*gorm.DB, error) {
	db, err := gorm.Open(mysql.Open(cfg.dsn()), &gorm.Config{})
	if err != nil {
		return nil, err
	}
	if cfg.Debug {
		db = db.Debug()
	}
	client, err := db.DB()
	if err != nil {
		return nil, err
	}

	client.SetMaxOpenConns(cfg.MaxOpenConn)
	client.SetMaxIdleConns(cfg.MaxIdleConn)
	return db, nil
}