package main

import (
	"context"
	"crypto/sha256"
	"encoding/hex"
	"errors"
	"html/template"
	"math/rand"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"time"

	"github.com/cloudwego/hertz/pkg/protocol/consts"
)

var funcMap = template.FuncMap{"md5": funcMD5, "basePath": funcBasePath, "T": funcT, "safeHTML": funcSafeHTML, "relURL": funcRelURL, "sass": funcSass}

// initTemplate 初始化模板
func initTemplate() error {
	h.SetFuncMap(funcMap)
	// h.LoadHTMLFiles(themePath + "index.html")
	// h.LoadHTMLGlob(datadir + "html/theme/default/*")
	// 手动声明template对象,自己控制文件路径,默认是使用文件名,多个文件夹会存在问题
	err := loadTemplate()
	// 设置默认的静态文件,实际路径会拼接为 datadir/public
	h.Static("/public", datadir)
	return err
}

// loadTemplate 用于更新重复加载
func loadTemplate() error {
	tmpl := template.New("").Delims("", "").Funcs(funcMap)
	err := filepath.Walk(templateDir, func(path string, info os.FileInfo, err error) error {
		// 分隔符统一为 / 斜杠
		path = filepath.ToSlash(path)
		// 相对路径
		relativePath := path[len(templateDir)-1:]
		// 如果是静态资源
		if strings.Contains(path, "/js/") || strings.Contains(path, "/css/") || strings.Contains(path, "/image/") {
			if !strings.HasSuffix(path, consts.FSCompressedFileSuffix) { // 过滤掉压缩包
				h.StaticFile(relativePath, path)
			}
		} else if strings.HasSuffix(path, ".html") { // 模板文件
			// 创建对应的模板
			t := tmpl.New(relativePath)
			b, err := os.ReadFile(path)
			if err != nil {
				return err
			}
			// 对应模板内容
			_, err = t.Parse(string(b))
			if err != nil {
				return err
			}
		}
		return nil
	})
	if err != nil {
		FuncLogError(err)
		// panic(err)
		return err
	}

	// 处理静态化文件
	filepath.Walk(statichtmlDir, func(path string, info os.FileInfo, err error) error {
		if info.IsDir() { // 只处理文件
			return nil
		}
		// 分隔符统一为 / 斜杠
		path = filepath.ToSlash(path)
		// 相对路径
		relativePath := path[len(statichtmlDir)-1:]
		// 设置静态化文件
		h.StaticFile(relativePath, path)
		return nil
	})

	// 设置模板
	h.SetHTMLTemplate(tmpl)
	return nil
}

// funcBasePath 基础路径,前端所有的资源请求必须带上 {{basePath}}
func funcBasePath() string {
	return config.basePath
}

// isInstalled 是否已经安装过了
func isInstalled() bool {
	// 依赖bleveStatus变量,确保bleve在isInstalled之前初始化
	if !bleveStatus {
		FuncLogError(errors.New("bleveStatus状态为false"))
	}
	return pathExists(templateDir + "admin/install.html")
}

// updateInstall 更新安装状态
func updateInstall(ctx context.Context) error {
	// 将config配置写入到索引,写入前先把config表清空
	err := insertConfig(ctx, config)
	if err != nil {
		return err
	}

	now := strconv.FormatInt(time.Now().UnixNano(), 10)
	// 删除 install 文件
	err = os.Rename(templateDir+"admin/install.html", templateDir+"admin/install.html."+now)
	if err != nil {
		return err
	}

	// install_config.json 重命名为 install_config.json_配置已失效_请通过后台设置管理
	err = os.Rename(datadir+"install_config.json", datadir+"install_config.json."+now)
	if err != nil {
		return err
	}
	// 更改安装状态
	installed = true
	return nil
}

// randStr 生成随机字符串
func randStr(n int) string {
	rand.Seed(time.Now().UnixNano())
	b := make([]byte, n)
	for i := range b {
		b[i] = letters[rand.Intn(len(letters))]
	}
	return string(b)
}

// hashSha256 使用sha256计算hash值
func hashSha256(str string) string {
	hashByte := sha256.Sum256([]byte(str))
	hashStr := hex.EncodeToString(hashByte[:])
	return hashStr
}