@primer/mcp 0.4.0-rc.fbf889cc5 → 0.5.0-rc.01176969a

package/dist/index.js

@@ -1,2 +1,2 @@
-import { t as server } from "./server-BTJ9W5jN.js";
+import { t as server } from "./server-rl4CHqO6.js";
 export { server };

package/dist/{server-BTJ9W5jN.js → server-rl4CHqO6.js}

@@ -692,7 +692,7 @@ function runStylelint(css) {
 //#region src/server.ts
 const server = new McpServer({
 	name: "Primer",
-	version: "0.4.0"
+	version: "0.5.0"
 });
 const turndownService = new TurndownService();
 const allTokensWithGuidelines = loadAllTokensWithGuidelines();
@@ -766,6 +766,43 @@ server.registerTool("get_component", {
 		content: []
 	};
 });
+server.registerTool("get_component_batch", {
+	description: "Retrieve documentation for multiple Primer React components in one call. Pass all component names you need at once. Docs are fetched in parallel to avoid repeated MCP round-trips when resolving several components. For a single component, prefer get_component instead.",
+	inputSchema: { names: z.array(z.string()).min(2).max(10).describe("Component names to retrieve (e.g. [\"ActionMenu\", \"Button\", \"Pagination\"])") },
+	annotations: { readOnlyHint: true }
+}, async ({ names }) => {
+	const seenNames = /* @__PURE__ */ new Set();
+	const deduped = names.filter((name) => {
+		const normalizedName = name.toLowerCase();
+		if (seenNames.has(normalizedName)) return false;
+		seenNames.add(normalizedName);
+		return true;
+	});
+	const components = listComponents();
+	return { content: await Promise.all(deduped.map(async (name) => {
+		const match = components.find((component) => component.name === name || component.name.toLowerCase() === name.toLowerCase());
+		if (!match) return {
+			type: "text",
+			text: `## ${name}\n\nStatus: not-found. No component named \`${name}\` exists in @primer/react. Use \`list_components\` for valid names.`
+		};
+		const controller = new AbortController();
+		const timeout = setTimeout(() => controller.abort(), 1e4);
+		try {
+			const llmsUrl = new URL(`/product/components/${match.slug}/llms.txt`, "https://primer.style");
+			const llmsResponse = await fetch(llmsUrl, { signal: controller.signal });
+			if (llmsResponse.ok) return {
+				type: "text",
+				text: await llmsResponse.text()
+			};
+		} catch (_) {} finally {
+			clearTimeout(timeout);
+		}
+		return {
+			type: "text",
+			text: `## ${match.name}\n\nStatus: fetch-error. Failed to load documentation for ${match.name}.`
+		};
+	})) };
+});
 server.registerTool("get_component_examples", {
 	description: "Get examples for how to use a component from Primer React",
 	inputSchema: { name: z.string().describe("The name of the component to retrieve") },

package/dist/stdio.js

@@ -1,4 +1,4 @@
-import { t as server } from "./server-BTJ9W5jN.js";
+import { t as server } from "./server-rl4CHqO6.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 //#region src/transports/stdio.ts
 const transport = new StdioServerTransport();

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@primer/mcp",
   "description": "An MCP server that connects AI tools to the Primer Design System",
-  "version": "0.4.0-rc.fbf889cc5",
+  "version": "0.5.0-rc.01176969a",
   "type": "module",
   "bin": {
     "mcp": "./bin/mcp.js"
@@ -29,6 +29,7 @@
   "scripts": {
     "clean": "rimraf dist",
     "build": "rolldown -c",
+    "test": "vitest --run",
     "type-check": "tsc --noEmit",
     "watch": "rolldown -c -w"
   },
@@ -36,16 +37,17 @@
     "@modelcontextprotocol/sdk": "^1.24.0",
     "@primer/octicons": "^19.15.5",
     "@primer/primitives": "10.x || 11.x",
-    "@primer/react": "^38.30.0",
+    "@primer/react": "^38.33.0",
     "cheerio": "^1.0.0",
     "turndown": "^7.2.0",
     "zod": "^4.3.5"
   },
   "devDependencies": {
     "@modelcontextprotocol/inspector": "^0.16.6",
+    "@primer/vitest-config": "^0.0.0",
     "@types/turndown": "^5.0.5",
     "rimraf": "^6.0.1",
-    "rolldown": "^1.1.2",
+    "rolldown": "^1.1.4",
     "rolldown-plugin-dts": "^0.26.0",
     "typescript": "^6.0.3"
   }

package/src/server.test.ts

@@ -0,0 +1,125 @@
+import {Client} from '@modelcontextprotocol/sdk/client/index.js'
+import {InMemoryTransport} from '@modelcontextprotocol/sdk/inMemory.js'
+import {afterAll, afterEach, beforeAll, beforeEach, describe, expect, it, vi} from 'vitest'
+import {server} from './server'
+
+describe('get_component_batch', () => {
+  const client = new Client({name: 'mcp-server-test', version: '1.0.0'})
+
+  beforeAll(async () => {
+    const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair()
+    await server.connect(serverTransport)
+    await client.connect(clientTransport)
+  })
+
+  beforeEach(() => {
+    vi.stubGlobal('fetch', vi.fn<typeof fetch>())
+  })
+
+  afterEach(() => {
+    vi.useRealTimers()
+    vi.unstubAllGlobals()
+  })
+
+  afterAll(async () => {
+    await client.close()
+    await server.close()
+  })
+
+  const callBatch = (names: string[]) => {
+    return client.callTool({
+      name: 'get_component_batch',
+      arguments: {names},
+    })
+  }
+
+  it('requires between 2 and 10 names', async () => {
+    const tooFew = await callBatch(['Button'])
+    const tooMany = await callBatch(Array.from({length: 11}, (_, index) => `Component${index}`))
+
+    expect(tooFew.isError).toBe(true)
+    expect(tooMany.isError).toBe(true)
+    expect(fetch).not.toHaveBeenCalled()
+  })
+
+  it('deduplicates names case-insensitively and preserves result order', async () => {
+    const fetchMock = vi.mocked(fetch)
+    fetchMock.mockImplementation(async input => {
+      const url = new URL(input.toString())
+      return new Response(`${url.pathname} docs`)
+    })
+
+    const result = await callBatch(['Button', 'button', 'Dialog'])
+
+    expect(fetchMock).toHaveBeenCalledTimes(2)
+    expect(result.content).toEqual([
+      {type: 'text', text: '/product/components/button/llms.txt docs'},
+      {type: 'text', text: '/product/components/dialog/llms.txt docs'},
+    ])
+  })
+
+  it('starts component fetches in parallel', async () => {
+    const fetchMock = vi.mocked(fetch)
+    const pendingResponses: Array<(response: Response) => void> = []
+    fetchMock.mockImplementation(
+      () =>
+        new Promise(resolve => {
+          pendingResponses.push(resolve)
+        }),
+    )
+
+    const resultPromise = callBatch(['Button', 'Dialog'])
+    await vi.waitFor(() => expect(fetchMock).toHaveBeenCalledTimes(2))
+
+    for (const resolve of pendingResponses) {
+      resolve(new Response('docs'))
+    }
+
+    await expect(resultPromise).resolves.toMatchObject({
+      content: [
+        {type: 'text', text: 'docs'},
+        {type: 'text', text: 'docs'},
+      ],
+    })
+  })
+
+  it('returns missing components and fetch failures in-band', async () => {
+    vi.mocked(fetch).mockResolvedValue(new Response(null, {status: 500}))
+
+    const result = await callBatch(['MissingComponent', 'button'])
+
+    expect(result.isError).not.toBe(true)
+    expect(result.content).toEqual([
+      {
+        type: 'text',
+        text: '## MissingComponent\n\nStatus: not-found. No component named `MissingComponent` exists in @primer/react. Use `list_components` for valid names.',
+      },
+      {
+        type: 'text',
+        text: '## Button\n\nStatus: fetch-error. Failed to load documentation for Button.',
+      },
+    ])
+  })
+
+  it('times out stalled requests without leaving timers active', async () => {
+    vi.useFakeTimers()
+    const fetchMock = vi.mocked(fetch)
+    fetchMock.mockImplementation(
+      (_input, init) =>
+        new Promise((_resolve, reject) => {
+          init?.signal?.addEventListener('abort', () => reject(init.signal?.reason), {once: true})
+        }),
+    )
+
+    const resultPromise = callBatch(['Button', 'Dialog'])
+    await vi.advanceTimersByTimeAsync(10_000)
+
+    await expect(resultPromise).resolves.toMatchObject({
+      content: [
+        {type: 'text', text: '## Button\n\nStatus: fetch-error. Failed to load documentation for Button.'},
+        {type: 'text', text: '## Dialog\n\nStatus: fetch-error. Failed to load documentation for Dialog.'},
+      ],
+    })
+    expect(vi.getTimerCount()).toBe(0)
+  })
+})

package/src/server.ts

@@ -161,6 +161,73 @@ server.registerTool(
   },
 )
 
+server.registerTool(
+  'get_component_batch',
+  {
+    description:
+      'Retrieve documentation for multiple Primer React components in one call. ' +
+      'Pass all component names you need at once. Docs are fetched in parallel to avoid repeated MCP round-trips when resolving several components. ' +
+      'For a single component, prefer get_component instead.',
+    inputSchema: {
+      names: z
+        .array(z.string())
+        .min(2)
+        .max(10)
+        .describe('Component names to retrieve (e.g. ["ActionMenu", "Button", "Pagination"])'),
+    },
+    annotations: {readOnlyHint: true},
+  },
+  async ({names}) => {
+    const seenNames = new Set<string>()
+    const deduped = names.filter(name => {
+      const normalizedName = name.toLowerCase()
+      if (seenNames.has(normalizedName)) {
+        return false
+      }
+      seenNames.add(normalizedName)
+      return true
+    })
+    const components = listComponents()
+
+    const results = await Promise.all(
+      deduped.map(async name => {
+        const match = components.find(
+          component => component.name === name || component.name.toLowerCase() === name.toLowerCase(),
+        )
+        if (!match) {
+          return {
+            type: 'text' as const,
+            text: `## ${name}\n\nStatus: not-found. No component named \`${name}\` exists in @primer/react. Use \`list_components\` for valid names.`,
+          }
+        }
+
+        const controller = new AbortController()
+        const timeout = setTimeout(() => controller.abort(), 10_000)
+
+        try {
+          const llmsUrl = new URL(`/product/components/${match.slug}/llms.txt`, 'https://primer.style')
+          const llmsResponse = await fetch(llmsUrl, {signal: controller.signal})
+          if (llmsResponse.ok) {
+            const text = await llmsResponse.text()
+            return {type: 'text' as const, text}
+          }
+        } catch (_: unknown) {
+          // Fall through to the in-band error result so other component requests can succeed.
+        } finally {
+          clearTimeout(timeout)
+        }
+
+        return {
+          type: 'text' as const,
+          text: `## ${match.name}\n\nStatus: fetch-error. Failed to load documentation for ${match.name}.`,
+        }
+      }),
+    )
+
+    return {content: results}
+  },
+)
+
 server.registerTool(
   'get_component_examples',
   {