package service

import (
	"basic.com/valib/logger.git"
	"errors"
	"fmt"
	"io"
	"net/http"
	"os"
	"path"
	"strconv"
	"strings"
	"sync"
	"vamicro/extend/util"
)

//正在安装列表
var insIngMap sync.Map

//判断是否在安装中
func IsInstalling(id string) bool {
	_, ok := insIngMap.Load(id)
	return ok
}

//安装进度
type InsProgress struct {
	Status   string `json:"status"`
	Progress int    `json:"progress"`
	Size     uint64 `json:"size"`
	IsDone   bool   `json:"isDone"` //不管成功或者失败，标志本次安装已完成
}

const (
	InsStatus_Downloading     = "下载中..."
	InsStatus_Err             = "安装包下载失败"
	InsStatus_UnPackErr       = "安装包解压失败"
	InsStatus_PackageChecking = "安装包校验中..."
	InsStatus_CheckErr        = "安装包校验失败"
	InsStatus_Installing      = "安装中..."
	InsStatus_InstallErr      = "安装失败"
	InsStatus_Done            = "已安装"
)

type WriteCounter struct {
	Total uint64

	iPgs *InsProgress
}

func (wc *WriteCounter) PrintProgress() {
	logger.Info(strings.Repeat(" ", 35))
	//logger.Info("Downloading... ",humanize.Bytes(wc.Total)," complete")

	percent := (wc.Total * 100) / wc.iPgs.Size

	wc.iPgs.Progress = int(percent)
}

func (wc *WriteCounter) Write(p []byte) (int, error) {
	n := len(p)
	wc.Total += uint64(n)
	wc.PrintProgress()
	return n, nil
}

//下载文件
func DownloadFile(fPath string, url string, iProgress *InsProgress) error {
	f, err := os.Create(fPath)
	if err != nil {
		return err
	}
	resp, err := http.Get(url)
	if err != nil {
		f.Close()
		return err
	}
	defer resp.Body.Close()
	counter := &WriteCounter{
		iPgs: iProgress,
	}
	if _, err = io.Copy(f, io.TeeReader(resp.Body, counter)); err != nil {
		f.Close()
		return err
	}
	fmt.Printf("\n")
	f.Close()
	//if err = os.Rename(fPath, fPath);err != nil {
	//	return err
	//}
	return nil
}

type downPara struct {
	URL      string //待下载文件的url
	FileName string //待下载文件的名称
}

//procPath 下载到目的文件夹
func downloadExtraFile(downloadDir string, extra downPara, fb func(length, downLen int64)) error {
	if extra.URL == "" {
		return nil
	}

	var (
		fsize   int64
		buf     = make([]byte, 32*1024)
		written int64
	)

	// 创建应用下载目录, 后续会直接剪切到编译环境
	//downloadDir := path.Join(procPath, "pre-install-sdks")
	//err := dirPing(downloadDir)
	//if err != nil {
	//	logger.Warn("pack error: can not create dir.", err)
	//	return errors.New("can not create dir")
	//}

	downloadPath := path.Join(downloadDir, extra.FileName)
	tmpFilePath := downloadPath + ".download"
	logger.Debug(tmpFilePath)
	//创建一个http client
	client := new(http.Client)
	//client.Timeout = time.Second * 60 //设置超时时间
	//get方法获取资源
	resp, err := client.Get(extra.URL)
	if err != nil {
		logger.Error("client.Get err:", err)
		return err
	}

	//读取服务器返回的文件大小
	fsize, err = strconv.ParseInt(resp.Header.Get("Content-Length"), 10, 32)
	if err != nil {
		logger.Error("读取服务器返回文件大小err: ", err)
	}

	if isFileExist(downloadPath, fsize) {
		return err
	}

	logger.Debug("fsize", fsize)
	//创建文件
	file, err := os.Create(tmpFilePath)
	if err != nil {
		logger.Error("os.Create err: ", err)
		return err
	}
	defer file.Close()
	if resp.Body == nil {
		return errors.New("body is null")
	}
	defer resp.Body.Close()
	//下面是 io.copyBuffer() 的简化版本
	for {
		//读取bytes
		nr, er := resp.Body.Read(buf)
		if nr > 0 {
			//写入bytes
			nw, ew := file.Write(buf[0:nr])
			//数据长度大于0
			if nw > 0 {
				written += int64(nw)
			}
			//写入出错
			if ew != nil {
				err = ew
				break
			}
			//读取是数据长度不等于写入的数据长度
			if nr != nw {
				err = io.ErrShortWrite
				break
			}
		}
		if er != nil {
			if er != io.EOF {
				err = er
			}
			break
		}
		//没有错误了快使用 callback
		if fb != nil {
			fb(fsize, written)
		}
	}
	logger.Debug(err)
	if err == nil {
		file.Close()
		err = os.Rename(tmpFilePath, downloadPath)
		logger.Debug(err)
	}
	return err
}

// 如果文件夹不存在就会创建
func dirPing(path string) error {
	if !util.DirExists(path) {
		e := os.MkdirAll(path, 00777)
		if e != nil {
			return e
		}
	}
	return nil
}

func isFileExist(filename string, filesize int64) bool {
	info, err := os.Stat(filename)
	if os.IsNotExist(err) {
		logger.Debug(info)
		return false
	}
	if filesize == info.Size() {
		logger.Debug("安装包已存在！", info.Name(), info.Size(), info.ModTime())
		return true
	}
	del := os.Remove(filename)
	if del != nil {
		logger.Error(del)
	}
	return false
}