use mysql

themes/manager.go

@@ -1,57 +1,164 @@
-package themes // Declare the package at the top
+package themes
+
 import (
+	"embed"
+	"errors"
 	"fmt"
 	"html/template"
-	"net/http"
+	"io/fs"
 	"os"
 	"path/filepath"
 	"strings"
+	"sync"
+
+	"github.com/gin-gonic/gin"
 )
 
-// ThemeManager manages themes for the blog system
-type ThemeManager struct {
-	CurrentTheme string                        // Name of the currently active theme
-	Templates    map[string]*template.Template // Cached compiled HTML templates
+// 主题元数据结构
+type ThemeMeta struct {
+	Name        string `yaml:"name"`
+	Author      string `yaml:"author"`
+	Version     string `yaml:"version"`
+	Description string `yaml:"description"`
 }
 
-// LoadTheme loads a specified theme by its name
-func (tm *ThemeManager) LoadTheme(themeName string) error {
-	// 1. 读取theme.yaml验证主题有效性
-	// 2. 预编译所有HTML模板，缓存到Templates
-	// 3. 注册静态资源路由：/themes/[name]/static/*filepath
-	tm.CurrentTheme = themeName
+// 主题管理器
+type ThemeManager struct {
+	mu            sync.RWMutex
+	currentTheme  string
+	templates     map[string]*template.Template
+	baseTemplates embed.FS // 可选：嵌入基础模板
+	themesDir     string
+}
 
-	// Step 1: Validate the theme by reading theme.yaml
-	themeConfigPath := fmt.Sprintf("h:/code/go_blog/web/themes/%s/theme.yaml", themeName)
-	if _, err := os.Stat(themeConfigPath); os.IsNotExist(err) {
-		return fmt.Errorf("theme %s does not exist or is invalid", themeName)
+// 创建新主题管理器
+func NewManager(themesDir string) *ThemeManager {
+	return &ThemeManager{
+		templates: make(map[string]*template.Template),
+		themesDir: themesDir,
+	}
+}
+
+// 核心方法：加载主题
+func (m *ThemeManager) LoadTheme(themeName string) error {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	// 1. 验证主题目录结构
+	themePath := filepath.Join(m.themesDir, themeName)
+	if !isValidTheme(themePath) {
+		return fmt.Errorf("invalid theme structure: %s", themeName)
 	}
 
-	// Step 2: Precompile all HTML templates and cache them
-	tm.Templates = make(map[string]*template.Template)
-	templateDir := fmt.Sprintf("h:/code/go_blog/web/themes/%s/tmplates/", themeName)
-	err := filepath.Walk(templateDir, func(path string, info os.FileInfo, err error) error {
-		if err != nil {
-			return err
-		}
-		if !info.IsDir() && strings.HasSuffix(info.Name(), ".tmpl") {
-			tmpl, err := template.ParseFiles(path)
-			if err != nil {
-				return err
-			}
-			tm.Templates[info.Name()] = tmpl
-		}
-		return nil
-	})
+	// 2. 加载模板文件
+	tpls, err := m.parseTemplates(themePath)
 	if err != nil {
-		return fmt.Errorf("failed to load templates: %v", err)
+		return fmt.Errorf("template parsing failed: %w", err)
 	}
 
-	// Step 3: Register static resource routes
-	http.Handle(fmt.Sprintf("/themes/%s/static/", themeName), http.StripPrefix(fmt.Sprintf("/themes/%s/static/", themeName), http.FileServer(http.Dir(fmt.Sprintf("h:/code/go_blog/themes/%s/static", themeName)))))
+	// 3. 更新当前主题
+	m.currentTheme = themeName
+	m.templates = tpls
 
 	return nil
 }
-func NewManager() *ThemeManager {
-	return &ThemeManager{}
+
+// 获取当前主题名称
+func (m *ThemeManager) CurrentTheme() string {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	return m.currentTheme
+}
+
+// 获取编译后的模板
+func (m *ThemeManager) GetTemplate(name string) *template.Template {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	return m.templates[name]
+}
+
+// 注册静态文件路由（Gin框架）
+func (m *ThemeManager) RegisterStaticRoutes(router *gin.Engine) {
+	staticPath := filepath.Join(m.themesDir, m.currentTheme, "static")
+	router.StaticFS("/theme/static", gin.Dir(staticPath, false))
+}
+
+// 校验主题完整性
+func isValidTheme(themePath string) bool {
+	requiredFiles := []string{
+		"theme.yaml",
+		"templates/home.html",
+		"templates/post.html",
+	}
+
+	for _, f := range requiredFiles {
+		if _, err := os.Stat(filepath.Join(themePath, f)); os.IsNotExist(err) {
+			return false
+		}
+	}
+	return true
+}
+
+// 解析模板文件（支持布局继承）
+func (m *ThemeManager) parseTemplates(themePath string) (map[string]*template.Template, error) {
+	templates := make(map[string]*template.Template)
+
+	// 加载公共基础模板（可选）
+	baseTpl := template.New("base").Funcs(template.FuncMap{
+		"safeHTML": func(s string) template.HTML { return template.HTML(s) },
+	})
+
+	// 从主题目录加载模板
+	tplDir := filepath.Join(themePath, "templates")
+	err := filepath.WalkDir(tplDir, func(path string, d fs.DirEntry, err error) error {
+		if err != nil || d.IsDir() || !strings.HasSuffix(path, ".html") {
+			return nil
+		}
+
+		// 读取模板内容
+		content, err := os.ReadFile(path)
+		if err != nil {
+			return err
+		}
+
+		// 生成模板名称（相对路径）
+		name := strings.TrimPrefix(path, tplDir+string(filepath.Separator))
+		name = strings.TrimSuffix(name, filepath.Ext(name))
+
+		// 克隆基础模板并解析
+		tpl := template.Must(baseTpl.Clone())
+		tpl, err = tpl.Parse(string(content))
+		if err != nil {
+			return fmt.Errorf("parse error in %s: %w", name, err)
+		}
+
+		templates[name] = tpl
+		return nil
+	})
+
+	if err != nil {
+		return nil, err
+	}
+
+	if len(templates) == 0 {
+		return nil, errors.New("no valid templates found")
+	}
+
+	return templates, nil
+}
+
+// 获取可用主题列表
+func (m *ThemeManager) ListThemes() ([]string, error) {
+	dirs, err := os.ReadDir(m.themesDir)
+	if err != nil {
+		return nil, err
+	}
+
+	var themes []string
+	for _, d := range dirs {
+		if d.IsDir() && isValidTheme(filepath.Join(m.themesDir, d.Name())) {
+			themes = append(themes, d.Name())
+		}
+	}
+	return themes, nil
 }