79 lines
1.6 KiB
Go
79 lines
1.6 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/distribyted/distribyted"
|
|
"github.com/rs/zerolog/log"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type EventFunc func(event string)
|
|
type ReloadFunc func(*Root, EventFunc) error
|
|
|
|
type Handler struct {
|
|
p string
|
|
}
|
|
|
|
func NewHandler(path string) *Handler {
|
|
return &Handler{p: path}
|
|
}
|
|
|
|
func (c *Handler) createFromTemplateFile() ([]byte, error) {
|
|
t, err := distribyted.HttpFS.Open("/templates/config_template.yaml")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer t.Close()
|
|
|
|
tb, err := ioutil.ReadAll(t)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := os.MkdirAll(filepath.Dir(c.p), 0744); err != nil {
|
|
return nil, fmt.Errorf("error creating path for configuration file: %s, %w", c.p, err)
|
|
}
|
|
return tb, ioutil.WriteFile(c.p, tb, 0644)
|
|
}
|
|
|
|
func (c *Handler) GetRaw() ([]byte, error) {
|
|
f, err := ioutil.ReadFile(c.p)
|
|
if os.IsNotExist(err) {
|
|
log.Info().Str("file", c.p).Msg("configuration file does not exist, creating from template file")
|
|
return c.createFromTemplateFile()
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error reading configuration file: %w", err)
|
|
}
|
|
|
|
return f, nil
|
|
}
|
|
|
|
func (c *Handler) Get() (*Root, error) {
|
|
b, err := c.GetRaw()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
conf := &Root{}
|
|
if err := yaml.Unmarshal(b, conf); err != nil {
|
|
return nil, fmt.Errorf("error parsing configuration file: %w", err)
|
|
}
|
|
|
|
conf = AddDefaults(conf)
|
|
|
|
return conf, nil
|
|
}
|
|
|
|
func (c *Handler) Set(b []byte) error {
|
|
if err := yaml.Unmarshal(b, &Root{}); err != nil {
|
|
return fmt.Errorf("error parsing configuration file: %w", err)
|
|
}
|
|
|
|
return ioutil.WriteFile(c.p, b, 0644)
|
|
}
|