my-log/config/load.go

package config

import (
	"encoding/json"
	"fmt"
	"os"
	fp "path/filepath"
	"time"

	"codeberg.org/danjones000/my-log/tools"
	"github.com/BurntSushi/toml"
	"github.com/caarlos0/env/v10"
	mapst "github.com/mitchellh/mapstructure"
)

var ConfigPath string
var Overrides = map[string]string{}

func init() {
	conf, _ := os.UserConfigDir()
	ConfigPath = fp.Join(conf, "my-log", "config.toml")
}

func Load() (Config, error) {
	c, _ := DefaultConfig()
	_, err := os.Stat(ConfigPath)
	if !os.IsNotExist(err) {
		_, err = toml.DecodeFile(ConfigPath, &c)
		if err != nil {
			return c, err
		}
	}
	env.Parse(&c)
	c.Outputs["stdout"] = loadStdout(c.Outputs["stdout"])
	c.Formatters["json"] = loadJsonFormat(c.Formatters["json"])

	l := ""
	for k, v := range Overrides {
		val := tools.ParseString(v)
		if val == nil {
			continue
		}
		if _, isJson := val.(json.RawMessage); isJson {
			continue
		}
		valout := fmt.Sprintf("%v", val)
		if vals, isString := val.(string); isString {
			valout = fmt.Sprintf(`"%s"`, vals)
		}
		if valt, isTime := val.(time.Time); isTime {
			valout = valt.Format(time.RFC3339)
		}
		l = l + "\n" + fmt.Sprintf("%s = %s", k, valout)
	}
	_, err = toml.Decode(l, &c)
	return c, err
}

func loadStdout(stdout Output) Output {
	st := stdoutEnabled{stdout.Enabled}
	env.Parse(&st)
	stdout.Enabled = st.Enabled
	var std Stdout
	mapst.Decode(stdout.Config, &std)
	env.Parse(&std)
	mapst.Decode(std, &stdout.Config)
	return stdout
}

func (oo Outputs) Stdout() (s Stdout, enabled bool) {
	o, ok := oo["stdout"]
	if !ok {
		return s, false
	}

	enabled = o.Enabled
	mapst.Decode(o.Config, &s)

	return
}

func loadJsonFormat(c map[string]any) map[string]any {
	jf := JsonFormat{}
	mapst.Decode(c, &jf)
	env.Parse(&jf)
	mapst.Decode(jf, &c)
	return c
}

func (ff Formatters) Json() (jf JsonFormat) {
	o, ok := ff["json"]
	if !ok {
		return
	}

	mapst.Decode(o, &jf)
	return
}