Autofocus input (#63)

* Implement custom AutoFocusPlugin

* Update comment

src/components/chat-view/chat-input/ChatUserInput.tsx

@@ -1,4 +1,3 @@
-import { AutoFocusPlugin } from '@lexical/react/LexicalAutoFocusPlugin'
 import {
   InitialConfigType,
   LexicalComposer,
@@ -34,6 +33,7 @@ import { MemoizedSyntaxHighlighterWrapper } from '../SyntaxHighlighterWrapper'
 
 import MentionableBadge from './MentionableBadge'
 import { ModelSelect } from './ModelSelect'
+import AutoFocusPlugin from './plugins/auto-focus/AutoFocusPlugin'
 import AutoLinkMentionPlugin from './plugins/mention/AutoLinkMentionPlugin'
 import { MentionNode } from './plugins/mention/MentionNode'
 import MentionPlugin from './plugins/mention/MentionPlugin'
@@ -318,7 +318,7 @@ const ChatUserInput = forwardRef<ChatUserInputRef, ChatUserInputProps>(
             ErrorBoundary={LexicalErrorBoundary}
           />
           <HistoryPlugin />
-          {autoFocus && <AutoFocusPlugin />}
+          {autoFocus && <AutoFocusPlugin defaultSelection="rootEnd" />}
           <MentionPlugin searchResultByQuery={searchResultByQuery} />
           <OnChangePlugin
             onChange={(editorState) => {

src/components/chat-view/chat-input/plugins/auto-focus/AutoFocusPlugin.tsx

@@ -0,0 +1,29 @@
+import { useLexicalComposerContext } from '@lexical/react/LexicalComposerContext'
+import { useEffect } from 'react'
+
+export type AutoFocusPluginProps = {
+  defaultSelection?: 'rootStart' | 'rootEnd'
+}
+
+export default function AutoFocusPlugin({
+  defaultSelection,
+}: AutoFocusPluginProps) {
+  const [editor] = useLexicalComposerContext()
+
+  useEffect(() => {
+    editor.focus(
+      () => {
+        const rootElement = editor.getRootElement()
+        if (rootElement) {
+          // requestAnimationFrame is required here for unknown reasons, possibly related to the Obsidian plugin environment.
+          requestAnimationFrame(() => {
+            rootElement.focus()
+          })
+        }
+      },
+      { defaultSelection },
+    )
+  }, [defaultSelection, editor])
+
+  return null
+}