@fairfox/polly 0.9.1 → 0.10.0

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

@@ -1573,17 +1573,21 @@ class HandlerExtractor {
     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);
     }
     this.debugLogExtractionResults(handlers.length, invalidMessageTypes.size);
     return {
       handlers,
-      messageTypes
+      messageTypes,
+      stateConstraints
     };
   }
   debugLogSourceFiles(sourceFiles) {
@@ -2415,6 +2419,84 @@ class HandlerExtractor {
       mutations.push({ field, line, afterAwait });
     }
   }
+  extractStateConstraintsFromFile(sourceFile) {
+    const constraints = [];
+    const filePath = sourceFile.getFilePath();
+    sourceFile.forEachDescendant((node) => {
+      const nodeConstraints = this.recognizeStateConstraint(node, filePath);
+      constraints.push(...nodeConstraints);
+    });
+    return constraints;
+  }
+  recognizeStateConstraint(node, filePath) {
+    if (!Node4.isCallExpression(node)) {
+      return [];
+    }
+    const expression = node.getExpression();
+    if (!Node4.isIdentifier(expression)) {
+      return [];
+    }
+    const functionName = expression.getText();
+    if (functionName !== "$constraints") {
+      return [];
+    }
+    const args = node.getArguments();
+    if (args.length < 2) {
+      return [];
+    }
+    const fieldArg = args[0];
+    if (!Node4.isStringLiteral(fieldArg)) {
+      return [];
+    }
+    const field = fieldArg.getLiteralValue();
+    const constraintsArg = args[1];
+    if (!Node4.isObjectLiteralExpression(constraintsArg)) {
+      return [];
+    }
+    const results = [];
+    for (const property of constraintsArg.getProperties()) {
+      if (!Node4.isPropertyAssignment(property)) {
+        continue;
+      }
+      const messageType = property.getName();
+      const initializer = property.getInitializer();
+      if (!initializer || !Node4.isObjectLiteralExpression(initializer)) {
+        continue;
+      }
+      let requires;
+      let ensures;
+      let message;
+      for (const constraintProp of initializer.getProperties()) {
+        if (!Node4.isPropertyAssignment(constraintProp)) {
+          continue;
+        }
+        const propName = constraintProp.getName();
+        const propValue = constraintProp.getInitializer();
+        if (!propValue) {
+          continue;
+        }
+        if (propName === "requires" && Node4.isStringLiteral(propValue)) {
+          requires = propValue.getLiteralValue();
+        } else if (propName === "ensures" && Node4.isStringLiteral(propValue)) {
+          ensures = propValue.getLiteralValue();
+        } else if (propName === "message" && Node4.isStringLiteral(propValue)) {
+          message = propValue.getLiteralValue();
+        }
+      }
+      results.push({
+        field,
+        messageType,
+        requires,
+        ensures,
+        message,
+        location: {
+          file: filePath,
+          line: property.getStartLineNumber()
+        }
+      });
+    }
+    return results;
+  }
 }
 
 // tools/analysis/src/extract/integrations.ts
@@ -4717,4 +4799,4 @@ main().catch((_error) => {
   process.exit(1);
 });
 
-//# debugId=C29862DDDCF6035364756E2164756E21
+//# debugId=4E2F22D76354977564756E2164756E21