import { expect, it } from 'vitest';
import { z } from 'zod';

import { createAgentWithInterruptibleTool, describeIf, getModel } from './helpers';
import { parseJudgeResponse } from '../../evals/parse-judge-response';
import { Agent, Tool, Eval, evaluate, evals } from '../../index';

/**
 * Create a fruit-bowl agent with a tool that generates random fruit coordinates.
 */
function createFruitBowlAgent(provider: 'anthropic' | 'openai'): Agent {
	// Simple seedable PRNG (mulberry32)
	function mulberry32(seed: number) {
		return function () {
			let t = (seed += 0x6d2b79f5);
			t = Math.imul(t ^ (t >>> 15), t | 1);
			t ^= t + Math.imul(t ^ (t >>> 7), t | 61);
			return ((t ^ (t >>> 14)) >>> 0) / 4294967296;
		};
	}

	const createFruitBowlTool = new Tool('create_fruit_bowl')
		.description(
			'Generate a fruit bowl with random 3D coordinates for fruits. Always use this tool when asked to create a fruit bowl.',
		)
		.input(
			z.object({
				num_apples: z.number().optional().describe('Number of apples (default: 3)'),
			}),
		)
		.handler(async (input) => {
			const numApples = input.num_apples ?? 3;
			const rand = mulberry32(42);
			const fruits = Array.from({ length: numApples }, () => ({
				type: 'apple',
				x: Math.round((rand() * 20 - 10) * 10) / 10,
				y: Math.round((rand() * 20 - 10) * 10) / 10,
				z: Math.round((rand() * 20 - 10) * 10) / 10,
			}));
			return { fruits };
		});

	return new Agent('fruit-bowl-bot')
		.model(getModel(provider))
		.instructions(
			'You are a fruit bowl generator. When asked to create a fruit bowl, use the create_fruit_bowl tool and then describe the contents including each fruit type and its x, y, z coordinates.',
		)
		.tool(createFruitBowlTool);
}

const describe = describeIf('anthropic');

describe('evaluate() integration', () => {
	it('runs deterministic evals against a fruit bowl agent', async () => {
		const mentionsFruit = new Eval('mentions-fruit')
			.description('Check if response mentions apples with coordinates')
			.check(({ output }) => {
				const lower = output.toLowerCase();
				const hasApple = lower.includes('apple');
				const hasCoord = /\d+\.\d/.test(output);
				return {
					pass: hasApple && hasCoord,
					reasoning:
						hasApple && hasCoord
							? 'Mentions apples with coordinates'
							: hasApple
								? 'Mentions apples but no coordinates'
								: 'No mention of apples',
				};
			});

		const usedTool = new Eval('used-tool')
			.description('Check if create_fruit_bowl tool was called')
			.check(({ toolCalls }) => {
				const used = (toolCalls ?? []).some((tc) => tc.tool === 'create_fruit_bowl');
				return {
					pass: used,
					reasoning: used ? 'Tool was called' : 'Tool was NOT called',
				};
			});

		const agent = createFruitBowlAgent('anthropic');

		const results = await evaluate(agent, {
			dataset: [{ input: 'Create a fruit bowl with 3 apples' }],
			evals: [mentionsFruit, usedTool],
		});

		expect(results.runs).toHaveLength(1);

		const run = results.runs[0];
		expect(run.output).toBeTruthy();

		expect(run.scores['mentions-fruit'].pass).toBe(true);
		expect(run.scores['used-tool'].pass).toBe(true);

		expect(results.summary['mentions-fruit'].total).toBe(1);
		expect(results.summary['used-tool'].passed).toBe(1);
	});

	it('runs multiple dataset rows in parallel', async () => {
		const hasContent = new Eval('has-content')
			.description('Check response is non-empty')
			.check(({ output }) => ({
				pass: output.length > 10,
				reasoning: `Response length: ${output.length}`,
			}));

		const agent = createFruitBowlAgent('anthropic');

		const results = await evaluate(agent, {
			dataset: [
				{ input: 'Create a fruit bowl with 2 apples' },
				{ input: 'Create a fruit bowl with 5 apples' },
			],
			evals: [hasContent],
		});

		expect(results.runs).toHaveLength(2);
		expect(results.summary['has-content'].total).toBe(2);
		expect(results.summary['has-content'].passed).toBe(2);
	});

	it('runs built-in string similarity eval', async () => {
		const agent = new Agent('echo-agent')
			.model(getModel('anthropic'))
			.instructions(
				'When asked "What is the capital of France?", reply with exactly: "Paris". Nothing else.',
			);

		const similarity = evals.stringSimilarity();

		const results = await evaluate(agent, {
			dataset: [{ input: 'What is the capital of France?', expected: 'Paris' }],
			evals: [similarity],
		});

		expect(results.runs).toHaveLength(1);
		expect(results.runs[0].scores['string-similarity'].pass).toBe(true);
	});

	it('runs LLM-as-judge correctness eval', async () => {
		const agent = new Agent('math-agent')
			.model(getModel('anthropic'))
			.instructions('Answer math questions with just the number. No explanation.');

		const correctness = evals.correctness().model(getModel('anthropic'));

		const results = await evaluate(agent, {
			dataset: [{ input: 'What is 2 + 2?', expected: '4' }],
			evals: [correctness],
		});

		expect(results.runs).toHaveLength(1);
		expect(results.runs[0].scores['correctness'].pass).toBe(true);
		expect(results.runs[0].scores['correctness'].reasoning).toBeTruthy();
	});

	it('runs LLM correctness eval on fruit bowl agent with expected output', async () => {
		const agent = createFruitBowlAgent('anthropic');

		const correctness = evals.correctness().model(getModel('anthropic'));

		const domainHelpfulness = new Eval('domain-helpfulness')
			.description('Judge helpfulness in the context of a fruit-picking robot simulation')
			.model(getModel('anthropic'))
			.judge(async ({ input, output, llm }) => {
				const result = await llm(
					[
						'You are evaluating a response from a simple fruit-picking robot simulation tool.',
						'This is a demo/toy agent. The robot generates fruit bowls with 3D coordinates.',
						'Judge ONLY whether the response fulfills what the user asked for — not production quality.',
						'',
						`User request: ${input}`,
						`Robot response: ${output}`,
						'',
						'Did the response deliver what was asked?',
						'Respond with ONLY JSON (no markdown fences): {"pass": true/false, "reasoning": "<explanation>"}',
					].join('\n'),
				);
				return parseJudgeResponse(result.text);
			});

		const results = await evaluate(agent, {
			dataset: [
				{
					input: 'Create a fruit bowl',
					expected: 'A fruit bowl with a number of apples and their coordinates',
				},
			],
			evals: [correctness, domainHelpfulness],
		});

		expect(results.runs).toHaveLength(1);

		const run = results.runs[0];
		expect(run.output.toLowerCase()).toContain('apple');

		expect(run.scores['correctness'].pass).toBe(true);
		expect(run.scores['correctness'].reasoning).toBeTruthy();

		expect(run.scores['domain-helpfulness'].pass).toBe(true);
		expect(run.scores['domain-helpfulness'].reasoning).toBeTruthy();
	});

	it('auto-resumes interruptible tool calls during eval', async () => {
		const { createAgentWithMixedTools } = await import('./helpers');
		const agent = createAgentWithMixedTools('anthropic');

		const usedTool = new Eval('used-list-tool')
			.description('Check if list_files was called')
			.check(({ toolCalls }) => {
				const used = (toolCalls ?? []).some((tc) => tc.tool === 'list_files');
				return {
					pass: used,
					reasoning: used ? 'Called list_files' : 'Did not call list_files',
				};
			});

		const hasOutput = new Eval('has-output')
			.description('Check response is non-empty')
			.check(({ output }) => ({
				pass: output.length > 5,
				reasoning: `Output length: ${output.length}`,
			}));

		const results = await evaluate(agent, {
			dataset: [{ input: 'List files in /home' }],
			evals: [usedTool, hasOutput],
		});

		expect(results.runs).toHaveLength(1);
		expect(results.runs[0].scores['used-list-tool'].pass).toBe(true);
		expect(results.runs[0].scores['has-output'].pass).toBe(true);
	});

	it('provides tool call inputs and outputs as JSON objects, not strings', async () => {
		const agent = createFruitBowlAgent('anthropic');

		const toolTypesEval = new Eval('tool-types')
			.description('Verify tool call inputs/outputs are JSON objects')
			.check(({ toolCalls }) => {
				if (!toolCalls || toolCalls.length === 0) {
					return { pass: false, reasoning: 'No tool calls' };
				}
				for (const tc of toolCalls) {
					if (typeof tc.input === 'string') {
						return { pass: false, reasoning: `Tool "${tc.tool}" input is a string: ${tc.input}` };
					}
					if (typeof tc.output === 'string') {
						return { pass: false, reasoning: `Tool "${tc.tool}" output is a string: ${tc.output}` };
					}
				}
				return { pass: true, reasoning: 'All tool inputs/outputs are JSON objects' };
			});

		const results = await evaluate(agent, {
			dataset: [{ input: 'Create a fruit bowl with 2 apples' }],
			evals: [toolTypesEval],
		});

		expect(results.runs).toHaveLength(1);
		expect(results.runs[0].scores['tool-types'].pass).toBe(true);
		expect(results.runs[0].scores['tool-types'].reasoning).toContain('JSON objects');
	});

	it('resume("generate") result includes the resumed tool call in toolCalls', async () => {
		const agent = createAgentWithInterruptibleTool('anthropic');

		// First generate: agent suspends on delete_file
		const first = await agent.generate('Delete the file /tmp/test.txt');

		expect(first.pendingSuspend).toBeDefined();
		const { runId, toolCallId } = first.pendingSuspend![0];

		// Resume with approval
		const resumed = await agent.resume('generate', { approved: true }, { runId, toolCallId });

		// The resumed tool call must appear in toolCalls.
		// Bug: toolCalls is undefined or empty because runGenerateLoop() starts
		// with a fresh toolCallSummary and the resume-phase tool execution is
		// never captured.
		expect(resumed.toolCalls).toBeDefined();
		expect(resumed.toolCalls!.length).toBeGreaterThan(0);

		const deletedCall = resumed.toolCalls!.find((tc) => tc.tool === 'delete_file');
		expect(deletedCall).toBeDefined();
		expect(deletedCall!.output).toMatchObject({ deleted: true, path: '/tmp/test.txt' });
	});

	it('resume("generate") result includes the resumed tool call when denied', async () => {
		const agent = createAgentWithInterruptibleTool('anthropic');

		const first = await agent.generate('Delete the file /tmp/secret.txt');
		expect(first.pendingSuspend).toBeDefined();
		const { runId, toolCallId } = first.pendingSuspend![0];

		const resumed = await agent.resume('generate', { approved: false }, { runId, toolCallId });

		expect(resumed.toolCalls).toBeDefined();
		const deletedCall = resumed.toolCalls!.find((tc) => tc.tool === 'delete_file');
		expect(deletedCall).toBeDefined();
		// denied: deleted should be false
		expect(deletedCall!.output).toMatchObject({ deleted: false });
	});

	it('evaluate() includes HITL tool calls in toolCalls passed to eval scorers', async () => {
		const agent = createAgentWithInterruptibleTool('anthropic');

		const sawDeleteCall = new Eval('saw-delete-call')
			.description('Check that delete_file tool call appears in toolCalls after auto-resume')
			.check(({ toolCalls }) => {
				const found = (toolCalls ?? []).some((tc) => tc.tool === 'delete_file');
				return {
					pass: found,
					reasoning: found
						? 'delete_file present in toolCalls'
						: `delete_file missing — toolCalls: ${JSON.stringify(toolCalls ?? [])}`,
				};
			});

		const results = await evaluate(agent, {
			dataset: [
				{
					input: 'Delete the file /tmp/test.txt',
					// auto-resume with approved: true (default) so the tool completes
				},
			],
			evals: [sawDeleteCall],
		});

		expect(results.runs).toHaveLength(1);
		// Bug: this fails because result.toolCalls is empty after resume,
		// so the eval scorer receives toolCalls=[] and pass=false.
		expect(results.runs[0].scores['saw-delete-call'].pass).toBe(true);
		expect(results.runs[0].scores['saw-delete-call'].reasoning).toContain('present');
	});

	it('evaluate() output is non-empty when agent only uses an interruptible tool (no text response)', async () => {
		// If the agent produces no text and only tool output, evaluate() uses
		// toolCalls to build the composite output string. With the bug, toolCalls
		// is empty after resume and output becomes "".
		const silentAgent = new Agent('silent-tool-agent')
			.model(getModel('anthropic'))
			.instructions(
				'When asked to delete a file, call delete_file and return ONLY the raw JSON tool result. Do not add any explanatory text — your entire response must be the tool result only.',
			)
			.tool(
				new Tool('delete_file')
					.description('Delete a file')
					.input(z.object({ path: z.string() }))
					.output(z.object({ deleted: z.boolean(), path: z.string() }))
					.suspend(z.object({ message: z.string(), severity: z.string() }))
					.resume(z.object({ approved: z.boolean() }))
					.handler(async ({ path }, ctx) => {
						if (!ctx.resumeData) {
							return await ctx.suspend({
								message: `Delete "${path}"?`,
								severity: 'destructive',
							});
						}
						return { deleted: ctx.resumeData.approved, path };
					}),
			)
			.checkpoint('memory');

		const hasOutput = new Eval('has-output')
			.description('Composite output must be non-empty after HITL auto-resume')
			.check(({ output, toolCalls }) => {
				const pass = output.length > 0;
				return {
					pass,
					reasoning: pass
						? `output="${output}"`
						: `output is empty; toolCalls=${JSON.stringify(toolCalls ?? [])}`,
				};
			});

		const results = await evaluate(silentAgent, {
			dataset: [{ input: 'Delete /tmp/test.txt' }],
			evals: [hasOutput],
		});

		expect(results.runs).toHaveLength(1);
		// Bug: output is "" because toolCalls is empty, so the fallback path in
		// evaluate() that builds output from tool outputs is never triggered.
		expect(results.runs[0].scores['has-output'].pass).toBe(true);
	});
});
