code-graph-context 2.4.1 → 2.4.2

package/dist/cli/cli.js

@@ -246,7 +246,7 @@ program
     .option('-p, --port <port>', 'Neo4j Bolt port', '7687')
     .option('--http-port <port>', 'Neo4j Browser port', '7474')
     .option('--password <password>', 'Neo4j password', 'PASSWORD')
-    .option('-m, --memory <size>', 'Max heap memory (e.g., 2G, 4G)', '2G')
+    .option('-m, --memory <size>', 'Max heap memory (e.g., 2G, 4G)', '4G')
     .option('-f, --force', 'Recreate container even if exists')
     .action(runInit);
 program.command('status').description('Check Neo4j and Docker status').action(runStatus);

package/dist/core/config/nestjs-framework-schema.js

@@ -434,7 +434,8 @@ export const NESTJS_FRAMEWORK_SCHEMA = {
                 },
                 {
                     type: 'function',
-                    pattern: (parsedNode) => parsedNode.sourceNode?.getName()?.endsWith('Service'),
+                    // Use pre-extracted name property (works after AST cleanup in streaming/chunking)
+                    pattern: (parsedNode) => parsedNode.properties?.name?.endsWith('Service'),
                     confidence: 0.7,
                     priority: 7,
                 },
@@ -516,13 +517,11 @@ export const NESTJS_FRAMEWORK_SCHEMA = {
             detectionPatterns: [
                 {
                     type: 'function',
+                    // Use pre-extracted decoratorNames from context (works after AST cleanup in streaming/chunking)
                     pattern: (parsedNode) => {
-                        const node = parsedNode.sourceNode;
-                        if (!node)
-                            return false;
-                        const decorators = node.getDecorators?.() ?? [];
+                        const decoratorNames = parsedNode.properties?.context?.decoratorNames ?? [];
                         const messageDecorators = ['MessagePattern', 'EventPattern'];
-                        return decorators.some((d) => messageDecorators.includes(d.getName()));
+                        return messageDecorators.some((d) => decoratorNames.includes(d));
                     },
                     confidence: 0.98,
                     priority: 15,
@@ -570,13 +569,11 @@ export const NESTJS_FRAMEWORK_SCHEMA = {
             detectionPatterns: [
                 {
                     type: 'function',
+                    // Use pre-extracted decoratorNames from context (works after AST cleanup in streaming/chunking)
                     pattern: (parsedNode) => {
-                        const node = parsedNode.sourceNode;
-                        if (!node)
-                            return false;
-                        const decorators = node.getDecorators?.() ?? [];
+                        const decoratorNames = parsedNode.properties?.context?.decoratorNames ?? [];
                         const httpDecorators = ['Get', 'Post', 'Put', 'Delete', 'Patch', 'Head', 'Options'];
-                        return decorators.some((d) => httpDecorators.includes(d.getName()));
+                        return httpDecorators.some((d) => decoratorNames.includes(d));
                     },
                     confidence: 0.98,
                     priority: 15,
@@ -729,16 +726,11 @@ export const NESTJS_FRAMEWORK_SCHEMA = {
                 if (parsedSourceNode.properties?.filePath !== parsedTargetNode.properties?.filePath) {
                     return false;
                 }
-                // Access AST nodes to check parent relationship
-                const sourceNode = parsedSourceNode.sourceNode;
-                const targetNode = parsedTargetNode.sourceNode;
-                if (sourceNode && targetNode) {
-                    const methodParent = targetNode.getParent();
-                    if (methodParent === sourceNode) {
-                        return true;
-                    }
-                }
-                return false;
+                // CRITICAL FIX: Use pre-extracted parentClassName property (works after AST cleanup in streaming/chunking)
+                // The method's parentClassName should match the controller's name
+                const targetParentClassName = parsedTargetNode.properties?.parentClassName;
+                const sourceName = parsedSourceNode.properties?.name;
+                return !!targetParentClassName && targetParentClassName === sourceName;
             },
             contextExtractor: (parsedSourceNode, parsedTargetNode) => ({
                 endpointType: 'RPC',
@@ -761,16 +753,11 @@ export const NESTJS_FRAMEWORK_SCHEMA = {
                 if (parsedSourceNode.properties?.filePath !== parsedTargetNode.properties?.filePath) {
                     return false;
                 }
-                // Access AST nodes to check parent relationship
-                const sourceNode = parsedSourceNode.sourceNode;
-                const targetNode = parsedTargetNode.sourceNode;
-                if (sourceNode && targetNode) {
-                    const methodParent = targetNode.getParent();
-                    if (methodParent === sourceNode) {
-                        return true;
-                    }
-                }
-                return false;
+                // CRITICAL FIX: Use pre-extracted parentClassName property (works after AST cleanup in streaming/chunking)
+                // The method's parentClassName should match the controller's name
+                const targetParentClassName = parsedTargetNode.properties?.parentClassName;
+                const sourceName = parsedSourceNode.properties?.name;
+                return !!targetParentClassName && targetParentClassName === sourceName;
             },
             contextExtractor: (parsedSourceNode, parsedTargetNode) => ({
                 httpMethod: parsedTargetNode.properties?.context?.httpMethod ?? '',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-graph-context",
-  "version": "2.4.1",
+  "version": "2.4.2",
   "description": "MCP server that builds code graphs to provide rich context to LLMs",
   "type": "module",
   "homepage": "https://github.com/drewdrewH/code-graph-context#readme",