66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
|
package config
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"reflect"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/heetch/confita"
|
||
|
|
"github.com/heetch/confita/backend"
|
||
|
|
"github.com/heetch/confita/backend/env"
|
||
|
|
"github.com/heetch/confita/backend/flags"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Config struct {
|
||
|
|
PIHoleHostname string `config:"pihole_hostname"`
|
||
|
|
PIHolePassword string `config:"pihole_password"`
|
||
|
|
|
||
|
|
Port string `config:"port"`
|
||
|
|
Interval time.Duration `config:"interval"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func getDefaultConfig() *Config {
|
||
|
|
return &Config{
|
||
|
|
PIHoleHostname: "127.0.0.1",
|
||
|
|
PIHolePassword: "",
|
||
|
|
|
||
|
|
Port: "9311",
|
||
|
|
Interval: 5 * time.Second,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func Load() *Config {
|
||
|
|
loaders := []backend.Backend{
|
||
|
|
env.NewBackend(),
|
||
|
|
flags.NewBackend(),
|
||
|
|
}
|
||
|
|
|
||
|
|
loader := confita.NewLoader(loaders...)
|
||
|
|
|
||
|
|
cfg := getDefaultConfig()
|
||
|
|
err := loader.Load(context.Background(), cfg)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
cfg.show()
|
||
|
|
|
||
|
|
return cfg
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c Config) show() {
|
||
|
|
val := reflect.ValueOf(&c).Elem()
|
||
|
|
log.Println("------------------------------------")
|
||
|
|
log.Println("- PI-Hole exporter configuration -")
|
||
|
|
log.Println("------------------------------------")
|
||
|
|
for i := 0; i < val.NumField(); i++ {
|
||
|
|
valueField := val.Field(i)
|
||
|
|
typeField := val.Type().Field(i)
|
||
|
|
|
||
|
|
log.Println(fmt.Sprintf("%s : %v", typeField.Name, valueField.Interface()))
|
||
|
|
}
|
||
|
|
log.Println("------------------------------------")
|
||
|
|
}
|