@huyooo/ai-chat-shared 0.2.13 → 0.2.15

package/dist/index.d.ts

@@ -22,17 +22,6 @@ interface CodeBlock extends ContentBlockBase {
 }
 /** 内容块联合类型 */
 type ContentBlock = TextBlock | CodeBlock;
-/** 工具渲染器 Props */
-interface ToolRendererProps {
-    /** 工具名称 */
-    toolName: string;
-    /** 工具参数 */
-    toolArgs: Record<string, unknown>;
-    /** 工具返回结果 */
-    toolResult: unknown;
-    /** 执行状态 */
-    status: 'running' | 'completed' | 'error';
-}
 /** 天气数据 */
 interface WeatherData {
     city: string;
@@ -95,6 +84,12 @@ interface StreamParseState {
     codeLanguage?: string;
     /** 当前代码块内容 */
     codeContent: string;
+    /** 当前代码块 ID（用于稳定 key，避免流式时每次 render 都生成新 id） */
+    codeBlockId: string | null;
+    /** 当前文本块内容（流式追加，避免把一句话拆成很多 text block） */
+    textContent: string;
+    /** 当前文本块 ID（稳定 key） */
+    textBlockId: string | null;
 }
 /** 创建初始流式解析状态 */
 declare function createStreamParseState(): StreamParseState;
@@ -133,7 +128,28 @@ declare function getLanguageDisplayName(language?: string): string;
 /**
  * Markdown 渲染工具
  * 提供统一的 Markdown 渲染函数，供 React 和 Vue 版本使用
+ * 支持 LaTeX 数学公式渲染和 Mermaid 图表
+ */
+/**
+ * Mermaid 源码编码成可安全放入 attribute 的 base64url（UTF-8）
+ * 给前端组件（非 markdown 渲染路径）复用，避免重复实现。
+ */
+declare function encodeMermaidCodeToBase64Url(code: string): string;
+/**
+ * 初始化 Mermaid（仅需调用一次）
+ */
+declare function initMermaid(): void;
+/**
+ * 渲染页面中的 Mermaid 图表
+ * 用户点击"显示图表"按钮后调用
+ * @param container 容器元素（可选，默认为 document）
+ */
+declare function renderMermaidDiagrams(container?: Element): Promise<void>;
+/**
+ * 渲染 fenced 的 LaTeX 代码块（```latex / ```katex），用于“先代码、后渲染”的视图切换。
+ * 注意：这里只返回 HTML 字符串，容器样式由 `.latex-block` 控制。
  */
+declare function renderLatexBlockToHtml(code: string): string;
 /**
  * 渲染 Markdown 为 HTML
  * @param markdown Markdown 文本
@@ -141,4 +157,18 @@ declare function getLanguageDisplayName(language?: string): string;
  */
 declare function renderMarkdown(markdown: string): string;
 
-export { type CodeBlock, type CodeExecutionResult, type ContentBlock, type ContentBlockBase, type ContentBlockType, type FileOperationResult, type SearchResultItem, type StreamParseState, type TextBlock, type ToolRendererProps, type WeatherData, type WeatherForecastItem, createStreamParseState, finishStreamParse, getLanguageDisplayName, highlightCode, parseContent, parseContentStream, renderMarkdown };
+declare function isMermaidLanguage(lang?: string): boolean;
+declare function isLatexLanguage(lang?: string): boolean;
+declare function isLatexDocument(code: string): boolean;
+declare function canVisualizeLatex(code: string): boolean;
+declare function hasLatexDelimiters(code: string): boolean;
+declare function shouldShowVisualToggle(lang?: string, code?: string): boolean;
+/**
+ * 将 fenced 的 latex/katex/tex 代码块渲染成 HTML（复用 shared 的 renderMarkdown + latex 预处理）。
+ * - 如果是完整 LaTeX 文档：返回提示 + code block（避免 KaTeX 报错）
+ * - 如果已经带分隔符：直接渲染（避免二次包裹）
+ * - 否则：自动包一层 $$...$$ 作为 display 公式渲染
+ */
+declare function renderFencedLatexToHtml(code: string): string;
+
+export { type CodeBlock, type CodeExecutionResult, type ContentBlock, type ContentBlockBase, type ContentBlockType, type FileOperationResult, type SearchResultItem, type StreamParseState, type TextBlock, type WeatherData, type WeatherForecastItem, canVisualizeLatex, createStreamParseState, encodeMermaidCodeToBase64Url, finishStreamParse, getLanguageDisplayName, hasLatexDelimiters, highlightCode, initMermaid, isLatexDocument, isLatexLanguage, isMermaidLanguage, parseContent, parseContentStream, renderFencedLatexToHtml, renderLatexBlockToHtml, renderMarkdown, renderMermaidDiagrams, shouldShowVisualToggle };

package/dist/index.js

@@ -1,262 +1 @@
-// src/parser.ts
-function generateId() {
-  return Math.random().toString(36).slice(2, 11);
-}
-var CODE_BLOCK_REGEX = /```(\w*)\n([\s\S]*?)```/g;
-function parseContent(raw) {
-  if (!raw) return [];
-  const blocks = [];
-  let lastIndex = 0;
-  let match;
-  const regex = new RegExp(CODE_BLOCK_REGEX.source, "g");
-  while ((match = regex.exec(raw)) !== null) {
-    if (match.index > lastIndex) {
-      const textContent = raw.slice(lastIndex, match.index);
-      if (textContent.trim()) {
-        blocks.push(createTextBlock(textContent));
-      }
-    }
-    const language = match[1] || void 0;
-    const code = match[2];
-    blocks.push(createCodeBlock(code, language));
-    lastIndex = match.index + match[0].length;
-  }
-  if (lastIndex < raw.length) {
-    const textContent = raw.slice(lastIndex);
-    if (textContent.trim()) {
-      blocks.push(createTextBlock(textContent));
-    }
-  }
-  return blocks;
-}
-function createTextBlock(content) {
-  return {
-    id: generateId(),
-    type: "text",
-    content: content.trim()
-  };
-}
-function createCodeBlock(content, language, filename) {
-  return {
-    id: generateId(),
-    type: "code",
-    content,
-    language,
-    filename
-  };
-}
-function createStreamParseState() {
-  return {
-    buffer: "",
-    blocks: [],
-    inCodeBlock: false,
-    codeLanguage: void 0,
-    codeContent: ""
-  };
-}
-function parseContentStream(chunk, state) {
-  const newState = { ...state };
-  newState.buffer += chunk;
-  while (true) {
-    if (newState.inCodeBlock) {
-      const endIndex = newState.buffer.indexOf("```");
-      if (endIndex !== -1) {
-        newState.codeContent += newState.buffer.slice(0, endIndex);
-        newState.blocks.push(createCodeBlock(
-          newState.codeContent.trim(),
-          newState.codeLanguage
-        ));
-        newState.buffer = newState.buffer.slice(endIndex + 3);
-        newState.inCodeBlock = false;
-        newState.codeLanguage = void 0;
-        newState.codeContent = "";
-      } else {
-        newState.codeContent += newState.buffer;
-        newState.buffer = "";
-        break;
-      }
-    } else {
-      const startIndex = newState.buffer.indexOf("```");
-      if (startIndex !== -1) {
-        const beforeCode = newState.buffer.slice(0, startIndex);
-        if (beforeCode.trim()) {
-          newState.blocks.push(createTextBlock(beforeCode));
-        }
-        const afterStart = newState.buffer.slice(startIndex + 3);
-        const newlineIndex = afterStart.indexOf("\n");
-        if (newlineIndex !== -1) {
-          newState.codeLanguage = afterStart.slice(0, newlineIndex).trim() || void 0;
-          newState.buffer = afterStart.slice(newlineIndex + 1);
-          newState.inCodeBlock = true;
-          newState.codeContent = "";
-        } else {
-          break;
-        }
-      } else {
-        const lastBackticks = newState.buffer.lastIndexOf("`");
-        if (lastBackticks !== -1 && newState.buffer.length - lastBackticks < 3) {
-          const safeText = newState.buffer.slice(0, lastBackticks);
-          if (safeText.trim()) {
-            newState.blocks.push(createTextBlock(safeText));
-          }
-          newState.buffer = newState.buffer.slice(lastBackticks);
-        } else {
-          if (newState.buffer.trim()) {
-            newState.blocks.push(createTextBlock(newState.buffer));
-          }
-          newState.buffer = "";
-        }
-        break;
-      }
-    }
-  }
-  return newState;
-}
-function finishStreamParse(state) {
-  const blocks = [...state.blocks];
-  if (state.inCodeBlock) {
-    blocks.push(createCodeBlock(
-      (state.codeContent + state.buffer).trim(),
-      state.codeLanguage
-    ));
-  } else if (state.buffer.trim()) {
-    blocks.push(createTextBlock(state.buffer));
-  }
-  return blocks;
-}
-
-// src/highlighter.ts
-import hljs from "highlight.js";
-var LANGUAGE_ALIASES = {
-  js: "javascript",
-  ts: "typescript",
-  tsx: "typescript",
-  jsx: "javascript",
-  py: "python",
-  rb: "ruby",
-  sh: "bash",
-  shell: "bash",
-  yml: "yaml",
-  md: "markdown"
-};
-function highlightCode(code, language) {
-  if (!language) {
-    try {
-      const result = hljs.highlightAuto(code);
-      return result.value;
-    } catch {
-      return escapeHtml(code);
-    }
-  }
-  const normalizedLang = LANGUAGE_ALIASES[language.toLowerCase()] || language.toLowerCase();
-  try {
-    if (hljs.getLanguage(normalizedLang)) {
-      return hljs.highlight(code, { language: normalizedLang }).value;
-    }
-    return escapeHtml(code);
-  } catch {
-    return escapeHtml(code);
-  }
-}
-function escapeHtml(text) {
-  return text.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;").replace(/"/g, "&quot;").replace(/'/g, "&#039;");
-}
-function getLanguageDisplayName(language) {
-  if (!language) return "Plain Text";
-  const normalizedLang = LANGUAGE_ALIASES[language.toLowerCase()] || language.toLowerCase();
-  const langDef = hljs.getLanguage(normalizedLang);
-  if (langDef) {
-    return normalizedLang.charAt(0).toUpperCase() + normalizedLang.slice(1);
-  }
-  return language;
-}
-
-// src/markdown.ts
-import { marked, Renderer } from "marked";
-import DOMPurify from "dompurify";
-function renderMarkdown(markdown) {
-  if (!markdown) return "";
-  const renderer = new Renderer();
-  renderer.code = (code, language) => {
-    const highlighted = highlightCode(code, language);
-    const lang = language || "plaintext";
-    return `<pre class="markdown-code-block"><code class="language-${lang}">${highlighted}</code></pre>`;
-  };
-  renderer.table = (header, body) => {
-    return `<div class="markdown-table-wrapper"><table class="markdown-table"><thead>${header}</thead><tbody>${body}</tbody></table></div>`;
-  };
-  renderer.link = (href, title, text) => {
-    const titleAttr = title ? ` title="${title}"` : "";
-    return `<a href="${href}"${titleAttr} target="_blank" rel="noopener noreferrer" class="markdown-link">${text}</a>`;
-  };
-  renderer.listitem = (text, task, checked) => {
-    if (task) {
-      const checkboxText = checked ? "[x]" : "[ ]";
-      const textWithoutCheckbox = text.replace(/<input[^>]*>/g, "").trim();
-      return `<li class="markdown-task-item"><span class="markdown-task-checkbox">${checkboxText}</span> ${textWithoutCheckbox}</li>`;
-    }
-    return `<li>${text}</li>`;
-  };
-  marked.setOptions({
-    breaks: true,
-    // 支持换行
-    gfm: true
-    // GitHub Flavored Markdown
-  });
-  let html = marked(markdown, { renderer });
-  html = DOMPurify.sanitize(html, {
-    ALLOWED_TAGS: [
-      "p",
-      "br",
-      "strong",
-      "em",
-      "u",
-      "s",
-      "code",
-      "pre",
-      "h1",
-      "h2",
-      "h3",
-      "h4",
-      "h5",
-      "h6",
-      "ul",
-      "ol",
-      "li",
-      "blockquote",
-      "table",
-      "thead",
-      "tbody",
-      "tr",
-      "th",
-      "td",
-      "a",
-      "img",
-      "hr",
-      "div",
-      "span"
-    ],
-    ALLOWED_ATTR: [
-      "href",
-      "title",
-      "target",
-      "rel",
-      "class",
-      "src",
-      "alt",
-      "width",
-      "height"
-    ]
-  });
-  return html;
-}
-export {
-  createStreamParseState,
-  finishStreamParse,
-  getLanguageDisplayName,
-  highlightCode,
-  parseContent,
-  parseContentStream,
-  renderMarkdown
-};
-//# sourceMappingURL=index.js.map
+function e(){return Math.random().toString(36).slice(2,11)}var t=/```(\w*)\n([\s\S]*?)```/g;function r(e){if(!e)return[];const r=[];let c,a=0;const i=new RegExp(t.source,"g");for(;null!==(c=i.exec(e));){if(c.index>a){const t=e.slice(a,c.index);t.trim()&&r.push(n(t,void 0,!0))}const t=c[1]||void 0,i=c[2];r.push(o(i,t)),a=c.index+c[0].length}if(a<e.length){const t=e.slice(a);t.trim()&&r.push(n(t,void 0,!0))}return r}function n(t,r,n=!0){return{id:r||e(),type:"text",content:n?t.trim():t}}function o(t,r,n,o){return{id:o||e(),type:"code",content:t,language:r,filename:n}}function c(){return{buffer:"",blocks:[],inCodeBlock:!1,codeLanguage:void 0,codeContent:"",codeBlockId:null,textContent:"",textBlockId:null}}function a(t,r){r&&(null===t.textBlockId&&(t.textBlockId=e()),t.textContent+=r)}function i(e){e.textContent.trim()&&(e.blocks.push(n(e.textContent,e.textBlockId||void 0,!1)),e.textContent="",e.textBlockId=null)}function s(t,r){const n={...r,blocks:[...r.blocks]};for(n.buffer+=t;;)if(n.inCodeBlock){const e=n.buffer.indexOf("```");if(-1===e){n.codeContent+=n.buffer,n.buffer="";break}n.codeContent+=n.buffer.slice(0,e),n.blocks.push(o(n.codeContent.trim(),n.codeLanguage,void 0,n.codeBlockId||void 0)),n.buffer=n.buffer.slice(e+3),n.inCodeBlock=!1,n.codeLanguage=void 0,n.codeContent="",n.codeBlockId=null}else{const t=n.buffer.indexOf("```");if(-1===t){const e=n.buffer.lastIndexOf("`");if(-1!==e&&n.buffer.length-e<3){const t=n.buffer.slice(0,e);a(n,t),n.buffer=n.buffer.slice(e)}else a(n,n.buffer),n.buffer="";break}{const r=n.buffer.slice(0,t);a(n,r),i(n);const o=n.buffer.slice(t+3),c=o.indexOf("\n");if(-1===c)break;n.codeLanguage=o.slice(0,c).trim()||void 0,n.buffer=o.slice(c+1),n.inCodeBlock=!0,n.codeContent="",n.codeBlockId=e()}}return n}function l(e){const t=[...e.blocks];if(e.inCodeBlock)t.push(o((e.codeContent+e.buffer).trim(),e.codeLanguage,void 0,e.codeBlockId||void 0));else{const r=e.textContent+e.buffer;r.trim()&&t.push(n(r,e.textBlockId||void 0,!1))}return t}import d from"highlight.js";var u={js:"javascript",ts:"typescript",tsx:"typescript",jsx:"javascript",py:"python",rb:"ruby",sh:"bash",shell:"bash",yml:"yaml",md:"markdown"};function m(e,t){if(!t)try{return d.highlightAuto(e).value}catch{return f(e)}const r=u[t.toLowerCase()]||t.toLowerCase();try{return d.getLanguage(r)?d.highlight(e,{language:r}).value:f(e)}catch{return f(e)}}function f(e){return e.replace(/&/g,"&amp;").replace(/</g,"&lt;").replace(/>/g,"&gt;").replace(/"/g,"&quot;").replace(/'/g,"&#039;")}function p(e){if(!e)return"Plain Text";const t=u[e.toLowerCase()]||e.toLowerCase();return d.getLanguage(t)?t.charAt(0).toUpperCase()+t.slice(1):e}import{marked as h,Renderer as g}from"marked";import b from"dompurify";import x from"katex";import $ from"mermaid";var k=!1;function v(e){const t=(new TextEncoder).encode(e);let r="";for(const e of t)r+=String.fromCharCode(e);return btoa(r).replace(/\+/g,"-").replace(/\//g,"_").replace(/=+$/g,"")}function L(e){return v(e)}function y(e){let t=e.replace(/\s/g,"+").replace(/-/g,"+").replace(/_/g,"/");const r=t.length%4;2===r?t+="==":3===r&&(t+="=");const n=atob(t),o=new Uint8Array(n.length);for(let e=0;e<n.length;e++)o[e]=n.charCodeAt(e);return(new TextDecoder).decode(o)}function w(){k||($.initialize({startOnLoad:!1,theme:"dark",securityLevel:"loose",fontFamily:"inherit",flowchart:{useMaxWidth:!0,htmlLabels:!0,curve:"basis"},sequence:{useMaxWidth:!0},gantt:{useMaxWidth:!0}}),k=!0)}async function C(e){k||w();const t=(e||document).querySelectorAll(".mermaid-placeholder:not(.mermaid-rendered):not(.mermaid-error-container)");for(const e of t){const t=e.getAttribute("data-mermaid-code-b64");if(!t)continue;let r="";try{r=y(t)}catch(t){e.innerHTML='<div class="mermaid-error">图表解码失败</div>',e.classList.remove("mermaid-placeholder"),e.classList.add("mermaid-error-container");continue}e.innerHTML='<div class="mermaid-loading">渲染中...</div>';try{const t=`mermaid-${Math.random().toString(36).slice(2,11)}`,{svg:n}=await $.render(t,r);e.innerHTML=n,e.classList.remove("mermaid-placeholder"),e.classList.add("mermaid-rendered"),e.removeAttribute("data-mermaid-code-b64")}catch(t){e.innerHTML=`<div class="mermaid-error">图表渲染失败</div><pre class="mermaid-source">${r.replace(/</g,"&lt;").replace(/>/g,"&gt;")}</pre>`,e.classList.remove("mermaid-placeholder"),e.classList.add("mermaid-error-container")}}}function B(e,t){try{return x.renderToString(e,{displayMode:t,throwOnError:!1,strict:!1,trust:!0,output:"html"})}catch(r){return t?`<div class="latex-error">$$${e}$$</div>`:`<span class="latex-error">$${e}$</span>`}}function A(e){return`<div class="latex-block chat-scrollbar">${B(e.trim(),!0)}</div>`}function T(e){if(!e)return"";const{processed:t,placeholders:r}=function(e){const t=new Map;let r=0;const n=()=>`\x3c!--LATEX:${r++}--\x3e`;let o=e;return o=o.replace(/\$\$([\s\S]*?)\$\$/g,(e,r)=>{const o=n(),c=B(r.trim(),!0);return t.set(o,`<div class="latex-block chat-scrollbar">${c}</div>`),o}),o=o.replace(/\\\[([\s\S]*?)\\\]/g,(e,r)=>{const o=n(),c=B(r.trim(),!0);return t.set(o,`<div class="latex-block chat-scrollbar">${c}</div>`),o}),o=o.replace(/(?<!\$)\$(?!\$)((?:\\.|[^$\n])+?)\$(?!\$)/g,(e,r)=>{const o=n(),c=B(r.trim(),!1);return t.set(o,`<span class="latex-inline">${c}</span>`),o}),o=o.replace(/\\\(([\s\S]*?)\\\)/g,(e,r)=>{const o=n(),c=B(r.trim(),!1);return t.set(o,`<span class="latex-inline">${c}</span>`),o}),{processed:o,placeholders:t}}(e),n=new g;n.code=(e,t)=>{const r=(t||"plaintext").toLowerCase();if("mermaid"===r){return`<div class="mermaid-placeholder" data-mermaid-code-b64="${v(e)}"><div class="mermaid-loading">加载图表中...</div></div>`}return`<pre class="markdown-code-block chat-scrollbar"><code class="language-${r}">${m(e,t)}</code></pre>`},n.table=(e,t)=>`<div class="markdown-table-wrapper chat-scrollbar"><table class="markdown-table"><thead>${e}</thead><tbody>${t}</tbody></table></div>`,n.link=(e,t,r)=>`<a href="${e}"${t?` title="${t}"`:""} target="_blank" rel="noopener noreferrer" class="markdown-link">${r}</a>`,n.listitem=(e,t,r)=>{if(t){return`<li class="markdown-task-item"><span class="markdown-task-checkbox">${r?"[x]":"[ ]"}</span> ${e.replace(/<input[^>]*>/g,"").trim()}</li>`}return`<li>${e}</li>`},h.setOptions({breaks:!0,gfm:!0});let o=h(t,{renderer:n});return o=function(e,t){let r=e;return t.forEach((e,t)=>{r=r.replace(t,e)}),r}(o,r),o=b.sanitize(o,{ALLOWED_TAGS:["p","br","strong","em","u","s","code","pre","h1","h2","h3","h4","h5","h6","ul","ol","li","blockquote","table","thead","tbody","tr","th","td","a","img","hr","div","span","math","semantics","mrow","mi","mn","mo","msup","msub","mfrac","mroot","msqrt","mtext","mspace","mtable","mtr","mtd","annotation","mover","munder","munderover","menclose","mpadded","svg","line","path","rect","circle","g","use","defs","symbol"],ALLOWED_ATTR:["href","title","target","rel","class","src","alt","width","height","data-mermaid-code-b64","style","xmlns","viewBox","preserveAspectRatio","d","x","y","x1","y1","x2","y2","r","cx","cy","fill","stroke","stroke-width","transform","xlink:href","aria-hidden","focusable","role","mathvariant","encoding","stretchy","fence","separator","lspace","rspace","minsize","maxsize","accent","accentunder"],ADD_ATTR:["xmlns:xlink"]}),o}function S(e){return(e||"").trim().toLowerCase()}function I(e){return"mermaid"===S(e)}function M(e){const t=S(e);return"latex"===t||"katex"===t||"tex"===t}function O(e){const t=(e||"").trim();return/\\documentclass\b|\\usepackage\b|\\begin\{document\}|\\end\{document\}/.test(t)}function E(e){const t=(e||"").trim();return!!t&&!O(t)}function _(e){const t=(e||"").trim();return!!t&&(/\$\$[\s\S]*\$\$/.test(t)||/\\\[[\s\S]*\\\]/.test(t)||/\\\([\s\S]*\\\)/.test(t)||/(?<!\$)\$(?!\$)[\s\S]*?\$(?!\$)/.test(t))}function j(e,t){return!!I(e)||!!M(e)&&E(t||"")}function q(e){const t=(e||"").trim();return t?E(t)?_(t)?T(t):T(`$$\n${t}\n$$`):T(`> 不支持将完整 LaTeX 文档作为公式渲染，请切换到代码视图查看。\n\n\`\`\`latex\n${t}\n\`\`\``):""}export{E as canVisualizeLatex,c as createStreamParseState,L as encodeMermaidCodeToBase64Url,l as finishStreamParse,p as getLanguageDisplayName,_ as hasLatexDelimiters,m as highlightCode,w as initMermaid,O as isLatexDocument,M as isLatexLanguage,I as isMermaidLanguage,r as parseContent,s as parseContentStream,q as renderFencedLatexToHtml,A as renderLatexBlockToHtml,T as renderMarkdown,C as renderMermaidDiagrams,j as shouldShowVisualToggle};