parseman 0.9.0 → 0.10.0

package/dist/plugin/index.js

@@ -1099,24 +1099,70 @@ function not(combinator) {
   };
 }
 
+// src/compiler/build-arity.ts
+var PARAM_LIST_RE = (
+  // (params) => ...   |   function name?(params) ...   |   single-ident arrow `x => ...`
+  /^(?:function\b[^(]*\(([^)]*)\)|\(([^)]*)\)\s*=>|([A-Za-z_$][\w$]*)\s*=>)/
+);
+var CONFIRMABLE_PARAM_RE = /^[A-Za-z_$][\w$]*\s*\??\s*(?::[^,=]+)?$/;
+function confirmedBuildArity(src) {
+  const s = src.trim();
+  const m = PARAM_LIST_RE.exec(s);
+  if (!m) return null;
+  if (m[3] !== void 0) return 1;
+  const inner = (m[1] ?? m[2] ?? "").trim();
+  if (inner === "") return 0;
+  const parts = inner.split(",");
+  for (const part of parts) {
+    if (!CONFIRMABLE_PARAM_RE.test(part.trim())) return null;
+  }
+  if (/\barguments\b/.test(s)) return null;
+  return parts.length;
+}
+function buildReadsTrivia(def) {
+  const src = def.buildSrc ?? def.build.toString();
+  const arity = confirmedBuildArity(src);
+  if (arity === null) return true;
+  return arity >= 4;
+}
+function buildReadsState(def) {
+  const src = def.buildSrc ?? def.build.toString();
+  const arity = confirmedBuildArity(src);
+  if (arity === null) return true;
+  return arity >= 5;
+}
+
 // src/combinators/node.ts
-function node(type, combinator, build) {
+function collapseChild(child) {
+  return child !== null && typeof child === "object" && child._tag === "leaf" ? child.value : child;
+}
+function node(type, combinator, build, opts) {
   const meta = {
     firstSet: combinator._meta.firstSet,
     canMatchNewline: combinator._meta.canMatchNewline,
     isTrivia: false
   };
+  const collapse = opts?.collapse === true;
+  const def = collapse ? { tag: "node", type, parser: combinator, build, collapse: true } : { tag: "node", type, parser: combinator, build };
+  const capturesTrivia = buildReadsTrivia(def);
+  const clonesState = buildReadsState(def);
   return {
     _tag: "node",
     _meta: meta,
-    _def: { tag: "node", type, parser: combinator, build },
+    _def: def,
     parse(input, pos, ctx) {
       const saved = beginCstNodeCapture(ctx);
+      if (!capturesTrivia) ctx.captureTrivia = false;
       const r = combinator.parse(input, pos, ctx);
       const { children, rawChildren, triviaLog } = endCstNodeCapture(ctx, saved);
       if (!r.ok) return r;
-      const state = ctx.state !== void 0 ? Object.assign({}, ctx.state) : void 0;
-      const built = build(children, rawChildren, r.span, triviaLog, state);
+      const built = collapse && children.length === 1 ? collapseChild(children[0]) : build(
+        children,
+        rawChildren,
+        r.span,
+        triviaLog,
+        clonesState && ctx.state !== void 0 ? Object.assign({}, ctx.state) : void 0
+      );
       const isNodeLike = typeof built === "object" && built !== null && built._tag === "node";
       const rawEntry = isNodeLike ? built : { _tag: "leaf", value: typeof built === "string" ? built : "", span: r.span };
       if (saved.buf !== void 0 || saved.ch !== void 0) {
@@ -2519,25 +2565,34 @@ function emitNode(def, ctx, pos) {
     ctx.buildSrcs.push(def.buildSrc ?? null);
   }
   const i = ind(ctx);
+  const capturesTrivia = mkType !== null || buildReadsTrivia(def);
+  const clonesState = buildReadsState(def);
   const chV = v(ctx, "_ch");
   const rawV = v(ctx, "_raw");
-  const tlV = v(ctx, "_tl");
+  const tlV = capturesTrivia ? v(ctx, "_tl") : "_EMPTY_TL";
+  if (!capturesTrivia) ctx.needsEmptyTl = true;
   const sc = v(ctx, "_sc"), sl = v(ctx, "_sl"), sr = v(ctx, "_sr"), st = v(ctx, "_st"), stl = v(ctx, "_stl");
+  const allocStmt = capturesTrivia ? `${i}const ${chV} = [], ${rawV} = [], ${tlV} = []` : `${i}const ${chV} = [], ${rawV} = []`;
+  const innerTl = capturesTrivia ? tlV : "undefined";
   const stmts = [
-    `${i}const ${chV} = [], ${rawV} = [], ${tlV} = []`,
+    allocStmt,
     `${i}const ${sc} = _ctx._cstChildren, ${sl} = _ctx._cstLeaves, ${sr} = _ctx._cstRawChildren, ${st} = _ctx.captureTrivia, ${stl} = _ctx._cstTriviaLog`,
-    `${i}_ctx._cstChildren = ${chV}; _ctx._cstLeaves = ${chV}; _ctx._cstRawChildren = ${rawV}; _ctx.captureTrivia = true; _ctx._cstTriviaLog = ${tlV}`
+    `${i}_ctx._cstChildren = ${chV}; _ctx._cstLeaves = ${chV}; _ctx._cstRawChildren = ${rawV}; _ctx.captureTrivia = true; _ctx._cstTriviaLog = ${innerTl}`
   ];
   const { stmts: innerStmts, okVar, endVar } = emitFallible(def.parser, ctx, pos);
   stmts.push(...innerStmts);
   stmts.push(`${i}_ctx._cstChildren = ${sc}; _ctx._cstLeaves = ${sl}; _ctx._cstRawChildren = ${sr}; _ctx.captureTrivia = ${st}; _ctx._cstTriviaLog = ${stl}`);
   stmts.push(...emitIfFail(ctx, `!${okVar}`, failBody(ctx, '"node"', pos)));
-  const stV = v(ctx, "_nst");
-  stmts.push(`${i}const ${stV} = _ctx.state !== undefined ? Object.assign({}, _ctx.state) : undefined`);
+  let stV = "undefined";
+  if (clonesState) {
+    stV = v(ctx, "_nst");
+    stmts.push(`${i}const ${stV} = _ctx.state !== undefined ? Object.assign({}, _ctx.state) : undefined`);
+  }
   const ndV = v(ctx, "_nd");
   const ndExpr = mkType ? emitInlineMkNodeExpr(mkType, chV, rawV, pos, endVar, tlV) : `_build[${buildIdx}](${chV}, ${rawV}, { start: ${pos}, end: ${endVar} }, ${tlV}, ${stV})`;
+  const finalExpr = def.collapse ? `${chV}.length === 1 ? (${chV}[0] !== null && typeof ${chV}[0] === 'object' && ${chV}[0]._tag === 'leaf' ? ${chV}[0].value : ${chV}[0]) : (${ndExpr})` : ndExpr;
   stmts.push(
-    `${i}const ${ndV} = ${ndExpr}`,
+    `${i}const ${ndV} = ${finalExpr}`,
     `${i}if (${sc}) ${sc}.push(${ndV})`,
     `${i}if (${sr}) ${sr}.push((typeof ${ndV} === 'object' && ${ndV} !== null && ${ndV}._tag === 'node') ? ${ndV} : { _tag: 'leaf', value: typeof ${ndV} === 'string' ? ${ndV} : '', span: { start: ${pos}, end: ${endVar} } })`
   );
@@ -2929,7 +2984,9 @@ function compile(combinator, mapFnSources) {
   const collatorDecl = ctx.needsCollator ? `const _collator = new Intl.Collator(undefined, { sensitivity: 'accent' })
 ` : "";
   const namedPrelude = ctx.namedFnDecls.length > 0 ? [...namedFnPrelude(), ""] : [];
+  const emptyTlDecls = ctx.needsEmptyTl ? ["const _EMPTY_TL = Object.freeze([])"] : [];
   const source = [
+    ...emptyTlDecls,
     ...ctx.regexDecls,
     "",
     ...namedPrelude,
@@ -2941,6 +2998,7 @@ function compile(combinator, mapFnSources) {
     `}`
   ].join("\n");
   const fn = new Function("input", "_pos", "_rp", "_mf", "_build", "_ctx", [
+    ...emptyTlDecls,
     ...ctx.regexDecls,
     collatorDecl,
     ...namedPrelude,
@@ -3013,6 +3071,7 @@ function compileRuleMap(ruleMap) {
   const buildDecl = buildSources?.length ? `  const _build = [${buildSources.join(", ")}]` : "";
   const namedPrelude = ctx.namedFnDecls.length > 0 ? namedFnPrelude() : [];
   const hoistedDecls = [
+    ctx.needsEmptyTl ? `  const _EMPTY_TL = Object.freeze([])` : "",
     ...ctx.regexDecls.map((d) => `  ${d}`),
     collatorDecl ? `  ${collatorDecl.trim()}` : "",
     mfDecl,
@@ -3054,10 +3113,12 @@ function buildInlineExpression(ctx, r, collatorDecl, mapFnSources, buildSources)
   ].join("\n");
   const mfDecl = mapFnSources?.length ? `  const _mf = [${mapFnSources.join(", ")}]` : "";
   const buildDecl = buildSources?.length ? `  const _build = [${buildSources.join(", ")}]` : "";
-  const needsWrapper = ctx.regexDecls.length > 0 || !!collatorDecl || ctx.namedFnDecls.length > 0 || !!mfDecl || !!buildDecl;
+  const emptyTlDecl = ctx.needsEmptyTl ? `  const _EMPTY_TL = Object.freeze([])` : "";
+  const needsWrapper = ctx.regexDecls.length > 0 || !!collatorDecl || ctx.namedFnDecls.length > 0 || !!mfDecl || !!buildDecl || !!emptyTlDecl;
   if (!needsWrapper) return innerFn;
   const namedPrelude = ctx.namedFnDecls.length > 0 ? namedFnPrelude() : [];
   const hoistedDecls = [
+    emptyTlDecl,
     ...ctx.regexDecls.map((d) => `  ${d}`),
     collatorDecl ? `  ${collatorDecl.trim()}` : "",
     mfDecl,
@@ -3258,6 +3319,20 @@ function scopeGet(scope, name, mfs) {
   if (isCombinator(entry)) return entry;
   return null;
 }
+function staticCollapseOption(expr) {
+  if (expr.type !== "ObjectExpression") return false;
+  for (const prop of expr.properties) {
+    const p = prop;
+    if (p.computed) continue;
+    const key = p.key;
+    const name = key.type === "Identifier" ? key.name : key.type === "Literal" ? String(key.value) : void 0;
+    if (name === "collapse") {
+      const val = p.value;
+      return (val.type === "Literal" || val.type === "BooleanLiteral") && val.value === true;
+    }
+  }
+  return false;
+}
 function exprToCombi(node2, scope, code, mfs) {
   if (node2.type === "Identifier") return scopeGet(scope, node2.name, mfs);
   if (node2.type !== "CallExpression") return null;
@@ -3306,15 +3381,16 @@ function exprToCombi(node2, scope, code, mfs) {
     }
   }
   if (callee.name === "node" && code !== void 0) {
-    const [typeArg, parserArg, buildArg] = node2.arguments;
+    const [typeArg, parserArg, buildArg, optsArg] = node2.arguments;
     if (!typeArg || !parserArg || !buildArg || typeArg.type === "SpreadElement" || parserArg.type === "SpreadElement" || buildArg.type === "SpreadElement") return null;
     const typeVal = anyValue(typeArg, scope, code);
     if (typeof typeVal !== "string") return null;
     const inner = anyValue(parserArg, scope, code, mfs);
     if (!isCombinator(inner)) return null;
     const buildSrc = code.slice(buildArg.start, buildArg.end);
+    const collapse = optsArg !== void 0 && optsArg.type !== "SpreadElement" ? staticCollapseOption(optsArg) : false;
     try {
-      const combi = node(typeVal, inner, () => null);
+      const combi = node(typeVal, inner, () => null, collapse ? { collapse: true } : void 0);
       if (combi._def.tag === "node") combi._def.buildSrc = buildSrc;
       return combi;
     } catch {