package controller

import (
	"basic.com/valib/logger.git"
	"fmt"
	"github.com/gin-gonic/gin"
	"io/ioutil"
	"net/http"
	"os"
	"path/filepath"
	"vamicro/update-service/models"
	"vamicro/update-service/upgrade"
	"vamicro/update-service/utils"
)

func UploadFiles(c *gin.Context) {
	file, err := c.FormFile("upload")
	if nil != err {
		c.JSON(http.StatusBadRequest, gin.H {
			"code": http.StatusBadRequest,
			"msg": err.Error(),
			"data": &struct {
			}{},
		})
		logger.Error("UploadFiles upload field is not filled with a file, err:", err.Error())
		return
	}

	if err = c.SaveUploadedFile(file, file.Filename); err != nil {
		c.JSON(http.StatusBadRequest, gin.H {
			"code": http.StatusBadRequest,
			"msg": err.Error(),
			"data": &struct {
			}{},
		})
		logger.Error("UploadFiles save file failed, err:", err.Error())
		return
	}

	c.JSON(http.StatusOK, gin.H {
		"code": http.StatusOK,
		"msg": "ok",
		"data": &struct {
		}{},
	})
}

func Download(c *gin.Context) {
	path := c.Query("filename")
	if len(path) == 0 || !utils.PathExists(path) {
		msg := fmt.Sprintf("%s not exists", path)
		c.JSON(http.StatusBadRequest, gin.H {
			"code": http.StatusBadRequest,
			"msg": msg,
			"data": &struct {
			}{},
		})
		logger.Error("UploadApps upload field is not filled with a file, err:", msg)
		return
	}

	filename := filepath.Base(path)
	c.Writer.Header().Add("Content-Disposition", fmt.Sprintf("attachment; filename=%s", filename))
	c.Writer.Header().Add("Content-Type", "application/octet-stream")

	c.File(path)
}

//根据platform获取zip包的地址，以及所有二进制文件的属性（md5，版本号，commit sha1，buildTime）
func GetAppVersion(c *gin.Context) {
	arch := c.Query("arch")
	if !models.IsValidPlatform(arch) {
		c.JSON(http.StatusBadRequest, gin.H {
			"code": http.StatusBadRequest,
			"msg": "platform is unspecified or invalid",
			"data": &struct {
			}{},
		})
		return
	}

	programs, err := models.FindAllPrograms(arch)
	if err != nil {
		c.JSON(http.StatusOK, gin.H {
			"code": http.StatusInternalServerError,
			"msg": err.Error(),
			"data": &struct {
			}{},
		})
	}

	var dist models.DistFile
	files, err := dist.FindAll(arch)
	if err != nil  {
		c.JSON(http.StatusOK, gin.H {
			"code": http.StatusInternalServerError,
			"msg": err.Error(),
			"data": &struct {
			}{},
		})
	}
	if len(files) < 1 {
		c.JSON(http.StatusOK, gin.H {
			"code": http.StatusInternalServerError,
			"msg": "no version found",
			"data": &struct {
			}{},
		})
	}

	c.JSON(http.StatusOK, gin.H {
		"code": http.StatusOK,
		"msg": "ok",
		"data": map[string]interface{}{
			"archive":files[0].Path,
			"programs":programs,
			"version":files[0].Version,
			"intro":files[0].Intro,
		},
	})
}

//上传执行make命令生成的zip程序包（包含所有的二进制可执行文件）
func UploadApps(c *gin.Context) {
	arch := c.Query("arch")
	version := c.Query("version")
	inrto := c.Query("intro")
	if !models.IsValidPlatform(arch) {
		c.JSON(http.StatusBadRequest, gin.H {
			"code": http.StatusBadRequest,
			"msg": "platform is unspecified or invalid",
			"data": []*models.Program{},
		})
		return
	}

	file, err := c.FormFile("archive")
	if nil != err {
		c.JSON(http.StatusBadRequest, gin.H {
			"code": http.StatusBadRequest,
			"msg": err.Error(),
			"data": []*models.Program{},
		})
		logger.Error("UploadApps upload field is not filled with a file, err:", err.Error())
		return
	}

	tmpFile, err := ioutil.TempFile("", "dist-*.zip")
	if nil != err {
		c.JSON(http.StatusInternalServerError, gin.H {
			"code": http.StatusInternalServerError,
			"msg": err.Error(),
			"data": []*models.Program{},
		})
		logger.Error("UploadApps create temp file failed, err:", err.Error())
		return
	}
	defer func() {
		tmpFile.Close()
		os.Remove(tmpFile.Name())
	}()

	if err = c.SaveUploadedFile(file, tmpFile.Name()); err != nil {
		c.JSON(http.StatusInternalServerError, gin.H {
			"code": http.StatusInternalServerError,
			"msg": err.Error(),
			"data": []*models.Program{},
		})
		logger.Error("UploadApps save file failed, err:", err.Error())
		return
	}

	logger.Info("UploadApps saved to temp file ok:", tmpFile.Name())

	ch := make(chan *upgrade.UpgradeReply)
	defer close(ch)
	msg := &upgrade.UpgradePara{
		Platform:    arch,
		ArchiveFile: tmpFile,
		ReplyCh:     ch,
		Version:     version,
		Intro:       inrto,
	}
	upgrade.PUshMsg(msg)
	reply := <- ch

	logger.Info("UploadApps get response from routing:", tmpFile.Name())

	c.JSON(reply.Status, reply.Body)
}