mcMMO/src/main/java/com/gmail/nossr50/util/TransientMetadataTools.java

package com.gmail.nossr50.util;

import com.gmail.nossr50.mcMMO;
import org.bukkit.entity.LivingEntity;

public class TransientMetadataTools {
    public static final String OLD_NAME_METAKEY = TransientMetadataTools.OLD_NAME_METAKEY;
    private final mcMMO pluginRef;

    public TransientMetadataTools(mcMMO pluginRef) {
        this.pluginRef = pluginRef;
    }

    public void cleanAllMobMetadata(LivingEntity livingEntity) {
        //Since its not written anywhere, apparently the GC won't touch objects with metadata still present on them
        if (livingEntity.hasMetadata(mcMMO.customNameKey)) {
            livingEntity.setCustomName(livingEntity.getMetadata(mcMMO.customNameKey).get(0).asString());
            livingEntity.removeMetadata(mcMMO.customNameKey, pluginRef);
        }

        if(livingEntity.hasMetadata(OLD_NAME_METAKEY)) {
            livingEntity.removeMetadata(OLD_NAME_METAKEY, pluginRef);
        }

        //Involved in changing mob names to hearts
        if (livingEntity.hasMetadata(mcMMO.customVisibleKey)) {
            livingEntity.setCustomNameVisible(livingEntity.getMetadata(mcMMO.customVisibleKey).get(0).asBoolean());
            livingEntity.removeMetadata(mcMMO.customVisibleKey, pluginRef);
        }

        //Gets assigned to endermen, potentially doesn't get cleared before this point
        if(livingEntity.hasMetadata(mcMMO.travelingBlock)) {
            livingEntity.removeMetadata(mcMMO.travelingBlock, pluginRef);
        }

        if(livingEntity.hasMetadata(mcMMO.REPLANT_META_KEY)) {
            livingEntity.removeMetadata(mcMMO.REPLANT_META_KEY, pluginRef);
        }


        //Cleanup mob metadata
        mcMMO.getCompatibilityManager().getPersistentDataLayer().removeMobFlags(livingEntity);
    }
}