updated logic to allow yaml configs

config.go

@@ -6,8 +6,10 @@ import (
 	"log"
 	"os"
 	"strconv"
+	"strings"
 
 	"github.com/joho/godotenv"
+	"gopkg.in/yaml.v3"
 )
 
 const (
@@ -17,6 +19,13 @@ const (
 	ConfigHosts  = "HOSTS"
 )
 
+type ConfigModel struct {
+	Email  string   `yaml:"Email"`
+	Token  string   `yaml:"Token"`
+	Domain string   `yaml:"Domain"`
+	Hosts  []string `yaml:"Hosts"`
+}
+
 type ConfigClient struct{}
 
 func NewConfigClient() ConfigClient {
@@ -26,6 +35,35 @@ func NewConfigClient() ConfigClient {
 	return c
 }
 
+func (cc *ConfigClient) LoadConfig() ConfigModel {
+	// load yaml first
+
+	model, err := LoadYaml("config.yaml")
+	if err != nil {
+		log.Print(err)
+	}
+
+	// refresh env to make sure its current
+	cc.RefreshEnv()
+
+	// if no domains pulled from yaml, load from env
+	if len(model.Hosts) == 0 {
+		envHosts := cc.GetConfig(ConfigHosts)
+		model.Hosts = append(model.Hosts, strings.Split(envHosts, ",")...)
+	}
+	if model.Domain == "" {
+		model.Domain = cc.GetConfig(ConfigDomain)
+	}
+	if model.Email == "" {
+		model.Email = cc.GetConfig(ConfigEmail)
+	}
+	if model.Token == "" {
+		model.Token = cc.GetConfig(ConfigToken)
+	}
+
+	return model
+}
+
 func (cc *ConfigClient) GetConfig(key string) string {
 	res, filled := os.LookupEnv(key)
 	if !filled {
@@ -70,3 +108,33 @@ func loadEnvFile() {
 		log.Fatalln(err)
 	}
 }
+
+func (cc *ConfigClient) IsConfigInCurrentDirectory(fileName string) error {
+	_, err := os.Stat(fileName)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func LoadYaml(yamlFile string) (ConfigModel, error) {
+	var results = ConfigModel{}
+
+	// check for the config in the current directory
+	_, err := os.Stat(yamlFile)
+	if err != nil {
+		return ConfigModel{}, err
+	}
+
+	content, err := os.ReadFile(yamlFile)
+	if err != nil {
+		return ConfigModel{}, err
+	}
+
+	err = yaml.Unmarshal(content, &results)
+	if err != nil {
+		return ConfigModel{}, err
+	}
+
+	return results, nil
+}

cron.go

@@ -47,7 +47,8 @@ func (c cronClient) RunCloudflareCheck(ApiToken string, Email string, Domain str
 			return
 		}
 
-		if dns.Result[0].Content != currentIp {
+		var result = dns.Result[0]
+		if result.Content != currentIp {
 			log.Println("IP Address no longer matches, sending an update")
 			err = cf.UpdateDnsEntry(domainDetails.Result[0].ID, dns, currentIp)
 			if err != nil {
@@ -57,3 +58,7 @@ func (c cronClient) RunCloudflareCheck(ApiToken string, Email string, Domain str
 	}
 	log.Println("Done!")
 }
+
+func (c cronClient) HelloWorldJob() {
+	log.Print("Hello World")
+}

go.sum

@@ -2,3 +2,6 @@ github.com/joho/godotenv v1.4.0 h1:3l4+N6zfMWnkbPEXKng2o2/MR5mSwTrBih4ZEkkz1lg=
 github.com/joho/godotenv v1.4.0/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
 github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

main.go

@@ -5,7 +5,6 @@ import (
 	"io"
 	"log"
 	"net/http"
-	"strings"
 	"time"
 )
 
@@ -38,33 +37,36 @@ func GetCurrentIpAddress() (string, error) {
 
 func main() {
 	config := NewConfigClient()
-	email := config.GetConfig(ConfigEmail)
-	if email == "" {
+	cfg := config.LoadConfig()
+
+	if cfg.Email == "" {
 		log.Println("Unable to find 'EMAIL' env value.")
 		return
 	}
 
-	token := config.GetConfig(ConfigToken)
-	if token == "" {
+	if cfg.Token == "" {
 		log.Println("Unable to find 'API_TOKEN' env value.")
+		return
 	}
 
-	domain := config.GetConfig(ConfigDomain)
-	if token == "" {
+	if cfg.Domain == "" {
 		log.Println("Unable to find 'DOMAIN' env value.")
+		return
 	}
 
-	hosts := config.GetConfig(ConfigHosts)
-	if token == "" {
+	if len(cfg.Hosts) == 0 {
 		log.Println("Unable to find 'HOSTS' env value.")
 	}
-	hostsArray := strings.Split(hosts, ",")
-	log.Println("Env Check: OK")
+
+	log.Println("Config Check: OK")
 
 	cron := NewCron()
 	log.Println("Cloudflare Check will run every 15 minutes.")
-	cron.scheduler.AddFunc("0,15,30,45 * * * *", func() { 
-		cron.RunCloudflareCheck(token, email, domain, hostsArray)
+	cron.scheduler.AddFunc("0/5 * * * *", func() {
+		cron.RunCloudflareCheck(cfg.Token, cfg.Email, cfg.Domain, cfg.Hosts)
+	})
+	cron.scheduler.AddFunc("0/1 * * * *", func() {
+		cron.HelloWorldJob()
 	})
 	cron.scheduler.Start()
 
@@ -74,4 +76,3 @@ func main() {
 	}
 
 }
-