2b63ddc6b8
This allows us to check for some required values and give an easy to respond to error at startup rather than a lot of validation during run time.
77 lines
1.3 KiB
Go
77 lines
1.3 KiB
Go
package config
|
|
|
|
import (
|
|
"io/ioutil"
|
|
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type Config struct {
|
|
Homeserver homeserver `yaml:"homeserver"`
|
|
Appservice appservice `yaml:"appservice"`
|
|
Bridge bridge `yaml:"bridge"`
|
|
}
|
|
|
|
func (cfg *Config) validate() error {
|
|
if err := cfg.Homeserver.validate(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := cfg.Appservice.validate(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := cfg.Bridge.validate(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (cfg *Config) UnmarshalYAML(unmarshal func(interface{}) error) error {
|
|
type rawConfig Config
|
|
|
|
raw := rawConfig{}
|
|
if err := unmarshal(&raw); err != nil {
|
|
return err
|
|
}
|
|
|
|
return cfg.validate()
|
|
}
|
|
|
|
func FromBytes(data []byte) (*Config, error) {
|
|
cfg := Config{}
|
|
|
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := cfg.validate(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &cfg, nil
|
|
}
|
|
|
|
func FromString(str string) (*Config, error) {
|
|
return FromBytes([]byte(str))
|
|
}
|
|
|
|
func FromFile(filename string) (*Config, error) {
|
|
data, err := ioutil.ReadFile(filename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return FromBytes(data)
|
|
}
|
|
|
|
func (cfg *Config) Save(filename string) error {
|
|
data, err := yaml.Marshal(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return ioutil.WriteFile(filename, data, 0600)
|
|
}
|