@ai-react-markdown/engine 2.4.2 → 2.4.3

package/dist/index.js

@@ -1056,6 +1056,8 @@ function rebaseDualWalk(node, segments, maxEnd, offsetDelta, lineDelta) {
     for (const child of children) rebaseDualWalk(child, segments, maxEnd, offsetDelta, lineDelta);
   }
 }
+var TABLE_PART_TAG_RE = /<(?:td|th|tr|tbody|thead|tfoot|caption|col|colgroup)\b/i;
+var STRAY_SYNTHESIZED_END_TAG_RE = /<\/(?:br|p)\b/i;
 function spliceTrees(input) {
   const { prevMdast, prevHast, tailMdast, tailHast, content, boundary, injectionPrefix, injectedSegments } = input;
   const injectedLen = injectionPrefix.length;
@@ -1112,6 +1114,12 @@ function spliceTrees(input) {
     return !(start !== void 0 && start < injectedLen);
   });
   const tailWrapVisible = tailMdastChildren.some((child) => !isWrapInvisible(child));
+  if (prefixMdast.some((c) => c.type === "html" && TABLE_PART_TAG_RE.test(c.value))) return null;
+  for (const child of tailMdastChildren) {
+    if (isWrapInvisible(child)) continue;
+    if (child.type !== "html") break;
+    if (STRAY_SYNTHESIZED_END_TAG_RE.test(child.value) || TABLE_PART_TAG_RE.test(child.value)) return null;
+  }
   const aligned = alignPrefixCut(prefixMdast, cutRegion, tailWrapVisible);
   if (aligned === null) return null;
   const hastChildren = aligned.children;
@@ -1237,6 +1245,9 @@ function alignPrefixCut(prefixMdast, cutRegion, tailWrapVisible) {
     return null;
   }
   const lastIsLiteral = last !== void 0 && last.type === "text" && last.value.trim() !== "";
+  if (lastIsLiteral && pairIdx >= 0 && visibles[pairIdx].type !== "html") {
+    return null;
+  }
   const litOwnerEnd = pairIdx >= 0 ? visibles[pairIdx].position?.end?.offset : void 0;
   const litEnd = lastIsLiteral ? last.position?.end?.offset : void 0;
   if (lastIsLiteral && last.position !== void 0 && (litEnd === void 0 || litOwnerEnd === void 0)) {
@@ -1325,8 +1336,9 @@ function stripInjectedHast(tailMdast, tailHast, injectedLen, tailWrapVisible) {
 }
 function tailLeadingTextIsHoist(tailMdastChildren, tailHastChildren) {
   const firstText = tailHastChildren[0];
-  if (!firstText) return false;
   const firstVisible = tailMdastChildren.find((c) => !isWrapInvisible(c));
+  if (firstVisible?.type === "html" && STRAY_SYNTHESIZED_END_TAG_RE.test(firstVisible.value)) return null;
+  if (!firstText) return false;
   if (!isSeparatorText(firstText)) {
     if (firstText.type === "text" && firstText.position === void 0 && firstVisible?.type === "html") {
       if (/^\s*<\/[A-Za-z][A-Za-z0-9-]*\s*>/.test(firstVisible.value)) return true;
@@ -1767,17 +1779,8 @@ function phantomSuffixCloser(content) {
 }
 
 // src/components/extractContributions.ts
-function fakeAnchorElement(url) {
-  return { type: "element", tagName: "a", properties: { href: url }, children: [] };
-}
-function sanitizeDefUrl(url, urlTransform) {
-  if (!urlTransform) return url;
-  const result = urlTransform(url, "href", fakeAnchorElement(url));
-  return result == null ? "" : String(result);
-}
 function* extractContributions(mdast, options = {}) {
   const phantomFn = options.phantomFootnoteLabels;
-  const urlTransform = options.urlTransform;
   const out = [];
   visit3(mdast, (n) => {
     if (n.type === "footnoteReference") {
@@ -1805,7 +1808,7 @@ function* extractContributions(mdast, options = {}) {
       out.push({
         kind: "linkDef",
         label: normalizeId(d.identifier),
-        url: sanitizeDefUrl(d.url, urlTransform),
+        url: d.url,
         title: d.title
       });
     }
@@ -3519,8 +3522,7 @@ function isProtocolAllowed(url, allowed) {
 }
 function sanitizeCrossChunkUrl(rawUrl, key, tagName, urlTransform, schema) {
   rawUrl = normalizeUri2(rawUrl);
-  const callerProtocols = schema.protocols;
-  const allowed = callerProtocols === void 0 || callerProtocols === null ? sanitizeSchema.protocols?.[key] : callerProtocols[key];
+  const allowed = Object.hasOwn(schema, "protocols") ? schema.protocols?.[key] : sanitizeSchema.protocols?.[key];
   if (allowed && allowed.length > 0 && !isProtocolAllowed(rawUrl, allowed)) return null;
   const transformed = urlTransform(rawUrl, key, fakeElement(tagName, key, rawUrl));
   if (transformed == null) return null;
@@ -3918,15 +3920,14 @@ function lineHasBacktick(content, pos) {
   }
   return false;
 }
-function isAtLineStart(content, pos) {
+function lineIndentBefore(content, pos) {
   let i = pos - 1;
-  let spaces = 0;
-  while (i >= 0 && content[i] === " ") {
-    spaces++;
-    if (spaces > 3) return false;
+  let indent = 0;
+  while (i >= 0 && (content[i] === " " || content[i] === "	")) {
+    indent += content[i] === "	" ? 4 : 1;
     i--;
   }
-  return i < 0 || content[i] === "\n" || content[i] === "\r";
+  return i < 0 || content[i] === "\n" || content[i] === "\r" ? indent : -1;
 }
 function findClosingBacktickRun(content, start, n) {
   let i = start;
@@ -3947,6 +3948,7 @@ function splitByProtectedRegions(content) {
   let multilineStart = -1;
   let multilineFenceMarker = null;
   let multilineFenceLength = 0;
+  let multilineFenceIndent = 0;
   function pushProtected(start, end) {
     if (start > lastIndex) {
       segments.push({ text: content.substring(lastIndex, start), isCode: false });
@@ -3960,7 +3962,8 @@ function splitByProtectedRegions(content) {
     if (multilineStart !== -1) {
       if (char === multilineFenceMarker) {
         const runLen = getRepeatedMarkerLength(content, i, multilineFenceMarker);
-        if (runLen >= multilineFenceLength && isAtLineStart(content, i) && restOfLineIsBlank(content, i + runLen)) {
+        const closerIndent = lineIndentBefore(content, i);
+        if (runLen >= multilineFenceLength && closerIndent !== -1 && closerIndent <= multilineFenceIndent + 3 && restOfLineIsBlank(content, i + runLen)) {
           pushProtected(multilineStart, i + runLen);
           multilineStart = -1;
           multilineFenceMarker = null;
@@ -3976,10 +3979,12 @@ function splitByProtectedRegions(content) {
     }
     if (char === "`" || char === "~") {
       const runLen = getRepeatedMarkerLength(content, i, char);
-      if (runLen >= 3 && isAtLineStart(content, i) && !(char === "`" && lineHasBacktick(content, i + runLen))) {
+      const openerIndent = lineIndentBefore(content, i);
+      if (runLen >= 3 && openerIndent !== -1 && !(char === "`" && lineHasBacktick(content, i + runLen))) {
         multilineStart = i;
         multilineFenceMarker = char;
         multilineFenceLength = runLen;
+        multilineFenceIndent = openerIndent;
         i += runLen;
         continue;
       }