package main import ( "os" "path/filepath" "strings" "github.com/knadh/koanf/parsers/toml" "github.com/knadh/koanf/providers/confmap" "github.com/knadh/koanf/providers/env" "github.com/knadh/koanf/providers/file" "go.arsenm.dev/logger" "go.arsenm.dev/logger/log" ) var cfgDir string func init() { etcPath := "/etc/itd.toml" // Set up logger log.Logger = logger.NewPretty(os.Stderr) // Get user's configuration directory userCfgDir, err := os.UserConfigDir() if err != nil { panic(err) } cfgDir = filepath.Join(userCfgDir, "itd") // If config dir is not readable if _, err = os.ReadDir(cfgDir); err != nil { // Create config dir with 700 permissions err = os.MkdirAll(cfgDir, 0o700) if err != nil { panic(err) } } // Get current and old config paths cfgPath := filepath.Join(cfgDir, "itd.toml") oldCfgPath := filepath.Join(userCfgDir, "itd.toml") // If old config path exists if _, err = os.Stat(oldCfgPath); err == nil { // Move old config to new path err = os.Rename(oldCfgPath, cfgPath) if err != nil { panic(err) } } // Set config defaults setCfgDefaults() // Load and watch config files loadAndwatchCfgFile(etcPath) loadAndwatchCfgFile(cfgPath) // Load envireonment variables k.Load(env.Provider("ITD_", "_", func(s string) string { return strings.ToLower(strings.TrimPrefix(s, "ITD_")) }), nil) } func loadAndwatchCfgFile(filename string) { provider := file.Provider(filename) if cfgError := k.Load(provider, toml.Parser()); cfgError != nil { log.Warn("Error while trying to read config file").Str("filename", filename).Err(cfgError).Send() } // Watch for changes and reload when detected provider.Watch(func(_ interface{}, err error) { if err != nil { return } if cfgError := k.Load(provider, toml.Parser()); cfgError != nil { log.Warn("Error while trying to read config file").Str("filename", filename).Err(cfgError).Send() } }) } func setCfgDefaults() { k.Load(confmap.Provider(map[string]interface{}{ "bluetooth.adapter": "hci0", "socket.path": "/tmp/itd/socket", "conn.reconnect": true, "conn.whitelist.enabled": false, "conn.whitelist.devices": []string{}, "on.connect.notify": true, "on.reconnect.notify": true, "on.reconnect.setTime": true, "notifs.translit.use": []string{"eASCII"}, "notifs.translit.custom": []string{}, "notifs.ignore.sender": []string{}, "notifs.ignore.summary": []string{"InfiniTime"}, "notifs.ignore.body": []string{}, "music.vol.interval": 5, }, "."), nil) }