block models and stuff

src/main/kotlin/org/neoflock/neocomputers/utils/GenericContainer.kt

@@ -1,9 +1,19 @@
 package org.neoflock.neocomputers.utils
 
 // based off the ImplementedContainer of https://docs.fabricmc.net/develop/blocks/block-containers
+import net.minecraft.client.gui.GuiGraphics
+import net.minecraft.client.gui.screens.inventory.AbstractContainerScreen
 import net.minecraft.world.Container;
 import net.minecraft.core.NonNullList;
+import net.minecraft.network.chat.Component
+import net.minecraft.resources.ResourceLocation
+import net.minecraft.server.packs.resources.Resource
 import net.minecraft.world.ContainerHelper
+import net.minecraft.world.entity.player.Inventory
+import net.minecraft.world.entity.player.Player
+import net.minecraft.world.inventory.AbstractContainerMenu
+import net.minecraft.world.inventory.MenuType
+import net.minecraft.world.inventory.Slot
 import net.minecraft.world.item.ItemStack
 
 // Common container interface, assumes the entire purpose is purely raw item storage
@@ -45,4 +55,87 @@ interface GenericContainer : Container {
     override fun clearContent() {
         getItems().clear()
     }
+}
+
+abstract class GenericContainerMenu(menuType: MenuType<*>, id: Int, var container: Container): AbstractContainerMenu(menuType, id) {
+    fun addInventorySlots(inventory: Inventory, x: Int, y: Int) {
+        // Based off the code in ChestMenu
+        for (i in 0..2) {
+            for (j in 0..8) {
+                this.addSlot(Slot(inventory, j + i * 9 + 9, 8 + j * 18, 84 + i * 18))
+            }
+        }
+
+        addInventoryHotbar(inventory, x, y + 3 * 18 + 4)
+    }
+
+    fun addInventoryHotbar(inventory: Inventory, x: Int, y: Int) {
+        for (i in 0..8) {
+            this.addSlot(Slot(inventory, i, x + i * 18, y))
+        }
+    }
+
+    // taken from https://docs.fabricmc.net/develop/blocks/container-menus
+    override fun quickMoveStack(player: Player, i: Int): ItemStack? {
+        val slot = slots[i]
+
+        if(!slot.hasItem()) return ItemStack.EMPTY
+
+        val stack = slot.item
+        val copied = stack.copy()
+        val contSize = container.containerSize
+
+        if(i < contSize) {
+            if(!this.moveItemStackTo(stack, contSize, slots.size, true)) {
+                return ItemStack.EMPTY
+            }
+        } else if(!this.moveItemStackTo(stack, 0, contSize, false)) {
+            return ItemStack.EMPTY
+        }
+
+        if(stack.isEmpty) {
+            slot.setByPlayer(ItemStack.EMPTY)
+        } else {
+            slot.setChanged()
+        }
+
+        return copied
+    }
+
+    override fun stillValid(player: Player): Boolean {
+        return container.stillValid(player)
+    }
+}
+
+abstract class GenericContainerScreen<T: GenericContainerMenu>(menu: T, inventory: Inventory, component: Component): AbstractContainerScreen<T>(menu, inventory, component) {
+    open fun shouldCenterTitle() = true
+    open fun shouldRenderTooltip() = true
+    open fun findMenuTexture(): ResourceLocation? = null
+
+    val imageX: Int
+        get() = (width - imageWidth) / 2
+
+    val imageY: Int
+        get() = (height - imageHeight) / 2
+
+    override fun init() {
+        super.init()
+
+        if(shouldCenterTitle()) this.titleLabelX = (this.imageWidth - this.font.width(this.title)) / 2
+    }
+
+    override fun renderBg(guiGraphics: GuiGraphics, f: Float, i: Int, j: Int) {
+        val menuTex = findMenuTexture()
+        if(menuTex != null) {
+            val cx = (width - imageWidth) / 2
+            val cy = (height - imageHeight) / 2
+
+            guiGraphics.blit(menuTex, imageX, imageY, 0, 0, imageWidth, imageHeight)
+        }
+    }
+
+    override fun render(graphics: GuiGraphics, mouseX: Int, mouseY: Int, something: Float) {
+        super.render(graphics, mouseX, mouseY, something)
+        if(shouldRenderTooltip()) super.renderTooltip(graphics, mouseX, mouseY)
+    }
 }