logmania/notify/config/config.go

package config

import (
	"encoding/json"
	"os"
	"regexp"
	"time"

	"github.com/genofire/logmania/log"
)

type NotifyState struct {
	Hostname  map[string]string           `json:"hostname"`
	HostTo    map[string][]string         `json:"host_to"`
	MaxPrioIn map[string]log.LogLevel     `json:"maxLevel"`
	RegexIn   map[string][]string         `json:"regexIn"`
	regexIn   map[string][]*regexp.Regexp `json:"-"`
}

func (state *NotifyState) SendTo(e *log.Entry) []string {
	if to, ok := state.HostTo[e.Hostname]; ok {
		var toList []string
		for _, toEntry := range to {
			if lvl := state.MaxPrioIn[toEntry]; e.Level < lvl {
				continue
			}
			toList = append(toList, toEntry)
		}
		e.Hostname = state.Hostname[e.Hostname]
		return toList
	}
	return nil
}

func ReadStateFile(path string) *NotifyState {
	var state *NotifyState
	if f, err := os.Open(path); err == nil { // transform data to legacy meshviewer
		if err = json.NewDecoder(f).Decode(state); err == nil {
			log.Info("loaded", len(state.HostTo), "nodes")
			state.regexIn = make(map[string][]*regexp.Regexp)
			return state
		} else {
			log.Error("failed to unmarshal nodes:", err)
		}
	} else {
		log.Error("failed to open state notify file: ", path, ":", err)
	}
	return &NotifyState{
		Hostname:  make(map[string]string),
		HostTo:    make(map[string][]string),
		MaxPrioIn: make(map[string]log.LogLevel),
		RegexIn:   make(map[string][]string),
		regexIn:   make(map[string][]*regexp.Regexp),
	}
}

func (state *NotifyState) Saver(path string) {
	c := time.Tick(time.Minute)

	for range c {
		state.SaveJSON(path)
	}
}

// SaveJSON to path
func (state *NotifyState) SaveJSON(outputFile string) {
	tmpFile := outputFile + ".tmp"

	f, err := os.OpenFile(tmpFile, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
	if err != nil {
		log.Panic(err)
	}

	err = json.NewEncoder(f).Encode(state)
	if err != nil {
		log.Panic(err)
	}

	f.Close()
	if err := os.Rename(tmpFile, outputFile); err != nil {
		log.Panic(err)
	}
}