@ai-react-markdown/core 1.2.5 → 1.2.7

package/dist/index.cjs

@@ -1572,11 +1572,20 @@ var AIMarkdownRenderStateProvider = ({
 var context_default = AIMarkdownRenderStateProvider;
 
 // src/preprocessors/latex.ts
+function getRepeatedMarkerLength(content, start, marker) {
+  let end = start;
+  while (end < content.length && content[end] === marker) {
+    end += 1;
+  }
+  return end - start;
+}
 function splitByProtectedRegions(content) {
   const segments = [];
   let lastIndex = 0;
   let inlineStart = -1;
   let multilineStart = -1;
+  let multilineFenceMarker = null;
+  let multilineFenceLength = 0;
   function pushProtected(start, end) {
     if (start > lastIndex) {
       segments.push({ text: content.substring(lastIndex, start), isCode: false });
@@ -1586,15 +1595,22 @@ function splitByProtectedRegions(content) {
   }
   for (let i = 0; i < content.length; i++) {
     const char = content[i];
-    if (char === "`" && i + 2 < content.length && content[i + 1] === "`" && content[i + 2] === "`") {
+    const fenceLength = char === "`" || char === "~" ? getRepeatedMarkerLength(content, i, char) : 0;
+    if ((char === "`" || char === "~") && fenceLength >= 3) {
       if (multilineStart === -1) {
         inlineStart = -1;
         multilineStart = i;
-        i += 2;
-      } else {
-        pushProtected(multilineStart, i + 3);
+        multilineFenceMarker = char;
+        multilineFenceLength = fenceLength;
+        i += fenceLength - 1;
+      } else if (char === multilineFenceMarker && fenceLength >= multilineFenceLength) {
+        pushProtected(multilineStart, i + fenceLength);
         multilineStart = -1;
-        i += 2;
+        multilineFenceMarker = null;
+        multilineFenceLength = 0;
+        i += fenceLength - 1;
+      } else {
+        i += fenceLength - 1;
       }
     } else if (char === "`" && multilineStart === -1) {
       if (inlineStart === -1) {
@@ -1614,6 +1630,9 @@ function splitByProtectedRegions(content) {
       }
     }
   }
+  if (multilineStart !== -1) {
+    pushProtected(multilineStart, content.length);
+  }
   if (lastIndex < content.length) {
     segments.push({ text: content.substring(lastIndex), isCode: false });
   }
@@ -1689,6 +1708,54 @@ function escapeLatexPipes(text) {
     return match;
   });
 }
+function escapeLatexPipesInUnclosed(text) {
+  let unclosedStart = -1;
+  let i = 0;
+  while (i < text.length) {
+    if (text[i] === "$" && i + 1 < text.length && text[i + 1] === "$") {
+      if (unclosedStart === -1) {
+        unclosedStart = i;
+        i += 2;
+      } else {
+        unclosedStart = -1;
+        i += 2;
+      }
+    } else if (text[i] === "$" && (i === 0 || text[i - 1] !== "\\") && (i + 1 >= text.length || text[i + 1] !== "$")) {
+      if (unclosedStart === -1) {
+        unclosedStart = i;
+      } else {
+        unclosedStart = -1;
+      }
+      i += 1;
+    } else {
+      i += 1;
+    }
+  }
+  if (unclosedStart === -1) return text;
+  const before = text.substring(0, unclosedStart);
+  const delimLen = text[unclosedStart + 1] === "$" ? 2 : 1;
+  const delim = text.substring(unclosedStart, unclosedStart + delimLen);
+  const tail = text.substring(unclosedStart + delimLen);
+  return before + delim + replaceUnescapedPipes(tail);
+}
+function truncateUnclosedLatexBlock(text) {
+  let unclosedStart = -1;
+  let i = 0;
+  while (i < text.length) {
+    if (text[i] === "$" && i + 1 < text.length && text[i + 1] === "$") {
+      if (unclosedStart === -1) {
+        unclosedStart = i;
+      } else {
+        unclosedStart = -1;
+      }
+      i += 2;
+    } else {
+      i += 1;
+    }
+  }
+  if (unclosedStart === -1) return text;
+  return text.substring(0, unclosedStart).trimEnd();
+}
 function escapeTextUnderscores(text) {
   return text.replaceAll(ESCAPE_TEXT_UNDERSCORES_REGEX, (_match, textContent) => {
     const escapedTextContent = textContent.replaceAll(/(?<!\\)_/g, "\\_");
@@ -1708,8 +1775,10 @@ function preprocessLaTeX(str) {
     text = escapeCurrencyDollarSigns(text);
     text = convertLatexDelimiters(text);
     text = escapeLatexPipes(text);
+    text = escapeLatexPipesInUnclosed(text);
     text = escapeTextUnderscores(text);
     text = convertSingleToDoubleDollar(text);
+    text = truncateUnclosedLatexBlock(text);
     return text;
   });
   return result.join("");