1 files changed, 141 insertions, 0 deletions
diff --git a/src/main/kotlin/features/items/recipes/ItemSlotWidget.kt b/src/main/kotlin/features/items/recipes/ItemSlotWidget.kt
new file mode 100644
index 0000000..b659643
--- /dev/null
+++ b/src/main/kotlin/features/items/recipes/ItemSlotWidget.kt
@@ -0,0 +1,141 @@
+package moe.nea.firmament.features.items.recipes
+
+import java.util.Optional
+import me.shedaniel.math.Dimension
+import me.shedaniel.math.Point
+import me.shedaniel.math.Rectangle
+import net.fabricmc.fabric.api.client.item.v1.ItemTooltipCallback
+import net.minecraft.client.gui.GuiGraphics
+import net.minecraft.network.chat.Component
+import net.minecraft.world.item.Item
+import net.minecraft.world.item.ItemStack
+import net.minecraft.world.item.TooltipFlag
+import moe.nea.firmament.api.v1.FirmamentItemWidget
+import moe.nea.firmament.events.ItemTooltipEvent
+import moe.nea.firmament.keybindings.SavedKeyBinding
+import moe.nea.firmament.repo.ExpensiveItemCacheApi
+import moe.nea.firmament.repo.SBItemStack
+import moe.nea.firmament.repo.recipes.RecipeLayouter
+import moe.nea.firmament.util.ErrorUtil
+import moe.nea.firmament.util.FirmFormatters.shortFormat
+import moe.nea.firmament.util.MC
+import moe.nea.firmament.util.darkGrey
+import moe.nea.firmament.util.mc.displayNameAccordingToNbt
+import moe.nea.firmament.util.mc.loreAccordingToNbt
+
+class ItemSlotWidget(
+	point: Point,
+	var content: List<SBItemStack>,
+	val slotKind: RecipeLayouter.SlotKind
+) : RecipeWidget(),
+	RecipeLayouter.CyclingItemSlot,
+	FirmamentItemWidget {
+	override var position = point
+	override val size get() = Dimension(16, 16)
+	val itemRect get() = Rectangle(position, Dimension(16, 16))
+
+	val backgroundTopLeft
+		get() =
+			if (slotKind.isBig) Point(position.x - 4, position.y - 4)
+			else Point(position.x - 1, position.y - 1)
+	val backgroundSize =
+		if (slotKind.isBig) Dimension(16 + 8, 16 + 8)
+		else Dimension(18, 18)
+	override val rect: Rectangle
+		get() = Rectangle(backgroundTopLeft, backgroundSize)
+
+	@OptIn(ExpensiveItemCacheApi::class)
+	override fun render(
+		guiGraphics: GuiGraphics,
+		mouseX: Int,
+		mouseY: Int,
+		partialTick: Float
+	) {
+		val stack = current().asImmutableItemStack()
+		// TODO: draw slot background
+		if (stack.isEmpty) return
+		guiGraphics.renderItem(stack, position.x, position.y)
+		guiGraphics.renderItemDecorations(
+			MC.font, stack, position.x, position.y,
+			if (stack.count >= SHORT_NUM_CUTOFF) shortFormat(stack.count.toDouble())
+			else null
+		)
+		if (itemRect.contains(mouseX, mouseY)
+			&& guiGraphics.containsPointInScissor(mouseX, mouseY)
+		) guiGraphics.setTooltipForNextFrame(
+			MC.font, getTooltip(stack), Optional.empty(),
+			mouseX, mouseY
+		)
+	}
+
+	companion object {
+		val SHORT_NUM_CUTOFF = 1000
+		var canUseTooltipEvent = true
+
+		fun getTooltip(itemStack: ItemStack): List<Component> {
+			val lore = mutableListOf(itemStack.displayNameAccordingToNbt)
+			lore.addAll(itemStack.loreAccordingToNbt)
+			if (canUseTooltipEvent) {
+				try {
+					ItemTooltipCallback.EVENT.invoker().getTooltip(
+						itemStack, Item.TooltipContext.EMPTY,
+						TooltipFlag.NORMAL, lore
+					)
+				} catch (ex: Exception) {
+					canUseTooltipEvent = false
+					ErrorUtil.softError("Failed to use vanilla tooltips", ex)
+				}
+			} else {
+				ItemTooltipEvent.publish(
+					ItemTooltipEvent(
+						itemStack,
+						Item.TooltipContext.EMPTY,
+						TooltipFlag.NORMAL,
+						lore
+					)
+				)
+			}
+			if (itemStack.count >= SHORT_NUM_CUTOFF && lore.isNotEmpty())
+				lore.add(1, Component.literal("${itemStack.count}x").darkGrey())
+			return lore
+		}
+	}
+
+
+	override fun tick() {
+		if (SavedKeyBinding.isShiftDown()) return
+		if (content.size <= 1) return
+		if (MC.currentTick % 5 != 0) return
+		index = (index + 1) % content.size
+	}
+
+	var index = 0
+	var onUpdate: () -> Unit = {}
+
+	override fun onUpdate(action: () -> Unit) {
+		this.onUpdate = action
+	}
+
+	override fun current(): SBItemStack {
+		return content.getOrElse(index) { SBItemStack.EMPTY }
+	}
+
+	override fun update(newValue: SBItemStack) {
+		content = listOf(newValue)
+		// SAFE: content was just assigned to a non-empty list
+		index = index.coerceIn(content.indices)
+	}
+
+	override fun getPlacement(): FirmamentItemWidget.Placement {
+		return FirmamentItemWidget.Placement.RECIPE_SCREEN
+	}
+
+	@OptIn(ExpensiveItemCacheApi::class)
+	override fun getItemStack(): ItemStack {
+		return current().asImmutableItemStack()
+	}
+
+	override fun getSkyBlockId(): String {
+		return current().skyblockId.neuItem
+	}
+}