import type { InstanceAiEvalExecutionResult, InstanceAiEvalNodeResult } from '@n8n/api-types';

import type { WorkflowResponse } from '../clients/n8n-client';
import { buildVerificationArtifact, selectScenarioWorkflowId } from '../harness/scenario-execution';
import type { ExecutionScenario } from '../types';

function makeNodeResult(
	overrides: Partial<InstanceAiEvalNodeResult> = {},
): InstanceAiEvalNodeResult {
	return {
		outputs: {},
		outputCount: 0,
		iterationCount: 0,
		interceptedRequests: [],
		executionMode: 'real',
		...overrides,
	};
}

const scenario: ExecutionScenario = {
	name: 'happy-path',
	description: 'baseline',
	dataSetup: 'three posts, two match the filter',
	successCriteria: 'two posts forwarded downstream',
};

function makeEvalResult(
	nodeResults: Record<string, InstanceAiEvalNodeResult>,
): InstanceAiEvalExecutionResult {
	return {
		executionId: 'exec-1',
		success: true,
		nodeResults,
		errors: [],
		hints: {
			globalContext: '',
			triggerContent: { foo: 1 },
			nodeHints: {},
			warnings: [],
			bypassPinData: {},
		},
		mockedCredentials: [],
	};
}

describe('buildVerificationArtifact', () => {
	it('splits artifact into a workflow block (cacheable) and a scenario block (fresh)', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'pipeline',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'Trigger',
					type: 'n8n-nodes-base.scheduleTrigger',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: { Trigger: { main: [[]] } },
		};
		const artifact = buildVerificationArtifact(scenario, makeEvalResult({}), [wf]);
		expect(artifact.workflowContext).toContain('Workflow structure');
		expect(artifact.workflowContext).toContain('Trigger');
		expect(artifact.workflowContext).toContain('Connections');
		expect(artifact.scenarioContext).toContain('happy-path');
		expect(artifact.scenarioContext).toContain('Execution trace');
		expect(artifact.scenarioContext).not.toContain('Workflow structure');
	});

	it('uses the routed workflow JSON for verification context', () => {
		const primary: WorkflowResponse = {
			id: 'w1',
			name: 'primary-router',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'Primary Switch',
					type: 'n8n-nodes-base.switch',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
				{
					id: 'b',
					name: 'Primary Action',
					type: 'n8n-nodes-base.noOp',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {
				'Primary Switch': { main: [[{ node: 'Primary Action', type: 'main', index: 0 }]] },
			},
		};
		const routed: WorkflowResponse = {
			id: 'w2',
			name: 'routed-router',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'c',
					name: 'Switch',
					type: 'n8n-nodes-base.switch',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
				{
					id: 'd',
					name: 'Routed Action',
					type: 'n8n-nodes-base.noOp',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {
				Switch: { main: [[{ node: 'Routed Action', type: 'main', index: 0 }]] },
			},
		};
		const evalResult = makeEvalResult({
			Switch: makeNodeResult({
				outputs: { main: [[{ json: { ok: true } }]] },
				outputCount: 1,
				iterationCount: 1,
			}),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [primary, routed], 'w2');

		expect(artifact.workflowContext).toContain('Routed Action');
		expect(artifact.workflowContext).not.toContain('Primary Action');
		expect(artifact.scenarioContext).toContain('Did not run');
		expect(artifact.scenarioContext).toContain('Routed Action');
		expect(artifact.scenarioContext).not.toContain('Primary Action');
	});

	it('does not route scenarios to sub-workflow or error-trigger workflows', () => {
		const logger = { info: vi.fn() };
		const primary: WorkflowResponse = {
			id: 'primary',
			name: 'incoming webhook',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'Webhook',
					type: 'n8n-nodes-base.webhook',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {},
		};
		const subWorkflow: WorkflowResponse = {
			id: 'sub',
			name: 'urgent invoice handler',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'b',
					name: 'Execute Workflow Trigger',
					type: 'n8n-nodes-base.executeWorkflowTrigger',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
				{
					id: 'c',
					name: 'Urgent Invoice Slack',
					type: 'n8n-nodes-base.slack',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {},
		};
		const errorWorkflow: WorkflowResponse = {
			id: 'error',
			name: 'urgent invoice error workflow',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'd',
					name: 'Error Trigger',
					type: 'n8n-nodes-base.errorTrigger',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {},
		};

		const selected = selectScenarioWorkflowId(
			{ ...scenario, dataSetup: 'urgent invoice should alert Slack' },
			'primary',
			[primary, subWorkflow, errorWorkflow],
			logger as never,
		);

		expect(selected).toBe('primary');
		expect(logger.info).not.toHaveBeenCalled();
	});

	it('routes to the entry point when the build saved a sub-workflow first', () => {
		// Sub-workflow tool-result arrives first → build.workflowId is the sub.
		// Executing it directly starts once with an empty payload; the caller's
		// entry point is the trigger-bearing sibling.
		const logger = { info: vi.fn() };
		const sub: WorkflowResponse = {
			id: 'process-order-sub',
			name: 'Process Order',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'Execute Workflow Trigger',
					type: 'n8n-nodes-base.executeWorkflowTrigger',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {},
		};
		const main: WorkflowResponse = {
			id: 'orders-main',
			name: 'Order Intake',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'b',
					name: 'Webhook',
					type: 'n8n-nodes-base.webhook',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
				{
					id: 'c',
					name: 'Run Process Order',
					type: 'n8n-nodes-base.executeWorkflow',
					typeVersion: 1,
					position: [0, 0],
					parameters: { workflowId: { value: 'process-order-sub' } },
				},
			],
			connections: {},
		};

		const selected = selectScenarioWorkflowId(
			{ ...scenario, name: 'three-orders-three-runs', dataSetup: 'Process Order runs three times' },
			'process-order-sub',
			[sub, main],
			logger as never,
		);

		expect(selected).toBe('orders-main');
		expect(logger.info).toHaveBeenCalledWith(expect.stringContaining('orders-main'));
	});

	it('demotes executeWorkflow-referenced candidates even when scenario tokens favor the sub', () => {
		const logger = { info: vi.fn() };
		const sub: WorkflowResponse = {
			id: 'sub-with-own-trigger',
			name: 'Process Order',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'Process Order Webhook',
					type: 'n8n-nodes-base.webhook',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {},
		};
		const main: WorkflowResponse = {
			id: 'orders-main',
			name: 'Order Intake',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'b',
					name: 'Webhook',
					type: 'n8n-nodes-base.webhook',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
				{
					id: 'c',
					name: 'Run Sub',
					type: 'n8n-nodes-base.executeWorkflow',
					typeVersion: 1,
					position: [0, 0],
					parameters: { workflowId: 'sub-with-own-trigger' },
				},
			],
			connections: {},
		};

		const selected = selectScenarioWorkflowId(
			{ ...scenario, name: 'process-order-runs', dataSetup: 'Process Order handles each order' },
			'orders-main',
			[main, sub],
			logger as never,
		);

		expect(selected).toBe('orders-main');
	});

	it('labels Filter branches with downstream node names so verifier can tell where items went', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'pipeline',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'Filter',
					type: 'n8n-nodes-base.filter',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
				{
					id: 'b',
					name: 'Aggregate Posts',
					type: 'n8n-nodes-base.aggregate',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {
				Filter: {
					main: [[{ node: 'Aggregate Posts', type: 'main', index: 0 }], []],
				},
			},
		};
		const evalResult = makeEvalResult({
			Filter: makeNodeResult({
				outputs: {
					main: [[{ json: { id: 1, kept: true } }], [{ json: { id: 2, dropped: true } }]],
				},
				outputCount: 2,
				iterationCount: 1,
			}),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [wf]);

		expect(artifact.scenarioContext).toContain('Output [main branch 0] → Aggregate Posts');
		expect(artifact.scenarioContext).toContain(
			'Output [main branch 1] → (no downstream connection)',
		);
		expect(artifact.scenarioContext).toContain('"id": 1');
		expect(artifact.scenarioContext).toContain('"id": 2');
	});

	it('renders Switch branches per route with their downstream targets', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'router',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'Switch',
					type: 'n8n-nodes-base.switch',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
				{
					id: 'b',
					name: 'Slack',
					type: 'n8n-nodes-base.slack',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
				{
					id: 'c',
					name: 'Email',
					type: 'n8n-nodes-base.emailSend',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {
				Switch: {
					main: [
						[{ node: 'Slack', type: 'main', index: 0 }],
						[{ node: 'Email', type: 'main', index: 0 }],
					],
				},
			},
		};
		const evalResult = makeEvalResult({
			Switch: makeNodeResult({
				outputs: { main: [[{ json: { route: 'a' } }], [{ json: { route: 'b' } }]] },
				outputCount: 2,
				iterationCount: 1,
			}),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [wf]);

		expect(artifact.scenarioContext).toContain('Output [main branch 0] → Slack');
		expect(artifact.scenarioContext).toContain('Output [main branch 1] → Email');
	});

	it('renders AI sub-node outputs under their non-main connection type', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'agent',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'OpenAI Chat Model',
					type: '@n8n/n8n-nodes-langchain.lmChatOpenAi',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
				{
					id: 'b',
					name: 'AI Agent',
					type: '@n8n/n8n-nodes-langchain.agent',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {
				'OpenAI Chat Model': {
					ai_languageModel: [[{ node: 'AI Agent', type: 'ai_languageModel', index: 0 }]],
				},
			},
		};
		const evalResult = makeEvalResult({
			'OpenAI Chat Model': makeNodeResult({
				outputs: { ai_languageModel: [[{ json: { reply: 'hi' } }]] },
				outputCount: 1,
				iterationCount: 1,
			}),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [wf]);

		expect(artifact.scenarioContext).toContain('Output [ai_languageModel branch 0] → AI Agent');
		expect(artifact.scenarioContext).toContain('"reply": "hi"');
	});

	it('flags truncation and reports the full count', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'big-output',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'HTTP',
					type: 'n8n-nodes-base.httpRequest',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: { HTTP: { main: [[]] } },
		};
		const evalResult = makeEvalResult({
			HTTP: makeNodeResult({
				outputs: { main: [Array.from({ length: 10 }, (_, i) => ({ json: { i } }))] },
				outputCount: 42,
				truncated: true,
				iterationCount: 1,
			}),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [wf]);

		expect(artifact.scenarioContext).toContain('full count across all branches: 42');
	});

	it('keeps pinned trigger nodes out of "Did not run" (they have synthetic input, no runData)', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'scheduled-pipeline',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'Schedule Trigger',
					type: 'n8n-nodes-base.scheduleTrigger',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: { 'Schedule Trigger': { main: [[]] } },
		};
		const evalResult = makeEvalResult({
			'Schedule Trigger': makeNodeResult({ executionMode: 'pinned' }),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [wf]);

		expect(artifact.scenarioContext).toContain(
			'**Pinned nodes** (synthetic input): Schedule Trigger',
		);
		expect(artifact.scenarioContext).toContain('**Did not run** (no execution data): none');
	});

	it('head/tail-truncates oversized JSON output blocks and reports chars saved', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'big-output',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'HTTP',
					type: 'n8n-nodes-base.httpRequest',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: { HTTP: { main: [[]] } },
		};
		const bigItems = Array.from({ length: 200 }, (_, i) => ({
			json: { i, blob: 'x'.repeat(200) },
		}));
		const evalResult = makeEvalResult({
			HTTP: makeNodeResult({
				outputs: { main: [bigItems] },
				outputCount: 200,
				iterationCount: 1,
			}),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [wf]);

		expect(artifact.scenarioContext).toContain('chars truncated]');
		// Head and tail survive the cut.
		expect(artifact.scenarioContext).toContain('"i": 0');
		expect(artifact.scenarioContext).toContain('"i": 199');
		expect(artifact.truncationSavedChars ?? 0).toBeGreaterThan(0);
	});

	it('elides the middle of an oversized intercepted-request list', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'paginated',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'HTTP',
					type: 'n8n-nodes-base.httpRequest',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: { HTTP: { main: [[]] } },
		};
		const evalResult = makeEvalResult({
			HTTP: makeNodeResult({
				interceptedRequests: Array.from({ length: 30 }, (_, i) => ({
					method: 'GET',
					url: `https://api.example.com/page/${i}`,
					nodeType: 'n8n-nodes-base.httpRequest',
					mockResponse: { page: i },
				})),
				iterationCount: 1,
			}),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [wf]);

		// First and last pages render; the middle is elided with a marker.
		expect(artifact.scenarioContext).toContain('https://api.example.com/page/0');
		expect(artifact.scenarioContext).toContain('https://api.example.com/page/29');
		expect(artifact.scenarioContext).toContain('18 further requests omitted for size');
		expect(artifact.scenarioContext).not.toContain('https://api.example.com/page/15');
		expect(artifact.truncationSavedChars ?? 0).toBeGreaterThan(0);
	});

	it('reports zero truncation savings for small traces', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'small',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'HTTP',
					type: 'n8n-nodes-base.httpRequest',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: { HTTP: { main: [[]] } },
		};
		const evalResult = makeEvalResult({
			HTTP: makeNodeResult({
				outputs: { main: [[{ json: { ok: true } }]] },
				outputCount: 1,
				iterationCount: 1,
			}),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [wf]);

		expect(artifact.truncationSavedChars).toBe(0);
		expect(artifact.scenarioContext).not.toContain('chars truncated]');
	});

	it('tags loop iterations and first-error iteration in the trace header', () => {
		const wf: WorkflowResponse = {
			id: 'w1',
			name: 'loop',
			active: false,
			versionId: 'v1',
			nodes: [
				{
					id: 'a',
					name: 'Loop',
					type: 'n8n-nodes-base.splitInBatches',
					typeVersion: 1,
					position: [0, 0],
					parameters: {},
				},
			],
			connections: {},
		};
		const evalResult = makeEvalResult({
			Loop: makeNodeResult({
				outputs: { main: [[{ json: { i: 4 } }]] },
				outputCount: 1,
				iterationCount: 5,
				firstErrorIteration: 3,
			}),
		});

		const artifact = buildVerificationArtifact(scenario, evalResult, [wf]);

		expect(artifact.scenarioContext).toContain('ran 5×');
		expect(artifact.scenarioContext).toContain('first error at iter 3');
	});
});