@fairfox/polly 0.3.1 → 0.3.4

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

@@ -1321,7 +1321,11 @@ class HandlerExtractor {
   typeGuardCache;
   constructor(tsConfigPath) {
     this.project = new Project4({
-      tsConfigFilePath: tsConfigPath
+      tsConfigFilePath: tsConfigPath,
+      compilerOptions: {
+        allowImportingTsExtensions: true,
+        moduleResolution: 99
+      }
     });
     this.typeGuardCache = new WeakMap;
   }
@@ -1329,6 +1333,14 @@ class HandlerExtractor {
     const handlers = [];
     const messageTypes = new Set;
     const sourceFiles = this.project.getSourceFiles();
+    if (process.env.POLLY_DEBUG) {
+      console.log(`[DEBUG] Loaded ${sourceFiles.length} source files`);
+      if (sourceFiles.length <= 20) {
+        for (const sf of sourceFiles) {
+          console.log(`[DEBUG]   - ${sf.getFilePath()}`);
+        }
+      }
+    }
     for (const sourceFile of sourceFiles) {
       const fileHandlers = this.extractFromFile(sourceFile);
       handlers.push(...fileHandlers);
@@ -1336,6 +1348,9 @@ class HandlerExtractor {
         messageTypes.add(handler.messageType);
       }
     }
+    if (process.env.POLLY_DEBUG) {
+      console.log(`[DEBUG] Total handlers extracted: ${handlers.length}`);
+    }
     return {
       handlers,
       messageTypes
@@ -1589,14 +1604,23 @@ class HandlerExtractor {
         typeGuards = this.findTypePredicateFunctions(sourceFile);
         this.typeGuardCache.set(sourceFile, typeGuards);
       }
-      if (typeGuards.size === 0) {
-        return handlers;
+      if (process.env.POLLY_DEBUG) {
+        console.log(`[DEBUG] File: ${sourceFile.getBaseName()}`);
+        console.log(`[DEBUG] Local type guards found: ${typeGuards.size}`);
+        if (typeGuards.size > 0) {
+          for (const [name, type] of typeGuards.entries()) {
+            console.log(`[DEBUG]   - ${name} → ${type}`);
+          }
+        }
       }
       let currentIf = ifNode;
       while (currentIf) {
         const handler = this.extractHandlerFromIfClause(currentIf, typeGuards, context, filePath);
         if (handler) {
           handlers.push(handler);
+          if (process.env.POLLY_DEBUG) {
+            console.log(`[DEBUG] Found handler: ${handler.messageType} at line ${handler.location.line}`);
+          }
         }
         const elseStatement = currentIf.getElseStatement();
         if (elseStatement && Node4.isIfStatement(elseStatement)) {
@@ -1605,7 +1629,11 @@ class HandlerExtractor {
           break;
         }
       }
-    } catch (error) {}
+    } catch (error) {
+      if (process.env.POLLY_DEBUG) {
+        console.log(`[DEBUG] Error in extractTypeGuardHandlers: ${error}`);
+      }
+    }
     return handlers;
   }
   extractHandlerFromIfClause(ifNode, typeGuards, context, filePath) {
@@ -1619,10 +1647,27 @@ class HandlerExtractor {
       if (Node4.isIdentifier(funcExpr)) {
         funcName = funcExpr.getText();
       }
-      if (!funcName || !typeGuards.has(funcName)) {
+      if (process.env.POLLY_DEBUG && funcName) {
+        console.log(`[DEBUG] Processing if condition with function: ${funcName}`);
+      }
+      let messageType = undefined;
+      if (funcName && typeGuards.has(funcName)) {
+        messageType = typeGuards.get(funcName);
+        if (process.env.POLLY_DEBUG) {
+          console.log(`[DEBUG] Found in local type guards: ${funcName} → ${messageType}`);
+        }
+      } else if (Node4.isIdentifier(funcExpr)) {
+        if (process.env.POLLY_DEBUG) {
+          console.log(`[DEBUG] Not found locally, trying import resolution for: ${funcName}`);
+        }
+        messageType = this.resolveImportedTypeGuard(funcExpr);
+      }
+      if (!messageType) {
+        if (process.env.POLLY_DEBUG && funcName) {
+          console.log(`[DEBUG] Could not resolve message type for: ${funcName}`);
+        }
         return null;
       }
-      const messageType = typeGuards.get(funcName);
       const line = ifNode.getStartLineNumber();
       return {
         messageType,
@@ -1683,6 +1728,55 @@ class HandlerExtractor {
     });
     return typeGuards;
   }
+  resolveImportedTypeGuard(identifier) {
+    try {
+      const funcName = identifier.getText();
+      const definitions = identifier.getDefinitionNodes();
+      if (definitions.length === 0) {
+        if (process.env.POLLY_DEBUG) {
+          console.log(`[DEBUG] No definitions found for imported function: ${funcName}`);
+        }
+        return null;
+      }
+      for (const def of definitions) {
+        if (Node4.isFunctionDeclaration(def) || Node4.isFunctionExpression(def) || Node4.isArrowFunction(def)) {
+          const returnType = def.getReturnType();
+          const returnTypeText = returnType.getText();
+          if (process.env.POLLY_DEBUG) {
+            console.log(`[DEBUG] Function ${funcName} return type: ${returnTypeText}`);
+          }
+          if (/is\s+\w+/.test(returnTypeText)) {
+            const typeMatch = returnTypeText.match(/is\s+(\w+)/);
+            if (typeMatch) {
+              const typeName = typeMatch[1];
+              const messageType = this.extractMessageTypeFromTypeName(typeName);
+              if (process.env.POLLY_DEBUG) {
+                console.log(`[DEBUG] Resolved ${funcName} → ${messageType}`);
+              }
+              return messageType;
+            }
+            const body = def.getBody();
+            if (body) {
+              const bodyText = body.getText();
+              const typeValueMatch = bodyText.match(/\.type\s*===?\s*['"](\w+)['"]/);
+              if (typeValueMatch) {
+                const messageType = typeValueMatch[1];
+                if (process.env.POLLY_DEBUG) {
+                  console.log(`[DEBUG] Resolved ${funcName} → ${messageType} (from body)`);
+                }
+                return messageType;
+              }
+            }
+          }
+        }
+      }
+    } catch (error) {
+      if (process.env.POLLY_DEBUG) {
+        console.log(`[DEBUG] Error resolving imported type guard: ${error}`);
+      }
+    }
+    return null;
+  }
   extractMessageTypeFromTypeName(typeName) {
     const messageType = typeName.replace(/Message$/, "").replace(/Event$/, "").replace(/Request$/, "").replace(/Command$/, "").replace(/Query$/, "").toLowerCase();
     return messageType;
@@ -2909,4 +3003,4 @@ Stack trace:`, COLORS.gray));
   process.exit(1);
 });
 
-//# debugId=2DAA9097DE2E4CFB64756E2164756E21
+//# debugId=FFB153526798575464756E2164756E21