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 registerLinksTool(server: McpServer): void {
server.tool(
"wiki_get_links",
"Get internal Wikipedia links from an article",
{
title: z.string().describe("Article title"),
limit: z.number().optional().default(100).describe("Maximum number of links (default 100)"),
},
async ({ title, limit }) => {
try {
const data = await queryWikipedia<WikiPagesResponse>({
action: "query",
titles: title,
prop: "links",
pllimit: String(limit),
plnamespace: "0", // Main namespace only (articles)
});
const pages = data.query?.pages ?? {};
const page = Object.values(pages)[0];
if (page?.missing !== undefined) {
return textResponse(`Article "${title}" not found`);
}
const links = page?.links?.map((link) => link.title) ?? [];
return jsonResponse({
title: page?.title,
links,
});
} catch (error) {
return errorResponse(error);
}
}
);
}
