2020-11-20 17:07:38 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"os"
|
2021-03-21 03:17:58 +00:00
|
|
|
"path"
|
2020-11-20 17:07:38 +00:00
|
|
|
"text/template"
|
|
|
|
|
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-03-20 20:29:06 +00:00
|
|
|
Title string `toml:"title"` // site title
|
|
|
|
URLs []string `toml:"urls"` // site URLs
|
|
|
|
Feeds map[string]string `toml:"feeds"` // site feeds
|
2021-03-21 03:44:13 +00:00
|
|
|
Tasks map[string]*Task `toml:"tasks"` // site tasks
|
2021-03-20 20:29:06 +00:00
|
|
|
Templates *Templates `toml:"-"` // site templates
|
2020-11-20 17:07:38 +00:00
|
|
|
}
|
|
|
|
|
2021-03-21 03:17:58 +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
|
2021-04-21 01:28:29 +00:00
|
|
|
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-03-21 03:17:58 +00:00
|
|
|
}
|
|
|
|
|
2021-04-21 01:28:29 +00:00
|
|
|
func (t Task) OutputPath(pagePath string) string {
|
|
|
|
return path.Join(pagePath, "index"+t.OutputExt)
|
2021-03-21 03:17:58 +00:00
|
|
|
}
|
|
|
|
|
2021-03-21 03:57:16 +00:00
|
|
|
// 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
|
|
|
|
}
|
|
|
|
|
|
|
|
return c, nil
|
2020-11-20 17:07:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// LoadTemplates loads templates from the provided path.
|
|
|
|
func (c *Config) LoadTemplates(path string) error {
|
|
|
|
// 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
|
|
|
}
|
|
|
|
|
|
|
|
// Load templates
|
|
|
|
c.Templates = NewTemplates()
|
|
|
|
c.Templates.Funcs(template.FuncMap{
|
|
|
|
"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
|
|
|
}
|
|
|
|
},
|
|
|
|
})
|
2020-11-22 20:14:50 +00:00
|
|
|
c.Templates.LoadDefault()
|
2020-11-20 17:07:38 +00:00
|
|
|
return c.Templates.Load(path)
|
|
|
|
}
|