feat: Add SQLITE database entry logging

1 files changed, 41 insertions, 30 deletions

diff --git a/src/main/kotlin/moe/nea/ledger/modules/BazaarDetection.kt b/src/main/kotlin/moe/nea/ledger/modules/BazaarDetection.kt
index 01c8bbc..522beed 100644
--- a/src/main/kotlin/moe/nea/ledger/modules/BazaarDetection.kt
+++ b/src/main/kotlin/moe/nea/ledger/modules/BazaarDetection.kt
@@ -1,10 +1,13 @@
 package moe.nea.ledger.modules
 
-import moe.nea.ledger.events.ChatReceived
+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.SHORT_NUMBER_PATTERN
+import moe.nea.ledger.TransactionType
+import moe.nea.ledger.events.ChatReceived
 import moe.nea.ledger.parseShortNumber
 import moe.nea.ledger.useMatcher
 import moe.nea.ledger.utils.Inject
@@ -13,35 +16,43 @@ import java.util.regex.Pattern
 
 class BazaarDetection @Inject constructor(val ledger: LedgerLogger, val ids: ItemIdProvider) {
 
-    val instaBuyPattern =
-        Pattern.compile("\\[Bazaar\\] Bought (?<count>$SHORT_NUMBER_PATTERN)x (?<what>.*) for (?<coins>$SHORT_NUMBER_PATTERN) coins!")
-    val instaSellPattern =
-        Pattern.compile("\\[Bazaar\\] Sold (?<count>$SHORT_NUMBER_PATTERN)x (?<what>.*) for (?<coins>$SHORT_NUMBER_PATTERN) coins!")
+	val instaBuyPattern =
+		Pattern.compile("\\[Bazaar\\] Bought (?<count>$SHORT_NUMBER_PATTERN)x (?<what>.*) for (?<coins>$SHORT_NUMBER_PATTERN) coins!")
+	val instaSellPattern =
+		Pattern.compile("\\[Bazaar\\] Sold (?<count>$SHORT_NUMBER_PATTERN)x (?<what>.*) for (?<coins>$SHORT_NUMBER_PATTERN) coins!")
 
 
-    @SubscribeEvent
-    fun onInstSellChat(event: ChatReceived) {
-        instaBuyPattern.useMatcher(event.message) {
-            ledger.logEntry(
-                LedgerEntry(
-	                "BAZAAR_BUY_INSTANT",
-	                event.timestamp,
-	                parseShortNumber(group("coins")),
-	                ids.findForName(group("what")),
-	                parseShortNumber(group("count")).toInt(),
-                )
-            )
-        }
-        instaSellPattern.useMatcher(event.message) {
-            ledger.logEntry(
-                LedgerEntry(
-	                "BAZAAR_SELL_INSTANT",
-	                event.timestamp,
-	                parseShortNumber(group("coins")),
-	                ids.findForName(group("what")),
-	                parseShortNumber(group("count")).toInt(),
-                )
-            )
-        }
-    }
+	@SubscribeEvent
+	fun onInstSellChat(event: ChatReceived) {
+		instaBuyPattern.useMatcher(event.message) {
+			ledger.logEntry(
+				LedgerEntry(
+					TransactionType.BAZAAR_BUY_INSTANT,
+					event.timestamp,
+					listOf(
+						ItemChange.loseCoins(parseShortNumber(group("coins"))),
+						ItemChange.gain(
+							ids.findForName(group("what")) ?: ItemId.NIL,
+							parseShortNumber(group("count"))
+						)
+					)
+				)
+			)
+		}
+		instaSellPattern.useMatcher(event.message) {
+			ledger.logEntry(
+				LedgerEntry(
+					TransactionType.BAZAAR_SELL_INSTANT,
+					event.timestamp,
+					listOf(
+						ItemChange.gainCoins(parseShortNumber(group("coins"))),
+						ItemChange.lose(
+							ids.findForName(group("what")) ?: ItemId.NIL,
+							parseShortNumber(group("count"))
+						)
+					),
+				)
+			)
+		}
+	}
 }