package cmd

import (
	"github.com/spf13/cobra"
	"io/ioutil"
	"kefu/common"
	"kefu/models"
	"kefu/tools"
	"log"
	"os"
	"strings"
	"time"
)

var installCmd = &cobra.Command{
	Use:   "install",
	Short: "安装导入数据",
	Run: func(cmd *cobra.Command, args []string) {
		install()
	},
}

func install() {
	if ok, _ := tools.IsFileNotExist("./install.lock"); !ok {
		log.Println("请先删除./install.lock")
		os.Exit(1)
	}
	sqlFile := "import.sql"
	isExit, _ := tools.IsFileExist(common.MysqlConf)
	dataExit, _ := tools.IsFileExist(sqlFile)
	if !isExit || !dataExit {
		log.Println("config/mysql.json 数据库配置文件或者数据库文件import.sql不存在")
		os.Exit(1)
	}
	err := models.NewConnect(common.MysqlConf)
	if err != nil {
		log.Println(err.Error())
		os.Exit(1)
	}
	sqls, _ := ioutil.ReadFile(sqlFile)
	sqlArr := strings.Split(string(sqls), ";")
	for _, sql := range sqlArr {
		sql = strings.TrimSpace(sql)
		if sql == "" {
			continue
		}
		err := models.Execute(sql)
		if err == nil {
			log.Println(sql, "\t success!")
		} else {
			log.Println(sql, err, "\t failed!", "数据库导入失败")
			os.Exit(1)
		}
	}
	installFile, _ := os.OpenFile("./install.lock", os.O_RDWR|os.O_CREATE, os.ModePerm)

	installTime := tools.Int64ToByte(time.Now().Unix())
	token, err := tools.AesEncrypt(installTime, []byte(common.AesKey))
	if err != nil {
		log.Println("数据库导入失败：", err)
	}
	installFile.Write(token)
	log.Println("数据库导入成功!")
}