@miyucy/storybook-mcp 1.0.5 → 1.0.7

package/dist/index.js

@@ -1,7 +1,6 @@
 #!/usr/bin/env node
-import { McpServer, ResourceTemplate, } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { McpServer, } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { McpError, ErrorCode } from "@modelcontextprotocol/sdk/types.js";
 import path from "node:path";
 import fs from "node:fs/promises";
 import process from "node:process";
@@ -28,17 +27,8 @@ class Storybook {
         await this.server.connect(new StdioServerTransport());
     }
     registerStorybook() {
-        this.server.resource("Storybookの情報を返します", "storybook://stories", async (uri) => {
-            const inf = [...this.componentInformation];
-            if (inf.length === 0) {
-                return { contents: [{ uri: uri.href, text: "No stories found." }] };
-            }
-            return {
-                contents: [{ uri: uri.href, text: JSON.stringify(inf, null, 2) }],
-            };
-        });
-        this.server.tool("get_storybooks", "Storybookの情報を返します", {}, async () => {
-            const inf = [...this.componentInformation];
+        this.server.tool("get_storybooks", "Storybookの情報を返します\nクエリーは大文字・小文字を区別しません", { query: z.string().nullish() }, async ({ query }) => {
+            const inf = this.queryComponentInformation([...this.componentInformation], query ?? null);
             if (inf.length === 0) {
                 return {
                     content: [{ type: "text", text: "No stories found." }],
@@ -51,41 +41,6 @@ class Storybook {
                 })),
             };
         });
-        this.server.resource("Storybookのタイトルや説明をクエリーします\nクエリーは大文字・小文字を区別しません", new ResourceTemplate("storybook://stories?{query}", { list: undefined }), async (uri, { query }) => {
-            const inf = [...this.componentInformation];
-            if (inf.length === 0) {
-                return { contents: [{ uri: uri.href, text: "No stories found." }] };
-            }
-            // queryがない場合はエラー
-            if (!query) {
-                throw new McpError(ErrorCode.InvalidParams, "Query is required.");
-            }
-            const filtered = this.queryComponentInformation(inf, query);
-            return {
-                contents: [
-                    { uri: uri.href, text: JSON.stringify(filtered, null, 2) },
-                ],
-            };
-        });
-        this.server.tool("query_storybooks", "Storybookのタイトルや説明をクエリーします\nクエリーは大文字・小文字を区別しません", { query: z.string() }, async ({ query }) => {
-            const inf = [...this.componentInformation];
-            if (inf.length === 0) {
-                return {
-                    content: [{ type: "text", text: "No stories found." }],
-                };
-            }
-            // queryがない場合はエラー
-            if (!query) {
-                throw new McpError(ErrorCode.InvalidParams, "Query is required.");
-            }
-            const filtered = this.queryComponentInformation(inf, query);
-            return {
-                content: filtered.map((entry) => ({
-                    type: "text",
-                    text: JSON.stringify(entry, null, 2),
-                })),
-            };
-        });
     }
     queryComponentInformation(inf, query) {
         const isMatch = (entry, q) => {
@@ -102,17 +57,17 @@ class Storybook {
                     return false;
                 }));
         };
-        if (Array.isArray(query)) {
-            return inf.filter((entry) => query.some((q) => isMatch(entry, q)));
+        if (!query) {
+            return inf;
         }
         return inf.filter((entry) => isMatch(entry, query));
     }
     async loadStorybookData() {
         // storybook-static/index.jsonを読み込む
-        console.debug("Loading storybook data...");
+        // console.debug("Loading storybook data...");
         const content = await fs.readFile(path.resolve(this.storybookRoot, "storybook-static", "index.json"), "utf-8");
         const data = JSON.parse(content);
-        console.debug("Loaded storybook data successfully.");
+        // console.debug("Loaded storybook data successfully.");
         if (data.v !== 5) {
             throw new Error("Invalid version (expected 5)");
         }
@@ -145,9 +100,9 @@ class Storybook {
             }
             entries.push(entry);
         }
-        console.debug(`Found ${entries.length} entries.`);
+        // console.debug(`Found ${entries.length} entries.`);
         this.loadStorybookEntries(entries);
-        console.debug("Component information loaded successfully.");
+        // console.debug("Component information loaded successfully.");
     }
     //
     async loadStorybookEntries(entries) {
@@ -182,19 +137,23 @@ class Storybook {
             const task = tasks.pop();
             if (!task) {
                 worker.terminate();
-                return processNext();
+                processNext();
+                return;
             }
             worker.postMessage(task);
         };
         for (let i = 0; i < this.AVAILABLE_PARALLELISM; i++) {
             const __filename = fileURLToPath(import.meta.url);
             const __dirname = path.dirname(__filename);
-            const worker = new Worker(path.resolve(__dirname, "worker.js"));
+            const worker = new Worker(path.resolve(__dirname, "parser.js"));
             worker.on("message", (event) => {
                 // console.log("message", event);
-                const { componentPath, data } = event;
+                const { componentPath, data, content } = event;
                 if (event.type === "error") {
-                    console.error(`Failed to get component info for ${event.filepath}:`, event.error);
+                    // console.error(
+                    // 	`Failed to get component info for ${event.filepath}:`,
+                    // 	event.error,
+                    // );
                 }
                 for (const entry of entries) {
                     if (entry.componentPath === componentPath) {
@@ -204,11 +163,12 @@ class Storybook {
                             name: entry.name,
                             importPath: entry.importPath,
                             componentPath: entry.componentPath,
+                            component: content,
                             docgenInfo: data || [],
                         });
                     }
                 }
-                console.log(`${this.componentInformation.length} components loaded.`);
+                // console.error(`${this.componentInformation.length} components loaded.`);
                 workers.push(worker);
                 processNext();
             });

package/package.json

@@ -1,10 +1,10 @@
 {
   "name": "@miyucy/storybook-mcp",
-  "version": "1.0.5",
+  "version": "1.0.7",
   "description": "",
   "main": "dist/index.js",
   "scripts": {
-    "build": "rm -rf dist && tsc",
+    "build": "rm -rf dist && tsc && chmod +x dist/index.js",
     "start": "node dist/index.js"
   },
   "author": "miyucy",
@@ -19,8 +19,8 @@
   "type": "module",
   "dependencies": {
     "@modelcontextprotocol/sdk": "1.10.1",
-    "fast-glob": "^3.3.3",
-    "react-docgen": "^7.1.1",
+    "fast-glob": "3.3.3",
+    "react-docgen": "7.1.1",
     "zod": "3.24.3"
   },
   "devDependencies": {