package trajectory

import (
	"encoding/json"
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"

	"reasonix/internal/event"
	"reasonix/internal/evidence"
)

type capabilitySink struct {
	events       []event.Event
	readiness    []evidence.ReadinessAudit
	anchorSafety []event.AnchorSafetyAudit
	recoveries   []event.ProtocolRecoveryAudit
	outcomes     []evidence.OutcomeSample
	reports      []event.CompletionReportAudit
	workspace    []event.WorkspaceMutation
	runBudgets   []event.RunBudgetSample
	turns        int
}

func (s *capabilitySink) Emit(e event.Event) { s.events = append(s.events, e) }
func (s *capabilitySink) RecordReadinessAudit(a evidence.ReadinessAudit) {
	s.readiness = append(s.readiness, a)
}
func (s *capabilitySink) RecordAnchorSafetyAudit(a event.AnchorSafetyAudit) {
	s.anchorSafety = append(s.anchorSafety, a)
}
func (s *capabilitySink) RecordProtocolRecovery(a event.ProtocolRecoveryAudit) {
	s.recoveries = append(s.recoveries, a)
}
func (s *capabilitySink) RecordTurnCompletion() { s.turns++ }
func (s *capabilitySink) RecordOutcomeProgress(sample evidence.OutcomeSample) {
	s.outcomes = append(s.outcomes, sample)
}
func (s *capabilitySink) RecordCompletionReport(a event.CompletionReportAudit) {
	s.reports = append(s.reports, a)
}
func (s *capabilitySink) RecordWorkspaceMutation(m event.WorkspaceMutation) {
	s.workspace = append(s.workspace, m)
}
func (s *capabilitySink) RecordRunBudget(sample event.RunBudgetSample) {
	s.runBudgets = append(s.runBudgets, sample)
}

func readRecords(t *testing.T, path string) []Record {
	t.Helper()
	data, err := os.ReadFile(path)
	if err != nil {
		t.Fatalf("read trajectory: %v", err)
	}
	var out []Record
	for line := range strings.SplitSeq(strings.TrimSpace(string(data)), "\n") {
		var r Record
		if err := json.Unmarshal([]byte(line), &r); err != nil {
			t.Fatalf("bad record %q: %v", line, err)
		}
		out = append(out, r)
	}
	return out
}

func TestRecorderAppendsOrderedTimestampedRecords(t *testing.T) {
	path := filepath.Join(t.TempDir(), "run.trajectory.jsonl")
	inner := &capabilitySink{}
	now := time.UnixMilli(1754500000000)
	r, err := New(inner, path, func() time.Time { return now })
	if err != nil {
		t.Fatalf("New: %v", err)
	}

	r.Emit(event.Event{Kind: event.ToolDispatch, Tool: event.Tool{ID: "c1", Name: "bash", Args: `{"command":"ls"}`}})
	now = now.Add(120 * time.Millisecond)
	r.Emit(event.Event{Kind: event.ToolResult, Tool: event.Tool{
		ID: "c1", Name: "bash", Output: "ok", DurationMs: 100,
		StartedAt: 1754500000010, EndedAt: 1754500000110,
	}})
	r.Emit(event.Event{Kind: event.Reasoning, Text: "thinking about the next step"})
	if err := r.Close(); err != nil {
		t.Fatalf("Close: %v", err)
	}

	recs := readRecords(t, path)
	if len(recs) != 3 {
		t.Fatalf("got %d records, want 3", len(recs))
	}
	for i, rec := range recs {
		if rec.Seq != uint64(i+1) {
			t.Errorf("record %d seq = %d, want %d", i, rec.Seq, i+1)
		}
		if rec.SchemaVersion != SchemaVersion {
			t.Errorf("record %d schema = %d, want %d", i, rec.SchemaVersion, SchemaVersion)
		}
		if rec.Event == nil {
			t.Fatalf("record %d has no event payload", i)
		}
	}
	if recs[0].TS != 1754500000000 || recs[1].TS != 1754500000120 {
		t.Errorf("timestamps = %d, %d, want recorder-clock values", recs[0].TS, recs[1].TS)
	}
	if recs[1].Event.Tool == nil || recs[1].Event.Tool.StartedAt != 1754500000010 || recs[1].Event.Tool.EndedAt != 1754500000110 {
		t.Errorf("tool result record lost execution bounds: %+v", recs[1].Event.Tool)
	}
	if recs[2].Event.Kind != "reasoning" || recs[2].Event.Text != "thinking about the next step" {
		t.Errorf("reasoning record = %+v", recs[2].Event)
	}
	if len(inner.events) != 3 {
		t.Errorf("inner sink saw %d events, want 3", len(inner.events))
	}
}

func TestRecorderRecordsAndForwardsOptionalCapabilities(t *testing.T) {
	path := filepath.Join(t.TempDir(), "run.trajectory.jsonl")
	inner := &capabilitySink{}
	r, err := New(inner, path, nil)
	if err != nil {
		t.Fatalf("New: %v", err)
	}

	r.RecordReadinessAudit(evidence.ReadinessAudit{Result: evidence.ReadinessBlocked, MissingVerification: 2})
	r.RecordProtocolRecovery(event.ProtocolRecoveryAudit{Kind: event.ProtocolRecoveryMissingReasoningDetected})
	r.RecordTurnCompletion()
	r.RecordOutcomeProgress(evidence.OutcomeSample{
		Round: 3, Exploration: 2, Objective: 1, LegacyGain: 4,
		Runway: 0, RunwayDry: 6, RunwayIdle: 6, RunwaySpent: true,
	})
	r.RecordDelegationAdmission(event.DelegationAdmissionAudit{Tool: "research", Verdict: "deny", Reason: "local_fix_no_external_need", Intent: "mutation"})
	r.RecordCompletionReport(event.CompletionReportAudit{
		Verdict: "partial", Changes: 2, ChangesUnreviewed: 1, Gaps: 1, GapKinds: []string{"unreviewed_change"},
	})
	r.RecordWorkspaceMutation(event.WorkspaceMutation{ToolName: "write_file"})
	r.RecordRunBudget(event.RunBudgetSample{Currency: "USD"})
	if err := r.Close(); err != nil {
		t.Fatalf("Close: %v", err)
	}

	recs := readRecords(t, path)
	if len(recs) != 6 {
		t.Fatalf("got %d records, want 6", len(recs))
	}
	if recs[0].ReadinessAudit == nil || recs[0].ReadinessAudit.Result != "blocked" || recs[0].ReadinessAudit.MissingVerification != 2 {
		t.Errorf("readiness record = %+v", recs[0].ReadinessAudit)
	}
	if recs[1].ProtocolRecovery != string(event.ProtocolRecoveryMissingReasoningDetected) {
		t.Errorf("protocol recovery record = %q", recs[1].ProtocolRecovery)
	}
	if !recs[2].TurnCompletion {
		t.Errorf("turn completion record = %+v", recs[2])
	}
	if recs[3].OutcomeProgress == nil || recs[3].OutcomeProgress.Round != 3 || recs[3].OutcomeProgress.Objective != 1 || recs[3].OutcomeProgress.LegacyGain != 4 {
		t.Errorf("outcome progress record = %+v", recs[3].OutcomeProgress)
	}
	if rec := recs[3].OutcomeProgress; rec.Runway == nil || *rec.Runway != 0 || rec.RunwayDry != 6 || rec.RunwayIdle != 6 || !rec.RunwaySpent {
		t.Errorf("runway shadow record = %+v, want explicit spent balance", rec)
	}
	if recs[4].DelegationAdmission == nil || recs[4].DelegationAdmission.Verdict != "deny" || recs[4].DelegationAdmission.Tool != "research" {
		t.Errorf("delegation admission record = %+v", recs[4].DelegationAdmission)
	}
	if rec := recs[5].CompletionReport; rec == nil || rec.Verdict != "partial" || rec.ChangesUnreviewed != 1 || len(rec.GapKinds) != 1 {
		t.Errorf("completion report record = %+v", recs[5].CompletionReport)
	}
	if len(inner.workspace) != 1 || len(inner.runBudgets) != 1 {
		t.Errorf("host capabilities not forwarded: workspace=%d run_budget=%d", len(inner.workspace), len(inner.runBudgets))
	}
	if len(inner.readiness) != 1 || len(inner.recoveries) != 1 || inner.turns != 1 || len(inner.outcomes) != 1 || len(inner.reports) != 1 {
		t.Errorf("inner capabilities = %d/%d/%d/%d/%d, want 1/1/1/1/1", len(inner.readiness), len(inner.recoveries), inner.turns, len(inner.outcomes), len(inner.reports))
	}
}

func TestOutcomeProgressRunwayIsAdditiveAndPresenceAware(t *testing.T) {
	var old Record
	if err := json.Unmarshal([]byte(`{"schema_version":1,"outcome_progress":{"round":1}}`), &old); err != nil {
		t.Fatalf("decode old record: %v", err)
	}
	if old.OutcomeProgress == nil || old.OutcomeProgress.Runway != nil {
		t.Fatalf("old record runway = %+v, want unobserved nil", old.OutcomeProgress)
	}

	zero := 0
	data, err := json.Marshal(Record{
		SchemaVersion:   SchemaVersion,
		OutcomeProgress: &OutcomeProgress{Round: 1, Runway: &zero, RunwaySpent: true},
	})
	if err != nil {
		t.Fatalf("encode new record: %v", err)
	}
	if !strings.Contains(string(data), `"runway":0`) {
		t.Fatalf("zero balance was omitted: %s", data)
	}
	// A previous reader ignores the additive fields and keeps its known data.
	var legacy struct {
		OutcomeProgress *struct {
			Round int `json:"round"`
		} `json:"outcome_progress"`
	}
	if err := json.Unmarshal(data, &legacy); err != nil || legacy.OutcomeProgress == nil || legacy.OutcomeProgress.Round != 1 {
		t.Fatalf("legacy decode = %+v, %v", legacy, err)
	}
}

func TestRecorderForwardsAfterCloseWithoutRecording(t *testing.T) {
	path := filepath.Join(t.TempDir(), "run.trajectory.jsonl")
	inner := &capabilitySink{}
	r, err := New(inner, path, nil)
	if err != nil {
		t.Fatalf("New: %v", err)
	}
	r.Emit(event.Event{Kind: event.Text, Text: "before"})
	if err := r.Close(); err != nil {
		t.Fatalf("Close: %v", err)
	}
	r.Emit(event.Event{Kind: event.Text, Text: "after"})

	if len(readRecords(t, path)) != 1 {
		t.Fatalf("post-close event must not be recorded")
	}
	if len(inner.events) != 2 {
		t.Fatalf("inner sink saw %d events, want 2 (forwarding survives Close)", len(inner.events))
	}
}

func TestNewFailsOnUnwritablePath(t *testing.T) {
	if _, err := New(event.Discard, filepath.Join(t.TempDir(), "missing", "run.jsonl"), nil); err == nil {
		t.Fatal("New must fail when the parent directory does not exist")
	}
}