kiln/config.go

84 lines
2.1 KiB
Go
Raw Normal View History

2020-11-20 17:07:38 +00:00
package main
import (
"fmt"
"html/template"
2020-11-20 17:07:38 +00:00
"os"
"strings"
2020-11-20 17:07:38 +00:00
2021-03-21 03:54:55 +00:00
"github.com/BurntSushi/toml"
2020-11-20 17:07:38 +00:00
)
// Config contains site configuration.
type Config struct {
2021-05-10 04:44:25 +00:00
Title string `toml:"title"`
URLs []string `toml:"urls"`
Feeds map[string]string `toml:"feeds"`
Tasks map[string]*Task `toml:"tasks"`
Permalinks map[string]string `toml:"permalinks"`
2021-05-09 00:27:13 +00:00
templates *Templates
2020-11-20 17:07:38 +00:00
}
// Task represents a site build task.
type Task struct {
2021-04-11 22:42:55 +00:00
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
2021-04-11 22:42:55 +00:00
PostProcess string `toml:"postprocess"` // postprocess command
StaticDir string `toml:"static_dir"` // static file directory
OutputDir string `toml:"output_dir"` // output directory
2021-05-10 04:44:25 +00:00
UglyURLs bool `toml:"ugly_urls"` // whether to use ugly URLs
}
// LoadConfig loads the configuration from the provided path.
func LoadConfig(path string) (*Config, error) {
2020-11-20 17:07:38 +00:00
f, err := os.Open(path)
if err != nil {
2021-03-20 06:02:36 +00:00
return nil, err
2020-11-20 17:07:38 +00:00
}
2021-03-20 06:02:36 +00:00
defer f.Close()
2020-11-20 17:07:38 +00:00
2021-04-20 20:16:12 +00:00
c := &Config{}
2021-03-21 03:54:55 +00:00
if _, err := toml.DecodeReader(f, c); err != nil {
2021-03-20 06:02:36 +00:00
return nil, err
}
2020-11-20 17:07:38 +00:00
// Site contains site metadata passed to templates
type Site struct {
Title string
2020-11-27 23:24:31 +00:00
URLs []string
2020-11-20 17:07:38 +00:00
}
2021-05-09 23:28:09 +00:00
// Initialize templates
2021-05-09 00:27:13 +00:00
c.templates = NewTemplates()
c.templates.Funcs(map[string]interface{}{
2020-11-20 17:07:38 +00:00
"site": func() Site {
return Site{
Title: c.Title,
2020-11-27 23:24:31 +00:00
URLs: c.URLs,
2020-11-20 17:07:38 +00:00
}
},
"partial": func(name string, data interface{}) (interface{}, error) {
2021-05-09 00:27:13 +00:00
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)
},
2020-11-20 17:07:38 +00:00
})
2021-05-09 23:28:09 +00:00
return c, nil
}
func (c *Config) LoadTemplates(path string, exts []string) error {
return c.templates.Load(path, exts)
2020-11-20 17:07:38 +00:00
}