package com.TheRPGAdventurer.ROTD.datafix;

import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.util.datafix.IFixableData;

public class DragonMountsItemDataFixes implements IFixableData {

    @Override
    public int getFixVersion() {
        return 2;
    }

    // is the fixTag proper (I think this is the cause tho) derived from PotionItem.class
    public NBTTagCompound fixTagCompound(NBTTagCompound compound) {
        if ("dragonmounts:end_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:nether_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:skeleton_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:wither_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:forest_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:fire_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:water_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:sunlight_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:enchant_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:terra_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        } else if ("dragonmounts:moonlight_dragon_amulet".equals(compound.getString("id"))) {
            compound.setString("id", "dragonmounts:dragon_amulet");
        }

        return compound;
    }
}