package at.hannibal2.skyhanni.data import at.hannibal2.skyhanni.SkyHanniMod import at.hannibal2.skyhanni.config.ConfigFileType import at.hannibal2.skyhanni.config.features.inventory.SackDisplayConfig.PriceFrom import at.hannibal2.skyhanni.events.InventoryCloseEvent import at.hannibal2.skyhanni.events.InventoryFullyOpenedEvent import at.hannibal2.skyhanni.events.LorenzChatEvent import at.hannibal2.skyhanni.events.SackChangeEvent import at.hannibal2.skyhanni.events.SackDataUpdateEvent import at.hannibal2.skyhanni.features.fishing.FishingAPI import at.hannibal2.skyhanni.features.fishing.trophy.TrophyRarity import at.hannibal2.skyhanni.features.inventory.SackDisplay import at.hannibal2.skyhanni.utils.CollectionUtils.editCopy import at.hannibal2.skyhanni.utils.ItemUtils.getInternalName import at.hannibal2.skyhanni.utils.ItemUtils.getLore import at.hannibal2.skyhanni.utils.ItemUtils.name import at.hannibal2.skyhanni.utils.NEUInternalName import at.hannibal2.skyhanni.utils.NEUInternalName.Companion.asInternalName import at.hannibal2.skyhanni.utils.NEUItems.getNpcPriceOrNull import at.hannibal2.skyhanni.utils.NEUItems.getPrice import at.hannibal2.skyhanni.utils.NumberUtil.formatInt import at.hannibal2.skyhanni.utils.NumberUtil.formatLong import at.hannibal2.skyhanni.utils.StringUtils.matchMatcher import at.hannibal2.skyhanni.utils.StringUtils.matches import at.hannibal2.skyhanni.utils.StringUtils.removeColor import at.hannibal2.skyhanni.utils.repopatterns.RepoPattern import com.google.gson.annotations.Expose import net.minecraft.item.ItemStack import net.minecraftforge.fml.common.eventhandler.SubscribeEvent object SackAPI { private val sackDisplayConfig get() = SkyHanniMod.feature.inventory.sackDisplay private val chatConfig get() = SkyHanniMod.feature.chat private var lastOpenedInventory = "" var inSackInventory = false private val patternGroup = RepoPattern.group("data.sacks") private val sackPattern by patternGroup.pattern( "sack", "^(.* Sack|Enchanted .* Sack)\$" ) private val numPattern by patternGroup.pattern( "number", "(?:(?:§[0-9a-f](?I{1,3})§7:)?|(?:§7Stored:)?) (?§[0-9a-f])(?[0-9.,kKmMbB]+)§7/(?\\d+(?:[0-9.,]+)?[kKmMbB]?)" ) private val gemstonePattern by patternGroup.pattern( "gemstone", " §[0-9a-f](?[A-z]*): §[0-9a-f](?\\d+(?:\\.\\d+)?(?:(?:,\\d+)?)+[kKmM]?)(?: §[0-9a-f]\\(\\d+(?:\\.\\d+)?(?:(?:,\\d+)?)+[kKmM]?\\))?" ) private var isRuneSack = false private var isGemstoneSack = false var isTrophySack = false private var sackRarity: TrophyRarity? = null val sackItem = mutableMapOf() val runeItem = mutableMapOf() val gemstoneItem = mutableMapOf() private val stackList = mutableMapOf() @SubscribeEvent fun onInventoryClose(event: InventoryCloseEvent) { inSackInventory = false isRuneSack = false isGemstoneSack = false isTrophySack = false runeItem.clear() gemstoneItem.clear() sackItem.clear() stackList.clear() } @SubscribeEvent fun onInventoryOpen(event: InventoryFullyOpenedEvent) { val inventoryName = event.inventoryName val isNewInventory = inventoryName != lastOpenedInventory lastOpenedInventory = inventoryName val match = sackPattern.matches(inventoryName) if (!match) return val stacks = event.inventoryItems isRuneSack = inventoryName == "Runes Sack" isGemstoneSack = inventoryName == "Gemstones Sack" isTrophySack = inventoryName.contains("Trophy Fishing Sack") sackRarity = inventoryName.getTrophyRarity() inSackInventory = true stackList.putAll(stacks) SackDisplay.update(isNewInventory) } private fun String.getTrophyRarity(): TrophyRarity? { return if (this.startsWith("Bronze")) TrophyRarity.BRONZE else if (this.startsWith("Silver")) TrophyRarity.SILVER else null } private fun NEUInternalName.sackPrice(stored: String) = when (sackDisplayConfig.priceFrom) { PriceFrom.BAZAAR -> (getPrice(true) * stored.formatLong()).toLong() .let { if (it < 0) 0L else it } PriceFrom.NPC -> try { val npcPrice = getNpcPriceOrNull() ?: 0.0 (npcPrice * stored.formatLong()).toLong() } catch (e: Exception) { 0L } else -> 0L } fun getSacksData(savingSacks: Boolean) { if (savingSacks) sackData = ProfileStorageData.sackProfiles?.sackContents ?: return for ((_, stack) in stackList) { val name = stack.name val lore = stack.getLore() val gem = SackGemstone() val rune = SackRune() val item = SackOtherItem() loop@ for (line in lore) { if (isGemstoneSack) { gemstonePattern.matchMatcher(line) { val rarity = group("gemrarity") val stored = group("stored") gem.internalName = gemstoneMap[name.removeColor()] ?: NEUInternalName.NONE if (gemstoneMap.containsKey(name.removeColor())) { val internalName = "${rarity.uppercase()}_${ name.uppercase().split(" ")[0].removeColor() }_GEM".asInternalName() when (rarity) { "Rough" -> { gem.rough = stored gem.roughPrice = internalName.sackPrice(stored) if (savingSacks) setSackItem(internalName, stored.formatLong()) } "Flawed" -> { gem.flawed = stored gem.flawedPrice = internalName.sackPrice(stored) if (savingSacks) setSackItem(internalName, stored.formatLong()) } "Fine" -> { gem.fine = stored gem.finePrice = internalName.sackPrice(stored) if (savingSacks) setSackItem(internalName, stored.formatLong()) } } gemstoneItem[name] = gem } } } else { numPattern.matchMatcher(line) { val stored = group("stored") val internalName = stack.getInternalName() item.internalName = internalName item.colorCode = group("color") item.stored = stored item.total = group("total") if (savingSacks) setSackItem(item.internalName, item.stored.formatLong()) item.price = if (isTrophySack) { val filletPerTrophy = FishingAPI.getFilletPerTrophy(stack.getInternalName()) val filletValue = filletPerTrophy * stored.formatLong() item.magmaFish = filletValue "MAGMA_FISH".asInternalName().sackPrice(filletValue.toString()) } else { internalName.sackPrice(stored).coerceAtLeast(0) } if (isRuneSack) { val level = group("level") rune.stack = stack if (level == "I") { rune.lvl1 = stored continue@loop } if (level == "II") { rune.lvl2 = stored continue@loop } if (level == "III") { rune.lvl3 = stored } runeItem.put(name, rune) } else { sackItem.put(name, item) } } } } } if (savingSacks) saveSackData() } private var sackData = mapOf() data class SackChange(val delta: Int, val internalName: NEUInternalName, val sacks: List) private val sackChangeRegex = Regex("""([+-][\d,]+) (.+) \((.+)\)""") @SubscribeEvent fun onChat(event: LorenzChatEvent) { if (!event.message.removeColor().startsWith("[Sacks]")) return val sackAddText = event.chatComponent.siblings.firstNotNullOfOrNull { sibling -> sibling.chatStyle?.chatHoverEvent?.value?.formattedText?.removeColor()?.takeIf { it.startsWith("Added") } } ?: "" val sackRemoveText = event.chatComponent.siblings.firstNotNullOfOrNull { sibling -> sibling.chatStyle?.chatHoverEvent?.value?.formattedText?.removeColor()?.takeIf { it.startsWith("Removed") } } ?: "" val sackChangeText = sackAddText + sackRemoveText if (sackChangeText.isEmpty()) return val otherItemsAdded = sackAddText.contains("other items") val otherItemsRemoved = sackRemoveText.contains("other items") val sackChanges = ArrayList() for (match in sackChangeRegex.findAll(sackChangeText)) { val delta = match.groups[1]!!.value.formatInt() val item = match.groups[2]!!.value val sacks = match.groups[3]!!.value.split(", ") val internalName = NEUInternalName.fromItemName(item) sackChanges.add(SackChange(delta, internalName, sacks)) } val sackEvent = SackChangeEvent(sackChanges, otherItemsAdded, otherItemsRemoved) updateSacks(sackEvent) sackEvent.postAndCatch() if (chatConfig.hideSacksChange) { event.blockedReason = "sacks_change" } } private fun updateSacks(changes: SackChangeEvent) { sackData = ProfileStorageData.sackProfiles?.sackContents ?: return // if it gets added and subtracted but only 1 shows it will be outdated val justChanged = mutableMapOf() for (change in changes.sackChanges) { if (change.internalName in justChanged) { justChanged[change.internalName] = (justChanged[change.internalName] ?: 0) + change.delta } else { justChanged[change.internalName] = change.delta } } for (item in justChanged) { if (sackData.containsKey(item.key)) { val oldData = sackData[item.key] var newAmount = oldData!!.amount + item.value var changed = (newAmount - oldData.amount).toInt() if (newAmount < 0) { newAmount = 0 changed = 0 } sackData = sackData.editCopy { this[item.key] = SackItem(newAmount, changed, oldData.getStatus()) } } else { val newAmount = if (item.value > 0) item.value else 0 sackData = sackData.editCopy { this[item.key] = SackItem(newAmount.toLong(), newAmount, SackStatus.OUTDATED) } } } if (changes.otherItemsAdded || changes.otherItemsRemoved) { for (item in sackData) { if (item.key in justChanged) continue val oldData = sackData[item.key] sackData = sackData.editCopy { this[item.key] = SackItem(oldData!!.amount, 0, SackStatus.ALRIGHT) } } } saveSackData() } private fun setSackItem(item: NEUInternalName, amount: Long) { sackData = sackData.editCopy { this[item] = SackItem(amount, 0, SackStatus.CORRECT) } } fun fetchSackItem(item: NEUInternalName): SackItem { sackData = ProfileStorageData.sackProfiles?.sackContents ?: return SackItem(0, 0, SackStatus.MISSING) if (sackData.containsKey(item)) { return sackData[item] ?: return SackItem(0, 0, SackStatus.MISSING) } sackData = sackData.editCopy { this[item] = SackItem(0, 0, SackStatus.MISSING) } return sackData[item] ?: return SackItem(0, 0, SackStatus.MISSING) } private fun saveSackData() { ProfileStorageData.sackProfiles?.sackContents = sackData SkyHanniMod.configManager.saveConfig(ConfigFileType.SACKS, "saving-data") SackDataUpdateEvent().postAndCatch() } data class SackGemstone( var internalName: NEUInternalName = NEUInternalName.NONE, var rough: String = "0", var flawed: String = "0", var fine: String = "0", var roughPrice: Long = 0, var flawedPrice: Long = 0, var finePrice: Long = 0, ) data class SackRune( var stack: ItemStack? = null, var lvl1: String = "0", var lvl2: String = "0", var lvl3: String = "0", ) data class SackOtherItem( var internalName: NEUInternalName = NEUInternalName.NONE, var colorCode: String = "", var stored: String = "0", var total: String = "0", var price: Long = 0, var magmaFish: Long = 0, ) } data class SackItem( @Expose val amount: Long, @Expose val lastChange: Int, @Expose private val status: SackStatus?, ) { fun getStatus() = status ?: SackStatus.MISSING fun statusIsCorrectOrAlright() = getStatus().let { it == SackStatus.CORRECT || it == SackStatus.ALRIGHT } } private val gemstoneMap = mapOf( "Jade Gemstones" to "ROUGH_JADE_GEM".asInternalName(), "Amber Gemstones" to "ROUGH_AMBER_GEM".asInternalName(), "Topaz Gemstones" to "ROUGH_TOPAZ_GEM".asInternalName(), "Sapphire Gemstones" to "ROUGH_SAPPHIRE_GEM".asInternalName(), "Amethyst Gemstones" to "ROUGH_AMETHYST_GEM".asInternalName(), "Jasper Gemstones" to "ROUGH_JASPER_GEM".asInternalName(), "Ruby Gemstones" to "ROUGH_RUBY_GEM".asInternalName(), "Opal Gemstones" to "ROUGH_OPAL_GEM".asInternalName(), ) // ideally should be correct but using alright should also be fine unless they sold their whole sacks enum class SackStatus { MISSING, CORRECT, ALRIGHT, OUTDATED; }