feat: Add kat upgrade and flower detection

1 files changed, 100 insertions, 0 deletions

diff --git a/src/main/kotlin/moe/nea/ledger/modules/KatDetection.kt b/src/main/kotlin/moe/nea/ledger/modules/KatDetection.kt
new file mode 100644
index 0000000..8a2aa19
--- /dev/null
+++ b/src/main/kotlin/moe/nea/ledger/modules/KatDetection.kt
@@ -0,0 +1,100 @@
+package moe.nea.ledger.modules
+
+import moe.nea.ledger.ItemChange
+import moe.nea.ledger.ItemId
+import moe.nea.ledger.ItemIdProvider
+import moe.nea.ledger.LedgerEntry
+import moe.nea.ledger.LedgerLogger
+import moe.nea.ledger.TransactionType
+import moe.nea.ledger.events.BeforeGuiAction
+import moe.nea.ledger.events.ChatReceived
+import moe.nea.ledger.getInternalId
+import moe.nea.ledger.getLore
+import moe.nea.ledger.unformattedString
+import moe.nea.ledger.useMatcher
+import moe.nea.ledger.utils.Inject
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+
+
+class KatDetection {
+	@Inject
+	lateinit var log: LedgerLogger
+
+	@Inject
+	lateinit var itemIdProvider: ItemIdProvider
+
+	val giftNameToIdMap = mapOf(
+		"flower" to ItemId("KAT_FLOWER"),
+		"bouquet" to ItemId("KAT_BOUQUET"),
+	)
+	val katGift = "\\[NPC\\] Kat: A (?<gift>.*)\\? For me\\? How sweet!".toPattern()
+
+	@SubscribeEvent
+	fun onChat(event: ChatReceived) {
+		katGift.useMatcher(event.message) {
+			val giftName = group("gift")
+			val giftId = giftNameToIdMap[giftName]
+			log.logEntry(LedgerEntry(
+				TransactionType.KAT_TIMESKIP,
+				event.timestamp,
+				listOf(
+					ItemChange.lose(giftId ?: ItemId.NIL, 1)
+				)
+			))
+		}
+	}
+
+	val confirmSlot = 9 + 9 + 4
+	val petSlot = 9 + 4
+
+	data class PetUpgrade(
+		val beforePetId: ItemId,
+		val cost: List<Pair<ItemId, Double>>
+	)
+
+	var lastPetUpgradeScheduled: PetUpgrade? = null
+
+	@SubscribeEvent
+	fun onClick(event: BeforeGuiAction) {
+		val slots = event.chestSlots ?: return
+		val petItem = slots.lowerChestInventory.getStackInSlot(petSlot) ?: return
+		val beforePetId = petItem.getInternalId() ?: return
+		val confirmItem = slots.lowerChestInventory.getStackInSlot(confirmSlot) ?: return
+		val lore = confirmItem.getLore()
+		val cost = lore.iterator().asSequence()
+			.dropWhile { it.unformattedString() != "Cost" }.drop(1)
+			.takeWhile { it != "" }
+			.map { itemIdProvider.findFromLore(it) ?: Pair(ItemId.NIL, 1.0) }
+			.toList()
+		lastPetUpgradeScheduled = PetUpgrade(beforePetId, cost)
+	}
+
+	val petUpgradeDialogue = "\\[NPC\\] Kat: I'll get your (?<type>.*) upgraded to (?<tier>.*) in no time!".toPattern()
+	fun upgradePetTier(itemId: ItemId): ItemId {
+		val str = itemId.string.split(";", limit = 2)
+		if (str.size == 2) {
+			val (type, tier) = str
+			val tierT = tier.toIntOrNull()
+			if (tierT != null)
+				return ItemId(type + ";" + (tierT + 1))
+		}
+		return itemId
+	}
+
+	@SubscribeEvent
+	fun onPetUpgrade(event: ChatReceived) {
+		petUpgradeDialogue.useMatcher(event.message) {
+			val upgrade = lastPetUpgradeScheduled ?: return
+			lastPetUpgradeScheduled = null
+			log.logEntry(LedgerEntry(
+				TransactionType.KAT_UPGRADE,
+				event.timestamp,
+				listOf(
+					ItemChange.lose(upgrade.beforePetId, 1),
+					ItemChange.gain(upgradePetTier(upgrade.beforePetId), 1),
+				) + upgrade.cost.map { ItemChange.lose(it.first, it.second) },
+			))
+		}
+	}
+
+}
\ No newline at end of file