Merge https://github.com/nea89o/LocalTransactionLedger/pull/2

1 files changed, 38 insertions, 0 deletions

diff --git a/src/main/kotlin/moe/nea/ledger/NpcDetection.kt b/src/main/kotlin/moe/nea/ledger/NpcDetection.kt
new file mode 100644
index 0000000..65693ef
--- /dev/null
+++ b/src/main/kotlin/moe/nea/ledger/NpcDetection.kt
@@ -0,0 +1,38 @@
+package moe.nea.ledger
+
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+import java.util.regex.Pattern
+
+class NpcDetection(val ledger: LedgerLogger, val ids: ItemIdProvider) {
+
+    val npcBuyPattern =
+        Pattern.compile("You bought (?<what>.*?) (x(?<count>$SHORT_NUMBER_PATTERN) )?for (?<coins>$SHORT_NUMBER_PATTERN) Coins!")
+    val npcSellPattern =
+        Pattern.compile("You sold (?<what>.*) (x(?<count>$SHORT_NUMBER_PATTERN) )?for (?<coins>$SHORT_NUMBER_PATTERN) Coins!")
+
+    @SubscribeEvent
+    fun onNpcBuy(event: ChatReceived) {
+        npcBuyPattern.useMatcher(event.message) {
+            ledger.logEntry(
+                LedgerEntry(
+                    "NPC_BUY",
+                    event.timestamp,
+                    parseShortNumber(group("coins")),
+                    ids.findForName(group("what")),
+                    group("count")?.let(::parseShortNumber)?.toInt() ?: 1,
+                )
+            )
+        }
+        npcSellPattern.useMatcher(event.message) {
+            ledger.logEntry(
+                LedgerEntry(
+                    "NPC_SELL",
+                    event.timestamp,
+                    parseShortNumber(group("coins")),
+                    ids.findForName(group("what")),
+                    group("count")?.let(::parseShortNumber)?.toInt() ?: 1,
+                )
+            )
+        }
+    }
+}