wild-central/internal/reconcile/reconciler.go

// Package reconcile orchestrates the domain→config→daemon pipeline.
// When domains change, Reconcile regenerates HAProxy routes, dnsmasq DNS
// entries, and related subsystem configs, then reloads the affected daemons.
package reconcile

import (
	"fmt"
	"log/slog"
	"os"
	"path/filepath"
	"strings"
	"time"

	"github.com/wild-cloud/wild-central/internal/authelia"
	"github.com/wild-cloud/wild-central/internal/certbot"
	"github.com/wild-cloud/wild-central/internal/config"
	"github.com/wild-cloud/wild-central/internal/dnsmasq"
	"github.com/wild-cloud/wild-central/internal/domains"
	"github.com/wild-cloud/wild-central/internal/haproxy"
	"github.com/wild-cloud/wild-central/internal/network"
	"github.com/wild-cloud/wild-central/internal/sse"
	"github.com/wild-cloud/wild-central/internal/storage"
)

// HAProxyManager is the subset of haproxy.Manager used by reconciliation.
type HAProxyManager interface {
	GenerateWithOpts(instances []haproxy.L4Route, custom []haproxy.CustomRoute, opts haproxy.GenerateOpts) string
	WriteConfig(content string) error
	ReloadService() error
}

// DNSManager is the subset of dnsmasq.Manager used by reconciliation.
type DNSManager interface {
	UpdateConfig(cfg *config.State, entries []dnsmasq.DNSEntry, restart bool) error
	SetFilterConfPath(path string)
	GetStatus() (*dnsmasq.Status, error)
}

// DomainManager is the subset of domains.Manager used by reconciliation.
type DomainManager interface {
	List() ([]domains.Domain, error)
}

// AuthManager is the subset of authelia.Manager used by reconciliation.
type AuthManager interface {
	UserCount() int
	RestartService() error
}

// DDNSRunner is the subset of ddns.Runner used by reconciliation.
type DDNSRunner interface {
	Trigger()
}

// DNSFilterManager is the subset of dnsfilter.Manager used by reconciliation.
type DNSFilterManager interface {
	HostsFilePath() string
}

// EventBroadcaster is the subset of sse.Manager used by reconciliation.
type EventBroadcaster interface {
	Broadcast(event *sse.Event)
}

// GenerateAutheliaConfigFn generates the Authelia config from current state.
// This is a callback because the logic depends on secrets and OIDC clients
// that the reconciler doesn't own.
type GenerateAutheliaConfigFn func(state *config.State) error

// Reconciler orchestrates config regeneration when domains change.
type Reconciler struct {
	Domains                DomainManager
	HAProxy                HAProxyManager
	DNS                    DNSManager
	Auth                   AuthManager
	DDNS                   DDNSRunner
	DNSFilter              DNSFilterManager
	SSE                    EventBroadcaster
	StatePath              string
	GenerateAutheliaConfig GenerateAutheliaConfigFn
}

// Reconcile reads all registered domains and regenerates dnsmasq DNS entries
// and HAProxy routes to match.
func (r *Reconciler) Reconcile() {
	doms, err := r.Domains.List()
	if err != nil {
		slog.Error("failed to list domains", "component", "reconcile", "error", err)
		return
	}

	globalCfg, err := config.LoadState(r.StatePath)
	if err != nil {
		slog.Warn("failed to load state, using empty", "component", "reconcile", "error", err)
		globalCfg = &config.State{}
	}

	l4Routes, httpRoutes := buildRoutes(doms)

	cleanZeroByteCerts("/etc/haproxy/certs/")

	// Only include L7 HTTP routes for domains that have a valid cert.
	var activeHTTPRoutes []haproxy.HTTPRoute
	for _, route := range httpRoutes {
		if hasCertForDomain(route.Domain) {
			activeHTTPRoutes = append(activeHTTPRoutes, route)
		} else {
			slog.Warn("skipping L7 route (no cert)", "component", "reconcile", "domain", route.Domain)
		}
	}

	r.writeHAProxyConfig(l4Routes, activeHTTPRoutes, globalCfg)

	// Build and apply DNS config
	centralIP := globalCfg.Cloud.Dnsmasq.IP
	if centralIP == "" {
		centralIP, _ = network.GetWildCentralIP()
	}
	if centralIP == "" {
		centralIP = "127.0.0.1"
	}

	dnsEntries := buildDNSEntries(doms, centralIP)

	if globalCfg.Cloud.DNSFilter.Enabled {
		hostsPath := r.DNSFilter.HostsFilePath()
		if storage.FileExists(hostsPath) {
			r.DNS.SetFilterConfPath(hostsPath)
		}
	} else {
		r.DNS.SetFilterConfPath("")
	}

	if err := r.DNS.UpdateConfig(globalCfg, dnsEntries, true); err != nil {
		slog.Error("failed to update dnsmasq", "component", "reconcile", "error", err)
	} else {
		r.broadcastDNSEvent("dnsmasq:config", "DNS config regenerated from registered domains")
	}

	if len(httpRoutes) > 0 {
		ensureTLSCerts(globalCfg, doms)
	}

	r.DDNS.Trigger()

	slog.Info("networking updated", "component", "reconcile",
		"domains", len(doms),
		"l4Routes", len(l4Routes),
		"l7Routes", len(httpRoutes),
	)
}

// buildRoutes converts registered domains into HAProxy L4 and L7 route lists.
func buildRoutes(doms []domains.Domain) ([]haproxy.L4Route, []haproxy.HTTPRoute) {
	var l4Routes []haproxy.L4Route
	var httpRoutes []haproxy.HTTPRoute

	for _, dom := range doms {
		switch dom.EffectiveBackendType() {
		case domains.BackendTCPPassthrough:
			l4Routes = append(l4Routes, haproxy.L4Route{
				Name:       dom.DomainName,
				Domain:     dom.DomainName,
				BackendIP:  extractHost(dom.EffectiveBackendAddress()),
				Subdomains: dom.Subdomains,
			})
		case domains.BackendDNSOnly:
			// dns-only: no gateway routing, just DNS resolution
		case domains.BackendHTTP:
			var routeBackends []haproxy.HTTPRouteBackend
			for _, route := range dom.EffectiveRoutes() {
				rb := haproxy.HTTPRouteBackend{
					Paths:      route.Paths,
					Backend:    route.Backend.Address,
					HealthPath: route.Backend.Health,
					Headers:    route.Headers,
					IPAllow:    route.IPAllow,
				}
				if dom.Auth != nil && dom.Auth.Enabled {
					rb.AuthEnabled = true
					rb.AuthPolicy = dom.Auth.Policy
				}
				routeBackends = append(routeBackends, rb)
			}
			httpRoutes = append(httpRoutes, haproxy.HTTPRoute{
				Name:       dom.DomainName,
				Domain:     dom.DomainName,
				Subdomains: dom.Subdomains,
				Routes:     routeBackends,
			})
		}
	}

	return l4Routes, httpRoutes
}

// buildDNSEntries converts domains to dnsmasq entries with appropriate IPs.
// HTTP domains point to centralIP (HAProxy terminates TLS).
// TCP passthrough and DNS-only domains point to their backend IP directly.
func buildDNSEntries(doms []domains.Domain, centralIP string) []dnsmasq.DNSEntry {
	var entries []dnsmasq.DNSEntry
	for _, dom := range doms {
		if dom.DomainName == "" {
			continue
		}

		dnsIP := centralIP
		bt := dom.EffectiveBackendType()
		if bt == domains.BackendTCPPassthrough || bt == domains.BackendDNSOnly {
			dnsIP = extractHost(dom.EffectiveBackendAddress())
		}

		entries = append(entries, dnsmasq.DNSEntry{
			Domain:   dom.DomainName,
			IP:       dnsIP,
			Wildcard: dom.Subdomains,
		})
	}
	return entries
}

// writeHAProxyConfig generates, validates, and writes the HAProxy config.
// On validation failure, it identifies broken domains and retries without them.
func (r *Reconciler) writeHAProxyConfig(l4Routes []haproxy.L4Route, httpRoutes []haproxy.HTTPRoute, globalCfg *config.State) {
	genOpts := haproxy.GenerateOpts{
		HTTPRoutes: httpRoutes,
		CertsDir:   "/etc/haproxy/certs/",
	}

	if globalCfg.Cloud.Authelia.Enabled && globalCfg.Cloud.Authelia.Domain != "" {
		genOpts.AuthEnabled = true
		genOpts.AuthBackend = "127.0.0.1:9091"
		genOpts.AuthDomain = globalCfg.Cloud.Authelia.Domain
		genOpts.AuthSessionDomain = authelia.SessionDomainFromAuthDomain(globalCfg.Cloud.Authelia.Domain)

		if r.GenerateAutheliaConfig != nil {
			if err := r.GenerateAutheliaConfig(globalCfg); err != nil {
				slog.Warn("failed to regenerate authelia config", "component", "reconcile", "error", err)
			} else if r.Auth.UserCount() > 0 {
				_ = r.Auth.RestartService()
			}
		}
	}

	cfg := r.HAProxy.GenerateWithOpts(l4Routes, nil, genOpts)

	if err := r.HAProxy.WriteConfig(cfg); err != nil {
		brokenDomains := haproxy.FindBrokenServices(cfg, haproxy.ParseValidationErrors(err.Error()))
		if len(brokenDomains) > 0 {
			slog.Error("excluding broken domains and retrying",
				"component", "reconcile", "broken", brokenDomains, "error", err)

			exclude := map[string]bool{}
			for _, d := range brokenDomains {
				exclude[d] = true
			}

			var filteredL4 []haproxy.L4Route
			for _, route := range l4Routes {
				if !exclude[route.Domain] {
					filteredL4 = append(filteredL4, route)
				}
			}
			var filteredHTTP []haproxy.HTTPRoute
			for _, route := range httpRoutes {
				if !exclude[route.Domain] {
					filteredHTTP = append(filteredHTTP, route)
				}
			}

			genOpts.HTTPRoutes = filteredHTTP
			cfg = r.HAProxy.GenerateWithOpts(filteredL4, nil, genOpts)
			if err := r.HAProxy.WriteConfig(cfg); err != nil {
				slog.Error("retry also failed", "component", "reconcile", "error", err)
			} else if err := r.HAProxy.ReloadService(); err != nil {
				slog.Warn("failed to reload HAProxy", "component", "reconcile", "error", err)
			} else {
				r.broadcastEvent("haproxy:config", "HAProxy config regenerated (excluded broken domains)")
			}
		} else {
			slog.Error("failed to write HAProxy config (no broken domains identified)", "component", "reconcile", "error", err)
		}
	} else if err := r.HAProxy.ReloadService(); err != nil {
		slog.Warn("failed to reload HAProxy", "component", "reconcile", "error", err)
	} else {
		r.broadcastEvent("haproxy:config", "HAProxy config regenerated from registered domains")
	}
}

// cleanZeroByteCerts removes 0-byte cert files that would poison HAProxy validation.
func cleanZeroByteCerts(certsDir string) {
	entries, err := os.ReadDir(certsDir)
	if err != nil {
		return
	}
	for _, e := range entries {
		if strings.HasSuffix(e.Name(), ".pem") {
			if info, err := e.Info(); err == nil && info.Size() == 0 {
				slog.Warn("removing 0-byte cert file", "component", "reconcile", "file", e.Name())
				_ = os.Remove(filepath.Join(certsDir, e.Name()))
			}
		}
	}
}

// broadcastEvent sends a simple SSE event.
func (r *Reconciler) broadcastEvent(eventType, message string) {
	if r.SSE == nil {
		return
	}
	r.SSE.Broadcast(&sse.Event{
		ID:           fmt.Sprintf("%s-%d", strings.SplitN(eventType, ":", 2)[0], time.Now().UnixNano()),
		Type:         eventType,
		InstanceName: "global",
		Timestamp:    time.Now(),
		Data:         map[string]any{"message": message},
	})
}

// broadcastDNSEvent sends a dnsmasq SSE event including current status.
func (r *Reconciler) broadcastDNSEvent(eventType, message string) {
	if r.SSE == nil {
		return
	}
	status, err := r.DNS.GetStatus()
	if err != nil {
		status = nil
	}
	r.SSE.Broadcast(&sse.Event{
		ID:           fmt.Sprintf("dnsmasq-%d", time.Now().UnixNano()),
		Type:         eventType,
		InstanceName: "global",
		Timestamp:    time.Now(),
		Data: map[string]any{
			"message": message,
			"status":  status,
		},
	})
}

// ensureTLSCerts logs which certificates are missing for registered domains.
// Does NOT auto-provision — cert provisioning is user-initiated.
func ensureTLSCerts(globalCfg *config.State, doms []domains.Domain) {
	gatewayDomain := ""
	if parts := strings.SplitN(globalCfg.Cloud.Central.Domain, ".", 2); len(parts) == 2 {
		gatewayDomain = parts[1]
	}

	for _, dom := range doms {
		if dom.TLS != domains.TLSTerminate || dom.DomainName == "" {
			continue
		}

		// Check if covered by wildcard
		if gatewayDomain != "" && strings.HasSuffix(dom.DomainName, "."+gatewayDomain) {
			wildcardCert := certbot.HAProxyCertPath(gatewayDomain)
			if _, err := os.Stat(wildcardCert); err != nil {
				slog.Warn("no wildcard cert for gateway domain — provision via Certificates page", "component", "reconcile",
					"domain", dom.DomainName, "needed", "*."+gatewayDomain)
			}
			continue
		}

		// Check individual cert
		if _, err := os.Stat(certbot.HAProxyCertPath(dom.DomainName)); err != nil {
			slog.Warn("no cert for domain — provision via Certificates page", "component", "reconcile",
				"domain", dom.DomainName)
		}
	}
}

// hasCertForDomain checks if a valid (non-empty) cert exists for a domain —
// either an individual cert (<domain>.pem) or a wildcard cert that covers it.
func hasCertForDomain(domain string) bool {
	if isValidCertFile(certbot.HAProxyCertPath(domain)) {
		return true
	}
	parts := strings.SplitN(domain, ".", 2)
	if len(parts) == 2 {
		if isValidCertFile(certbot.HAProxyCertPath(parts[1])) {
			return true
		}
	}
	return false
}

func isValidCertFile(path string) bool {
	info, err := os.Stat(path)
	return err == nil && info.Size() > 0
}

func extractHost(addr string) string {
	for i := len(addr) - 1; i >= 0; i-- {
		if addr[i] == ':' {
			return addr[:i]
		}
	}
	return addr
}