import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
import {
	prepareCodeModeToolsPrompt,
} from "./custom-tool-prompt.js";
import type { SharedCodeModeRuntime } from "./shared-runtime.js";

export function registerCodeModeEvents(
	pi: ExtensionAPI,
	runtime: SharedCodeModeRuntime,
): void {
	pi.on("session_start", () => {
		runtime.resetPromptTools();
	});
	pi.on("model_select", () => {
		runtime.resetPromptTools();
	});
	pi.on("before_agent_start", (event, ctx) => {
		const requiredTools = runtime.executionKind(ctx) === "notebook"
			? ["exec", "wait", "notebook"]
			: ["exec", "wait"];
		const activeProviders = runtime.activeProviders(ctx);
		if (activeProviders.length === 0) return undefined;
		void runtime.prepare(ctx)?.catch(() => undefined);
		const documentationPath = activeProviders.find(
			(provider) => provider.documentationPath,
		)?.documentationPath;
		const promptTools = runtime.collectPromptTools(ctx);
		prepareCodeModeToolsPrompt(
			event.systemPromptOptions,
			promptTools,
			documentationPath,
			() => requiredTools.every((name) => event.systemPromptOptions.selectedTools.includes(name)),
		);
		return undefined;
	});
	pi.on("tool_result", (event) => {
		if (
			(event.toolName === "exec" || event.toolName === "wait") &&
			event.details &&
			typeof event.details === "object" &&
			"codeMode" in event.details &&
			event.details.codeMode === true &&
			"scriptError" in event.details &&
			typeof event.details.scriptError === "string"
		)
			return { isError: true };
		return undefined;
	});
}