watchyourlan/pkg/conf/getconfig.go
2022-11-16 22:37:54 +07:00

41 lines
1 KiB
Go

package conf
import (
. "github.com/aceberg/WatchYourLAN/pkg/models"
"github.com/spf13/viper"
)
const configPath = "/data/config"
func GetConfig() (config Conf) {
viper.SetDefault("IFACE", "enp1s0")
viper.SetDefault("DBPATH", "/data/db.sqlite")
viper.SetDefault("GUIIP", "localhost")
viper.SetDefault("GUIPORT", "8840")
viper.SetDefault("TIMEOUT", "60")
viper.SetDefault("SHOUTRRR_URL", "")
viper.SetDefault("THEME", "solar")
viper.SetConfigFile(configPath)
viper.SetConfigType("env")
viper.ReadInConfig()
viper.AutomaticEnv() // Get ENVIRONMENT variables
config.Iface = viper.Get("IFACE").(string)
config.DbPath = viper.Get("DBPATH").(string)
config.GuiIP = viper.Get("GUIIP").(string)
config.GuiPort = viper.Get("GUIPORT").(string)
config.Timeout = viper.GetInt("TIMEOUT")
config.ShoutUrl = viper.Get("SHOUTRRR_URL").(string)
config.Theme = viper.Get("THEME").(string)
return config
}
func WriteConfig() {
viper.SetConfigFile(configPath)
viper.SetConfigType("env")
viper.Set("THEME", AppConfig.Theme)
viper.WriteConfig()
}