@fairfox/polly 0.3.2 → 0.3.6

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

@@ -1329,6 +1329,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 +1344,9 @@ class HandlerExtractor {
         messageTypes.add(handler.messageType);
       }
     }
+    if (process.env.POLLY_DEBUG) {
+      console.log(`[DEBUG] Total handlers extracted: ${handlers.length}`);
+    }
     return {
       handlers,
       messageTypes
@@ -1367,8 +1378,12 @@ class HandlerExtractor {
         handlers.push(...mapHandlers);
       }
       if (Node4.isIfStatement(node)) {
-        const typeGuardHandlers = this.extractTypeGuardHandlers(node, context, filePath);
-        handlers.push(...typeGuardHandlers);
+        const parent = node.getParent();
+        const isElseIf = parent && Node4.isIfStatement(parent);
+        if (!isElseIf) {
+          const typeGuardHandlers = this.extractTypeGuardHandlers(node, context, filePath);
+          handlers.push(...typeGuardHandlers);
+        }
       }
     });
     return handlers;
@@ -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,13 +1647,25 @@ class HandlerExtractor {
       if (Node4.isIdentifier(funcExpr)) {
         funcName = funcExpr.getText();
       }
+      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 line = ifNode.getStartLineNumber();
@@ -1645,9 +1685,8 @@ class HandlerExtractor {
     const typeGuards = new Map;
     sourceFile.forEachDescendant((node) => {
       if (Node4.isFunctionDeclaration(node) || Node4.isFunctionExpression(node) || Node4.isArrowFunction(node)) {
-        const returnType = node.getReturnType();
-        const returnTypeText = returnType.getText();
-        if (/is\s+\w+/.test(returnTypeText)) {
+        const returnTypeNode = node.getReturnTypeNode();
+        if (returnTypeNode && Node4.isTypePredicate(returnTypeNode)) {
           let functionName;
           if (Node4.isFunctionDeclaration(node)) {
             functionName = node.getName();
@@ -1663,10 +1702,10 @@ class HandlerExtractor {
             }
           }
           if (functionName) {
+            const typeNode = returnTypeNode.getTypeNode();
             let messageType = null;
-            const typeMatch = returnTypeText.match(/is\s+(\w+)/);
-            if (typeMatch) {
-              const typeName = typeMatch[1];
+            if (typeNode) {
+              const typeName = typeNode.getText();
               messageType = this.extractMessageTypeFromTypeName(typeName);
             }
             if (!messageType) {
@@ -1690,29 +1729,55 @@ class HandlerExtractor {
   }
   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 (/is\s+\w+/.test(returnTypeText)) {
-            const typeMatch = returnTypeText.match(/is\s+(\w+)/);
-            if (typeMatch) {
-              const typeName = typeMatch[1];
-              return this.extractMessageTypeFromTypeName(typeName);
+          const returnTypeNode = def.getReturnTypeNode();
+          if (process.env.POLLY_DEBUG) {
+            const returnType = def.getReturnType().getText();
+            console.log(`[DEBUG] Function ${funcName} return type (resolved): ${returnType}`);
+            console.log(`[DEBUG] Has return type node: ${!!returnTypeNode}`);
+            console.log(`[DEBUG] Is type predicate node: ${returnTypeNode && Node4.isTypePredicate(returnTypeNode)}`);
+          }
+          if (returnTypeNode && Node4.isTypePredicate(returnTypeNode)) {
+            const typeNode = returnTypeNode.getTypeNode();
+            if (typeNode) {
+              const typeName = typeNode.getText();
+              const messageType = this.extractMessageTypeFromTypeName(typeName);
+              if (messageType) {
+                if (process.env.POLLY_DEBUG) {
+                  console.log(`[DEBUG] Resolved ${funcName} → ${messageType} (from AST type predicate)`);
+                }
+                return messageType;
+              }
             }
-            const body = def.getBody();
-            if (body) {
-              const bodyText = body.getText();
-              const typeValueMatch = bodyText.match(/\.type\s*===?\s*['"](\w+)['"]/);
-              if (typeValueMatch) {
-                return typeValueMatch[1];
+          }
+          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) {}
+    } catch (error) {
+      if (process.env.POLLY_DEBUG) {
+        console.log(`[DEBUG] Error resolving imported type guard: ${error}`);
+      }
+    }
     return null;
   }
   extractMessageTypeFromTypeName(typeName) {
@@ -2941,4 +3006,4 @@ Stack trace:`, COLORS.gray));
   process.exit(1);
 });
 
-//# debugId=0C75E6084607493864756E2164756E21
+//# debugId=A5E7982E558C442F64756E2164756E21