package service

import (
	"basic.com/valib/bhomedbapi.git"
	"basic.com/valib/licence.git"
	"basic.com/valib/logger.git"
	"encoding/json"
	"errors"
	"github.com/mitchellh/mapstructure"
	"io/ioutil"
	"os"
	"path"
	"strings"
	"sync"
	"time"
	"vamicro/config"
	"vamicro/extend/util"
)

//安装依赖
func installDepend(insDir string) error {
	depDir := insDir + "/toolkits"
	depDef := depDir + "/depend_enc.def"
	if util.DirExists(depDir) && util.FileExists(depDef) {
		depB, err := util.CallDecFileContent(depDef)
		if err != nil {
			logger.Error("CallDecFileContent depDef err:", err)
			return err
		}
		depArr := strings.Split(string(depB), ":")
		for _, depName := range depArr {
			if util.DirExists(depDir+"/"+depName) && util.DirExists("/opt/toolkits") && !util.DirExists("/opt/toolkits/"+depName) {
				_, err = util.CopyDirByCmd(depDir+"/"+depName, "/opt/toolkits/")
				if err != nil {
					logger.Error("copy dep ", depName, " err:", err)
					continue
				}
			}
		}
	}
	return nil
}

type sdkConfig struct {
	RunTime string `json:"runtime"`
}

//动态监测算法依赖文件是否存在，不存在则尝试在线下载
const binPath = "/opt/vasystem/bin"

var depInstallingMap sync.Map

func downloadDepend(sdks []SdkIns) error {
	for _, s := range sdks {
		sdkTyp := s.SdkType
		jsf := binPath + "/zconf/" + sdkTyp + ".json"
		if util.FileExists(jsf) {
			fData, err := ioutil.ReadFile(jsf)
			if err != nil {
				logger.Error("open ", sdkTyp, " config err:", err)
				continue
			}
			var sc sdkConfig
			err = json.Unmarshal(fData, &sc)
			if err != nil {
				logger.Error("unmarshal sdkConfig err:", err)
				continue
			}
			depArr := strings.Split(sc.RunTime, ":")
			for _, dep := range depArr {
				if dep == "/opt/vasystem/libs/"+sdkTyp {
					continue
				}
				if strings.HasPrefix(dep, "/opt/toolkits/") {
					tmp := strings.Replace(dep, "/opt/toolkits", "", -1)
					idx := strings.Index(tmp, "/")
					if idx > 0 {
						depName := tmp[:idx]
						if util.DirExists("/opt/toolkits") && !util.DirExists("/opt/toolkits/"+depName) {
							_, ok := depInstallingMap.Load(depName)
							if !ok {
								go depOnline(depName)
							}
						}
					}
				}
			}
		}
	}
	return nil
}

//在线下载依赖
func depOnline(depName string) error {
	depInstallingMap.Store(depName, depName)
	defer depInstallingMap.Delete(depName)
	url := "http://" + util.GetShopUrl() + "/data/api-s/sdk/depend"
	machineCode := licence.GetMachineCode()
	if machineCode == "" {
		logger.Debug("depOnline 获取机器码失败")
		return errors.New("获取机器码失败")
	}
	paramBody := map[string]interface{}{
		"depName":     depName,
		"machineCode": machineCode,
		"serverId":    config.Server.AnalyServerId,
	}
	header := map[string]string{
		"Authorization": token,
	}
	respBody, err := util.DoPostRequest(url, util.CONTENT_TYPE_JSON, paramBody, nil, header, time.Second*60)
	if err != nil {
		logger.Debug("DoPostRequest err:", err)
		return err
	}
	var res bhomedbapi.Result
	if err = json.Unmarshal(respBody, &res); err != nil {
		logger.Debug("unmarshal err:", err)
		return err
	}
	if !res.Success {
		logger.Debug("res.Data:", res.Data)
		return errors.New("请求商城失败")
	}
	logger.Debug("res.Data:", res.Data)
	var resp downOrUpResp
	if err := mapstructure.Decode(res.Data.(map[string]interface{}), &resp); err != nil {
		logger.Debug("mapstructure.Decode err:", err)
		return err
	}
	logger.Debug("resp:", resp)
	if resp.Url == "" || resp.Md5 == "" {
		return errors.New("获取依赖包下载地址失败")
	}
	logger.Info("依赖项:", depName, " 在线安装包大小:", resp.Size)
	configPatchPath := ""
	if config.Server.PatchPath != "" {
		configPatchPath = config.Server.PatchPath
	} else {
		configPatchPath = "/opt/vasystem/patch"
	}
	if !util.DirExists(configPatchPath) {
		os.Mkdir(configPatchPath, 0777)
	}
	filenameWithSuffix := path.Base(resp.Url)
	ext := path.Ext(filenameWithSuffix)
	gzFilePath := configPatchPath + "/" + resp.Md5 + ext

	ip := &InsProgress{
		Status:   InsStatus_Downloading,
		Progress: 0,
		Size:     resp.Size,
	}
	defer func() {
		if util.FileExists(gzFilePath) {
			os.Remove(gzFilePath)
		}
	}()
	if !util.FileExists(gzFilePath) {
		err := DownloadFile(gzFilePath, resp.Url, ip)
		if err != nil {
			logger.Debug("DownloadFile err:", err)
			return err
		}
		rmd5, err := util.FileMd5(gzFilePath)
		if err != nil {
			logger.Error("FileMd5 err:", err)
			return err
		}
		if rmd5 != resp.Md5 {
			err = errors.New("下载的依赖文件校验md5失败，请重新下载")
			logger.Error(err)
			return err
		}
	}
	//解压安装依赖
	depPath := "/opt/toolkits/" + depName
	if !util.DirExists(depPath) {
		os.Mkdir(configPatchPath, os.ModePerm)
	}
	out, err := util.UnTarGzByCmd(gzFilePath, depPath)
	logger.Info("UnTarGzByCmd depName:", depName, " err:", err, " out:", out)
	if err != nil {
		os.RemoveAll(depPath)
		return err
	}

	return nil
}