fixup

internal/templates/templates.go

@@ -0,0 +1,57 @@
+// Package templates 負責載入與渲染 html/template 模板。
+package templates
+
+import (
+	"bytes"
+	"fmt"
+	"html/template"
+	"io/fs"
+	"net/http"
+	"path"
+)
+
+// layoutFile 基礎版型，定義 "layout" 模板並引用各頁面的 "content" 區塊。
+const layoutFile = "layout.html"
+
+// Engine 依頁面名稱持有各自獨立的模板集合（layout + 單一頁面），
+// 讓每個頁面都能定義自己的 "content"、"title" 區塊而不互相覆蓋。
+type Engine struct {
+	pages map[string]*template.Template
+}
+
+// New 載入 fsys 根目錄的 layout.html 與 pages/ 下所有頁面模板。
+func New(fsys fs.FS) (*Engine, error) {
+	pageFiles, err := fs.Glob(fsys, path.Join("pages", "*.html"))
+	if err != nil {
+		return nil, err
+	}
+	if len(pageFiles) == 0 {
+		return nil, fmt.Errorf("templates: pages/ 下找不到任何模板")
+	}
+	e := &Engine{pages: make(map[string]*template.Template, len(pageFiles))}
+	for _, page := range pageFiles {
+		t, err := template.New(layoutFile).ParseFS(fsys, layoutFile, page)
+		if err != nil {
+			return nil, fmt.Errorf("templates: 解析 %s: %w", page, err)
+		}
+		e.pages[path.Base(page)] = t
+	}
+	return e, nil
+}
+
+// Render 先將頁面完整渲染至緩衝區再以 status 寫出，
+// 避免中途失敗時輸出殘缺的 HTML。頁面名稱為檔名，如 "login.html"。
+func (e *Engine) Render(w http.ResponseWriter, status int, page string, data any) error {
+	t, ok := e.pages[page]
+	if !ok {
+		return fmt.Errorf("templates: 找不到頁面 %q", page)
+	}
+	var buf bytes.Buffer
+	if err := t.ExecuteTemplate(&buf, "layout", data); err != nil {
+		return fmt.Errorf("templates: 渲染 %q: %w", page, err)
+	}
+	w.Header().Set("Content-Type", "text/html; charset=utf-8")
+	w.WriteHeader(status)
+	_, err := w.Write(buf.Bytes())
+	return err
+}