package magicbees.bees;

import magicbees.api.MagicBeesAPI;
import magicbees.bees.allele.effect.AlleleEffectCrumbling;
import magicbees.bees.allele.effect.AlleleEffectCure;
import magicbees.bees.allele.effect.AlleleEffectEmpowering;
import magicbees.bees.allele.effect.AlleleEffectNodeConversion;
import magicbees.bees.allele.effect.AlleleEffectNodeRepair;
import magicbees.bees.allele.effect.AlleleEffectPlaceholder;
import magicbees.bees.allele.effect.AlleleEffectPotion;
import magicbees.bees.allele.effect.AlleleEffectRecharge;
import magicbees.bees.allele.effect.AlleleEffectSpawnMob;
import magicbees.bees.allele.effect.AlleleEffectSpawnMobWeighted;
import magicbees.bees.allele.effect.AlleleEffectSpawnWisp;
import magicbees.bees.allele.effect.AlleleEffectTransmuting;
import magicbees.bees.allele.effect.TransmutationEffectController;
import magicbees.bees.allele.effect.TransmutationEffectLBotaniaLiving;
import magicbees.bees.allele.effect.TransmutationEffectRailcraft;
import magicbees.bees.allele.effect.TransmutationEffectVanilla;
import magicbees.bees.allele.flowerProvider.FlowerProviderArsMagicaFlower;
import magicbees.bees.allele.flowerProvider.FlowerProviderAuraNode;
import magicbees.bees.allele.flowerProvider.FlowerProviderBookshelf;
import magicbees.bees.allele.flowerProvider.FlowerProviderBotania;
import magicbees.bees.allele.flowerProvider.FlowerProviderThaumcraftFlower;
import magicbees.main.utils.LocalizationManager;
import magicbees.main.utils.compat.ArsMagicaHelper;
import magicbees.main.utils.compat.BotaniaHelper;
import magicbees.main.utils.compat.ThaumcraftHelper;
import magicbees.main.utils.compat.ThermalModsHelper;
import net.minecraft.potion.Potion;
import thaumcraft.api.nodes.NodeType;

import forestry.api.apiculture.EnumBeeChromosome;
import forestry.api.apiculture.IAlleleBeeEffect;
import forestry.api.apiculture.IAlleleBeeSpecies;
import forestry.api.genetics.AlleleManager;
import forestry.api.genetics.IAllele;
import forestry.api.genetics.IAlleleEffect;
import forestry.api.genetics.IAlleleFlowers;
import forestry.api.genetics.IAlleleRegistry;
import forestry.api.genetics.IChromosomeType;
import forestry.api.genetics.IFlowerProvider;

public class Allele implements IAllele {
	
	private static final String FLOWER_ARS_MAGICA_PLANT = "flowerArsMagicaPlant";
	private static final String FLOWER_BOTANIA = "flowerBotania";
	private static final String EFFECT_TE_BLIZZY = "TEBlizzy";
	private static final String EFFECT_TE_BLITZ = "TEBlitz";
	private static final String EFFECT_TE_BASALZ = "TEBasalz";
	private static final String EFFECT_AM_WISP = "AMWisp";
	private static final String EFFECT_MANA_DRAIN = "ManaDrain";
	private static final String EFFECT_DREAMING = "Dreaming";
	private static final String EFFECT_WISPY = "Wispy";
	private static final String EFFECT_BATTY = "Batty";
	private static final String EFFECT_BRAINY = "Brainy";
	private static final String EFFECT_NODE_RAVENING = "NodeRavening";
	private static final String EFFECT_NODE_PURIFYING = "NodePurifying";
	private static final String EFFECT_NODE_TAINTING = "NodeTainting";
	private static final String EFFECT_NODE_REPAIR = "NodeRepair";
	private static final String EFFECT_NODE_EMPOWER = "NodeEmpower";
	private static final String EFFECT_VIS_RECHARGE = "VisRecharge";
	private static final String FLOWERS_AURA_NODE = "AuraNode";
	private static final String FLOWERS_THAUMCRAFT_PLANT = "ThaumcraftPlant";

	public static AlleleFloat speedBlinding;

	public static IAlleleBeeEffect forestryBaseEffect;

	public static IAlleleFlowers flowerBookshelf;
	public static IAlleleFlowers flowerThaumcraft;
	public static IAlleleFlowers flowerArsMagica;
	public static IAlleleFlowers flowerAuraNode;
	public static IAlleleFlowers flowerBotania;

	public static IAlleleEffect effectCleansing;
	public static IAlleleEffect effectDigSpeed;
	public static IAlleleEffect effectMoveSpeed;
	public static IAlleleEffect effectSlowSpeed;
	public static IAlleleEffect effectWithering;
	public static IAlleleEffect effectVisRecharge;
	public static IAlleleEffect effectNodeEmpower;
	public static IAlleleEffect effectNodeRepair;
	public static IAlleleEffect effectNodeConversionTaint;
	public static IAlleleEffect effectNodeConversionPure;
	public static IAlleleEffect effectNodeConversionHungry;
	public static IAlleleEffect effectTransmuting;
	public static IAlleleEffect effectCrumbling;
	public static IAlleleEffect effectInvisibility;
	public static IAlleleEffect effectDreaming;

	public static IAlleleEffect spawnWolf;
	public static IAlleleEffect spawnBats;
	public static IAlleleEffect spawnCow;
	public static IAlleleEffect spawnChicken;
	public static IAlleleEffect spawnPig;
	public static IAlleleEffect spawnSheep;
	public static IAlleleEffect spawnCat;
	public static IAlleleEffect spawnHorse;
	public static IAlleleEffect spawnGhast;
	public static IAlleleEffect spawnSpider;
	public static IAlleleEffect spawnBlaze;
	public static IAlleleEffect spawnBrainyZombie;
	public static IAlleleEffect spawnWisp;
	public static IAlleleEffect spawnBlizz;
	public static IAlleleEffect spawnBlitz;
	public static IAlleleEffect spawnBasalz;
	public static IAlleleEffect spawnManaDrainer;
	public static IAlleleEffect spawnWispOrHecate;

	public static void setupAdditionalAlleles() {
		forestryBaseEffect = (IAlleleBeeEffect) getBaseAllele("effectNone");
		IFlowerProvider vanillaFlowers = ((IAlleleFlowers)getBaseAllele("flowersVanilla")).getProvider();

		speedBlinding = new AlleleFloat("speedBlinding", 2f, false, EnumBeeChromosome.SPEED);
		flowerBookshelf = new AlleleFlower("Bookshelf", new FlowerProviderBookshelf(), true);

		effectCleansing = new AlleleEffectCure("Curative", false);
		effectDigSpeed = new AlleleEffectPotion("DigSpeed", Potion.digSpeed, 15, false);
		effectMoveSpeed = new AlleleEffectPotion("MoveSpeed", Potion.moveSpeed, 10, false);
		effectSlowSpeed = new AlleleEffectPotion("SlowSpeed", Potion.moveSlowdown, 3, false).setMalicious();
		effectWithering = new AlleleEffectPotion("Withering", Potion.wither, 10, false).setMalicious();

		TransmutationEffectController controller = new TransmutationEffectController(
				new TransmutationEffectRailcraft(),
				new TransmutationEffectVanilla()
		);
		MagicBeesAPI.transmutationEffectController = controller;
		effectTransmuting = new AlleleEffectTransmuting("Transmuting", true, controller, 200);
		effectCrumbling = new AlleleEffectCrumbling("Crumbling", true);

		effectInvisibility = new AlleleEffectPotion("Invisibility", Potion.invisibility, 10, false);

		spawnGhast = new AlleleEffectSpawnMob("Ghastly", false, "Ghast", "mob.ghast.moan")
				.setThrottle(2060)
				.setChanceToSpawn(10)
				.setMaxMobsInSpawnZone(1);
		spawnSpider = new AlleleEffectSpawnMob("Spidery", false, "Spider", "mob.spider.step")
				.setThrottle(400)
				.setChanceToSpawn(70)
				.setMaxMobsInSpawnZone(4);
		spawnBlaze = new AlleleEffectSpawnMob("Ablaze", false, "Blaze", "mob.blaze.breathe")
				.setThrottle(800)
				.setChanceToSpawn(60)
				.setMaxMobsInSpawnZone(2);
		spawnWolf = new AlleleEffectSpawnMob("Canine", false , "Wolf", "mob.wolf.panting")
				.setThrottle(650)
				.setChanceToSpawn(40)
				.setMaxMobsInSpawnZone(2);
		spawnPig = new AlleleEffectSpawnMob("Porcine", true, "Pig", "mob.pig.say")
				.setThrottle(350)
				.setMaxMobsInSpawnZone(4);
		spawnCow = new AlleleEffectSpawnMob("Bovine", true, "Cow", "mob.cow.say")
				.setThrottle(640)
				.setMaxMobsInSpawnZone(3);
		spawnChicken = new AlleleEffectSpawnMob("Chicken", true, "Chicken", "mob.chicken.hurt")
				.setThrottle(20)
				.setMaxMobsInSpawnZone(20);
		spawnSheep = new AlleleEffectSpawnMob("Sheep", false, "Sheep", "mob.sheep.say")
				.setThrottle(450)
				.setMaxMobsInSpawnZone(5);
		spawnCat = new AlleleEffectSpawnMob("Catty", false, "Ozelot", "mob.cat.meow")
				.setThrottle(702)
				.setChanceToSpawn(60)
				.setMaxMobsInSpawnZone(2);
		spawnHorse = new AlleleEffectSpawnMob("Horse", false, "EntityHorse", "mob.horse.idle")
				.setThrottle(450)
				.setChanceToSpawn(59)
				.setMaxMobsInSpawnZone(2);

		if (ThaumcraftHelper.isActive()) {
			flowerThaumcraft = new AlleleFlower(FLOWERS_THAUMCRAFT_PLANT, new FlowerProviderThaumcraftFlower(), false);
			flowerAuraNode = new AlleleFlower(FLOWERS_AURA_NODE, new FlowerProviderAuraNode(), true);
			
			effectVisRecharge = new AlleleEffectRecharge(EFFECT_VIS_RECHARGE, false);
			effectNodeEmpower = new AlleleEffectEmpowering(EFFECT_NODE_EMPOWER, false);
			effectNodeRepair = new AlleleEffectNodeRepair(EFFECT_NODE_REPAIR, false);
			effectNodeConversionTaint = new AlleleEffectNodeConversion(EFFECT_NODE_TAINTING, NodeType.TAINTED, false, 250);
			effectNodeConversionPure = new AlleleEffectNodeConversion(EFFECT_NODE_PURIFYING, NodeType.PURE, false, 250);
			effectNodeConversionHungry = new AlleleEffectNodeConversion(EFFECT_NODE_RAVENING, NodeType.HUNGRY, false, 2);

			spawnBrainyZombie = new AlleleEffectSpawnMob(EFFECT_BRAINY, false, ThaumcraftHelper.Entity.BRAINY_ZOMBIE.entityID)
				.setAggrosPlayerOnSpawn()
				.setThrottle(800)
				.setSpawnsOnPlayerNear(null)
				.setMaxMobsInSpawnZone(2);

			spawnBats = new AlleleEffectSpawnMob(EFFECT_BATTY, false, ThaumcraftHelper.Entity.FIREBAT.entityID)
				.setThrottle(300)
				.setSpawnsOnPlayerNear("Bat");

			spawnWisp = new AlleleEffectSpawnWisp(EFFECT_WISPY, false, ThaumcraftHelper.Entity.WISP.entityID, "thaumcraft.wisplive")
				.setThrottle(1800)
				.setChanceToSpawn(79);
		}
		else {
			spawnBats = new AlleleEffectSpawnMob(EFFECT_BATTY, false, "Bat")
				.setThrottle(150);
			
			flowerThaumcraft = new AlleleFlower(FLOWERS_THAUMCRAFT_PLANT, vanillaFlowers, false);
			flowerAuraNode = new AlleleFlower(FLOWERS_AURA_NODE, vanillaFlowers, true);Allele.effectVisRecharge = new AlleleEffectRecharge(EFFECT_VIS_RECHARGE, false);
			effectNodeEmpower = new AlleleEffectPlaceholder(EFFECT_NODE_EMPOWER, false);
			effectNodeRepair = new AlleleEffectPlaceholder(EFFECT_NODE_REPAIR, false);
			effectNodeConversionTaint = new AlleleEffectPlaceholder(EFFECT_NODE_TAINTING, false);
			effectNodeConversionPure = new AlleleEffectPlaceholder(EFFECT_NODE_PURIFYING, false);
			effectNodeConversionHungry = new AlleleEffectPlaceholder(EFFECT_NODE_RAVENING, false);
			spawnBrainyZombie = new AlleleEffectPlaceholder(EFFECT_BRAINY, false);
			spawnWisp = new AlleleEffectPlaceholder(EFFECT_WISPY, false);
		}

		if (BotaniaHelper.isActive()) {
			flowerBotania = new AlleleFlower(FLOWER_BOTANIA, new FlowerProviderBotania(), true);
			effectDreaming = new AlleleEffectTransmuting(EFFECT_DREAMING, false,
					new TransmutationEffectController(new TransmutationEffectLBotaniaLiving()), 100);
		}
		else {
			flowerBotania = new AlleleFlower(FLOWER_BOTANIA, vanillaFlowers, true);
			effectDreaming = new AlleleEffectPlaceholder(EFFECT_DREAMING, false);
		}

		if (ArsMagicaHelper.isActive()) {
			flowerArsMagica = new AlleleFlower(FLOWER_ARS_MAGICA_PLANT, new FlowerProviderArsMagicaFlower(), false);
			spawnManaDrainer = new AlleleEffectSpawnMobWeighted(EFFECT_MANA_DRAIN, true, 20,
					new String[] { ArsMagicaHelper.Name + ".MobManaCreeper", ArsMagicaHelper.Name + ".ManaVortex" },
					new int[] { 60, 2 }
			);

			spawnWispOrHecate = new AlleleEffectSpawnMobWeighted(EFFECT_AM_WISP, true, 20,
					new String[] { ArsMagicaHelper.Name + ".MobWisp", ArsMagicaHelper.Name + ".MobHecate" },
					new int[] { 40, 3 }
			);
		}
		else {
			flowerArsMagica =  new AlleleFlower(FLOWER_ARS_MAGICA_PLANT, vanillaFlowers, false);
			spawnManaDrainer = new AlleleEffectPlaceholder(EFFECT_MANA_DRAIN, true);
			spawnWispOrHecate = new AlleleEffectPlaceholder(EFFECT_AM_WISP, true);
		}

		//TODO TE spawning
		if (ThermalModsHelper.isActive()) {
			spawnBlizz = new AlleleEffectSpawnMob(EFFECT_TE_BLIZZY, true, ThermalModsHelper.Entity.BLIZZ.entityID, ThermalModsHelper.Entity.BLIZZ.soundName)
			.setThrottle(100)
			.setChanceToSpawn(80);
			spawnBlitz = new AlleleEffectSpawnMob(EFFECT_TE_BLITZ, true, ThermalModsHelper.Entity.BLITZ.entityID, ThermalModsHelper.Entity.BLITZ.soundName)
			.setThrottle(100)
			.setChanceToSpawn(80);
			spawnBasalz = new AlleleEffectSpawnMob(EFFECT_TE_BASALZ, true, ThermalModsHelper.Entity.BASALZ.entityID, ThermalModsHelper.Entity.BASALZ.soundName)
			.setThrottle(100)
			.setChanceToSpawn(80);
		}
		else {
			spawnBlizz = new AlleleEffectPlaceholder(EFFECT_TE_BLIZZY, true);
			spawnBlitz = new AlleleEffectPlaceholder(EFFECT_TE_BLITZ, true);
			spawnBasalz = new AlleleEffectPlaceholder(EFFECT_TE_BASALZ, true);
		}
	}

	public static void registerDeprecatedAlleleReplacements() {
		IAlleleRegistry registry = AlleleManager.alleleRegistry;
		
		registry.registerDeprecatedAlleleReplacement("magicbees.effectNodeAttract", effectNodeEmpower);
		registry.registerDeprecatedAlleleReplacement("magicbees.effectNodePurify", effectNodeConversionPure);
		registry.registerDeprecatedAlleleReplacement("magicbees.effectNodeFlux", effectNodeConversionTaint);
		registry.registerDeprecatedAlleleReplacement("magicbees.effectNodeCharge", effectNodeEmpower);
		registry.registerDeprecatedAlleleReplacement("magicbees.speciesTCAttractive", BeeSpecies.TC_EMPOWERING.getSpecies());
		registry.registerDeprecatedAlleleReplacement("magicbees.speciesBMBloodyBMBloody", BeeSpecies.EARTHY.getSpecies());
		registry.registerDeprecatedAlleleReplacement("magicbees.speciesBMBloodyBMBound", BeeSpecies.ARCANE.getSpecies());
	}

	public static IAlleleBeeSpecies getBaseSpecies(String name) {
		return (IAlleleBeeSpecies) AlleleManager.alleleRegistry.getAllele((new StringBuilder()).append("forestry.species").append(name).toString());
	}

	public static IAlleleBeeSpecies getExtraSpecies(String name) {
		return (IAlleleBeeSpecies) AlleleManager.alleleRegistry.getAllele((new StringBuilder()).append("extrabees.species.").append(name.toLowerCase())
				.toString());
	}

	public static IAllele getBaseAllele(String name) {
		return AlleleManager.alleleRegistry.getAllele("forestry." + name);
	}

	public static IAllele getAllele(String name) {
		return AlleleManager.alleleRegistry.getAllele(name);
	}

	private String uid;
	private boolean dominant;

	public Allele(String id, boolean isDominant, IChromosomeType... chromosomeTypes) {
		this.uid = "magicbees." + id;
		this.dominant = isDominant;
		AlleleManager.alleleRegistry.registerAllele(this, chromosomeTypes);
	}

	@Override
	public String getUID() {
		return this.uid;
	}

	@Override
	public boolean isDominant() {
		return this.dominant;
	}

	@Override
	public String getName() {
		return LocalizationManager.getLocalizedString(getUID());
	}

	@Override
	public String getUnlocalizedName() {
		return this.uid;
	}

}