package service

import (
	"context"
	"fmt"
	"os"
	"os/exec"
	"path/filepath"
	"time"
	"vamicro/extend/util"
	"vamicro/version-control/models"

	"basic.com/valib/logger.git"
)

/*
	系统备份、备份还原、按设置周期性备份
	备份的数据主要包括sqlite数据、config以及exe(按需要)
	按钮可以触发立即备份
	参数:
		1.备份开关
		2.备份目录文件夹(/opt/vasystem/backup_auto)
		3.备份周期(每7天备份一次,备份时间是夜间0点)
		4.备份数据保存时间
*/
func BakSchedule(ctx context.Context) {
	// tk := time.NewTicker(time.Minute)
	tk := time.Tick(24 * time.Hour)
	for {
		select {
		case <-ctx.Done():
			return
		case <-tk:
			removeCheck()
			backupCheck()
		}
	}
}

//自动清理大于保存时间的备份数据
func removeCheck() {
	var ab models.AutoBackupConf
	i, e := ab.Select()
	if e != nil || i <= 0 {
		return
	}
	if ab.Dir == "" {
		logger.Error("备份目录未配置")
		return
	}

	if ab.Period < 1 || ab.SaveDays < 1 {
		logger.Error("周期和保存天数不合法")
		return
	}

	t := time.Now()
	err := filepath.Walk(ab.Dir, func(src string, f os.FileInfo, err error) error {
		if f.IsDir() {
			nt, e := time.ParseInLocation("2006-01-02-15-04-05", f.Name(), time.Local)
			if e != nil {
				return e
			}

			if int(t.Sub(nt).Hours()) > ab.SaveDays*24 {
				os.RemoveAll(ab.Dir + "/" + f.Name())
			}
		}
		return nil
	})
	if err != nil {
		logger.Error(err)
	}
}

func backupCheck() {
	//检查当前时间
	t := time.Now()
	if t.Hour() > 1 {
		return
	}
	var ab models.AutoBackupConf
	i, e := ab.Select()
	if e != nil || i <= 0 {
		return
	}
	if !ab.Enable {
		logger.Info("自动备份开关未打开")
		return
	}
	if ab.Dir == "" {
		logger.Error("备份目录未配置")
		return
	}
	if ab.Period < 1 || ab.SaveDays < 1 {
		logger.Error("周期和保存天数不合法")
		return
	}

	if !util.DirExists(ab.Dir) {
		os.Mkdir(ab.Dir, os.ModePerm)
	}
	lbd, err := exec.Command(fmt.Sprintf("ls %s | sort -r|head -1", ab.Dir)).Output()
	if err != nil {
		logger.Error("get last auto backup day err:", err)
		return
	}

	sbd := string(lbd)
	if sbd == "" { //未备份过，则执行备份
		DoAutoBackup(&ab)
	} else {
		tl, _ := time.ParseInLocation("2006-01-02-15-04-05", sbd, time.Local)
		if int(t.Sub(tl).Hours()) >= ab.Period*24 {
			DoAutoBackup(&ab)
		}
	}
}

func DoAutoBackup(ab *models.AutoBackupConf) error {
	backDirName := time.Now().Format("2006-01-02-15-04-05")
	dir := ab.Dir + "/" + backDirName
	if !util.DirExists(dir) {
		if err := os.Mkdir(dir, os.ModePerm); err != nil {
			return err
		}
	}
	//备份数据库和配置文件
	_, err := util.CopyDirByCmd("/opt/vasystem/config", dir)
	if err != nil {
		return err
	}
	return nil
}
func DoBackupRecover(dir string) error {
	//防止还原失败，做一个还原点备份
	//先停止所有进程
	//还原
	//重启系统
	recoverBakDir := util.GetVamicroPath() + "/recover_backup"
	if util.DirExists(recoverBakDir) {
		os.RemoveAll(recoverBakDir)
	}
	os.Mkdir(recoverBakDir, os.ModePerm)
	util.CopyDirByCmd("/opt/vasystem/config", recoverBakDir)

	_, err := exec.Command("/bin/sh", "-c", fmt.Sprintf(`
		/opt/vasystem/procs/killAllProcs.sh && 
		cp -rf %s/* /opt/vasystem/config/ && 
		/opt/vasystem/daemon.sh`, dir)).Output()
	return err
}