Add current pet highlighter

1 files changed, 40 insertions, 0 deletions

diff --git a/src/main/kotlin/features/inventory/PetFeatures.kt b/src/main/kotlin/features/inventory/PetFeatures.kt
new file mode 100644
index 0000000..81124f3
--- /dev/null
+++ b/src/main/kotlin/features/inventory/PetFeatures.kt
@@ -0,0 +1,40 @@
+package moe.nea.firmament.features.inventory
+
+import net.minecraft.util.Identifier
+import moe.nea.firmament.annotations.Subscribe
+import moe.nea.firmament.events.SlotRenderEvents
+import moe.nea.firmament.features.FirmamentFeature
+import moe.nea.firmament.gui.config.ManagedConfig
+import moe.nea.firmament.util.MC
+import moe.nea.firmament.util.petData
+import moe.nea.firmament.util.unformattedString
+import moe.nea.firmament.util.useMatch
+
+object PetFeatures : FirmamentFeature {
+	override val identifier: String
+		get() = "pets"
+
+	override val config: ManagedConfig?
+		get() = TConfig
+
+	object TConfig : ManagedConfig(identifier) {
+		val highlightEquippedPet by toggle("highlight-pet") { true }
+	}
+
+	val petMenuTitle = "Pets(?: \\([0-9]+/[0-9]+\\))?".toPattern()
+
+	@Subscribe
+	fun onSlotRender(event: SlotRenderEvents.Before) {
+		if (!TConfig.highlightEquippedPet) return
+		val stack = event.slot.stack
+		if (stack.petData?.active == true)
+			petMenuTitle.useMatch(MC.screenName ?: return) {
+				event.context.drawSprite(
+					event.slot.x, event.slot.y, 0, 16, 16,
+					MC.guiAtlasManager.getSprite(Identifier.of("firmament:selected_pet_background"))
+				)
+			}
+	}
+
+
+}