code-graph-context 1.0.0 → 2.0.0

package/dist/core/parsers/typescript-parser.js

@@ -2,25 +2,31 @@
 import crypto from 'crypto';
 import fs from 'fs/promises';
 import path from 'node:path';
+import { glob } from 'glob';
 import { minimatch } from 'minimatch';
 import { Project, Node } from 'ts-morph';
+import { createFrameworkEdgeData } from '../utils/edge-factory.js';
 /**
  * Generate a deterministic node ID based on stable properties.
  * This ensures the same node gets the same ID across reparses.
  *
- * Identity is based on: coreType + filePath + name (+ parentId for nested nodes)
+ * Identity is based on: projectId + coreType + filePath + name (+ parentId for nested nodes)
  * This is stable because when it matters (one side of edge not reparsed),
  * names are guaranteed unchanged (or imports would break, triggering reparse).
+ *
+ * Including projectId ensures nodes from different projects have unique IDs
+ * even if they have identical file paths and names.
  */
-const generateDeterministicId = (coreType, filePath, name, parentId) => {
-    const parts = parentId ? [coreType, filePath, parentId, name] : [coreType, filePath, name];
+const generateDeterministicId = (projectId, coreType, filePath, name, parentId) => {
+    const parts = parentId ? [projectId, coreType, filePath, parentId, name] : [projectId, coreType, filePath, name];
     const identity = parts.join('::');
     const hash = crypto.createHash('sha256').update(identity).digest('hex').substring(0, 16);
-    return `${coreType}:${hash}`;
+    return `${projectId}:${coreType}:${hash}`;
 };
-import { hashFile } from '../../utils/file-utils.js';
+import { debugLog, hashFile } from '../utils/file-utils.js';
 import { NESTJS_FRAMEWORK_SCHEMA } from '../config/nestjs-framework-schema.js';
 import { CoreNodeType, CORE_TYPESCRIPT_SCHEMA, DEFAULT_PARSE_OPTIONS, CoreEdgeType, } from '../config/schema.js';
+import { resolveProjectId } from '../utils/project-id.js';
 export class TypeScriptParser {
     workspacePath;
     tsConfigPath;
@@ -33,24 +39,63 @@ export class TypeScriptParser {
     existingNodes = new Map(); // Nodes from Neo4j for edge target matching
     deferredEdges = [];
     sharedContext = new Map(); // Shared context for custom data
-    constructor(workspacePath, tsConfigPath = 'tsconfig.json', coreSchema = CORE_TYPESCRIPT_SCHEMA, frameworkSchemas = [NESTJS_FRAMEWORK_SCHEMA], parseConfig = DEFAULT_PARSE_OPTIONS) {
+    projectId; // Project identifier for multi-project isolation
+    lazyLoad; // Whether to use lazy file loading for large projects
+    discoveredFiles = null; // Cached file discovery results
+    deferEdgeEnhancements = false; // When true, skip edge enhancements (parent will handle)
+    constructor(workspacePath, tsConfigPath = 'tsconfig.json', coreSchema = CORE_TYPESCRIPT_SCHEMA, frameworkSchemas = [NESTJS_FRAMEWORK_SCHEMA], parseConfig = DEFAULT_PARSE_OPTIONS, projectId, // Optional - derived from workspacePath if not provided
+    lazyLoad = false) {
         this.workspacePath = workspacePath;
         this.tsConfigPath = tsConfigPath;
         this.coreSchema = coreSchema;
         this.frameworkSchemas = frameworkSchemas;
         this.parseConfig = parseConfig;
-        this.project = new Project({
-            tsConfigFilePath: tsConfigPath,
-            skipAddingFilesFromTsConfig: false,
-            compilerOptions: {
-                experimentalDecorators: true,
-                emitDecoratorMetadata: true,
-                target: 7,
-                module: 1,
-                esModuleInterop: true,
-            },
-        });
-        this.project.addSourceFilesAtPaths(path.join(workspacePath, '**/*.ts'));
+        this.projectId = resolveProjectId(workspacePath, projectId);
+        this.lazyLoad = lazyLoad;
+        console.log(`🆔 Project ID: ${this.projectId}`);
+        console.log(`📂 Lazy loading: ${lazyLoad ? 'enabled' : 'disabled'}`);
+        if (lazyLoad) {
+            // Lazy mode: create Project without loading any files
+            // Files will be added just-in-time during parseChunk()
+            this.project = new Project({
+                tsConfigFilePath: tsConfigPath,
+                skipAddingFilesFromTsConfig: true, // Don't load files from tsconfig
+                skipFileDependencyResolution: true, // Don't load node_modules types
+                compilerOptions: {
+                    experimentalDecorators: true,
+                    emitDecoratorMetadata: true,
+                    target: 7,
+                    module: 1,
+                    esModuleInterop: true,
+                    skipLibCheck: true,
+                },
+            });
+        }
+        else {
+            // Eager mode: load all files upfront (original behavior for small projects)
+            this.project = new Project({
+                tsConfigFilePath: tsConfigPath,
+                skipAddingFilesFromTsConfig: false,
+                skipFileDependencyResolution: true,
+                compilerOptions: {
+                    experimentalDecorators: true,
+                    emitDecoratorMetadata: true,
+                    target: 7,
+                    module: 1,
+                    esModuleInterop: true,
+                    skipLibCheck: true,
+                },
+            });
+            // Include both .ts and .tsx files
+            this.project.addSourceFilesAtPaths(path.join(workspacePath, '**/*.{ts,tsx}'));
+        }
+    }
+    /**
+     * Get the projectId for this parser instance.
+     * This is used by tools to pass projectId to Neo4j queries.
+     */
+    getProjectId() {
+        return this.projectId;
     }
     /**
      * Set existing nodes from Neo4j for edge target matching during incremental parsing.
@@ -67,6 +112,7 @@ export class TypeScriptParser {
                 labels: node.labels,
                 properties: {
                     id: node.id,
+                    projectId: this.projectId,
                     name: node.name,
                     coreType: node.coreType,
                     filePath: node.filePath,
@@ -81,8 +127,20 @@ export class TypeScriptParser {
     async parseWorkspace(filesToParse) {
         let sourceFiles;
         if (filesToParse && filesToParse.length > 0) {
+            // In lazy mode, files may not be loaded yet - add them if needed
             sourceFiles = filesToParse
-                .map((filePath) => this.project.getSourceFile(filePath))
+                .map((filePath) => {
+                const existing = this.project.getSourceFile(filePath);
+                if (existing)
+                    return existing;
+                // Add file to project if not already loaded (lazy mode)
+                try {
+                    return this.project.addSourceFileAtPath(filePath);
+                }
+                catch {
+                    return undefined;
+                }
+            })
                 .filter((sf) => sf !== undefined);
         }
         else {
@@ -93,7 +151,7 @@ export class TypeScriptParser {
                 continue;
             await this.parseCoreTypeScriptV2(sourceFile);
         }
-        this.resolveDeferredEdges();
+        await this.resolveDeferredEdges();
         await this.applyContextExtractors();
         if (this.frameworkSchemas.length > 0) {
             await this.applyFrameworkEnhancements();
@@ -124,13 +182,45 @@ export class TypeScriptParser {
         const filePath = sourceFile.getFilePath();
         const stats = await fs.stat(filePath);
         const fileTrackingProperties = {
-            size: stats.size,
-            mtime: stats.mtimeMs,
+            size: Number(stats.size),
+            mtime: Number(stats.mtimeMs),
             contentHash: await hashFile(filePath),
         };
         const sourceFileNode = this.createCoreNode(sourceFile, CoreNodeType.SOURCE_FILE, fileTrackingProperties);
         this.addNode(sourceFileNode);
         await this.parseChildNodes(this.coreSchema.nodeTypes[CoreNodeType.SOURCE_FILE], sourceFileNode, sourceFile);
+        // Queue IMPORTS edges for deferred resolution
+        // Note: ImportDeclaration nodes are already created by parseChildNodes via the schema
+        // This adds SourceFile → SourceFile IMPORTS edges for cross-file dependency tracking
+        for (const importDecl of sourceFile.getImportDeclarations()) {
+            const moduleSpecifier = importDecl.getModuleSpecifierValue();
+            // Skip external modules (node_modules) - only process relative and scoped imports
+            if (!moduleSpecifier.startsWith('.') && !moduleSpecifier.startsWith('@')) {
+                continue;
+            }
+            // Use ts-morph's module resolution to get the actual file path
+            // This correctly resolves relative imports like './auth.controller' to absolute paths
+            try {
+                const targetSourceFile = importDecl.getModuleSpecifierSourceFile();
+                if (targetSourceFile) {
+                    this.deferredEdges.push({
+                        edgeType: CoreEdgeType.IMPORTS,
+                        sourceNodeId: sourceFileNode.id,
+                        targetName: targetSourceFile.getFilePath(), // Store resolved absolute path
+                        targetType: CoreNodeType.SOURCE_FILE,
+                    });
+                }
+            }
+            catch {
+                // If resolution fails, fall back to raw module specifier
+                this.deferredEdges.push({
+                    edgeType: CoreEdgeType.IMPORTS,
+                    sourceNodeId: sourceFileNode.id,
+                    targetName: moduleSpecifier,
+                    targetType: CoreNodeType.SOURCE_FILE,
+                });
+            }
+        }
         if (this.shouldParseVariables(sourceFile.getFilePath())) {
             for (const varStatement of sourceFile.getVariableStatements()) {
                 for (const varDecl of varStatement.getDeclarations()) {
@@ -171,6 +261,14 @@ export class TypeScriptParser {
                 this.addNode(coreNode);
                 const coreEdge = this.createCoreEdge(edgeType, parentNode.id, coreNode.id);
                 this.addEdge(coreEdge);
+                const SKELETONIZE_TYPES = new Set([
+                    CoreNodeType.METHOD_DECLARATION,
+                    CoreNodeType.FUNCTION_DECLARATION,
+                    CoreNodeType.PROPERTY_DECLARATION,
+                ]);
+                if (SKELETONIZE_TYPES.has(type)) {
+                    this.skeletonizeChildInParent(parentNode, coreNode);
+                }
                 const childNodeConfig = this.coreSchema.nodeTypes[type];
                 if (childNodeConfig) {
                     this.queueRelationshipNodes(childNodeConfig, coreNode, child);
@@ -179,6 +277,15 @@ export class TypeScriptParser {
             }
         }
     }
+    skeletonizeChildInParent(parent, child) {
+        const childText = child.properties.sourceCode;
+        const bodyStart = childText.indexOf('{');
+        if (bodyStart > -1) {
+            const signature = childText.substring(0, bodyStart).trim();
+            const placeholder = `${signature} { /* NodeID: ${child.id} */ }`;
+            parent.properties.sourceCode = parent.properties.sourceCode.replace(childText, placeholder);
+        }
+    }
     /**
      * Queue relationship edges for deferred processing
      * These are resolved after all nodes are parsed since the target may not exist yet
@@ -201,15 +308,50 @@ export class TypeScriptParser {
                 const targetName = this.extractRelationshipTargetName(target);
                 if (!targetName)
                     continue;
+                // For EXTENDS/IMPLEMENTS, try to get the file path from the resolved declaration
+                let targetFilePath;
+                if (edgeType === CoreEdgeType.EXTENDS || edgeType === CoreEdgeType.IMPLEMENTS) {
+                    targetFilePath = this.extractTargetFilePath(target);
+                }
                 this.deferredEdges.push({
                     edgeType: edgeType,
                     sourceNodeId: parsedNode.id,
                     targetName,
                     targetType: targetNodeType,
+                    targetFilePath,
                 });
             }
         }
     }
+    /**
+     * Extract the file path from a resolved target declaration.
+     * Used for EXTENDS/IMPLEMENTS to enable precise matching.
+     */
+    extractTargetFilePath(target) {
+        try {
+            // If target is already a ClassDeclaration or InterfaceDeclaration, get its source file
+            if (Node.isClassDeclaration(target) || Node.isInterfaceDeclaration(target)) {
+                return target.getSourceFile().getFilePath();
+            }
+            // If target is ExpressionWithTypeArguments (e.g., extends Foo<T>), resolve the type
+            if (Node.isExpressionWithTypeArguments(target)) {
+                const expression = target.getExpression();
+                if (Node.isIdentifier(expression)) {
+                    // Try to get the definition of the type
+                    const definitions = expression.getDefinitionNodes();
+                    for (const def of definitions) {
+                        if (Node.isClassDeclaration(def) || Node.isInterfaceDeclaration(def)) {
+                            return def.getSourceFile().getFilePath();
+                        }
+                    }
+                }
+            }
+        }
+        catch {
+            // If resolution fails (e.g., external type), return undefined
+        }
+        return undefined;
+    }
     /**
      * Extract the target name from an AST node returned by relationship methods
      */
@@ -218,15 +360,103 @@ export class TypeScriptParser {
             return target.getName();
         if (Node.isInterfaceDeclaration(target))
             return target.getName();
-        if (Node.isExpressionWithTypeArguments(target))
-            return target.getExpression().getText();
+        if (Node.isExpressionWithTypeArguments(target)) {
+            const expression = target.getExpression();
+            const text = expression.getText();
+            const genericIndex = text.indexOf('<');
+            return genericIndex > 0 ? text.substring(0, genericIndex) : text;
+        }
         return undefined;
     }
     /**
      * Find a parsed node by name and core type
+     * For SourceFiles, implements smart import resolution:
+     * - Direct file path match
+     * - Relative import resolution (./foo, ../bar)
+     * - Scoped package imports (@workspace/ui, @ui/core)
+     *
+     * For ClassDeclaration/InterfaceDeclaration with filePath, uses precise matching.
      */
-    findNodeByNameAndType(name, coreType) {
-        for (const node of this.parsedNodes.values()) {
+    findNodeByNameAndType(name, coreType, filePath) {
+        // Combine both node collections for searching
+        const allNodes = [...this.parsedNodes.values(), ...this.existingNodes.values()];
+        // If we have a file path and it's not a SOURCE_FILE, use precise matching first
+        if (filePath && coreType !== CoreNodeType.SOURCE_FILE) {
+            for (const node of allNodes) {
+                if (node.coreType === coreType && node.properties.name === name && node.properties.filePath === filePath) {
+                    return node;
+                }
+            }
+            // If precise match fails, fall through to name-only matching below
+        }
+        // For SOURCE_FILE with import specifier, try multiple matching strategies
+        if (coreType === CoreNodeType.SOURCE_FILE) {
+            // Strategy 1: Direct file path match
+            for (const node of allNodes) {
+                if (node.coreType === coreType && node.properties.filePath === name) {
+                    return node;
+                }
+            }
+            // Strategy 2: Resolve relative imports (./foo, ../bar, ../../baz)
+            if (name.startsWith('.')) {
+                // Normalize: remove all leading ./ or ../ segments (handles ../../foo, ./bar, etc.)
+                const normalizedPath = name.replace(/^(\.\.?\/)+/, '');
+                // Try matching with common extensions
+                const extensions = ['', '.ts', '.tsx', '/index.ts', '/index.tsx'];
+                for (const ext of extensions) {
+                    const searchPath = normalizedPath + ext;
+                    for (const node of allNodes) {
+                        if (node.coreType === coreType) {
+                            // Match if filePath ends with the normalized path
+                            if (node.properties.filePath.endsWith(searchPath) ||
+                                node.properties.filePath.endsWith('/' + searchPath)) {
+                                return node;
+                            }
+                        }
+                    }
+                }
+            }
+            // Strategy 3: Workspace package imports (@workspace/ui, @ui/core)
+            if (name.startsWith('@')) {
+                const parts = name.split('/');
+                const packageName = parts.slice(0, 2).join('/'); // @scope/package
+                const subPath = parts.slice(2).join('/'); // rest of path after package name
+                // First, try to find an exact match with subpath
+                if (subPath) {
+                    const extensions = ['', '.ts', '.tsx', '/index.ts', '/index.tsx'];
+                    for (const ext of extensions) {
+                        const searchPath = subPath + ext;
+                        for (const node of allNodes) {
+                            if (node.coreType === coreType && node.properties.packageName === packageName) {
+                                if (node.properties.filePath.endsWith(searchPath) ||
+                                    node.properties.filePath.endsWith('/' + searchPath)) {
+                                    return node;
+                                }
+                            }
+                        }
+                    }
+                }
+                // For bare package imports (@workspace/ui), look for index files
+                if (!subPath) {
+                    for (const node of allNodes) {
+                        if (node.coreType === coreType && node.properties.packageName === packageName) {
+                            const fileName = node.properties.name;
+                            if (fileName === 'index.ts' || fileName === 'index.tsx') {
+                                return node;
+                            }
+                        }
+                    }
+                    // If no index file, return any file from the package as a fallback
+                    for (const node of allNodes) {
+                        if (node.coreType === coreType && node.properties.packageName === packageName) {
+                            return node;
+                        }
+                    }
+                }
+            }
+        }
+        // Default: exact name match (for non-SourceFile types like classes, interfaces)
+        for (const node of allNodes) {
             if (node.coreType === coreType && node.properties.name === name) {
                 return node;
             }
@@ -236,14 +466,68 @@ export class TypeScriptParser {
     /**
      * Resolve deferred edges after all nodes have been parsed
      */
-    resolveDeferredEdges() {
+    async resolveDeferredEdges() {
+        // Count edges by type for logging
+        const importsCount = this.deferredEdges.filter((e) => e.edgeType === CoreEdgeType.IMPORTS).length;
+        const extendsCount = this.deferredEdges.filter((e) => e.edgeType === CoreEdgeType.EXTENDS).length;
+        const implementsCount = this.deferredEdges.filter((e) => e.edgeType === CoreEdgeType.IMPLEMENTS).length;
+        let importsResolved = 0;
+        let extendsResolved = 0;
+        let implementsResolved = 0;
+        const unresolvedImports = [];
+        const unresolvedExtends = [];
+        const unresolvedImplements = [];
         for (const deferred of this.deferredEdges) {
-            const targetNode = this.findNodeByNameAndType(deferred.targetName, deferred.targetType);
+            // Pass filePath for precise matching (especially important for EXTENDS/IMPLEMENTS)
+            const targetNode = this.findNodeByNameAndType(deferred.targetName, deferred.targetType, deferred.targetFilePath);
             if (targetNode) {
                 const edge = this.createCoreEdge(deferred.edgeType, deferred.sourceNodeId, targetNode.id);
                 this.addEdge(edge);
+                // Track resolution by type
+                if (deferred.edgeType === CoreEdgeType.IMPORTS) {
+                    importsResolved++;
+                }
+                else if (deferred.edgeType === CoreEdgeType.EXTENDS) {
+                    extendsResolved++;
+                }
+                else if (deferred.edgeType === CoreEdgeType.IMPLEMENTS) {
+                    implementsResolved++;
+                }
+            }
+            else {
+                // Track unresolved by type
+                if (deferred.edgeType === CoreEdgeType.IMPORTS) {
+                    unresolvedImports.push(deferred.targetName);
+                }
+                else if (deferred.edgeType === CoreEdgeType.EXTENDS) {
+                    unresolvedExtends.push(deferred.targetName);
+                }
+                else if (deferred.edgeType === CoreEdgeType.IMPLEMENTS) {
+                    unresolvedImplements.push(deferred.targetName);
+                }
             }
-            // If not found, it's likely an external type (from node_modules) - skip silently
+        }
+        // Log import resolution stats
+        if (importsCount > 0) {
+            await debugLog('Import edge resolution', {
+                totalImports: importsCount,
+                resolved: importsResolved,
+                unresolvedCount: unresolvedImports.length,
+                unresolvedSample: unresolvedImports.slice(0, 10),
+            });
+        }
+        // Log inheritance (EXTENDS/IMPLEMENTS) resolution stats
+        if (extendsCount > 0 || implementsCount > 0) {
+            await debugLog('Inheritance edge resolution', {
+                extendsQueued: extendsCount,
+                extendsResolved,
+                extendsUnresolved: unresolvedExtends.length,
+                unresolvedExtendsSample: unresolvedExtends.slice(0, 10),
+                implementsQueued: implementsCount,
+                implementsResolved,
+                implementsUnresolved: unresolvedImplements.length,
+                unresolvedImplementsSample: unresolvedImplements.slice(0, 10),
+            });
         }
         this.deferredEdges = [];
     }
@@ -347,6 +631,23 @@ export class TypeScriptParser {
                 // File contains import relationship
                 const containsEdge = this.createCoreEdge(CoreEdgeType.CONTAINS, sourceFileNode.id, importNode.id);
                 this.addEdge(containsEdge);
+                // Try to resolve import to create SourceFile -> SourceFile IMPORTS edge
+                try {
+                    const targetSourceFile = importDecl.getModuleSpecifierSourceFile();
+                    if (targetSourceFile) {
+                        const targetFilePath = targetSourceFile.getFilePath();
+                        // Queue deferred edge - will be resolved after all files are parsed
+                        this.deferredEdges.push({
+                            edgeType: CoreEdgeType.IMPORTS,
+                            sourceNodeId: sourceFileNode.id,
+                            targetName: targetFilePath, // Use file path as "name" for SourceFiles
+                            targetType: CoreNodeType.SOURCE_FILE,
+                        });
+                    }
+                }
+                catch {
+                    // Module resolution failed - external dependency, skip
+                }
             }
             // Parse variable declarations if framework schema specifies this file should have them parsed
             if (this.shouldParseVariables(sourceFile.getFilePath())) {
@@ -368,10 +669,11 @@ export class TypeScriptParser {
     createCoreNode(astNode, coreType, baseProperties = {}, parentId) {
         const name = this.extractNodeName(astNode, coreType);
         const filePath = astNode.getSourceFile().getFilePath();
-        const nodeId = generateDeterministicId(coreType, filePath, name, parentId);
+        const nodeId = generateDeterministicId(this.projectId, coreType, filePath, name, parentId);
         // Extract base properties using schema
         const properties = {
             id: nodeId,
+            projectId: this.projectId,
             name,
             coreType,
             filePath,
@@ -449,6 +751,7 @@ export class TypeScriptParser {
             targetNodeId,
             properties: {
                 coreType: relationshipType,
+                projectId: this.projectId,
                 source: 'ast',
                 confidence: 1.0,
                 relationshipWeight,
@@ -600,22 +903,16 @@ export class TypeScriptParser {
         }
     }
     createFrameworkEdge(semanticType, relationshipType, sourceNodeId, targetNodeId, context = {}, relationshipWeight = 0.5) {
-        // Generate deterministic edge ID based on type + source + target
-        const edgeIdentity = `${semanticType}::${sourceNodeId}::${targetNodeId}`;
-        const edgeHash = crypto.createHash('sha256').update(edgeIdentity).digest('hex').substring(0, 16);
-        const edgeId = `${semanticType}:${edgeHash}`;
-        const properties = {
-            coreType: semanticType, // This might need adjustment based on schema
+        const { id, properties } = createFrameworkEdgeData({
             semanticType,
-            source: 'pattern',
-            confidence: 0.8,
-            relationshipWeight,
-            filePath: '',
-            createdAt: new Date().toISOString(),
+            sourceNodeId,
+            targetNodeId,
+            projectId: this.projectId,
             context,
-        };
+            relationshipWeight,
+        });
         return {
-            id: edgeId,
+            id,
             relationshipType,
             sourceNodeId,
             targetNodeId,
@@ -712,11 +1009,29 @@ export class TypeScriptParser {
         const excludedNodeTypes = this.parseConfig.excludedNodeTypes ?? [];
         return excludedNodeTypes.includes(node.getKindName());
     }
+    /**
+     * Safely test if a file path matches a pattern (string or regex).
+     * Falls back to literal string matching if the pattern is an invalid regex.
+     */
+    matchesPattern(filePath, pattern) {
+        // First try literal string match (always safe)
+        if (filePath.includes(pattern)) {
+            return true;
+        }
+        // Then try regex match with error handling
+        try {
+            return new RegExp(pattern).test(filePath);
+        }
+        catch {
+            // Invalid regex pattern - already checked via includes() above
+            return false;
+        }
+    }
     shouldSkipFile(sourceFile) {
         const filePath = sourceFile.getFilePath();
         const excludedPatterns = this.parseConfig.excludePatterns ?? [];
         for (const pattern of excludedPatterns) {
-            if (filePath.includes(pattern) || filePath.match(new RegExp(pattern))) {
+            if (this.matchesPattern(filePath, pattern)) {
                 return true;
             }
         }
@@ -778,4 +1093,284 @@ export class TypeScriptParser {
         }));
         return { nodes, edges };
     }
+    // ============================================
+    // CHUNK-AWARE PARSING METHODS
+    // For streaming/chunked parsing of large codebases
+    // ============================================
+    /**
+     * Export current chunk results without clearing internal state.
+     * Use this when importing chunks incrementally.
+     */
+    exportChunkResults() {
+        const nodes = Array.from(this.parsedNodes.values()).map(this.toNeo4jNode);
+        const edges = Array.from(this.parsedEdges.values()).map(this.toNeo4jEdge);
+        return {
+            nodes,
+            edges,
+            deferredEdges: [...this.deferredEdges],
+        };
+    }
+    /**
+     * Clear all parsed data (nodes, edges, deferred edges).
+     * Call this after importing a chunk to free memory.
+     */
+    clearParsedData() {
+        this.parsedNodes.clear();
+        this.parsedEdges.clear();
+        this.deferredEdges = [];
+    }
+    /**
+     * Get count of currently parsed nodes and edges.
+     * Useful for progress reporting.
+     */
+    getCurrentCounts() {
+        return {
+            nodes: this.parsedNodes.size,
+            edges: this.parsedEdges.size,
+            deferredEdges: this.deferredEdges.length,
+        };
+    }
+    /**
+     * Set the shared context for this parser.
+     * Use this to share context across multiple parsers (e.g., in WorkspaceParser).
+     * @param context The shared context map to use
+     */
+    setSharedContext(context) {
+        this.sharedContext = context;
+    }
+    /**
+     * Get the shared context from this parser.
+     * Useful for aggregating context across multiple parsers.
+     */
+    getSharedContext() {
+        return this.sharedContext;
+    }
+    /**
+     * Get all parsed nodes (for cross-parser edge resolution).
+     * Returns the internal Map of ParsedNodes.
+     */
+    getParsedNodes() {
+        return this.parsedNodes;
+    }
+    /**
+     * Get the framework schemas used by this parser.
+     * Useful for WorkspaceParser to apply cross-package edge enhancements.
+     */
+    getFrameworkSchemas() {
+        return this.frameworkSchemas;
+    }
+    /**
+     * Defer edge enhancements to a parent parser (e.g., WorkspaceParser).
+     * When true, parseChunk() will skip applyEdgeEnhancements().
+     * The parent is responsible for calling applyEdgeEnhancementsManually() at the end.
+     */
+    setDeferEdgeEnhancements(defer) {
+        this.deferEdgeEnhancements = defer;
+    }
+    /**
+     * Get list of source files in the project.
+     * In lazy mode, uses glob to discover files without loading them into memory.
+     * Useful for determining total work and creating chunks.
+     */
+    async discoverSourceFiles() {
+        if (this.discoveredFiles !== null) {
+            return this.discoveredFiles;
+        }
+        if (this.lazyLoad) {
+            // Use glob to find files without loading them into ts-morph
+            // Include both .ts and .tsx files
+            const pattern = path.join(this.workspacePath, '**/*.{ts,tsx}');
+            const allFiles = await glob(pattern, {
+                ignore: ['**/node_modules/**', '**/*.d.ts'],
+                absolute: true,
+            });
+            // Apply exclude patterns from parseConfig
+            const excludedPatterns = this.parseConfig.excludePatterns ?? [];
+            this.discoveredFiles = allFiles.filter((filePath) => {
+                for (const excludePattern of excludedPatterns) {
+                    if (this.matchesPattern(filePath, excludePattern)) {
+                        return false;
+                    }
+                }
+                return true;
+            });
+            console.log(`🔍 Discovered ${this.discoveredFiles.length} TypeScript files (lazy mode)`);
+            return this.discoveredFiles;
+        }
+        else {
+            // Eager mode - files are already loaded
+            this.discoveredFiles = this.project
+                .getSourceFiles()
+                .filter((sf) => !this.shouldSkipFile(sf))
+                .map((sf) => sf.getFilePath());
+            return this.discoveredFiles;
+        }
+    }
+    /**
+     * @deprecated Use discoverSourceFiles() instead for async file discovery
+     */
+    getSourceFilePaths() {
+        if (this.lazyLoad) {
+            throw new Error('getSourceFilePaths() is not supported in lazy mode. Use discoverSourceFiles() instead.');
+        }
+        return this.project
+            .getSourceFiles()
+            .filter((sf) => !this.shouldSkipFile(sf))
+            .map((sf) => sf.getFilePath());
+    }
+    /**
+     * Parse a chunk of files without resolving deferred edges.
+     * Use this for streaming parsing where edges are resolved after all chunks.
+     * In lazy mode, files are added to the project just-in-time and removed after parsing.
+     * @param filePaths Specific file paths to parse
+     * @param skipEdgeResolution If true, deferred edges are not resolved (default: false)
+     */
+    async parseChunk(filePaths, skipEdgeResolution = false) {
+        // Declare sourceFiles outside try so it's available in finally
+        const sourceFiles = [];
+        try {
+            if (this.lazyLoad) {
+                // Lazy mode: add files to project just-in-time
+                for (const filePath of filePaths) {
+                    try {
+                        // Check if file already exists in project (shouldn't happen in lazy mode)
+                        // Add the file to the project if not already present
+                        const sourceFile = this.project.getSourceFile(filePath) ?? this.project.addSourceFileAtPath(filePath);
+                        sourceFiles.push(sourceFile);
+                    }
+                    catch (error) {
+                        console.warn(`Failed to add source file ${filePath}:`, error);
+                    }
+                }
+            }
+            else {
+                // Eager mode: files are already loaded
+                const loadedFiles = filePaths
+                    .map((filePath) => this.project.getSourceFile(filePath))
+                    .filter((sf) => sf !== undefined);
+                sourceFiles.push(...loadedFiles);
+            }
+            for (const sourceFile of sourceFiles) {
+                if (this.shouldSkipFile(sourceFile))
+                    continue;
+                await this.parseCoreTypeScriptV2(sourceFile);
+            }
+            // Only resolve edges if not skipping
+            if (!skipEdgeResolution) {
+                await this.resolveDeferredEdges();
+            }
+            await this.applyContextExtractors();
+            if (this.frameworkSchemas.length > 0) {
+                await this.applyFrameworkEnhancements();
+            }
+            // Apply edge enhancements unless deferred to parent (e.g., WorkspaceParser)
+            // When deferred, parent will call applyEdgeEnhancementsManually() at the end
+            // with all accumulated nodes for cross-package edge detection
+            if (!this.deferEdgeEnhancements) {
+                await this.applyEdgeEnhancements();
+            }
+            const neo4jNodes = Array.from(this.parsedNodes.values()).map(this.toNeo4jNode);
+            const neo4jEdges = Array.from(this.parsedEdges.values()).map(this.toNeo4jEdge);
+            return { nodes: neo4jNodes, edges: neo4jEdges };
+        }
+        finally {
+            // Always clean up in lazy mode to prevent memory leaks
+            if (this.lazyLoad) {
+                for (const sourceFile of sourceFiles) {
+                    try {
+                        this.project.removeSourceFile(sourceFile);
+                    }
+                    catch {
+                        // Ignore errors when removing files
+                    }
+                }
+            }
+        }
+    }
+    /**
+     * Resolve deferred edges against both parsed nodes and existing nodes.
+     * Call this after all chunks have been parsed.
+     * @returns Resolved edges
+     */
+    async resolveDeferredEdgesManually() {
+        const resolvedEdges = [];
+        // Count edges by type for logging
+        const extendsCount = this.deferredEdges.filter((e) => e.edgeType === CoreEdgeType.EXTENDS).length;
+        const implementsCount = this.deferredEdges.filter((e) => e.edgeType === CoreEdgeType.IMPLEMENTS).length;
+        let extendsResolved = 0;
+        let implementsResolved = 0;
+        const unresolvedExtends = [];
+        const unresolvedImplements = [];
+        for (const deferred of this.deferredEdges) {
+            // Pass filePath for precise matching (especially important for EXTENDS/IMPLEMENTS)
+            const targetNode = this.findNodeByNameAndType(deferred.targetName, deferred.targetType, deferred.targetFilePath);
+            if (targetNode) {
+                const edge = this.createCoreEdge(deferred.edgeType, deferred.sourceNodeId, targetNode.id);
+                resolvedEdges.push(edge);
+                this.addEdge(edge);
+                if (deferred.edgeType === CoreEdgeType.EXTENDS) {
+                    extendsResolved++;
+                }
+                else if (deferred.edgeType === CoreEdgeType.IMPLEMENTS) {
+                    implementsResolved++;
+                }
+            }
+            else {
+                if (deferred.edgeType === CoreEdgeType.EXTENDS) {
+                    unresolvedExtends.push(deferred.targetName);
+                }
+                else if (deferred.edgeType === CoreEdgeType.IMPLEMENTS) {
+                    unresolvedImplements.push(deferred.targetName);
+                }
+            }
+        }
+        // Log inheritance resolution stats
+        if (extendsCount > 0 || implementsCount > 0) {
+            await debugLog('Inheritance edge resolution (manual)', {
+                extendsQueued: extendsCount,
+                extendsResolved,
+                extendsUnresolved: unresolvedExtends.length,
+                unresolvedExtendsSample: unresolvedExtends.slice(0, 10),
+                implementsQueued: implementsCount,
+                implementsResolved,
+                implementsUnresolved: unresolvedImplements.length,
+                unresolvedImplementsSample: unresolvedImplements.slice(0, 10),
+            });
+        }
+        this.deferredEdges = [];
+        return resolvedEdges.map(this.toNeo4jEdge);
+    }
+    /**
+     * Apply edge enhancements on all accumulated nodes.
+     * Call this after all chunks have been parsed for streaming mode.
+     * This allows context-dependent edges (like INTERNAL_API_CALL) to be detected
+     * after all nodes and their context have been collected.
+     * @returns New edges created by edge enhancements
+     */
+    async applyEdgeEnhancementsManually() {
+        const edgeCountBefore = this.parsedEdges.size;
+        console.log(`🔗 Applying edge enhancements on ${this.parsedNodes.size} accumulated nodes...`);
+        await this.applyEdgeEnhancements();
+        const newEdgeCount = this.parsedEdges.size - edgeCountBefore;
+        console.log(`   ✅ Created ${newEdgeCount} edges from edge enhancements`);
+        // Return only the new edges (those created by edge enhancements)
+        const allEdges = Array.from(this.parsedEdges.values()).map(this.toNeo4jEdge);
+        return allEdges.slice(edgeCountBefore);
+    }
+    /**
+     * Add nodes to the existing nodes map for cross-chunk edge resolution.
+     * These nodes are considered as potential edge targets but won't be exported.
+     */
+    addExistingNodesFromChunk(nodes) {
+        for (const node of nodes) {
+            const parsedNode = {
+                id: node.id,
+                coreType: node.properties.coreType,
+                semanticType: node.properties.semanticType,
+                labels: node.labels,
+                properties: node.properties,
+            };
+            this.existingNodes.set(node.id, parsedNode);
+        }
+    }
 }