package main

import (
	"embed"
	"flag"
	"fmt"
	"go_dreamfactory/cmd/upgrade/tools"
	"html/template"
	"io"
	"io/ioutil"
	"log"
	"net/http"
	"os"
	"path"
	"strconv"
	"strings"
	"time"

	"github.com/gin-gonic/gin"
)

// func main() {
// 	http.HandleFunc("/download", FileDownload)
// 	http.HandleFunc("/upload", FileUpload)

// 	log.Print("http server start")
// 	if err := http.ListenAndServe(":8910", nil); err != nil {
// 		log.Fatal(err)
// 	}
// }

func FileUpload(w http.ResponseWriter, r *http.Request) {
	//获取文件流,第三个返回值是错误对象
	file, header, _ := r.FormFile("file")
	//读取文件流为[]byte
	b, _ := ioutil.ReadAll(file)
	//把文件保存到指定位置
	ioutil.WriteFile("/opt/upgrade", b, 0777)
	//输出上传时文件名
	fmt.Println("上传文件名:", header.Filename)
}

func FileDownload(w http.ResponseWriter, r *http.Request) {
	filename := "E:\\projects\\workspace\\go_dreamfactory\\cmd\\v2\\RobotGUI.exe"

	file, _ := os.Open(filename)
	defer file.Close()

	fileHeader := make([]byte, 512)
	file.Read(fileHeader)

	fileStat, _ := file.Stat()

	w.Header().Set("Content-Disposition", "attachment; filename="+filename)
	w.Header().Set("Content-Type", http.DetectContentType(fileHeader))
	w.Header().Set("Content-Length", strconv.FormatInt(fileStat.Size(), 10))

	file.Seek(0, 0)
	io.Copy(w, file)

	return
}

//go:embed views/*
var f embed.FS

var version = "0.0.1"

func main() {

	host := flag.String("host", "10.0.0.9", "Host")
	port := flag.String("port", "8080", "Port")
	uploadsDir := flag.String("uploadsDir", "./uploads", "上传文件存储地址")
	wwwDir := flag.String("wwwDir", "./www", "www服务地址，即解压地址")

	flag.Parse()

	gin.SetMode(gin.ReleaseMode)
	r := gin.Default()
	r.Static("/prd", "./www")
	tmpl := template.Must(template.New("").ParseFS(f, "views/*"))
	r.SetHTMLTemplate(tmpl)

	r.GET("/version", func(c *gin.Context) {
		c.JSON(http.StatusOK, gin.H{
			"version": version,
		})
	})

	r.GET("/", func(c *gin.Context) {
		fmt.Println(tmpl.DefinedTemplates())
		c.HTML(http.StatusOK, "index.html", nil)
	})

	r.POST("/upload", func(c *gin.Context) {
		f, err := c.FormFile("file")
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{
				"error": err.Error(),
			})
		}

		//从上传的文件名获取到版本号
		s := strings.SplitN(f.Filename, "-", 2)
		if len(s) != 2 {
			c.JSON(http.StatusBadRequest, gin.H{
				"error": "上传的文件名中没有包含版本号 eg. update.zip-0.0.1",
			})
			return
		}
		version = s[1] //设置版本号

		dst := path.Join(*uploadsDir, f.Filename)
		err = c.SaveUploadedFile(f, dst)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{
				"error": err.Error(),
			})
		}
		_, err = tools.Unzip(dst, *wwwDir)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{
				"error": err.Error(),
			})
		}

		c.JSON(http.StatusOK, gin.H{
			"msg": "upload success",
			"url": strings.Join([]string{"http://" + *host + ":" + *port, "/prd/", strings.Split(f.Filename, ".")[0], "/index.html"}, ""),
		})
	})

	r.GET("/dirs", func(c *gin.Context) {
		dir, err := ioutil.ReadDir(*wwwDir)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{
				"error": err.Error(),
			})
		}

		type Item struct {
			Name     string `json:"name"`
			Url      string `json:"url"`
			DateTime string `json:"dateTime"`
		}

		var dirs []Item
		for _, f := range dir {
			// if f.IsDir() && len(f.Name()) > 0 && f.Name() != "__MACOSX" {
			dirs = append(dirs, Item{
				Name:     f.Name(),
				Url:      strings.Join([]string{"http://" + *host + ":" + *port, "/prd/", f.Name()}, ""),
				DateTime: f.ModTime().Format(time.RFC3339),
			})
			// }
		}

		c.JSON(http.StatusOK, gin.H{
			"dirs": dirs,
		})
	})

	err := r.Run("0.0.0.0:" + *port)
	if err != nil {
		log.Fatal(err)
	}
}