Hbm-s-Nuclear-Tech-GIT/src/main/java/com/hbm/util/EnchantmentUtil.java

package com.hbm.util;

import net.minecraft.enchantment.Enchantment;
import net.minecraft.item.ItemStack;

public class EnchantmentUtil {
	
	public static void addEnchantment(ItemStack stack, Enchantment enchantment, int level) {

		stack.addEnchantment(enchantment, level);
	}
	
	public static void removeEnchantment(ItemStack stack, Enchantment enchantment) {
		
		if(stack.getEnchantmentTagList() == null)
			return;
		
		int i = 0;
		for( ; i < stack.getEnchantmentTagList().tagCount(); i++) {
			if(stack.getEnchantmentTagList().getCompoundTagAt(i).getShort("id") == enchantment.effectId)
				break;
		}
		
		if(i < stack.getEnchantmentTagList().tagCount())
			stack.getEnchantmentTagList().removeTag(i);
		
		if(stack.getEnchantmentTagList().tagCount() == 0)
			stack.getTagCompound().removeTag("ench");
	}
}