2020-11-20 17:07:38 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"os"
|
|
|
|
"text/template"
|
|
|
|
|
2021-03-21 00:35:48 +00:00
|
|
|
"github.com/pelletier/go-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
|
|
|
|
Templates *Templates `toml:"-"` // site templates
|
2020-11-20 17:07:38 +00:00
|
|
|
}
|
|
|
|
|
2021-03-20 06:02:36 +00:00
|
|
|
// LoadConfig loads the configuration from the provided path.
|
|
|
|
func LoadConfig(path string) (*Config, error) {
|
|
|
|
c := new(Config)
|
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-03-21 00:35:48 +00:00
|
|
|
if err := toml.NewDecoder(f).Decode(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)
|
|
|
|
}
|