aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/util/HoveredItemStack.kt
blob: 91202dd913a173fe875ec155d707a17df56e69fa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package moe.nea.firmament.util

import com.google.auto.service.AutoService
import kotlin.jvm.optionals.getOrNull
import net.minecraft.client.gui.screens.Screen
import net.minecraft.client.gui.screens.inventory.AbstractContainerScreen
import net.minecraft.world.item.ItemStack
import moe.nea.firmament.api.v1.FirmamentAPI
import moe.nea.firmament.mixins.accessor.AccessorHandledScreen
import moe.nea.firmament.util.compatloader.CompatLoader

interface HoveredItemStackProvider : Comparable<HoveredItemStackProvider> {
	fun provideHoveredItemStack(screen: Screen): ItemStack?
	override fun compareTo(other: HoveredItemStackProvider): Int {
		return compareValues(this.prio, other.prio)
	}

	val prio: Int get() = 0

	companion object : CompatLoader<HoveredItemStackProvider>(HoveredItemStackProvider::class) {
		val sorted = HoveredItemStackProvider.allValidInstances.sorted()
	}
}

@AutoService(HoveredItemStackProvider::class)
class VanillaScreenProvider : HoveredItemStackProvider {

	override fun provideHoveredItemStack(screen: Screen): ItemStack? {
		if (screen !is AccessorHandledScreen) return null
		val vanillaSlot = screen.focusedSlot_Firmament?.item
		return vanillaSlot
	}

	override val prio: Int
		get() = -1
}

@AutoService(HoveredItemStackProvider::class)
class FirmamentStackScreenProvider : HoveredItemStackProvider {
	override fun provideHoveredItemStack(screen: Screen): ItemStack? {
		return FirmamentAPI.getInstance()
			.hoveredItemWidget
			.getOrNull()
			?.itemStack
	}
}

val Screen.focusedItemStack: ItemStack?
	get() =
		HoveredItemStackProvider.sorted
			.firstNotNullOfOrNull { it.provideHoveredItemStack(this)?.takeIf { !it.isEmpty } }