package model

import (
	"aps_crm/conf"
	"aps_crm/pkg/logx"
	"aps_crm/pkg/mysqlx"
	"fmt"
	"gorm.io/gorm"
)

type Mysql struct{}

func NewMysql() *Mysql {
	return &Mysql{}
}

func (slf *Mysql) CreateUser(userName string, password string, database string) error {
	db := mysqlx.GetDB()
	err := db.Transaction(func(tx *gorm.DB) error {
		userSql := fmt.Sprintf("create user if not exists %v@'%v' identified by '%v'", userName, conf.Conf.Mysql.Host, password)
		privilegeSql := fmt.Sprintf("grant all privileges on %v.* to %v@'%v'", database, userName, conf.Conf.Mysql.Host)
		flushSql := "flush privileges"

		err := tx.Exec(userSql).Error
		if err != nil {
			logx.Errorf("create mysql-user err:%v", err)
			return err
		}

		err = tx.Exec(privilegeSql).Error
		if err != nil {
			logx.Errorf("grant mysql-privileges err:%v", err)
			return err
		}

		err = tx.Exec(flushSql).Error
		if err != nil {
			logx.Errorf("flush mysql-privilege err:%v", err)
			return err
		}
		return nil
	})
	return err
}

func (slf *Mysql) CreateDatabase(database string) error {
	db := mysqlx.GetDB()
	//sql := fmt.Sprintf("create database if not exists %v default charset utf8mb4 collate utf8mb4_general_ci", database)
	sql := fmt.Sprintf("CREATE DATABASE IF NOT EXISTS `%s` DEFAULT CHARACTER SET utf8mb4 DEFAULT COLLATE utf8mb4_general_ci;", database)
	err := db.Exec(sql).Error
	if err != nil {
		logx.Errorf("create database err:%v", err)
		return err
	}
	return nil
}