adguard-exporter/config/configuration.go

package config

import (
	"context"
	"fmt"
	"log"
	"os"
	"reflect"
	"time"

	"github.com/heetch/confita"
	"github.com/heetch/confita/backend"
	"github.com/heetch/confita/backend/env"
	"github.com/heetch/confita/backend/flags"
)

// Config is the exporter CLI configuration.
type Config struct {
	AdguardProtocol string        `config:"adguard_protocol"`
	AdguardHostname string        `config:"adguard_hostname"`
	AdguardUsername string        `config:"adguard_username"`
	AdguardPassword string        `config:"adguard_password"`
	AdguardPort     string        `config:"adguard_port"`
	ServerPort      string        `config:"server_port"`
	Interval        time.Duration `config:"interval"`
	LogLimit        string        `config:"log_limit"`
}

func getDefaultConfig() *Config {
	return &Config{
		AdguardProtocol: "http",
		AdguardHostname: "127.0.0.1",
		AdguardUsername: "",
		AdguardPassword: "",
		AdguardPort:     "80",
		ServerPort:      "9617",
		Interval:        10 * time.Second,
		LogLimit:        "1000",
	}
}

// Load method loads the configuration by using both flag or environment variables.
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 {
		log.Printf("Could not load the configuration...")
		os.Exit(1)
	}

	//Set the adguard port based on the input configuration
	if cfg.AdguardPort == "" {
		if cfg.AdguardProtocol == "http" {
			cfg.AdguardPort = "80"
		} else if cfg.AdguardProtocol == "https" {
			cfg.AdguardPort = "443"
		} else {
			log.Printf("protocol %s is invalid. Must be http or https.", cfg.AdguardProtocol)
			os.Exit(1)
		}
	}

	cfg.show()

	return cfg
}

func (c Config) show() {
	val := reflect.ValueOf(&c).Elem()
	log.Println("---------------------------------------")
	log.Println("- AdGuard Home exporter configuration -")
	log.Println("---------------------------------------")
	for i := 0; i < val.NumField(); i++ {
		valueField := val.Field(i)
		typeField := val.Type().Field(i)

		// Do not print password or api token but do print the authentication method
		if typeField.Name != "AdguardPassword" {
			log.Println(fmt.Sprintf("%s : %v", typeField.Name, valueField.Interface()))
		} else {
			showAuthenticationMethod(typeField.Name, valueField.String())
		}
	}
	log.Println("---------------------------------------")
}

func showAuthenticationMethod(name, value string) {
	if len(value) > 0 {
		log.Println(fmt.Sprintf("AdGuard Authentication Method : %s", name))
	}
}