@lobehub/ui 2.15.4 → 2.15.5

package/es/hooks/useHighlight.js

@@ -2,11 +2,27 @@
 
 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 _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 _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; }
+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 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); }
 import { transformerNotationDiff, transformerNotationErrorLevel, transformerNotationFocus, transformerNotationHighlight, transformerNotationWordHighlight } from '@shikijs/transformers';
 import { useTheme, useThemeMode } from 'antd-style';
-import { useMemo } from 'react';
+import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
+import { ShikiStreamTokenizer } from 'shiki-stream';
 import useSWR from 'swr';
 import { Md5 } from 'ts-md5';
 import { getCodeLanguageByInput } from "../Highlighter/const";
@@ -17,10 +33,15 @@ var MD5_LENGTH_THRESHOLD = 10000; // Use async MD5 for text exceeding this lengt
 // Color replacement mapping type
 
 // Lazy load shiki
-var loadShiki = function loadShiki() {
+var loadShikiModule = function loadShikiModule() {
   if (typeof window === 'undefined') return Promise.resolve(null);
-  return import('shiki').then(function (mod) {
-    return mod.codeToHtml;
+  return import('shiki');
+};
+var shikiModulePromise = loadShikiModule();
+var loadShiki = function loadShiki() {
+  return shikiModulePromise.then(function (mod) {
+    var _mod$codeToHtml;
+    return (_mod$codeToHtml = mod === null || mod === void 0 ? void 0 : mod.codeToHtml) !== null && _mod$codeToHtml !== void 0 ? _mod$codeToHtml : null;
   });
 };
 var shikiPromise = loadShiki();
@@ -29,12 +50,248 @@ var shikiPromise = loadShiki();
 var escapeHtml = function escapeHtml(str) {
   return str.replaceAll('&', '&amp;').replaceAll('<', '&lt;').replaceAll('>', '&gt;').replaceAll('"', '&quot;').replaceAll("'", '&#039;');
 };
+var tokensToLineTokens = function tokensToLineTokens(tokens) {
+  if (!tokens.length) return [[]];
+  var lines = [[]];
+  var currentLine = lines[0];
+  var startNewLine = function startNewLine() {
+    currentLine = [];
+    lines.push(currentLine);
+  };
+  tokens.forEach(function (token) {
+    var _token$content;
+    var content = (_token$content = token.content) !== null && _token$content !== void 0 ? _token$content : '';
+    if (content === '\n') {
+      startNewLine();
+      return;
+    }
+    if (!content.includes('\n')) {
+      currentLine.push(token);
+      return;
+    }
+    var segments = content.split('\n');
+    segments.forEach(function (segment, index) {
+      if (segment) {
+        currentLine.push(_objectSpread(_objectSpread({}, token), {}, {
+          content: segment
+        }));
+      }
+      if (index < segments.length - 1) {
+        startNewLine();
+      }
+    });
+  });
+  if (lines.length === 0) return [[]];
+  return lines;
+};
+var createPreStyle = function createPreStyle(bg, fg) {
+  if (!bg && !fg) return undefined;
+  return {
+    backgroundColor: bg,
+    color: fg
+  };
+};
+var useStreamingHighlighter = function useStreamingHighlighter(text, options) {
+  var colorReplacements = options.colorReplacements,
+    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 latestTextRef = useRef(text);
+  var preStyleRef = useRef(undefined);
+  var colorReplacementsRef = useRef(colorReplacements);
+  var linesRef = useRef([[]]);
+  useEffect(function () {
+    latestTextRef.current = text;
+  }, [text]);
+  useEffect(function () {
+    colorReplacementsRef.current = colorReplacements;
+  }, [colorReplacements]);
+  var setStreamingResult = useCallback(function (rawLines) {
+    var previousLines = linesRef.current;
+    var mergedLines = rawLines.map(function (line, index) {
+      var previousLine = previousLines[index];
+      if (previousLine && previousLine.length === line.length && previousLine.every(function (token, tokenIndex) {
+        return token === line[tokenIndex];
+      })) {
+        return previousLine;
+      }
+      return line;
+    });
+    linesRef.current = mergedLines;
+    setResult({
+      colorReplacements: colorReplacementsRef.current,
+      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,
+        mergedTokens,
+        _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 = 14;
+              break;
+            }
+            mergedTokens = [].concat(_toConsumableArray(tokenizer.tokensStable), _toConsumableArray(tokenizer.tokensUnstable));
+            setStreamingResult(mergedTokens.length ? tokensToLineTokens(mergedTokens) : [[]]);
+            return _context.abrupt("return");
+          case 14:
+            _context.prev = 14;
+            _context.next = 17;
+            return tokenizer.enqueue(chunk);
+          case 17:
+            _mergedTokens = [].concat(_toConsumableArray(tokenizer.tokensStable), _toConsumableArray(tokenizer.tokensUnstable));
+            setStreamingResult(tokensToLineTokens(_mergedTokens));
+            _context.next = 24;
+            break;
+          case 21:
+            _context.prev = 21;
+            _context.t0 = _context["catch"](14);
+            console.error('Streaming highlighting failed:', _context.t0);
+          case 24:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, null, [[14, 21]]);
+    }));
+    return function (_x) {
+      return _ref.apply(this, arguments);
+    };
+  }(), [setStreamingResult]);
+  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);
+      return;
+    }
+    var cancelled = false;
+    _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2() {
+      var mod, highlighter, 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;
+            _context2.next = 8;
+            return mod.getSingletonHighlighter({
+              langs: language ? [language] : [],
+              themes: [theme]
+            });
+          case 8:
+            highlighter = _context2.sent;
+            if (!(!highlighter || cancelled)) {
+              _context2.next = 11;
+              break;
+            }
+            return _context2.abrupt("return");
+          case 11:
+            tokenizer = new ShikiStreamTokenizer({
+              highlighter: highlighter,
+              lang: language,
+              theme: theme
+            });
+            tokenizerRef.current = tokenizer;
+            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 = 23;
+              break;
+            }
+            _context2.next = 21;
+            return updateTokens(currentText, true);
+          case 21:
+            _context2.next = 24;
+            break;
+          case 23:
+            setStreamingResult([[]]);
+          case 24:
+            _context2.next = 29;
+            break;
+          case 26:
+            _context2.prev = 26;
+            _context2.t0 = _context2["catch"](5);
+            console.error('Streaming highlighter initialization failed:', _context2.t0);
+          case 29:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2, null, [[5, 26]]);
+    }))();
+    return function () {
+      var _tokenizerRef$current2;
+      cancelled = true;
+      (_tokenizerRef$current2 = tokenizerRef.current) === null || _tokenizerRef$current2 === void 0 || _tokenizerRef$current2.clear();
+      tokenizerRef.current = null;
+      previousTextRef.current = '';
+    };
+  }, [enabled, language, setStreamingResult, theme, updateTokens]);
+  useEffect(function () {
+    if (!enabled) return;
+    if (!tokenizerRef.current) return;
+    updateTokens(text);
+  }, [enabled, text, updateTokens]);
+  return result;
+};
 
 // Main highlight component
-export var useHighlight = function useHighlight(text, _ref) {
-  var language = _ref.language,
-    enableTransformer = _ref.enableTransformer,
-    builtinTheme = _ref.theme;
+export var useHighlight = function useHighlight(text, _ref3) {
+  var language = _ref3.language,
+    enableTransformer = _ref3.enableTransformer,
+    builtinTheme = _ref3.theme,
+    streaming = _ref3.streaming;
   var _useThemeMode = useThemeMode(),
     isDarkMode = _useThemeMode.isDarkMode;
   var theme = useTheme();
@@ -88,23 +345,23 @@ export var useHighlight = function useHighlight(text, _ref) {
   }, [text, matchedLanguage, isDarkMode, builtinTheme]);
 
   // Use SWR to get highlighted HTML
-  return useSWR(cacheKey, /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
+  var response = useSWR(streaming ? null : cacheKey, /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3() {
     var codeToHtml, html, _codeToHtml, _html, fallbackHtml;
-    return _regeneratorRuntime().wrap(function _callee$(_context) {
-      while (1) switch (_context.prev = _context.next) {
+    return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+      while (1) switch (_context3.prev = _context3.next) {
         case 0:
-          _context.prev = 0;
-          _context.next = 3;
+          _context3.prev = 0;
+          _context3.next = 3;
           return shikiPromise;
         case 3:
-          codeToHtml = _context.sent;
+          codeToHtml = _context3.sent;
           if (codeToHtml) {
-            _context.next = 6;
+            _context3.next = 6;
             break;
           }
-          return _context.abrupt("return", text);
+          return _context3.abrupt("return", text);
         case 6:
-          _context.next = 8;
+          _context3.next = 8;
           return codeToHtml(text, {
             colorReplacements: builtinTheme ? undefined : colorReplacements,
             lang: matchedLanguage,
@@ -112,42 +369,42 @@ export var useHighlight = function useHighlight(text, _ref) {
             transformers: transformers
           });
         case 8:
-          html = _context.sent;
-          return _context.abrupt("return", html);
+          html = _context3.sent;
+          return _context3.abrupt("return", html);
         case 12:
-          _context.prev = 12;
-          _context.t0 = _context["catch"](0);
-          console.error('Advanced rendering failed:', _context.t0);
-          _context.prev = 15;
-          _context.next = 18;
+          _context3.prev = 12;
+          _context3.t0 = _context3["catch"](0);
+          console.error('Advanced rendering failed:', _context3.t0);
+          _context3.prev = 15;
+          _context3.next = 18;
           return shikiPromise;
         case 18:
-          _codeToHtml = _context.sent;
+          _codeToHtml = _context3.sent;
           if (_codeToHtml) {
-            _context.next = 21;
+            _context3.next = 21;
             break;
           }
-          return _context.abrupt("return", text);
+          return _context3.abrupt("return", text);
         case 21:
-          _context.next = 23;
+          _context3.next = 23;
           return _codeToHtml(text, {
             lang: matchedLanguage,
             theme: isDarkMode ? 'dark-plus' : 'light-plus'
           });
         case 23:
-          _html = _context.sent;
-          return _context.abrupt("return", _html);
+          _html = _context3.sent;
+          return _context3.abrupt("return", _html);
         case 27:
-          _context.prev = 27;
-          _context.t1 = _context["catch"](15);
+          _context3.prev = 27;
+          _context3.t1 = _context3["catch"](15);
           // Fallback to plain text
           fallbackHtml = "<pre class=\"fallback\"><code>".concat(escapeHtml(text), "</code></pre>");
-          return _context.abrupt("return", fallbackHtml);
+          return _context3.abrupt("return", fallbackHtml);
         case 31:
         case "end":
-          return _context.stop();
+          return _context3.stop();
       }
-    }, _callee, null, [[0, 12], [15, 27]]);
+    }, _callee3, null, [[0, 12], [15, 27]]);
   })), {
     dedupingInterval: 3000,
     // Only execute once for the same request within 3 seconds
@@ -156,5 +413,16 @@ export var useHighlight = function useHighlight(text, _ref) {
     revalidateOnFocus: false,
     revalidateOnReconnect: false
   });
+  var effectiveTheme = builtinTheme || (isDarkMode ? 'slack-dark' : 'slack-ochin');
+  var streamingResult = useStreamingHighlighter(text, {
+    colorReplacements: builtinTheme ? undefined : colorReplacements[effectiveTheme],
+    enabled: streaming,
+    language: matchedLanguage,
+    theme: effectiveTheme
+  });
+  return _objectSpread(_objectSpread({}, response), {}, {
+    colorReplacements: colorReplacements,
+    streaming: streamingResult
+  });
 };
 export { escapeHtml, loadShiki, MD5_LENGTH_THRESHOLD, shikiPromise };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/ui",
-  "version": "2.15.4",
+  "version": "2.15.5",
   "description": "Lobe UI is an open-source UI component library for building AIGC web apps",
   "keywords": [
     "lobehub",
@@ -59,6 +59,7 @@
     "@mdx-js/mdx": "^3.1.1",
     "@mdx-js/react": "^3.1.1",
     "@radix-ui/react-slot": "^1.2.4",
+    "@shikijs/core": "^3.15.0",
     "@shikijs/transformers": "^3.15.0",
     "@splinetool/runtime": "0.9.526",
     "ahooks": "^3.9.6",
@@ -102,6 +103,7 @@
     "remark-github": "^12.0.0",
     "remark-math": "^6.0.0",
     "shiki": "^3.15.0",
+    "shiki-stream": "^0.1.3",
     "swr": "^2.3.6",
     "ts-md5": "^2.0.1",
     "unified": "^11.0.5",

package/es/Highlighter/SyntaxHighlighter/Line.d.ts

@@ -1,5 +0,0 @@
-/// <reference types="react" />
-declare const Line: import("react").NamedExoticComponent<{
-    children?: string | undefined;
-}>;
-export default Line;

package/es/Highlighter/SyntaxHighlighter/Line.js

@@ -1,80 +0,0 @@
-'use client';
-
-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 _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 _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 _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; }
-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; }
-import { memo, useEffect, useState } from 'react';
-import Span from "./Span";
-import { jsx as _jsx } from "react/jsx-runtime";
-var Line = /*#__PURE__*/memo(function (_ref) {
-  var children = _ref.children;
-  var _useState = useState([]),
-    _useState2 = _slicedToArray(_useState, 2),
-    contents = _useState2[0],
-    setContents = _useState2[1];
-  useEffect(function () {
-    if (children && typeof children === 'string') {
-      // Extract all lines from the HTML content
-      // We need to handle the structure from shiki which gives us HTML with a <pre><code> structure
-      var parser = new DOMParser();
-      var doc = parser.parseFromString(children, 'text/html');
-      var codeElement = doc.querySelector('span.line');
-      if (codeElement) {
-        var spanLines = codeElement.querySelectorAll('span');
-        var newLines = _toConsumableArray(spanLines).map(function (line) {
-          return line.outerHTML;
-        });
-        setContents(function (prevLines) {
-          if (prevLines.length !== newLines.length) return newLines;
-          var hasChanged = false;
-          var _iterator = _createForOfIteratorHelper(newLines.entries()),
-            _step;
-          try {
-            for (_iterator.s(); !(_step = _iterator.n()).done;) {
-              var _step$value = _slicedToArray(_step.value, 2),
-                i = _step$value[0],
-                newLine = _step$value[1];
-              if (prevLines[i] !== newLine) {
-                hasChanged = true;
-                break;
-              }
-            }
-          } catch (err) {
-            _iterator.e(err);
-          } finally {
-            _iterator.f();
-          }
-          return hasChanged ? newLines : prevLines;
-        });
-      } else {
-        // Fallback if the structure is different
-        var htmlLines = children.split('\n').map(function (line) {
-          return "<span>".concat(line, "</span>");
-        });
-        setContents(htmlLines);
-      }
-    }
-  }, [children]);
-  return /*#__PURE__*/_jsx("span", {
-    className: 'line',
-    children: contents && contents.length > 0 ? contents.map(function (span, index) {
-      return /*#__PURE__*/_jsx(Span, {
-        children: span
-      }, index);
-    }) : /*#__PURE__*/_jsx("span", {
-      children: " "
-    })
-  });
-}, function (prevProps, nextProps) {
-  return prevProps.children === nextProps.children;
-});
-Line.displayName = 'HighlighterLine';
-export default Line;

package/es/Highlighter/SyntaxHighlighter/Span.d.ts

@@ -1,5 +0,0 @@
-/// <reference types="react" />
-declare const Span: import("react").NamedExoticComponent<{
-    children?: string | undefined;
-}>;
-export default Span;

package/es/Highlighter/SyntaxHighlighter/Span.js

@@ -1,16 +0,0 @@
-'use client';
-
-import { memo } from 'react';
-import { jsx as _jsx } from "react/jsx-runtime";
-var Span = /*#__PURE__*/memo(function (_ref) {
-  var children = _ref.children;
-  return /*#__PURE__*/_jsx("span", {
-    dangerouslySetInnerHTML: {
-      __html: children || ''
-    }
-  });
-}, function (prevProps, nextProps) {
-  return prevProps.children === nextProps.children;
-});
-Span.displayName = 'HighlighterSpan';
-export default Span;