Lexical: Started code block node implementation

[bookstack] / resources / js / wysiwyg / ui / index.ts

diff --git a/resources/js/wysiwyg/ui/index.ts b/resources/js/wysiwyg/ui/index.ts
index 8227dec6838f4cb23077db9c307ccd41a55d02f2..1ad1395dc7781289d449ba294dfbf96fe7779f61 100644 (file)
--- a/resources/js/wysiwyg/ui/index.ts
+++ b/resources/js/wysiwyg/ui/index.ts
@@ -1,22 +1,17 @@
-import {
-    $getSelection,
-    COMMAND_PRIORITY_LOW,
-    LexicalEditor,
-    SELECTION_CHANGE_COMMAND
-} from "lexical";
-import {getMainEditorFullToolbar} from "./toolbars";
+import {LexicalEditor} from "lexical";
+import {getImageToolbarContent, getLinkToolbarContent, getMainEditorFullToolbar} from "./toolbars";
 import {EditorUIManager} from "./framework/manager";
-import {image as imageFormDefinition, link as linkFormDefinition} from "./defaults/form-definitions";
-import {DecoratorListener} from "lexical/LexicalEditor";
-import type {NodeKey} from "lexical/LexicalNode";
-import {EditorDecoratorAdapter} from "./framework/decorator";
+import {image as imageFormDefinition, link as linkFormDefinition, source as sourceFormDefinition} from "./defaults/form-definitions";
 import {ImageDecorator} from "./decorators/image";
 import {EditorUiContext} from "./framework/core";
+import {CodeBlockDecorator} from "./decorators/code-block";
 
-export function buildEditorUI(element: HTMLElement, editor: LexicalEditor) {
+export function buildEditorUI(container: HTMLElement, element: HTMLElement, editor: LexicalEditor) {
     const manager = new EditorUIManager();
     const context: EditorUiContext = {
         editor,
+        containerDOM: container,
+        editorDOM: element,
         manager,
         translate: (text: string): string => text,
         lastSelection: null,
@@ -24,9 +19,7 @@ export function buildEditorUI(element: HTMLElement, editor: LexicalEditor) {
     manager.setContext(context);
 
     // Create primary toolbar
-    const toolbar = getMainEditorFullToolbar();
-    toolbar.setContext(context);
-    element.before(toolbar.getDOMElement());
+    manager.setToolbar(getMainEditorFullToolbar());
 
     // Register modals
     manager.registerModal('link', {
@@ -36,31 +29,26 @@ export function buildEditorUI(element: HTMLElement, editor: LexicalEditor) {
     manager.registerModal('image', {
         title: 'Insert/Edit Image',
         form: imageFormDefinition
-    })
+    });
+    manager.registerModal('source', {
+        title: 'Source code',
+        form: sourceFormDefinition,
+    });
 
-    // Register decorator listener
-    // Maybe move to manager?
-    manager.registerDecoratorType('image', ImageDecorator);
-    const domDecorateListener: DecoratorListener<EditorDecoratorAdapter> = (decorators: Record<NodeKey, EditorDecoratorAdapter>) => {
-        const keys = Object.keys(decorators);
-        for (const key of keys) {
-            const decoratedEl = editor.getElementByKey(key);
-            const adapter = decorators[key];
-            const decorator = manager.getDecorator(adapter.type, key);
-            decorator.setNode(adapter.getNode());
-            const decoratorEl = decorator.render(context);
-            if (decoratedEl) {
-                decoratedEl.append(decoratorEl);
-            }
+    // Register context toolbars
+    manager.registerContextToolbar('image', {
+        selector: 'img',
+        content: getImageToolbarContent(),
+        displayTargetLocator(originalTarget: HTMLElement) {
+            return originalTarget.closest('a') || originalTarget;
         }
-    }
-    editor.registerDecoratorListener(domDecorateListener);
+    });
+    manager.registerContextToolbar('link', {
+        selector: 'a',
+        content: getLinkToolbarContent(),
+    });
 
-    // Update button states on editor selection change
-    editor.registerCommand(SELECTION_CHANGE_COMMAND, () => {
-        const selection = $getSelection();
-        toolbar.updateState({editor, selection});
-        context.lastSelection = selection;
-        return false;
-    }, COMMAND_PRIORITY_LOW);
+    // Register image decorator listener
+    manager.registerDecoratorType('image', ImageDecorator);
+    manager.registerDecoratorType('code', CodeBlockDecorator);
 }
\ No newline at end of file