import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { z } from "zod";
import type { WikiPagesResponse } from "../types.js";
import { queryWikipedia, textResponse, errorResponse } from "../api.js";
export function registerArticleTool(server: McpServer): void {
server.tool(
"wiki_get_article",
"Get the full content of a Wikipedia article",
{
title: z.string().describe("Article title"),
},
async ({ title }) => {
try {
const data = await queryWikipedia<WikiPagesResponse>({
action: "query",
titles: title,
prop: "extracts",
explaintext: "true",
});
const pages = data.query?.pages ?? {};
const page = Object.values(pages)[0];
if (page?.missing !== undefined) {
return textResponse(`Article "${title}" not found`);
}
return textResponse(page?.extract ?? "No content available");
} catch (error) {
return errorResponse(error);
}
}
);
}
