@fairfox/polly 0.33.0 → 0.34.0

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

@@ -2583,15 +2583,7 @@ class HandlerExtractor {
   }
   extractPropertyAssignment(prop, assignments, signalName) {
     if (Node4.isPropertyAssignment(prop)) {
-      const name = prop.getName();
-      const initializer = prop.getInitializer();
-      if (!name || !initializer)
-        return;
-      const value = this.extractValue(initializer);
-      if (value === undefined)
-        return;
-      const field = signalName ? `${signalName}_${name}` : name;
-      assignments.push({ field, value });
+      this.extractRegularPropertyAssignment(prop, assignments, signalName);
       return;
     }
     if (Node4.isShorthandPropertyAssignment(prop)) {
@@ -2604,6 +2596,37 @@ class HandlerExtractor {
       }
     }
   }
+  extractRegularPropertyAssignment(prop, assignments, signalName) {
+    if (!Node4.isPropertyAssignment(prop))
+      return;
+    const name = prop.getName();
+    const initializer = prop.getInitializer();
+    if (!name || !initializer)
+      return;
+    const field = signalName ? `${signalName}_${name}` : name;
+    const value = this.extractValue(initializer);
+    if (value !== undefined) {
+      assignments.push({ field, value });
+      return;
+    }
+    const paramName = this.extractPayloadPropertyParam(initializer);
+    if (paramName !== null) {
+      assignments.push({ field, value: `param:${paramName}` });
+    }
+  }
+  extractPayloadPropertyParam(initializer) {
+    if (!Node4.isPropertyAccessExpression(initializer))
+      return null;
+    const parts = this.getPropertyPath(initializer).split(".");
+    if (parts.length !== 2)
+      return null;
+    const [paramName, fieldName] = parts;
+    if (paramName === undefined || fieldName === undefined)
+      return null;
+    if (!this.currentFunctionParams.includes(paramName))
+      return null;
+    return fieldName;
+  }
   extractElementAccessAssignment(left, right, assignments) {
     if (!Node4.isElementAccessExpression(left))
       return;
@@ -6212,4 +6235,4 @@ main().catch((_error) => {
   process.exit(1);
 });
 
-//# debugId=84905818D609523864756E2164756E21
+//# debugId=ED3EBA8D328D579464756E2164756E21