@suss/extractor 0.8.0 → 0.10.0

package/dist/index.cjs

@@ -47,7 +47,7 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 var import_node_crypto = require("crypto");
-var import_behavioral_ir = require("@suss/behavioral-ir");
+var import_behavioral_ir2 = require("@suss/behavioral-ir");
 
 // src/inputReads.ts
 function inputReadsOf(args) {
@@ -167,6 +167,7 @@ function unwrapJsonStringify(body) {
 }
 
 // src/paths/enumeratePaths.ts
+var import_behavioral_ir = require("@suss/behavioral-ir");
 var MAX_PATHS = 256;
 var UnmodeledFlow = class extends Error {
 };
@@ -210,7 +211,7 @@ var loopCompletedCond = (headerText, exit) => fixedCond(
   exit === "throw" ? "earlyThrow" : "earlyReturn"
 );
 var catchEntryCond = () => fixedCond(
-  { sourceText: "catch", expression: null },
+  { sourceText: import_behavioral_ir.CATCH_ENTRY_TEXT, expression: null },
   "positive",
   "catchBlock"
 );
@@ -220,7 +221,7 @@ function classify(path, home, isAncestorOrSelf) {
       return cond.info;
     }
     const encloses = home !== null && isAncestorOrSelf(cond.branchStmt, home);
-    if (encloses || cond.info.polarity === "positive") {
+    if (encloses || cond.info.polarity === "positive" || cond.oppositeExit === null) {
       return { ...cond.info, source: "explicit" };
     }
     const source = cond.oppositeExit === "throw" ? "earlyThrow" : "earlyReturn";
@@ -228,6 +229,9 @@ function classify(path, home, isAncestorOrSelf) {
   });
 }
 function childrenOf(stmt) {
+  return [...ownBlocksOf(stmt), ...(stmt.callbacks ?? []).flat()];
+}
+function ownBlocksOf(stmt) {
   if (stmt.kind === "if") {
     return stmt.elseBody === null ? stmt.thenBody : [...stmt.thenBody, ...stmt.elseBody];
   }
@@ -254,7 +258,7 @@ function hasStrayBreak(nodes) {
     if (node.kind === "switch" || node.kind === "loop") {
       return false;
     }
-    return hasStrayBreak(childrenOf(node));
+    return hasStrayBreak(ownBlocksOf(node));
   });
 }
 function exitKindOfList(stmts) {
@@ -291,11 +295,6 @@ function containsTerminal(ctx, stmt) {
   return childrenOf(stmt).some((child) => containsTerminal(ctx, child));
 }
 function stepIf(stmt, path, ctx) {
-  const armsExit = stmt.exitKind !== null;
-  const armsHaveTerminals = containsTerminal(ctx, stmt);
-  if (!armsExit && !armsHaveTerminals) {
-    return [path];
-  }
   const thenExit = exitKindOfList(stmt.thenBody);
   const elseExit = stmt.elseBody === null ? null : exitKindOfList(stmt.elseBody);
   const thenPath = [
@@ -304,7 +303,12 @@ function stepIf(stmt, path, ctx) {
   ];
   const elsePath = [
     ...path,
-    branchCond(stmt, stmt.condition, "negative", thenExit)
+    branchCond(
+      stmt,
+      stmt.condition,
+      "negative",
+      stmt.elseBody === null ? thenExit : null
+    )
   ];
   const out = [];
   out.push(...enumerate(ctx, stmt.thenBody, thenPath));
@@ -376,7 +380,11 @@ function stepTry(stmt, path, ctx) {
   }
   return out;
 }
-function stepExit(_stmt, _path, _ctx) {
+function stepExit(stmt, path, ctx) {
+  const collector = ctx.state.callbackExits[ctx.state.callbackExits.length - 1];
+  if (collector !== void 0 && (stmt.exit === "return" || stmt.exit === "throw")) {
+    collector.push(path);
+  }
   return [];
 }
 function stepOpaque(_stmt, path, _ctx) {
@@ -390,16 +398,36 @@ var stepHandlers = {
   exit: stepExit,
   opaque: stepOpaque
 };
+function stepCallback(ctx, body, path) {
+  const exits = [];
+  ctx.state.callbackExits.push(exits);
+  let fellThrough;
+  try {
+    fellThrough = enumerate(ctx, body, path);
+  } finally {
+    ctx.state.callbackExits.pop();
+  }
+  const out = [...fellThrough, ...exits];
+  return out.length === 0 ? [path] : out;
+}
 function stepStatement(ctx, stmt, path) {
+  let paths = [path];
+  for (const body of stmt.callbacks ?? []) {
+    paths = paths.flatMap((each) => stepCallback(ctx, body, each));
+  }
   const ownTerminals = ctx.terminalsByStmt.get(stmt) ?? [];
-  for (const terminal of ownTerminals) {
-    recordTerminal(ctx, terminal, stmt, path);
+  for (const each of withRejoined(paths)) {
+    for (const terminal of ownTerminals) {
+      recordTerminal(ctx, terminal, stmt, each);
+    }
   }
   const handler = stepHandlers[stmt.kind];
-  return handler(
-    stmt,
-    path,
-    ctx
+  return paths.flatMap(
+    (each) => handler(
+      stmt,
+      each,
+      ctx
+    )
   );
 }
 function soleDisagreement(left, right) {
@@ -456,14 +484,18 @@ function mergeRejoined(paths) {
   }
   return current;
 }
+function withRejoined(paths) {
+  const merged = mergeRejoined(paths);
+  return merged.length === paths.length ? paths : [...merged, ...paths];
+}
 function enumerate(ctx, stmts, prefix) {
   let frontiers = [[...prefix]];
   for (const stmt of stmts) {
     const nextFrontiers = [];
-    for (const path of frontiers) {
+    for (const path of mergeRejoined(frontiers)) {
       nextFrontiers.push(...stepStatement(ctx, stmt, path));
     }
-    frontiers = mergeRejoined(nextFrontiers);
+    frontiers = nextFrontiers;
     if (frontiers.length > MAX_PATHS) {
       throw new PathBudgetExceeded(
         `path budget exceeded, more than ${MAX_PATHS} paths`
@@ -503,14 +535,15 @@ function enumerateStructuredPaths(input) {
   const state = {
     byTerminal: /* @__PURE__ */ new Map(),
     fallthrough: [],
-    pathCount: 0
+    pathCount: 0,
+    callbackExits: []
   };
   const ctx = {
     terminalsByStmt: input.terminalsByStmt,
     state,
     isAncestorOrSelf
   };
-  const continuations = enumerate(ctx, input.statements, []);
+  const continuations = withRejoined(enumerate(ctx, input.statements, []));
   for (const path of continuations) {
     chargeBudget(state);
     state.fallthrough.push(classify(path, null, isAncestorOrSelf));
@@ -803,11 +836,16 @@ function buildMetadata(raw) {
   let metadata = {};
   const http = buildHttpMetadataValue(raw);
   if (http !== null) {
-    metadata = (0, import_behavioral_ir.withHttpMetadata)(metadata, http);
+    metadata = (0, import_behavioral_ir2.withHttpMetadata)(metadata, http);
   }
   const graphql = buildGraphqlMetadataValue(raw);
   if (graphql !== null) {
-    metadata = (0, import_behavioral_ir.withGraphqlMetadata)(metadata, graphql);
+    metadata = (0, import_behavioral_ir2.withGraphqlMetadata)(metadata, graphql);
+  }
+  if (raw.sourceDocumentLabel !== void 0) {
+    metadata = (0, import_behavioral_ir2.withSourceDocumentMetadata)(metadata, {
+      label: raw.sourceDocumentLabel
+    });
   }
   return Object.keys(metadata).length > 0 ? metadata : null;
 }
@@ -826,6 +864,12 @@ function buildHttpMetadataValue(raw) {
   if (raw.statusAccessors !== void 0 && raw.statusAccessors.length > 0) {
     http.statusAccessors = raw.statusAccessors;
   }
+  if (raw.successAccessors !== void 0 && raw.successAccessors.length > 0) {
+    http.successAccessors = raw.successAccessors;
+  }
+  if (raw.failureDelivery !== void 0) {
+    http.failureDelivery = raw.failureDelivery;
+  }
   return Object.keys(http).length > 0 ? http : null;
 }
 function buildGraphqlMetadataValue(raw) {
@@ -848,7 +892,7 @@ function answersWithHttpResponses(raw) {
   if (raw.boundaryBinding === null || raw.boundaryBinding === void 0) {
     return true;
   }
-  return (0, import_behavioral_ir.exchangesHttpResponses)(raw.boundaryBinding);
+  return (0, import_behavioral_ir2.exchangesHttpResponses)(raw.boundaryBinding);
 }
 function detectGaps(raw, transitions, options) {
   if (options.gapHandling === "silent") {