AnimalContestEntry.java

package joshie.harvest.quests.town.festivals.contest.animal;

import joshie.harvest.api.HFApi;
import joshie.harvest.api.animals.AnimalStats;
import joshie.harvest.api.animals.AnimalTest;
import joshie.harvest.api.npc.NPC;
import joshie.harvest.core.helpers.EntityHelper;
import joshie.harvest.core.helpers.SpawnItemHelper;
import joshie.harvest.core.helpers.TextHelper;
import joshie.harvest.npcs.entity.EntityNPC;
import joshie.harvest.quests.base.QuestAnimalContest;
import joshie.harvest.quests.town.festivals.Place;
import joshie.harvest.quests.town.festivals.contest.ContestEntry;
import net.minecraft.entity.passive.EntityAnimal;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.util.EnumHand;
import net.minecraft.util.ResourceLocation;
import net.minecraft.util.math.BlockPos;
import net.minecraft.world.World;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.List;
import java.util.UUID;

@SuppressWarnings("WeakerAccess")
public class AnimalContestEntry extends ContestEntry<QuestAnimalContest> {
    private final UUID entity;

    public AnimalContestEntry(UUID player, UUID entity, int stall) {
        super(player, stall);
        this.entity = entity;
    }

    public AnimalContestEntry(NPC npc, UUID entity, int stall) {
        super(npc, stall);
        this.entity = entity;
    }

    @Nullable
    private AnimalStats getStats(World world) {
        EntityAnimal animal  = EntityHelper.getAnimalFromUUID(world, entity);
        if (animal != null) {
            return EntityHelper.getStats(animal);
        } else return null;
    }

    @SuppressWarnings("ConstantConditions")
    public int getScore(QuestAnimalContest quest, World world) {
        EntityAnimal animal  = EntityHelper.getAnimalFromUUID(world, entity);
        int score = 0;
        if (animal != null) {
            AnimalStats stats = EntityHelper.getStats(animal);
            score += stats.getHappiness(); //Base level
            //Add bonuses, if the animal has had everything done today
            if (stats.performTest(AnimalTest.CAN_CLEAN)) {
                if (stats.performTest(AnimalTest.HAS_EATEN)) score += 1000;
                if (stats.performTest(AnimalTest.HAD_TREAT)) score += 3000;
                if (stats.performTest(AnimalTest.IS_CLEAN)) score += 3000;
                if (stats.performTest(AnimalTest.BEEN_LOVED)) score += 2000;
            } else {
                if (stats.performTest(AnimalTest.HAS_EATEN)) score += 1000;
                if (stats.performTest(AnimalTest.HAD_TREAT)) score += 5000;
                if (stats.performTest(AnimalTest.BEEN_LOVED)) score += 3000;
            }

            //Reduce the score if the animal is sick
            if (stats.performTest(AnimalTest.IS_SICK)) {
                score -= 25000;
            }
        }

        return score;
    }

    @Override
    public String getTextFromScore(String unlocalised, int score) {
        return TextHelper.localize(unlocalised + "." + Math.max(0, Math.min(9, (int)Math.floor(((double)score) / 3000))));
    }

    @Nullable
    private EntityAnimal getAnimalEntity(World world) {
        return EntityHelper.getAnimalFromUUID(world, entity);
    }

    @Override
    public boolean isInvalid(World world) {
        return getAnimalEntity(world) == null;
    }

    @Nonnull
    UUID getAnimalUUID() {
        return entity;
    }

    @Override
    public String getName(World world) {
        EntityAnimal animal = EntityHelper.getAnimalFromUUID(world, entity);
        return animal == null ? "" : animal.getName();
    }

    public void reward(World world, Place place, NPC[] npcs, ItemStack reward) {
        AnimalStats stats = getStats(world);
        EntityPlayer player = getPlayer(world);
        if (stats != null && player != null) { //Give the rewards for this
            SpawnItemHelper.addToPlayerInventory(player, reward);
            stats.affectHappiness(place.happiness); //Make the animal happier, and the npcs that took part v
            for (NPC npc: npcs) {
                HFApi.player.getRelationsForPlayer(player).affectRelationship(npc, place.happiness);
            }
        } else if (npc != null) {
            EntityAnimal animal = getAnimalEntity(world);
            if (animal != null) {
                List<EntityNPC> npcList = EntityHelper.getEntities(EntityNPC.class, world, new BlockPos(animal), 64D, 64D);
                for (EntityNPC aNPC: npcList) {
                    if (aNPC.getNPC() == npc) {
                        aNPC.setHeldItem(EnumHand.OFF_HAND, reward);
                        break;
                    }
                }
            }
        }
    }

    @Nullable
    public static AnimalContestEntry fromNBT(NBTTagCompound tag) {
        UUID animal = UUID.fromString(tag.getString("Animal"));
        Integer stall = tag.getInteger("Stall");
        if (tag.hasKey("Player")) {
            UUID player = UUID.fromString(tag.getString("Player"));
            return new AnimalContestEntry(player, animal, stall);
        } else if (tag.hasKey("NPC")) {
            NPC npc = NPC.REGISTRY.get(new ResourceLocation(tag.getString("NPC")));
            return new AnimalContestEntry(npc, animal, stall);
        } else return null;
    }

    @Override
    public NBTTagCompound toNBT() {
        NBTTagCompound tag = super.toNBT();
        tag.setString("Animal", entity.toString());
        return tag;
    }
}