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

2 files changed, 45 insertions, 1 deletions

diff --git a/src/main/kotlin/moe/nea/ledger/Ledger.kt b/src/main/kotlin/moe/nea/ledger/Ledger.kt
index b4fe2c4..23d1265 100644
--- a/src/main/kotlin/moe/nea/ledger/Ledger.kt
+++ b/src/main/kotlin/moe/nea/ledger/Ledger.kt
@@ -46,7 +46,12 @@ class Ledger {
 	You purchased 2x Walnut for 69 coins!
 	You purchased ◆ Ice Rune I for 4,000 coins!
 
-	TODO: TRADING, FORGE, COOKIE_EATEN, NPC_SELL, NPC_BUY
+	// NPC
+
+	// You bought Cactus x32 for 465.6 Coins!
+	// You sold Cactus x1 for 3 Coins!
+
+	TODO: TRADING, FORGE, COOKIE_EATEN
 	*/
 	companion object {
 		val dataFolder = File("money-ledger").apply { mkdirs() }
@@ -104,6 +109,7 @@ class Ledger {
 			BitsDetection(ledger),
 			BitsShop(ledger),
 			MinionDetection(ledger),
+			NpcDetection(ledger, ids),
 		).forEach(MinecraftForge.EVENT_BUS::register)
 	}
 
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,
+                )
+            )
+        }
+    }
+}