ds-markdown 0.1.4-beta.0 → 0.1.4

package/dist/esm/index.js

@@ -1,2 +1,664 @@
-import{jsxs as e,jsx as n}from"react/jsx-runtime";import{memo as r,useMemo as t,useRef as c,useEffect as a,forwardRef as s,useState as i,useImperativeHandle as u}from"react";import o from"react-markdown";import{Prism as l}from"react-syntax-highlighter";import h from"remark-gfm";import d from"classnames";const p=({children:r,language:t,theme:c="light"})=>e("div",{className:`md-code-block md-code-block-${c}`,children:[n("div",{className:"md-code-block-banner-wrap",children:n("div",{className:"md-code-block-banner md-code-block-banner-lite",children:n("div",{className:"md-code-block-banner-content",children:n("div",{className:"md-code-block-language",children:t})})})}),n("div",{className:"md-code-block-content",children:r})]}),g="development"===process.env.NODE_ENV,m=Symbol("katex");var w=r(({theme:e="light",children:r,math:c,plugins:a,...s})=>{const i=c?.splitSymbol??"dollar",{remarkPlugins:u,rehypePlugins:d,hasKatexPlugin:g}=t(()=>{let e=!1;const n=[h],r=[];return a?(a.forEach(t=>{t.id===m&&(e=!0,n.push(t.remarkPlugin),r.push(t.rehypePlugin))}),{remarkPlugins:n,rehypePlugins:r,hasKatexPlugin:e}):{remarkPlugins:n,rehypePlugins:r}},[a]),w=t(()=>g&&"bracket"===i?(e=>{const n=[];let r=e.replace(/\\*\[([\s\S]+?)\\*\]/g,(e,r)=>(n.push(r),`__BLOCK_MATH_${n.length-1}__`));return r=r.replace(/\$\$([\s\S]+?)\$\$/g,(e,r)=>(n.push(r),`__BLOCK_MATH_${n.length-1}__`)),r=r.replace(/\\*\(([^)]+?)\\*\)/g,(e,n)=>"$"+n+"$"),r=r.replace(/__BLOCK_MATH_(\d+)__/g,(e,r)=>"$$"+n[Number(r)]+"$$"),r})(r):r,[g,i,r]);return n(o,{remarkPlugins:u,rehypePlugins:d,components:{code:({className:r,children:t,...c})=>{const a=/language-(\w+)/.exec(r||"");return a?n(p,{language:a[1],theme:e,children:n(l,{useInlineStyles:!1,language:a[1],style:{},children:String(t).replace(/\n$/,"")})}):n("code",{className:r,...c,children:t})},table:({children:e,...r})=>n("div",{className:"markdown-table-wrapper",children:n("table",{className:"ds-markdown-table",children:e})})},...s,children:w})});const k=e=>{const{timerType:n="setTimeout",interval:r,charsRef:t,onEnd:s,onStart:i,onBeforeTypedChar:u,onTypedChar:o,processCharDisplay:l,wholeContentRef:h,disableTyping:d,triggerUpdate:p,resetWholeContent:g}=e,m=c(!1),w=c(!1),k=c(null),y=c(null),f=c(void 0),T=c(!1),v=c(d);v.current=d;const b=c(r);b.current=r;const S=()=>t.current;a(()=>(m.current=!1,()=>{m.current=!0,$()}),[]);const C=e=>{if(!i)return;const n=h.current[e.answerType].content;i({currentIndex:n.length,currentChar:e.content,answerType:e.answerType,prevStr:n})},x=e=>{s&&s({str:h.current.answer.content,answerStr:h.current.answer.content,thinkingStr:h.current.thinking.content,manual:e?.manual??!1})},_=async e=>{if(!u)return;const{answerType:n,content:r,index:t}=e,c=h.current.allLength,a=e.index/c*100;await u({currentIndex:t,currentChar:r,answerType:n,prevStr:h.current[n].content,percent:a})},N=async e=>{if(!o)return;const{answerType:n,content:r,index:t}=e,c=h.current.allLength,a=(e.index+1)/c*100;o({currentIndex:t,currentChar:r,answerType:n,prevStr:h.current[n].content.slice(0,t),currentStr:h.current[n].content,percent:a})},$=()=>{k.current&&(cancelAnimationFrame(k.current),k.current=null),y.current&&(clearTimeout(y.current),y.current=null),w.current=!1,f.current=void 0},L=()=>{T.current||w.current||("requestAnimationFrame"===n?P():A())};async function M(){const e=S(),n=e.filter(e=>"thinking"===e.answerType).map(e=>e.content).join(""),r=e.filter(e=>"answer"===e.answerType).map(e=>e.content).join("");n&&await(u?.({currentIndex:h.current.thinking.length,currentChar:n,answerType:"thinking",prevStr:h.current.thinking.content,percent:100})),r&&await(u?.({currentIndex:h.current.answer.length,currentChar:r,answerType:"answer",prevStr:h.current.answer.content,percent:100})),h.current.thinking.content+=n,h.current.thinking.length+=n.length,h.current.answer.content+=r,h.current.answer.length+=r.length,h.current.allLength+=n.length+r.length,t.current=[],w.current=!1,x(),p()}const P=()=>{let e=performance.now();const n=async r=>{if(v.current)return void await M();const t=S();if(m.current)return;if(0===t.length)return void I();const c=r-e;let a=Math.max(0,Math.floor(c/b.current));if(a=Math.min(a,t.length),a>0){for(let e=0;e<a;e++){const e=t.shift();if(void 0===e)break;w.current||(w.current=!0,C(e)),await _(e),l(e),N(e)}e=performance.now(),t.length>0?k.current=requestAnimationFrame(n):(w.current=!1,x())}else k.current=requestAnimationFrame(n)};k.current=requestAnimationFrame(n)},I=(e=!1)=>{w.current=!1,k.current&&(cancelAnimationFrame(k.current),k.current=null),e||x({manual:e})},A=()=>{const e=async(n=!1)=>{if(v.current)return void M();const r=S();if(m.current)return;w.current=!0;const t=r.shift();void 0!==t?(n&&C(t),await _(t),l(t),N(t),0!==S().length?y.current=setTimeout(e,b.current):E()):E()};e(!0)},E=()=>{w.current=!1,y.current&&(clearTimeout(y.current),y.current=null),x()},R=()=>{"requestAnimationFrame"===n?I():E()};return{start:L,restart:function(){R(),t.current.unshift(...h.current.thinking.content.split("").map(e=>({content:e,answerType:"thinking",tokenId:0,index:0}))),t.current.unshift(...h.current.answer.content.split("").map(e=>({content:e,answerType:"answer",tokenId:0,index:0}))),g(),p(),L()},stop:()=>{T.current=!0,R()},resume:function(){T.current=!1,L()},clear:function(){$()},isTyping:()=>w.current,typedIsManualStopRef:T}},y=s(({interval:r=30,onEnd:t,onStart:a,onTypedChar:s,onBeforeTypedChar:o,timerType:l="setTimeout",theme:h="light",math:p,plugins:g,disableTyping:m=!1,autoStartTyping:y=!0},f)=>{const T=c(y),v=c(!1),b=c([]),S=c(!1),C=c(0),x=c({thinking:{content:"",length:0,prevLength:0},answer:{content:"",length:0,prevLength:0},allLength:0}),[,_]=i(0),N=()=>{_(e=>e+1)},$=()=>{x.current.thinking.content="",x.current.thinking.length=0,x.current.thinking.prevLength=0,x.current.answer.content="",x.current.answer.length=0,x.current.answer.prevLength=0,x.current.allLength=0},L=k({timerType:l,interval:r,charsRef:b,onEnd:t,onStart:a,onTypedChar:s,onBeforeTypedChar:o,processCharDisplay:e=>{v.current||(v.current=!0),"thinking"===e.answerType?(x.current.thinking.content+=e.content,x.current.thinking.length+=1):(x.current.answer.content+=e.content,x.current.answer.length+=1),N()},wholeContentRef:x,disableTyping:m,triggerUpdate:N,resetWholeContent:$});u(f,()=>({push:(e,n="answer")=>{m?((e,n)=>{x.current[n].content+=e,x.current[n].prevLength=x.current[n].length,x.current[n].length+=e.length,N(),t?.({str:e,answerStr:x.current.answer.content,thinkingStr:x.current.thinking.content,manual:!1})})(e,n):((e,n)=>{0!==e.length&&(b.current.push(...e.split("").map(e=>{const r=C.current++;return{content:e,answerType:n,tokenId:0,index:r}})),x.current.allLength+=e.length,(T.current||v.current)&&(L.isTyping()||L.start()))})(e,n)},clear:()=>{L.stop(),L.typedIsManualStopRef.current=!1,b.current=[],$(),S.current=!1,C.current=0,v.current=!1,N()},start:()=>{T.current||L.start()},stop:()=>{L.stop()},resume:()=>{L.resume()},triggerWholeEnd:()=>{S.current=!0,L.isTyping()||t?.({str:x.current.answer.content,answerStr:x.current.answer.content,thinkingStr:x.current.thinking.content,manual:!0})},restart:()=>{L.restart()}}));const M=e=>{const r=x.current[e].content||"";return n("div",{className:`ds-markdown-paragraph ds-typed-${e}`,children:n(w,{theme:h,math:p,plugins:g,children:r})})};return e("div",{className:d({"ds-markdown":!0,apple:!0,"ds-markdown-dark":"dark"===h}),children:[n("div",{className:"ds-markdown-thinking",children:M("thinking")}),n("div",{className:"ds-markdown-answer",children:M("answer")})]})});g&&(y.displayName="MarkdownCMD");const f=({children:e="",answerType:r,markdownRef:s,...i})=>{const o=c(null),l=c(""),h=t(()=>"string"==typeof e?e:(g&&console.error("Markdown组件的子元素必须是一个字符串"),""),[e]);return a(()=>{if(l.current!==h){let e="";""===l.current?e=h:h.startsWith(l.current)?e=h.slice(l.current.length):(e=h,o.current.clear()),o.current.push(e,r),l.current=h}},[r,h]),u(s,()=>({stop:()=>{o.current.stop()},resume:()=>{o.current.resume()},start:()=>{o.current.start()},restart:()=>{o.current.restart()}})),n(y,{ref:o,...i})};var T=r(s((e,r)=>{const{children:t="",answerType:c="answer"}=e;if(g){if(!["thinking","answer"].includes(c))throw new Error("Markdown组件的answerType必须是thinking或answer");if("string"!=typeof t)throw new Error("Markdown组件的子元素必须是一个字符串")}return n(f,{...e,answerType:c,markdownRef:r})}));export{T as Markdown,y as MarkdownCMD,T as default};
+import { jsxs, jsx } from 'react/jsx-runtime';
+import { memo, useMemo, useRef, useEffect, forwardRef, useState, useImperativeHandle } from 'react';
+import Markdown$2 from 'react-markdown';
+import { Prism } from 'react-syntax-highlighter';
+import gfmPlugin from 'remark-gfm';
+import classNames from 'classnames';
+
+const replaceMathBracket = (value) => {
+    // 1. 提取所有块级公式内容，临时替换为占位符, [...]
+    const blockMatches = [];
+    let replaced = value.replace(/\\*\[([\s\S]+?)\\*\]/g, (_m, p1) => {
+        blockMatches.push(p1);
+        return `__BLOCK_MATH_${blockMatches.length - 1}__`;
+    });
+    // 也需要兼容 $$ xxxx $$ 这种写法
+    replaced = replaced.replace(/\$\$([\s\S]+?)\$\$/g, (_m, p1) => {
+        blockMatches.push(p1);
+        return `__BLOCK_MATH_${blockMatches.length - 1}__`;
+    });
+    // 2. 替换块级公式外部的 ( ... ) 为 $...$
+    replaced = replaced.replace(/\\*\(([^)]+?)\\*\)/g, (_m, p1) => {
+        return '$' + p1 + '$';
+    });
+    // 3. 还原块级公式内容，保持其内部小括号原样
+    replaced = replaced.replace(/__BLOCK_MATH_(\d+)__/g, (_m, idx) => {
+        return '$$' + blockMatches[Number(idx)] + '$$';
+    });
+    return replaced;
+};
+
+const BlockWrap = ({ children, language, theme = 'light' }) => {
+    return (jsxs("div", { className: `md-code-block md-code-block-${theme}`, children: [jsx("div", { className: "md-code-block-banner-wrap", children: jsx("div", { className: "md-code-block-banner md-code-block-banner-lite", children: jsx("div", { className: "md-code-block-banner-content", children: jsx("div", { className: "md-code-block-language", children: language }) }) }) }), jsx("div", { className: "md-code-block-content", children: children })] }));
+};
+
+const __DEV__ = process.env.NODE_ENV === 'development';
+/** 数学公式插件id */
+const katexId = Symbol('katex');
+
+const HighReactMarkdown = ({ theme = 'light', children: _children, math, plugins, ...props }) => {
+    const mathSplitSymbol = math?.splitSymbol ?? 'dollar';
+    const { remarkPlugins, rehypePlugins, hasKatexPlugin } = useMemo(() => {
+        let hasKatexPlugin = false;
+        const remarkPlugins = [gfmPlugin];
+        const rehypePlugins = [];
+        if (!plugins) {
+            return {
+                remarkPlugins,
+                rehypePlugins,
+            };
+        }
+        plugins.forEach((plugin) => {
+            if (plugin.id === katexId) {
+                hasKatexPlugin = true;
+                remarkPlugins.push(plugin.remarkPlugin);
+                rehypePlugins.push(plugin.rehypePlugin);
+            }
+        });
+        return {
+            remarkPlugins,
+            rehypePlugins,
+            hasKatexPlugin,
+        };
+    }, [plugins]);
+    const children = useMemo(() => {
+        /** 如果存在数学公式插件，并且数学公式分隔符为括号，则替换成 $ 符号 */
+        if (hasKatexPlugin && mathSplitSymbol === 'bracket') {
+            return replaceMathBracket(_children);
+        }
+        return _children;
+    }, [hasKatexPlugin, mathSplitSymbol, _children]);
+    return (jsx(Markdown$2, { remarkPlugins: remarkPlugins, rehypePlugins: rehypePlugins, components: {
+            code: ({ className, children, ...props }) => {
+                const match = /language-(\w+)/.exec(className || '');
+                return match ? (jsx(BlockWrap, { language: match[1], theme: theme, children: jsx(Prism, { useInlineStyles: false, language: match[1], style: {}, children: String(children).replace(/\n$/, '') }) })) : (jsx("code", { className: className, ...props, children: children }));
+            },
+            table: ({ children, ...props }) => {
+                return (jsx("div", { className: "markdown-table-wrapper", children: jsx("table", { className: "ds-markdown-table", children: children }) }));
+            },
+        }, ...props, children: children }));
+};
+var HighReactMarkdown$1 = memo(HighReactMarkdown);
+
+const useTypingTask = (options) => {
+    const { timerType = 'setTimeout', interval, charsRef, onEnd, onStart, onBeforeTypedChar, onTypedChar, processCharDisplay, wholeContentRef, disableTyping, triggerUpdate, resetWholeContent, } = options;
+    /** 是否卸载 */
+    const isUnmountRef = useRef(false);
+    /** 是否正在打字 */
+    const isTypingRef = useRef(false);
+    /** 动画帧ID */
+    const animationFrameRef = useRef(null);
+    /** 传统定时器（兼容模式） */
+    const timerRef = useRef(null);
+    // 已经打过的字记录
+    const typedCharsRef = useRef(undefined);
+    // 是否主动调用 stop 方法
+    const typedIsManualStopRef = useRef(false);
+    const disableTypingRef = useRef(disableTyping);
+    disableTypingRef.current = disableTyping;
+    const intervalRef = useRef(interval);
+    intervalRef.current = interval;
+    const getChars = () => {
+        return charsRef.current;
+    };
+    useEffect(() => {
+        isUnmountRef.current = false;
+        return () => {
+            isUnmountRef.current = true;
+            clearTimer();
+        };
+    }, []);
+    /**
+     * 触发打字开始回调
+     * @param char 当前字符
+     */
+    const triggerOnStart = (char) => {
+        if (!onStart) {
+            return;
+        }
+        const prevStr = wholeContentRef.current[char.answerType].content;
+        onStart({
+            currentIndex: prevStr.length,
+            currentChar: char.content,
+            answerType: char.answerType,
+            prevStr,
+        });
+    };
+    /**
+     * 触发打字结束回调
+     */
+    const triggerOnEnd = (data) => {
+        if (!onEnd) {
+            return;
+        }
+        onEnd({
+            str: wholeContentRef.current.answer.content,
+            answerStr: wholeContentRef.current.answer.content,
+            thinkingStr: wholeContentRef.current.thinking.content,
+            manual: data?.manual ?? false,
+        });
+    };
+    /**
+     * 触发打字过程中回调
+     * @param char 当前字符
+     */
+    const triggerOnBeforeTypedChar = async (char) => {
+        if (!onBeforeTypedChar) {
+            return;
+        }
+        const { answerType, content, index } = char;
+        const allLength = wholeContentRef.current.allLength;
+        // 计算之前字符的百分比
+        const percent = (char.index / allLength) * 100;
+        await onBeforeTypedChar({
+            currentIndex: index,
+            currentChar: content,
+            answerType: answerType,
+            prevStr: wholeContentRef.current[answerType].content,
+            percent,
+        });
+    };
+    /** 打字完成回调 */
+    const triggerOnTypedChar = async (char) => {
+        if (!onTypedChar) {
+            return;
+        }
+        const { answerType, content, index } = char;
+        const allLength = wholeContentRef.current.allLength;
+        const percent = ((char.index + 1) / allLength) * 100;
+        onTypedChar({
+            currentIndex: index,
+            currentChar: content,
+            answerType: answerType,
+            prevStr: wholeContentRef.current[answerType].content.slice(0, index),
+            currentStr: wholeContentRef.current[answerType].content,
+            percent,
+        });
+    };
+    /** 清除定时器 */
+    const clearTimer = () => {
+        // 清理 requestAnimationFrame
+        if (animationFrameRef.current) {
+            cancelAnimationFrame(animationFrameRef.current);
+            animationFrameRef.current = null;
+        }
+        // 清理 setTimeout (可能被 timestamp 模式使用)
+        if (timerRef.current) {
+            clearTimeout(timerRef.current);
+            timerRef.current = null;
+        }
+        isTypingRef.current = false;
+        typedCharsRef.current = undefined;
+    };
+    /** 开始打字任务 */
+    const startTypedTask = () => {
+        /** 如果手动调用 stop 方法，则不重新开始打字 */
+        if (typedIsManualStopRef.current) {
+            return;
+        }
+        if (isTypingRef.current) {
+            return;
+        }
+        if (timerType === 'requestAnimationFrame') {
+            startAnimationFrameMode();
+        }
+        else {
+            startTimeoutMode();
+        }
+    };
+    /** 打字机打完所有字符 */
+    async function typingRemainAll() {
+        const chars = getChars();
+        const thinkingCharsStr = chars
+            .filter((char) => char.answerType === 'thinking')
+            .map((char) => char.content)
+            .join('');
+        const answerCharsStr = chars
+            .filter((char) => char.answerType === 'answer')
+            .map((char) => char.content)
+            .join('');
+        if (thinkingCharsStr) {
+            await onBeforeTypedChar?.({
+                currentIndex: wholeContentRef.current.thinking.length,
+                currentChar: thinkingCharsStr,
+                answerType: 'thinking',
+                prevStr: wholeContentRef.current.thinking.content,
+                percent: 100,
+            });
+        }
+        if (answerCharsStr) {
+            await onBeforeTypedChar?.({
+                currentIndex: wholeContentRef.current.answer.length,
+                currentChar: answerCharsStr,
+                answerType: 'answer',
+                prevStr: wholeContentRef.current.answer.content,
+                percent: 100,
+            });
+        }
+        wholeContentRef.current.thinking.content += thinkingCharsStr;
+        wholeContentRef.current.thinking.length += thinkingCharsStr.length;
+        wholeContentRef.current.answer.content += answerCharsStr;
+        wholeContentRef.current.answer.length += answerCharsStr.length;
+        wholeContentRef.current.allLength += thinkingCharsStr.length + answerCharsStr.length;
+        charsRef.current = [];
+        isTypingRef.current = false;
+        triggerOnEnd();
+        triggerUpdate();
+    }
+    /** requestAnimationFrame 模式 */
+    const startAnimationFrameMode = () => {
+        let lastFrameTime = performance.now();
+        const frameLoop = async (currentTime) => {
+            // 如果关闭打字机效果，则打完所有字符
+            if (disableTypingRef.current) {
+                await typingRemainAll();
+                return;
+            }
+            const chars = getChars();
+            if (isUnmountRef.current)
+                return;
+            if (chars.length === 0) {
+                stopAnimationFrame();
+                return;
+            }
+            const deltaTime = currentTime - lastFrameTime;
+            let needToTypingCharsLength = Math.max(0, Math.floor(deltaTime / intervalRef.current));
+            needToTypingCharsLength = Math.min(needToTypingCharsLength, chars.length);
+            if (needToTypingCharsLength > 0) {
+                // 处理字符
+                for (let i = 0; i < needToTypingCharsLength; i++) {
+                    const char = chars.shift();
+                    if (char === undefined)
+                        break;
+                    if (!isTypingRef.current) {
+                        isTypingRef.current = true;
+                        triggerOnStart(char);
+                    }
+                    /** 打字前回调 */
+                    await triggerOnBeforeTypedChar(char);
+                    processCharDisplay(char);
+                    /** 打字完成回调 */
+                    triggerOnTypedChar(char);
+                }
+                lastFrameTime = performance.now();
+                // 继续下一帧
+                if (chars.length > 0) {
+                    animationFrameRef.current = requestAnimationFrame(frameLoop);
+                }
+                else {
+                    isTypingRef.current = false;
+                    triggerOnEnd();
+                }
+            }
+            else {
+                // 本次你不需要打字，继续下一帧
+                animationFrameRef.current = requestAnimationFrame(frameLoop);
+            }
+        };
+        animationFrameRef.current = requestAnimationFrame(frameLoop);
+    };
+    /** 停止动画帧模式 */
+    const stopAnimationFrame = (manual = false) => {
+        isTypingRef.current = false;
+        if (animationFrameRef.current) {
+            cancelAnimationFrame(animationFrameRef.current);
+            animationFrameRef.current = null;
+        }
+        if (!manual) {
+            triggerOnEnd({ manual });
+        }
+    };
+    /** setTimeout 模式 */
+    const startTimeoutMode = () => {
+        const nextTyped = () => {
+            const chars = getChars();
+            if (chars.length === 0) {
+                stopTimeout();
+                return;
+            }
+            timerRef.current = setTimeout(startTyped, intervalRef.current);
+        };
+        const startTyped = async (isStartPoint = false) => {
+            // 如果关闭打字机效果，则打完所有字符
+            if (disableTypingRef.current) {
+                typingRemainAll();
+                return;
+            }
+            const chars = getChars();
+            if (isUnmountRef.current)
+                return;
+            isTypingRef.current = true;
+            const char = chars.shift();
+            if (char === undefined) {
+                stopTimeout();
+                return;
+            }
+            if (isStartPoint) {
+                triggerOnStart(char);
+            }
+            /** 打字前回调 */
+            await triggerOnBeforeTypedChar(char);
+            processCharDisplay(char);
+            /** 打字完成回调 */
+            triggerOnTypedChar(char);
+            nextTyped();
+        };
+        startTyped(true);
+    };
+    /** 停止超时模式 */
+    const stopTimeout = () => {
+        isTypingRef.current = false;
+        if (timerRef.current) {
+            clearTimeout(timerRef.current);
+            timerRef.current = null;
+        }
+        triggerOnEnd();
+    };
+    const cancelTask = () => {
+        if (timerType === 'requestAnimationFrame') {
+            stopAnimationFrame();
+        }
+        else {
+            stopTimeout();
+        }
+    };
+    /** 暂时停止 */
+    const stopTask = () => {
+        typedIsManualStopRef.current = true;
+        cancelTask();
+    };
+    /** 停止打字任务 */
+    const endTask = () => {
+        cancelTask();
+    };
+    function restartTypedTask() {
+        endTask();
+        // 将wholeContentRef的内容放到charsRef中
+        charsRef.current.unshift(...wholeContentRef.current.thinking.content.split('').map((charUnit) => {
+            const char = {
+                content: charUnit,
+                answerType: 'thinking',
+                tokenId: 0,
+                index: 0,
+            };
+            return char;
+        }));
+        charsRef.current.unshift(...wholeContentRef.current.answer.content.split('').map((charUnit) => {
+            const char = {
+                content: charUnit,
+                answerType: 'answer',
+                tokenId: 0,
+                index: 0,
+            };
+            return char;
+        }));
+        resetWholeContent();
+        triggerUpdate();
+        startTypedTask();
+    }
+    function clear() {
+        clearTimer();
+    }
+    function resume() {
+        typedIsManualStopRef.current = false;
+        startTypedTask();
+    }
+    return {
+        start: startTypedTask,
+        restart: restartTypedTask,
+        stop: stopTask,
+        resume: resume,
+        clear: clear,
+        isTyping: () => isTypingRef.current,
+        typedIsManualStopRef,
+    };
+};
+
+const MarkdownCMD = forwardRef(({ interval = 30, onEnd, onStart, onTypedChar, onBeforeTypedChar, timerType = 'setTimeout', theme = 'light', math, plugins, disableTyping = false, autoStartTyping = true }, ref) => {
+    /** 是否自动开启打字动画, 后面发生变化将不会生效 */
+    const autoStartTypingRef = useRef(autoStartTyping);
+    /** 是否打过字 */
+    const isStartedTypingRef = useRef(false);
+    /** 当前需要打字的内容 */
+    const charsRef = useRef([]);
+    /**
+     * 打字是否已经完全结束
+     * 如果打字已经完全结束，则不会再触发打字效果
+     */
+    const isWholeTypedEndRef = useRef(false);
+    const charIndexRef = useRef(0);
+    /** 整个内容引用 */
+    const wholeContentRef = useRef({
+        thinking: {
+            content: '',
+            length: 0,
+            prevLength: 0,
+        },
+        answer: {
+            content: '',
+            length: 0,
+            prevLength: 0,
+        },
+        allLength: 0,
+    });
+    const [, setUpdate] = useState(0);
+    const triggerUpdate = () => {
+        setUpdate((prev) => prev + 1);
+    };
+    /**
+     * 处理字符显示逻辑
+     */
+    const processCharDisplay = (char) => {
+        if (!isStartedTypingRef.current) {
+            isStartedTypingRef.current = true;
+        }
+        if (char.answerType === 'thinking') {
+            wholeContentRef.current.thinking.content += char.content;
+            wholeContentRef.current.thinking.length += 1;
+        }
+        else {
+            wholeContentRef.current.answer.content += char.content;
+            wholeContentRef.current.answer.length += 1;
+        }
+        triggerUpdate();
+    };
+    const resetWholeContent = () => {
+        wholeContentRef.current.thinking.content = '';
+        wholeContentRef.current.thinking.length = 0;
+        wholeContentRef.current.thinking.prevLength = 0;
+        wholeContentRef.current.answer.content = '';
+        wholeContentRef.current.answer.length = 0;
+        wholeContentRef.current.answer.prevLength = 0;
+        wholeContentRef.current.allLength = 0;
+    };
+    // 使用新的打字任务 hook
+    const typingTask = useTypingTask({
+        timerType,
+        interval,
+        charsRef,
+        onEnd,
+        onStart,
+        onTypedChar,
+        onBeforeTypedChar,
+        processCharDisplay,
+        wholeContentRef,
+        disableTyping,
+        triggerUpdate,
+        resetWholeContent,
+    });
+    /**
+     * 内部推送处理逻辑
+     */
+    const processHasTypingPush = (content, answerType) => {
+        if (content.length === 0) {
+            return;
+        }
+        charsRef.current.push(...content.split('').map((chatStr) => {
+            const index = charIndexRef.current++;
+            const charObj = {
+                content: chatStr,
+                answerType,
+                tokenId: 0,
+                index,
+            };
+            return charObj;
+        }));
+        wholeContentRef.current.allLength += content.length;
+        // 如果关闭了自动打字， 并且没有打过字， 则不开启打字动画
+        if (!autoStartTypingRef.current && !isStartedTypingRef.current) {
+            return;
+        }
+        if (!typingTask.isTyping()) {
+            typingTask.start();
+        }
+    };
+    const processNoTypingPush = (content, answerType) => {
+        wholeContentRef.current[answerType].content += content;
+        // 记录打字前的长度
+        wholeContentRef.current[answerType].prevLength = wholeContentRef.current[answerType].length;
+        wholeContentRef.current[answerType].length += content.length;
+        triggerUpdate();
+        onEnd?.({
+            str: content,
+            answerStr: wholeContentRef.current.answer.content,
+            thinkingStr: wholeContentRef.current.thinking.content,
+            manual: false,
+        });
+    };
+    useImperativeHandle(ref, () => ({
+        /**
+         * 添加内容
+         * @param content 内容 {string}
+         * @param answerType 回答类型 {AnswerType}
+         */
+        push: (content, answerType = 'answer') => {
+            if (disableTyping) {
+                processNoTypingPush(content, answerType);
+                return;
+            }
+            processHasTypingPush(content, answerType);
+        },
+        /**
+         * 清除打字任务
+         */
+        clear: () => {
+            typingTask.stop();
+            typingTask.typedIsManualStopRef.current = false;
+            charsRef.current = [];
+            resetWholeContent();
+            isWholeTypedEndRef.current = false;
+            charIndexRef.current = 0;
+            isStartedTypingRef.current = false;
+            triggerUpdate();
+        },
+        /** 开启打字，只有在关闭了自动打字才生效 */
+        start: () => {
+            if (!autoStartTypingRef.current) {
+                typingTask.start();
+            }
+        },
+        /** 停止打字任务 */
+        stop: () => {
+            typingTask.stop();
+        },
+        /** 重新开始打字任务 */
+        resume: () => {
+            typingTask.resume();
+        },
+        /**
+         * 主动触发打字结束
+         */
+        triggerWholeEnd: () => {
+            isWholeTypedEndRef.current = true;
+            if (!typingTask.isTyping()) {
+                // 这里需要手动触发结束回调，因为 hook 中的 triggerOnEnd 不能直接调用
+                onEnd?.({
+                    str: wholeContentRef.current.answer.content,
+                    answerStr: wholeContentRef.current.answer.content,
+                    thinkingStr: wholeContentRef.current.thinking.content,
+                    manual: true,
+                });
+            }
+        },
+        /** 重新开始打字任务 */
+        restart: () => {
+            typingTask.restart();
+        },
+    }));
+    const getParagraphs = (answerType) => {
+        const content = wholeContentRef.current[answerType].content || '';
+        return (jsx("div", { className: `ds-markdown-paragraph ds-typed-${answerType}`, children: jsx(HighReactMarkdown$1, { theme: theme, math: math, plugins: plugins, children: content }) }));
+    };
+    return (jsxs("div", { className: classNames({
+            'ds-markdown': true,
+            apple: true,
+            'ds-markdown-dark': theme === 'dark',
+        }), children: [jsx("div", { className: "ds-markdown-thinking", children: getParagraphs('thinking') }), jsx("div", { className: "ds-markdown-answer", children: getParagraphs('answer') })] }));
+});
+if (__DEV__) {
+    MarkdownCMD.displayName = 'MarkdownCMD';
+}
+
+const MarkdownInner = ({ children: _children = '', answerType, markdownRef, ...rest }) => {
+    const cmdRef = useRef(null);
+    const prefixRef = useRef('');
+    const content = useMemo(() => {
+        if (typeof _children === 'string') {
+            return _children;
+        }
+        if (__DEV__) {
+            console.error('Markdown组件的子元素必须是一个字符串');
+        }
+        return '';
+    }, [_children]);
+    useEffect(() => {
+        if (prefixRef.current !== content) {
+            let newContent = '';
+            if (prefixRef.current === '') {
+                newContent = content;
+            }
+            else {
+                if (content.startsWith(prefixRef.current)) {
+                    newContent = content.slice(prefixRef.current.length);
+                }
+                else {
+                    newContent = content;
+                    cmdRef.current.clear();
+                }
+            }
+            cmdRef.current.push(newContent, answerType);
+            prefixRef.current = content;
+        }
+    }, [answerType, content]);
+    useImperativeHandle(markdownRef, () => ({
+        stop: () => {
+            cmdRef.current.stop();
+        },
+        resume: () => {
+            cmdRef.current.resume();
+        },
+        start: () => {
+            cmdRef.current.start();
+        },
+        restart: () => {
+            cmdRef.current.restart();
+        },
+    }));
+    return jsx(MarkdownCMD, { ref: cmdRef, ...rest });
+};
+const Markdown = forwardRef((props, ref) => {
+    const { children = '', answerType = 'answer' } = props;
+    if (__DEV__) {
+        if (!['thinking', 'answer'].includes(answerType)) {
+            throw new Error('Markdown组件的answerType必须是thinking或answer');
+        }
+        if (typeof children !== 'string') {
+            throw new Error('Markdown组件的子元素必须是一个字符串');
+        }
+    }
+    return jsx(MarkdownInner, { ...props, answerType: answerType, markdownRef: ref });
+});
+var Markdown$1 = memo(Markdown);
+
+export { Markdown$1 as Markdown, MarkdownCMD, Markdown$1 as default };
 //# sourceMappingURL=index.js.map