Implemented Enchantment API

This commit is contained in:
Nathan Adams
2011-11-27 00:39:25 +00:00
parent ba608f55e4
commit eb7bd5f23b
7 changed files with 545 additions and 0 deletions

View File

@@ -1,5 +1,12 @@
package org.bukkit.craftbukkit.inventory;
import java.util.HashMap;
import java.util.Map;
import net.minecraft.server.EnchantmentManager;
import net.minecraft.server.NBTBase;
import net.minecraft.server.NBTTagCompound;
import net.minecraft.server.NBTTagList;
import org.bukkit.enchantments.Enchantment;
import org.bukkit.inventory.ItemStack;
import org.bukkit.Material;
import org.bukkit.configuration.serialization.DelegateDeserialization;
@@ -129,4 +136,70 @@ public class CraftItemStack extends ItemStack {
public int getMaxStackSize() {
return item.getItem().getMaxStackSize();
}
@Override
public void addUnsafeEnchantment(Enchantment ench, int level) {
Map<Enchantment, Integer> enchantments = getEnchantments();
enchantments.put(ench, level);
rebuildEnchantments(enchantments);
}
@Override
public boolean containsEnchantment(Enchantment ench) {
return getEnchantmentLevel(ench) > 0;
}
@Override
public int getEnchantmentLevel(Enchantment ench) {
return EnchantmentManager.b(ench.getId(), item);
}
@Override
public int removeEnchantment(Enchantment ench) {
Map<Enchantment, Integer> enchantments = getEnchantments();
Integer previous = enchantments.remove(ench);
rebuildEnchantments(enchantments);
return (previous == null) ? 0 : previous;
}
@Override
public Map<Enchantment, Integer> getEnchantments() {
Map<Enchantment, Integer> result = new HashMap<Enchantment, Integer>();
NBTTagList list = item.p();
if (list == null) {
return result;
}
for (int i = 0; i < list.d(); i++) {
short id = ((NBTTagCompound)list.a(i)).e("id");
short level = ((NBTTagCompound)list.a(i)).e("lvl");
result.put(Enchantment.getById(id), (int)level);
}
return result;
}
private void rebuildEnchantments(Map<Enchantment, Integer> enchantments) {
NBTTagCompound tag = item.tag;
NBTTagList list = new NBTTagList("ench");
if (tag == null) {
tag = item.tag = new NBTTagCompound();
}
for (Map.Entry<Enchantment, Integer> entry : enchantments.entrySet()) {
NBTTagCompound subtag = new NBTTagCompound();
subtag.a("id", (short)entry.getKey().getId());
subtag.a("lvl", (short)(int)entry.getValue());
list.a(subtag);
}
tag.a("ench", (NBTBase)list);
}
}