@lobehub/ui
Version:
Lobe UI is an open-source UI component library for building AIGC web apps
428 lines (417 loc) • 28.3 kB
JavaScript
'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 _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 { 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";
// Application-level cache to avoid repeated calculations
var MD5_LENGTH_THRESHOLD = 10000; // Use async MD5 for text exceeding this length
// Color replacement mapping type
// Lazy load shiki
var loadShikiModule = function loadShikiModule() {
if (typeof window === 'undefined') return Promise.resolve(null);
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();
// Helper function: Safe HTML escaping
var escapeHtml = function escapeHtml(str) {
return str.replaceAll('&', '&').replaceAll('<', '<').replaceAll('>', '>').replaceAll('"', '"').replaceAll("'", ''');
};
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, _ref3) {
var language = _ref3.language,
enableTransformer = _ref3.enableTransformer,
builtinTheme = _ref3.theme,
streaming = _ref3.streaming;
var _useThemeMode = useThemeMode(),
isDarkMode = _useThemeMode.isDarkMode;
var theme = useTheme();
var lang = language.toLowerCase();
// Match supported languages
var matchedLanguage = useMemo(function () {
return getCodeLanguageByInput(lang);
}, [lang]);
// Optimize transformer creation
var transformers = useMemo(function () {
if (!enableTransformer) return;
return [transformerNotationDiff(), transformerNotationHighlight(), transformerNotationWordHighlight(), transformerNotationFocus(), transformerNotationErrorLevel()];
}, [enableTransformer]);
// Optimize color replacement configuration
var colorReplacements = useMemo(function () {
return {
'slack-dark': {
'#4ec9b0': theme.yellow,
'#569cd6': theme.colorError,
'#6a9955': theme.gray,
'#9cdcfe': theme.colorText,
'#b5cea8': theme.purple10,
'#c586c0': theme.colorInfo,
'#ce9178': theme.colorSuccess,
'#dcdcaa': theme.colorWarning,
'#e6e6e6': theme.colorText
},
'slack-ochin': {
'#002339': theme.colorText,
'#0444ac': theme.geekblue,
'#0991b6': theme.colorError,
'#174781': theme.purple10,
'#2f86d2': theme.colorText,
'#357b42': theme.gray,
'#7b30d0': theme.colorInfo,
'#7eb233': theme.colorWarningTextActive,
'#a44185': theme.colorSuccess,
'#dc3eb7': theme.yellow11
}
};
}, [theme]);
// Build cache key
var cacheKey = useMemo(function () {
// Use hash for long text
var hash = text.length < MD5_LENGTH_THRESHOLD ? text : Md5.hashStr(text);
return [matchedLanguage, builtinTheme || (isDarkMode ? 'd' : 'l'), hash].filter(Boolean).join('-');
}, [text, matchedLanguage, isDarkMode, builtinTheme]);
// Use SWR to get highlighted HTML
var response = useSWR(streaming ? null : cacheKey, /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3() {
var codeToHtml, html, _codeToHtml, _html, fallbackHtml;
return _regeneratorRuntime().wrap(function _callee3$(_context3) {
while (1) switch (_context3.prev = _context3.next) {
case 0:
_context3.prev = 0;
_context3.next = 3;
return shikiPromise;
case 3:
codeToHtml = _context3.sent;
if (codeToHtml) {
_context3.next = 6;
break;
}
return _context3.abrupt("return", text);
case 6:
_context3.next = 8;
return codeToHtml(text, {
colorReplacements: builtinTheme ? undefined : colorReplacements,
lang: matchedLanguage,
theme: builtinTheme || (isDarkMode ? 'slack-dark' : 'slack-ochin'),
transformers: transformers
});
case 8:
html = _context3.sent;
return _context3.abrupt("return", html);
case 12:
_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 = _context3.sent;
if (_codeToHtml) {
_context3.next = 21;
break;
}
return _context3.abrupt("return", text);
case 21:
_context3.next = 23;
return _codeToHtml(text, {
lang: matchedLanguage,
theme: isDarkMode ? 'dark-plus' : 'light-plus'
});
case 23:
_html = _context3.sent;
return _context3.abrupt("return", _html);
case 27:
_context3.prev = 27;
_context3.t1 = _context3["catch"](15);
// Fallback to plain text
fallbackHtml = "<pre class=\"fallback\"><code>".concat(escapeHtml(text), "</code></pre>");
return _context3.abrupt("return", fallbackHtml);
case 31:
case "end":
return _context3.stop();
}
}, _callee3, null, [[0, 12], [15, 27]]);
})), {
dedupingInterval: 3000,
// Only execute once for the same request within 3 seconds
errorRetryCount: 2,
// Retry at most 2 times
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 };