From d7d13aa6e2ce7ec138eb02261b41ab0c21d05f65 Mon Sep 17 00:00:00 2001
From: Yuxin <55794321+yvonneyx@users.noreply.github.com>
Date: Tue, 16 Jul 2024 16:26:13 +0800
Subject: [PATCH] fix: fix issue that history beforeAddCommand returns false
 (#6042)

* fix: fix issue that history beforeAddCommand returns false

* test: supplement history test
---
 .../plugins/history/plugin-history.spec.ts    | 23 +++++++++++++++++++
 packages/g6/src/plugins/history/index.ts      | 19 ++++++++-------
 2 files changed, 32 insertions(+), 10 deletions(-)

diff --git a/packages/g6/__tests__/unit/plugins/history/plugin-history.spec.ts b/packages/g6/__tests__/unit/plugins/history/plugin-history.spec.ts
index ec3c201d36b..f97cfa11cf8 100644
--- a/packages/g6/__tests__/unit/plugins/history/plugin-history.spec.ts
+++ b/packages/g6/__tests__/unit/plugins/history/plugin-history.spec.ts
@@ -105,4 +105,27 @@ describe('history plugin', () => {
     await expect(graph).toMatchSnapshot(__filename, 'setElementZIndex-redo');
     history.undo();
   });
+
+  it('beforeAddCommand', async () => {
+    const undoStackLen = history.undoStack.length;
+
+    graph.updatePlugin({ key: 'history', beforeAddCommand: () => false });
+    graph.setElementVisibility('node-1', 'hidden');
+    await graph.draw();
+    expect(history.undoStack.length).toEqual(undoStackLen);
+
+    graph.updatePlugin({ key: 'history', beforeAddCommand: () => true });
+    graph.setElementVisibility('node-1', 'visible');
+    await graph.draw();
+    expect(history.undoStack.length).toEqual(undoStackLen + 1);
+  });
+
+  it('canUndo/canRedo/clear', async () => {
+    expect(history.canUndo()).toBeTruthy();
+    expect(history.canRedo()).toBeTruthy();
+    history.clear();
+    expect(history.undoStack.length).toEqual(0);
+    expect(history.canUndo()).toBeFalsy();
+    expect(history.canRedo()).toBeFalsy();
+  });
 });
diff --git a/packages/g6/src/plugins/history/index.ts b/packages/g6/src/plugins/history/index.ts
index 0c27c375bed..6eab6597f52 100644
--- a/packages/g6/src/plugins/history/index.ts
+++ b/packages/g6/src/plugins/history/index.ts
@@ -28,13 +28,13 @@ export interface HistoryOptions extends BasePluginOptions {
    *
    * <en/> Called before a command is added to the Undo/Redo queue. If this method returns false, the command will not be added to the queue. revert is true for undo operations and false for redo operations
    */
-  beforeAddCommand?: (cmd: Command, revert?: boolean) => boolean | void;
+  beforeAddCommand?: (cmd: Command, revert: boolean) => boolean | void;
   /**
    * <zh/> 当一个命令被添加到 Undo/Redo 队列后被调用。revert 为 true 时表示撤销操作，为 false 时表示重做操作
    *
    * <en/> Called after a command is added to the Undo/Redo queue. revert is true for undo operations and false for redo operations
    */
-  afterAddCommand?: (cmd: Command, revert?: boolean) => void;
+  afterAddCommand?: (cmd: Command, revert: boolean) => void;
   /**
    * <zh/> 执行命令时的回调函数
    *
@@ -61,7 +61,6 @@ export class History extends BasePlugin<HistoryOptions> {
   private batchAnimation = false;
   public undoStack: Command[] = [];
   public redoStack: Command[] = [];
-  private isFirstDraw = true;
   private freezed = false;
 
   constructor(context: RuntimeContext, options: HistoryOptions) {
@@ -105,7 +104,10 @@ export class History extends BasePlugin<HistoryOptions> {
     const cmd = this.undoStack.pop();
     if (cmd) {
       this.executeCommand(cmd);
-      this.options.beforeAddCommand?.(cmd, false);
+
+      const before = this.options.beforeAddCommand?.(cmd, false);
+      if (before === false) return;
+
       this.redoStack.push(cmd);
       this.options.afterAddCommand?.(cmd, false);
       this.notify(HistoryEvent.UNDO, cmd);
@@ -160,11 +162,6 @@ export class History extends BasePlugin<HistoryOptions> {
   };
 
   private addCommand = (event: GraphLifeCycleEvent) => {
-    if (this.isFirstDraw) {
-      this.isFirstDraw = false;
-      return;
-    }
-
     if (this.freezed) return;
 
     if (event.type === GraphEvent.AFTER_DRAW) {
@@ -202,7 +199,9 @@ export class History extends BasePlugin<HistoryOptions> {
       this.undoStack.shift();
     }
 
-    this.options.beforeAddCommand?.(cmd, true);
+    const before = this.options.beforeAddCommand?.(cmd, true);
+    if (before === false) return;
+
     this.undoStack.push(cmd);
     this.options.afterAddCommand?.(cmd, true);
   }