Farming fortune accessory support (#37)

1 files changed, 124 insertions, 0 deletions

diff --git a/src/main/java/at/hannibal2/skyhanni/data/CropAccessoryData.kt b/src/main/java/at/hannibal2/skyhanni/data/CropAccessoryData.kt
new file mode 100644
index 000000000..ea50257c3
--- /dev/null
+++ b/src/main/java/at/hannibal2/skyhanni/data/CropAccessoryData.kt
@@ -0,0 +1,124 @@
+package at.hannibal2.skyhanni.data
+
+import at.hannibal2.skyhanni.SkyHanniMod
+import at.hannibal2.skyhanni.events.GuiContainerEvent
+import at.hannibal2.skyhanni.events.InventoryOpenEvent
+import at.hannibal2.skyhanni.events.ProfileApiDataLoadedEvent
+import at.hannibal2.skyhanni.events.ProfileJoinEvent
+import at.hannibal2.skyhanni.features.garden.CropAccessory
+import at.hannibal2.skyhanni.utils.InventoryUtils
+import at.hannibal2.skyhanni.utils.ItemUtils.getInternalName
+import at.hannibal2.skyhanni.utils.LorenzUtils
+import at.hannibal2.skyhanni.utils.NEUItems
+import com.google.gson.JsonElement
+import net.minecraft.item.ItemStack
+import net.minecraft.nbt.CompressedStreamTools
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+import net.minecraftforge.fml.common.gameevent.TickEvent
+import java.io.ByteArrayInputStream
+import java.util.*
+
+class CropAccessoryData {
+    private val accessoryBagNamePattern = "Accessory Bag \\((\\d)/(\\d)\\)".toRegex()
+    private var loadedAccessoryThisProfile = false
+    private var ticks = 0
+    private var accessoryInBag: CropAccessory? = null
+    private var accessoryInInventory = CropAccessory.NONE
+
+    private var accessoryBagPageNumber = 0
+
+    // Handle API detection
+    @SubscribeEvent
+    fun onProfileJoin(event: ProfileJoinEvent) {
+        loadedAccessoryThisProfile = false
+    }
+
+    @SubscribeEvent
+    fun onProfileDataLoad(event: ProfileApiDataLoadedEvent) {
+        if (loadedAccessoryThisProfile) return
+        val inventoryData = event.profileData["inv_contents"] ?: return
+        val accessories = getCropAccessories(event.profileData["talisman_bag"]).also {
+            it.addAll(getCropAccessories(inventoryData))
+        }
+        cropAccessory = accessories.maxOrNull() ?: CropAccessory.NONE
+        loadedAccessoryThisProfile = true
+    }
+
+    // Handle accessory bag detection
+    @SubscribeEvent
+    fun onGuiDraw(event: InventoryOpenEvent) {
+        val groups = accessoryBagNamePattern.matchEntire(event.inventoryName)?.groups ?: return
+        accessoryBagPageCount = groups[2]!!.value.toInt()
+        accessoryBagPageNumber = groups[1]!!.value.toInt()
+        isLoadingAccessories = true
+
+        val bestCropAccessoryPage = bestCropAccessory(event.inventoryItems.values)
+        accessoryPage[accessoryBagPageNumber] = bestCropAccessoryPage
+        if (accessoryBagPageCount == accessoryPage.size) {
+            accessoryInBag = accessoryPage.values.max().also {
+                cropAccessory = maxOf(it, accessoryInInventory)
+            }
+            loadedAccessoryThisProfile = true
+        }
+    }
+
+    @SubscribeEvent
+    fun onCloseWindow(event: GuiContainerEvent.CloseWindowEvent) {
+        isLoadingAccessories = false
+    }
+
+    // Handle inventory detection
+    @SubscribeEvent
+    fun onTick(event: TickEvent.ClientTickEvent) {
+        if (event.phase != TickEvent.Phase.START || ticks++ % 20 != 0) return
+        if (!LorenzUtils.inSkyBlock) return
+        accessoryInInventory = bestCropAccessory(InventoryUtils.getItemsInOwnInventory())
+        if (accessoryInInventory == CropAccessory.NONE) return
+        if (accessoryInInventory > (accessoryInBag ?: CropAccessory.NONE)) {
+            cropAccessory = accessoryInInventory
+        }
+    }
+
+
+    private fun bestCropAccessory(items: Iterable<ItemStack>): CropAccessory {
+        return items.mapNotNull { item -> CropAccessory.getByName(item.getInternalName()) }
+            .maxOrNull() ?: CropAccessory.NONE
+    }
+
+    companion object {
+        var accessoryBagPageCount = 0
+            private set
+
+        private var accessoryPage = mutableMapOf<Int, CropAccessory>()
+
+        var isLoadingAccessories = false
+            private set
+
+        val pagesLoaded get() = accessoryPage.size
+
+        var cropAccessory: CropAccessory?
+            get() = SkyHanniMod.feature.hidden.savedCropAccessory
+            private set(accessory) {
+                SkyHanniMod.feature.hidden.savedCropAccessory = accessory
+            }
+
+        // Derived partially from NotEnoughUpdates/NotEnoughUpdates, ProfileViewer.Profile#getInventoryInfo
+        private fun getCropAccessories(inventory: JsonElement?): MutableList<CropAccessory> {
+            if (inventory == null) return mutableListOf()
+            val cropAccessories = mutableListOf<CropAccessory>()
+            val data = inventory.asJsonObject["data"]?.asString
+            val accessoryBagItems = CompressedStreamTools.readCompressed(
+                ByteArrayInputStream(Base64.getDecoder().decode(data))
+            ).getTagList("i", 10)
+            for (j in 0 until accessoryBagItems.tagCount()) {
+                val itemStackTag = accessoryBagItems.getCompoundTagAt(j)
+                if (!itemStackTag.hasKey("tag")) continue
+                val itemTag = itemStackTag.getCompoundTag("tag")
+                val itemName = NEUItems.getInternalNameOrNull(itemTag) ?: continue
+                val itemAsCropAccessory = CropAccessory.getByName(itemName) ?: continue
+                cropAccessories.add(itemAsCropAccessory)
+            }
+            return cropAccessories
+        }
+    }
+}