@fairfox/polly 0.10.1 → 0.11.0

package/dist/tools/visualize/src/cli.js

@@ -1562,40 +1562,116 @@ class HandlerExtractor {
   project;
   typeGuardCache;
   relationshipExtractor;
+  analyzedFiles;
+  packageRoot;
   constructor(tsConfigPath) {
     this.project = new Project3({
       tsConfigFilePath: tsConfigPath
     });
     this.typeGuardCache = new WeakMap;
     this.relationshipExtractor = new RelationshipExtractor;
+    this.analyzedFiles = new Set;
+    this.packageRoot = this.findPackageRoot(tsConfigPath);
+  }
+  findPackageRoot(tsConfigPath) {
+    let dir = tsConfigPath.substring(0, tsConfigPath.lastIndexOf("/"));
+    while (dir.length > 1) {
+      try {
+        const packageJsonPath = `${dir}/package.json`;
+        const file = Bun.file(packageJsonPath);
+        if (file.size > 0) {
+          return dir;
+        }
+      } catch {}
+      const parentDir = dir.substring(0, dir.lastIndexOf("/"));
+      if (parentDir === dir)
+        break;
+      dir = parentDir;
+    }
+    return tsConfigPath.substring(0, tsConfigPath.lastIndexOf("/"));
   }
   extractHandlers() {
     const handlers = [];
     const messageTypes = new Set;
     const invalidMessageTypes = new Set;
     const stateConstraints = [];
-    const sourceFiles = this.project.getSourceFiles();
-    this.debugLogSourceFiles(sourceFiles);
-    for (const sourceFile of sourceFiles) {
-      const fileHandlers = this.extractFromFile(sourceFile);
-      handlers.push(...fileHandlers);
-      this.categorizeHandlerMessageTypes(fileHandlers, messageTypes, invalidMessageTypes);
-      const fileConstraints = this.extractStateConstraintsFromFile(sourceFile);
-      stateConstraints.push(...fileConstraints);
+    const allSourceFiles = this.project.getSourceFiles();
+    const entryPoints = allSourceFiles.filter((f) => this.isWithinPackage(f.getFilePath()));
+    this.debugLogSourceFiles(allSourceFiles, entryPoints);
+    for (const entryPoint of entryPoints) {
+      this.analyzeFileAndImports(entryPoint, handlers, messageTypes, invalidMessageTypes, stateConstraints);
     }
     this.debugLogExtractionResults(handlers.length, invalidMessageTypes.size);
+    this.debugLogAnalysisStats(allSourceFiles.length, entryPoints.length);
     return {
       handlers,
       messageTypes,
       stateConstraints
     };
   }
-  debugLogSourceFiles(sourceFiles) {
+  analyzeFileAndImports(sourceFile, handlers, messageTypes, invalidMessageTypes, stateConstraints) {
+    const filePath = sourceFile.getFilePath();
+    if (this.analyzedFiles.has(filePath)) {
+      return;
+    }
+    this.analyzedFiles.add(filePath);
+    if (process.env["POLLY_DEBUG"]) {
+      console.log(`[DEBUG] Analyzing: ${filePath}`);
+    }
+    const fileHandlers = this.extractFromFile(sourceFile);
+    handlers.push(...fileHandlers);
+    this.categorizeHandlerMessageTypes(fileHandlers, messageTypes, invalidMessageTypes);
+    const fileConstraints = this.extractStateConstraintsFromFile(sourceFile);
+    stateConstraints.push(...fileConstraints);
+    const importDeclarations = sourceFile.getImportDeclarations();
+    for (const importDecl of importDeclarations) {
+      const importedFile = importDecl.getModuleSpecifierSourceFile();
+      if (importedFile) {
+        const importedPath = importedFile.getFilePath();
+        if (!this.isWithinPackage(importedPath)) {
+          if (process.env["POLLY_DEBUG"]) {
+            console.log(`[DEBUG] Skipping external import: ${importedPath}`);
+          }
+          continue;
+        }
+        this.analyzeFileAndImports(importedFile, handlers, messageTypes, invalidMessageTypes, stateConstraints);
+      } else if (process.env["POLLY_DEBUG"]) {
+        const specifier = importDecl.getModuleSpecifierValue();
+        if (!specifier.startsWith("node:") && !this.isNodeModuleImport(specifier)) {
+          console.log(`[DEBUG] Could not resolve import: ${specifier} in ${filePath}`);
+        }
+      }
+    }
+  }
+  isWithinPackage(filePath) {
+    if (!filePath.startsWith(this.packageRoot)) {
+      return false;
+    }
+    if (filePath.includes("/node_modules/")) {
+      return false;
+    }
+    return true;
+  }
+  isNodeModuleImport(specifier) {
+    return !specifier.startsWith(".") && !specifier.startsWith("/");
+  }
+  debugLogAnalysisStats(totalSourceFiles, entryPointCount) {
+    if (!process.env["POLLY_DEBUG"])
+      return;
+    console.log(`[DEBUG] Analysis Statistics:`);
+    console.log(`[DEBUG]   Package root: ${this.packageRoot}`);
+    console.log(`[DEBUG]   Source files from tsconfig: ${totalSourceFiles}`);
+    console.log(`[DEBUG]   Entry points (in package): ${entryPointCount}`);
+    console.log(`[DEBUG]   Files analyzed (including imports): ${this.analyzedFiles.size}`);
+    console.log(`[DEBUG]   Additional files discovered: ${this.analyzedFiles.size - entryPointCount}`);
+  }
+  debugLogSourceFiles(allSourceFiles, entryPoints) {
     if (!process.env["POLLY_DEBUG"])
       return;
-    console.log(`[DEBUG] Loaded ${sourceFiles.length} source files`);
-    if (sourceFiles.length <= 20) {
-      for (const sf of sourceFiles) {
+    console.log(`[DEBUG] Loaded ${allSourceFiles.length} source files from tsconfig`);
+    console.log(`[DEBUG] Filtered to ${entryPoints.length} entry points within package`);
+    if (entryPoints.length <= 20) {
+      for (const sf of entryPoints) {
         console.log(`[DEBUG]   - ${sf.getFilePath()}`);
       }
     }
@@ -4799,4 +4875,4 @@ main().catch((_error) => {
   process.exit(1);
 });
 
-//# debugId=4E2F22D76354977564756E2164756E21
+//# debugId=95CA9FB262DAF8A164756E2164756E21