From 49bb5c3a6692f6f0030e3cd233c7a6a4863f2cb8 Mon Sep 17 00:00:00 2001
From: Piratux <58703216+Piratux@users.noreply.github.com>
Date: Mon, 22 Jan 2024 20:38:50 +0200
Subject: [PATCH] add undo/redo

---
 connect_points/index.html |  8 ++++-
 connect_points/index.js   | 76 ++++++++++++++++++++++++++++++++++++++-
 2 files changed, 82 insertions(+), 2 deletions(-)

diff --git a/connect_points/index.html b/connect_points/index.html
index a34b78d..313b58d 100644
--- a/connect_points/index.html
+++ b/connect_points/index.html
@@ -65,7 +65,9 @@ <h1>
                 Auto connect circles
                 <br />
 
-                <button onclick="reset_grid()">Reset connections</button>
+                <button onclick="reset_grid()" style="margin-right: 2em;">Reset connections</button>
+                <button onclick="perform_undo()">Undo</button>
+                <button onclick="perform_redo()">Redo</button>
             </div>
 
             <div class="function_column">
@@ -104,6 +106,10 @@ <h2>
                 - Hold left mouse button: connect circles with an edge.
                 <br />
                 - Hold right mouse button: disconnect circles by removing edge between them (if there is one).
+                <br />
+                - CTRL+Z: undo connection.
+                <br />
+                - CTRL+Y: redo connection.
             </p>
         </div>
 
diff --git a/connect_points/index.js b/connect_points/index.js
index 1228615..c50433b 100644
--- a/connect_points/index.js
+++ b/connect_points/index.js
@@ -33,6 +33,11 @@ let last_grid_circle_y = -1;
 // Array of connection defining arrays
 let grid_connection_data = [];
 
+// Undo/redo data (stored as stacks)
+let max_memento_size = 100;
+let undo_mementos = [];
+let redo_mementos = [];
+
 // Array of 1 and 0s where 1 = vertex is taken
 let taken_vertices = [];
 
@@ -79,6 +84,18 @@ window.onload = function () {
     setInterval(draw_everything, 1000 / 60);
 }
 
+document.addEventListener('keydown', function (event) {
+    // Check if Ctrl key is pressed and "Z" key is pressed
+    if (event.ctrlKey && event.key === 'z') {
+        // Call your function here
+        perform_undo();
+    }
+    else if (event.ctrlKey && event.key === 'y') {
+        // Call your function here
+        perform_redo();
+    }
+});
+
 function draw_everything() {
     canvas_ctx.fillStyle = 'white';
     canvas_ctx.fillRect(0, 0, canvas.width, canvas.height);
@@ -169,7 +186,6 @@ function draw_grid_circles() {
                 else {
                     draw_circle(pos_x, pos_y, circle_radius, colour_blue);
                 }
-                // draw_circle(pos_x, pos_y, circle_radius, colour_blue);
             }
         }
     }
@@ -281,6 +297,9 @@ function draw_line(from_x, from_y, to_x, to_y, style) {
 }
 
 function reset_grid() {
+    undo_mementos.push(structuredClone(grid_connection_data));
+    redo_mementos = [];
+
     grid_connection_data.length = 0;
     update_connection_info();
 }
@@ -439,6 +458,8 @@ function add_grid_entry_if_needed(mouse_pos_x, mouse_pos_y) {
         simplified_vector.y *= -1;
     }
 
+    undo_mementos.push(structuredClone(grid_connection_data));
+
     for (let i = 0; i < vector_count; i++) {
         grid_connection_data.push([
             a_x + simplified_vector.x * i,
@@ -452,6 +473,20 @@ function add_grid_entry_if_needed(mouse_pos_x, mouse_pos_y) {
     // https://stackoverflow.com/a/44014849
     grid_connection_data = Array.from(new Set(grid_connection_data.map(JSON.stringify)), JSON.parse)
 
+    // revert new action from undo stack, if nothing was added
+    if (undo_mementos[undo_mementos.length - 1].length === grid_connection_data.length) {
+        undo_mementos.pop();
+    }
+    else {
+        // otherwise, new action appeared, clear redo stack
+        redo_mementos = [];
+
+        // don't store too many undo actions
+        if (undo_mementos.length > max_memento_size) {
+            undo_mementos.shift();
+        }
+    }
+
     update_connection_info();
 }
 
@@ -471,6 +506,8 @@ function remove_grid_entry_if_needed(mouse_pos_x, mouse_pos_y) {
         return;
     }
 
+    undo_mementos.push(structuredClone(grid_connection_data));
+
     let arr_to_remove1 = [old_last_grid_circle_x, old_last_grid_circle_y, last_grid_circle_x, last_grid_circle_y];
     let arr_to_remove2 = [last_grid_circle_x, last_grid_circle_y, old_last_grid_circle_x, old_last_grid_circle_y];
 
@@ -491,6 +528,43 @@ function remove_grid_entry_if_needed(mouse_pos_x, mouse_pos_y) {
         }
     }
 
+    // revert new action from undo stack, if nothing was added
+    if (undo_mementos[undo_mementos.length - 1].length === grid_connection_data.length) {
+        undo_mementos.pop();
+    }
+    else {
+        // otherwise, new action appeared, clear redo stack
+        redo_mementos = [];
+
+        // don't store too many undo actions
+        if (undo_mementos.length > max_memento_size) {
+            undo_mementos.shift();
+        }
+    }
+
+    update_connection_info();
+}
+
+// NOTE: Only works with connections. Doesn't work with grid size changes or other parameters.
+function perform_undo() {
+    if (undo_mementos.length === 0) {
+        return;
+    }
+
+    redo_mementos.push(structuredClone(grid_connection_data));
+    grid_connection_data = undo_mementos.pop();
+
+    update_connection_info();
+}
+
+function perform_redo() {
+    if (redo_mementos.length === 0) {
+        return;
+    }
+
+    undo_mementos.push(structuredClone(grid_connection_data));
+    grid_connection_data = redo_mementos.pop();
+
     update_connection_info();
 }