feat: remove related Line, when there is only one Block (#50)

orzyyyy · Apr 14, 2019 · 397f376 · 397f376
1 parent 20d366e
commit 397f376
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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({});
   };