@marcomigozzi/trailhead 0.13.0 → 0.14.0

package/plugins/trailhead/hooks/lib/shell-scan.js

@@ -42,87 +42,512 @@ function stripQuotes(t) {
 
 // --- heredoc bodies ------------------------------------------------------------
 
-// Matches a heredoc introducer anywhere it can appear on a line: `<<TAG`,
-// `<<'TAG'`, `<<"TAG"`, and `<<-TAG` (which also allows the terminator line to
-// be indented with leading tabs). Captures the operator's `-` and the tag
-// (quoted or bare) separately.
-const HEREDOC_RE = /<<(-?)\s*(?:'([^']+)'|"([^"]+)"|([A-Za-z0-9_]+))/;
-
-// A heredoc BODY is literal data (e.g. a script being written to a file), not
-// shell statements executed in the current shell. Scan the command line by
-// line; whenever a line introduces a heredoc, skip every following line up to
-// and including the matching terminator line before it ever reaches the
-// statement splitter below. Handles multiple heredocs on the same command and
-// both quoted and unquoted tags. Best-effort: not a full shell parser.
-function stripHeredocBodies(cmd) {
+// What follows a `<<` introducer: an optional `-` (the terminator line may then
+// be indented with leading tabs), then the tag, quoted or bare. Captures the
+// `-` and the tag (single-quoted, double-quoted or bare) separately. Sticky, so
+// it only matches right at the operator.
+const HEREDOC_TAG_RE = /(-?)[ \t]*(?:'([^']+)'|"([^"]+)"|([A-Za-z0-9_]+))/y;
+
+// A `#` starts a comment, and a `((` an arithmetic command, only at a word start.
+const WORD_START_PREV_RE = /[\s;&|()]/;
+
+// The ONE lexer behind heredoc detection and statement splitting. It scans a
+// line with the shell context a bare regex lacks: a `<<` inside quotes, a
+// comment, `$((..))`, `$[..]`, `((..))` or after a backslash is not an
+// introducer, and `<<<` is a herestring. `st.stack` holds the open contexts as
+// uniform { k, ... } frames: k is 'sq', 'dq', 'ansi' (`$'..'`), 'br' (`${..}`),
+// 'sub' (`$(..)`, with a paren count n), 'ar' (an arithmetic span) or 'hd' (the
+// body of an unquoted heredoc, only ever the bottom frame of a dedicated scan,
+// see heredocSubstitutionBodies); `st.cont` is set when the line
+// ends in a backslash-newline continuation. It is mutated and carried to the next line, so a quote or arithmetic
+// span that runs across lines keeps suppressing introducers. A `$(..)` is code
+// again, even inside double quotes, so its heredocs count. Returns
+// [{ at, depth, dash, tag, quoted }] (`at` is the index of the `<<`, `depth` the
+// number of open contexts there). When `seps` is given, the statement separators found
+// in splittable code (top level or inside `$(..)`, never inside a quote,
+// arithmetic span or comment) are pushed to it as { at, len, op }, so the
+// splitters cut exactly where this lexer says code is. Best-effort: not a full
+// shell parser (`$(( (a) ))` nesting tricks and `<<\TAG` are not modelled,
+// nor is a backtick nested inside a backtick span other than as an escaped
+// char; nor is the body order of a heredoc nested in another heredoc's pending `"$(..)"`, e.g. `cat <<E "$(cat <<A ..)"`).
+// When `subs` is given ({ open(at, bq), close(at) }), it is told where each
+// OUTERMOST `$(..)` / `<(..)` / `>(..)` / backtick body starts (the index after
+// the opener; `bq` is true for a backtick) and where it ends (the index of the
+// closing `)` or backtick), so substitutionBodies reads the bodies off this
+// lexer instead of running a second one.
+// A backtick span is NOT a stack frame: it lives in `st.bq` ({ outer } or unset)
+// and is opened by an unescaped backtick in code, `dq`, `br` or `sub` context
+// (never in `sq`, `ansi`, arithmetic or a comment) and closed by the next
+// unescaped one. The span is OPAQUE: while it is open only a backslash (skips
+// the next char) and the closing backtick are looked at, as in the shell, which
+// finds the closing backtick first and parses the body separately (so a quote
+// inside it cannot desync this lexer). Its separators and heredocs therefore
+// do not count; the guards recurse into the body via substitutionBodies. It is
+// kept off the stack so heredoc depth is untouched. `st.bq.dq` records whether
+// it sits in double quotes (nearest `dq`/`sub` frame is `dq`), where the shell
+// also unescapes `\"` in the body. A span that never closes runs to the end of
+// the input.
+function scanLine(line, st, seps, subs) {
+  const found = [];
+  const stack = st.stack;
+  const parens = {}; // parenIsArith's per-line match cache
+  const cont0 = st.cont; // the previous line ended in a backslash: this one continues it
+  let escAt = -2; // index of the last escaped char
+  // Index of the last bare `$` (not escaped, not the 2nd of `$$`). A bare `$` that
+  // ended the previous line before its `\<nl>` is still adjacent: it counts as index -1.
+  let bareAt = cont0 && st.bareEnd ? -1 : -2;
+  st.cont = false;
+  st.bareEnd = false;
+  // Is the char before index i a word boundary? An escaped char (`\ #x`) is not.
+  // The `)` that closes an expansion (`$(..)`, `<(..)`, `$((..))`) ends no word
+  // either: `$(x)# y` is not a comment, `(x)# y` is.
+  let expCloseAt = -2;
+  // Push a `$(..)` / `<(..)` frame; the outermost one is reported to `subs`.
+  const openSub = (bodyAt) => {
+    const outer = !stack.some((f) => f.k === 'sub');
+    stack.push({ k: 'sub', n: 0, outer });
+    if (subs && outer) subs.open(bodyAt);
+  };
+  const isBoundary = (i) => WORD_START_PREV_RE.test(line[i - 1]) && escAt !== i - 1 && expCloseAt !== i - 1;
+  for (let i = 0; i < line.length; i++) {
+    const c = line[i];
+    if (st.bq) { // opaque backtick span: only a backslash and the closing backtick matter
+      if (c === '\\') i++;
+      else if (c === '`') { if (subs && st.bq.outer) subs.close(i); st.bq = null; }
+      continue;
+    }
+    const top = stack[stack.length - 1];
+    if (top && top.k === 'sq') { if (c === "'") stack.pop(); continue; }
+    if (top && top.k === 'ansi') { // $'..': a backslash escapes the next char, even a quote
+      if (c === '\\') i++;
+      else if (c === "'") stack.pop();
+      continue;
+    }
+    if (top && top.k === 'ar') {
+      // A `$(..)` inside arithmetic runs: push a sub frame (its own parens
+      // balance there), unless it is itself an arithmetic `$((..))`.
+      if (c === '$' && line[i + 1] === '(' && !(line[i + 2] === '(' && parenIsArith(line, i + 3, parens))) { openSub(i + 2); i++; continue; }
+      if (top.mode === 'bracket') { if (c === ']') stack.pop(); continue; }
+      if (c === '(') top.depth++;
+      else if (c === ')' && --top.depth === 0) { stack.pop(); if (top.exp) expCloseAt = i; }
+      continue;
+    }
+    if (c === '\\') { // escaped char stays literal
+      if (i + 1 >= line.length) { // backslash-newline: the line continues
+        st.cont = true;
+        // The next line is glued to this text: it starts a word only if the
+        // char before the backslash was a boundary (`a \<nl>#x` is a comment,
+        // `a\<nl>#x` is not). A line of just `\` inherits the earlier answer.
+        st.contBoundary = i === 0 ? (cont0 ? st.contBoundary : true) : isBoundary(i);
+        st.bareEnd = i === 0 ? bareAt === -1 : bareAt === i - 1; // carry the bare-dollar bit over
+        if (st.onCont) st.onCont(i, st.contBoundary, top); // optional observer (inWordContinuations)
+      }
+      i++;
+      escAt = i;
+      continue;
+    }
+    if (c === '`') { // a backtick substitution opens here (its close is handled above)
+      const outer = !stack.some((f) => f.k === 'sub');
+      const near = stack.slice().reverse().find((f) => f.k === 'dq' || f.k === 'sub');
+      const dq = !!near && near.k === 'dq';
+      st.bq = { outer, dq };
+      if (subs && outer) subs.open(i + 1, true, dq);
+      continue;
+    }
+    // Only a bare `$` opens `$'..'`, `${..}`, `$[..]`, `$(..)` or `$((..))`: not an escaped one
+    // (skipped above) and not the 2nd of `$$` (the PID, so `$${` is PID + a literal brace).
+    const bare = c === '$' && bareAt !== i - 1;
+    if (bare) bareAt = i;
+    if (bare && line[i + 1] === '(' && line[i + 2] === '(' && parenIsArith(line, i + 3, parens)) { stack.push({ k: 'ar', mode: 'paren', depth: 2, exp: true }); i += 2; continue; }
+    if (bare && line[i + 1] === '(') { openSub(i + 2); i++; continue; }
+    if (bare && line[i + 1] === '{') {
+      // `${ ` / `${<tab>` / `${|` / `${` at the end of a line is a bash 5.3 function
+      // substitution (it runs code); the optional observer `st.funsubs` records it.
+      const funsub = st.funsubs && (i + 2 >= line.length || ' \t|'.includes(line[i + 2]));
+      const entry = funsub ? { line: st.lineNo, at: i } : null;
+      if (entry) st.funsubs.push(entry);
+      stack.push(entry ? { k: 'br', fs: entry } : { k: 'br' });
+      i++;
+      continue;
+    }
+    if (top && top.k === 'dq') {
+      if (c === '"') stack.pop();
+      continue;
+    }
+    // An unquoted heredoc body is like double quotes without a closing `"`: the
+    // expansions above stay live, but quotes, comments, separators and
+    // introducers are literal text.
+    if (top && top.k === 'hd') continue;
+    if (bare && line[i + 1] === "'") { stack.push({ k: 'ansi' }); i++; continue; }
+    // A `'` glued to a bare `$` that ended the previous line (`$\<nl>'..'`) is ANSI-C too.
+    if (c === "'" && i === 0 && bareAt === -1) { stack.push({ k: 'ansi' }); continue; }
+    if (c === "'") { stack.push({ k: 'sq' }); continue; }
+    if (c === '"') { stack.push({ k: 'dq' }); continue; }
+    if (bare && line[i + 1] === '[') { stack.push({ k: 'ar', mode: 'bracket' }); i++; continue; }
+    if (top && top.k === 'br') { // ${..}: no code until the matching }
+      if (c === '}') {
+        stack.pop();
+        if (top.fs) { top.fs.endLine = st.lineNo; top.fs.endAt = i; }
+      }
+      continue;
+    }
+    const wordStart = i === 0 ? (cont0 ? st.contBoundary : true) : isBoundary(i);
+    if (c === '#' && wordStart) break; // comment to end of line
+    if (c === '(' && line[i + 1] === '(' && wordStart && parenIsArith(line, i + 2, parens)) { stack.push({ k: 'ar', mode: 'paren', depth: 2 }); i++; continue; }
+    if (top && top.k === 'sub') {
+      if (c === '(') { top.n++; continue; }
+      if (c === ')') { if (top.n === 0) { stack.pop(); expCloseAt = i; if (subs && top.outer) subs.close(i); } else top.n--; continue; }
+    }
+    // A process substitution `<(..)` / `>(..)` is code, like `$(..)`.
+    if ((c === '<' || c === '>') && line[i + 1] === '(') { openSub(i + 2); i++; continue; }
+    if (seps && splittable(stack)) {
+      const two = c + line[i + 1];
+      if (two === '&&' || two === '||') { seps.push({ at: i, len: 2, op: two }); i++; continue; }
+      if (c === ';' || c === '|') { seps.push({ at: i, len: 1, op: c }); continue; }
+      // A lone `&` backgrounds, unless it is part of a redirection (`2>&1`, `>&2`, `&>f`).
+      if (c === '&' && line[i - 1] !== '>' && line[i - 1] !== '<' && line[i + 1] !== '>') { seps.push({ at: i, len: 1, op: '&' }); continue; }
+    }
+    if (c === '<' && line[i + 1] === '<') {
+      if (line[i + 2] === '<') { i += 2; continue; } // herestring
+      HEREDOC_TAG_RE.lastIndex = i + 2;
+      const m = HEREDOC_TAG_RE.exec(line);
+      if (m) {
+        found.push({ at: i, depth: stack.length, dash: m[1] === '-', tag: m[2] || m[3] || m[4], quoted: m[2] !== undefined || m[3] !== undefined });
+        i = HEREDOC_TAG_RE.lastIndex - 1;
+      } else {
+        i++;
+      }
+    }
+  }
+  return found;
+}
+
+// `((` / `$((` open arithmetic only when the paren that closes the inner `(` is
+// directly followed by `)`; `((a); b)` is nested subshells (and `$((a); b)` a
+// substitution). `from` is the index just after the two opening parens. A span
+// that runs past the end of the line is assumed to be arithmetic. `cache` is one
+// object per line: a single left-to-right pass (skipping quoted spans and
+// escapes) matches every `(` it visits to its `)`, so the many calls on a line
+// full of `((` stay linear instead of each rescanning the rest of the line.
+function parenIsArith(line, from, cache) {
+  const open = from - 1; // the inner `(`
+  if (!cache.close || !cache.close.has(open)) cache.close = matchParens(line, from);
+  const close = cache.close.get(open);
+  return close < 0 ? true : line[close + 1] === ')';
+}
+
+// Match every `(` visited from `from` on to its `)`: Map(open -> close, or -1
+// when it never closes or an unclosed quote hides the rest of the line). The
+// paren just before `from` is the inner one of a `((`, so it is seeded as open.
+function matchParens(line, from) {
+  const close = new Map([[from - 1, -1]]);
+  const opens = [from - 1];
+  for (let j = from; j < line.length; j++) {
+    const c = line[j];
+    if (c === '\\') { j++; continue; }
+    if (c === "'" || c === '"') { // skip a quoted span on this line
+      const end = line.indexOf(c, j + 1);
+      if (end < 0) break;
+      j = end;
+    } else if (c === '(') { close.set(j, -1); opens.push(j); }
+    else if (c === ')' && opens.length > 0) close.set(opens.pop(), j);
+  }
+  return close;
+}
+
+// Code that can be split into statements: the innermost open context is a
+// `$(..)` substitution (or nothing). Frames are uniform { k, ... } objects, so
+// no string can ever hit a String.prototype property by accident.
+const splittable = (stack) => stack.length === 0 || stack[stack.length - 1].k === 'sub';
+
+// The one walk shared by stripHeredocBodies, heredocBodies, countHeredocs and
+// both splitters, so they always agree (secret-guard pairs heredocBodies with
+// per-statement countHeredocs by position). A heredoc BODY is literal data
+// (e.g. a script being written to a file), not shell statements run in the
+// current shell. Whenever a line introduces heredocs, every following line up
+// to and including each terminator is body. Returns { lines, keep, docs, seps,
+// nl, cont, hd }: `keep[i]` is false for body and terminator lines; `docs` is
+// [{ quoted, body }] in order of appearance (`hd[i]` holds the introducer positions on line i) (`quoted` is true for a quoted
+// delimiter, whose body is literal text); `seps[i]` are the separators inside
+// line i and `nl[i]` says whether the newline after it separates statements;
+// `cont[i]` says whether line i ends in a live backslash-newline continuation.
+function scanHeredocs(cmd, initialStack) {
   const lines = String(cmd).split('\n');
-  const out = [];
+  const keep = new Array(lines.length).fill(true);
+  const docs = [];
+  const seps = lines.map(() => []);
+  const nl = new Array(lines.length).fill(false);
+  const cont = new Array(lines.length).fill(false);
+  const hd = lines.map(() => []); // hd[i]: indexes of the heredoc introducers on line i
+  const st = { stack: (initialStack || []).map((f) => ({ ...f })) }; // a copy: never shared with the caller
   for (let i = 0; i < lines.length; i++) {
-    let line = lines[i];
-    out.push(line);
-    let m;
-    // A single line can introduce more than one heredoc (`cmd <<A <<B`); walk
-    // left to right and skip each body in turn.
-    while ((m = HEREDOC_RE.exec(line))) {
-      const dashStrip = m[1] === '-';
-      const tag = m[2] || m[3] || m[4];
-      line = line.slice(m.index + m[0].length); // keep scanning the rest of this line for more `<<TAG`s
-      if (!tag) break;
-      // Skip lines until the terminator: an exact match, or (for `<<-`) the
-      // tag preceded only by tabs.
+    const found = scanLine(lines[i], st, seps[i]);
+    hd[i] = found.map((f) => f.at);
+    nl[i] = !st.cont && !st.bq && splittable(st.stack);
+    cont[i] = st.cont;
+    // A line ending in a backslash continues on the next: that is ONE logical
+    // line, and any heredoc body starts after its last physical line. So does a
+    // quote opened after the introducer (`cat <<E 'a<nl>b'`): bash starts the
+    // body after the logical line, which ends only when the quote closes.
+    // A heredoc introduced inside the innermost open `$(..)` (the `"$(cat <<EOF`
+    // idiom) starts its body on the next physical line; one introduced at an
+    // outer depth waits until every context opened after it has closed.
+    while ((st.cont || (st.bq && found.length > 0) || found.some((f) => st.stack.length > f.depth)) && i + 1 < lines.length) {
+      i++;
+      const more = scanLine(lines[i], st, seps[i]);
+      hd[i] = more.map((f) => f.at);
+      found.push(...more);
+      nl[i] = !st.cont && !st.bq && splittable(st.stack);
+      cont[i] = st.cont;
+    }
+    // A single line can introduce more than one heredoc (`cmd <<A <<B`); the
+    // bodies follow in order.
+    for (const { dash, tag, quoted } of found) {
+      const body = [];
       while (i + 1 < lines.length) {
         i++;
-        const body = lines[i];
-        const term = dashStrip ? body.replace(/^\t+/, '') : body;
-        if (term === tag) break; // terminator line: consumed, not a candidate statement either
-        // else: heredoc data line, deliberately NOT pushed to `out` - it must
-        // never become a candidate statement.
+        keep[i] = false; // body or terminator: never a candidate statement
+        const l = lines[i];
+        if ((dash ? l.replace(/^\t+/, '') : l) === tag) break; // exact match, or (for `<<-`) tabs then the tag
+        body.push(l);
       }
+      docs.push({ quoted, body: body.join('\n') });
+    }
+  }
+  return { lines, keep, docs, seps, nl, cont, hd };
+}
+
+// Remove every heredoc body (and terminator line) before the statement splitter
+// sees the command. Handles multiple heredocs on one line and quoted tags.
+function stripHeredocBodies(cmd) {
+  const { lines, keep } = scanHeredocs(cmd);
+  return lines.filter((_, i) => keep[i]).join('\n');
+}
+
+// The heredoc bodies of a command, in order of appearance: [{ quoted, body }].
+function heredocBodies(cmd) {
+  return scanHeredocs(cmd).docs;
+}
+
+function lexCount(text) {
+  const st = { stack: [] };
+  return String(text).split('\n').reduce((n, line) => n + scanLine(line, st).length, 0);
+}
+
+// How many heredoc introducers a statement holds. A splitter entry carries the
+// count from the single whole-command scan (`heredocs`), which is what callers
+// pairing statements with heredocBodies must use. A plain string is lexed
+// standalone from an empty stack: pure, no memory of any earlier split, so a
+// fragment cut out of a larger command may count differently from its entry.
+function countHeredocs(stmt) {
+  if (stmt && typeof stmt === 'object' && typeof stmt.heredocs === 'number') return stmt.heredocs;
+  return lexCount(stmt && typeof stmt === 'object' ? stmt.text : stmt);
+}
+
+// The text of every outermost command substitution in a command: `$(..)`,
+// `<(..)`, `>(..)` and backticks, including those inside double quotes (the
+// shell runs them), never one in single quotes, `$'..'`, arithmetic, a comment or a
+// heredoc body. Nested ones are not returned: callers recurse into a body to
+// reach them. A backtick body is returned as the shell would run it: `\``,
+// `\\` and `\$` (and `\"` inside double quotes) are unescaped, so a nested `\`..\`` becomes a real backtick
+// span in the body. A body can span lines (joined with '\n'); one that never closes
+// runs to the end of the input (fail toward scanning). Read off scanLine's
+// `sub` frames, so it agrees with the splitters on what is code.
+function substitutionBodies(cmd) {
+  // Cut from the ORIGINAL text: a heredoc inside a body keeps its body and
+  // terminator lines there (the caller's own scan strips them), while lines
+  // that are heredoc body are never lexed, so a `$(..)` in one is not returned.
+  const { lines, keep } = scanHeredocs(cmd);
+  return collectSubstitutionBodies(lines, keep, { stack: [] });
+}
+
+// The outermost substitution bodies of an UNQUOTED heredoc body (#218): the
+// shell runs `$(..)` and backticks there, though substitutionBodies never lexes
+// a body line. The body is lexed as the shell reads a heredoc body, not as
+// command text: quotes are literal (`it's $(cat .env)` still has its
+// substitution), `\$` and `\`` stay escaped. Every line is lexed (a heredoc
+// introducer in the body is text, not another heredoc). Same result shape and
+// backtick unescaping as substitutionBodies. A quoted-delimiter body is literal
+// and must not be passed here.
+function heredocSubstitutionBodies(body) {
+  const hd = () => [{ k: 'hd' }];
+  // Scan once for the keep marks: a heredoc introduced inside a substitution in
+  // the body owns the lines after it (its body and terminator are data, so a `)`
+  // there does not close the substitution). Then collect from a fresh state.
+  const { lines, keep } = scanHeredocs(joinLineContinuations(body), hd());
+  return collectSubstitutionBodies(lines, keep, { stack: hd() });
+}
+
+// Remove every backslash-newline pair whose backslash is itself unescaped, as
+// the shell does in an unquoted heredoc body before it expands anything. A run
+// of backslashes before the newline pairs up from the left: an odd run leaves
+// one live backslash (joined), an even run is escaped backslashes plus a real
+// newline (kept).
+function joinLineContinuations(text) {
+  const s = String(text);
+  let out = '';
+  for (let i = 0; i < s.length; i++) {
+    if (s[i] === '\\') {
+      if (s[i + 1] === '\n') { i++; continue; } // live backslash-newline: dropped
+      out += s[i] + (s[i + 1] === undefined ? '' : s[i + 1]); // escaped pair: kept whole
+      i++;
+      continue;
     }
+    out += s[i];
   }
-  return out.join('\n');
+  return out;
+}
+
+// The one collector behind substitutionBodies and heredocSubstitutionBodies:
+// lex the `keep`-marked lines from the start state `st` and read the outermost
+// body texts off scanLine's `subs` callbacks.
+function collectSubstitutionBodies(lines, keep, st) {
+  const out = [];
+  let cur = null; // text of the open body so far, or null
+  let from = 0; // where the open body's text starts on the current line
+  let bq = null; // the open body is a backtick span: null, or { dq } (in double quotes)
+  const unescape = (t) => (bq ? t.replace(bq.dq ? /\\([`\\$"])/g : /\\([`\\$])/g, '$1') : t);
+  const subs = {
+    open: (at, isBq, dq) => { cur = ''; from = at; bq = isBq ? { dq: !!dq } : null; },
+    close: (at) => { out.push(unescape(cur + lines[li].slice(from, at))); cur = null; },
+  };
+  let li = 0;
+  for (; li < lines.length; li++) {
+    if (!keep[li]) { if (cur !== null) cur += lines[li] + '\n'; continue; }
+    from = 0;
+    scanLine(lines[li], st, undefined, subs);
+    if (cur !== null) cur += lines[li].slice(from) + '\n';
+  }
+  if (cur !== null) out.push(unescape(cur.slice(0, -1))); // unterminated: the remainder
+  return out;
+}
+
+// Lex the `keep`-marked lines of a command from the start state `st`, telling the
+// optional observers on `st` (funsubs, onCont) which line they are on.
+function lexKeptLines(lines, keep, st) {
+  for (let li = 0; li < lines.length; li++) {
+    if (!keep[li]) continue;
+    st.lineNo = li;
+    scanLine(lines[li], st);
+  }
+}
+
+// Every live bash 5.3 function substitution in a command: `${ cmd; }`, `${| cmd; }`
+// and `${` at the end of a line. They run code the statement splitter treats as
+// ordinary text. Read off scanLine, so a `${ ` in single quotes, `$'..'`, a comment,
+// arithmetic or a quoted-delimiter heredoc body is not reported, while one in double
+// quotes, `$(..)` or an unquoted heredoc body is. Returns [{ at, payload }]: `at`
+// is the offset of the `${` in `cmd` (-1 inside a heredoc body), `payload` the code
+// up to the closing brace (to the end of the text when it never closes). A backtick
+// span is opaque to the lexer: callers recurse into substitutionBodies for those.
+function functionSubstitutions(cmd) {
+  const out = [];
+  const lexText = (text, initial, absolute) => {
+    const { lines, keep } = scanHeredocs(text, initial);
+    const starts = [];
+    lines.reduce((n, l, i) => { starts[i] = n; return n + l.length + 1; }, 0);
+    const st = { stack: (initial || []).map((f) => ({ ...f })), funsubs: [] };
+    lexKeptLines(lines, keep, st);
+    for (const f of st.funsubs) {
+      const endLine = f.endLine === undefined ? lines.length - 1 : f.endLine;
+      const endAt = f.endAt === undefined ? lines[endLine].length : f.endAt;
+      const parts = lines.slice(f.line, endLine + 1);
+      parts[parts.length - 1] = parts[parts.length - 1].slice(0, endAt);
+      parts[0] = parts[0].slice(f.at + 2);
+      out.push({ at: absolute ? starts[f.line] + f.at : -1, payload: parts.join('\n') });
+    }
+  };
+  lexText(String(cmd), [], true);
+  for (const doc of heredocBodies(cmd)) {
+    if (!doc.quoted) lexText(joinLineContinuations(doc.body), [{ k: 'hd' }], false);
+  }
+  return out;
+}
+
+// Every live backslash-newline that splits a WORD: the char before the backslash is
+// not a word boundary and the next line starts with a non-blank (`cat .e\<nl>nv`).
+// A continuation between words (` \<nl>  --body x`, the usual multi-line form) is
+// not reported, nor is one inside quotes, a comment or a heredoc body. Returns
+// [{ line, junction }]: `junction` is the glued text around the join (`.env`).
+function inWordContinuations(cmd) {
+  const out = [];
+  const { lines, keep } = scanHeredocs(cmd);
+  const st = {
+    stack: [],
+    onCont: (at, boundary, top) => {
+      const next = lines[st.lineNo + 1];
+      if (boundary || next === undefined || !/^\S/.test(next)) return;
+      if (top && top.k !== 'sub') return; // inside quotes the existing emulation joins it
+      const before = /[^\s;&|()<>]{0,6}$/.exec(lines[st.lineNo].slice(0, at))[0];
+      out.push({ line: st.lineNo, junction: before + /^[^\s;&|()<>]{0,6}/.exec(next)[0] });
+    },
+  };
+  lexKeptLines(lines, keep, st);
+  return out;
 }
 
 // --- statement splitting -------------------------------------------------------
 
+// Cut a command at the separators the lexer reports. Heredoc bodies are skipped
+// (literal data, never statements). Returns [{ text, sep }] with the operator
+// that ENDED each statement (null for the last) and the number of heredoc
+// introducers it holds (counted from the scan positions), untrimmed and unfiltered.
+// A live backslash-newline continuation is removed like bash does (the backslash
+// and the newline both go), so a word split across lines is one word again.
+function cutStatements(cmd, ops) {
+  const { lines, keep, seps, nl, cont, hd } = scanHeredocs(cmd);
+  const kept = lines.map((_, i) => i).filter((i) => keep[i]);
+  const out = [];
+  let cur = '';
+  let heredocs = 0;
+  kept.forEach((i, k) => {
+    let from = 0;
+    const cuts = seps[i].filter((s) => ops.includes(s.op));
+    let h = 0;
+    for (const { at, len, op } of cuts) {
+      while (h < hd[i].length && hd[i][h] < at) { heredocs++; h++; }
+      out.push({ text: cur + lines[i].slice(from, at), sep: op, heredocs });
+      cur = '';
+      heredocs = 0;
+      from = at + len;
+    }
+    heredocs += hd[i].length - h;
+    cur += lines[i].slice(from);
+    if (k === kept.length - 1) return;
+    if (cont[i] && cur.endsWith('\\')) { cur = cur.slice(0, -1); return; } // joined: no separator, no newline
+    if (nl[i]) { out.push({ text: cur, sep: '\n', heredocs }); cur = ''; heredocs = 0; } else cur += '\n';
+  });
+  out.push({ text: cur, sep: null, heredocs });
+  return out;
+}
+
 // Split a shell string into top-level statements on `&&`, `||`, `;`, `|`, and
-// newlines, respecting single/double quotes (so a separator INSIDE a quoted
-// argument, e.g. a grep pattern, does not fracture the statement). Heredoc
-// bodies are stripped first (see stripHeredocBodies) so literal data written
-// via `<<TAG` is never mistaken for statements. Splitting is purely
+// newlines, respecting quotes, `$((..))`, `$[..]`, `((..))` and comments (so a
+// separator INSIDE any of them, e.g. a grep pattern, does not fracture the
+// statement). Heredoc bodies are stripped first so literal data written via
+// `<<TAG` is never mistaken for statements. Splitting is purely
 // character-driven (not whitespace-driven), so a glued separator with no
 // surrounding spaces (`cat .env|grep KEY`, `cat .env;echo done`) still splits
 // correctly. Best-effort: not a full shell parser, just enough to avoid the
 // common false splits.
 function splitStatements(cmd) {
-  const s = stripHeredocBodies(String(cmd));
-  const stmts = [];
-  let cur = '';
-  let inSingle = false;
-  let inDouble = false;
-  for (let i = 0; i < s.length; i++) {
-    const c = s[i];
-    if (inSingle) {
-      cur += c;
-      if (c === "'") inSingle = false;
-      continue;
-    }
-    if (inDouble) {
-      cur += c;
-      if (c === '\\' && i + 1 < s.length) { cur += s[++i]; continue; } // escaped char stays literal
-      if (c === '"') inDouble = false;
-      continue;
-    }
-    if (c === "'") { inSingle = true; cur += c; continue; }
-    if (c === '"') { inDouble = true; cur += c; continue; }
-    if (c === '\\' && i + 1 < s.length) { cur += c + s[++i]; continue; }
-    if (c === '&' && s[i + 1] === '&') { stmts.push(cur); cur = ''; i++; continue; }
-    if (c === '|' && s[i + 1] === '|') { stmts.push(cur); cur = ''; i++; continue; }
-    if (c === ';' || c === '|' || c === '\n') { stmts.push(cur); cur = ''; continue; }
-    cur += c;
-  }
-  stmts.push(cur);
-  return stmts.map((t) => t.trim()).filter(Boolean);
+  return cutStatements(cmd, [';', '|', '&&', '||'])
+    .map((x) => ({ text: x.text.trim(), heredocs: x.heredocs }))
+    .filter((x) => x.text)
+    .map((x) => x.text);
+}
+
+// Like splitStatements, but each statement carries the operator that ENDED it
+// (`;`, `\n`, `&&`, `||`, `|`, `&`, or null for the last one), and a lone `&`
+// (background) also splits. Callers use the separators to tell which statements
+// ran in the current shell and which in a subshell (pipeline or background).
+// Same lexing and heredoc rules as splitStatements, which stays unchanged for
+// its existing consumers. A `&` that is part of a redirection (`2>&1`, `>&2`,
+// `&>f`) is not a separator.
+function splitStatementsWithSeparators(cmd) {
+  return cutStatements(cmd, [';', '|', '&&', '||', '&'])
+    .map((x) => ({ text: x.text.trim(), sep: x.sep, heredocs: x.heredocs }))
+    .filter((x) => x.text);
 }
 
 // --- tokenizing ----------------------------------------------------------------
@@ -237,7 +662,14 @@ module.exports = {
   basename,
   stripQuotes,
   stripHeredocBodies,
+  heredocBodies,
+  countHeredocs,
   splitStatements,
+  splitStatementsWithSeparators,
+  substitutionBodies,
+  heredocSubstitutionBodies,
+  functionSubstitutions,
+  inWordContinuations,
   tokenize,
   skipEnvPrefix,
   GREP_VALUE_FLAGS,