import type { AgentTool, AgentToolResult } from "@oh-my-pi/pi-agent-core";
import { logger, untilAborted } from "@oh-my-pi/pi-utils";
import * as z from "zod/v4";
import { ensureBankMission } from "../hindsight/bank";
import reflectDescription from "../prompts/tools/reflect.md" with { type: "text" };
import type { ToolSession } from ".";

const hindsightReflectSchema = z.object({
	query: z.string().describe("question to answer"),
	context: z.string().describe("optional context").optional(),
});

export type HindsightReflectParams = z.infer<typeof hindsightReflectSchema>;

export class HindsightReflectTool implements AgentTool<typeof hindsightReflectSchema> {
	readonly name = "reflect";
	readonly label = "Reflect";
	readonly description = reflectDescription;
	readonly parameters = hindsightReflectSchema;
	readonly strict = true;
	readonly loadMode = "discoverable";
	readonly summary = "Reflect on recent work and write hindsight memory";

	constructor(private readonly session: ToolSession) {}

	static createIf(session: ToolSession): HindsightReflectTool | null {
		if (session.settings.get("memory.backend") !== "hindsight") return null;
		return new HindsightReflectTool(session);
	}

	async execute(_id: string, params: HindsightReflectParams, signal?: AbortSignal): Promise<AgentToolResult> {
		return untilAborted(signal, async () => {
			const state = this.session.getHindsightSessionState?.();
			if (!state) {
				throw new Error("Hindsight backend is not initialised for this session.");
			}

			try {
				await ensureBankMission(state.client, state.bankId, state.config, state.missionsSet);
				const response = await state.client.reflect(state.bankId, params.query, {
					context: params.context,
					budget: state.config.recallBudget,
					tags: state.recallTags,
					tagsMatch: state.recallTagsMatch,
				});
				const text = response.text?.trim() || "No relevant information found to reflect on.";
				return {
					content: [{ type: "text", text }],
					details: {},
				};
			} catch (err) {
				logger.warn("reflect failed", { bankId: state.bankId, error: String(err) });
				throw err instanceof Error ? err : new Error(String(err));
			}
		});
	}
}