package agent

import "reasonix/internal/provider"

// ContextMaintenanceSnapshot is a read-only view of the current provider-bound
// context. It separates present composition from cumulative summary-call cost.
type ContextMaintenanceSnapshot struct {
	CanonicalTokens   int
	ProjectedTokens   int
	SummaryTokens     int
	LastSavedTokens   int
	SnipTrigger       int
	FoldTrigger       int
	ForceTrigger      int
	TriggerTokens     int
	CheckpointState   string
	HardInputCeiling  int
	Headroom          int
	ProjectionVersion uint64
	Blocked           bool
	LastReceipt       *ContextMaintenanceReceipt
	ContextBudget     *ContextBudgetSnapshot
}

// ContextBudgetSnapshot is the optional send-time admission view for the
// desktop Context Panel. All fields are optional for older frontends.
type ContextBudgetSnapshot struct {
	WindowMode            string
	LimitMode             string
	Source                string
	WindowTokens          int
	PromptTokens          int
	AutoOutputTokens      int
	MaxOutputTokens       int
	RequestedOutputTokens int
	EffectiveOutputTokens int
	ReserveTokens         int
	PhysicalRemaining     int
	Clipped               bool
	LastRecovery          string
	ObservedWindow        int
	ObservedPrompt        int
	ObservedCompletion    int
}

func (a *Agent) ContextMaintenanceSnapshot() ContextMaintenanceSnapshot {
	if a == nil || a.sess.conversation == nil {
		return ContextMaintenanceSnapshot{}
	}
	canonical, _ := a.sess.conversation.snapshotMessagesVersion()
	a.sess.compactionMu.Lock()
	state := a.sess.compactionState
	checkpointState := a.sess.checkpointState
	a.sess.compactionMu.Unlock()
	visible := canonical
	valid := projectionValid(state, canonical, a.currentPromptCacheKey())
	if valid {
		if projected := modelVisibleFromProjection(state.Projection, canonical); len(projected) > 0 {
			visible = projected
		}
	}
	trigger := a.compactTrigger()
	// UI checkpoint label requires a still-valid covered prefix, not merely
	// that the sidecar loaded.
	uiCheckpoint := "none"
	if valid && len(state.Projection.Messages) > 0 {
		uiCheckpoint = stateCheckpointState(checkpointState, state)
	}
	snapshot := ContextMaintenanceSnapshot{
		CanonicalTokens:   a.estimatedVisibleRequestTokens(canonical),
		ProjectedTokens:   a.estimatedVisibleRequestTokens(visible),
		FoldTrigger:       trigger,
		TriggerTokens:     trigger,
		CheckpointState:   uiCheckpoint,
		HardInputCeiling:  a.hardInputCeiling(),
		ProjectionVersion: state.Projection.ProjectionVersion,
	}
	for _, msg := range visible {
		if isCompactionSummary(msg) {
			snapshot.SummaryTokens += a.estimatedPromptTokens([]provider.Message{msg})
		}
	}
	snapshot.Headroom = max(0, snapshot.HardInputCeiling-snapshot.ProjectedTokens)
	currentHash := a.contextMaintenanceInputHash(visible)
	if state.LastReceipt != nil {
		receipt := *state.LastReceipt
		snapshot.LastReceipt = &receipt
		if receipt.Status == "applied" && (receipt.Action == "prune" || receipt.Action == "summary") {
			snapshot.LastSavedTokens = receipt.SavedTokens
		}
		// Generation-scoped blocked/failed receipts match contextMaintenanceBlocked.
		if receipt.Status == "blocked" || receipt.Status == "failed" {
			snapshot.Blocked = true
		}
	}
	// Legacy sidecars may only have top-level BlockedInputHash.
	if !snapshot.Blocked && state.BlockedInputHash != "" && state.BlockedInputHash == currentHash {
		snapshot.Blocked = true
	}
	if budget := a.lastAdmission(); budget.WindowTokens > 0 || budget.PromptTokens > 0 || budget.LastRecovery != "" && budget.LastRecovery != contextRecoveryNone {
		snapshot.ContextBudget = &ContextBudgetSnapshot{
			WindowMode: budget.WindowMode, LimitMode: budget.LimitMode, Source: budget.Source,
			WindowTokens: budget.WindowTokens, PromptTokens: budget.PromptTokens,
			AutoOutputTokens: budget.AutoOutputTokens, MaxOutputTokens: budget.MaxOutputTokens,
			RequestedOutputTokens: budget.RequestedOutputTokens, EffectiveOutputTokens: budget.EffectiveOutputTokens,
			ReserveTokens: budget.ReserveTokens, PhysicalRemaining: budget.PhysicalRemaining,
			Clipped: budget.Clipped, LastRecovery: budget.LastRecovery,
			ObservedWindow: budget.ObservedWindow, ObservedPrompt: budget.ObservedPrompt,
			ObservedCompletion: budget.ObservedCompletion,
		}
	}
	return snapshot
}

func stateCheckpointState(runtimeState string, state CompactionState) string {
	if len(state.Projection.Messages) == 0 {
		return "none"
	}
	if runtimeState == "applied" {
		return "applied"
	}
	return "restored"
}