@ai-react-markdown/core 1.0.0 → 1.0.2

package/dist/index.cjs

@@ -34,15 +34,18 @@ __export(index_exports, {
   AIMarkdownRenderDisplayOptimizeAbility: () => AIMarkdownRenderDisplayOptimizeAbility,
   AIMarkdownRenderExtraSyntax: () => AIMarkdownRenderExtraSyntax,
   default: () => index_default,
-  useAIMarkdownRenderState: () => useAIMarkdownRenderState
+  defaultAIMarkdownRenderConfig: () => defaultAIMarkdownRenderConfig,
+  useAIMarkdownMetadata: () => useAIMarkdownMetadata,
+  useAIMarkdownRenderState: () => useAIMarkdownRenderState,
+  useStableValue: () => useStableValue
 });
 module.exports = __toCommonJS(index_exports);
 var import_react5 = require("react");
 
 // src/context.tsx
 var import_react = require("react");
-var import_cloneDeep = __toESM(require("lodash/cloneDeep"), 1);
-var import_mergeWith = __toESM(require("lodash/mergeWith"), 1);
+var import_cloneDeep = __toESM(require("lodash-es/cloneDeep"), 1);
+var import_mergeWith = __toESM(require("lodash-es/mergeWith"), 1);
 
 // src/defs.ts
 var AIMarkdownRenderExtraSyntax = /* @__PURE__ */ ((AIMarkdownRenderExtraSyntax2) => {
@@ -57,22 +60,23 @@ var AIMarkdownRenderDisplayOptimizeAbility = /* @__PURE__ */ ((AIMarkdownRenderD
   AIMarkdownRenderDisplayOptimizeAbility2["PANGU"] = "PANGU";
   return AIMarkdownRenderDisplayOptimizeAbility2;
 })(AIMarkdownRenderDisplayOptimizeAbility || {});
-var defaultMRMarkdownRenderConfig = {
-  extraSyntaxSupported: [
+var defaultAIMarkdownRenderConfig = Object.freeze({
+  extraSyntaxSupported: Object.freeze([
     "HIGHLIGHT" /* HIGHLIGHT */,
     "DEFINITION_LIST" /* DEFINITION_LIST */,
     "SUBSCRIPT" /* SUBSCRIPT */
-  ],
-  displayOptimizeAbilities: [
+  ]),
+  displayOptimizeAbilities: Object.freeze([
     "REMOVE_COMMENTS" /* REMOVE_COMMENTS */,
     "SMARTYPANTS" /* SMARTYPANTS */,
     "PANGU" /* PANGU */
-  ]
-};
+  ])
+});
 
 // src/context.tsx
 var import_jsx_runtime = require("react/jsx-runtime");
 var AIMarkdownRenderStateContext = (0, import_react.createContext)(null);
+var AIMarkdownMetadataContext = (0, import_react.createContext)(void 0);
 function useAIMarkdownRenderState() {
   const context = (0, import_react.useContext)(AIMarkdownRenderStateContext);
   if (!context) {
@@ -80,51 +84,70 @@ function useAIMarkdownRenderState() {
   }
   return context;
 }
+function useAIMarkdownMetadata() {
+  return (0, import_react.useContext)(AIMarkdownMetadataContext);
+}
 var configMergeCustomizer = (_objValue, srcValue, _key, _object, _source, _stack) => {
   if (Array.isArray(srcValue)) {
     return srcValue;
   }
 };
+var AIMarkdownMetadataProvider = ({
+  metadata,
+  children
+}) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(AIMarkdownMetadataContext.Provider, { value: metadata, children });
+};
 var AIMarkdownRenderStateProvider = ({
   streaming,
   fontSize,
+  variant,
+  colorScheme,
+  defaultConfig,
   config,
-  metadata,
   children
 }) => {
+  const baseConfig = defaultConfig ?? defaultAIMarkdownRenderConfig;
   const mergedConfig = (0, import_react.useMemo)(
-    () => config ? (0, import_mergeWith.default)((0, import_cloneDeep.default)(defaultMRMarkdownRenderConfig), config, configMergeCustomizer) : defaultMRMarkdownRenderConfig,
-    [config]
+    () => config ? (0, import_mergeWith.default)((0, import_cloneDeep.default)(baseConfig), config, configMergeCustomizer) : baseConfig,
+    [baseConfig, config]
   );
   const state = (0, import_react.useMemo)(
-    () => ({
+    () => Object.freeze({
       streaming,
       fontSize,
-      config: mergedConfig,
-      metadata
+      variant,
+      colorScheme,
+      config: mergedConfig
     }),
-    [streaming, fontSize, mergedConfig, metadata]
+    [streaming, fontSize, variant, colorScheme, mergedConfig]
   );
   return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(AIMarkdownRenderStateContext.Provider, { value: state, children });
 };
 var context_default = AIMarkdownRenderStateProvider;
 
 // src/preprocessors/latex.ts
-function escapeMhchemCommands(text) {
-  return text.replaceAll("$\\ce{", "$\\\\ce{").replaceAll("$\\pu{", "$\\\\pu{");
-}
-function findCodeBlockRegions(content) {
-  const regions = [];
+function splitByProtectedRegions(content) {
+  const segments = [];
+  let lastIndex = 0;
   let inlineStart = -1;
   let multilineStart = -1;
+  function pushProtected(start, end) {
+    if (start > lastIndex) {
+      segments.push({ text: content.substring(lastIndex, start), isCode: false });
+    }
+    segments.push({ text: content.substring(start, end), isCode: true });
+    lastIndex = end;
+  }
   for (let i = 0; i < content.length; i++) {
     const char = content[i];
     if (char === "`" && i + 2 < content.length && content[i + 1] === "`" && content[i + 2] === "`") {
       if (multilineStart === -1) {
+        inlineStart = -1;
         multilineStart = i;
         i += 2;
       } else {
-        regions.push([multilineStart, i + 2]);
+        pushProtected(multilineStart, i + 3);
         multilineStart = -1;
         i += 2;
       }
@@ -132,43 +155,71 @@ function findCodeBlockRegions(content) {
       if (inlineStart === -1) {
         inlineStart = i;
       } else {
-        regions.push([inlineStart, i]);
+        pushProtected(inlineStart, i + 1);
         inlineStart = -1;
       }
+    } else if (char === "<" && multilineStart === -1 && inlineStart === -1) {
+      const rest = content.substring(i);
+      const tagMatch = rest.match(
+        /^<\/?(span|div|p|br|hr|img|a|em|strong|b|i|u|s|sub|sup|code|pre|table|tr|td|th|thead|tbody|tfoot|ul|ol|li|dl|dt|dd|h[1-6]|blockquote|details|summary|figure|figcaption|section|article|aside|nav|header|footer|main|mark|del|ins|small|abbr|cite|dfn|kbd|samp|var|ruby|rt|rp|bdo|wbr|input|button|select|textarea|label|fieldset|legend|output|iframe|video|audio|source|canvas|svg|math|time)(?:\s[^>]*)?\/?>/i
+      );
+      if (tagMatch) {
+        pushProtected(i, i + tagMatch[0].length);
+        i += tagMatch[0].length - 1;
+      }
     }
   }
-  return regions;
-}
-function isInCodeBlock(position, codeRegions) {
-  let left = 0;
-  let right = codeRegions.length - 1;
-  while (left <= right) {
-    const mid = Math.floor((left + right) / 2);
-    const [start, end] = codeRegions[mid];
-    if (position >= start && position <= end) {
-      return true;
-    } else if (position < start) {
-      right = mid - 1;
-    } else {
-      left = mid + 1;
-    }
+  if (lastIndex < content.length) {
+    segments.push({ text: content.substring(lastIndex), isCode: false });
   }
-  return false;
+  return segments;
+}
+function escapeMhchemCommands(text) {
+  return text.replaceAll("$\\ce{", "$\\\\ce{").replaceAll("$\\pu{", "$\\\\pu{");
 }
 var CURRENCY_REGEX = /(?<![\\$])\$(?!\$)(?=\d+(?:,\d{3})*(?:\.\d+)?(?:[KMBkmb])?(?:\s|$|[^a-zA-Z\d]))/g;
 var NO_ESCAPED_DOLLAR_REGEX = /(?<![\\$])\$(?!\$)/g;
-var DELIMITERS_REGEX = /\\\[([\S\s]*?[^\\])\\]|\\\((.*?)\\\)/g;
-var UNESCAPED_PIPES_REGEX = /(?<!\\)\|/g;
-var LATEX_BLOCK_REGEX = /\$\$([\S\s]*?)\$\$|(?<![\\$])\$(?!\$)((?:.|\n)*?)(?<![\\`])\$(?!\$)/g;
+var DELIMITERS_REGEX = /(?<!!)\\\[([\S\s]*?[^\\])\\](?!\()|\\\((.*?)\\\)/g;
+var ARRAY_COL_SPEC_OR_PIPE_REGEX = /(\\begin\{(?:array|tabular[x*]?)\}\{[^}]*\})|(?<!\\)\|/g;
+var LATEX_BLOCK_REGEX = /\$\$([\S\s]*?)\$\$|(?<![\\$])\$(?!\$)((?:[^$\n]|\\\$)*?)(?<![\\`])\$(?!\$)/g;
 var ESCAPE_TEXT_UNDERSCORES_REGEX = /\\text{([^}]*)}/g;
 var SINGLE_DOLLAR_REGEX = /(?<![\\$])\$(?!\$)((?:[^$\n]|\\[$])+?)(?<!\\)(?<!`)\$(?!\$)/g;
-function convertLatexDelimiters(text, codeRegions) {
+function escapeCurrencyDollarSigns(text) {
+  const parts = [];
+  let lastIndex = 0;
+  const currencyMatches = Array.from(text.matchAll(CURRENCY_REGEX));
+  for (let i = 0; i < currencyMatches.length; i++) {
+    const match = currencyMatches[i];
+    parts.push(text.substring(lastIndex, match.index));
+    let needEscape = true;
+    let restBeforeNextMatchOrEnd = "";
+    if (i < currencyMatches.length - 1) {
+      const nextMatch = currencyMatches[i + 1];
+      if (nextMatch.index - match.index > 1) {
+        restBeforeNextMatchOrEnd = text.substring(match.index + 1, nextMatch.index);
+      }
+    } else {
+      restBeforeNextMatchOrEnd = text.substring(match.index + 1);
+    }
+    const firstLineBeforeNextMatch = restBeforeNextMatchOrEnd.split(/\r\n|\r|\n/g)[0];
+    if (Array.from(firstLineBeforeNextMatch.matchAll(NO_ESCAPED_DOLLAR_REGEX)).length % 2 !== 0) {
+      const previousNewContent = parts.join("");
+      const previousLastLineContent = previousNewContent.split(/\r\n|\r|\n/g).pop();
+      const wholeLineBeforeNextMatchWithoutCurrentDollar = previousLastLineContent + firstLineBeforeNextMatch;
+      if (Array.from(wholeLineBeforeNextMatchWithoutCurrentDollar.matchAll(NO_ESCAPED_DOLLAR_REGEX)).length % 2 !== 0) {
+        needEscape = false;
+      }
+    }
+    parts.push(needEscape ? "\\$" : "$");
+    lastIndex = match.index + 1;
+  }
+  parts.push(text.substring(lastIndex));
+  return parts.join("");
+}
+function convertLatexDelimiters(text) {
   return text.replaceAll(
     DELIMITERS_REGEX,
-    (match, squareBracket, roundBracket, index) => {
-      if (isInCodeBlock(index, codeRegions)) {
-        return match;
-      }
+    (match, squareBracket, roundBracket) => {
       if (squareBracket !== void 0) {
         return `$$${squareBracket}$$`;
       } else if (roundBracket !== void 0) {
@@ -179,83 +230,43 @@ function convertLatexDelimiters(text, codeRegions) {
   );
 }
 var replaceUnescapedPipes = (formula) => (
-  // Use \vert{} so the control sequence terminates before the next token
-  formula.replaceAll(UNESCAPED_PIPES_REGEX, "\\vert{}")
+  // Use \vert{} so the control sequence terminates before the next token.
+  // Preserve `|` inside \begin{array}{...} / \begin{tabular}{...} column specifiers.
+  formula.replaceAll(
+    ARRAY_COL_SPEC_OR_PIPE_REGEX,
+    (match, colSpec) => colSpec !== void 0 ? match : "\\vert{}"
+  )
 );
-function escapeLatexPipes(text, codeRegions) {
-  return text.replaceAll(LATEX_BLOCK_REGEX, (match, display, inline, index) => {
-    if (isInCodeBlock(index, codeRegions)) {
-      return match;
-    }
+function escapeLatexPipes(text) {
+  return text.replaceAll(LATEX_BLOCK_REGEX, (match, display, inline) => {
     if (display !== void 0) return `$$${replaceUnescapedPipes(display)}$$`;
     if (inline !== void 0) return `$${replaceUnescapedPipes(inline)}$`;
     return match;
   });
 }
-function escapeTextUnderscores(text, codeRegions) {
-  return text.replaceAll(ESCAPE_TEXT_UNDERSCORES_REGEX, (match, textContent, index) => {
-    if (isInCodeBlock(index, codeRegions)) {
-      return match;
-    }
+function escapeTextUnderscores(text) {
+  return text.replaceAll(ESCAPE_TEXT_UNDERSCORES_REGEX, (_match, textContent) => {
     const escapedTextContent = textContent.replaceAll(/(?<!\\)_/g, "\\_");
     return `\\text{${escapedTextContent}}`;
   });
 }
+function convertSingleToDoubleDollar(text) {
+  return text.replaceAll(SINGLE_DOLLAR_REGEX, (_match, content) => `$$${content}$$`);
+}
 function preprocessLaTeX(str) {
   if (!str.includes("$") && !str.includes("\\[") && !str.includes("\\(")) return str;
-  let processed = str;
-  processed = escapeMhchemCommands(processed);
-  const codeRegions = findCodeBlockRegions(processed);
-  const parts = [];
-  let lastIndex = 0;
-  const currencyMatchesIterator = processed.matchAll(CURRENCY_REGEX);
-  const currencyMatches = Array.from(currencyMatchesIterator);
-  for (let i = 0; i < currencyMatches.length; i++) {
-    const match = currencyMatches[i];
-    parts.push(processed.substring(lastIndex, match.index));
-    let needEscape = true;
-    if (!isInCodeBlock(match.index, codeRegions)) {
-      let restBeforeNextMatchOrEnd = "";
-      if (i < currencyMatches.length - 1) {
-        const nextMatch = currencyMatches[i + 1];
-        if (nextMatch.index - match.index > 1) {
-          restBeforeNextMatchOrEnd = processed.substring(match.index + 1, nextMatch.index);
-        }
-      } else {
-        restBeforeNextMatchOrEnd = processed.substring(match.index + 1, processed.length);
-      }
-      const firstLineBeforeNextMatch = restBeforeNextMatchOrEnd.split(/\r\n|\r|\n/g)[0];
-      if (Array.from(firstLineBeforeNextMatch.matchAll(NO_ESCAPED_DOLLAR_REGEX)).length % 2 !== 0) {
-        const previousNewCotent = parts.join("");
-        const previousLastLineContent = previousNewCotent.split(/\r\n|\r|\n/g).pop();
-        const wholeLineBeforeNextMatchWithoutCurrentDollar = previousLastLineContent + firstLineBeforeNextMatch;
-        if (Array.from(wholeLineBeforeNextMatchWithoutCurrentDollar.matchAll(NO_ESCAPED_DOLLAR_REGEX)).length % 2 !== 0) {
-          needEscape = false;
-        }
-      }
-    } else {
-      needEscape = false;
-    }
-    parts.push(needEscape ? "\\$" : "$");
-    lastIndex = match.index + 1;
-  }
-  parts.push(processed.substring(lastIndex));
-  processed = parts.join("");
-  console.log("processed", processed);
-  processed = convertLatexDelimiters(processed, codeRegions);
-  processed = escapeLatexPipes(processed, codeRegions);
-  processed = escapeTextUnderscores(processed, codeRegions);
-  const result = [];
-  lastIndex = 0;
-  const singleDollarMatchesIterator = processed.matchAll(SINGLE_DOLLAR_REGEX);
-  for (const match of singleDollarMatchesIterator) {
-    if (!isInCodeBlock(match.index, codeRegions)) {
-      result.push(processed.substring(lastIndex, match.index));
-      result.push(`$$${match[1]}$$`);
-      lastIndex = match.index + match[0].length;
-    }
-  }
-  result.push(processed.substring(lastIndex));
+  const segments = splitByProtectedRegions(str);
+  const result = segments.map((segment) => {
+    if (segment.isCode) return segment.text;
+    let text = segment.text;
+    text = escapeMhchemCommands(text);
+    text = escapeCurrencyDollarSigns(text);
+    text = convertLatexDelimiters(text);
+    text = escapeLatexPipes(text);
+    text = escapeTextUnderscores(text);
+    text = convertSingleToDoubleDollar(text);
+    return text;
+  });
   return result.join("");
 }
 
@@ -319,28 +330,36 @@ var AIMarkdownContent = (0, import_react2.memo)(({ content, customComponents })
     import_react_markdown.default,
     {
       remarkPlugins: [
+        // --- Core plugins (always active) ---
         import_remark_gfm.default,
         [
           import_remark_math.default,
           {
+            // Disable single-dollar inline math to avoid conflicts with currency
+            // signs and other dollar usages; the preprocessor converts $...$ to $$...$$.
             singleDollarTextMath: false
           }
         ],
+        // --- Configurable extra syntax plugins ---
         ...extraSyntaxRemarkPlugins,
+        // --- Formatting & normalization ---
         import_remark_breaks.default,
         import_remark_emoji.default,
         import_remark_squeeze_paragraphs.default,
         import_remark_cjk_friendly.default,
         import_remark_cjk_friendly_gfm_strikethrough.default,
+        // --- Configurable display optimizations ---
         ...displayOptimizeRemarkPlugins
       ],
       rehypePlugins: [
+        // Allow raw HTML through so rehype-sanitize can handle it.
         [
           import_rehype_raw.default,
           {
             passThrough: []
           }
         ],
+        // Sanitize HTML while allowing <mark> (highlight) and KaTeX class names.
         [
           import_rehype_sanitize.default,
           {
@@ -359,6 +378,7 @@ var AIMarkdownContent = (0, import_react2.memo)(({ content, customComponents })
       remarkRehypeOptions: {
         allowDangerousHtml: true,
         handlers: {
+          // Inject definition-list HAST handlers when the extension is active.
           ...enableDefinitionList ? import_remark_definition_list.defListHastHandlers : {}
         }
       },
@@ -372,7 +392,7 @@ var MarkdownContent_default = AIMarkdownContent;
 
 // src/hooks/useStableValue.ts
 var import_react3 = require("react");
-var import_isEqual = __toESM(require("lodash/isEqual"), 1);
+var import_isEqual = __toESM(require("lodash-es/isEqual"), 1);
 function useStableValue(value) {
   const ref = (0, import_react3.useRef)(value);
   const prev = ref.current;
@@ -405,14 +425,16 @@ var AIMarkdownComponent = ({
   fontSize,
   contentPreprocessors,
   customComponents,
+  defaultConfig,
   config,
   metadata,
-  typography: Typography = Default_default,
-  extraStyle: ExtraStyle,
+  Typography = Default_default,
+  ExtraStyles,
   variant = "default",
   colorScheme = "light"
 }) => {
   const usedFontSize = fontSize ? typeof fontSize === "number" ? `${fontSize}px` : fontSize : "0.875rem";
+  const stableDefaultConfig = useStableValue(defaultConfig);
   const stableConfig = useStableValue(config);
   const stablePreprocessors = useStableValue(contentPreprocessors);
   const stableCustomComponents = useStableValue(customComponents);
@@ -420,16 +442,18 @@ var AIMarkdownComponent = ({
     () => content ? preprocessAIMDContent(content, stablePreprocessors) : content,
     [content, stablePreprocessors]
   );
-  return /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(
+  return /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(AIMarkdownMetadataProvider, { metadata, children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(
     context_default,
     {
       streaming,
       fontSize: usedFontSize,
+      variant,
+      colorScheme,
+      defaultConfig: stableDefaultConfig,
       config: stableConfig,
-      metadata,
-      children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(Typography, { fontSize: usedFontSize, variant, colorScheme, children: ExtraStyle ? /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(ExtraStyle, { children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(MarkdownContent_default, { content: usedContent, customComponents: stableCustomComponents }) }) : /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(MarkdownContent_default, { content: usedContent, customComponents: stableCustomComponents }) })
+      children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(Typography, { fontSize: usedFontSize, variant, colorScheme, children: ExtraStyles ? /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(ExtraStyles, { children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(MarkdownContent_default, { content: usedContent, customComponents: stableCustomComponents }) }) : /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(MarkdownContent_default, { content: usedContent, customComponents: stableCustomComponents }) })
     }
-  );
+  ) });
 };
 var AIMarkdown = (0, import_react5.memo)(AIMarkdownComponent);
 AIMarkdown.displayName = "AIMarkdown";
@@ -438,6 +462,9 @@ var index_default = AIMarkdown;
 0 && (module.exports = {
   AIMarkdownRenderDisplayOptimizeAbility,
   AIMarkdownRenderExtraSyntax,
-  useAIMarkdownRenderState
+  defaultAIMarkdownRenderConfig,
+  useAIMarkdownMetadata,
+  useAIMarkdownRenderState,
+  useStableValue
 });
 //# sourceMappingURL=index.cjs.map