LogBlock/src/main/java/de/diddiz/util/BukkitUtils.java

package de.diddiz.util;

import static de.diddiz.util.MaterialName.materialName;
import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import org.bukkit.ChatColor;
import org.bukkit.Chunk;
import org.bukkit.Location;
import org.bukkit.Material;
import org.bukkit.World;
import org.bukkit.block.BlockFace;
import org.bukkit.block.BlockState;
import org.bukkit.block.DoubleChest;
import org.bukkit.entity.Entity;
import org.bukkit.entity.Player;
import org.bukkit.entity.TNTPrimed;
import org.bukkit.inventory.Inventory;
import org.bukkit.inventory.InventoryHolder;
import org.bukkit.inventory.ItemStack;

public class BukkitUtils
{
	private static final Set<Set<Integer>> blockEquivalents;
	private static final Set<Integer> relativeBreakable;
	private static final Set<Integer> relativeTopBreakable;

	static {
		blockEquivalents = new HashSet<Set<Integer>>(7);
		blockEquivalents.add(new HashSet<Integer>(Arrays.asList(2, 3, 60)));
		blockEquivalents.add(new HashSet<Integer>(Arrays.asList(8, 9, 79)));
		blockEquivalents.add(new HashSet<Integer>(Arrays.asList(10, 11)));
		blockEquivalents.add(new HashSet<Integer>(Arrays.asList(61, 62)));
		blockEquivalents.add(new HashSet<Integer>(Arrays.asList(73, 74)));
		blockEquivalents.add(new HashSet<Integer>(Arrays.asList(75, 76)));
		blockEquivalents.add(new HashSet<Integer>(Arrays.asList(93, 94)));

		relativeBreakable = new HashSet<Integer>(2);
		relativeBreakable.add(63); // Sign
		relativeBreakable.add(68); // Sign
		relativeBreakable.add(65); // Ladder
		relativeBreakable.add(77); // Button

		relativeTopBreakable = new HashSet<Integer>(19);
		relativeTopBreakable.add(6);   ////Vegetation start////
		relativeTopBreakable.add(31);  //                  
		relativeTopBreakable.add(32);  //                  
		relativeTopBreakable.add(37);  //                  
		relativeTopBreakable.add(38);  //                 
		relativeTopBreakable.add(39);  //                 
		relativeTopBreakable.add(40);  //                 
		relativeTopBreakable.add(59);  //                
		relativeTopBreakable.add(81);  //                
		relativeTopBreakable.add(83);  ////Vegetation end////
		relativeTopBreakable.add(27);  // Powered rail
		relativeTopBreakable.add(28);  // Detector rail
		relativeTopBreakable.add(66);  // Rails
		relativeTopBreakable.add(55);  // Redstone
		relativeTopBreakable.add(70);  // Stone pressure plate
		relativeTopBreakable.add(72);  // Wood pressure plate
		relativeTopBreakable.add(78);  // Snow
		relativeTopBreakable.add(93);  // Redstone repeater
		relativeTopBreakable.add(94);  // Redstone repeater
	}

	/**
	 * Returns a list of block locations around the block that are of the type specified by the integer list parameter
	 * 
	 * @param block
	 * @param type
	 * @return List of block locations around the block that are of the type specified by the integer list parameter
	 */
	public static List<Location> getBlocksNearby(org.bukkit.block.Block block, Set<Integer> type) {
		ArrayList<Location> blocks = new ArrayList<Location>();
		if (type.contains(block.getRelative(BlockFace.EAST).getTypeId()))
			blocks.add(block.getRelative(BlockFace.EAST).getLocation());
		if (type.contains(block.getRelative(BlockFace.WEST).getTypeId()))
			blocks.add(block.getRelative(BlockFace.WEST).getLocation());
		if (type.contains(block.getRelative(BlockFace.NORTH).getTypeId()))
			blocks.add(block.getRelative(BlockFace.NORTH).getLocation());
		if (type.contains(block.getRelative(BlockFace.SOUTH).getTypeId()))
			blocks.add(block.getRelative(BlockFace.SOUTH).getLocation());
		if (type.contains(block.getRelative(BlockFace.UP).getTypeId()))
			blocks.add(block.getRelative(BlockFace.UP).getLocation());
		return blocks;
	}

	public static int getInventoryHolderType(InventoryHolder holder) {
		if (holder instanceof DoubleChest) {
			return ((DoubleChest)holder).getLocation().getBlock().getTypeId();
		} else if (holder instanceof BlockState) {
			return ((BlockState)holder).getTypeId();
		} else {
			return -1;
		}
	}

	public static Location getInventoryHolderLocation(InventoryHolder holder) {
		if (holder instanceof DoubleChest) {
			return ((DoubleChest)holder).getLocation();
		} else if (holder instanceof BlockState) {
			return ((BlockState)holder).getLocation();
		} else {
			return null;
		}
	}

	public static ItemStack[] compareInventories(ItemStack[] items1, ItemStack[] items2) {
		final ItemStackComparator comperator = new ItemStackComparator();
		final ArrayList<ItemStack> diff = new ArrayList<ItemStack>();
		final int l1 = items1.length, l2 = items2.length;
		int c1 = 0, c2 = 0;
		while (c1 < l1 || c2 < l2) {
			if (c1 >= l1) {
				diff.add(items2[c2]);
				c2++;
				continue;
			}
			if (c2 >= l2) {
				items1[c1].setAmount(items1[c1].getAmount() * -1);
				diff.add(items1[c1]);
				c1++;
				continue;
			}
			final int comp = comperator.compare(items1[c1], items2[c2]);
			if (comp < 0) {
				items1[c1].setAmount(items1[c1].getAmount() * -1);
				diff.add(items1[c1]);
				c1++;
			} else if (comp > 0) {
				diff.add(items2[c2]);
				c2++;
			} else {
				final int amount = items2[c2].getAmount() - items1[c1].getAmount();
				if (amount != 0) {
					items1[c1].setAmount(amount);
					diff.add(items1[c1]);
				}
				c1++;
				c2++;
			}
		}
		return diff.toArray(new ItemStack[diff.size()]);
	}

	public static ItemStack[] compressInventory(ItemStack[] items) {
		final ArrayList<ItemStack> compressed = new ArrayList<ItemStack>();
		for (final ItemStack item : items)
			if (item != null) {
				final int type = item.getTypeId();
				final byte data = rawData(item);
				boolean found = false;
				for (final ItemStack item2 : compressed)
					if (type == item2.getTypeId() && data == rawData(item2)) {
						item2.setAmount(item2.getAmount() + item.getAmount());
						found = true;
						break;
					}
				if (!found)
					compressed.add(new ItemStack(type, item.getAmount(), (short)0, data));
			}
		Collections.sort(compressed, new ItemStackComparator());
		return compressed.toArray(new ItemStack[compressed.size()]);
	}

	public static boolean equalTypes(int type1, int type2) {
		if (type1 == type2)
			return true;
		for (final Set<Integer> equivalent : blockEquivalents)
			if (equivalent.contains(type1) && equivalent.contains(type2))
				return true;
		return false;
	}

	public static String friendlyWorldname(String worldName) {
		return new File(worldName).getName();
	}

	public static Set<Set<Integer>> getBlockEquivalents() {
		return blockEquivalents;
	}

	public static Set<Integer> getRelativeBreakables() {
		return relativeBreakable;
	}

	public static Set<Integer> getRelativeTopBreakabls() {
		return relativeTopBreakable;
	}

	public static String entityName(Entity entity) {
		if (entity instanceof Player)
			return ((Player)entity).getName();
		if (entity instanceof TNTPrimed)
			return "TNT";
		return entity.getClass().getSimpleName().substring(5);
	}

	public static void giveTool(Player player, int type) {
		final Inventory inv = player.getInventory();
		if (inv.contains(type))
			player.sendMessage(ChatColor.RED + "You have already a " + materialName(type));
		else {
			final int free = inv.firstEmpty();
			if (free >= 0) {
				if (player.getItemInHand() != null && player.getItemInHand().getTypeId() != 0)
					inv.setItem(free, player.getItemInHand());
				player.setItemInHand(new ItemStack(type, 1));
				player.sendMessage(ChatColor.GREEN + "Here's your " + materialName(type));
			} else
				player.sendMessage(ChatColor.RED + "You have no empty slot in your inventory");
		}
	}

	public static byte rawData(ItemStack item) {
		return item.getType() != null ? item.getData() != null ? item.getData().getData() : 0 : 0;
	}

	public static int saveSpawnHeight(Location loc) {
		final World world = loc.getWorld();
		final Chunk chunk = world.getChunkAt(loc);
		if (!world.isChunkLoaded(chunk))
			world.loadChunk(chunk);
		final int x = loc.getBlockX(), z = loc.getBlockZ();
		int y = loc.getBlockY();
		boolean lower = world.getBlockTypeIdAt(x, y, z) == 0, upper = world.getBlockTypeIdAt(x, y + 1, z) == 0;
		while ((!lower || !upper) && y != 127) {
			lower = upper;
			upper = world.getBlockTypeIdAt(x, ++y, z) == 0;
		}
		while (world.getBlockTypeIdAt(x, y - 1, z) == 0 && y != 0)
			y--;
		return y;
	}

	public static int modifyContainer(BlockState b, ItemStack item) {
		if (b instanceof InventoryHolder) {
			final Inventory inv = ((InventoryHolder)b).getInventory();
			if (item.getAmount() < 0) {
				item.setAmount(-item.getAmount());
				final ItemStack tmp = inv.removeItem(item).get(0);
				return tmp != null ? tmp.getAmount() : 0;
			} else if (item.getAmount() > 0) {
				final ItemStack tmp = inv.addItem(item).get(0);
				return tmp != null ? tmp.getAmount() : 0;
			}
		}
		return 0;
	}

	public static boolean canFall(World world, int x, int y, int z) {
		Material mat = world.getBlockAt(x, y, z).getType();

		// Air
		if (mat == Material.AIR) {
			return true;
		} else if (mat == Material.WATER || mat == Material.STATIONARY_WATER || mat == Material.LAVA || mat == Material.STATIONARY_LAVA) { // Fluids
			return true;
		} else if (mat == Material.SIGN || mat == Material.FIRE) { // Misc.
			return true;
		}
		return false;
	}

	public static class ItemStackComparator implements Comparator<ItemStack>
	{
		@Override
		public int compare(ItemStack a, ItemStack b) {
			final int aType = a.getTypeId(), bType = b.getTypeId();
			if (aType < bType)
				return -1;
			if (aType > bType)
				return 1;
			final byte aData = rawData(a), bData = rawData(b);
			if (aData < bData)
				return -1;
			if (aData > bData)
				return 1;
			return 0;
		}
	}
}