untitledui-mcp 0.1.3 → 0.1.4

package/dist/index.js

@@ -378,6 +378,28 @@ function getBaseComponentNames(files) {
   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);
@@ -455,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: {
@@ -467,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: {
@@ -477,6 +499,19 @@ 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: "Browse available page examples. Call without path to see categories, then drill down.",
@@ -575,7 +610,20 @@ function createServer(licenseKey) {
             };
             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;
@@ -602,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
@@ -617,15 +672,77 @@ 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 "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({
+                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}`;
@@ -675,17 +792,25 @@ function createServer(licenseKey) {
             };
           }
           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: example.content.files,
+                  files,
+                  fileList: getFileTokenList(files),
                   dependencies: example.content.dependencies || [],
                   devDependencies: example.content.devDependencies || [],
                   components: example.content.components || [],
-                  fileCount: example.content.files?.length || 0
+                  fileCount: files.length,
+                  estimatedTokens,
+                  ...tooLarge && {
+                    warning: `Response is large (${estimatedTokens} tokens). Consider fetching specific component files instead.`
+                  }
                 }, null, 2)
               }]
             };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "untitledui-mcp",
-  "version": "0.1.3",
+  "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",