90 lines
3.0 KiB
Kotlin
90 lines
3.0 KiB
Kotlin
package net.shadowfacts.phycon.network.block.netinterface
|
|
|
|
import alexiil.mc.lib.attributes.SearchOptions
|
|
import alexiil.mc.lib.attributes.Simulation
|
|
import alexiil.mc.lib.attributes.item.GroupedItemInv
|
|
import alexiil.mc.lib.attributes.item.ItemAttributes
|
|
import net.minecraft.item.ItemStack
|
|
import net.minecraft.util.math.Direction
|
|
import net.shadowfacts.phycon.api.packet.Packet
|
|
import net.shadowfacts.phycon.init.PhyBlockEntities
|
|
import net.shadowfacts.phycon.network.DeviceBlockEntity
|
|
import net.shadowfacts.phycon.network.component.ItemStackPacketHandler
|
|
import net.shadowfacts.phycon.network.component.handleItemStack
|
|
import net.shadowfacts.phycon.network.packet.*
|
|
|
|
/**
|
|
* @author shadowfacts
|
|
*/
|
|
class InterfaceBlockEntity: DeviceBlockEntity(PhyBlockEntities.INTERFACE), ItemStackPacketHandler {
|
|
|
|
private val facing: Direction
|
|
get() = world!!.getBlockState(pos)[InterfaceBlock.FACING]
|
|
|
|
// todo: should this be a weak ref?
|
|
private var inventory: GroupedItemInv? = null
|
|
|
|
fun updateInventory() {
|
|
val offsetPos = pos.offset(facing)
|
|
val option = SearchOptions.inDirection(facing)
|
|
inventory = ItemAttributes.GROUPED_INV.getFirstOrNull(world, offsetPos, option)
|
|
}
|
|
|
|
private fun getInventory(): GroupedItemInv? {
|
|
// if we don't have an inventory, try to get one
|
|
// this happens when readAll is called before a neighbor state changes, such as immediately after world load
|
|
if (inventory == null) updateInventory()
|
|
return inventory
|
|
}
|
|
|
|
override fun handlePacket(packet: Packet) {
|
|
when (packet) {
|
|
is RequestInventoryPacket -> handleRequestInventory(packet)
|
|
is LocateStackPacket -> handleLocateStack(packet)
|
|
is ExtractStackPacket -> handleExtractStack(packet)
|
|
is CheckCapacityPacket -> handleCheckCapacity(packet)
|
|
is ItemStackPacket -> handleItemStack(packet)
|
|
}
|
|
}
|
|
|
|
private fun handleRequestInventory(packet: RequestInventoryPacket) {
|
|
getInventory()?.also { inv ->
|
|
sendToSingle(ReadInventoryPacket(inv, macAddress, packet.source))
|
|
}
|
|
}
|
|
|
|
private fun handleLocateStack(packet: LocateStackPacket) {
|
|
getInventory()?.also { inv ->
|
|
val amount = inv.getAmount(packet.stack)
|
|
sendToSingle(StackLocationPacket(packet.stack, amount, this, macAddress, packet.source))
|
|
}
|
|
}
|
|
|
|
private fun handleExtractStack(packet: ExtractStackPacket) {
|
|
getInventory()?.also { inv ->
|
|
val extracted = inv.extract(packet.stack, packet.amount)
|
|
sendToSingle(ItemStackPacket(extracted, macAddress, packet.source))
|
|
}
|
|
}
|
|
|
|
private fun handleCheckCapacity(packet: CheckCapacityPacket) {
|
|
getInventory()?.also { inv ->
|
|
val remaining = inv.attemptInsertion(packet.stack, Simulation.SIMULATE)
|
|
val couldAccept = packet.stack.count - remaining.count
|
|
sendToSingle(CapacityPacket(packet.stack, couldAccept, this, macAddress, packet.source))
|
|
}
|
|
}
|
|
|
|
override fun doHandleItemStack(packet: ItemStackPacket): ItemStack {
|
|
val inventory = getInventory()
|
|
if (inventory != null) {
|
|
val remaining = inventory.insert(packet.stack)
|
|
// whatever could not be inserted will be sent back to the packet's source
|
|
return remaining
|
|
} else {
|
|
return packet.stack
|
|
}
|
|
}
|
|
|
|
}
|