package service

import (
	"aps_crm/conf"
	"aps_crm/pkg/logx"
	"aps_crm/utils"
	"context"
	"encoding/json"
	"errors"
	"os"
	"os/exec"
	"time"
)

// DockerImpl 平台下的用户容器管理
type DockerImpl struct{}

func (slf *DockerImpl) Image() string {
	return "/var/data/aps/docker/image"
}

func (slf *DockerImpl) GetContainerRoot() string {
	return "/var/data/aps/docker"
}

// Init 初始化(创建容器，并启动)
func (slf *DockerImpl) Init(cid string) error {
	uDir := slf.GetContainerRoot() + "/" + cid
	//if !utils.DirExists(uDir + "/db") {
	if !utils.DirExists(uDir) {
		err := os.MkdirAll(uDir, os.ModePerm)
		if err != nil {
			logx.Errorf("MkdirAll err:", err)
			return err
		}
	}
	var err error
	defer func() {
		if err != nil {
			logx.Errorf("docker init defer err:", err)
			os.RemoveAll(uDir)
		}
	}()

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

type ContainerDockerInfo struct {
	Name  string `json:"Name"`
	Image string `json:"Image"`
	State struct {
		Status  string `json:"SalesReturnStatus"`
		Running bool   `json:"Running"`
	} `json:"State"`
}

// Exist 判断容器是否存在
func (slf *DockerImpl) Exist(cid string) (*ContainerDockerInfo, error) {
	sudoPwd := conf.Conf.System.SudoPassword
	if sudoPwd == "" {
		return nil, errors.New("sudoPassword 配置有误")
	}
	sh := "echo " + sudoPwd + "|sudo -S docker inspect " + cid
	cmd := exec.Command("sh", "-c", sh)
	ret, _ := cmd.Output()
	//当前cid容器不存在时，会报exit status 1错误并且 ret是空数组。所以不再判断Output的输出Error
	var cdi []ContainerDockerInfo
	e := json.Unmarshal(ret, &cdi)
	if e != nil {
		logx.Errorf("unmarshal e:", e)
		return nil, e
	}
	if len(cdi) == 1 {
		return &cdi[0], nil
	}

	return nil, ContainerNotFound
}

// Start 启动容器
func (slf *DockerImpl) Start(cid string) error {
	sudoPwd := conf.Conf.System.SudoPassword
	if sudoPwd == "" {
		return errors.New("sudoPassword 配置有误")
	}

	// 判断容器是否存在
	cdi, ce := slf.Exist(cid)
	logx.Infof("slf.Exist containerInfo:", cdi, " ce:", ce)
	if ce != nil { // 容器不存在
		uhome := slf.GetContainerRoot() + "/" + cid
		//if !utils.DirExists(uhome + "/db")
		if !utils.DirExists(uhome) {
			//err := os.MkdirAll(uhome+"/db", os.ModePerm)
			err := os.MkdirAll(uhome, os.ModePerm)
			if err != nil {
				logx.Errorf("docker Init err:", err)
				return err
			}
		}
		//sh := "echo " + sudoPwd + "|sudo -S docker run -d --restart=always --net host --name " + cid + " -v " + uhome + "/db:/saas/db saasuser:latest start.sh " + cid
		sh := "echo " + sudoPwd + "|sudo -S docker run -d --restart=always --net host --name " + cid + " aps-admin:latest /bin/bash /aps/start.sh " + cid
		cmd := exec.Command("sh", "-c", sh)
		ret, e := cmd.Output()
		if e != nil {
			return e
		}
		logx.Infof("docker start container:", cid, " ret:", ret)
		return nil
	}

	if cdi.State.Running { // 容器存在，运行状态
		return nil
	}

	// 容器存在，停止状态
	{
		sh := "echo " + sudoPwd + "|sudo -S docker start " + cid
		cmd := exec.Command("sh", "-c", sh)
		ret, e := cmd.Output()
		if e != nil {
			return e
		}
		logx.Infof("docker start container:", cid, " ret:", ret)
		return nil
	}
}

// Stop 停止容器
func (slf *DockerImpl) Stop(cid string) error {
	sudoPwd := conf.Conf.System.SudoPassword
	if sudoPwd == "" {
		return errors.New("sudoPassword 配置有误")
	}
	sh := "echo " + sudoPwd + "|sudo -S docker stop " + cid + ""
	cmd := exec.Command("sh", "-c", sh)
	ret, e := cmd.Output()
	if e != nil {
		return e
	}
	logx.Infof("docker stop container:", cid, " ret:", ret)
	return nil
}

// Restart 重启容器
func (slf *DockerImpl) Restart(cid string) error {
	sudoPwd := conf.Conf.System.SudoPassword
	if sudoPwd == "" {
		return errors.New("sudoPassword 配置有误")
	}
	sh := "echo " + sudoPwd + "|sudo -S docker restart " + cid + ""
	cmd := exec.Command("sh", "-c", sh)
	ret, e := cmd.Output()
	if e != nil {
		return e
	}
	logx.Infof("docker restart container:", cid, " ret:", ret)
	return nil
}

// BakData 备份数据
func (slf *DockerImpl) BakData(cid string) error {
	sudoPwd := conf.Conf.System.SudoPassword
	if sudoPwd == "" {
		return errors.New("sudoPassword 配置有误")
	}
	bdir := slf.GetContainerRoot() + "/bak"
	if !utils.DirExists(bdir) {
		err := os.MkdirAll(bdir, os.ModePerm)
		if err != nil {
			return err
		}
	}
	sh := "echo " + sudoPwd + "|sudo -S cp -r " + slf.GetContainerRoot() + "/" + cid + " " + bdir
	cmd := exec.Command("sh", "-c", sh)
	ret, e := cmd.Output()
	if e != nil {
		return e
	}
	logx.Infof("docker Bake container:", cid, " ret:", ret)
	return nil
}

// Destroy 销毁容器包括数据(销毁前先备份)
func (slf *DockerImpl) Destroy(cid string) error {
	sudoPwd := conf.Conf.System.SudoPassword
	if sudoPwd == "" {
		return errors.New("sudoPassword 配置有误")
	}
	//删除前先备份数据
	if err := slf.BakData(cid); err != nil {
		return err
	}
	sh := "echo " + sudoPwd + "|sudo -S docker rm -f " + cid + ""
	cmd := exec.Command("sh", "-c", sh)
	ret, e := cmd.Output()
	if e != nil {
		return e
	}
	logx.Infof("docker rm container:", cid, " ret:", ret)

	dsh := "echo " + sudoPwd + "|sudo -S rm -rf " + slf.GetContainerRoot() + "/" + cid
	dmd := exec.Command("sh", "-c", dsh)
	dr, de := dmd.Output()
	if de != nil {
		return de
	}
	logx.Infof("rm -rf volume: ", slf.GetContainerRoot()+"/"+cid, " dret:", string(dr))

	return nil
}

// 监控是否有未启动的用户容器
func (slf *DockerImpl) watch() {
	ctx := context.Background()
	for {
		select {
		case <-ctx.Done():
			return
		default:
			time.Sleep(60 * time.Second)
			//slf.initContainerNotFound()
		}
	}
}

func (slf *DockerImpl) initContainerNotFound() {
	//users, _ := model.NewUserSearch(nil).FindNotTotal()
	//for _, u := range users {
	//	if u.ParentId == "" { //是主账号
	//		//判断容器是否存在时是否需要加锁？用户注册主账号时是异步请求，创建docker容器没有冲突，因为容器的id不同。
	//		//拉起容器和用户主动启动容器也没有时间冲突
	//		_, e := slf.Exist(u.UUID)
	//		if e != ContainerNotFound {
	//			continue
	//		}
	//		if err := slf.Init(u.UUID); err != nil {
	//			logx.Errorf("init ContainerNotFound err:", err)
	//		}
	//	}
	//}
}