aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/moe/nea/ledger/modules/NpcDetection.kt
blob: 95b8aa550411b0c13e48489422ff0d19f9c43715 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
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.SHORT_NUMBER_PATTERN
import moe.nea.ledger.TransactionType
import moe.nea.ledger.asIterable
import moe.nea.ledger.events.BeforeGuiAction
import moe.nea.ledger.events.ChatReceived
import moe.nea.ledger.events.ExtraSupplyIdEvent
import moe.nea.ledger.getDisplayNameU
import moe.nea.ledger.getInternalId
import moe.nea.ledger.getLore
import moe.nea.ledger.parseShortNumber
import moe.nea.ledger.unformattedString
import moe.nea.ledger.useMatcher
import moe.nea.ledger.utils.ErrorUtil
import moe.nea.ledger.utils.di.Inject
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
import java.util.regex.Pattern

class NpcDetection @Inject constructor(val ledger: LedgerLogger, val ids: ItemIdProvider) {

	val npcBuyPattern =
		Pattern.compile("You bought (back )?(?<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?!")

	// You bought InfiniDirt™ Wand!
	// You bought Prismapump x4!
	val npcBuyWithItemPattern =
		"You bought (?<what>.*?)!".toPattern()
	var storedPurchases = mutableMapOf<String, List<ItemChange>>()

	@SubscribeEvent
	fun onClick(event: BeforeGuiAction) {
		(event.chestSlots?.lowerChestInventory?.asIterable() ?: listOf())
			.filterNotNull().forEach {
				val name = it.getDisplayNameU().unformattedString()
				val id = it.getInternalId() ?: return@forEach
				val count = it.stackSize
				val cost = ids.findCostItemsFromSpan(it.getLore())
				storedPurchases[name] = listOf(ItemChange.gain(id, count)) + cost.map { ItemChange.unpairLose(it) }
			}
	}

	@SubscribeEvent
	fun addChocolate(event: ExtraSupplyIdEvent) {
		event.store("Chocolate", ItemId("SKYBLOCK_CHOCOLATE"))
	}

	@Inject
	lateinit var errorUtil: ErrorUtil

	@SubscribeEvent
	fun onBarteredItemBought(event: ChatReceived) {
		npcBuyWithItemPattern.useMatcher(event.message) {
			val changes = storedPurchases[group("what")]
			if (changes == null) {
				errorUtil.reportAdHoc("Item bought for items without associated cost")
			}
			storedPurchases.clear()
			ledger.logEntry(
				LedgerEntry(
					TransactionType.NPC_BUY,
					event.timestamp,
					changes ?: listOf()
				)
			)
		}
	}

	@SubscribeEvent
	fun onNpcBuy(event: ChatReceived) {
		npcBuyPattern.useMatcher(event.message) {
			ledger.logEntry(
				LedgerEntry(
					TransactionType.NPC_BUY,
					event.timestamp,
					listOf(
						ItemChange.loseCoins(
							parseShortNumber(group("coins")),
						),
						ItemChange.gain(
							ids.findForName(group("what")) ?: ItemId.NIL,
							group("count")?.let(::parseShortNumber) ?: 1,
						)
					)
				)
			)
		}
		npcSellPattern.useMatcher(event.message) {
			ledger.logEntry(
				LedgerEntry(
					TransactionType.NPC_SELL,
					event.timestamp,
					listOf(
						ItemChange.gainCoins(parseShortNumber(group("coins"))),
						ItemChange.lose(
							ids.findForName(group("what")) ?: ItemId.NIL,
							group("count")?.let(::parseShortNumber)?.toInt() ?: 1,
						)
					)
				)
			)
		}
	}
}