feat: God Pot and God Pot Mixin Detection

5 files changed, 79 insertions, 0 deletions

diff --git a/src/main/kotlin/moe/nea/ledger/ItemId.kt b/src/main/kotlin/moe/nea/ledger/ItemId.kt
index fdfa19f..a940495 100644
--- a/src/main/kotlin/moe/nea/ledger/ItemId.kt
+++ b/src/main/kotlin/moe/nea/ledger/ItemId.kt
@@ -31,6 +31,7 @@ value class ItemId(
 		val ARCHFIEND_LOW_CLASS = ItemId("ARCHFIEND_DICE")
 		val ARCHFIEND_HIGH_CLASS = ItemId("HIGH_CLASS_ARCHFIEND_DICE")
 		val CAP_EYEDROPS = ItemId("CAPSAICIN_EYEDROPS_NO_CHARGES")
+		val GOD_POTION = ItemId("GOD_POTION_2")
 		val ARCHFIEND_DYE = ItemId("DYE_ARCHFIEND")
 		val SLEEPING_EYE = ItemId("SLEEPING_EYE")
 		val SUMMONING_EYE = ItemId("SUMMONING_EYE")
diff --git a/src/main/kotlin/moe/nea/ledger/Ledger.kt b/src/main/kotlin/moe/nea/ledger/Ledger.kt
index d2d09eb..5682797 100644
--- a/src/main/kotlin/moe/nea/ledger/Ledger.kt
+++ b/src/main/kotlin/moe/nea/ledger/Ledger.kt
@@ -24,6 +24,8 @@ import moe.nea.ledger.modules.ExternalDataProvider
 import moe.nea.ledger.modules.EyedropsDetection
 import moe.nea.ledger.modules.ForgeDetection
 import moe.nea.ledger.modules.GambleDetection
+import moe.nea.ledger.modules.GodPotionDetection
+import moe.nea.ledger.modules.GodPotionMixinDetection
 import moe.nea.ledger.modules.KatDetection
 import moe.nea.ledger.modules.KuudraChestDetection
 import moe.nea.ledger.modules.MineshaftCorpseDetection
@@ -131,6 +133,8 @@ class Ledger {
 			EyedropsDetection::class.java,
 			ForgeDetection::class.java,
 			GambleDetection::class.java,
+			GodPotionDetection::class.java,
+			GodPotionMixinDetection::class.java,
 			ItemIdProvider::class.java,
 			KatDetection::class.java,
 			KuudraChestDetection::class.java,
diff --git a/src/main/kotlin/moe/nea/ledger/TransactionType.kt b/src/main/kotlin/moe/nea/ledger/TransactionType.kt
index 00feebb..26749d6 100644
--- a/src/main/kotlin/moe/nea/ledger/TransactionType.kt
+++ b/src/main/kotlin/moe/nea/ledger/TransactionType.kt
@@ -20,6 +20,8 @@ enum class TransactionType {
 	DRACONIC_SACRIFICE,
 	DUNGEON_CHEST_OPEN,
 	FORGED,
+	GOD_POTION_DRANK,
+	GOD_POTION_MIXIN_DRANK,
 	KAT_TIMESKIP,
 	KAT_UPGRADE,
 	KISMET_REROLL,
diff --git a/src/main/kotlin/moe/nea/ledger/modules/GodPotionDetection.kt b/src/main/kotlin/moe/nea/ledger/modules/GodPotionDetection.kt
new file mode 100644
index 0000000..806feb0
--- /dev/null
+++ b/src/main/kotlin/moe/nea/ledger/modules/GodPotionDetection.kt
@@ -0,0 +1,34 @@
+package moe.nea.ledger.modules
+
+import moe.nea.ledger.ItemChange
+import moe.nea.ledger.ItemId
+import moe.nea.ledger.LedgerEntry
+import moe.nea.ledger.LedgerLogger
+import moe.nea.ledger.TransactionType
+import moe.nea.ledger.events.ChatReceived
+import moe.nea.ledger.useMatcher
+import moe.nea.ledger.utils.di.Inject
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+
+class GodPotionDetection {
+
+    val godPotionDrank = "(SIP|SLURP|GULP|CHUGALUG)! The God Potion grants you powers for .*!".toPattern()
+
+    @Inject
+    lateinit var logger: LedgerLogger
+
+    @SubscribeEvent
+    fun onChat(event: ChatReceived) {
+        godPotionDrank.useMatcher(event.message) {
+            logger.logEntry(
+                LedgerEntry(
+                    TransactionType.GOD_POTION_DRANK,
+                    event.timestamp,
+                    listOf(
+                        ItemChange.lose(ItemId.GOD_POTION, 1)
+                    )
+                )
+            )
+        }
+    }
+}
\ No newline at end of file
diff --git a/src/main/kotlin/moe/nea/ledger/modules/GodPotionMixinDetection.kt b/src/main/kotlin/moe/nea/ledger/modules/GodPotionMixinDetection.kt
new file mode 100644
index 0000000..b96a24a
--- /dev/null
+++ b/src/main/kotlin/moe/nea/ledger/modules/GodPotionMixinDetection.kt
@@ -0,0 +1,38 @@
+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.ChatReceived
+import moe.nea.ledger.useMatcher
+import moe.nea.ledger.utils.di.Inject
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+
+class GodPotionMixinDetection {
+
+    val godPotionMixinDrank = "SCHLURP! The (effects of the )?(?<what>.*?) (grants you effects|have been extended by) .*! They will pause if your God Potion expires.".toPattern()
+
+    @Inject
+    lateinit var logger: LedgerLogger
+
+    @Inject
+    lateinit var itemIdProvider: ItemIdProvider
+
+    @SubscribeEvent
+    fun onChat(event: ChatReceived) {
+        godPotionMixinDrank.useMatcher(event.message) {
+            logger.logEntry(
+                LedgerEntry(
+                    TransactionType.GOD_POTION_MIXIN_DRANK,
+                    event.timestamp,
+                    listOf(
+                        ItemChange.lose(itemIdProvider.findForName(group("what")) ?: ItemId.NIL, 1)
+                    )
+                )
+            )
+        }
+    }
+}
\ No newline at end of file