import { defineDebug } from "./pi-ext-debug.ts";

export const { dbg, span, closeDebug } = defineDebug({
	events: [
		"session.start",
		"session.shutdown",
		"compaction.plan",
		"compaction.memory",
		"compaction.outcome",
		"classifier.decision",
	],
	spans: [
		"compaction.prepare",
		"compaction.complete",
		"classifier.request",
		"summarizer.request",
	],
	fields: {
		outcome: [
			"model_unavailable",
			"cancelled",
			"fallback",
			"salvaged",
			"success",
			"ignored",
		],
		stage: [
			"model",
			"prepare",
			"classifier",
			"reconcile",
			"merge",
			"summarizer",
			"cleanup",
		],
		requestResult: [
			"accepted",
			"omitted",
			"unknown_tool",
			"invalid_record",
			"wrong_shard",
			"conflicting_records",
			"output_limit",
			"incomplete_stream",
			"provider_error",
			"record_log_error",
			"unexpected_tool",
			"empty_summary",
			"cancelled",
			"other",
		],
		decision: ["retry", "switch_model", "give_up"],
		sourceKind: ["legacy-summary", "history", "split-prefix"],
		modelRole: ["selected", "current_fallback"],
		stopReason: [
			"stop",
			"toolUse",
			"length",
			"error",
			"aborted",
			"deferred",
			"pending",
		],
		runIndex: "number",
		chunkIndex: "number",
		chunkCount: "number",
		attempt: "number",
		sourceTokens: "number",
		largestChunkTokens: "number",
		overTargetChunks: "number",
		previousAtoms: "number",
		memoryAtoms: "number",
		catalogAtoms: "number",
		catalogAvailableAtoms: "number",
		catalogChars: "number",
		toolCalls: "number",
		validRecords: "number",
		committedRecords: "number",
		receivedAtoms: "number",
		recordAtoms: "number",
		activeAtoms: "number",
		selectedAtoms: "number",
		protectedAtoms: "number",
		allocatedTokens: "number",
		atomGoals: "number",
		atomConstraints: "number",
		atomDecisions: "number",
		atomStates: "number",
		atomBlockers: "number",
		atomExacts: "number",
		atomContexts: "number",
		summaryTokens: "number",
		reportedInputTokens: "number",
		reportedOutputTokens: "number",
		reportedCacheReadTokens: "number",
		reportedCost: "number",
		outputBudget: "number",
		tokensBefore: "number",
		keepRecentTokens: "number",
		legacySummary: "boolean",
		transient: "boolean",
	},
} as const);