76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"git.binarythought.com/cdramey/alrm/alarm"
|
|
"time"
|
|
)
|
|
|
|
type Config struct {
|
|
Groups map[string]*Group
|
|
Alarms map[string]alarm.Alarm
|
|
Interval time.Duration
|
|
Listen string
|
|
Path string
|
|
}
|
|
|
|
func NewConfig() *Config {
|
|
return &Config{
|
|
// Default check interval, 30 seconds
|
|
Interval: time.Second * 30,
|
|
// Default listen address
|
|
Listen: "127.0.0.1:8282",
|
|
}
|
|
}
|
|
|
|
func (c *Config) NewAlarm(name string, typename string) (alarm.Alarm, error) {
|
|
if c.Alarms == nil {
|
|
c.Alarms = make(map[string]alarm.Alarm)
|
|
}
|
|
|
|
if _, exists := c.Alarms[name]; exists {
|
|
return nil, fmt.Errorf("alarm %s already exists", name)
|
|
}
|
|
|
|
a, err := alarm.NewAlarm(name, typename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
c.Alarms[name] = a
|
|
|
|
return a, nil
|
|
}
|
|
|
|
func (c *Config) NewGroup(name string) (*Group, error) {
|
|
if c.Groups == nil {
|
|
c.Groups = make(map[string]*Group)
|
|
}
|
|
|
|
if _, exists := c.Groups[name]; exists {
|
|
return nil, fmt.Errorf("group %s already exists", name)
|
|
}
|
|
|
|
group := &Group{Name: name}
|
|
c.Groups[name] = group
|
|
return group, nil
|
|
}
|
|
|
|
func (c *Config) SetInterval(val string) error {
|
|
interval, err := time.ParseDuration(val)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.Interval = interval
|
|
return nil
|
|
}
|
|
|
|
func ReadConfig(fn string, debuglvl int) (*Config, error) {
|
|
parser := &Parser{DebugLevel: debuglvl}
|
|
config, err := parser.Parse(fn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return config, nil
|
|
}
|