feat: allow redo/undo customlink

SSShooter · Nov 4, 2023 · 7323255 · 7323255
1 parent 323d156
commit 7323255
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 8 deletions.
diff --git a/src/customLink.ts b/src/customLink.ts
@@ -129,7 +129,7 @@ export const drawCustomLink = function (this: MindElixirInstance, from: Topic, t
 }
 
 export const createLink = function (this: MindElixirInstance, from: Topic, to: Topic) {
-  const newLinkObj = {
+  const linkObj = {
     id: generateUUID(),
     label: 'Custom Link',
     from: from.nodeObj.id,
@@ -143,7 +143,12 @@ export const createLink = function (this: MindElixirInstance, from: Topic, to: T
       y: -200,
     },
   }
-  this.drawCustomLink(from, to, newLinkObj)
+  this.drawCustomLink(from, to, linkObj)
+
+  this.bus.fire('operation', {
+    name: 'createCustomLink',
+    obj: linkObj,
+  })
 }
 
 export const removeLink = function (this: MindElixirInstance, linkSvg?: CustomSvg) {
@@ -158,6 +163,12 @@ export const removeLink = function (this: MindElixirInstance, linkSvg?: CustomSv
   const id = link.linkObj!.id
   delete this.linkData[id]
   link.remove()
+  this.bus.fire('operation', {
+    name: 'removeCustomLink',
+    obj: {
+      id,
+    },
+  })
 }
 
 export const selectLink = function (this: MindElixirInstance, link: CustomSvg) {
@@ -307,11 +318,10 @@ export function editCutsomLinkLabel(this: MindElixirInstance, el: CustomSvg) {
     if (text === origin) return
     textEl.innerHTML = node.label
     this.linkDiv()
-    // this.bus.fire('operation', {
-    //   name: 'finishEditSummary',
-    //   obj: node,
-    //   origin,
-    // })
+    this.bus.fire('operation', {
+      name: 'finishEditCustomLinkLabel',
+      obj: node,
+    })
   })
   console.timeEnd('editSummary')
 }

diff --git a/src/plugin/operationHistory.ts b/src/plugin/operationHistory.ts
@@ -23,6 +23,11 @@ const calcCurentObject = function (operation: Operation): History['currentObject
       type: 'summary',
       value: (operation as any).obj.id,
     }
+  } else if (['createCustomLink', 'removeCustomLink', 'finishEditCustomLinkLabel'].includes(operation.name)) {
+    return {
+      type: 'customLink',
+      value: (operation as any).obj.id,
+    }
   } else if (['removeNodes'].includes(operation.name)) {
     return {
       type: 'nodes',

diff --git a/src/utils/pubsub.ts b/src/utils/pubsub.ts
@@ -1,3 +1,4 @@
+import type { LinkItem } from '../customLink'
 import type { Summary } from '../summary'
 import type { NodeObj } from '../types/index'
 
@@ -47,7 +48,21 @@ export type SummaryOperation =
       obj: Summary
     }
 
-export type Operation = NodeOperation | SummaryOperation
+export type CustomLinkOperation =
+  | {
+      name: 'createCustomLink'
+      obj: LinkItem
+    }
+  | {
+      name: 'removeCustomLink'
+      obj: { id: string }
+    }
+  | {
+      name: 'finishEditCustomLinkLabel'
+      obj: LinkItem
+    }
+
+export type Operation = NodeOperation | SummaryOperation | CustomLinkOperation
 export type OperationType = Operation['name']
 
 export type EventMap = {