mirror of
https://git.sr.ht/~adnano/kiln
synced 2024-10-30 01:13:08 +00:00
88 lines
2 KiB
Go
88 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"html/template"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
)
|
|
|
|
// Config contains site configuration.
|
|
type Config struct {
|
|
Title string
|
|
URLs []string
|
|
Feeds map[string]string
|
|
Tasks map[string]*Task
|
|
Permalinks map[string]string
|
|
templates *Templates
|
|
}
|
|
|
|
// Task represents a site build task.
|
|
type Task struct {
|
|
InputExt string `toml:"input_ext"` // input file extension
|
|
OutputExt string `toml:"output_ext"` // output file extension
|
|
TemplateExt string `toml:"template_ext"` // template file extension
|
|
PreProcess string `toml:"preprocess"` // preprocess command
|
|
PostProcess string `toml:"postprocess"` // postprocess command
|
|
StaticDir string `toml:"static_dir"` // static file directory
|
|
OutputDir string `toml:"output_dir"` // output directory
|
|
}
|
|
|
|
func (t Task) OutputPath(pagePath string) string {
|
|
return path.Join(pagePath, "index"+t.OutputExt)
|
|
}
|
|
|
|
// LoadConfig loads the configuration from the provided path.
|
|
func LoadConfig(path string) (*Config, error) {
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
|
|
c := &Config{}
|
|
if _, err := toml.DecodeReader(f, c); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Site contains site metadata passed to templates
|
|
type Site struct {
|
|
Title string
|
|
URLs []string
|
|
}
|
|
|
|
// Initialize templates
|
|
c.templates = NewTemplates()
|
|
c.templates.Funcs(map[string]interface{}{
|
|
"site": func() Site {
|
|
return Site{
|
|
Title: c.Title,
|
|
URLs: c.URLs,
|
|
}
|
|
},
|
|
"partial": func(name string, data interface{}) (interface{}, error) {
|
|
t, ok := c.templates.FindPartial(name)
|
|
if !ok {
|
|
return "", fmt.Errorf("Error: partial %q not found", name)
|
|
}
|
|
var b strings.Builder
|
|
if err := t.Execute(&b, data); err != nil {
|
|
return "", err
|
|
}
|
|
return b.String(), nil
|
|
},
|
|
"html": func(s string) template.HTML {
|
|
return template.HTML(s)
|
|
},
|
|
})
|
|
|
|
return c, nil
|
|
}
|
|
|
|
func (c *Config) LoadTemplates(path string, exts []string) error {
|
|
return c.templates.Load(path, exts)
|
|
}
|