import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { z } from "zod";
import type { WikiPagesResponse } from "../types.js";
import { queryWikipedia, jsonResponse, textResponse, errorResponse } from "../api.js";
export function registerReferencesTool(server: McpServer): void {
server.tool(
"wiki_get_references",
"Get external references/links from a Wikipedia article",
{
title: z.string().describe("Article title"),
limit: z.number().optional().default(50).describe("Maximum number of references (default 50)"),
},
async ({ title, limit }) => {
try {
const data = await queryWikipedia<WikiPagesResponse>({
action: "query",
titles: title,
prop: "extlinks",
ellimit: String(limit),
});
const pages = data.query?.pages ?? {};
const page = Object.values(pages)[0];
if (page?.missing !== undefined) {
return textResponse(`Article "${title}" not found`);
}
const references = page?.extlinks?.map((link) => link["*"]) ?? [];
return jsonResponse({
title: page?.title,
references,
});
} catch (error) {
return errorResponse(error);
}
}
);
}
