package tmpl

import (
	"encoding/json"
	"fmt"
	"html/template"
	"io/fs"
	"log"
	"os"
	"path/filepath"
	"strings"

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

func New(app *gin.Engine) {
	app.Static("/assets", "./res/assets")

	funcMap := template.FuncMap{
		"mod":  GetRemainder,
		"add":  GetAdd,
		"div":  GetDivision,
		"sub":  Sub,
		"seq":  Seq,
		"mul":  GetMultiply,
		"test": TestFunc,
	}
	app.SetFuncMap(funcMap)

	// 加载所有HTML模板
	templ := loadTemplates("./res/templates")
	app.SetHTMLTemplate(templ)
}

// loadTemplates 加载所有HTML模板
func loadTemplates(templatesDir string) *template.Template {
	tmpl := template.New("")

	// 首先加载组件和布局模板（.tpl文件）
	err := filepath.Walk(templatesDir, func(path string, info fs.FileInfo, err error) error {
		if err != nil {
			return err
		}
		// 加载.tpl文件（组件和布局）
		if !info.IsDir() && strings.HasSuffix(path, ".html") {
			content, err := os.ReadFile(path)
			if err != nil {
				return err
			}
			// 使用相对路径作为模板名，保持目录结构
			relPath, _ := filepath.Rel(templatesDir, path)
			templateName := strings.ReplaceAll(relPath, "\\", "/")

			// 如果不是 components 或 layouts 目录中的文件，且包含 {{define "content"}}
			// 则将其重命名为唯一的 content 名称（基于文件名）
			contentStr := string(content)
			_, err = tmpl.New(templateName).Parse(contentStr)
			if err != nil {
				return fmt.Errorf("解析模板 %s 失败: %v", path, err)
			}
			fmt.Printf("已加载组件/布局模板: %s\n", templateName)
		}
		return nil
	})

	if err != nil {
		log.Fatalf("加载组件模板失败: %v", err)
	}

	return tmpl
}

func TestFunc(in string) string {
	return strings.ToUpper(in)
}

func AnyToBytes(data any) ([]byte, error) {
	return json.Marshal(data)
}

// 定义取余函数
func GetRemainder(a, b int) int {
	return a % b
}

// 定义 整数相加
func GetAdd(a, b int) int {
	return a + b
}

// 定义 整数相乘
func GetMultiply(a, b int) int {
	return a * b
}

// 定义 整数相除
func GetDivision(a, b int) int {
	if b == 0 {
		return 0 // 当除数为 0 时返回 0，避免 panic
	}
	return a / b
}

// 定义 整数相减
func Sub(a, b int) int {
	return a - b
}

// 定义 获取页码
func Seq(start, end int) []int {
	result := make([]int, end-start+1)
	for i := range result {
		result[i] = start + i
	}
	return result
}