58 lines
1.7 KiB
Go
58 lines
1.7 KiB
Go
// 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
|
|
}
|