@fairfox/polly 0.18.0 → 0.19.0

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

@@ -2212,7 +2212,9 @@ class HandlerExtractor {
       return;
     if (this.tryExtractSetConstructorPattern(fieldPath, right, assignments))
       return;
-    this.tryExtractMapConstructorPattern(fieldPath, right, assignments);
+    if (this.tryExtractMapConstructorPattern(fieldPath, right, assignments))
+      return;
+    this.tryExtractSignalDirectValuePattern(fieldPath, right, assignments);
   }
   tryExtractStateFieldPattern(fieldPath, right, assignments) {
     if (!fieldPath.startsWith("state."))
@@ -2331,6 +2333,25 @@ class HandlerExtractor {
     }
     return true;
   }
+  tryExtractSignalDirectValuePattern(fieldPath, right, assignments) {
+    if (!fieldPath.endsWith(".value"))
+      return false;
+    const signalName = fieldPath.slice(0, -6);
+    const literalValue = this.extractValue(right);
+    if (literalValue !== undefined) {
+      assignments.push({ field: signalName, value: literalValue });
+      return true;
+    }
+    if (Node4.isPropertyAccessExpression(right)) {
+      const rightPath = this.getPropertyPath(right);
+      const parts = rightPath.split(".");
+      if (parts.length === 2 && parts[0] !== undefined && parts[1] !== undefined && this.currentFunctionParams.includes(parts[0])) {
+        assignments.push({ field: signalName, value: `param:${parts[1]}` });
+        return true;
+      }
+    }
+    return false;
+  }
   extractSetOperation(newExpr, fieldPath, signalName) {
     const args = newExpr.getArguments();
     if (args.length === 0) {
@@ -2447,11 +2468,11 @@ class HandlerExtractor {
       return null;
     switch (methodName) {
       case "filter":
-        return { field: signalName, value: "SelectSeq(@, LAMBDA t: TRUE)" };
+        return { field: signalName, value: "NDET:FILTER" };
       case "map":
-        return { field: signalName, value: "[i \\in DOMAIN @ |-> @[i]]" };
+        return { field: signalName, value: "NDET:MAP" };
       case "slice":
-        return { field: signalName, value: "SubSeq(@, 1, Len(@))" };
+        return { field: signalName, value: "NDET:FILTER" };
       case "concat":
         return { field: signalName, value: "@ \\o <<payload>>" };
       case "reverse":
@@ -6052,4 +6073,4 @@ main().catch((_error) => {
   process.exit(1);
 });
 
-//# debugId=345DE519CB00B03D64756E2164756E21
+//# debugId=D42055AB70D110FF64756E2164756E21