package models

import (
	"errors"
	"fmt"
	"go-admin/common/global"
	"io/ioutil"
	"log"
	"os"
	"strings"

	"gorm.io/gorm"
)

func InitDb(db *gorm.DB) (err error) {
	filePath := "config/db.sql"
	if global.Driver == "postgres" {
		filePath := "config/db.sql"
		if err = ExecSql(db, filePath); err != nil {
			return err
		}
		filePath = "config/pg.sql"
		err = ExecSql(db, filePath)
	} else if global.Driver == "mysql" {
		filePath = "config/db-begin-mysql.sql"
		if err = ExecSql(db, filePath); err != nil {
			return err
		}
		filePath = "config/db.sql"
		if err = ExecSql(db, filePath); err != nil {
			return err
		}
		filePath = "config/db-end-mysql.sql"
		err = ExecSql(db, filePath)
	} else {
		err = ExecSql(db, filePath)
	}
	if err != nil {
		return err
	}
	return bootstrapAdmin(db, os.Getenv)
}

func bootstrapAdmin(db *gorm.DB, getenv func(string) string) error {
	username := strings.TrimSpace(getenv("BELL_BOOTSTRAP_USERNAME"))
	password := getenv("BELL_BOOTSTRAP_PASSWORD")
	if username == "" || password == "" {
		return errors.New("BELL_BOOTSTRAP_USERNAME and BELL_BOOTSTRAP_PASSWORD are required for the initial migration")
	}
	if len(password) < 8 {
		return errors.New("BELL_BOOTSTRAP_PASSWORD must contain at least 8 characters")
	}
	var count int64
	if err := db.Model(&SysUser{}).Where("username = ?", username).Count(&count).Error; err != nil {
		return err
	}
	if count != 0 {
		return fmt.Errorf("bootstrap user %q already exists", username)
	}
	return db.Create(&SysUser{
		Username: username,
		Password: password,
		NickName: "Bell 管理员",
		RoleId:   1,
		DeptId:   1,
		PostId:   1,
		Status:   "2",
		ControlBy: ControlBy{
			CreateBy: 1,
			UpdateBy: 1,
		},
	}).Error
}

func ExecSql(db *gorm.DB, filePath string) error {
	sql, err := Ioutil(filePath)
	if err != nil {
		fmt.Println("数据库基础数据初始化脚本读取失败！原因:", err.Error())
		return err
	}
	sqlList := strings.Split(sql, ";")
	for i := 0; i < len(sqlList)-1; i++ {
		if strings.Contains(sqlList[i], "--") {
			fmt.Println(sqlList[i])
			continue
		}
		sql := strings.Replace(sqlList[i]+";", "\n", "", -1)
		sql = strings.TrimSpace(sql)
		if err = db.Exec(sql).Error; err != nil {
			log.Printf("error sql: %s", sql)
			if !strings.Contains(err.Error(), "Query was empty") {
				return err
			}
		}
	}
	return nil
}

func Ioutil(filePath string) (string, error) {
	if contents, err := ioutil.ReadFile(filePath); err == nil {
		//因为contents是[]byte类型，直接转换成string类型后会多一行空格,需要使用strings.Replace替换换行符
		result := strings.Replace(string(contents), "\n", "", 1)
		fmt.Println("Use ioutil.ReadFile to read a file:", result)
		return result, nil
	} else {
		return "", err
	}
}