@fairfox/polly 0.36.0 → 0.38.0

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

@@ -401,6 +401,32 @@ async function generateTLA(config, analysis) {
   const generator = new TLAGenerator;
   return await generator.generate(config, analysis);
 }
+function applySubsystemBounds(messages, subsystem, handlerNames) {
+  if (messages.perMessageBounds) {
+    const filtered = {};
+    for (const [msg, bound] of Object.entries(messages.perMessageBounds)) {
+      if (handlerNames.has(msg)) {
+        filtered[msg] = bound;
+      }
+    }
+    messages.perMessageBounds = filtered;
+  }
+  const override = subsystem.bounds;
+  if (!override)
+    return;
+  if (override.maxInFlight !== undefined) {
+    messages.maxInFlight = override.maxInFlight;
+  }
+  if (override.perMessageBounds) {
+    const merged = { ...messages.perMessageBounds ?? {} };
+    for (const [msg, bound] of Object.entries(override.perMessageBounds)) {
+      if (handlerNames.has(msg)) {
+        merged[msg] = bound;
+      }
+    }
+    messages.perMessageBounds = merged;
+  }
+}
 async function generateSubsystemTLA(_subsystemName, subsystem, config, analysis) {
   const stateFields = new Set(subsystem.state);
   const handlerNames = new Set(subsystem.handlers);
@@ -415,15 +441,7 @@ async function generateSubsystemTLA(_subsystemName, subsystem, config, analysis)
     include: subsystem.handlers
   };
   filteredMessages.exclude = undefined;
-  if (filteredMessages.perMessageBounds) {
-    const filteredBounds = {};
-    for (const [msg, bound] of Object.entries(filteredMessages.perMessageBounds)) {
-      if (handlerNames.has(msg)) {
-        filteredBounds[msg] = bound;
-      }
-    }
-    filteredMessages.perMessageBounds = filteredBounds;
-  }
+  applySubsystemBounds(filteredMessages, subsystem, handlerNames);
   const filteredConfig = {
     ...config,
     state: filteredState,
@@ -5289,6 +5307,13 @@ class HandlerExtractor {
       assignments.push({ field, value });
       return;
     }
+    if (Node2.isIdentifier(initializer)) {
+      const paramName2 = initializer.getText();
+      if (this.currentFunctionParams.includes(paramName2)) {
+        assignments.push({ field, value: `param:${paramName2}` });
+        return;
+      }
+    }
     const paramName = this.extractPayloadPropertyParam(initializer);
     if (paramName !== null) {
       assignments.push({ field, value: `param:${paramName}` });
@@ -7717,4 +7742,4 @@ main().catch((error) => {
   process.exit(1);
 });
 
-//# debugId=F1333D1589EFA94964756E2164756E21
+//# debugId=93D3E9EFDD87727A64756E2164756E21