extras/src/main/java/pw/kaboom/extras/modules/block/BlockPhysics.java

package pw.kaboom.extras.modules.block;

import java.util.HashSet;

import org.bukkit.Bukkit;
import org.bukkit.Material;
import org.bukkit.block.BlockFace;
import org.bukkit.entity.EntityType;
import org.bukkit.event.EventHandler;
import org.bukkit.event.Listener;
import org.bukkit.event.block.BlockFormEvent;
import org.bukkit.event.block.BlockFromToEvent;
import org.bukkit.event.block.BlockPhysicsEvent;
import org.bukkit.event.block.BlockRedstoneEvent;
import org.bukkit.event.entity.EntityChangeBlockEvent;

import com.destroystokyo.paper.event.block.BlockDestroyEvent;

public class BlockPhysics implements Listener {
	public static HashSet<BlockFace> blockFaces = new HashSet<BlockFace>();

	@EventHandler
	void onBlockForm(BlockFormEvent event) {
		if (event.getBlock().getType() == Material.LAVA ||
			event.getBlock().getType() == Material.WATER) {
			for (BlockFace face : blockFaces) {
				if (event.getBlock().getRelative(face).getType() != Material.LAVA &&
					event.getBlock().getRelative(face).getType() != Material.WATER) {
					return;
				}
				event.setCancelled(true);
			}
		}
	}

	@EventHandler
	void onBlockFromTo(BlockFromToEvent event) {
		if (event.getBlock().getType() == Material.LAVA ||
			event.getBlock().getType() == Material.WATER) {
			boolean lavaFound = false;
			boolean waterFound = false;

			for (BlockFace face : blockFaces) {
				if (event.getBlock().getRelative(face).getType() == Material.LAVA) {
					lavaFound = true;
				} else if (event.getBlock().getRelative(face).getType() == Material.WATER) {
					waterFound = true;
				}

				if (lavaFound && waterFound) {
					event.setCancelled(true);
				}
			}
		}
	}

	@EventHandler
	void onBlockDestroy(BlockDestroyEvent event) {
		if (!event.getBlock().getType().isSolid()) {
			for (BlockFace face : blockFaces) {
				if (event.getBlock().getRelative(face).getType() != event.getBlock().getType()) {
					return;
				}
				event.getBlock().setType(Material.AIR, false);
				event.setCancelled(true);
			}
		}
	}

	@EventHandler
	void onBlockPhysics(BlockPhysicsEvent event) {
		switch (event.getChangedType()) {
		case COMPARATOR:
		case REDSTONE_TORCH:
		case REDSTONE_WIRE:
		case REPEATER:
			/*for (BlockFace face : blockFaces) {
				if (event.getBlock().getRelative(face).getType() != event.getChangedType()) {
					return;
				}
				event.setCancelled(true);
			}*/
			event.setCancelled(true);
			break;
		case ACTIVATOR_RAIL:
		case DETECTOR_RAIL:
		case POWERED_RAIL:
		case RAIL:
			for (BlockFace face : blockFaces) {
				if (event.getBlock().getRelative(face).getType() != event.getChangedType()) {
					return;
				}
				event.getBlock().setType(Material.AIR, false);
				event.setCancelled(true);
			}
		default:
			break;
		}
	}

	@EventHandler
	void onBlockRedstone(BlockRedstoneEvent event) {
		final double tps = Bukkit.getServer().getTPS()[0];

		if (tps < 10) {
			event.setNewCurrent(0);
		}
	}

	int fallingBlockCount;

	@EventHandler
	void onEntityChangeBlock(EntityChangeBlockEvent event) {
		if (event.getEntityType() == EntityType.FALLING_BLOCK &&
			event.getTo() == Material.AIR) {
			fallingBlockCount++;

			if (fallingBlockCount == 10) {
				event.setCancelled(true);
				fallingBlockCount = 0;
			}
		}
	}
}