BloodMagic/src/main/java/WayofTime/bloodmagic/compat/waila/provider/DataProviderRitualController.java

package WayofTime.bloodmagic.compat.waila.provider;

import WayofTime.bloodmagic.util.Constants;
import WayofTime.bloodmagic.ritual.imperfect.ImperfectRitualRegistry;
import WayofTime.bloodmagic.ritual.RitualRegistry;
import WayofTime.bloodmagic.ritual.imperfect.ImperfectRitual;
import WayofTime.bloodmagic.util.helper.PlayerHelper;
import WayofTime.bloodmagic.tile.TileMasterRitualStone;
import WayofTime.bloodmagic.util.helper.TextHelper;
import mcp.mobius.waila.api.IWailaConfigHandler;
import mcp.mobius.waila.api.IWailaDataAccessor;
import mcp.mobius.waila.api.IWailaDataProvider;
import net.minecraft.entity.player.EntityPlayerMP;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.util.math.BlockPos;
import net.minecraft.world.World;

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

public class DataProviderRitualController implements IWailaDataProvider {

    public static final IWailaDataProvider INSTANCE = new DataProviderRitualController();

    @Nonnull
    @Override
    public List<String> getWailaBody(ItemStack itemStack, List<String> currenttip, IWailaDataAccessor accessor, IWailaConfigHandler config) {
        if (!config.getConfig(Constants.Compat.WAILA_CONFIG_RITUAL))
            return currenttip;

        NBTTagCompound tag = accessor.getNBTData();
        if (tag.getBoolean("master")) {
            if (tag.hasKey("ritual"))
                currenttip.add(TextHelper.localizeEffect(tag.getString("ritual")));
            if (tag.hasKey("owner"))
                currenttip.add(TextHelper.localizeEffect("tooltip.bloodmagic.currentOwner", tag.getString("owner")));
            if (!tag.getBoolean("active"))
                currenttip.add(TextHelper.localizeEffect("tooltip.bloodmagic.deactivated"));
            if (!tag.getBoolean("enabled"))
                currenttip.add(TextHelper.localizeEffect("tooltip.bloodmagic.config.disabled"));
        } else {
            if (tag.hasKey("ritual")) {
                currenttip.add(TextHelper.localizeEffect(tag.getString("ritual")));
                if (!tag.getBoolean("enabled"))
                    currenttip.add(TextHelper.localizeEffect("tooltip.bloodmagic.config.disabled"));
            }
        }

        return currenttip;
    }

    @Nonnull
    @Override
    public NBTTagCompound getNBTData(EntityPlayerMP player, TileEntity te, NBTTagCompound tag, World world, BlockPos pos) {

        if (te instanceof TileMasterRitualStone) {
            TileMasterRitualStone mrs = (TileMasterRitualStone) te;
            tag.setBoolean("master", true);
            if (mrs.getCurrentRitual() != null) {
                tag.setString("ritual", mrs.getCurrentRitual().getUnlocalizedName());
                tag.setBoolean("active", mrs.isActive());
                if (mrs.getOwner() != null)
                    tag.setString("owner", PlayerHelper.getUsernameFromUUID(mrs.getOwner()));
                tag.setBoolean("enabled", RitualRegistry.ritualEnabled(mrs.getCurrentRitual()));
            }
        } else {
            tag.setBoolean("master", false);

            ImperfectRitual ritual = ImperfectRitualRegistry.getRitualForBlock(world.getBlockState(pos.up()));
            if (ritual != null) {
                tag.setString("ritual", ritual.getUnlocalizedName());
                tag.setBoolean("enabled", ImperfectRitualRegistry.ritualEnabled(ritual));
            }
        }

        return tag;
    }
}