alrm/config/config.go

package config

import (
	"fmt"
	"git.binarythought.com/cdramey/alrm/alarm"
	"os"
	"time"
)

type Config struct {
	Groups     map[string]*Group
	Alarms     map[string]alarm.Alarm
	Interval   time.Duration
	DebugLevel int
	Listen     string
	Path       string
	APIKey     string
	APIKeyFile string
}

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) {
	cfg := &Config{
		// Default check interval, 30 seconds
		Interval: time.Second * 30,
		// Default listen address
		Listen:     "127.0.0.1:8282",
		DebugLevel: debuglvl,
		Path:       fn,
		// API keyfile defaults to alrmrc.key
		APIKeyFile: fn + ".key",
	}

	pr := &parser{config: cfg}
	if err := pr.parse(); err != nil {
		return nil, err
	}

	if cfg.APIKey == "" {
		b, err := os.ReadFile(cfg.APIKeyFile)
		if err != nil {
			return nil, err
		}
		cfg.APIKey = string(b)
	}

	return cfg, nil
}