package main

import (
	"bytes"
	"embed"
	"flag"
	"html/template"
	"io/fs"
	"log"
	"net/http"
	"os"
	"sort"
	"strconv"
	"toolbox/pkg/base"
	_ "toolbox/pkg/learnnumber"
	_ "toolbox/pkg/zitie" // 匿名导入以触发 init()

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

//go:embed web/*
var webFS embed.FS

func main() {
	portFlag := flag.Int("port", 0, "端口号 (优先于环境变量 PORT)")
	gaFlag := flag.String("ga", "", "Google Analytics ID")
	flag.Parse()

	gaID := *gaFlag
	if gaID == "" {
		gaID = os.Getenv("GA_ID")
	}

	if os.Getenv("GIN_MODE") == "release" {
		gin.SetMode(gin.ReleaseMode)
	}

	r := gin.New()
	r.Use(gin.Logger(), gin.Recovery())
	r.RedirectTrailingSlash = false
	r.RedirectFixedPath = false
	_ = r.SetTrustedProxies(nil)

	// 1. 自动发现并注册工具路由
	api := r.Group("/api")
	{
		for id, tool := range base.Registry {
			if err := tool.Init(); err != nil {
				log.Fatalf("Failed to initialize tool %s: %v", id, err)
			}
			tool.RegisterRoutes(api.Group("/" + id))
		}
	}

	// 2. 静态资源
	subFS, _ := fs.Sub(webFS, "web")
	r.StaticFS("/static", http.FS(subFS))

	// 3. 模板引擎初始化
	// 递归加载 web 目录下所有的 .html 文件
	tmpl, err := template.ParseFS(subFS, "layout.html", "index.html", "tools/*.html")
	if err != nil {
		log.Fatalf("Failed to parse templates: %v", err)
	}

	// 通用页面渲染函数
	serveIndex := func(c *gin.Context) {
		data := gin.H{
			"Title": "Own-Tools",
			"GA_ID": gaID,
		}
		
		var buf bytes.Buffer
		// 因为已经 Sub 了，模板名就是文件名
		if err := tmpl.ExecuteTemplate(&buf, "layout.html", data); err != nil {
			c.String(http.StatusInternalServerError, "Template error: %v", err)
			return
		}
		c.Data(http.StatusOK, "text/html; charset=utf-8", buf.Bytes())
	}

	r.GET("/", serveIndex)
	r.GET("/api/tools", func(c *gin.Context) {
		var list []map[string]string
		ids := make([]string, 0, len(base.Registry))
		for id := range base.Registry { ids = append(ids, id) }
		sort.Strings(ids)

		for _, id := range ids {
			t := base.Registry[id]
			list = append(list, map[string]string{
				"id":    t.ID(),
				"name":  t.Name(),
				"desc":  t.Description(),
				"emoji": t.Emoji(),
			})
		}
		c.JSON(http.StatusOK, list)
	})

	// 4. 全路径回退 (SPA Routing)
	r.NoRoute(serveIndex)

	// 5. 启动
	var finalPort string
	if *portFlag > 0 {
		finalPort = strconv.Itoa(*portFlag)
	} else if envPort := os.Getenv("PORT"); envPort != "" {
		finalPort = envPort
	} else {
		finalPort = "8080"
	}

	log.Printf("Toolbox server starting on :%s", finalPort)
	if err := r.Run(":" + finalPort); err != nil {
		log.Fatalf("Server failed: %v", err)
	}
}