// 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
}