[Bleeding] Implemented EnchantmentTable API.

1 files changed, 226 insertions, 0 deletions

diff --git a/src/main/java/net/minecraft/server/ContainerEnchantTable.java b/src/main/java/net/minecraft/server/ContainerEnchantTable.java
new file mode 100644
index 00000000..cce9a7f6
--- /dev/null
+++ b/src/main/java/net/minecraft/server/ContainerEnchantTable.java
@@ -0,0 +1,226 @@
+package net.minecraft.server;
+
+import java.util.Iterator;
+import java.util.List;
+import java.util.Random;
+
+// CraftBukkit start
+import java.util.Map;
+import java.util.HashMap;
+
+import org.bukkit.craftbukkit.inventory.CraftItemStack;
+import org.bukkit.event.enchantment.EnchantItemEvent;
+import org.bukkit.event.enchantment.PrepareItemEnchantEvent;
+import org.bukkit.entity.Player;
+// CraftBukkit end
+
+public class ContainerEnchantTable extends Container {
+
+    public IInventory a = new ContainerEnchantTableInventory(this, "Enchant", 1);
+    private World h;
+    private int i;
+    private int j;
+    private int k;
+    private Random l = new Random();
+    public long b;
+    public int[] c = new int[3];
+    private Player player;  // CraftBukkit
+
+    public ContainerEnchantTable(PlayerInventory playerinventory, World world, int i, int j, int k) {
+        this.h = world;
+        this.i = i;
+        this.j = j;
+        this.k = k;
+        this.a((Slot) (new SlotEnchant(this, this.a, 0, 25, 47)));
+
+        int l;
+
+        for (l = 0; l < 3; ++l) {
+            for (int i1 = 0; i1 < 9; ++i1) {
+                this.a(new Slot(playerinventory, i1 + l * 9 + 9, 8 + i1 * 18, 84 + l * 18));
+            }
+        }
+
+        for (l = 0; l < 9; ++l) {
+            this.a(new Slot(playerinventory, l, 8 + l * 18, 142));
+        }
+        player = (Player) playerinventory.d.bukkitEntity;    // CraftBukkit
+    }
+
+    public void a(ICrafting icrafting) {
+        super.a(icrafting);
+        icrafting.a(this, 0, this.c[0]);
+        icrafting.a(this, 1, this.c[1]);
+        icrafting.a(this, 2, this.c[2]);
+    }
+
+    public void a() {
+        super.a();
+
+        for (int i = 0; i < this.listeners.size(); ++i) {
+            ICrafting icrafting = (ICrafting) this.listeners.get(i);
+
+            icrafting.a(this, 0, this.c[0]);
+            icrafting.a(this, 1, this.c[1]);
+            icrafting.a(this, 2, this.c[2]);
+        }
+    }
+
+    public void a(IInventory iinventory) {
+        if (iinventory == this.a) {
+            ItemStack itemstack = iinventory.getItem(0);
+            int i;
+
+            if (itemstack != null && itemstack.q()) {
+                this.b = this.l.nextLong();
+                if (!this.h.isStatic) {
+                    i = 0;
+
+                    int j;
+
+                    for (j = -1; j <= 1; ++j) {
+                        for (int k = -1; k <= 1; ++k) {
+                            if ((j != 0 || k != 0) && this.h.isEmpty(this.i + k, this.j, this.k + j) && this.h.isEmpty(this.i + k, this.j + 1, this.k + j)) {
+                                if (this.h.getTypeId(this.i + k * 2, this.j, this.k + j * 2) == Block.BOOKSHELF.id) {
+                                    ++i;
+                                }
+
+                                if (this.h.getTypeId(this.i + k * 2, this.j + 1, this.k + j * 2) == Block.BOOKSHELF.id) {
+                                    ++i;
+                                }
+
+                                if (k != 0 && j != 0) {
+                                    if (this.h.getTypeId(this.i + k * 2, this.j, this.k + j) == Block.BOOKSHELF.id) {
+                                        ++i;
+                                    }
+
+                                    if (this.h.getTypeId(this.i + k * 2, this.j + 1, this.k + j) == Block.BOOKSHELF.id) {
+                                        ++i;
+                                    }
+
+                                    if (this.h.getTypeId(this.i + k, this.j, this.k + j * 2) == Block.BOOKSHELF.id) {
+                                        ++i;
+                                    }
+
+                                    if (this.h.getTypeId(this.i + k, this.j + 1, this.k + j * 2) == Block.BOOKSHELF.id) {
+                                        ++i;
+                                    }
+                                }
+                            }
+                        }
+                    }
+
+                    for (j = 0; j < 3; ++j) {
+                        this.c[j] = EnchantmentManager.a(this.l, j, i, itemstack);
+                    }
+
+                    // CraftBukkit start
+                    CraftItemStack item = new CraftItemStack(itemstack);
+                    PrepareItemEnchantEvent event = new PrepareItemEnchantEvent(player, h.getWorld().getBlockAt(this.i, this.j, this.k), item, this.c, i);
+                    h.getServer().getPluginManager().callEvent(event);
+                    if (event.isCancelled()) {
+                        for (i = 0; i < 3; ++i) {
+                            this.c[i] = 0;
+                        }
+                        return;
+                    }
+                    // CraftBukkit end
+                    this.a();
+                }
+            } else {
+                for (i = 0; i < 3; ++i) {
+                    this.c[i] = 0;
+                }
+            }
+        }
+    }
+
+    public boolean a(EntityHuman entityhuman, int i) {
+        ItemStack itemstack = this.a.getItem(0);
+
+        if (this.c[i] > 0 && itemstack != null && entityhuman.expLevel >= this.c[i]) {
+            if (!this.h.isStatic) {
+                List list = EnchantmentManager.a(this.l, itemstack, this.c[i]);
+
+                // CraftBukkit start
+                Map<org.bukkit.enchantments.Enchantment, Integer> enchants = new HashMap<org.bukkit.enchantments.Enchantment, Integer>();
+                for (Object o : list) {
+                    EnchantmentInstance e = (EnchantmentInstance) o;
+                    enchants.put(org.bukkit.enchantments.Enchantment.getById(e.a.id), e.b);
+                }
+                CraftItemStack item = new CraftItemStack(itemstack);
+                EnchantItemEvent event = new EnchantItemEvent((Player) entityhuman.bukkitEntity, h.getWorld().getBlockAt(this.i, this.j, this.k), item, this.c[i], enchants);
+                h.getServer().getPluginManager().callEvent(event);
+
+                int level = event.getExpLevelCost();
+                if (event.isCancelled() || (level > entityhuman.expLevel)) {
+                    return false;
+                }
+                if (enchants.isEmpty() == false) {
+                    entityhuman.levelDown(level);
+                    for (Map.Entry<org.bukkit.enchantments.Enchantment, Integer> ent : event.getEnchantsToAdd().entrySet()) {
+                        try {
+                            item.addEnchantment(ent.getKey(), ent.getValue());
+                        } catch (IllegalArgumentException iax) {
+                            /* Just swallow invalid enchantments */
+                        }
+                    }
+                    // CraftBukkit end
+
+                    this.a(this.a);
+                }
+            }
+
+            return true;
+        } else {
+            return false;
+        }
+    }
+
+    public void a(EntityHuman entityhuman) {
+        super.a(entityhuman);
+        if (!this.h.isStatic) {
+            ItemStack itemstack = this.a.getItem(0);
+
+            if (itemstack != null) {
+                entityhuman.b(itemstack);
+            }
+        }
+    }
+
+    public boolean b(EntityHuman entityhuman) {
+        return this.h.getTypeId(this.i, this.j, this.k) != Block.ENCHANTMENT_TABLE.id ? false : entityhuman.e((double) this.i + 0.5D, (double) this.j + 0.5D, (double) this.k + 0.5D) <= 64.0D;
+    }
+
+    public ItemStack a(int i) {
+        ItemStack itemstack = null;
+        Slot slot = (Slot) this.e.get(i);
+
+        if (slot != null && slot.c()) {
+            ItemStack itemstack1 = slot.getItem();
+
+            itemstack = itemstack1.cloneItemStack();
+            if (i != 0) {
+                return null;
+            }
+
+            if (!this.a(itemstack1, 1, 37, true)) {
+                return null;
+            }
+
+            if (itemstack1.count == 0) {
+                slot.c((ItemStack) null);
+            } else {
+                slot.d();
+            }
+
+            if (itemstack1.count == itemstack.count) {
+                return null;
+            }
+
+            slot.b(itemstack1);
+        }
+
+        return itemstack;
+    }
+}