parseman 0.26.1 → 0.26.3

package/dist/plugin/index.js

@@ -5530,7 +5530,7 @@ function emitDispatch(p, ctx, pos) {
         ctx.indent = 1;
         ctx.failLabel = "_pfail";
         ctx.recordFail = true;
-        const innerR = emit(innerParser, ctx, "_pos");
+        const innerR = emitDispatch(innerParser, ctx, "_pos");
         ctx.indent = savedIndent;
         ctx.failLabel = savedFailLabel;
         ctx.recordFail = savedRecord;
@@ -6247,6 +6247,23 @@ function escapeClassChar(ch) {
   return ch.replace(/[\\\]^-]/g, "\\$&");
 }
 
+// src/combinators/withCtx.ts
+function withCtx(extra, combinator) {
+  const meta = {
+    firstSet: combinator._meta.firstSet,
+    canMatchNewline: combinator._meta.canMatchNewline,
+    isTrivia: false
+  };
+  return {
+    _tag: "withCtx",
+    _meta: meta,
+    _def: { tag: "withCtx", extra, parser: combinator },
+    parse(input, pos, ctx) {
+      return combinator.parse(input, pos, { ...ctx, state: extra });
+    }
+  };
+}
+
 // src/compiler/ir-serialize.ts
 function childrenOf2(def) {
   switch (def.tag) {
@@ -6263,6 +6280,7 @@ function childrenOf2(def) {
     case "field":
     case "not":
     case "node":
+    case "withCtx":
     case "expect":
       return [def.parser];
     case "grammar":
@@ -6278,7 +6296,6 @@ function childrenOf2(def) {
     case "regex":
     case "keywords":
     case "guard":
-    case "withCtx":
     case "recover":
     case "unknown":
       return [];
@@ -6325,6 +6342,45 @@ function evalRuleMapIR(ir) {
     n._def.buildSrc = src;
     return n;
   };
+  const _gch = (items) => {
+    const gateSrcs = [];
+    const arms = items.map((it) => {
+      if (Array.isArray(it)) {
+        const [src, comb] = it;
+        gateSrcs.push(src);
+        let gate;
+        try {
+          gate = (0, eval)(`(${src})`);
+        } catch {
+          gate = () => {
+            throw new Error("IR gate fn not materialized");
+          };
+        }
+        return { gate, combinator: comb };
+      }
+      gateSrcs.push(null);
+      return it;
+    });
+    const c = choice(...arms);
+    const gates = c._def.gates;
+    if (gates && gates.length !== gateSrcs.length) {
+      throw new Error(`_gch: gateSrcs length ${gateSrcs.length} != choice gates length ${gates.length}`);
+    }
+    ;
+    c._def.gateSrcs = gateSrcs;
+    return c;
+  };
+  const _wc = (src, inner) => {
+    let extra;
+    try {
+      extra = (0, eval)(`(${src})`);
+    } catch {
+      extra = void 0;
+    }
+    const w = withCtx(extra, inner);
+    w._def.extraSrc = src;
+    return w;
+  };
   const fn = new Function(
     "rules",
     "ref",
@@ -6350,6 +6406,8 @@ function evalRuleMapIR(ir) {
     "expect",
     "_tf",
     "_nd",
+    "_gch",
+    "_wc",
     `return (${ir})`
   );
   const map = fn(
@@ -6376,7 +6434,9 @@ function evalRuleMapIR(ir) {
     field,
     expect,
     _tf,
-    _nd
+    _nd,
+    _gch,
+    _wc
   );
   return Object.entries(map);
 }
@@ -6432,7 +6492,7 @@ ${body}
       this.analyze(target, active);
       return;
     }
-    if (def.tag === "guard" || def.tag === "withCtx" || def.tag === "recover" || def.tag === "unknown") {
+    if (def.tag === "guard" || def.tag === "recover" || def.tag === "unknown") {
       throw new Unserializable(`unsupported tag ${def.tag}`);
     }
     this.counts.set(c, (this.counts.get(c) ?? 0) + 1);
@@ -6488,8 +6548,16 @@ ${body}
       case "sequence":
         return `sequence(${def.parsers.map(kid).join(", ")})`;
       case "choice": {
-        if (def.gates.some((g) => g !== null)) throw new Unserializable("choice with a gate() has no source");
-        return `choice(${def.parsers.map(kid).join(", ")})`;
+        const gates = def.gates;
+        if (gates.every((g) => g === null)) return `choice(${def.parsers.map(kid).join(", ")})`;
+        const gateSrcs = def.gateSrcs;
+        const items = def.parsers.map((p, i) => {
+          if (gates[i] === null) return kid(p);
+          const src = gateSrcs?.[i];
+          if (src == null) throw new Unserializable("choice gate() has no captured source");
+          return `[${JSON.stringify(src)}, ${kid(p)}]`;
+        });
+        return `_gch([${items.join(", ")}])`;
       }
       case "many":
         return `many(${kid(def.parser)})`;
@@ -6533,8 +6601,11 @@ ${body}
         const trivia2 = def.clearTrivia ? "null" : def.triviaParser ? kid(def.triviaParser) : "undefined";
         return `parser({ trivia: ${trivia2}${def.trackLines ? ", trackLines: true" : ""} }, ${kid(def.parser)})`;
       }
+      case "withCtx": {
+        if (def.extraSrc === void 0) throw new Unserializable("withCtx without extraSrc");
+        return `_wc(${JSON.stringify(def.extraSrc)}, ${kid(def.parser)})`;
+      }
       case "guard":
-      case "withCtx":
       case "recover":
       case "unknown":
         throw new Unserializable(`unsupported tag ${def.tag}`);
@@ -6646,23 +6717,6 @@ function guard(predicate) {
   };
 }
 
-// src/combinators/withCtx.ts
-function withCtx(extra, combinator) {
-  const meta = {
-    firstSet: combinator._meta.firstSet,
-    canMatchNewline: combinator._meta.canMatchNewline,
-    isTrivia: false
-  };
-  return {
-    _tag: "withCtx",
-    _meta: meta,
-    _def: { tag: "withCtx", extra, parser: combinator },
-    parse(input, pos, ctx) {
-      return combinator.parse(input, pos, { ...ctx, state: extra });
-    }
-  };
-}
-
 // src/cst/offset-model.ts
 var { isFinite } = Number;
 
@@ -6670,6 +6724,48 @@ var { isFinite } = Number;
 registerRegexAnalyzer(firstSetFromRegex);
 
 // src/plugin/evaluator.ts
+function stripTsFromSource(node2, code) {
+  const cuts = [];
+  const walk2 = (n) => {
+    if (!n || typeof n !== "object") return;
+    const rec = n;
+    if (typeof rec.type === "string" && rec.type.startsWith("TS") && typeof rec.start === "number" && typeof rec.end === "number") {
+      const ex = rec.expression;
+      if ((rec.type === "TSAsExpression" || rec.type === "TSSatisfiesExpression" || rec.type === "TSNonNullExpression" || rec.type === "TSInstantiationExpression") && ex && typeof ex.end === "number") {
+        cuts.push([ex.end, rec.end]);
+        walk2(rec.expression);
+        return;
+      }
+      if (rec.type === "TSTypeAssertion" && ex && typeof ex.start === "number") {
+        cuts.push([rec.start, ex.start]);
+        walk2(rec.expression);
+        return;
+      }
+      cuts.push([rec.start, rec.end]);
+      return;
+    }
+    for (const key of Object.keys(rec)) {
+      if (key === "type" || key === "start" || key === "end") continue;
+      const v2 = rec[key];
+      if (Array.isArray(v2)) {
+        for (const item of v2) walk2(item);
+      } else if (v2 && typeof v2 === "object") walk2(v2);
+    }
+  };
+  walk2(node2);
+  const start = node2.start;
+  const end = node2.end;
+  if (cuts.length === 0) return code.slice(start, end);
+  cuts.sort((a, b) => a[0] - b[0]);
+  let out = "";
+  let cur = start;
+  for (const [s, e] of cuts) {
+    if (s < cur) continue;
+    out += code.slice(cur, s);
+    cur = e;
+  }
+  return out + code.slice(cur, end);
+}
 function isWordFactory(v2) {
   return !!v2 && typeof v2 === "object" && v2.tag === "wordFactory";
 }
@@ -6945,7 +7041,7 @@ function exprToCombi(node2, scope, code, mfs) {
       if (isGatedArmExpr(argNode)) {
         const parts = gatedArmParts(argNode);
         if (!parts) return null;
-        const gateSrc = code.slice(parts.gate.start, parts.gate.end);
+        const gateSrc = stripTsFromSource(parts.gate, code);
         mfs.push(gateSrc);
         const combi = anyValue(parts.combinator, scope, code, mfs);
         if (!isCombinator(combi)) return null;