47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
|
package templatelib
|
||
|
|
|
||
|
|
import (
|
||
|
|
"html/template"
|
||
|
|
"path/filepath"
|
||
|
|
)
|
||
|
|
|
||
|
|
type TemplateLibrary struct {
|
||
|
|
Templates map[string]*template.Template
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewTemplateLibrary(dir string) (TemplateLibrary, error) {
|
||
|
|
t := make(map[string]*template.Template)
|
||
|
|
|
||
|
|
pages, err := filepath.Glob(filepath.Join(dir, "*.page.tmpl"))
|
||
|
|
if err != nil {
|
||
|
|
return TemplateLibrary{}, err
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, page := range pages {
|
||
|
|
name := filepath.Base(page)
|
||
|
|
ts, err := template.ParseFiles(page)
|
||
|
|
if err != nil {
|
||
|
|
return TemplateLibrary{}, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Use the ParseGlob method to add any 'layout' templates to the
|
||
|
|
// template set (in our case, it's just the 'base' layout at the
|
||
|
|
// moment).
|
||
|
|
ts, err = ts.ParseGlob(filepath.Join(dir, "*.layout.tmpl"))
|
||
|
|
if err != nil {
|
||
|
|
return TemplateLibrary{}, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Use the ParseGlob method to add any 'partial' templates to the
|
||
|
|
// template set (in our case, it's just the 'footer' partial at the
|
||
|
|
// moment).
|
||
|
|
ts, err = ts.ParseGlob(filepath.Join(dir, "*.partial.tmpl"))
|
||
|
|
if err != nil {
|
||
|
|
return TemplateLibrary{}, err
|
||
|
|
}
|
||
|
|
t[name] = ts
|
||
|
|
}
|
||
|
|
|
||
|
|
return TemplateLibrary{Templates: t}, nil
|
||
|
|
}
|