@lobehub/editor
Version:
A powerful and extensible rich text editor built on Meta's Lexical framework, providing a modern editing experience with React integration.
380 lines (363 loc) • 22.6 kB
JavaScript
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 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); }); }; }
/**
* Copyright (c) Meta Platforms, Inc. and affiliates.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*
*/
import { $createCodeHighlightNode, $isCodeNode } from '@lexical/code';
import { ShikiLobeTheme } from '@lobehub/ui';
import { createHighlighterCoreSync, isSpecialLang, isSpecialTheme, stringifyTokenStyle } from '@shikijs/core';
import { createJavaScriptRegexEngine } from '@shikijs/engine-javascript';
import { $createLineBreakNode, $createTabNode, $getNodeByKey } from 'lexical';
import { bundledLanguagesInfo, bundledThemesInfo } from 'shiki';
import { INodeHelper } from "../../../editor-kernel/inode/helper";
// Color replacements types for Shiki
var shiki = createHighlighterCoreSync({
engine: createJavaScriptRegexEngine(),
langs: [],
themes: []
});
// Load ShikiLobeTheme immediately after initialization
shiki.loadTheme(ShikiLobeTheme);
function getDiffedLanguage(language) {
var DIFF_LANGUAGE_REGEX = /^diff-([\w-]+)/i;
var diffLanguageMatch = DIFF_LANGUAGE_REGEX.exec(language);
return diffLanguageMatch ? diffLanguageMatch[1] : null;
}
/**
* Creates a simple color replacement map
* @param replacements - Object mapping from old color to new color
* @returns ColorReplacements object
*/
export function createColorReplacements(replacements) {
return replacements;
}
/**
* Creates scoped color replacements for multiple themes
* @param scopedReplacements - Object mapping theme names to their color replacements
* @returns ScopedColorReplacements object
*/
export function createScopedColorReplacements(scopedReplacements) {
return scopedReplacements;
}
/**
* Validates that a color value is a valid CSS color
* @param color - The color string to validate
* @returns true if the color is valid
*/
export function isValidColor(color) {
// Basic validation for common color formats
var hexColorRegex = /^#[\dA-Fa-f]{3,8}$/;
var rgbColorRegex = /^rgb\((?:\s*\d+\s*,){2}\s*\d+\s*\)$/;
var rgbaColorRegex = /^rgba\((?:\s*\d+\s*,){3}\s*[\d.]+\s*\)$/;
var hslColorRegex = /^hsl\(\s*\d+(?:\s*,\s*\d+%){2}\s*\)$/;
var hslaColorRegex = /^hsla\(\s*\d+(?:\s*,\s*\d+%){2}\s*,\s*[\d.]+\s*\)$/;
var cssVariableRegex = /^var\(--[\w-]+\)$/;
return hexColorRegex.test(color) || rgbColorRegex.test(color) || rgbaColorRegex.test(color) || hslColorRegex.test(color) || hslaColorRegex.test(color) || cssVariableRegex.test(color);
}
export function isCodeLanguageLoaded(language) {
var diffedLanguage = getDiffedLanguage(language);
var langId = diffedLanguage || language;
// handle shiki Hard-coded languages ['ansi', '', 'plaintext', 'txt', 'text', 'plain']
if (isSpecialLang(langId)) {
return true;
}
// note: getLoadedLanguages() also returns aliases
return shiki.getLoadedLanguages().includes(langId);
}
export function loadCodeLanguage(language, editor, codeNodeKey) {
var diffedLanguage = getDiffedLanguage(language);
var langId = diffedLanguage ? diffedLanguage : language;
if (!isCodeLanguageLoaded(langId)) {
var languageInfo = bundledLanguagesInfo.find(function (desc) {
return desc.id === langId || desc.aliases && desc.aliases.includes(langId);
});
if (languageInfo) {
// in case we arrive here concurrently (not yet loaded language is loaded twice)
// shiki's synchronous checks make sure to load it only once
shiki.loadLanguage(languageInfo.import()).then(function () {
// here we know that the language is loaded
// make sure the code is highlighed with the correct language
if (editor && codeNodeKey) {
editor.update(function () {
var codeNode = $getNodeByKey(codeNodeKey);
if ($isCodeNode(codeNode) && codeNode.getLanguage() === language && !codeNode.getIsSyntaxHighlightSupported()) {
codeNode.setIsSyntaxHighlightSupported(true);
}
});
}
});
}
}
}
function _isCodeThemeLoaded(theme) {
var themeId = theme;
// handle shiki special theme ['none']
if (isSpecialTheme(themeId)) {
return true;
}
return shiki.getLoadedThemes().includes(themeId);
}
export function isCodeThemeLoaded(theme) {
var themes = theme.split(' ');
return themes.every(function (t) {
return _isCodeThemeLoaded(t);
});
}
function _loadCodeTheme(_x, _x2, _x3) {
return _loadCodeTheme2.apply(this, arguments);
}
function _loadCodeTheme2() {
_loadCodeTheme2 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(theme, editor, codeNodeKey) {
var themeInfo;
return _regeneratorRuntime().wrap(function _callee$(_context) {
while (1) switch (_context.prev = _context.next) {
case 0:
if (isCodeThemeLoaded(theme)) {
_context.next = 6;
break;
}
if (!(theme === 'lobe-theme')) {
_context.next = 4;
break;
}
// ShikiLobeTheme is already loaded in shiki instance initialization
if (editor && codeNodeKey) {
editor.update(function () {
var codeNode = $getNodeByKey(codeNodeKey);
if ($isCodeNode(codeNode)) {
codeNode.markDirty();
}
});
}
return _context.abrupt("return");
case 4:
themeInfo = bundledThemesInfo.find(function (info) {
return info.id === theme;
});
if (themeInfo) {
shiki.loadTheme(themeInfo.import()).then(function () {
if (editor && codeNodeKey) {
editor.update(function () {
var codeNode = $getNodeByKey(codeNodeKey);
if ($isCodeNode(codeNode)) {
codeNode.markDirty();
}
});
}
});
}
case 6:
case "end":
return _context.stop();
}
}, _callee);
}));
return _loadCodeTheme2.apply(this, arguments);
}
export function loadCodeTheme(_x4, _x5, _x6) {
return _loadCodeTheme3.apply(this, arguments);
}
function _loadCodeTheme3() {
_loadCodeTheme3 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(theme, editor, codeNodeKey) {
var themes;
return _regeneratorRuntime().wrap(function _callee2$(_context2) {
while (1) switch (_context2.prev = _context2.next) {
case 0:
themes = theme.split(' ');
_context2.next = 3;
return Promise.all(themes.map(function (t) {
return _loadCodeTheme(t, editor, codeNodeKey);
}));
case 3:
case "end":
return _context2.stop();
}
}, _callee2);
}));
return _loadCodeTheme3.apply(this, arguments);
}
export function getCodeLanguageOptions() {
return bundledLanguagesInfo.map(function (i) {
return [i.id, i.name];
});
}
export function getCodeThemeOptions() {
return bundledThemesInfo.map(function (i) {
return [i.id, i.displayName];
});
}
export function normalizeCodeLanguage(language) {
var langId = language;
var languageInfo = bundledLanguagesInfo.find(function (desc) {
return desc.id === langId || desc.aliases && desc.aliases.includes(langId);
});
if (languageInfo) {
return languageInfo.id;
}
return language;
}
function getTokenStyleObject(token) {
var style = '';
if (token.color) {
style += "color: ".concat(token.color, ";");
}
if (token.bgColor) {
style += "background-color: ".concat(token.bgColor, ";");
}
return style;
}
function mapTokensToLexicalSerialized(tokens, diff) {
var nodes = [];
tokens.forEach(function (line, idx) {
if (idx) {
nodes.push({
type: 'linebreak',
version: 1
});
}
line.forEach(function (token, tidx) {
var text = token.content;
// implement diff-xxxx languages
if (diff && tidx === 0 && text.length > 0) {
var prefixes = ['+', '-', '>', '<', ' '];
var prefixTypes = ['inserted', 'deleted', 'inserted', 'deleted', 'unchanged'];
var prefixIndex = prefixes.indexOf(text[0]);
if (prefixIndex !== -1) {
nodes.push(INodeHelper.createLikeTextNode('code-highlight', prefixes[prefixIndex], {
highlightType: prefixTypes[prefixIndex]
}));
text = text.slice(1);
}
}
var parts = text.split('\t');
parts.forEach(function (part, pidx) {
if (pidx) {
nodes.push(INodeHelper.createLikeTextNode('tab', '\t'));
}
if (part !== '') {
var node = INodeHelper.createLikeTextNode('code-highlight', part);
var style = stringifyTokenStyle(token.htmlStyle || getTokenStyleObject(token));
node.style = style;
nodes.push(node);
}
});
});
});
return nodes;
}
function mapTokensToLexicalStructure(tokens, diff) {
var nodes = [];
tokens.forEach(function (line, idx) {
if (idx) {
nodes.push($createLineBreakNode());
}
line.forEach(function (token, tidx) {
var text = token.content;
// implement diff-xxxx languages
if (diff && tidx === 0 && text.length > 0) {
var prefixes = ['+', '-', '>', '<', ' '];
var prefixTypes = ['inserted', 'deleted', 'inserted', 'deleted', 'unchanged'];
var prefixIndex = prefixes.indexOf(text[0]);
if (prefixIndex !== -1) {
nodes.push($createCodeHighlightNode(prefixes[prefixIndex], prefixTypes[prefixIndex]));
text = text.slice(1);
}
}
var parts = text.split('\t');
parts.forEach(function (part, pidx) {
if (pidx) {
nodes.push($createTabNode());
}
if (part !== '') {
var node = $createCodeHighlightNode(part);
var style = stringifyTokenStyle(token.htmlStyle || getTokenStyleObject(token));
node.setStyle(style);
nodes.push(node);
}
});
});
});
return nodes;
}
var DIFF_LANGUAGE_REGEX = /^diff-([\w-]+)/i;
export function getHighlightSerializeNode(code, language) {
var theme = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : 'lobe-theme';
// Implementation goes here
var diffLanguageMatch = DIFF_LANGUAGE_REGEX.exec(language);
var themes = theme.split(' ');
// Build the options for codeToTokens
var options = themes.length > 1 ? {
defaultColor: false,
lang: diffLanguageMatch ? diffLanguageMatch[1] : language,
themes: {
dark: themes[1],
light: themes[0]
}
} : {
lang: language,
theme: themes[0]
};
var tokensResult = shiki.codeToTokens(code, options);
var tokens = tokensResult.tokens;
return mapTokensToLexicalSerialized(tokens, !!diffLanguageMatch);
}
export function $getHighlightNodes(codeNode, language) {
var diffLanguageMatch = DIFF_LANGUAGE_REGEX.exec(language);
var code = codeNode.getTextContent();
var theme = codeNode.getTheme() || 'lobe-theme';
var themes = theme.split(' ');
// Build the options for codeToTokens
var options = themes.length > 1 ? {
defaultColor: false,
lang: diffLanguageMatch ? diffLanguageMatch[1] : language,
themes: {
dark: themes[1],
light: themes[0]
}
} : {
lang: language,
theme: themes[0]
};
var tokensResult = shiki.codeToTokens(code, options);
var tokens = tokensResult.tokens;
// let style = '';
// if (bg) {
// style += `background-color: ${bg};`;
// }
// if (fg) {
// style += `color: ${fg};`;
// }
// if (codeNode.getStyle() !== style) {
// codeNode.setStyle(style);
// }
return mapTokensToLexicalStructure(tokens, !!diffLanguageMatch);
}
/**
* Extended version of $getHighlightNodes with additional options
* @param codeNode - The CodeNode to highlight
* @param language - The programming language
* @param options - Additional highlighting options
*/
export function $getHighlightNodesWithOptions(codeNode, language, options) {
// If theme is provided in options, temporarily override the CodeNode's theme
var originalGetTheme = codeNode.getTheme;
if (options !== null && options !== void 0 && options.theme) {
codeNode.getTheme = function () {
return options.theme;
};
}
try {
return $getHighlightNodes(codeNode, language);
} finally {
// Restore original getTheme method
if (options !== null && options !== void 0 && options.theme) {
codeNode.getTheme = originalGetTheme;
}
}
}