@fairfox/polly 0.3.1 → 0.3.2

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

@@ -1619,10 +1619,15 @@ class HandlerExtractor {
       if (Node4.isIdentifier(funcExpr)) {
         funcName = funcExpr.getText();
       }
-      if (!funcName || !typeGuards.has(funcName)) {
+      let messageType = undefined;
+      if (funcName && typeGuards.has(funcName)) {
+        messageType = typeGuards.get(funcName);
+      } else if (Node4.isIdentifier(funcExpr)) {
+        messageType = this.resolveImportedTypeGuard(funcExpr);
+      }
+      if (!messageType) {
         return null;
       }
-      const messageType = typeGuards.get(funcName);
       const line = ifNode.getStartLineNumber();
       return {
         messageType,
@@ -1683,6 +1688,33 @@ class HandlerExtractor {
     });
     return typeGuards;
   }
+  resolveImportedTypeGuard(identifier) {
+    try {
+      const definitions = identifier.getDefinitionNodes();
+      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 body = def.getBody();
+            if (body) {
+              const bodyText = body.getText();
+              const typeValueMatch = bodyText.match(/\.type\s*===?\s*['"](\w+)['"]/);
+              if (typeValueMatch) {
+                return typeValueMatch[1];
+              }
+            }
+          }
+        }
+      }
+    } catch (error) {}
+    return null;
+  }
   extractMessageTypeFromTypeName(typeName) {
     const messageType = typeName.replace(/Message$/, "").replace(/Event$/, "").replace(/Request$/, "").replace(/Command$/, "").replace(/Query$/, "").toLowerCase();
     return messageType;
@@ -2909,4 +2941,4 @@ Stack trace:`, COLORS.gray));
   process.exit(1);
 });
 
-//# debugId=2DAA9097DE2E4CFB64756E2164756E21
+//# debugId=0C75E6084607493864756E2164756E21