Merge pull request #153 from Saibot393/master

Allow undo for movementHUD
theripper93 · Mar 10, 2024 · 05298e2 · 05298e2
2 parents 76c98d0 + ace377c
commit 05298e2
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/scripts/core/components/main/movementHud.js b/scripts/core/components/main/movementHud.js
@@ -17,7 +17,7 @@ export class MovementHud extends ArgonComponent {
   }
 
   set movementUsed(value) {
-    this._movementUsed = value;
+    this._movementUsed = Math.max(value, 0);
   }
 
   get movementMax() { }
@@ -26,15 +26,20 @@ export class MovementHud extends ArgonComponent {
     return movementColors[Math.min(Math.floor((this.movementUsed) / this.movementMax), 2)];
   }
 
-  onTokenUpdate(updates) {
+  onTokenUpdate(updates, context) {
     if (updates.x === undefined && updates.y === undefined) return;
     const ray = new Ray({ x: this.token.x, y: this.token.y }, { x: updates.x ?? this.token.x, y: updates.y ?? this.token.y });
     const segments = [{ ray }];
     const distance = Math.floor(
       canvas.grid.measureDistances(segments, { gridSpaces: true }) /
         canvas.dimensions.distance
     );
-    this.movementUsed += distance;
+    if (context?.isUndo) {
+      this.movementUsed -= distance;
+    }
+    else {
+      this.movementUsed += distance;
+    }
     this.updateMovement();
   }