Add IFluidStack to IData expansion

CraftTweaker · Jan 31, 2023 · d5cb865 · d5cb865
1 parent 4661605
commit d5cb865
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/Forge/src/main/java/com/blamejared/crafttweaker/api/fluid/IFluidStack.java b/Forge/src/main/java/com/blamejared/crafttweaker/api/fluid/IFluidStack.java
@@ -4,6 +4,10 @@
 import com.blamejared.crafttweaker.api.bracket.CommandStringDisplayable;
 import com.blamejared.crafttweaker.api.data.MapData;
 import com.blamejared.crafttweaker.platform.Services;
+import com.blamejared.crafttweaker.api.data.IData;
+import com.blamejared.crafttweaker.api.data.IntData;
+import com.blamejared.crafttweaker.api.data.MapData;
+import com.blamejared.crafttweaker.natives.resource.ExpandResourceLocation;
 import com.blamejared.crafttweaker_annotations.annotations.Document;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.world.level.material.Fluid;
@@ -177,4 +181,13 @@ default CTFluidIngredient asList(CTFluidIngredient other) {
         return new CTFluidIngredient.CompoundFluidIngredient(elements);
     }
 
+    @ZenCodeType.Method
+    @ZenCodeType.Caster(implicit = true)
+    default IData asIData() {
+        MapData data = new MapData();
+        data.put("fluid", ExpandResourceLocation.asData(this.getRegistryName()));
+        data.put("amount", new IntData(this.getAmount()));
+        if (this.getTag() != null) data.put("nbt", this.getTag());
+        return data;
+    }
 }