Add bits stuff

1 files changed, 48 insertions, 0 deletions

diff --git a/src/main/kotlin/moe/nea/ledger/BitsShop.kt b/src/main/kotlin/moe/nea/ledger/BitsShop.kt
new file mode 100644
index 0000000..ea9e4d2
--- /dev/null
+++ b/src/main/kotlin/moe/nea/ledger/BitsShop.kt
@@ -0,0 +1,48 @@
+package moe.nea.ledger
+
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+import java.time.Instant
+
+class BitsShop(val ledger: LedgerLogger) {
+
+
+    data class BitShopEntry(
+        val id: String,
+        val bitPrice: Int,
+        val timestamp: Long = System.currentTimeMillis()
+    )
+
+    var lastClickedBitShopItem: BitShopEntry? = null
+    var bitCostPattern = "(?<cost>$SHORT_NUMBER_PATTERN) Bits".toPattern()
+
+    @SubscribeEvent
+    fun recordLastBitPrice(event: GuiClickEvent) {
+        val slot = event.slotIn ?: return
+        val name = slot.inventory.displayName.unformattedText.unformattedString()
+        if (name != "Community Shop" && !name.startsWith("Bits Shop"))
+            return
+        val stack = slot.stack ?: return
+        val id = stack.getInternalId() ?: return
+        val bitPrice = stack.getLore()
+            .firstNotNullOfOrNull { bitCostPattern.useMatcher(it.unformattedString()) { parseShortNumber(group("cost")).toInt() } }
+            ?: return
+        lastClickedBitShopItem = BitShopEntry(id, bitPrice)
+    }
+
+    @SubscribeEvent
+    fun onChat(event: ChatReceived) {
+        if (event.message.startsWith("You bought ")) {
+            val lastBit = lastClickedBitShopItem ?: return
+            if (System.currentTimeMillis() - lastBit.timestamp > 5000) return
+            ledger.logEntry(
+                LedgerEntry(
+                    "COMMUNITY_SHOP_BUY", Instant.now(),
+                    lastBit.bitPrice.toDouble(),
+                    lastBit.id,
+                    1
+                )
+            )
+        }
+    }
+
+}
\ No newline at end of file