From 8067b9bb461ead73b79b039bc3a7d66fc0db40a7 Mon Sep 17 00:00:00 2001
From: Skyler Lehmkuhl <skycooler@gmail.com>
Date: Sat, 7 Dec 2024 15:23:13 -0500
Subject: [PATCH] Revert

---
 src/main.js | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/main.js b/src/main.js
index 02ba0c4..d0db930 100644
--- a/src/main.js
+++ b/src/main.js
@@ -48,6 +48,7 @@ let maxSmoothAngle = 0.6;
 
 let undoStack = [];
 let redoStack = [];
+let lastSaveIndex = 0;
 
 let layoutElements = []
 
@@ -2177,6 +2178,7 @@ async function _save(path) {
     const contents = JSON.stringify(fileData);
     await writeTextFile(path, contents)
     filePath = path
+    lastSaveIndex = undoStack.length - 1;
     console.log(`${path} saved successfully!`);
   } catch (error) {
     console.error("Error saving text file:", error);
@@ -2240,6 +2242,7 @@ async function open() {
             await actions[action.name].execute(action.action)
             undoStack.push(action)
           }
+          lastSaveIndex = undoStack.length - 1;
           updateUI()
         } else {
           await messageDialog(`File ${path} was created in a newer version of Lightningbeam and cannot be opened in this version.`, { title: 'File version mismatch', kind: 'error' });
@@ -2258,6 +2261,12 @@ async function open() {
   }
 }
 
+function revert() {
+  for (let _=0; undoStack.length > lastSaveIndex+1; _++) {
+    undo()
+  }
+}
+
 async function importImage() {
   const path = await openFileDialog({
     multiple: false,
@@ -3428,6 +3437,11 @@ async function updateMenu() {
         enabled: true,
         action: open,
       },
+      {
+        text: 'Revert',
+        enabled: undoStack.length > lastSaveIndex,
+        action: revert,
+      },
       {
         text: 'Import Image...',
         enabled: true,