@fairfox/polly 0.3.0 → 0.3.2

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

@@ -1318,10 +1318,12 @@ import { Project as Project4, SyntaxKind as SyntaxKind2, Node as Node4 } from "t
 
 class HandlerExtractor {
   project;
+  typeGuardCache;
   constructor(tsConfigPath) {
     this.project = new Project4({
       tsConfigFilePath: tsConfigPath
     });
+    this.typeGuardCache = new WeakMap;
   }
   extractHandlers() {
     const handlers = [];
@@ -1364,6 +1366,10 @@ class HandlerExtractor {
         const mapHandlers = this.extractHandlerMapPattern(node, context, filePath);
         handlers.push(...mapHandlers);
       }
+      if (Node4.isIfStatement(node)) {
+        const typeGuardHandlers = this.extractTypeGuardHandlers(node, context, filePath);
+        handlers.push(...typeGuardHandlers);
+      }
     });
     return handlers;
   }
@@ -1574,6 +1580,145 @@ class HandlerExtractor {
     } catch (error) {}
     return handlers;
   }
+  extractTypeGuardHandlers(ifNode, context, filePath) {
+    const handlers = [];
+    try {
+      const sourceFile = ifNode.getSourceFile();
+      let typeGuards = this.typeGuardCache.get(sourceFile);
+      if (!typeGuards) {
+        typeGuards = this.findTypePredicateFunctions(sourceFile);
+        this.typeGuardCache.set(sourceFile, typeGuards);
+      }
+      if (typeGuards.size === 0) {
+        return handlers;
+      }
+      let currentIf = ifNode;
+      while (currentIf) {
+        const handler = this.extractHandlerFromIfClause(currentIf, typeGuards, context, filePath);
+        if (handler) {
+          handlers.push(handler);
+        }
+        const elseStatement = currentIf.getElseStatement();
+        if (elseStatement && Node4.isIfStatement(elseStatement)) {
+          currentIf = elseStatement;
+        } else {
+          break;
+        }
+      }
+    } catch (error) {}
+    return handlers;
+  }
+  extractHandlerFromIfClause(ifNode, typeGuards, context, filePath) {
+    try {
+      const condition = ifNode.getExpression();
+      if (!Node4.isCallExpression(condition)) {
+        return null;
+      }
+      const funcExpr = condition.getExpression();
+      let funcName;
+      if (Node4.isIdentifier(funcExpr)) {
+        funcName = funcExpr.getText();
+      }
+      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 line = ifNode.getStartLineNumber();
+      return {
+        messageType,
+        node: context,
+        assignments: [],
+        preconditions: [],
+        postconditions: [],
+        location: { file: filePath, line }
+      };
+    } catch (error) {
+      return null;
+    }
+  }
+  findTypePredicateFunctions(sourceFile) {
+    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)) {
+          let functionName;
+          if (Node4.isFunctionDeclaration(node)) {
+            functionName = node.getName();
+          } else if (Node4.isFunctionExpression(node)) {
+            const parent = node.getParent();
+            if (Node4.isVariableDeclaration(parent)) {
+              functionName = parent.getName();
+            }
+          } else if (Node4.isArrowFunction(node)) {
+            const parent = node.getParent();
+            if (Node4.isVariableDeclaration(parent)) {
+              functionName = parent.getName();
+            }
+          }
+          if (functionName) {
+            let messageType = null;
+            const typeMatch = returnTypeText.match(/is\s+(\w+)/);
+            if (typeMatch) {
+              const typeName = typeMatch[1];
+              messageType = this.extractMessageTypeFromTypeName(typeName);
+            }
+            if (!messageType) {
+              const body = node.getBody();
+              if (body) {
+                const bodyText = body.getText();
+                const typeValueMatch = bodyText.match(/\.type\s*===?\s*['"](\w+)['"]/);
+                if (typeValueMatch) {
+                  messageType = typeValueMatch[1];
+                }
+              }
+            }
+            if (messageType) {
+              typeGuards.set(functionName, messageType);
+            }
+          }
+        }
+      }
+    });
+    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;
+  }
   inferContext(filePath) {
     const path2 = filePath.toLowerCase();
     if (path2.includes("/background/") || path2.includes("\\background\\")) {
@@ -2796,4 +2941,4 @@ Stack trace:`, COLORS.gray));
   process.exit(1);
 });
 
-//# debugId=37FBF573C9C57A4C64756E2164756E21
+//# debugId=0C75E6084607493864756E2164756E21