untitledui-mcp 0.1.1 → 0.1.4

package/README.md

@@ -74,7 +74,8 @@ Your AI gets these tools:
 | `list_components` | Browse a category |
 | `get_component_with_deps` | Fetch component + all dependencies |
 | `get_component` | Fetch component only |
-| `get_example` | Get complete page template |
+| `list_examples` | Browse available page templates |
+| `get_example` | Fetch a specific page template |
 
 ### Example: Add a modal
 
@@ -95,13 +96,22 @@ AI calls list_components { type: "application", subfolder: "sidebars" }
 → Returns all sidebar options for you to choose from
 ```
 
-### Example: Start from template
+### Example: Start from a page template
 
 ```
-You: "Set up a dashboard layout"
+You: "Show me available dashboard templates"
 
-AI calls get_example { name: "application" }
-→ Returns complete dashboard with sidebar, header, and sample pages
+AI calls list_examples { path: "" }
+→ Returns: application, marketing
+
+AI calls list_examples { path: "application" }
+→ Returns: dashboards-01, dashboards-02, settings-01, ...
+
+AI calls list_examples { path: "application/dashboards-01" }
+→ Returns: 01, 02, 03, ... (individual pages)
+
+AI calls get_example { path: "application/dashboards-01/01" }
+→ Returns complete page with 27 files, all dependencies
 ```
 
 ## Response Format

package/dist/index.js

@@ -126,17 +126,60 @@ var UntitledUIClient = class {
     }
     return data.components;
   }
-  async fetchExample(name) {
+  /**
+   * Browse examples hierarchy or fetch example content.
+   * - "" → lists types (application, marketing)
+   * - "application" → lists examples (dashboards-01, settings-01, etc.)
+   * - "application/dashboards-01" → lists pages (01, 02, 03, ...)
+   * - "application/dashboards-01/01" → returns actual content
+   */
+  async fetchExample(path2) {
     const response = await fetch(ENDPOINTS.fetchExample, {
       method: "POST",
       headers: { "Content-Type": "application/json" },
       body: JSON.stringify({
-        example: name,
+        example: path2,
         key: this.licenseKey
       })
     });
     return response.json();
   }
+  /**
+   * Fetch all pages in an example and combine them.
+   * Path should be "application/dashboards-01" (without page number).
+   */
+  async fetchFullExample(examplePath) {
+    const listing = await this.fetchExample(examplePath);
+    if (listing.type !== "directory" || !listing.results) {
+      throw new Error(`Invalid example path: ${examplePath}. Expected a directory with pages.`);
+    }
+    const pages = listing.results;
+    const allDeps = /* @__PURE__ */ new Set();
+    const allDevDeps = /* @__PURE__ */ new Set();
+    const fetchedPages = [];
+    for (const page of pages) {
+      const pageData = await this.fetchExample(`${examplePath}/${page}`);
+      if (pageData.type === "json-file" && pageData.content) {
+        const content = pageData.content;
+        fetchedPages.push({
+          page,
+          files: content.files || [],
+          dependencies: content.dependencies || [],
+          devDependencies: content.devDependencies || []
+        });
+        content.dependencies?.forEach((d) => allDeps.add(d));
+        content.devDependencies?.forEach((d) => allDevDeps.add(d));
+      }
+    }
+    const totalFiles = fetchedPages.reduce((sum, p) => sum + p.files.length, 0);
+    return {
+      name: examplePath,
+      pages: fetchedPages,
+      allDependencies: Array.from(allDeps),
+      allDevDependencies: Array.from(allDevDeps),
+      totalFiles
+    };
+  }
 };
 
 // src/cache/memory-cache.ts
@@ -289,6 +332,74 @@ function generateDescription(name, type) {
   return `UI component: ${name.replace(/-/g, " ")}`;
 }
 
+// src/utils/parse-deps.ts
+function parseComponentImports(files) {
+  const imports = /* @__PURE__ */ new Set();
+  const regex = /@\/components\/(base|application|foundations|shared-assets)\/([^"']+)/g;
+  for (const file of files) {
+    const code = file.content || file.code || "";
+    let match;
+    while ((match = regex.exec(code)) !== null) {
+      imports.add(match[0]);
+    }
+  }
+  const parsed = [];
+  for (const imp of imports) {
+    const match = imp.match(/@\/components\/(base|application|foundations|shared-assets)\/(.+)/);
+    if (match) {
+      parsed.push({
+        type: match[1],
+        path: match[2],
+        fullImport: imp
+      });
+    }
+  }
+  return parsed;
+}
+function extractBaseComponentPaths(files) {
+  const deps = parseComponentImports(files);
+  const basePaths = /* @__PURE__ */ new Set();
+  for (const dep of deps) {
+    if (dep.type === "base") {
+      basePaths.add(dep.path);
+    }
+  }
+  return Array.from(basePaths);
+}
+function getBaseComponentNames(files) {
+  const paths = extractBaseComponentPaths(files);
+  const names = /* @__PURE__ */ new Set();
+  for (const path2 of paths) {
+    const parts = path2.split("/");
+    if (parts.length >= 1) {
+      names.add(parts[0]);
+    }
+  }
+  return Array.from(names);
+}
+
+// src/utils/tokens.ts
+function estimateTokens(content) {
+  return Math.ceil(content.length / 4);
+}
+function estimateFileTokens(file) {
+  const content = file.code || file.content || "";
+  return estimateTokens(content);
+}
+function estimateComponentTokens(files) {
+  return files.reduce((sum, file) => sum + estimateFileTokens(file), 0);
+}
+function getFileTokenList(files) {
+  return files.map((file) => ({
+    path: file.path,
+    tokens: estimateFileTokens(file)
+  }));
+}
+var CLAUDE_READ_TOKEN_LIMIT = 25e3;
+function isLikelyTooLarge(estimatedTokens) {
+  return estimatedTokens > CLAUDE_READ_TOKEN_LIMIT;
+}
+
 // src/server.ts
 function createServer(licenseKey) {
   const client = new UntitledUIClient(licenseKey);
@@ -366,7 +477,7 @@ function createServer(licenseKey) {
       },
       {
         name: "get_component",
-        description: "Get a single component's code. Does NOT include dependencies - use get_component_with_deps for that.",
+        description: "Get a single component's code with token estimates. Returns estimatedTokens and file list. If estimatedTokens > 25000, consider using get_component_file for specific files instead.",
         inputSchema: {
           type: "object",
           properties: {
@@ -378,7 +489,7 @@ function createServer(licenseKey) {
       },
       {
         name: "get_component_with_deps",
-        description: "Get a component with all its base component dependencies included",
+        description: "Get a component with all base dependencies. Returns estimatedTokens. If response is too large (>25000 tokens), use get_component_file to fetch specific files individually.",
         inputSchema: {
           type: "object",
           properties: {
@@ -388,20 +499,38 @@ function createServer(licenseKey) {
           required: ["type", "name"]
         }
       },
+      {
+        name: "get_component_file",
+        description: "Get a single file from a component. Use this when get_component or get_component_with_deps returns a large response (>25000 tokens). First call get_component to see the file list, then fetch specific files as needed.",
+        inputSchema: {
+          type: "object",
+          properties: {
+            type: { type: "string", description: "Component type" },
+            name: { type: "string", description: "Component name" },
+            file: { type: "string", description: "File path within the component (e.g., 'Button.tsx' or 'variants/InputPhone.tsx')" }
+          },
+          required: ["type", "name", "file"]
+        }
+      },
       {
         name: "list_examples",
-        description: "List available page examples (dashboards, marketing pages, etc.)",
-        inputSchema: { type: "object", properties: {} }
+        description: "Browse available page examples. Call without path to see categories, then drill down.",
+        inputSchema: {
+          type: "object",
+          properties: {
+            path: { type: "string", description: "Path to browse (e.g., '', 'application', 'application/dashboards-01')" }
+          }
+        }
       },
       {
         name: "get_example",
-        description: "Get a complete page example with all files",
+        description: "Get a single example page with all files. Requires full path including page number.",
         inputSchema: {
           type: "object",
           properties: {
-            name: { type: "string", description: "Example name (e.g., 'application', 'marketing')" }
+            path: { type: "string", description: "Full path to example page (e.g., 'application/dashboards-01/01')" }
           },
-          required: ["name"]
+          required: ["path"]
         }
       },
       {
@@ -477,11 +606,24 @@ function createServer(licenseKey) {
               files: fetched.files,
               dependencies: fetched.dependencies || [],
               devDependencies: fetched.devDependencies || [],
-              baseComponents: (fetched.components || []).map((c) => c.name)
+              baseComponents: getBaseComponentNames(fetched.files)
             };
             cache.set(cacheKey, component, CACHE_TTL.componentCode);
           }
-          return { content: [{ type: "text", text: JSON.stringify(component, null, 2) }] };
+          const estimatedTokens = estimateComponentTokens(component.files);
+          const fileTokens = getFileTokenList(component.files);
+          const tooLarge = isLikelyTooLarge(estimatedTokens);
+          const result = {
+            ...component,
+            estimatedTokens,
+            fileCount: component.files.length,
+            fileList: fileTokens,
+            ...tooLarge && {
+              warning: `Response is large (${estimatedTokens} tokens). Consider using get_component_file for specific files.`,
+              hint: "Use get_component_file with type, name, and file path to fetch individual files."
+            }
+          };
+          return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
         }
         case "get_component_with_deps": {
           const { type, name: componentName } = args;
@@ -500,7 +642,7 @@ function createServer(licenseKey) {
               }]
             };
           }
-          const baseComponentNames = (primary.components || []).map((c) => c.name);
+          const baseComponentNames = getBaseComponentNames(primary.files);
           const baseComponents = baseComponentNames.length > 0 ? await client.fetchComponents("base", baseComponentNames) : [];
           const allDeps = /* @__PURE__ */ new Set();
           const allDevDeps = /* @__PURE__ */ new Set();
@@ -508,12 +650,19 @@ function createServer(licenseKey) {
             c.dependencies?.forEach((d) => allDeps.add(d));
             c.devDependencies?.forEach((d) => allDevDeps.add(d));
           });
+          const allFiles = [
+            ...primary.files,
+            ...baseComponents.flatMap((c) => c.files)
+          ];
+          const estimatedTokens = estimateComponentTokens(allFiles);
+          const tooLarge = isLikelyTooLarge(estimatedTokens);
           const result = {
             primary: {
               name: primary.name,
               type,
               description: generateDescription(primary.name, type),
               files: primary.files,
+              fileList: getFileTokenList(primary.files),
               dependencies: primary.dependencies || [],
               devDependencies: primary.devDependencies || [],
               baseComponents: baseComponentNames
@@ -523,38 +672,149 @@ function createServer(licenseKey) {
               type: "base",
               description: generateDescription(c.name, "base"),
               files: c.files,
+              fileList: getFileTokenList(c.files),
               dependencies: c.dependencies || [],
               devDependencies: c.devDependencies || []
             })),
             totalFiles: primary.files.length + baseComponents.reduce((sum, c) => sum + c.files.length, 0),
+            estimatedTokens,
             allDependencies: Array.from(allDeps),
-            allDevDependencies: Array.from(allDevDeps)
+            allDevDependencies: Array.from(allDevDeps),
+            ...tooLarge && {
+              warning: `Response is large (${estimatedTokens} tokens, limit is ${CLAUDE_READ_TOKEN_LIMIT}). Consider using get_component_file for specific files.`,
+              hint: "To fetch individual files, use get_component_file with the component type, name, and file path from fileList."
+            }
           };
           return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
         }
-        case "list_examples": {
+        case "get_component_file": {
+          const { type, name: componentName, file: filePath } = args;
+          const cacheKey = `component:${type}:${componentName}`;
+          let files;
+          const cached = cache.get(cacheKey);
+          if (cached) {
+            files = cached.files;
+          } else {
+            const fetched = await client.fetchComponent(type, componentName);
+            if (!fetched) {
+              const index = await buildSearchIndex();
+              const suggestions = fuzzySearch(componentName, index, 5).map((r) => r.fullPath);
+              return {
+                content: [{
+                  type: "text",
+                  text: JSON.stringify({
+                    error: `Component "${componentName}" not found`,
+                    code: "NOT_FOUND",
+                    suggestions
+                  }, null, 2)
+                }]
+              };
+            }
+            files = fetched.files;
+          }
+          const file = files.find(
+            (f) => f.path === filePath || f.path.endsWith(`/${filePath}`) || f.path.endsWith(filePath)
+          );
+          if (!file) {
+            const availableFiles = files.map((f) => f.path);
+            return {
+              content: [{
+                type: "text",
+                text: JSON.stringify({
+                  error: `File "${filePath}" not found in ${type}/${componentName}`,
+                  code: "FILE_NOT_FOUND",
+                  availableFiles,
+                  hint: "Use one of the file paths from availableFiles"
+                }, null, 2)
+              }]
+            };
+          }
           return {
             content: [{
               type: "text",
               text: JSON.stringify({
-                examples: [
-                  { name: "application", type: "application", description: "Dashboard application example" },
-                  { name: "marketing", type: "marketing", description: "Marketing landing page example" }
-                ]
+                component: `${type}/${componentName}`,
+                file: {
+                  path: file.path,
+                  code: file.code
+                },
+                estimatedTokens: estimateComponentTokens([file])
               }, null, 2)
             }]
           };
         }
+        case "list_examples": {
+          const { path: path2 = "" } = args;
+          const cacheKey = `examples:list:${path2}`;
+          let listing = cache.get(cacheKey);
+          if (!listing) {
+            listing = await client.fetchExample(path2);
+            if (listing) {
+              cache.set(cacheKey, listing, CACHE_TTL.componentList);
+            }
+          }
+          if (listing.type === "directory" && listing.results) {
+            return {
+              content: [{
+                type: "text",
+                text: JSON.stringify({
+                  path: path2 || "(root)",
+                  type: "directory",
+                  items: listing.results,
+                  hint: path2 === "" ? "Use list_examples with path='application' or path='marketing' to see available examples" : path2.split("/").length === 1 ? `Use list_examples with path='${path2}/<example>' to see pages` : `Use get_example with path='${path2}/<page>' to fetch a specific page`
+                }, null, 2)
+              }]
+            };
+          }
+          return { content: [{ type: "text", text: JSON.stringify(listing, null, 2) }] };
+        }
         case "get_example": {
-          const { name: exampleName } = args;
-          const cacheKey = `example:${exampleName}`;
+          const { path: examplePath } = args;
+          const cacheKey = `example:${examplePath}`;
           let example = cache.get(cacheKey);
           if (!example) {
-            example = await client.fetchExample(exampleName);
+            example = await client.fetchExample(examplePath);
             if (example) {
               cache.set(cacheKey, example, CACHE_TTL.examples);
             }
           }
+          if (example.type === "directory") {
+            return {
+              content: [{
+                type: "text",
+                text: JSON.stringify({
+                  error: "Path is a directory, not a page",
+                  path: examplePath,
+                  availableItems: example.results,
+                  hint: `Use get_example with path='${examplePath}/<item>' to fetch a specific page`
+                }, null, 2)
+              }]
+            };
+          }
+          if (example.type === "json-file" && example.content) {
+            const files = example.content.files || [];
+            const estimatedTokens = estimateComponentTokens(files);
+            const tooLarge = isLikelyTooLarge(estimatedTokens);
+            return {
+              content: [{
+                type: "text",
+                text: JSON.stringify({
+                  path: examplePath,
+                  name: example.content.name,
+                  files,
+                  fileList: getFileTokenList(files),
+                  dependencies: example.content.dependencies || [],
+                  devDependencies: example.content.devDependencies || [],
+                  components: example.content.components || [],
+                  fileCount: files.length,
+                  estimatedTokens,
+                  ...tooLarge && {
+                    warning: `Response is large (${estimatedTokens} tokens). Consider fetching specific component files instead.`
+                  }
+                }, null, 2)
+              }]
+            };
+          }
           return { content: [{ type: "text", text: JSON.stringify(example, null, 2) }] };
         }
         case "validate_license": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "untitledui-mcp",
-  "version": "0.1.1",
+  "version": "0.1.4",
   "description": "MCP server for UntitledUI Pro components - browse, search, and retrieve UI components via Claude Code",
   "type": "module",
   "main": "dist/index.js",