Lexical: Focus editor by hotkey

Signed-off-by: marcin mikołajczak <git@mkljczk.pl>

app/soapbox/features/compose/editor/plugins/focus-plugin.tsx

@@ -1,27 +1,35 @@
 import { useLexicalComposerContext } from '@lexical/react/LexicalComposerContext';
+import { COMMAND_PRIORITY_NORMAL, createCommand, type LexicalCommand } from 'lexical';
 import { useEffect } from 'react';
 
 interface IFocusPlugin {
   autoFocus?: boolean
 }
 
+export const FOCUS_EDITOR_COMMAND: LexicalCommand<void> = createCommand();
+
 const FocusPlugin: React.FC<IFocusPlugin> = ({ autoFocus }) => {
   const [editor] = useLexicalComposerContext();
 
   const focus = () => {
+    editor.dispatchCommand(FOCUS_EDITOR_COMMAND, undefined);
+  };
+
+  useEffect(() => editor.registerCommand(FOCUS_EDITOR_COMMAND, () => {
     editor.focus(
       () => {
         const activeElement = document.activeElement;
         const rootElement = editor.getRootElement();
         if (
           rootElement !== null &&
-        (activeElement === null || !rootElement.contains(activeElement))
+          (activeElement === null || !rootElement.contains(activeElement))
         ) {
           rootElement.focus({ preventScroll: true });
         }
       }, { defaultSelection: 'rootEnd' },
     );
-  };
+    return true;
+  }, COMMAND_PRIORITY_NORMAL));
 
   useEffect(() => {
     if (autoFocus) focus();