1 files changed, 42 insertions, 0 deletions
diff --git a/src/main/java/gregtech/common/net/MessageSetFlaskCapacity.java b/src/main/java/gregtech/common/net/MessageSetFlaskCapacity.java
new file mode 100644
index 0000000000..c20b4002c3
--- /dev/null
+++ b/src/main/java/gregtech/common/net/MessageSetFlaskCapacity.java
@@ -0,0 +1,42 @@
+package gregtech.common.net;
+
+import cpw.mods.fml.common.network.simpleimpl.IMessage;
+import cpw.mods.fml.common.network.simpleimpl.IMessageHandler;
+import cpw.mods.fml.common.network.simpleimpl.MessageContext;
+import gregtech.common.items.GT_VolumetricFlask;
+import io.netty.buffer.ByteBuf;
+import net.minecraft.item.Item;
+import net.minecraft.item.ItemStack;
+
+public final class MessageSetFlaskCapacity implements IMessage {
+    private int capacity;
+
+    public MessageSetFlaskCapacity() {
+    }
+
+    public MessageSetFlaskCapacity(int capacity) {
+        this.capacity = capacity;
+    }
+
+
+    public void fromBytes(ByteBuf buf) {
+        this.capacity = buf.readInt();
+    }
+
+    public void toBytes(ByteBuf buf) {
+        buf.writeInt(this.capacity);
+    }
+
+    public static final class Handler
+            implements IMessageHandler<MessageSetFlaskCapacity, IMessage> {
+        public IMessage onMessage(MessageSetFlaskCapacity message, MessageContext ctx) {
+            ItemStack stack = ctx.getServerHandler().playerEntity.getHeldItem();
+            if ((stack != null) && (stack.stackSize > 0)) {
+                Item item = stack.getItem();
+                if ((item instanceof GT_VolumetricFlask))
+                    ((GT_VolumetricFlask) item).setCapacity(stack, message.capacity);
+            }
+            return null;
+        }
+    }
+}
+\ No newline at end of file