package service

import (
	"aps_crm/pkg/logx"
	"aps_crm/utils"
	"errors"
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
)

/*
	管理dir类型的container的生命周期的管理器
*/

type DirContainerImpl struct{}

// Image 获取容器化用户一整套运行环境的基础目录
func (slf *DirContainerImpl) Image() string {
	return "/var/data/aps/sys"
}

// GetContainerRoot 获取容器目录的根路径地址
func (slf *DirContainerImpl) GetContainerRoot() string {
	return "/var/data/aps/container"
}

/*
	将生成的用户id写入到新分配的container(配置文件)中，container中所有进程启动后注册到etcd

时，内部服务的key都包含有此主账号的id。container内部中的服务调用时，只会调用到内部注册有此用户id的
key的服务，所有container对外部来说是隔离的。
*/
func (slf *DirContainerImpl) Init(cId string) error {
	logx.Infof("DirContainerImpl Init cId:", cId)
	root := slf.GetContainerRoot()
	uDir := root + "/" + cId
	if !utils.DirExists(uDir) {
		if err := os.MkdirAll(uDir, os.ModePerm); err != nil {
			logx.Errorf("mkdir container Root err:", err)
			return err
		}
	} else {
		logx.Errorf("container ", cId, " exist")
		return errors.New("container " + cId + " exist")
	}
	if !utils.DirExists(slf.Image()) {
		logx.Errorf("saas 系统运行环境文件不存在,请检查")
		return errors.New("saas 系统运行环境文件不存在,请检查")
	}
	var err error
	defer func() {
		if err != nil {
			logx.Errorf("rollback err:", err)
			os.RemoveAll(uDir)
		}
	}()
	_, err = utils.CopyDirByCmd(slf.Image()+"/*", uDir)
	if err != nil {
		logx.Errorf("CopyDirByCmd err:", err)
		return err
	}
	//将创建容器的主账号id写入到容器中
	err = ioutil.WriteFile(uDir+"/bin/container.txt", []byte(cId), os.ModePerm)
	if err != nil {
		logx.Errorf("ioutil.WriteFile err:", err)
		return err
	}
	//配置开机启动

	//启动容器
	err = slf.Start(cId)
	if err != nil {
		logx.Errorf("start container err:", err, " id:", cId)
		return err
	}

	return nil
}

// Start 启动指定的用户容器
func (slf *DirContainerImpl) Start(cId string) error {
	uroot := slf.GetContainerRoot() + "/" + cId
	sh := fmt.Sprintf("%s %s &", uroot+"/start.sh", uroot)
	cmd := exec.Command("sh", "-C", sh)
	ret, e := cmd.Output()
	if e != nil {
		return e
	}
	logx.Infof("start container "+cId+", ret: ", ret)
	return nil
}

// Stop 停止指定的用户容器
func (slf *DirContainerImpl) Stop(cId string) error {
	uroot := slf.GetContainerRoot() + "/" + cId
	sh := fmt.Sprintf("%s %s", uroot+"/stop.sh", uroot)
	cmd := exec.Command("sh", "-C", sh)
	ret, e := cmd.Output()
	if e != nil {
		return e
	}
	logx.Infof("stop container "+cId+", ret: ", ret)
	return nil
}

// Restart 重新启动指定的用户容器
func (slf *DirContainerImpl) Restart(cId string) error {
	err := slf.Stop(cId)
	if err != nil {
		return err
	}
	err = slf.Start(cId)
	if err != nil {
		return err
	}
	return nil
}

// BakData 备份容器中的用户数据
func (slf *DirContainerImpl) BakData(cId string) error {
	bdir := slf.GetContainerRoot() + "/bak"
	if !utils.DirExists(bdir) {
		err := os.MkdirAll(bdir, os.ModePerm)
		if err != nil {
			return err
		}
	}
	ret, err := utils.CopyDirByCmd(slf.GetContainerRoot()+"/"+cId, bdir)
	if err != nil {
		return err
	}
	logx.Infof("Bake container:", cId, " ret:", ret)
	return nil
}

// Destroy 删除指定的用户容器
func (slf *DirContainerImpl) Destroy(cId string) error {
	if err := slf.BakData(cId); err != nil {
		return err
	}
	err := slf.Stop(cId)
	if err != nil {
		return err
	}
	uroot := slf.GetContainerRoot() + "/" + cId
	if utils.DirExists(uroot) {
		os.RemoveAll(uroot)
	}
	return nil
}

func (slf *DirContainerImpl) watch() {
	return
}