/** @file Pi tool adapter for robots, sitemap, and llms.txt URL maps. */
import { Type, type Static } from "typebox";

import { loadEffectiveConfig } from "../config.ts";
import { discoverSiteUrls } from "../map/discover.ts";
import { storeResponse } from "../storage/responses/store.ts";
import { toolCall } from "../tui/index.ts";
import { renderWebMapResult } from "../tui/renderers/map.ts";
import { defineWebTool } from "./infra/define.ts";
import { emitProgress } from "./infra/progress.ts";
import { toolResult } from "./infra/result.ts";
import { urlProperty } from "./infra/schemas.ts";

export const webMapSchema = Type.Object({
	url: urlProperty(),
	maxSitemaps: Type.Optional(Type.Unsafe<number>({})),
});

type Params = Static<typeof webMapSchema>;

export const webMapTool = defineWebTool({
	name: "web_map",
	label: "Map",
	description: "robots/sitemaps/llms no bodies",
	parameters: webMapSchema,
	async execute(_toolCallId, params: Params, signal, onUpdate) {
		const config = await loadEffectiveConfig();
		await emitProgress(onUpdate, {
			state: "loading",
			url: params.url,
			message: "discovering robots/sitemaps/llms",
		});
		const map = await discoverSiteUrls(
			params.url,
			{
				cacheTtlSeconds: config.scrapeDefaults.cacheTtlSeconds,
				maxAgeSeconds: config.scrapeDefaults.maxAgeSeconds,
				refresh: config.scrapeDefaults.refresh,
				...params,
			},
			{},
			signal,
		);
		const metadata = await storeResponse(map);
		await emitProgress(onUpdate, {
			state: "done",
			url: params.url,
			current: map.urls.length,
			message: "map complete",
		});
		return toolResult({
			text: `Mapped ${map.urls.length} URL(s) from ${map.sitemaps.length} sitemap candidate(s). responseId: ${metadata.responseId}`,
			data: map,
			url: params.url,
			responseId: metadata.responseId,
			fullOutputPath: metadata.fullOutputPath,
			truncated: map.urls.length > 50,
		});
	},
	renderCall: (args, theme) => toolCall("web_map", [args.url], theme),
	renderResult: (result, { expanded }, theme) => renderWebMapResult(result, expanded, theme),
});