@fairfox/polly 0.13.2 → 0.14.0

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

@@ -1874,6 +1874,39 @@ class HandlerExtractor {
       if (value !== undefined) {
         assignments.push({ field, value });
       }
+      return;
+    }
+    const valueMatch = fieldPath.match(/\.value\.(.+)$/);
+    if (valueMatch && valueMatch[1]) {
+      const field = valueMatch[1];
+      const value = this.extractValue(right);
+      if (value !== undefined) {
+        assignments.push({ field, value });
+      }
+      return;
+    }
+    if (fieldPath.endsWith(".value") && Node4.isObjectLiteralExpression(right)) {
+      this.extractObjectLiteralAssignments(right, assignments);
+    }
+  }
+  extractObjectLiteralAssignments(objectLiteral, assignments) {
+    if (!Node4.isObjectLiteralExpression(objectLiteral))
+      return;
+    for (const prop of objectLiteral.getProperties()) {
+      if (Node4.isPropertyAssignment(prop)) {
+        const name = prop.getName();
+        const initializer = prop.getInitializer();
+        if (!name || !initializer)
+          continue;
+        const value = this.extractValue(initializer);
+        if (value !== undefined) {
+          assignments.push({ field: name, value });
+        }
+      }
+      if (Node4.isShorthandPropertyAssignment(prop)) {
+        const name = prop.getName();
+        assignments.push({ field: name, value: "@" });
+      }
     }
   }
   extractElementAccessAssignment(left, right, assignments) {
@@ -1883,9 +1916,9 @@ class HandlerExtractor {
     if (!Node4.isPropertyAccessExpression(expr))
       return;
     const fieldPath = this.getPropertyPath(expr);
-    if (!fieldPath.startsWith("state."))
+    const field = this.extractFieldFromStatePath(fieldPath);
+    if (!field)
       return;
-    const field = fieldPath.substring(6);
     const indexExpr = left.getArgumentExpression();
     const index = indexExpr ? indexExpr.getText() : "0";
     const value = this.extractValue(right);
@@ -1897,6 +1930,16 @@ class HandlerExtractor {
       });
     }
   }
+  extractFieldFromStatePath(fieldPath) {
+    if (fieldPath.startsWith("state.")) {
+      return fieldPath.substring(6);
+    }
+    const valueMatch = fieldPath.match(/\.value\.(.+)$/);
+    if (valueMatch && valueMatch[1]) {
+      return valueMatch[1];
+    }
+    return null;
+  }
   extractCompoundAssignment(node, assignments) {
     if (!Node4.isBinaryExpression(node))
       return;
@@ -1905,8 +1948,8 @@ class HandlerExtractor {
     const right = node.getRight();
     if (Node4.isPropertyAccessExpression(left)) {
       const fieldPath = this.getPropertyPath(left);
-      if (fieldPath.startsWith("state.")) {
-        const field = fieldPath.substring(6);
+      const field = this.extractFieldFromStatePath(fieldPath);
+      if (field) {
         const rightValue = right.getText();
         const tlaOp = operator.slice(0, -1);
         assignments.push({
@@ -1940,8 +1983,8 @@ class HandlerExtractor {
     if (!Node4.isPropertyAccessExpression(operand))
       return;
     const fieldPath = this.getPropertyPath(operand);
-    if (fieldPath.startsWith("state.")) {
-      const field = fieldPath.substring(6);
+    const field = this.extractFieldFromStatePath(fieldPath);
+    if (field) {
       const value = operatorText === "++" ? "@ + 1" : "@ - 1";
       assignments.push({ field, value });
     }
@@ -1956,8 +1999,8 @@ class HandlerExtractor {
     const object = expr.getExpression();
     if (Node4.isPropertyAccessExpression(object)) {
       const fieldPath = this.getPropertyPath(object);
-      if (fieldPath.startsWith("state.")) {
-        const field = fieldPath.substring(6);
+      const field = this.extractFieldFromStatePath(fieldPath);
+      if (field) {
         const args = node.getArguments().map((arg) => arg.getText());
         const tlaValue = this.translateArrayMethod(methodName, args, fieldPath);
         if (tlaValue) {
@@ -2597,8 +2640,8 @@ class HandlerExtractor {
     if (!Node4.isPropertyAccessExpression(left) && !Node4.isElementAccessExpression(left))
       return;
     const fieldPath = Node4.isPropertyAccessExpression(left) ? this.getPropertyPath(left) : this.getPropertyPath(left.getExpression());
-    if (fieldPath.startsWith("state.")) {
-      const field = fieldPath.substring(6);
+    const field = this.extractFieldFromStatePath(fieldPath);
+    if (field) {
       const line = node.getStartLineNumber();
       const afterAwait = node.getStart() > firstAwaitPos;
       mutations.push({ field, line, afterAwait });
@@ -2615,10 +2658,8 @@ class HandlerExtractor {
     if (!Node4.isPropertyAccessExpression(object))
       return;
     const fieldPath = this.getPropertyPath(object);
-    if (!fieldPath.startsWith("state."))
-      return;
-    if (["push", "pop", "shift", "unshift", "splice"].includes(methodName)) {
-      const field = fieldPath.substring(6);
+    const field = this.extractFieldFromStatePath(fieldPath);
+    if (field && ["push", "pop", "shift", "unshift", "splice"].includes(methodName)) {
       const line = node.getStartLineNumber();
       const afterAwait = node.getStart() > firstAwaitPos;
       mutations.push({ field, line, afterAwait });
@@ -5019,4 +5060,4 @@ main().catch((_error) => {
   process.exit(1);
 });
 
-//# debugId=29B7FAEF737837BA64756E2164756E21
+//# debugId=7860F724A3DD29E164756E2164756E21