NetworkAuth/web/public.go

package web

import (
	"embed"
	"html/template"
	"io/fs"
	"log"
	"os"
	"path/filepath"

	"github.com/spf13/viper"
)

// TemplatesFS 嵌入模板的文件系统
//
//go:embed template/*.html template/admin/*.html
var templatesFS embed.FS

// StaticFS 嵌入静态资源的文件系统（包含 CSS/JS 的 static 与 图片/字体等资源的 assets）
//
//go:embed static/* assets/*
var staticFS embed.FS

// getDistRootFS 获取基于 server.dist 的本地文件系统
// 当 server.dist 非空且路径存在时，返回对应的本地只读 FS；否则返回 nil
func getDistRootFS() fs.FS {
	// 从配置中读取 server.dist
	distPath := viper.GetString("server.dist")
	if distPath == "" {
		return nil
	}
	// 归一化路径，兼容相对/绝对
	absPath := distPath
	if !filepath.IsAbs(distPath) {
		if p, err := filepath.Abs(distPath); err == nil {
			absPath = p
		}
	}
	// 检查目录是否存在
	if info, err := os.Stat(absPath); err == nil && info.IsDir() {
		return os.DirFS(absPath)
	}
	log.Printf("server.dist 路径无效或不可访问：%s，将回退使用嵌入资源", distPath)
	return nil
}

// ParseTemplates 解析模板
// 优先从 server.dist 指定目录加载（当配置非空且有效），否则回退到嵌入模板
func ParseTemplates() (*template.Template, error) { // Go 顶级函数不支持箭头写法
	if distFS := getDistRootFS(); distFS != nil {
		// 期望 dist 目录下存在 template 与 template/admin 结构
		// 如：{dist}/template/*.html 与 {dist}/template/admin/*.html
		return template.ParseFS(distFS, "template/*.html", "template/admin/*.html")
	}
	// 默认：使用嵌入模板
	return template.ParseFS(templatesFS, "template/*.html", "template/admin/*.html")
}

// GetStaticFS 返回静态资源文件系统（包含 static 与 assets 目录）
// 优先使用 server.dist 指定的本地目录；否则回退到嵌入静态资源
func GetStaticFS() (fs.FS, error) { // Go 顶级函数不支持箭头写法
	if distFS := getDistRootFS(); distFS != nil {
		// 直接返回以 dist 根为起点的 FS，routes 中会再基于此 FS Sub 出 static 与 assets
		return distFS, nil
	}
	return staticFS, nil
}

// IsDevMode 检查是否为开发模式
// 注意：这个函数保留用于向后兼容，建议使用 middleware.IsDevMode()
func IsDevMode() bool {
	return viper.GetBool("server.dev_mode")
}