ddns-updater/internal/config.go

package internal

import (
	"fmt"
	"log"
	"os"
	"strconv"
	"time"
)

const (
	defaultCheckInterval       = 5 * 60
	envKeyDnsToCheck           = "DNS_NAMES"
	envKeyPublicIpResolverTag  = "PUBLIC_IP_RESOLVER"
	envKeyCloudflareApiKey     = "CLOUDFLARE_API_KEY"
	envKeyCloudflareZone       = "CLOUDFLARE_ZONE"
	envKeyDirectadminUser      = "DA_USER"
	envKeyDirectadminKey       = "DA_LOGIN_KEY"
	envKeyDirectadminUrl       = "DA_LOGIN_URL"
	envKeyOnChangeComment      = "ON_CHANGE_COMMENT"
	envKeyCheckIntervalSeconds = "CHECK_INTERVAL_SECONDS"
	envKeyNotifiers            = "NOTIFIERS"
)

type Config struct {
	DnsRecordsToCheck   []string
	PublicIpResolverTag string
	DNSProviderTag      string
	DirectadminUsername string
	DirectadminKey      string
	DirectadminUrl      string
	ApiToken            string
	WebhookToken        string
	CloudflareZone      string
	OnChangeComment     string
	Notifiers           []string
	CheckInterval       time.Duration
}

func (c Config) Validate() error {
	switch c.DNSProviderTag {
	case "cloudflare":
		if c.ApiToken == "" {
			return fmt.Errorf("empty api token env key %s", envKeyCloudflareApiKey)
		}

		// if c.CloudflareZone == "" {
		// 	return fmt.Errorf("empty zone in env key %s", envKeyCloudflareZone)
		// }

	case "directadmin":
		if c.DirectadminUrl == "" {
			return fmt.Errorf("empty DirectAdmin URL env key %s", envKeyDirectadminUrl)
		}

		if c.DirectadminUsername == "" {
			return fmt.Errorf("empty Username in env key %s", envKeyDirectadminUser)
		}

		if c.DirectadminKey == "" {
			return fmt.Errorf("empty Login Key in env key %s", envKeyDirectadminKey)
		}
	}

	if len(c.DnsRecordsToCheck) == 0 {
		return fmt.Errorf("no dns to check defined in env key %s", envKeyDnsToCheck)
	}

	return nil
}

func NewConfig() Config {
	checkInterval, err := strconv.ParseInt(os.Getenv(envKeyCheckIntervalSeconds), 10, 64)
	if err != nil {
		log.Printf("wrong `%s` value. Check interval set default(%ds)", envKeyCheckIntervalSeconds, defaultCheckInterval)
		checkInterval = defaultCheckInterval
	}

	return Config{
		DnsRecordsToCheck:   parseCommaDelimited(os.Getenv(envKeyDnsToCheck)),
		PublicIpResolverTag: os.Getenv(envKeyPublicIpResolverTag),
		ApiToken:            os.Getenv(envKeyCloudflareApiKey),
		DirectadminUsername: os.Getenv(envKeyDirectadminUser),
		DirectadminKey:      os.Getenv(envKeyDirectadminKey),
		DirectadminUrl:      os.Getenv(envKeyDirectadminUrl),
		CloudflareZone:      os.Getenv(envKeyCloudflareZone),
		OnChangeComment:     os.Getenv(envKeyOnChangeComment),
		Notifiers:           parseCommaDelimited(os.Getenv(envKeyNotifiers)),
		CheckInterval:       time.Duration(checkInterval) * time.Second,
		WebhookToken:        os.Getenv("WEBHOOK_TOKEN"),
	}
}