package dao

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"key-manager/conf"
)

type Dao struct {
	db *gorm.DB
}

func InitMysqlDB() (*Dao, error) {
	conf := conf.GetConfig()
	dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8&parseTime=True&loc=Local",
		conf.Mysql.User,
		conf.Mysql.Password,
		conf.Mysql.Host,
		conf.Mysql.DB,
	)

	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
		//Logger: logger.Default.LogMode(logger.Info),
		Logger: logger.Default.LogMode(logger.Silent),
	})
	if err != nil {
		return nil, err
	}

	db.AutoMigrate(&EthAddressInfo{})
	db.AutoMigrate(&TronAddressInfo{})
	db.AutoMigrate(&Mnemonic{})
	db.AutoMigrate(&Password{})
	db.AutoMigrate(&Whitelist{})
	db.AutoMigrate(&SignTx{})

	return &Dao{
		db: db,
	}, nil
}