czm.go

package czm

import (
	"github.com/quan-to/slog"
	"time"
)

type CloudflareZoneManager struct {
	ConfigMap ConfigMap
	SLog *slog.Instance
	Srv struct {
		CF *Cloudflare
		Reporter string
		Mods Modules
	}
}

func (e *CloudflareZoneManager) Init() {
	e.SLog = slog.Scope(`CZM`)

	e.ConfigMap.ReadConfigMap()
	e.InitServices()

	for true {
		e.VerifyAndUpdateZones()
		time.Sleep(time.Second * 60)
	}
}

func (e *CloudflareZoneManager) InitServices() {
	e.Srv.CF = e.ConfigMap.Cloudflare.Initialize()
	e.Srv.Mods.LoadAllModules()
}

func (e *CloudflareZoneManager) VerifyAndUpdateZones() {
	var zones = &e.ConfigMap.Zones

	for _, zone := range *zones {
		zoneLog := e.SLog.SubScope(zone.Hostname)

		if !e.Srv.CF.ExistsZone(&zone) {
			continue
		}

		e.Srv.CF.LoadDNSRecords(&zone)

		for _, dns := range zone.Dns {
			dns.Module.Mods = &e.Srv.Mods // append address to dns mods
			exists := e.Srv.CF.ExistsDNSRecord(&zone, &dns)
			rules := Rules{
				dns.Rules.NotExist,
				dns.Rules.Update,
			}

			if !exists {
				if rules.VerifyRule(RULES_NEXISTS_TAG) {
					zoneLog.SubScope(dns.Name).Warn(`DNS Zone dosen't exists... skip`)
					continue
				}

				err := e.Srv.CF.CreateDNSRecord(&zone, &dns)

				if err != nil {
					e.SLog.Error(err)
					continue
				}

				continue
			}

			// Resolve content of DNS with Modules
			if dns.Content == "" {
				dns.Content = dns.Module.Resolve()
			}

			if e.Srv.CF.DNSRecordHasDiff(&zone, &dns) {
				if rules.VerifyRule(RULES_UPDATE_TAG) {
					zoneLog.SubScope(dns.Name).Warn(`DNS Zone has diff but rule say is mark to skip`)
					continue
				}

				err := e.Srv.CF.UpdateDNSRecord(&zone, &dns)

				if err != nil {
					e.SLog.Error(err)
					continue
				}
			}
		}
	}
}