From 854ef71f9d26635f73780cac7a65e7fc23b37bab Mon Sep 17 00:00:00 2001
From: zy410419243 <zy410419243@gmail.com>
Date: Sun, 14 Apr 2019 11:41:37 +0800
Subject: [PATCH] feat: remove related Line, when there is only one Block

---
 src/canvas/core.tsx | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/canvas/core.tsx b/src/canvas/core.tsx
index f2a3c141..544f425e 100644
--- a/src/canvas/core.tsx
+++ b/src/canvas/core.tsx
@@ -161,6 +161,12 @@ export default class Canvas extends Component<CanvasProps, CanvasState> {
   handleRightClick = ({ key, event, group }: ContextMenuProps) => {
     preventDefault(event);
     delete dataCollector[group][key];
+    for (const lineKey of Object.keys(dataCollector.LineGroup)) {
+      const { fromKey, toKey } = dataCollector.LineGroup[lineKey];
+      if (group === 'BlockGroup' && (fromKey === key || toKey === key)) {
+        delete dataCollector.LineGroup[lineKey];
+      }
+    }
     this.setState({});
   };