From edead7e2511b6c59843bb5b0336ef89fd3240f77 Mon Sep 17 00:00:00 2001
From: Aman Harwara <amanharwara@protonmail.com>
Date: Wed, 19 Jul 2023 12:37:02 +0530
Subject: [PATCH] feat: Added Undo/Redo buttons to Super mobile toolbar

---
 .../Components/Icon/IconNameToSvgMapping.tsx  |  2 +
 .../MobileToolbarPlugin.tsx                   | 57 ++++++++++++++++++-
 2 files changed, 56 insertions(+), 3 deletions(-)

diff --git a/packages/web/src/javascripts/Components/Icon/IconNameToSvgMapping.tsx b/packages/web/src/javascripts/Components/Icon/IconNameToSvgMapping.tsx
index 32a8483c6..8848c67b9 100644
--- a/packages/web/src/javascripts/Components/Icon/IconNameToSvgMapping.tsx
+++ b/packages/web/src/javascripts/Components/Icon/IconNameToSvgMapping.tsx
@@ -111,6 +111,7 @@ export const IconNameToSvgMapping = {
   password: icons.PasswordIcon,
   pencil: icons.PencilIcon,
   pin: icons.PinIcon,
+  redo: icons.RedoIcon,
   restore: icons.RestoreIcon,
   save: icons.SaveIcon,
   search: icons.SearchIcon,
@@ -135,6 +136,7 @@ export const IconNameToSvgMapping = {
   tune: icons.TuneIcon,
   unarchive: icons.UnarchiveIcon,
   underline: icons.UnderlineIcon,
+  undo: icons.UndoIcon,
   unpin: icons.UnpinIcon,
   upload: icons.UploadIcon,
   user: icons.UserIcon,
diff --git a/packages/web/src/javascripts/Components/SuperEditor/Plugins/MobileToolbarPlugin/MobileToolbarPlugin.tsx b/packages/web/src/javascripts/Components/SuperEditor/Plugins/MobileToolbarPlugin/MobileToolbarPlugin.tsx
index 878b4f7dc..d539fedc7 100644
--- a/packages/web/src/javascripts/Components/SuperEditor/Plugins/MobileToolbarPlugin/MobileToolbarPlugin.tsx
+++ b/packages/web/src/javascripts/Components/SuperEditor/Plugins/MobileToolbarPlugin/MobileToolbarPlugin.tsx
@@ -4,7 +4,17 @@ import useModal from '../../Lexical/Hooks/useModal'
 import { InsertTableDialog } from '../../Plugins/TablePlugin'
 import { getSelectedNode } from '../../Lexical/Utils/getSelectedNode'
 import { sanitizeUrl } from '../../Lexical/Utils/sanitizeUrl'
-import { $getSelection, $isRangeSelection, FORMAT_TEXT_COMMAND } from 'lexical'
+import {
+  $getSelection,
+  $isRangeSelection,
+  CAN_REDO_COMMAND,
+  CAN_UNDO_COMMAND,
+  COMMAND_PRIORITY_CRITICAL,
+  FORMAT_TEXT_COMMAND,
+  REDO_COMMAND,
+  UNDO_COMMAND,
+} from 'lexical'
+import { mergeRegister } from '@lexical/utils'
 import { $isLinkNode, TOGGLE_LINK_COMMAND } from '@lexical/link'
 import { useCallback, useEffect, useMemo, useRef, useState } from 'react'
 import { GetAlignmentBlocks } from '../Blocks/Alignment'
@@ -64,13 +74,53 @@ const MobileToolbarPlugin = () => {
     }
   }, [editor])
 
+  const [canUndo, setCanUndo] = useState(false)
+  const [canRedo, setCanRedo] = useState(false)
+  useEffect(() => {
+    return mergeRegister(
+      editor.registerCommand<boolean>(
+        CAN_UNDO_COMMAND,
+        (payload) => {
+          setCanUndo(payload)
+          return false
+        },
+        COMMAND_PRIORITY_CRITICAL,
+      ),
+      editor.registerCommand<boolean>(
+        CAN_REDO_COMMAND,
+        (payload) => {
+          setCanRedo(payload)
+          return false
+        },
+        COMMAND_PRIORITY_CRITICAL,
+      ),
+    )
+  }, [editor])
+
   const items = useMemo(
     (): {
       name: string
       iconName: string
       keywords?: string[]
+      disabled?: boolean
       onSelect: () => void
     }[] => [
+      {
+        name: 'Undo',
+        iconName: 'undo',
+        disabled: !canUndo,
+        onSelect: () => {
+          editor.dispatchCommand(UNDO_COMMAND, undefined)
+        },
+      },
+      {
+        name: 'Redo',
+        iconName: 'redo',
+        disabled: !canRedo,
+        onSelect: () => {
+          editor.dispatchCommand(REDO_COMMAND, undefined)
+        },
+      },
       {
         name: 'Bold',
         iconName: 'bold',
@@ -150,7 +200,7 @@ const MobileToolbarPlugin = () => {
       GetCollapsibleBlock(editor),
       ...GetEmbedsBlocks(editor),
     ],
-    [application.keyboardService, editor, insertLink, showModal],
+    [application.keyboardService, canRedo, canUndo, editor, insertLink, showModal],
   )
 
   useEffect(() => {
@@ -222,10 +272,11 @@ const MobileToolbarPlugin = () => {
           {items.map((item) => {
             return (
               <button
-                className="flex items-center justify-center rounded px-3 py-3"
+                className="flex items-center justify-center rounded px-3 py-3 disabled:opacity-50"
                 aria-label={item.name}
                 onClick={item.onSelect}
                 key={item.name}
+                disabled={item.disabled}
               >
                 <Icon type={item.iconName} size="medium" className="!text-current [&>path]:!text-current" />
               </button>