Drop an ES System when the block gets broken.

SeanOMik · Jul 8, 2020 · 77a426a · 77a426a
1 parent 2e67719
commit 77a426a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/net/seanomik/energeticstorage/listeners/BlockBreakListener.java b/src/main/java/net/seanomik/energeticstorage/listeners/BlockBreakListener.java
@@ -5,6 +5,7 @@
 import net.seanomik.energeticstorage.files.PlayersFile;
 import net.seanomik.energeticstorage.objects.ESDrive;
 import net.seanomik.energeticstorage.objects.ESSystem;
+import net.seanomik.energeticstorage.utils.ItemConstructor;
 import net.seanomik.energeticstorage.utils.PermissionChecks;
 import net.seanomik.energeticstorage.utils.Reference;
 import net.seanomik.energeticstorage.utils.Utils;
@@ -34,7 +35,7 @@ public void onBlockBreakListener(BlockBreakEvent event) {
                 if (esSystem != null) {
                     if (esSystem.isPlayerTrusted(player) || esSystem.getOwner().equals(player.getUniqueId()) || PermissionChecks.canDestroyUntrustedSystems(player)) {
                         for (ESDrive drive : esSystem.getESDrives()) {
-                            block.getLocation().getWorld().dropItem(block.getLocation(), drive.getDriveItem());
+                            block.getLocation().getWorld().dropItemNaturally(block.getLocation(), drive.getDriveItem());
                         }
 
                         // Remove the system from cache and storage
@@ -43,6 +44,10 @@ public void onBlockBreakListener(BlockBreakEvent event) {
                         List<ESSystem> systems = new LinkedList<>(Reference.ES_SYSTEMS.get(player.getUniqueId()));
                         systems.removeIf(esSystem::equals);
                         Reference.ES_SYSTEMS.replace(player.getUniqueId(), systems);
+
+                        // Drop an ES System
+                        event.setDropItems(false);
+                        event.getBlock().getLocation().getWorld().dropItemNaturally(event.getBlock().getLocation(), ItemConstructor.createSystemBlock());
                     } else {
                         event.setCancelled(true);
                         player.sendMessage(Reference.PREFIX + ChatColor.RED + "You are not trusted to this system!");