Fixed SmartReset logic for user placed blocks

(MysticCity/MinigamesAPI#169)
MysticCity · Oct 16, 2016 · 6b072b7 · 6b072b7
1 parent 58c410f
commit 6b072b7
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/main/java/com/comze_instancelabs/mgskywars/Main.java b/src/main/java/com/comze_instancelabs/mgskywars/Main.java
@@ -167,11 +167,20 @@ public void onBlockPlace(BlockPlaceEvent event) {
 				event.setCancelled(true);
 			}
 		}
-		if (event.getBlock().getType() == Material.CHEST) {
+		else if (event.getBlock().getType() == Material.CHEST) {
 			if (pli.global_players.containsKey(event.getPlayer().getName())) {
 				temp.add(event.getBlock().getLocation());
 			}
 		}
+		else {
+			if (pli.global_players.containsKey(event.getPlayer().getName())) {
+				final Arena a = this.pli.global_players.get(event.getPlayer().getName());
+				if (a.getArenaState() == ArenaState.INGAME)
+				{
+					 a.getSmartReset().addChanged(event.getBlock(), event.getBlockReplacedState());
+				}
+			}
+		}
 	}
 
 	@EventHandler