From 0669e73a5b7de2a6f612ab9f86aee2356371f2ae Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=A1=97=E8=A7=92=E5=B0=8F=E6=9E=97?= <1013335014@qq.com>
Date: Mon, 29 Jul 2024 09:49:05 +0800
Subject: [PATCH] =?UTF-8?q?Feat=EF=BC=9A=E6=94=AF=E6=8C=81=E7=82=B9?=
 =?UTF-8?q?=E5=87=BB=E7=94=BB=E5=B8=83=E5=8F=96=E6=B6=88=E5=88=9B=E5=BB=BA?=
 =?UTF-8?q?=E5=85=B3=E8=81=94=E7=BA=BF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/plugins/AssociativeLine.js            | 30 +++++++++++++------
 1 file changed, 21 insertions(+), 9 deletions(-)

diff --git a/simple-mind-map/src/plugins/AssociativeLine.js b/simple-mind-map/src/plugins/AssociativeLine.js
index 3878e3df..51ee8b2f 100644
--- a/simple-mind-map/src/plugins/AssociativeLine.js
+++ b/simple-mind-map/src/plugins/AssociativeLine.js
@@ -68,6 +68,7 @@ class AssociativeLine {
     this.onNodeDragging = this.onNodeDragging.bind(this)
     this.onNodeDragend = this.onNodeDragend.bind(this)
     this.onControlPointMouseup = this.onControlPointMouseup.bind(this)
+    this.cancelCreateLine = this.cancelCreateLine.bind(this)
 
     // 节点树渲染完毕后渲染连接线
     this.mindMap.on('node_tree_render_end', this.renderAllLines)
@@ -76,6 +77,7 @@ class AssociativeLine {
     // 监听画布和节点点击事件，用于清除当前激活的连接线
     this.mindMap.on('draw_click', this.onDrawClick)
     this.mindMap.on('node_click', this.onNodeClick)
+    this.mindMap.on('contextmenu', this.cancelCreateLine)
     // 注册删除快捷键
     this.mindMap.keyCommand.addShortcut('Del|Backspace', this.removeLine)
     // 注册添加连接线的命令
@@ -97,6 +99,7 @@ class AssociativeLine {
     this.mindMap.off('data_change', this.renderAllLines)
     this.mindMap.off('draw_click', this.onDrawClick)
     this.mindMap.off('node_click', this.onNodeClick)
+    this.mindMap.off('contextmenu', this.cancelCreateLine)
     this.mindMap.keyCommand.removeShortcut('Del|Backspace', this.removeLine)
     this.mindMap.command.remove('ADD_ASSOCIATIVE_LINE', this.addLine)
     this.mindMap.off('mousemove', this.onMousemove)
@@ -108,10 +111,14 @@ class AssociativeLine {
 
   // 画布点击事件
   onDrawClick() {
-    if (this.isControlPointMousedown) {
-      return
+    // 取消创建关联线
+    if (this.isCreatingLine) {
+      this.cancelCreateLine()
+    }
+    // 取消激活关联线
+    if (this.isControlPointMousedown) {
+      this.clearActiveLine()
     }
-    this.clearActiveLine()
   }
 
   // 节点点击事件
@@ -346,6 +353,16 @@ class AssociativeLine {
     this.creatingLine.marker('end', this.marker)
   }
 
+  // 取消创建关联线
+  cancelCreateLine() {
+    this.isCreatingLine = false
+    this.creatingStartNode = null
+    this.creatingLine.remove()
+    this.creatingLine = null
+    this.overlapNode = null
+    this.back()
+  }
+
   // 鼠标移动事件
   onMousemove(e) {
     this.onControlPointMousemove(e)
@@ -420,12 +437,7 @@ class AssociativeLine {
     if (this.overlapNode && this.overlapNode.getData('isActive')) {
       this.mindMap.execCommand('SET_NODE_ACTIVE', this.overlapNode, false)
     }
-    this.isCreatingLine = false
-    this.creatingStartNode = null
-    this.creatingLine.remove()
-    this.creatingLine = null
-    this.overlapNode = null
-    this.back()
+    this.cancelCreateLine()
   }
 
   // 添加连接线