parseman 0.26.3 → 0.27.1

package/dist/language-service/index.js

@@ -112,7 +112,7 @@ var REC = { scan: recoverScan, at: matchesAt, or: orSentinel, sentinel: firstSet
 
 // src/functional/run.ts
 var invoke = (r, input, pos, ctx) => typeof r === "function" ? r(input, pos, ctx) : r.parse(input, pos, ctx);
-function run(entry, input, options = {}) {
+function runOnce(entry, input, options, phase, profileState) {
   if (typeof entry !== "function" && typeof entry?.parse !== "function") {
     throw new TypeError(
       `run(): start production is ${entry === null ? "null" : typeof entry}, not a rule \u2014 the requested grammar rule does not exist (check the rule name).`
@@ -120,11 +120,12 @@ function run(entry, input, options = {}) {
   }
   const triviaLog = [];
   const errors = [];
+  const profile = profileState ?? (phase === void 0 ? void 0 : { phase, nodes: 0, childSlots: 0, rawSlots: 0, triviaSlots: 0, fieldSlots: 0, hostCalls: 0 });
+  const skipGlobalSinks = phase === "recognizer" || phase === "capture";
   const grammarTrivia = typeof entry !== "function" ? entry._meta.grammarTrivia : void 0;
   const ctx = {
     trackLines: false,
-    _triviaLog: triviaLog,
-    _errors: errors,
+    ...skipGlobalSinks ? { _pmProfile: profile } : { _triviaLog: triviaLog, _errors: errors, ...profile === void 0 ? {} : { _pmProfile: profile } },
     build: options.build,
     state: options.state,
     ...grammarTrivia !== void 0 ? { trivia: grammarTrivia, ...grammarTrivia._meta.triviaKindLabels ? { triviaKindLabels: grammarTrivia._meta.triviaKindLabels } : {} } : {},
@@ -151,6 +152,38 @@ function run(entry, input, options = {}) {
     unconsumedFrom
   };
 }
+function clonePassState(state) {
+  if (state === null || typeof state !== "object") return state;
+  return Array.isArray(state) ? [...state] : { ...state };
+}
+function profilePass(entry, input, options, phase) {
+  const state = { phase, nodes: 0, childSlots: 0, rawSlots: 0, triviaSlots: 0, fieldSlots: 0, hostCalls: 0 };
+  const passOptions = options.state === void 0 ? options : { ...options, state: clonePassState(options.state) };
+  const start = performance.now();
+  const result = runOnce(entry, input, passOptions, phase, state);
+  const { phase: _phase, ...counts } = state;
+  return { result, profile: { ms: performance.now() - start, ...counts } };
+}
+function run(entry, input, options = {}) {
+  if (!options.profile) return runOnce(entry, input, options);
+  if (typeof entry !== "function") {
+    throw new TypeError("run({ profile: true }) requires a compiled parser entry");
+  }
+  const recognizer = profilePass(entry, input, options, "recognizer");
+  const capture = profilePass(entry, input, options, "capture");
+  const host = profilePass(entry, input, options, "host");
+  if (recognizer.result.ok !== host.result.ok || capture.result.ok !== host.result.ok || recognizer.result.unconsumedFrom !== host.result.unconsumedFrom || capture.result.unconsumedFrom !== host.result.unconsumedFrom) {
+    throw new Error("run({ profile: true }) changed recognition; the grammar is not profile-safe");
+  }
+  return {
+    ...host.result,
+    profile: {
+      recognizer: recognizer.profile,
+      structuralCapture: capture.profile,
+      hostConstruction: host.profile
+    }
+  };
+}
 
 // src/combinators/completions.ts
 function completionsAt(target, input, offset, options = {}) {
@@ -175,7 +208,7 @@ var META = new Set("()[]{}*+?|^$.".split(""));
 var CAP_RECORD = [
   `  if (_cap && _e > _pos) {`,
   `    if (_ctx._triviaLog !== undefined) _ctx._triviaLog.push(_pos, _e)`,
-  `    if (_ctx._cstTriviaLog !== undefined) _ctx._cstTriviaLog.push(_pos, _e, _ctx._cstRawChildren ? _ctx._cstRawChildren.length : 0)`,
+  `    if (_ctx._cstTriviaLog !== undefined && _ctx.captureTrivia) _ctx._cstTriviaLog.push(_pos, _e, _ctx._cstRawChildren ? _ctx._cstRawChildren.length : 0)`,
   `  }`
 ].join("\n");
 
@@ -197,9 +230,11 @@ function cstBuildHost(typeOrOptions, children, _fields, span, rawChildren, trivi
   }
   const collapse = normalizeCstCollapse(typeOrOptions?.collapse);
   const host = (type, children2, fields, span2, rawChildren2, triviaLog2, state2) => buildCstNode(type, children2, fields, span2, rawChildren2, triviaLog2, state2);
+  host._parsemanCstOutput = true;
   if (collapse) host._parsemanCstCollapse = collapse;
   return host;
 }
+cstBuildHost._parsemanCstOutput = true;
 
 // src/cst/walk.ts
 function walk(root, visitor, ctx = void 0) {