Merge pull request #343 from MovieReviewComment/feature/issue-302/his…

…tory-actions

[#302] Implement useHistoryActions

ui/src/hooks/editor/use-history-actions.ts

@@ -0,0 +1,54 @@
+'use client';
+
+import { mergeRegister } from '@lexical/utils';
+import {
+  CAN_REDO_COMMAND,
+  CAN_UNDO_COMMAND,
+  COMMAND_PRIORITY_CRITICAL,
+  REDO_COMMAND,
+  UNDO_COMMAND,
+} from 'lexical';
+import type { LexicalEditor } from 'lexical';
+import { useEffect, useState } from 'react';
+
+export function useHistoryActions(editor: LexicalEditor | undefined | null): {
+  canUndo: boolean;
+  canRedo: boolean;
+  undo: () => void;
+  redo: () => void;
+} {
+  const [canUndo, setCanUndo] = useState(false);
+  const [canRedo, setCanRedo] = useState(false);
+
+  useEffect(() => {
+    if (!editor) {
+      return;
+    }
+
+    return mergeRegister(
+      editor.registerCommand(
+        CAN_UNDO_COMMAND,
+        (payload) => {
+          setCanUndo(payload);
+          return false;
+        },
+        COMMAND_PRIORITY_CRITICAL
+      ),
+      editor.registerCommand(
+        CAN_REDO_COMMAND,
+        (payload) => {
+          setCanRedo(payload);
+          return false;
+        },
+        COMMAND_PRIORITY_CRITICAL
+      )
+    );
+  }, [editor]);
+
+  return {
+    canUndo,
+    canRedo,
+    undo: () => editor?.dispatchCommand(UNDO_COMMAND, undefined),
+    redo: () => editor?.dispatchCommand(REDO_COMMAND, undefined),
+  };
+}