Slimefun4/src/me/mrCookieSlime/Slimefun/api/energy/ChargableBlock.java

package me.mrCookieSlime.Slimefun.api.energy;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import me.mrCookieSlime.CSCoreLibPlugin.Configuration.Config;
import me.mrCookieSlime.CSCoreLibPlugin.general.Math.DoubleHandler;
import me.mrCookieSlime.CSCoreLibPlugin.general.World.CustomSkull;
import me.mrCookieSlime.Slimefun.SlimefunStartup;
import me.mrCookieSlime.Slimefun.api.BlockStorage;

import org.bukkit.Bukkit;
import org.bukkit.Location;
import org.bukkit.block.Block;
import org.bukkit.block.Skull;

public class ChargableBlock {
	
	public static Map<String, Integer> max_charges = new HashMap<String, Integer>();
	public static Set<String> rechargeable = new HashSet<String>();
	public static Set<String> capacitors = new HashSet<String>();
	
	public static void registerChargableBlock(String id, int capacity, boolean recharge) {
		max_charges.put(id, capacity);
		if (recharge) rechargeable.add(id);
	}
	
	public static void registerCapacitor(String id, int capacity) {
		max_charges.put(id, capacity);
		rechargeable.add(id);
		capacitors.add(id);
	}
	
	public static boolean isChargable(Block b) {
		return isChargable(b.getLocation());
	}
	
	public static boolean isChargable(Location l) {
		if (!BlockStorage.hasBlockInfo(l)) return false;
		return max_charges.containsKey(BlockStorage.checkID(l));
	}
	
	public static boolean isRechargable(Block b) {
		if (!BlockStorage.hasBlockInfo(b)) return false;
		String id = BlockStorage.checkID(b);
		return max_charges.containsKey(id) && rechargeable.contains(id);
	}
	
	public static boolean isCapacitor(Block b) {
		return isCapacitor(b.getLocation());
	}
	
	public static boolean isCapacitor(Location l) {
		if (!BlockStorage.hasBlockInfo(l)) return false;
		return capacitors.contains(BlockStorage.checkID(l));
	}
	
	public static int getDefaultCapacity(Block b) {
		return getDefaultCapacity(b.getLocation());
	}
	
	public static int getDefaultCapacity(Location l) {
		return max_charges.get(BlockStorage.checkID(l));
	}
	
	public static int getCharge(Block b) {
		return getCharge(b.getLocation());
	}
	
	public static int getCharge(Location l) {
		String charge = BlockStorage.getBlockInfo(l, "energy-charge");
		if (charge != null) return Integer.parseInt(charge);
		else {
			BlockStorage.addBlockInfo(l, "energy-charge", "0", false);
			return 0;
		}
	}
	
	public static void setCharge(Block b, int charge) {
		setCharge(b.getLocation(), charge);
	}
	
	public static void setCharge(Location l, int charge) {
		if (charge < 0) charge = 0;
		else {
			int capacity = getMaxCharge(l);
			if (charge > capacity) charge = capacity;
		}
		if (charge != getCharge(l)) BlockStorage.addBlockInfo(l, "energy-charge", String.valueOf(charge), false);
	}
	
	public static void setUnsafeCharge(Location l, int charge, boolean updateTexture) {
		if (charge != getCharge(l)) {
			BlockStorage.addBlockInfo(l, "energy-charge", String.valueOf(charge), false);
			if (updateTexture) {
				try {
					updateTexture(l);
				} catch (Exception e) {
					e.printStackTrace();
				}
			}
		}
	}
	
	private static void updateTexture(final Location l) throws Exception {
		Bukkit.getScheduler().scheduleSyncDelayedTask(SlimefunStartup.instance, new Runnable() {
			
			@Override
			public void run() {
				try {
					Block b = l.getBlock();
					int charge = getCharge(b), capacity = getMaxCharge(b);
					if (b.getState() instanceof Skull) {
						if (charge < (int) (capacity * 0.25D)) CustomSkull.setSkull(b, "eyJ0ZXh0dXJlcyI6eyJTS0lOIjp7InVybCI6Imh0dHA6Ly90ZXh0dXJlcy5taW5lY3JhZnQubmV0L3RleHR1cmUvOTEzNjFlNTc2YjQ5M2NiZmRmYWUzMjg2NjFjZWRkMWFkZDU1ZmFiNGU1ZWI0MThiOTJjZWJmNjI3NWY4YmI0In19fQ==");
						else if (charge < (int) (capacity * 0.5D)) CustomSkull.setSkull(b, "eyJ0ZXh0dXJlcyI6eyJTS0lOIjp7InVybCI6Imh0dHA6Ly90ZXh0dXJlcy5taW5lY3JhZnQubmV0L3RleHR1cmUvMzA1MzIzMzk0YTdkOTFiZmIzM2RmMDZkOTJiNjNjYjQxNGVmODBmMDU0ZDA0NzM0ZWEwMTVhMjNjNTM5In19fQ==");
						else if (charge < (int) (capacity * 0.75D)) CustomSkull.setSkull(b, "eyJ0ZXh0dXJlcyI6eyJTS0lOIjp7InVybCI6Imh0dHA6Ly90ZXh0dXJlcy5taW5lY3JhZnQubmV0L3RleHR1cmUvNTU4NDQzMmFmNmYzODIxNjcxMjAyNThkMWVlZThjODdjNmU3NWQ5ZTQ3OWU3YjBkNGM3YjZhZDQ4Y2ZlZWYifX19");
						else CustomSkull.setSkull(b, "eyJ0ZXh0dXJlcyI6eyJTS0lOIjp7InVybCI6Imh0dHA6Ly90ZXh0dXJlcy5taW5lY3JhZnQubmV0L3RleHR1cmUvN2EyNTY5NDE1YzE0ZTMxYzk4ZWM5OTNhMmY5OWU2ZDY0ODQ2ZGIzNjdhMTNiMTk5OTY1YWQ5OWM0MzhjODZjIn19fQ==");
					}
				} catch (Exception e) {
					e.printStackTrace();
				}
			}
		});
	}
	
	public static String formatEnergy(Block b) {
		return DoubleHandler.getFancyDouble(getCharge(b)) + " J";
	}
	
	public static int addCharge(Block b, int charge) {
		return addCharge(b.getLocation(), charge);
	}
	
	public static int addCharge(Location l, int charge) {
		int space = getMaxCharge(l) - getCharge(l);
		int rest = charge;
		if (space > 0) {
			if (space > charge) {
				setCharge(l, getCharge(l) + charge);
				rest = 0;
			}
			else {
				rest = charge - space;
				setCharge(l, getMaxCharge(l));
			}
			if (capacitors.contains(BlockStorage.checkID(l))) {
				try {
					updateTexture(l);
				} catch (Exception e) {
					e.printStackTrace();
				}
			}
		}
		return rest;
	}
	
	public static int getMaxCharge(Block b) {
		return getMaxCharge(b.getLocation());
	}
	
	public static int getMaxCharge(Location l) {
		Config cfg = BlockStorage.getBlockInfo(l);
		if (cfg.contains("energy-capacity")) return Integer.parseInt(cfg.getString("energy-capacity"));
		else {
			BlockStorage.addBlockInfo(l, "energy-capacity", String.valueOf(getDefaultCapacity(l)), false);
			return getDefaultCapacity(l);
		}
	}

}