jsc-typescript-ast-mcp 1.1.0 → 1.1.1

package/dist/ts-morph/utils/component.js

@@ -1,16 +1,52 @@
-import { SyntaxKind } from 'ts-morph';
-import { getComponentName } from './nameHelper.js';
+import { Node, SyntaxKind } from 'ts-morph';
+const isComponentVariableDeclaration = (node) => {
+    if (!Node.isVariableDeclaration(node)) {
+        return false;
+    }
+    const initializer = node.getInitializer();
+    if (!initializer) {
+        return false;
+    }
+    return (initializer.getKind() === SyntaxKind.ArrowFunction ||
+        initializer.getKind() === SyntaxKind.FunctionExpression);
+};
+const isComponentDeclaration = (node) => {
+    if (Node.isFunctionDeclaration(node)) {
+        return true;
+    }
+    return isComponentVariableDeclaration(node);
+};
+const getFirstExportedComponent = (sourceFile) => {
+    const exportedDeclarations = sourceFile.getExportedDeclarations();
+    for (const declarations of exportedDeclarations.values()) {
+        const componentDeclaration = declarations.find(isComponentDeclaration);
+        if (componentDeclaration) {
+            return componentDeclaration;
+        }
+    }
+    return undefined;
+};
 export const getComponent = (sourceFile) => {
-    const component = sourceFile.getDefaultExportSymbol()?.getDeclarations()?.[0];
-    if (!component) {
-        throw new Error(`Component not found in ${sourceFile.getFilePath()}`);
-    }
-    const componentName = getComponentName(component);
-    if (componentName !== 'AnonymousComponent')
-        return component;
-    const components = sourceFile.getVariableDeclarations().filter((v) => {
-        const initializer = v.getInitializer();
-        return initializer?.getKind() === SyntaxKind.ArrowFunction;
-    });
-    return components[0];
+    const defaultExportDeclaration = sourceFile
+        .getDefaultExportSymbol()
+        ?.getDeclarations()
+        ?.find(isComponentDeclaration);
+    if (defaultExportDeclaration) {
+        return defaultExportDeclaration;
+    }
+    const firstExportedComponent = getFirstExportedComponent(sourceFile);
+    if (firstExportedComponent) {
+        return firstExportedComponent;
+    }
+    const localVariableComponent = sourceFile
+        .getVariableDeclarations()
+        .find(isComponentVariableDeclaration);
+    if (localVariableComponent) {
+        return localVariableComponent;
+    }
+    const localFunctionComponent = sourceFile.getFunctions()[0];
+    if (localFunctionComponent) {
+        return localFunctionComponent;
+    }
+    throw new Error(`Component not found in ${sourceFile.getFilePath()}`);
 };

package/dist/ts-morph/utils/extractJsx.js

@@ -91,19 +91,64 @@ const resolveComponentFile = (node) => {
         const defaultImport = imp.getDefaultImport();
         if (defaultImport?.getText() === tagName) {
             const file = imp.getModuleSpecifierSourceFile();
-            const component = getComponent(file);
-            return component ?? null;
+            if (!file || file.getFilePath().includes('/node_modules/')) {
+                return null;
+            }
+            try {
+                const component = getComponent(file);
+                return component ?? null;
+            }
+            catch {
+                return null;
+            }
         }
         for (const n of named) {
             if (n.getName() === tagName) {
                 const file = imp.getModuleSpecifierSourceFile();
-                const component = getComponent(file);
-                return component ?? null;
+                if (!file || file.getFilePath().includes('/node_modules/')) {
+                    return null;
+                }
+                try {
+                    const component = getComponent(file);
+                    return component ?? null;
+                }
+                catch {
+                    return null;
+                }
             }
         }
     }
     return null;
 };
+const resolveLocalComponentFilePath = (node) => {
+    const tagName = getTagName(node);
+    const sourceFile = node.getSourceFile();
+    for (const imp of sourceFile.getImportDeclarations()) {
+        const matchesDefaultImport = imp.getDefaultImport()?.getText() === tagName;
+        const matchesNamedImport = imp
+            .getNamedImports()
+            .some((namedImport) => namedImport.getName() === tagName);
+        if (!matchesDefaultImport && !matchesNamedImport) {
+            continue;
+        }
+        const moduleSpecifier = imp.getModuleSpecifierValue();
+        const isLocalImport = moduleSpecifier.startsWith('.') || moduleSpecifier.startsWith('/');
+        if (!isLocalImport) {
+            return undefined;
+        }
+        const importedSourceFile = imp.getModuleSpecifierSourceFile();
+        if (!importedSourceFile) {
+            return undefined;
+        }
+        const importedFilePath = importedSourceFile.getFilePath();
+        if (importedFilePath.includes('/node_modules/')) {
+            return undefined;
+        }
+        return importedFilePath;
+    }
+    // If there is no matching import, treat it as a component from the current local source file.
+    return sourceFile.getFilePath();
+};
 const buildNodeFromJSX = (node, project, options, depth) => {
     const tagName = getTagName(node);
     const isHtml = tagName[0] === tagName[0].toLowerCase();
@@ -113,7 +158,10 @@ const buildNodeFromJSX = (node, project, options, depth) => {
         children: [],
     };
     if (!isHtml) {
-        treeNode.filePath = node.getSourceFile().getFilePath();
+        const localComponentFilePath = resolveLocalComponentFilePath(node);
+        if (localComponentFilePath) {
+            treeNode.filePath = localComponentFilePath;
+        }
     }
     const props = extractPropsFromNode(node);
     if (props) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jsc-typescript-ast-mcp",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "mcpName": "io.github.jscoobyced/jsc-typescript-ast-mcp",
   "description": "A Model Context Protocol (MCP) server that provides an abstract syntax tree (AST) representation of TypeScript code using the ts-morph library. It allows clients to analyze and manipulate TypeScript code structures, making it easier for AI models to understand and work with TypeScript projects. You can create a JSON representation of your React components, and use it for various purposes such as documentation, code analysis, or even generating new code based on existing components.",
   "main": "dist/index.js",