@4399ywkf/design 2.0.0 → 2.1.1

package/dist/AiConversation/_lobe/hooks/useNativeButton.js

@@ -0,0 +1,121 @@
+function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
+import { isValidElement, useMemo } from 'react';
+/**
+ * Map of component displayNames to their nativeButton values.
+ * Components that render native <button> elements should be true,
+ * components that render non-button elements should be false.
+ */
+var NATIVE_BUTTON_MAP = {
+  A: false,
+  ActionIcon: false,
+  ActionIconGroup: false,
+  Alert: false,
+  Avatar: false,
+  AvatarGroup: false,
+  Block: false,
+  BottomGradientButton: true,
+  Burger: false,
+  Button: true,
+  Center: false,
+  Checkbox: false,
+  CheckboxGroup: false,
+  Collapse: false,
+  ColorSwatches: false,
+  CopyButton: false,
+  DownloadButton: false,
+  EditableText: false,
+  Empty: false,
+  FileTypeIcon: false,
+  Flexbox: false,
+  FluentEmoji: false,
+  GradientButton: true,
+  Highlighter: false,
+  Hotkey: false,
+  Icon: false,
+  Image: false,
+  Img: false,
+  Input: false,
+  InputNumber: false,
+  InputPassword: false,
+  List: false,
+  ListItem: false,
+  Select: false,
+  Switch: false,
+  Markdown: false,
+  MaterialFileTypeIcon: false,
+  Segmented: false,
+  Skeleton: false,
+  SkeletonAvatar: false,
+  SkeletonBlock: false,
+  SkeletonButton: false,
+  SkeletonParagraph: false,
+  SkeletonTags: false,
+  SkeletonTitle: false,
+  Snippet: false,
+  Tag: false,
+  Text: false,
+  TextArea: false,
+  ThemeSwitch: false,
+  Video: false
+};
+
+/**
+ * Get the displayName of a React component from an element.
+ * Handles function components, forwardRef, memo, etc.
+ */
+function getComponentDisplayName(element) {
+  var type = element.type;
+  if (typeof type === 'string') return undefined;
+  if (typeof type === 'function') {
+    return type.displayName || type.name;
+  }
+  if (_typeof(type) === 'object' && type !== null) {
+    var _render, _render2, _type, _type2;
+    // Handle forwardRef, memo, etc.
+    var displayName = type.displayName || ((_render = type.render) === null || _render === void 0 ? void 0 : _render.displayName) || ((_render2 = type.render) === null || _render2 === void 0 ? void 0 : _render2.name) || ((_type = type.type) === null || _type === void 0 ? void 0 : _type.displayName) || ((_type2 = type.type) === null || _type2 === void 0 ? void 0 : _type2.name);
+    return displayName;
+  }
+  return undefined;
+}
+
+/**
+ * Hook to resolve nativeButton prop for Base UI trigger components.
+ *
+ * When using `render`, Base UI expects the rendered element to be a native <button> by default.
+ * If we can infer it's not, we opt out to avoid warnings (users can still override via `nativeButton`).
+ */
+export function useNativeButton(_ref) {
+  var children = _ref.children,
+    nativeButton = _ref.nativeButton,
+    triggerNativeButton = _ref.triggerNativeButton;
+  var isNativeButtonTriggerElement = useMemo(function () {
+    if (! /*#__PURE__*/isValidElement(children)) return false;
+    return typeof children.type === 'string' && children.type === 'button';
+  }, [children]);
+  var resolvedNativeButton = useMemo(function () {
+    // User-provided nativeButton takes highest priority
+    if (nativeButton !== undefined) return nativeButton;
+    // Trigger props nativeButton (for DropdownMenu) takes second priority
+    if (triggerNativeButton !== undefined) return triggerNativeButton;
+    // If it's a native button element, return true
+    if (isNativeButtonTriggerElement) return true;
+    // If children is not a valid element, let Base UI decide
+    if (! /*#__PURE__*/isValidElement(children)) return undefined;
+    // If it's a string type but not a button (e.g., 'div'), return false
+    if (typeof children.type === 'string') return false;
+
+    // Check if it's a known component from the library
+    var displayName = getComponentDisplayName(children);
+    if (displayName && displayName in NATIVE_BUTTON_MAP) {
+      return NATIVE_BUTTON_MAP[displayName];
+    }
+
+    // For unknown React components, default to false to avoid warnings
+    // since most custom components don't render native buttons
+    return false;
+  }, [children, isNativeButtonTriggerElement, nativeButton, triggerNativeButton]);
+  return {
+    isNativeButtonTriggerElement: isNativeButtonTriggerElement,
+    resolvedNativeButton: resolvedNativeButton
+  };
+}

package/dist/AiConversation/_lobe/hooks/useStreamHighlight.d.ts

@@ -0,0 +1,8 @@
+import { type BuiltinTheme } from 'shiki';
+import { type StreamingHighlightResult } from './useHighlight';
+export declare const useStreamHighlight: (text: string, { language, theme: builtinTheme, streaming, }: {
+    enableTransformer?: boolean | undefined;
+    language: string;
+    streaming?: boolean | undefined;
+    theme?: import("shiki/dist/themes-oX7Feg4r.mjs").t | undefined;
+}) => StreamingHighlightResult | undefined;

package/dist/AiConversation/_lobe/hooks/useStreamHighlight.js

@@ -0,0 +1,414 @@
+'use client';
+
+function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
+function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return e; }; var t, e = {}, r = Object.prototype, n = r.hasOwnProperty, o = Object.defineProperty || function (t, e, r) { t[e] = r.value; }, i = "function" == typeof Symbol ? Symbol : {}, a = i.iterator || "@@iterator", c = i.asyncIterator || "@@asyncIterator", u = i.toStringTag || "@@toStringTag"; function define(t, e, r) { return Object.defineProperty(t, e, { value: r, enumerable: !0, configurable: !0, writable: !0 }), t[e]; } try { define({}, ""); } catch (t) { define = function define(t, e, r) { return t[e] = r; }; } function wrap(t, e, r, n) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype), c = new Context(n || []); return o(a, "_invoke", { value: makeInvokeMethod(t, r, c) }), a; } function tryCatch(t, e, r) { try { return { type: "normal", arg: t.call(e, r) }; } catch (t) { return { type: "throw", arg: t }; } } e.wrap = wrap; var h = "suspendedStart", l = "suspendedYield", f = "executing", s = "completed", y = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var p = {}; define(p, a, function () { return this; }); var d = Object.getPrototypeOf, v = d && d(d(values([]))); v && v !== r && n.call(v, a) && (p = v); var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p); function defineIteratorMethods(t) { ["next", "throw", "return"].forEach(function (e) { define(t, e, function (t) { return this._invoke(e, t); }); }); } function AsyncIterator(t, e) { function invoke(r, o, i, a) { var c = tryCatch(t[r], t, o); if ("throw" !== c.type) { var u = c.arg, h = u.value; return h && "object" == _typeof(h) && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) { invoke("next", t, i, a); }, function (t) { invoke("throw", t, i, a); }) : e.resolve(h).then(function (t) { u.value = t, i(u); }, function (t) { return invoke("throw", t, i, a); }); } a(c.arg); } var r; o(this, "_invoke", { value: function value(t, n) { function callInvokeWithMethodAndArg() { return new e(function (e, r) { invoke(t, n, e, r); }); } return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(e, r, n) { var o = h; return function (i, a) { if (o === f) throw new Error("Generator is already running"); if (o === s) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var c = n.delegate; if (c) { var u = maybeInvokeDelegate(c, n); if (u) { if (u === y) continue; return u; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (o === h) throw o = s, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = f; var p = tryCatch(e, r, n); if ("normal" === p.type) { if (o = n.done ? s : l, p.arg === y) continue; return { value: p.arg, done: n.done }; } "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg); } }; } function maybeInvokeDelegate(e, r) { var n = r.method, o = e.iterator[n]; if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y; var i = tryCatch(o, e.iterator, r.arg); if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y; var a = i.arg; return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y); } function pushTryEntry(t) { var e = { tryLoc: t[0] }; 1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e); } function resetTryEntry(t) { var e = t.completion || {}; e.type = "normal", delete e.arg, t.completion = e; } function Context(t) { this.tryEntries = [{ tryLoc: "root" }], t.forEach(pushTryEntry, this), this.reset(!0); } function values(e) { if (e || "" === e) { var r = e[a]; if (r) return r.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) { var o = -1, i = function next() { for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next; return next.value = t, next.done = !0, next; }; return i.next = i; } } throw new TypeError(_typeof(e) + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), o(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) { var e = "function" == typeof t && t.constructor; return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name)); }, e.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t; }, e.awrap = function (t) { return { __await: t }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () { return this; }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(wrap(t, r, n, o), i); return e.isGeneratorFunction(r) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () { return this; }), define(g, "toString", function () { return "[object Generator]"; }), e.keys = function (t) { var e = Object(t), r = []; for (var n in e) r.push(n); return r.reverse(), function next() { for (; r.length;) { var t = r.pop(); if (t in e) return next.value = t, next.done = !1, next; } return next.done = !0, next; }; }, e.values = values, Context.prototype = { constructor: Context, reset: function reset(e) { if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0].completion; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(e) { if (this.done) throw e; var r = this; function handle(n, o) { return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o; } for (var o = this.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i.completion; if ("root" === i.tryLoc) return handle("end"); if (i.tryLoc <= this.prev) { var c = n.call(i, "catchLoc"), u = n.call(i, "finallyLoc"); if (c && u) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } else if (c) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); } else { if (!u) throw new Error("try statement without catch or finally"); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } } } }, abrupt: function abrupt(t, e) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var o = this.tryEntries[r]; if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) { var i = o; break; } } i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null); var a = i ? i.completion : {}; return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a); }, complete: function complete(t, e) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y; }, finish: function finish(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y; } }, catch: function _catch(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.tryLoc === t) { var n = r.completion; if ("throw" === n.type) { var o = n.arg; resetTryEntry(r); } return o; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(e, r, n) { return this.delegate = { iterator: values(e), resultName: r, nextLoc: n }, "next" === this.method && (this.arg = t), y; } }, e; }
+function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
+function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
+function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
+function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
+function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
+function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
+function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
+function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
+function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); }
+function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
+function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
+function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
+function _createForOfIteratorHelper(o, allowArrayLike) { var it = typeof Symbol !== "undefined" && o[Symbol.iterator] || o["@@iterator"]; if (!it) { if (Array.isArray(o) || (it = _unsupportedIterableToArray(o)) || allowArrayLike && o && typeof o.length === "number") { if (it) o = it; var i = 0; var F = function F() {}; return { s: F, n: function n() { if (i >= o.length) return { done: true }; return { done: false, value: o[i++] }; }, e: function e(_e) { throw _e; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var normalCompletion = true, didErr = false, err; return { s: function s() { it = it.call(o); }, n: function n() { var step = it.next(); normalCompletion = step.done; return step; }, e: function e(_e2) { didErr = true; err = _e2; }, f: function f() { try { if (!normalCompletion && it.return != null) it.return(); } finally { if (didErr) throw err; } } }; }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
+import { ShikiStreamTokenizer } from 'shiki-stream';
+import { getCodeLanguageByInput } from "../Highlighter/const";
+import lobeTheme from "../Highlighter/theme/lobe-theme";
+import { shikiModulePromise } from "./useHighlight";
+// Optimized version: reduce array allocations and object spreading
+var tokensToLineTokens = function tokensToLineTokens(tokens) {
+  if (!tokens.length) return [[]];
+  var lines = [];
+  var currentLine = [];
+  var _iterator = _createForOfIteratorHelper(tokens),
+    _step;
+  try {
+    for (_iterator.s(); !(_step = _iterator.n()).done;) {
+      var _token$content;
+      var token = _step.value;
+      var content = (_token$content = token.content) !== null && _token$content !== void 0 ? _token$content : '';
+      if (content === '\n') {
+        lines.push(currentLine);
+        currentLine = [];
+        continue;
+      }
+      var newlineIndex = content.indexOf('\n');
+      if (newlineIndex === -1) {
+        // No newline, add token directly
+        currentLine.push(token);
+      } else {
+        // Split on newlines
+        var segments = content.split('\n');
+        var _iterator2 = _createForOfIteratorHelper(segments.entries()),
+          _step2;
+        try {
+          for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
+            var _step2$value = _slicedToArray(_step2.value, 2),
+              j = _step2$value[0],
+              segment = _step2$value[1];
+            if (segment) {
+              // Only create new object if we need to modify content
+              currentLine.push(j === 0 && segment === content ? token : _objectSpread(_objectSpread({}, token), {}, {
+                content: segment
+              }));
+            }
+            if (j < segments.length - 1) {
+              lines.push(currentLine);
+              currentLine = [];
+            }
+          }
+        } catch (err) {
+          _iterator2.e(err);
+        } finally {
+          _iterator2.f();
+        }
+      }
+    }
+
+    // Don't forget the last line
+  } catch (err) {
+    _iterator.e(err);
+  } finally {
+    _iterator.f();
+  }
+  if (currentLine.length > 0 || lines.length === 0) {
+    lines.push(currentLine);
+  }
+  return lines.length > 0 ? lines : [[]];
+};
+var createPreStyle = function createPreStyle(bg, fg) {
+  if (!bg && !fg) return undefined;
+  return {
+    backgroundColor: bg,
+    color: fg
+  };
+};
+var useStreamingHighlighter = function useStreamingHighlighter(text, options) {
+  var customThemes = options.customThemes,
+    enabled = options.enabled,
+    language = options.language,
+    theme = options.theme;
+  var _useState = useState(),
+    _useState2 = _slicedToArray(_useState, 2),
+    result = _useState2[0],
+    setResult = _useState2[1];
+  var tokenizerRef = useRef(null);
+  var previousTextRef = useRef('');
+  var safeText = text !== null && text !== void 0 ? text : '';
+  var latestTextRef = useRef(safeText);
+  var preStyleRef = useRef(undefined);
+  var linesRef = useRef([[]]);
+  useEffect(function () {
+    latestTextRef.current = safeText;
+  }, [safeText]);
+
+  // Use ref to store callback to avoid recreating it
+  var setStreamingResultRef = useRef(function (rawLines) {
+    var previousLines = linesRef.current;
+    var newLinesLength = rawLines.length;
+    var prevLinesLength = previousLines.length;
+
+    // Fast path: if lengths differ or it's a complete reset, use new lines directly
+    if (newLinesLength !== prevLinesLength || newLinesLength === 0) {
+      linesRef.current = rawLines;
+      setResult({
+        lines: rawLines,
+        preStyle: preStyleRef.current
+      });
+      return;
+    }
+
+    // Optimized comparison: only check changed lines
+    var hasChanges = false;
+    var mergedLines = [];
+    for (var i = 0; i < newLinesLength; i++) {
+      var newLine = rawLines[i];
+      var prevLine = previousLines[i];
+
+      // Quick reference equality check first
+      if (prevLine === newLine) {
+        mergedLines[i] = prevLine;
+        continue;
+      }
+
+      // Length check
+      if (!prevLine || prevLine.length !== newLine.length) {
+        mergedLines[i] = newLine;
+        hasChanges = true;
+        continue;
+      }
+
+      // Deep comparison only for lines that might have changed
+      var lineChanged = false;
+      var _iterator3 = _createForOfIteratorHelper(newLine.entries()),
+        _step3;
+      try {
+        for (_iterator3.s(); !(_step3 = _iterator3.n()).done;) {
+          var _step3$value = _slicedToArray(_step3.value, 2),
+            j = _step3$value[0],
+            newToken = _step3$value[1];
+          if (prevLine[j] !== newToken) {
+            lineChanged = true;
+            break;
+          }
+        }
+      } catch (err) {
+        _iterator3.e(err);
+      } finally {
+        _iterator3.f();
+      }
+      if (lineChanged) {
+        mergedLines[i] = newLine;
+        hasChanges = true;
+      } else {
+        mergedLines[i] = prevLine;
+      }
+    }
+
+    // Only update state if there are actual changes
+    if (hasChanges) {
+      linesRef.current = mergedLines;
+      setResult({
+        lines: mergedLines,
+        preStyle: preStyleRef.current
+      });
+    }
+  });
+  var updateTokens = useCallback( /*#__PURE__*/function () {
+    var _ref = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(nextText) {
+      var forceReset,
+        tokenizer,
+        previousText,
+        chunk,
+        canAppend,
+        stableTokens,
+        unstableTokens,
+        totalLength,
+        mergedTokens,
+        _stableTokens,
+        _unstableTokens,
+        _mergedTokens,
+        _args = arguments;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            forceReset = _args.length > 1 && _args[1] !== undefined ? _args[1] : false;
+            tokenizer = tokenizerRef.current;
+            if (tokenizer) {
+              _context.next = 4;
+              break;
+            }
+            return _context.abrupt("return");
+          case 4:
+            if (forceReset) {
+              tokenizer.clear();
+              previousTextRef.current = '';
+            }
+            previousText = previousTextRef.current;
+            chunk = nextText;
+            canAppend = !forceReset && nextText.startsWith(previousText);
+            if (canAppend) {
+              chunk = nextText.slice(previousText.length);
+            } else if (!forceReset) {
+              tokenizer.clear();
+            }
+            previousTextRef.current = nextText;
+            if (chunk) {
+              _context.next = 20;
+              break;
+            }
+            // Optimize: avoid array spread if possible
+            stableTokens = tokenizer.tokensStable;
+            unstableTokens = tokenizer.tokensUnstable;
+            totalLength = stableTokens.length + unstableTokens.length;
+            if (!(totalLength === 0)) {
+              _context.next = 17;
+              break;
+            }
+            setStreamingResultRef.current([[]]);
+            return _context.abrupt("return");
+          case 17:
+            // Only create merged array if we have both stable and unstable tokens
+            mergedTokens = stableTokens.length === 0 ? unstableTokens : unstableTokens.length === 0 ? stableTokens : [].concat(_toConsumableArray(stableTokens), _toConsumableArray(unstableTokens));
+            setStreamingResultRef.current(tokensToLineTokens(mergedTokens));
+            return _context.abrupt("return");
+          case 20:
+            _context.prev = 20;
+            _context.next = 23;
+            return tokenizer.enqueue(chunk);
+          case 23:
+            // Optimize: avoid array spread if possible
+            _stableTokens = tokenizer.tokensStable;
+            _unstableTokens = tokenizer.tokensUnstable;
+            _mergedTokens = _stableTokens.length === 0 ? _unstableTokens : _unstableTokens.length === 0 ? _stableTokens : [].concat(_toConsumableArray(_stableTokens), _toConsumableArray(_unstableTokens));
+            setStreamingResultRef.current(tokensToLineTokens(_mergedTokens));
+            _context.next = 32;
+            break;
+          case 29:
+            _context.prev = 29;
+            _context.t0 = _context["catch"](20);
+            console.error('Streaming highlighting failed:', _context.t0);
+          case 32:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, null, [[20, 29]]);
+    }));
+    return function (_x) {
+      return _ref.apply(this, arguments);
+    };
+  }(), []);
+
+  // Cache highlighter key to avoid unnecessary recreations
+  var highlighterKeyRef = useRef('');
+  useEffect(function () {
+    if (!enabled) {
+      var _tokenizerRef$current;
+      (_tokenizerRef$current = tokenizerRef.current) === null || _tokenizerRef$current === void 0 || _tokenizerRef$current.clear();
+      tokenizerRef.current = null;
+      previousTextRef.current = '';
+      preStyleRef.current = undefined;
+      linesRef.current = [[]];
+      setResult(undefined);
+      highlighterKeyRef.current = '';
+      return;
+    }
+
+    // Skip if language/theme combination hasn't changed
+    var currentKey = "".concat(language, "-").concat(theme);
+    if (highlighterKeyRef.current === currentKey && tokenizerRef.current) {
+      return;
+    }
+    var cancelled = false;
+    _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2() {
+      var mod, themesToLoad, customTheme, highlighter, _tokenizerRef$current2, tokenizer, themeInfo, currentText;
+      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
+          case 0:
+            _context2.next = 2;
+            return shikiModulePromise;
+          case 2:
+            mod = _context2.sent;
+            if (!(!mod || cancelled)) {
+              _context2.next = 5;
+              break;
+            }
+            return _context2.abrupt("return");
+          case 5:
+            _context2.prev = 5;
+            // Load custom theme if using slack-dark or slack-ochin
+            themesToLoad = [theme];
+            if (customThemes && theme === 'lobe-theme') {
+              customTheme = customThemes[theme];
+              if (customTheme) {
+                themesToLoad = [customTheme];
+              }
+            }
+
+            // Only load the specific language and theme needed
+            // getSingletonHighlighter will cache the instance internally
+            _context2.next = 10;
+            return mod.getSingletonHighlighter({
+              langs: language ? [language] : ['plaintext'],
+              themes: themesToLoad
+            });
+          case 10:
+            highlighter = _context2.sent;
+            if (!(!highlighter || cancelled)) {
+              _context2.next = 13;
+              break;
+            }
+            return _context2.abrupt("return");
+          case 13:
+            // Only create new tokenizer if key changed
+            if (highlighterKeyRef.current !== currentKey) {
+              // Clear old tokenizer
+              (_tokenizerRef$current2 = tokenizerRef.current) === null || _tokenizerRef$current2 === void 0 || _tokenizerRef$current2.clear();
+              tokenizer = new ShikiStreamTokenizer({
+                highlighter: highlighter,
+                lang: language,
+                theme: theme
+              });
+              tokenizerRef.current = tokenizer;
+              highlighterKeyRef.current = currentKey;
+              previousTextRef.current = '';
+              linesRef.current = [[]];
+              themeInfo = highlighter.getTheme(theme);
+              preStyleRef.current = createPreStyle(themeInfo === null || themeInfo === void 0 ? void 0 : themeInfo.bg, themeInfo === null || themeInfo === void 0 ? void 0 : themeInfo.fg);
+            }
+            currentText = latestTextRef.current;
+            if (!currentText) {
+              _context2.next = 20;
+              break;
+            }
+            _context2.next = 18;
+            return updateTokens(currentText, true);
+          case 18:
+            _context2.next = 21;
+            break;
+          case 20:
+            setStreamingResultRef.current([[]]);
+          case 21:
+            _context2.next = 27;
+            break;
+          case 23:
+            _context2.prev = 23;
+            _context2.t0 = _context2["catch"](5);
+            console.error('Streaming highlighter initialization failed:', _context2.t0);
+            // Reset on error
+            highlighterKeyRef.current = '';
+          case 27:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2, null, [[5, 23]]);
+    }))();
+    return function () {
+      cancelled = true;
+      // Cleanup only if this effect was cancelled before completion
+      // The next effect will handle cleanup if key changed
+    };
+  }, [enabled, language, theme, updateTokens, customThemes]);
+
+  // Separate effect for text updates to avoid unnecessary tokenizer recreation
+  useEffect(function () {
+    if (!enabled) return;
+    if (!tokenizerRef.current) return;
+    // Use ref to get latest text to avoid stale closures
+    var currentText = latestTextRef.current;
+    updateTokens(currentText);
+  }, [enabled, safeText, updateTokens]);
+  return result;
+};
+export var useStreamHighlight = function useStreamHighlight(text, _ref3) {
+  var language = _ref3.language,
+    builtinTheme = _ref3.theme,
+    streaming = _ref3.streaming;
+  // Safely handle language and text with boundary checks
+  var safeText = text !== null && text !== void 0 ? text : '';
+  var lang = (language !== null && language !== void 0 ? language : 'plaintext').toLowerCase();
+
+  // Match supported languages
+  var matchedLanguage = useMemo(function () {
+    return getCodeLanguageByInput(lang);
+  }, [lang]);
+  var effectiveTheme = builtinTheme || 'lobe-theme';
+  return useStreamingHighlighter(safeText, {
+    customThemes: {
+      'lobe-theme': lobeTheme
+    },
+    enabled: streaming,
+    language: matchedLanguage,
+    theme: effectiveTheme
+  });
+};

package/dist/AiConversation/_lobe/hooks/useStreamMermaid.d.ts

@@ -0,0 +1,9 @@
+import type { MermaidConfig } from 'mermaid';
+/**
+ * 流式 Mermaid 渲染 - 支持内容逐步更新
+ */
+export declare const useStreamMermaid: (content: string, { enabled, id, theme: customTheme, }: {
+    enabled?: boolean | undefined;
+    id: string;
+    theme?: MermaidConfig['theme'];
+}) => string;