Fix several issues with ender tank covers (#3377)

Co-authored-by: Martin Robertz <dream-master@gmx.net>
Co-authored-by: Ethryan <3237986+Ethryan@users.noreply.github.com>

1 files changed, 22 insertions, 0 deletions

diff --git a/src/main/java/tectech/mechanics/enderStorage/EnderFluidContainer.java b/src/main/java/tectech/mechanics/enderStorage/EnderFluidContainer.java
index f77a5a1b51..9995ec5da0 100644
--- a/src/main/java/tectech/mechanics/enderStorage/EnderFluidContainer.java
+++ b/src/main/java/tectech/mechanics/enderStorage/EnderFluidContainer.java
@@ -131,4 +131,26 @@ public class EnderFluidContainer implements IFluidHandler, Serializable {
         }
         if (fluidStack != null) fluidStack.tag = tag;
     }
+
+    public NBTTagCompound save() {
+        NBTTagCompound data = new NBTTagCompound();
+
+        if (fluidStack != null) {
+            fluidStack.writeToNBT(data);
+        }
+
+        return data;
+    }
+
+    public static EnderFluidContainer load(NBTTagCompound data) {
+        EnderFluidContainer container = new EnderFluidContainer();
+
+        if (!"".equals(data.getString("FluidName"))) {
+            container.fluidStack = FluidStack.loadFluidStackFromNBT(data);
+        } else {
+            container.fluidStack = null;
+        }
+
+        return container;
+    }
 }