/**
 * ask-claude — Stream Claude agent reviews into pi.
 *
 * For AGENTS to use. Delegates to specialized Claude agents or raw models
 * for review, analysis, debugging, and second opinions.
 *
 * Tool (callable by the LLM):
 *   ask_claude(prompt, agent?, model?, question?, session_id?)
 *     agent      — any agent name from ~/.claude/agents/ (e.g. "plan_review", "code_review", "oracle", "debug")
 *     model      — model override: "opus", "sonnet", or full model ID
 *     question   — specific focus prepended as a review header
 *     session_id — resume a prior conversation (returned in every response)
 *
 * Commands:
 *   /review-plan  — editor → Claude Opus plan_review → inject result
 *   /review-code  — editor → Claude Sonnet code_review → inject result
 */

import { defineTool, getMarkdownTheme } from "@mariozechner/pi-coding-agent";
import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
import { Type } from "@mariozechner/pi-ai";
import { Box, Container, Markdown, Spacer, Text } from "@mariozechner/pi-tui";
import {
	type ClaudeDetails,
	type Theme,
	renderToolCallLine,
	renderToolResultBox,
	renderToolBlock,
	formatUsage,
	buildEditDiff,
	formatAnthropicError,
	runClaude,
	type RunClaudeResult,
} from "../shared/claude-stream.js";

// =============================================================================
// Rendering (ask-claude specific — uses shared tool renderers)
// =============================================================================

function buildLabel(agent?: string, model?: string): string {
	if (agent) return `Claude [${agent}]`;
	if (model) return `Claude [${model}]`;
	return "Claude Sonnet";
}

// =============================================================================
// Tool definition
// =============================================================================

const AskClaudeParams = Type.Object({
	prompt: Type.String({
		description: "Full content to review/analyze. Include all relevant context: CLAUDE.md conventions, files explored, code or plan to review.",
	}),
	agent: Type.Optional(Type.String({
		description: "Agent name from ~/.claude/agents/ (e.g. 'plan_review', 'code_review', 'oracle', 'debug'). Omit to use model= directly.",
	})),
	model: Type.Optional(Type.String({
		description: "Model override: 'opus', 'sonnet', 'haiku', or a full model ID. When agent is set this overrides the agent's default. When agent is omitted this selects the model directly.",
	})),
	question: Type.Optional(Type.String({
		description: "Specific question or focus area prepended to the prompt (e.g. 'Focus on security', 'Are there race conditions?').",
	})),
	session_id: Type.Optional(Type.String({
		description: "Resume a prior conversation. Pass the session_id returned from a previous ask_claude call.",
	})),
});

const askClaudeTool = defineTool({
	name: "ask_claude",
	label: "Ask Claude",
	description: [
		"Ask a Claude agent or model for review, analysis, or a second opinion.",
		"agent=<name> runs a named agent from ~/.claude/agents/ (e.g. 'plan_review', 'code_review', 'oracle', 'debug').",
		"Use model= alone for free-form requests without an agent. Use question= to specify a focus.",
		"Pass session_id from a prior response to continue the same conversation across turns.",
		"CLAUDE.md and .claude/skills are loaded automatically from the project root.",
	].join(" "),
	promptSnippet: "Ask a Claude agent or model for review, analysis, or a second opinion",
	promptGuidelines: [
		"Use ask_claude(agent=<name>) to invoke a specialized agent — include all relevant context in the prompt.",
		"Use ask_claude(model='opus', question='...') for free-form deep analysis.",
		"Always include the artifact to review (plan, code, problem description) in the prompt.",
		"Pass session_id back from the previous response to continue the conversation.",
	],
	parameters: AskClaudeParams,

	async execute(_toolCallId, params, signal, onUpdate, ctx) {
		const fullPrompt = params.question
			? `## Review Focus\n\n${params.question}\n\n## Content\n\n${params.prompt}`
			: params.prompt;

		const label = buildLabel(params.agent, params.model);
		const details: ClaudeDetails = { label, done: false, blocks: [], finalText: "", isResume: !!params.session_id };

		try {
			const result: RunClaudeResult = await runClaude(fullPrompt, {
				agent: params.agent,
				model: params.model,
				sessionId: params.session_id,
				enrichEditDiffs: true,   // ask-claude wants diff enrichment
				cwd: ctx.cwd,
				signal,
				onUpdate: (partial) => {
					Object.assign(details, partial);
					onUpdate?.({
						content: [{ type: "text", text: details.finalText || "(thinking…)" }],
						details: { ...details },
					});
				},
			});

			details.done      = true;
			details.finalText = result.finalText;
			details.blocks    = result.blocks;
			details.sessionId = result.sessionId;
			details.costUsd   = result.costUsd;
			details.inputTokens  = result.inputTokens;
			details.outputTokens = result.outputTokens;
			details.cacheReadTokens  = result.cacheReadTokens;
			details.cacheWriteTokens = result.cacheWriteTokens;

			const sessionNote = result.sessionId
				? `\n\n---\n*session_id: \`${result.sessionId}\`*`
				: "";

			return {
				content: [{ type: "text", text: (result.finalText || "(no output)") + sessionNote }],
				details: { ...details },
			};
		} catch (err) {
			const errMsg = err instanceof Error ? err.message : String(err);
			details.done = true;
			details.finalText = `Error: ${errMsg}`;
			return {
				content: [{ type: "text", text: `**Claude error:** ${errMsg}` }],
				details: { ...details },
			};
		}
	},

	renderCall(args, theme, _ctx) {
		const label = buildLabel(args.agent, args.model);
		let text = theme.fg("toolTitle", theme.bold("ask_claude ")) + theme.fg("accent", `[${label}]`);
		if (args.question) {
			text += "\n  " + theme.fg("dim", theme.italic(args.question));
		} else {
			const lines = args.prompt.split("\n").filter((l) => l.trim()).slice(0, 3);
			text += "\n  " + theme.fg("dim", lines.join("\n  "));
		}
		return new Text(text, 0, 0);
	},

	renderResult(result, { isPartial }, theme, _ctx) {
		const d = result.details as ClaudeDetails | undefined;
		if (!d) return new Text(theme.fg("muted", "…"), 0, 0);

		const isDone = d.done && !isPartial;
		const statusIcon = isDone ? theme.fg("success", "✓ ") : theme.fg("warning", "⏳ ");
		const c = new Container();

		const resume = d.isResume ? theme.fg("dim", " ↩") : "";
		c.addChild(new Text(statusIcon + theme.fg("toolTitle", theme.bold(d.label)) + resume, 0, 0));

		for (const block of d.blocks ?? []) {
			if (block.type === "thinking" && block.text.trim()) {
				c.addChild(new Text(theme.fg("dim", theme.italic(block.text.trimEnd())), 0, 0));
			} else if (block.type === "tool") {
				c.addChild(renderToolBlock(block, theme as any));
			} else if (block.type === "text" && block.text.trim()) {
				c.addChild(new Spacer(1));
				if (isDone) {
					c.addChild(new Markdown(block.text.trim(), 0, 0, getMarkdownTheme()));
				} else {
					c.addChild(new Text(theme.fg("dim", block.text.trimEnd()), 0, 0));
				}
			}
		}

		if (isDone) {
			const usageLine = formatUsage(d);
			const parts: string[] = [];
			if (usageLine) parts.push(usageLine);
			if (d.sessionId) parts.push(`session:${d.sessionId.slice(0, 8)}`);
			if (parts.length > 0) {
				c.addChild(new Spacer(1));
				c.addChild(new Text(theme.fg("dim", parts.join("  ")), 0, 0));
			}
		}

		return c;
	},
});

// =============================================================================
// Extension entry point
// =============================================================================

export default function (pi: ExtensionAPI) {
	pi.registerTool(askClaudeTool);

	// ── /review-plan ───────────────────────────────────────────────────────

	pi.registerCommand("review-plan", {
		description: "Editor → Claude Opus plan_review → inject review into conversation",
		handler: async (_args, ctx) => {
			const input = await ctx.ui.editor(
				"Plan Review · Claude Opus",
				"Paste your plan or strategy. Claude will review for correctness, completeness, and risk.",
			);
			if (!input?.trim()) { ctx.ui.notify("Cancelled.", "info"); return; }
			ctx.ui.setStatus("ask-claude", "Asking Claude Opus (plan_review)…");
			try {
				const r = await runClaude(input, { agent: "plan_review", cwd: ctx.cwd, onUpdate: () => {} });
				if (!r.finalText.trim()) { ctx.ui.notify("No output from Claude.", "warning"); return; }
				pi.sendMessage(
					{ customType: "ask-claude-review", content: r.finalText.trim(), display: true,
					  details: { label: "Claude Opus · plan_review", output: r.finalText } },
					{ triggerTurn: true },
				);
			} catch (err) {
				ctx.ui.notify(`Claude error: ${err instanceof Error ? err.message : String(err)}`, "error");
			} finally { ctx.ui.setStatus("ask-claude", undefined); }
		},
	});

	// ── /review-code ───────────────────────────────────────────────────────

	pi.registerCommand("review-code", {
		description: "Editor → Claude Sonnet code_review → inject review into conversation",
		handler: async (_args, ctx) => {
			const input = await ctx.ui.editor(
				"Code Review · Claude Sonnet",
				"Paste code to review. Include the plan it implements and any specific concerns.",
			);
			if (!input?.trim()) { ctx.ui.notify("Cancelled.", "info"); return; }
			ctx.ui.setStatus("ask-claude", "Asking Claude Sonnet (code_review)…");
			try {
				const r = await runClaude(input, { agent: "code_review", cwd: ctx.cwd, onUpdate: () => {} });
				if (!r.finalText.trim()) { ctx.ui.notify("No output from Claude.", "warning"); return; }
				pi.sendMessage(
					{ customType: "ask-claude-review", content: r.finalText.trim(), display: true,
					  details: { label: "Claude Sonnet · code_review", output: r.finalText } },
					{ triggerTurn: true },
				);
			} catch (err) {
				ctx.ui.notify(`Claude error: ${err instanceof Error ? err.message : String(err)}`, "error");
			} finally { ctx.ui.setStatus("ask-claude", undefined); }
		},
	});

	// ── Message renderer for injected reviews ──────────────────────────────

	pi.registerMessageRenderer("ask-claude-review", (message, { expanded }, theme) => {
		const d = message.details as { label?: string; output?: string } | undefined;
		const label = d?.label ?? "Claude";
		const output = (d?.output ?? "").trim();

		if (expanded) {
			const c = new Container();
			c.addChild(new Text(theme.fg("accent", "◆ ") + theme.fg("toolTitle", theme.bold(label)), 0, 0));
			c.addChild(new Spacer(1));
			c.addChild(new Markdown(output, 0, 0, getMarkdownTheme()));
			return c;
		}

		let text = theme.fg("accent", "◆ ") + theme.fg("toolTitle", theme.bold(label));
		const lines = output.split("\n").filter((l) => l.trim());
		const preview = lines.slice(0, 4).join("\n");
		if (preview) {
			text += "\n" + theme.fg("dim", preview);
			if (lines.length > 4) text += "\n" + theme.fg("muted", `… ${lines.length - 4} more (Ctrl+O)`);
		}
		return new Text(text, 0, 0);
	});
}