Merge pull request #110 from BlockHorizons/Chris-Prime-patch-1

Improved inputs for Undo command (/undo all)
BlockHorizons · Oct 5, 2018 · 00b94a7 · 00b94a7
2 parents 681f411 + 98196a0
commit 00b94a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/BlockHorizons/BlockSniper/commands/UndoCommand.php b/src/BlockHorizons/BlockSniper/commands/UndoCommand.php
@@ -29,7 +29,8 @@ public function onExecute(CommandSender $sender, string $commandLabel, array $ar
 		$undoAmount = 1;
 		if(isset($args[0])){
 			$undoAmount = (int) $args[0];
-			if($undoAmount > ($totalUndo = SessionManager::getPlayerSession($sender)->getRevertStore()->getTotalStores(Revert::TYPE_UNDO))){
+			$totalUndo = SessionManager::getPlayerSession($sender)->getRevertStore()->getTotalStores(Revert::TYPE_UNDO);
+			if($undoAmount > $totalUndo || $args[0] === "all"){
 				$undoAmount = $totalUndo;
 			}
 		}