Iris/src/main/java/com/volmit/iris/object/IrisEntity.java

package com.volmit.iris.object;

import com.volmit.iris.Iris;
import com.volmit.iris.scaffold.engine.Engine;
import com.volmit.iris.util.*;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.NoArgsConstructor;
import lombok.experimental.Accessors;
import org.bukkit.Location;
import org.bukkit.NamespacedKey;
import org.bukkit.attribute.Attributable;
import org.bukkit.entity.*;
import org.bukkit.entity.Panda.Gene;
import org.bukkit.inventory.Inventory;
import org.bukkit.inventory.ItemStack;
import org.bukkit.loot.LootContext;
import org.bukkit.loot.LootTable;
import org.bukkit.loot.Lootable;

import java.util.Collection;
import java.util.Random;

@Accessors(chain = true)
@NoArgsConstructor
@AllArgsConstructor
@DontObfuscate
@Desc("Represents an iris entity.")
@Data
@EqualsAndHashCode(callSuper = false)
public class IrisEntity extends IrisRegistrant
{
	@Required
	@DontObfuscate
	@Desc("The type of entity to spawn. To spawn a mythic mob, set this type to unknown and define mythic type.")
	private EntityType type = EntityType.UNKNOWN;

	@RegistryListMythical
	@Desc("The type of mythic mob (if mythic mobs is installed). If this is set, make sure to set 'type' to UNKNOWN")
	@DontObfuscate
	private String mythicalType = "";

	@DontObfuscate
	@Desc("The custom name of this entity")
	private String customName = "";

	@DontObfuscate
	@Desc("Should the name on this entity be visible even if you arent looking at it.")
	private boolean customNameVisible = false;

	@DontObfuscate
	@Desc("If this entity type is a mob, should it be aware of it's surroundings & interact with the world.")
	private boolean aware = true;

	@DontObfuscate
	@Desc("If this entity type is a creature, should it have ai goals.")
	private boolean ai = true;

	@DontObfuscate
	@Desc("Should this entity be glowing")
	private boolean glowing = false;

	@DontObfuscate
	@Desc("Should gravity apply to this entity")
	private boolean gravity = true;

	@DontObfuscate
	@Desc("When an entity is invulnerable it can only be damaged by players increative mode.")
	private boolean invulnerable = false;

	@DontObfuscate
	@Desc("When an entity is silent it will not produce any sound.")
	private boolean silent = false;

	@DontObfuscate
	@Desc("Should this entity be allowed to pickup items")
	private boolean pickupItems = false;

	@DontObfuscate
	@Desc("Should this entity be removed when far away")
	private boolean removable = true;

	@DontObfuscate
	@Desc("Entity helmet equipment")
	private IrisLoot helmet = null;

	@DontObfuscate
	@Desc("Entity chestplate equipment")
	private IrisLoot chestplate = null;

	@DontObfuscate
	@Desc("Entity boots equipment")
	private IrisLoot boots = null;

	@DontObfuscate
	@Desc("Entity leggings equipment")
	private IrisLoot leggings = null;

	@DontObfuscate
	@Desc("Entity main hand equipment")
	private IrisLoot mainHand = null;

	@DontObfuscate
	@Desc("Entity off hand equipment")
	private IrisLoot offHand = null;

	@DontObfuscate
	@Desc("Make other entities ride this entity")
	@ArrayType(min = 1, type = IrisEntity.class)
	private KList<IrisEntity> passengers = new KList<>();

	@DontObfuscate
	@Desc("Attribute modifiers for this entity")
	@ArrayType(min = 1, type = IrisAttributeModifier.class)
	private KList<IrisAttributeModifier> attributes = new KList<>();

	@DontObfuscate
	@Desc("Loot tables for drops")
	private IrisLootReference loot = new IrisLootReference();

	@DontObfuscate
	@Desc("If specified, this entity will be leashed by this entity. I.e. THIS ENTITY Leashed by SPECIFIED. This has no effect on EnderDragons, Withers, Players, or Bats.Non-living entities excluding leashes will not persist as leashholders.")
	private IrisEntity leashHolder = null;

	@DontObfuscate
	@Desc("The main gene for a panda if the entity type is a panda")
	private Gene pandaMainGene = Gene.NORMAL;

	@DontObfuscate
	@Desc("The hidden gene for a panda if the entity type is a panda")
	private Gene pandaHiddenGene = Gene.NORMAL;

	@DontObfuscate
	@Desc("The this entity is ageable, set it's baby status")
	private boolean baby = false;

	public Entity spawn(Engine gen, Location at)
	{
		return spawn(gen, at, new RNG(at.hashCode()));
	}

	public Entity spawn(Engine gen, Location at, RNG rng)
	{
		Entity e = doSpawn(at);
		e.setCustomName(getCustomName() != null ? C.translateAlternateColorCodes('&', getCustomName()) : null);
		e.setCustomNameVisible(isCustomNameVisible());
		e.setGlowing(isGlowing());
		e.setGravity(isGravity());
		e.setInvulnerable(isInvulnerable());
		e.setSilent(isSilent());

		int gg = 0;
		for(IrisEntity i : passengers)
		{
			e.addPassenger(i.spawn(gen, at, rng.nextParallelRNG(234858 + gg++)));
		}

		if(e instanceof Attributable)
		{
			Attributable a = (Attributable) e;

			for(IrisAttributeModifier i : getAttributes())
			{
				i.apply(rng, a);
			}
		}

		if(e instanceof Lootable)
		{
			Lootable l = (Lootable) e;

			if(getLoot().getTables().isNotEmpty())
			{
				l.setLootTable(new LootTable()
				{
					@Override
					public NamespacedKey getKey()
					{
						return new NamespacedKey(Iris.instance, "loot-" + IrisEntity.this.hashCode());
					}

					@Override
					public Collection<ItemStack> populateLoot(Random random, LootContext context)
					{
						KList<ItemStack> items = new KList<>();

						for(String fi : getLoot().getTables())
						{
							IrisLootTable i = gen.getData().getLootLoader().load(fi);
							items.addAll(i.getLoot(gen.isStudio(), false, rng.nextParallelRNG(345911), InventorySlotType.STORAGE, at.getBlockX(), at.getBlockY(), at.getBlockZ(), 8, 4));
						}

						return items;
					}

					@Override
					public void fillInventory(Inventory inventory, Random random, LootContext context)
					{
						for(ItemStack i : populateLoot(random, context))
						{
							inventory.addItem(i);
						}

						gen.getCompound().getEngine(at.getBlockY()).scramble(inventory, rng);
					}
				});
			}
		}

		if(e instanceof LivingEntity)
		{
			LivingEntity l = (LivingEntity) e;
			l.setAI(isAi());
			l.setCanPickupItems(isPickupItems());

			if(getLeashHolder() != null)
			{
				l.setLeashHolder(getLeashHolder().spawn(gen, at, rng.nextParallelRNG(234548)));
			}

			l.setRemoveWhenFarAway(isRemovable());

			if(getHelmet() != null && rng.i(1, getHelmet().getRarity()) == 1)
			{
				l.getEquipment().setHelmet(getHelmet().get(gen.isStudio(), rng));
			}

			if(getChestplate() != null && rng.i(1, getChestplate().getRarity()) == 1)
			{
				l.getEquipment().setChestplate(getChestplate().get(gen.isStudio(), rng));
			}

			if(getLeggings() != null && rng.i(1, getLeggings().getRarity()) == 1)
			{
				l.getEquipment().setLeggings(getLeggings().get(gen.isStudio(), rng));
			}

			if(getBoots() != null && rng.i(1, getBoots().getRarity()) == 1)
			{
				l.getEquipment().setBoots(getBoots().get(gen.isStudio(), rng));
			}

			if(getMainHand() != null && rng.i(1, getMainHand().getRarity()) == 1)
			{
				l.getEquipment().setItemInMainHand(getMainHand().get(gen.isStudio(), rng));
			}

			if(getOffHand() != null && rng.i(1, getOffHand().getRarity()) == 1)
			{
				l.getEquipment().setItemInOffHand(getOffHand().get(gen.isStudio(), rng));
			}
		}

		if(e instanceof Ageable && isBaby())
		{
			((Ageable) e).setBaby();
		}

		if(e instanceof Panda)
		{
			((Panda) e).setMainGene(getPandaMainGene());
			((Panda) e).setMainGene(getPandaHiddenGene());
		}

		if(Iris.awareEntities && e instanceof Mob)
		{
			Mob m = (Mob) e;
			m.setAware(isAware());
		}

		return e;
	}

	private Entity doSpawn(Location at)
	{
		if(isMythical())
		{
			return Iris.linkMythicMobs.spawn(getMythicalType(), at);
		}

		if(isCitizens())
		{
			// TODO: return Iris.linkCitizens.spawn(getType(), at); SPAWN SOME TYPE TOO
		}

		return at.getWorld().spawnEntity(at, getType());
	}

	public boolean isMythical()
	{
		return Iris.linkMythicMobs.supported() && !getMythicalType().trim().isEmpty();
	}

	public boolean isCitizens()
	{
		return false;

		// TODO: return Iris.linkCitizens.supported() && someType is not empty;
	}
}