From cd45c3a01b36110bc42c7b821d022b66fe7a24da Mon Sep 17 00:00:00 2001
From: Azgaar <azgaar.fmg@yandex.com>
Date: Sun, 23 Jun 2024 14:18:57 +0200
Subject: [PATCH] feat: update data on control point remove

---
 modules/ui/routes-editor.js | 36 +++++++++++++++++++++++++++---------
 1 file changed, 27 insertions(+), 9 deletions(-)

diff --git a/modules/ui/routes-editor.js b/modules/ui/routes-editor.js
index 93ed0e1a..c68061be 100644
--- a/modules/ui/routes-editor.js
+++ b/modules/ui/routes-editor.js
@@ -47,8 +47,7 @@ function editRoute(id) {
 
   function getRoute() {
     const routeId = +elSelected.attr("id").slice(5);
-    const route = pack.routes.find(r => r.i === routeId);
-    return route;
+    return pack.routes.find(route => route.i === routeId);
   }
 
   function updateRouteData() {
@@ -145,7 +144,6 @@ function editRoute(id) {
 
   function addControlPoint() {
     const [x, y] = d3.mouse(this);
-
     const route = getRoute();
     if (!route.points) route.points = debug.selectAll("#controlPoints > *").data();
 
@@ -172,6 +170,32 @@ function editRoute(id) {
     redrawRoute();
   }
 
+  function removeControlPoint() {
+    const route = getRoute();
+
+    if (!route.points) route.points = debug.selectAll("#controlPoints > *").data();
+    const [x, y] = d3.select(this).datum();
+    const cellId = findCell(x, y);
+    const routeAllCells = route.points.map(([x, y]) => findCell(x, y));
+
+    const isOnlyPointInCell = routeAllCells.filter(cell => cell === cellId).length === 1;
+    if (isOnlyPointInCell) {
+      const index = route.cells.indexOf(cellId);
+      const prev = route.cells[index - 1];
+      const next = route.cells[index + 1];
+      if (prev) removeConnection(prev, cellId);
+      if (next) removeConnection(cellId, next);
+      if (prev && next) addConnection(prev, next, route.i);
+    }
+
+    this.remove();
+    route.points = debug.selectAll("#controlPoints > *").data();
+    route.cells = unique(route.points.map(([x, y]) => findCell(x, y)));
+
+    drawCells();
+    redrawRoute();
+  }
+
   function showCreationDialog() {
     const route = getRoute();
     createRoute(route.group);
@@ -193,12 +217,6 @@ function editRoute(id) {
     routes[to][from] = routeId;
   }
 
-  function removeControlPoint() {
-    this.remove();
-    redrawRoute();
-    drawCells();
-  }
-
   function changeName() {
     getRoute().name = this.value;
   }