From e8266229d46433b6bbbff1d2165bf5c695e455ec Mon Sep 17 00:00:00 2001
From: MuXiu1997 <muxiu1997@gmail.com>
Date: Tue, 26 Apr 2022 18:57:40 +0800
Subject: [PATCH] WorldFluidContainerAnalytics

---
 .../cil/oc/server/component/Transposer.scala  |  2 +-
 .../traits/WorldFluidContainerAnalytics.scala | 68 +++++++++++++++++++
 2 files changed, 69 insertions(+), 1 deletion(-)
 create mode 100644 src/main/scala/li/cil/oc/server/component/traits/WorldFluidContainerAnalytics.scala

diff --git a/src/main/scala/li/cil/oc/server/component/Transposer.scala b/src/main/scala/li/cil/oc/server/component/Transposer.scala
index 2340b4bbb..52037c398 100644
--- a/src/main/scala/li/cil/oc/server/component/Transposer.scala
+++ b/src/main/scala/li/cil/oc/server/component/Transposer.scala
@@ -22,7 +22,7 @@ import scala.language.existentials
 
 object Transposer {
 
-  abstract class Common extends prefab.ManagedEnvironment with traits.WorldInventoryAnalytics with traits.WorldTankAnalytics with traits.InventoryTransfer with DeviceInfo {
+  abstract class Common extends prefab.ManagedEnvironment with traits.WorldInventoryAnalytics with traits.WorldTankAnalytics with traits.WorldFluidContainerAnalytics with traits.InventoryTransfer with DeviceInfo {
     override val node = api.Network.newNode(this, Visibility.Network).
       withComponent("transposer").
       withConnector().
diff --git a/src/main/scala/li/cil/oc/server/component/traits/WorldFluidContainerAnalytics.scala b/src/main/scala/li/cil/oc/server/component/traits/WorldFluidContainerAnalytics.scala
new file mode 100644
index 000000000..c704f0007
--- /dev/null
+++ b/src/main/scala/li/cil/oc/server/component/traits/WorldFluidContainerAnalytics.scala
@@ -0,0 +1,68 @@
+package li.cil.oc.server.component.traits
+
+import li.cil.oc.Settings
+import li.cil.oc.api.machine.{Arguments, Callback, Context}
+import li.cil.oc.server.component.result
+import li.cil.oc.util.ExtendedArguments.extendedArguments
+import li.cil.oc.util.InventoryUtils
+import net.minecraft.inventory.IInventory
+import net.minecraft.item.ItemStack
+import net.minecraftforge.common.util.ForgeDirection
+import net.minecraftforge.fluids.{FluidContainerRegistry, FluidStack, IFluidContainerItem}
+
+trait WorldFluidContainerAnalytics extends WorldAware with SideRestricted with NetworkAware {
+
+  @Callback(doc = """function(side:number, slot:number):number -- Get the capacity of the fluid container in the specified slot of the inventory on the specified side of the device.""")
+  def getContainerCapacityInSlot(context: Context, args: Arguments): Array[AnyRef] = if (Settings.get.allowItemStackInspection) {
+    val facing = checkSideForAction(args, 0)
+    withInventory(facing, inventory => {
+      val stack = inventory.getStackInSlot(args.checkSlot(inventory, 1))
+      withFluidInfo(stack, (_, capacity) => result(capacity))
+    })
+  }
+  else result(Unit, "not enabled in config")
+
+  @Callback(doc = """function(side:number, slot:number):number -- Get the capacity the fluid container in the specified slot of the inventory on the specified side of the device.""")
+  def getContainerLevelInSlot(context: Context, args: Arguments): Array[AnyRef] = if (Settings.get.allowItemStackInspection) {
+    val facing = checkSideForAction(args, 0)
+    withInventory(facing, inventory => {
+      val stack = inventory.getStackInSlot(args.checkSlot(inventory, 1))
+      withFluidInfo(stack, (fluid, _) => result(fluid.amount))
+    })
+  }
+  else result(Unit, "not enabled in config")
+
+  @Callback(doc = """function(side:number, slot:number):table -- Get a description of the fluid in the fluid container in the specified slot of the inventory on the specified side of the device.""")
+  def getFluidInContainerInSlot(context: Context, args: Arguments): Array[AnyRef] = if (Settings.get.allowItemStackInspection) {
+    val facing = checkSideForAction(args, 0)
+    withInventory(facing, inventory => {
+      val stack = inventory.getStackInSlot(args.checkSlot(inventory, 1))
+      withFluidInfo(stack, (fluid, _) => result(fluid))
+    })
+  }
+  else result(Unit, "not enabled in config")
+
+  private def withFluidInfo(stack: ItemStack, f: (FluidStack, Int) => Array[AnyRef]) = {
+    val fluidInfo = if (FluidContainerRegistry.isFilledContainer(stack)) {
+      Option((FluidContainerRegistry.getFluidForFilledItem(stack), FluidContainerRegistry.getContainerCapacity(stack)))
+    }
+    else if (FluidContainerRegistry.isEmptyContainer(stack)) {
+      Option((new FluidStack(0, 0), FluidContainerRegistry.getContainerCapacity(stack)))
+    }
+    else stack.getItem match {
+      case from: IFluidContainerItem => Option((from.getFluid(stack), from.getCapacity(stack)))
+      case _ => None
+    }
+    fluidInfo match {
+      case Some((fluid, capacity)) => f(fluid, capacity)
+      case _ => result(Unit, "item is not a fluid container")
+    }
+  }
+
+
+  private def withInventory(side: ForgeDirection, f: IInventory => Array[AnyRef]) =
+    InventoryUtils.inventoryAt(position.offset(side)) match {
+      case Some(inventory) if inventory.isUseableByPlayer(fakePlayer) && mayInteract(position.offset(side), side.getOpposite) => f(inventory)
+      case _ => result(Unit, "no inventory")
+    }
+}