Fixed schematic importing for sponge

src/main/java/com/plotsquared/object/schematic/StateWrapper.java

@@ -0,0 +1,122 @@
+package com.plotsquared.object.schematic;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import org.bukkit.block.BlockState;
+import org.bukkit.enchantments.Enchantment;
+import org.bukkit.inventory.InventoryHolder;
+import org.bukkit.inventory.ItemStack;
+
+import com.intellectualcrafters.jnbt.ByteTag;
+import com.intellectualcrafters.jnbt.CompoundTag;
+import com.intellectualcrafters.jnbt.ListTag;
+import com.intellectualcrafters.jnbt.ShortTag;
+import com.intellectualcrafters.jnbt.Tag;
+import com.intellectualcrafters.plot.object.schematic.ItemType;
+import com.intellectualcrafters.plot.object.schematic.PlotItem;
+import com.intellectualcrafters.plot.util.MathMan;
+import com.intellectualcrafters.plot.util.SchematicHandler.Schematic;
+
+public class StateWrapper {
+    
+    public BlockState state = null;
+    public CompoundTag tag = null;
+    
+    public StateWrapper(BlockState state) {
+        this.state = state;
+    }
+    
+    public StateWrapper(CompoundTag tag) {
+        this.tag = tag;
+    }
+    
+    public boolean restoreTag(short x, short y, short z, Schematic schematic) {
+        if (this.tag == null) {
+            return false;
+        }
+        List<Tag> itemsTag = this.tag.getListTag("Items").getValue();
+        int length = itemsTag.size();
+        short[] ids = new short[length];
+        byte[] datas = new byte[length];
+        byte[] amounts = new byte[length];
+        for (int i = 0; i < length; i++) {
+            Tag itemTag = itemsTag.get(i);
+            CompoundTag itemComp = (CompoundTag) itemTag;
+            short id = itemComp.getShort("id");
+            String idStr = itemComp.getString("id");
+            if (idStr != null && !MathMan.isInteger(idStr)) {
+                idStr = idStr.split(":")[1].toLowerCase();
+                id = (short) ItemType.getId(idStr);
+            }
+            ids[i] = id;
+            datas[i] = (byte) itemComp.getShort("Damage");
+            amounts[i] = itemComp.getByte("Count");
+        }
+        if (length != 0) {
+            schematic.addItem(new PlotItem(x, y, z, ids, datas, amounts));
+        }
+        return true;
+    }
+    
+    public CompoundTag getTag() {
+        if (this.tag != null) {
+            return this.tag;
+        }
+        if (state instanceof InventoryHolder) {
+            InventoryHolder inv = (InventoryHolder) state;
+            ItemStack[] contents = inv.getInventory().getContents();
+            Map<String, Tag> values = new HashMap<String, Tag>();
+            values.put("Items", new ListTag("Items", CompoundTag.class, serializeInventory(contents)));
+            return new CompoundTag(values);
+        }
+        return null;
+    }
+    
+    public String getId() {
+        return "Chest";
+    }
+    
+    public List<CompoundTag> serializeInventory(ItemStack[] items) {
+        List<CompoundTag> tags = new ArrayList<CompoundTag>();
+        for (int i = 0; i < items.length; ++i) {
+            if (items[i] != null) {
+                Map<String, Tag> tagData = serializeItem(items[i]);
+                tagData.put("Slot", new ByteTag("Slot", (byte) i));
+                tags.add(new CompoundTag(tagData));
+            }
+        }
+        return tags;
+    }
+    
+    public Map<String, Tag> serializeItem(org.spongepowered.api.item.inventory.ItemStack item) {
+        Map<String, Tag> data = new HashMap<String, Tag>();
+        
+        // FIXME serialize sponge item
+        
+        return data;
+    }
+    
+    public Map<String, Tag> serializeItem(ItemStack item) {
+        Map<String, Tag> data = new HashMap<String, Tag>();
+        data.put("id", new ShortTag("id", (short) item.getTypeId()));
+        data.put("Damage", new ShortTag("Damage", item.getDurability()));
+        data.put("Count", new ByteTag("Count", (byte) item.getAmount()));
+        if (!item.getEnchantments().isEmpty()) {
+            List<CompoundTag> enchantmentList = new ArrayList<CompoundTag>();
+            for(Entry<Enchantment, Integer> entry : item.getEnchantments().entrySet()) {
+                Map<String, Tag> enchantment = new HashMap<String, Tag>();
+                enchantment.put("id", new ShortTag("id", (short) entry.getKey().getId()));
+                enchantment.put("lvl", new ShortTag("lvl", entry.getValue().shortValue()));
+                enchantmentList.add(new CompoundTag(enchantment));
+            }
+            Map<String, Tag> auxData = new HashMap<String, Tag>();
+            auxData.put("ench", new ListTag("ench", CompoundTag.class, enchantmentList));
+            data.put("tag", new CompoundTag("tag", auxData));
+        }
+        return data;
+    }
+}