@ai-react-markdown/engine 2.5.5 → 2.6.0

package/dist/index.js

@@ -558,7 +558,8 @@ function classifyBlockStart(text, indent, defListEnabled) {
 function computeFreezeBoundary(text, options, resume) {
   const mathFlow = options.mathFlow ?? true;
   const referenceTaint = options.referenceTaint ?? true;
-  const cp = resume && resume.defListEnabled === options.defListEnabled && resume.mathFlow === mathFlow && resume.referenceTaint === referenceTaint && resume.confirmedOffset <= text.length ? resume : freshCheckpoint(options.defListEnabled, mathFlow, referenceTaint);
+  const prev = resume;
+  const cp = prev && prev.defListEnabled === options.defListEnabled && prev.mathFlow === mathFlow && prev.referenceTaint === referenceTaint && prev.confirmedOffset <= text.length ? prev : freshCheckpoint(options.defListEnabled, mathFlow, referenceTaint);
   let start = cp.confirmedOffset;
   let tailLine = null;
   while (start < text.length) {
@@ -621,6 +622,14 @@ function computeFreezeBoundary(text, options, resume) {
   }
   return { boundary, checkpoint: cp };
 }
+function pendingFenceCloser(checkpoint) {
+  const cp = checkpoint;
+  if (cp.phasePoisonedAt !== Infinity) return "";
+  if (cp.openIndent !== 0) return "";
+  if (cp.inFence) return cp.fenceChar.repeat(cp.fenceLen);
+  if (cp.inMath) return "$".repeat(cp.mathFenceLen);
+  return "";
+}
 function floatingResidue(text, commentOpenAtStart) {
   let out = "";
   let open = commentOpenAtStart;
@@ -2172,12 +2181,10 @@ function phantomSuffixCloser(content) {
   const endsWithNewline = content.endsWith("\n");
   const confirmed = endsWithNewline ? content : content + "\n";
   const { checkpoint } = computeFreezeBoundary(confirmed, { defListEnabled: false, referenceTaint: false });
-  if (checkpoint.phasePoisonedAt !== Infinity) return "";
-  if (checkpoint.openIndent !== 0) return "";
-  const nl = endsWithNewline ? "" : "\n";
-  if (checkpoint.inFence) return `${nl}${checkpoint.fenceChar.repeat(checkpoint.fenceLen)}`;
-  if (checkpoint.inMath) return `${nl}${"$".repeat(checkpoint.mathFenceLen)}`;
-  return "";
+  const closer = pendingFenceCloser(checkpoint);
+  if (closer === "") return "";
+  return endsWithNewline ? closer : `
+${closer}`;
 }
 
 // src/components/extractContributions.ts