Fix: add class to elements exported from CodeNode (#5558)

facebook · Jan 29, 2024 · b0cc9c4 · b0cc9c4 · vercel · Jan 29, 2024
1 parent 805ed3b
commit b0cc9c4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/packages/lexical-code/src/CodeNode.ts b/packages/lexical-code/src/CodeNode.ts
@@ -12,6 +12,7 @@ import type {
   DOMConversionOutput,
   DOMExportOutput,
   EditorConfig,
+  LexicalEditor,
   LexicalNode,
   NodeKey,
   ParagraphNode,
@@ -129,8 +130,9 @@ export class CodeNode extends ElementNode {
     return false;
   }
 
-  exportDOM(): DOMExportOutput {
+  exportDOM(editor: LexicalEditor): DOMExportOutput {
     const element = document.createElement('pre');
+    addClassNamesToElement(element, editor._config.theme.code);
     element.setAttribute('spellcheck', 'false');
     const language = this.getLanguage();
     if (language) {