From 3ad6d7dda04d3ff0f53cbf31b4f95b1fc2dbf7b4 Mon Sep 17 00:00:00 2001
From: josh <josh@mjbrowns.com>
Date: Thu, 14 Sep 2023 23:08:06 -0500
Subject: [PATCH] feat!: pass preventScroll to focus and autoFocus

BREAKING CHANGE: use opts object instead of strings | boolean
---
 src/MDXEditor.tsx         |  8 ++++----
 src/plugins/core/index.ts | 16 +++++++++++++---
 2 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/src/MDXEditor.tsx b/src/MDXEditor.tsx
index 5b2bd99c..ab592558 100644
--- a/src/MDXEditor.tsx
+++ b/src/MDXEditor.tsx
@@ -104,7 +104,7 @@ export interface MDXEditorProps {
   /**
    * pass if you would like to have the editor automatically focused when mounted.
    */
-  autoFocus?: boolean | 'rootStart' | 'rootEnd'
+  autoFocus?: boolean | { defaultSelection?: 'rootStart' | 'rootEnd'; preventScroll?: boolean }
   /**
    * Triggered when focus leaves the editor
    */
@@ -147,7 +147,7 @@ export interface MDXEditorMethods {
   /**
    * Sets focus on input
    */
-  focus: (callbackFn?: (() => void) | undefined, defaultSelection?: 'rootStart' | 'rootEnd') => void
+  focus: (callbackFn?: (() => void) | undefined, opts?: { defaultSelection?: 'rootStart' | 'rootEnd'; preventScroll?: boolean }) => void
 }
 
 const RenderRecurisveWrappers: React.FC<{ wrappers: React.ComponentType<{ children: React.ReactNode }>[]; children: React.ReactNode }> = ({
@@ -203,8 +203,8 @@ const Methods: React.FC<{ mdxRef: React.ForwardedRef<MDXEditorMethods> }> = ({ m
         setMarkdown: (markdown) => {
           realm.pubKey('setMarkdown', markdown)
         },
-        focus: (callbackFn?: (() => void) | undefined, defaultSelection?: 'rootStart' | 'rootEnd') => {
-          realm.getKeyValue('rootEditor')?.focus(callbackFn, { defaultSelection })
+        focus: (callbackFn?: (() => void) | undefined, opts?: { defaultSelection?: 'rootStart' | 'rootEnd'; preventScroll?: boolean }) => {
+          realm.getKeyValue('rootEditor')?.focus(callbackFn, opts)
         }
       }
     },
diff --git a/src/plugins/core/index.ts b/src/plugins/core/index.ts
index 74d33bf4..7fd3c1ba 100644
--- a/src/plugins/core/index.ts
+++ b/src/plugins/core/index.ts
@@ -94,7 +94,7 @@ export const coreSystem = system((r) => {
   const contentEditableClassName = r.node<string>('')
   const readOnly = r.node<boolean>(false)
   const placeholder = r.node<React.ReactNode>('')
-  const autoFocus = r.node<boolean | 'rootStart' | 'rootEnd'>(false)
+  const autoFocus = r.node<boolean | { defaultSelection?: 'rootStart' | 'rootEnd'; preventScroll: boolean }>(false)
   const inFocus = r.node(false, true)
   const currentFormat = r.node(0, true)
 
@@ -251,7 +251,17 @@ export const coreSystem = system((r) => {
 
     const autoFocusValue = r.getValue(autoFocus)
     if (autoFocusValue) {
-      setTimeout(() => theRootEditor.focus(noop, { defaultSelection: autoFocusValue === true ? 'rootStart' : autoFocusValue }))
+      if (autoFocusValue === true) {
+        // Default 'on' state
+        setTimeout(() => theRootEditor.focus(noop, { defaultSelection: 'rootStart' }))
+        return
+      }
+      setTimeout(() =>
+        theRootEditor.focus(noop, {
+          defaultSelection: autoFocusValue.defaultSelection ?? 'rootStart',
+          preventScroll: autoFocusValue.preventScroll ?? false
+        })
+      )
     }
   })
 
@@ -493,7 +503,7 @@ interface CorePluginParams {
   initialMarkdown: string
   contentEditableClassName: string
   placeholder?: React.ReactNode
-  autoFocus: boolean | 'rootStart' | 'rootEnd'
+  autoFocus: boolean | { defaultSelection?: 'rootStart' | 'rootEnd'; preventScroll?: boolean }
   onChange: (markdown: string) => void
   onBlur?: (e: FocusEvent) => void
   toMarkdownOptions: NonNullable<LexicalConvertOptions['toMarkdownOptions']>