// Package desired bridges the hub's "Down" channel (the control-envelope generation signal +
// the desired-state fetch) to the reconcile engine's provider (slice 10A). It implements
// hub.EnvelopeObserver: on each heartbeat it inspects the envelope's DesiredGeneration and, only
// when it has ADVANCED past the cached one, fetches the full desired-state and updates the
// engine's CachingProvider. So the heartbeat stays light; the heavy state moves on change.
//
// It lives in its own package because it imports BOTH hub (the wire client + types) and reconcile
// (the domain DesiredState + CachingProvider). hub does not import it (the loop sees only the
// hub.EnvelopeObserver seam) and reconcile does not import it — so there is no import cycle.
package desired

import (
	"context"
	"log/slog"
	"time"

	"gitea.dooplex.hu/admin/felhom-agent/internal/hub"
	"gitea.dooplex.hu/admin/felhom-agent/internal/reconcile"
)

// Fetcher fetches this host's desired-state from the hub. Satisfied by *hub.Client.
type Fetcher interface {
	FetchDesiredState(ctx context.Context) (*hub.DesiredStateResponse, error)
}

// RawConsumer receives the FULL fetched desired-state document after each successful
// generation-advance fetch (S3 seam — internal/wgtunnel consumes its wireguard block this way
// without the reconcile engine learning about tunnels). Implementations must not block: do the
// cheap store-and-nudge, never network/exec inline.
type RawConsumer interface {
	OnDesiredState(ctx context.Context, resp *hub.DesiredStateResponse)
}

// Syncer keeps the engine's CachingProvider in step with the hub's authoritative desired-state.
type Syncer struct {
	fetcher   Fetcher
	provider  *reconcile.CachingProvider
	consumers []RawConsumer
	logger    *slog.Logger
}

// AddConsumer registers a raw desired-state consumer (nil-safe no-op). Not concurrency-safe —
// call during wiring, before the hub loop starts.
func (s *Syncer) AddConsumer(c RawConsumer) {
	if c != nil {
		s.consumers = append(s.consumers, c)
	}
}

// NewSyncer builds a Syncer over the hub fetcher and the engine's provider.
func NewSyncer(fetcher Fetcher, provider *reconcile.CachingProvider, logger *slog.Logger) *Syncer {
	if logger == nil {
		logger = slog.Default()
	}
	return &Syncer{fetcher: fetcher, provider: provider, logger: logger}
}

// OnEnvelope implements hub.EnvelopeObserver. It fetches + caches the desired-state ONLY when the
// envelope's generation advances past the provider's cached generation — otherwise it is a no-op
// (the cached state is already current). A fetch failure keeps the last-known state (the engine
// keeps reconciling toward it) and is retried on the next advance signal.
func (s *Syncer) OnEnvelope(ctx context.Context, env *hub.ControlEnvelope) {
	if env == nil || s.provider == nil {
		return
	}
	have := s.provider.Generation()
	if env.DesiredGeneration <= have {
		return // cached: the heavy desired-state moves only on a generation advance
	}
	s.logger.Debug("desired: generation advanced — fetching desired-state",
		"have_generation", have, "envelope_generation", env.DesiredGeneration)
	start := time.Now()
	resp, err := s.fetcher.FetchDesiredState(ctx)
	if err != nil {
		s.logger.Warn("desired: fetch failed; keeping cached desired-state",
			"have_generation", have, "envelope_generation", env.DesiredGeneration, "err", err)
		return
	}
	s.logger.Debug("desired: fetched", "generation", resp.Generation,
		"duration_ms", time.Since(start).Milliseconds())
	state := mapWire(resp.DesiredState, s.logger)
	// Cache against the FETCHED generation (not the envelope's) — robust to a generation that
	// advanced again between the heartbeat and this fetch (we won't re-fetch the same state).
	s.provider.Update(resp.Generation, state)
	s.logger.Info("desired: updated from hub",
		"generation", resp.Generation, "guests", len(state.Guests))
	// S3: fan the raw document out to registered consumers (wgtunnel etc). A panicking consumer
	// is contained — the guest reconcile path must never break over a tunnel add-on.
	for _, c := range s.consumers {
		s.notifyConsumer(ctx, c, resp)
	}
	if env.HasSignedOps {
		// 10A only notes the flag; fetching + verifying + executing signed ops is slice 10B.
		s.logger.Info("desired: hub reports pending signed ops (fetch/execute is slice 10B)")
	}
}

// notifyConsumer delivers one raw document with panic containment.
func (s *Syncer) notifyConsumer(ctx context.Context, c RawConsumer, resp *hub.DesiredStateResponse) {
	defer func() {
		if r := recover(); r != nil {
			s.logger.Error("desired: raw consumer panicked (contained)", "panic", r)
		}
	}()
	c.OnDesiredState(ctx, resp)
}

// mapWire maps the hub wire desired-state to the reconcile domain. 10A acts only on guests; the
// forward-compat fields (restore_directive — 10D — etc.) are carried on the wire and logged, but
// not translated into actions here.
func mapWire(w hub.WireDesiredState, logger *slog.Logger) reconcile.DesiredState {
	guests := make(map[int]reconcile.DesiredGuest, len(w.Guests))
	for _, g := range w.Guests {
		dg := reconcile.DesiredGuest{
			VMID:         g.VMID,
			Spec:         g.Spec,
			Description:  g.Description,
			Decommission: g.Decommission,
		}
		switch g.Run {
		case "running":
			dg.Run = reconcile.RunRunning
		case "stopped":
			dg.Run = reconcile.RunStopped
		default:
			dg.Run = reconcile.RunUnspecified // unknown/empty → unmanaged (planner leaves run alone)
		}
		guests[g.VMID] = dg
	}
	if w.RestoreDirective != nil {
		// The reconcile mapping does NOT act on the directive; the DR consumer (raw-consumer seam,
		// S5 internal/dr) surfaces it as an inspectable restore PLAN — no restore is executed here.
		logger.Info("desired: restore_directive present (surfaced as a restore PLAN by the DR consumer; not acted on in the reconcile mapping)",
			"mode", w.RestoreDirective.Mode)
	}
	return reconcile.DesiredState{Guests: guests}
}