From 6113603872de2ffdf59283e016f6218a4fbd3054 Mon Sep 17 00:00:00 2001
From: Hadi Elghoul <hadielghoul@Hadis-MacBook-Pro.local>
Date: Tue, 3 Dec 2024 17:02:38 -0500
Subject: [PATCH] Unify  implementations and deprecate @lexical/selection

---
 packages/lexical-selection/src/index.ts       |  5 ++-
 .../lexical-selection/src/range-selection.ts  | 34 +-------------
 packages/lexical/src/LexicalUtils.ts          | 45 +++++++++++++++++--
 3 files changed, 47 insertions(+), 37 deletions(-)

diff --git a/packages/lexical-selection/src/index.ts b/packages/lexical-selection/src/index.ts
index d901ab4d4d9..eb4c5a5a328 100644
--- a/packages/lexical-selection/src/index.ts
+++ b/packages/lexical-selection/src/index.ts
@@ -18,7 +18,6 @@ import {
   $isParentElementRTL,
   $moveCaretSelection,
   $moveCharacter,
-  $selectAll,
   $setBlocksType,
   $shouldOverrideDefaultCharacterSelection,
   $wrapNodes,
@@ -30,6 +29,9 @@ import {
   getStyleObjectFromCSS,
 } from './utils';
 
+export {
+  /** @deprecated moved to the lexical package */ $selectAll,
+} from 'lexical';
 export {
   /** @deprecated moved to the lexical package */ $cloneWithProperties,
 } from 'lexical';
@@ -48,7 +50,6 @@ export {
   $isParentElementRTL,
   $moveCaretSelection,
   $moveCharacter,
-  $selectAll,
   $setBlocksType,
   $shouldOverrideDefaultCharacterSelection,
   $wrapNodes,
diff --git a/packages/lexical-selection/src/range-selection.ts b/packages/lexical-selection/src/range-selection.ts
index e92a81b6188..eec61beba8d 100644
--- a/packages/lexical-selection/src/range-selection.ts
+++ b/packages/lexical-selection/src/range-selection.ts
@@ -453,39 +453,9 @@ export function $moveCharacter(
 }
 
 /**
- * Expands the current Selection to cover all of the content in the editor.
- * @param selection - The current selection.
+ * @deprecated This function has been moved to the lexical package. Use the $selectAll function from the lexical package instead.
  */
-export function $selectAll(selection: RangeSelection): void {
-  const anchor = selection.anchor;
-  const focus = selection.focus;
-  const anchorNode = anchor.getNode();
-  const topParent = anchorNode.getTopLevelElementOrThrow();
-  const root = topParent.getParentOrThrow();
-  let firstNode = root.getFirstDescendant();
-  let lastNode = root.getLastDescendant();
-  let firstType: 'element' | 'text' = 'element';
-  let lastType: 'element' | 'text' = 'element';
-  let lastOffset = 0;
-
-  if ($isTextNode(firstNode)) {
-    firstType = 'text';
-  } else if (!$isElementNode(firstNode) && firstNode !== null) {
-    firstNode = firstNode.getParentOrThrow();
-  }
-
-  if ($isTextNode(lastNode)) {
-    lastType = 'text';
-    lastOffset = lastNode.getTextContentSize();
-  } else if (!$isElementNode(lastNode) && lastNode !== null) {
-    lastNode = lastNode.getParentOrThrow();
-  }
-
-  if (firstNode && lastNode) {
-    anchor.set(firstNode.getKey(), 0, firstType);
-    focus.set(lastNode.getKey(), lastOffset, lastType);
-  }
-}
+export {$selectAll} from 'lexical';
 
 /**
  * Returns the current value of a CSS property for Nodes, if set. If not set, it returns the defaultValue.
diff --git a/packages/lexical/src/LexicalUtils.ts b/packages/lexical/src/LexicalUtils.ts
index f4cc88ac86a..78bd1081983 100644
--- a/packages/lexical/src/LexicalUtils.ts
+++ b/packages/lexical/src/LexicalUtils.ts
@@ -1084,10 +1084,49 @@ export function isSelectAll(
   return key.toLowerCase() === 'a' && controlOrMeta(metaKey, ctrlKey);
 }
 
-export function $selectAll(): void {
+export function $selectAll(selection?: RangeSelection | null): RangeSelection {
   const root = $getRoot();
-  const selection = root.select(0, root.getChildrenSize());
-  $setSelection($normalizeSelection(selection));
+
+  if (selection && $isRangeSelection(selection)) {
+    // Expand the existing RangeSelection
+    const anchor = selection.anchor;
+    const focus = selection.focus;
+    const anchorNode = anchor.getNode();
+    const topParent = anchorNode.getTopLevelElementOrThrow();
+    const rootNode = topParent.getParentOrThrow();
+    let firstNode = rootNode.getFirstDescendant();
+    let lastNode = rootNode.getLastDescendant();
+    let firstType: 'element' | 'text' = 'element';
+    let lastType: 'element' | 'text' = 'element';
+    let lastOffset = 0;
+
+    if ($isTextNode(firstNode)) {
+      firstType = 'text';
+    } else if (!$isElementNode(firstNode) && firstNode !== null) {
+      firstNode = firstNode.getParentOrThrow();
+    }
+
+    if ($isTextNode(lastNode)) {
+      lastType = 'text';
+      lastOffset = lastNode.getTextContentSize();
+    } else if (!$isElementNode(lastNode) && lastNode !== null) {
+      lastNode = lastNode.getParentOrThrow();
+    }
+
+    if (firstNode && lastNode) {
+      anchor.set(firstNode.getKey(), 0, firstType);
+      focus.set(lastNode.getKey(), lastOffset, lastType);
+    }
+
+    // Normalize and set the updated selection
+    $setSelection($normalizeSelection(selection));
+    return selection;
+  } else {
+    // Create a new RangeSelection
+    const newSelection = root.select(0, root.getChildrenSize());
+    $setSelection($normalizeSelection(newSelection));
+    return newSelection;
+  }
 }
 
 export function getCachedClassNameArray(