79 lines
1.8 KiB
Go
79 lines
1.8 KiB
Go
package config
|
|
|
|
import (
|
|
"automation/device"
|
|
"encoding/base64"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/kelseyhightower/envconfig"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type config struct {
|
|
Hue struct {
|
|
Token string `yaml:"token" envconfig:"HUE_TOKEN"`
|
|
IP string `yaml:"ip" envconfig:"HUE_IP"`
|
|
} `yaml:"hue"`
|
|
|
|
Ntfy struct {
|
|
Topic string `yaml:"topic" envconfig:"NTFY_TOPIC"`
|
|
} `yaml:"ntfy"`
|
|
|
|
MQTT struct {
|
|
Host string `yaml:"host" envconfig:"MQTT_HOST"`
|
|
Port int `yaml:"port" envconfig:"MQTT_PORT"`
|
|
Username string `yaml:"username" envconfig:"MQTT_USERNAME"`
|
|
Password string `yaml:"password" envconfig:"MQTT_PASSWORD"`
|
|
ClientID string `yaml:"client_id" envconfig:"MQTT_CLIENT_ID"`
|
|
} `yaml:"mqtt"`
|
|
|
|
Kasa struct {
|
|
Outlets map[device.InternalName]string `yaml:"outlets"`
|
|
} `yaml:"kasa"`
|
|
|
|
Computer map[device.InternalName]struct {
|
|
MACAddress string `yaml:"mac"`
|
|
Url string `yaml:"url"`
|
|
} `yaml:"computers"`
|
|
|
|
Google struct {
|
|
Username string `yaml:"username" envconfig:"GOOGLE_USERNAME"`
|
|
Credentials Credentials `yaml:"credentials" envconfig:"GOOGLE_CREDENTIALS"`
|
|
} `yaml:"google"`
|
|
}
|
|
|
|
type Credentials []byte
|
|
|
|
func (c *Credentials) Decode(value string) error {
|
|
b, err := base64.StdEncoding.DecodeString(value)
|
|
*c = b
|
|
|
|
return err
|
|
}
|
|
|
|
func Get() config {
|
|
// First load the config from the yaml file
|
|
f, err := os.Open("config.yml")
|
|
if err != nil {
|
|
log.Fatalln("Failed to open config file", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
var cfg config
|
|
decoder := yaml.NewDecoder(f)
|
|
err = decoder.Decode(&cfg)
|
|
if err != nil {
|
|
log.Fatalln("Failed to parse config file", err)
|
|
}
|
|
|
|
// Then load values from environment
|
|
// This can be used to either override the config or pass in secrets
|
|
err = envconfig.Process("", &cfg)
|
|
if err != nil {
|
|
log.Fatalln("Failed to parse environmet config", err)
|
|
}
|
|
|
|
return cfg
|
|
}
|