package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"

	"gitee.com/flyiot/flylibs/FlyWebRoots"
)

var ()

func main() {
	if len(os.Args) < 2 {
		printHelp()
	}
	switch os.Args[1] {
	case "build_wwwroot_go":
		cmd_build_wwwroot_go()
	case "build_wwwroot_file":
		cmd_build_wwwroot_file()
	default:
		printHelp()
	}

}

func printHelp() {
	fmt.Println("")
	fmt.Println("flylibs tools")
	fmt.Println("")
	fmt.Println("flylibs build_wwwroot_go <wwwroot dir>")
	fmt.Println("")
	fmt.Println("flylibs build_wwwroot_file <wwwroot dir> <file name>")
	fmt.Println("")

	os.Exit(0)
}

func cmd_build_wwwroot_file() error {
	p := "."
	if len(os.Args) >= 3 && os.Args[2] != "" {
		p = os.Args[2]
	}
	t := "./wwwroot"
	if len(os.Args) >= 4 && os.Args[3] != "" {
		t = os.Args[3]
	}

	fwr := FlyWebRoots.NewFlyWebRoot()
	encodeString, err := fwr.BuildWebRoot(p)
	if err != nil {
		fmt.Println(err)
		return err
	}
	err = ioutil.WriteFile(t, []byte(encodeString), 0777)
	if err != nil {
		fmt.Println(err)
		return err
	}
	fmt.Println("Done !")
	return nil
}

func cmd_build_wwwroot_go() {
	p := "."
	if len(os.Args) >= 3 && os.Args[2] != "" {
		p = os.Args[2]
	}
	t := "./"
	if len(os.Args) >= 4 && os.Args[3] != "" {
		t = os.Args[3]
	}
	if err := buildWebRootFromDir(p, t); err != nil {
		fmt.Println(err)
	} else {
		fmt.Println("Done !")
	}
}

// 将目录打包成wwwroot.go
func buildWebRootFromDir(dir, target string) error {
	fwr := FlyWebRoots.NewFlyWebRoot()
	// 完成zip打包，进行编码，写入文件
	encodeString, err := fwr.BuildWebRoot(dir)
	if err != nil {
		return err
	}

	c := `package DefaultWebRoot

import (
	flywwwroot "gitee.com/flyiot/flylibs/FlyWebRoots"
)

func init(){
	flywwwroot.LoadDefaultData("` + encodeString + `")
}`
	// 写入文件

	os.MkdirAll(filepath.Join(target, "DefaultWebRoot"), os.ModePerm)
	ioutil.WriteFile(filepath.Join(target, "DefaultWebRoot/wwwroot.go"), []byte(c), 0777)
	return nil
}