aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/features/chat/QuickCommands.kt
blob: 7963171cd81e44caf5d54844b59ed44e1acc0b14 (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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
package moe.nea.firmament.features.chat

import com.mojang.brigadier.CommandDispatcher
import com.mojang.brigadier.context.CommandContext
import net.fabricmc.fabric.api.client.command.v2.ClientCommandRegistrationCallback
import net.fabricmc.fabric.api.client.command.v2.FabricClientCommandSource
import net.fabricmc.fabric.impl.command.client.ClientCommandInternals
import net.minecraft.command.CommandRegistryAccess
import net.minecraft.network.packet.s2c.play.CommandTreeS2CPacket
import net.minecraft.text.Text
import moe.nea.firmament.annotations.Subscribe
import moe.nea.firmament.commands.DefaultSource
import moe.nea.firmament.commands.RestArgumentType
import moe.nea.firmament.commands.get
import moe.nea.firmament.commands.thenArgument
import moe.nea.firmament.commands.thenExecute
import moe.nea.firmament.events.CommandEvent
import moe.nea.firmament.features.FirmamentFeature
import moe.nea.firmament.gui.config.ManagedConfig
import moe.nea.firmament.gui.config.ManagedOption
import moe.nea.firmament.util.MC
import moe.nea.firmament.util.SBData
import moe.nea.firmament.util.grey
import moe.nea.firmament.util.tr

object QuickCommands : FirmamentFeature {
	override val identifier: String
		get() = "quick-commands"

	object TConfig : ManagedConfig("quick-commands", Category.CHAT) {
		val enableJoin by toggle("join") { true }
		val enableDh by toggle("dh") { true }
		override fun onChange(option: ManagedOption<*>) {
			reloadCommands()
		}
	}

	fun reloadCommands() {
		val lastPacket = lastReceivedTreePacket ?: return
		val network = MC.networkHandler ?: return
		val fallback = ClientCommandInternals.getActiveDispatcher()
		try {
			val dispatcher = CommandDispatcher<FabricClientCommandSource>()
			ClientCommandInternals.setActiveDispatcher(dispatcher)
			ClientCommandRegistrationCallback.EVENT.invoker()
				.register(dispatcher, CommandRegistryAccess.of(network.combinedDynamicRegistries,
				                                               network.enabledFeatures))
			ClientCommandInternals.finalizeInit()
			network.onCommandTree(lastPacket)
		} catch (ex: Exception) {
			ClientCommandInternals.setActiveDispatcher(fallback)
			throw ex
		}
	}


	fun removePartialPrefix(text: String, prefix: String): String? {
		var lf: String? = null
		for (i in 1..prefix.length) {
			if (text.startsWith(prefix.substring(0, i))) {
				lf = text.substring(i)
			}
		}
		return lf
	}

	var lastReceivedTreePacket: CommandTreeS2CPacket? = null

	val kuudraLevelNames = listOf("NORMAL", "HOT", "BURNING", "FIERY", "INFERNAL")
	val dungeonLevelNames = listOf("ONE", "TWO", "THREE", "FOUR", "FIVE", "SIX", "SEVEN")

	@Subscribe
	fun registerDh(event: CommandEvent) {
		if (!TConfig.enableDh) return
		event.register("dh") {
			thenExecute {
				MC.sendCommand("warp dhub")
			}
		}
		event.register("dn") {
			thenExecute {
				MC.sendChat(tr("firmament.quickwarp.deez-nutz", "Warping to... Deez Nuts!").grey())
				MC.sendCommand("warp dhub")
			}
		}
	}

	@Subscribe
	fun registerJoin(it: CommandEvent) {
		if (!TConfig.enableJoin) return
		it.register("join") {
			thenArgument("what", RestArgumentType) { what ->
				thenExecute {
					val what = this[what]
					if (!SBData.isOnSkyblock) {
						MC.sendCommand("join $what")
						return@thenExecute
					}
					val joinName = getNameForFloor(what.replace(" ", "").lowercase())
					if (joinName == null) {
						source.sendFeedback(Text.stringifiedTranslatable("firmament.quick-commands.join.unknown", what))
					} else {
						source.sendFeedback(Text.stringifiedTranslatable("firmament.quick-commands.join.success",
						                                                 joinName))
						MC.sendCommand("joininstance $joinName")
					}
				}
			}
			thenExecute {
				source.sendFeedback(Text.translatable("firmament.quick-commands.join.explain"))
			}
		}
	}

	fun CommandContext<DefaultSource>.getNameForFloor(w: String): String? {
		val kuudraLevel = removePartialPrefix(w, "kuudratier") ?: removePartialPrefix(w, "tier")
		if (kuudraLevel != null) {
			val l = kuudraLevel.toIntOrNull()?.let { it - 1 } ?: kuudraLevelNames.indexOfFirst {
				it.startsWith(
					kuudraLevel,
					true
				)
			}
			if (l !in kuudraLevelNames.indices) {
				source.sendFeedback(Text.stringifiedTranslatable("firmament.quick-commands.join.unknown-kuudra",
				                                                 kuudraLevel))
				return null
			}
			return "KUUDRA_${kuudraLevelNames[l]}"
		}
		val masterLevel = removePartialPrefix(w, "master")
		val normalLevel =
			removePartialPrefix(w, "floor") ?: removePartialPrefix(w, "catacombs") ?: removePartialPrefix(w, "dungeons")
		val dungeonLevel = masterLevel ?: normalLevel
		if (dungeonLevel != null) {
			val l = dungeonLevel.toIntOrNull()?.let { it - 1 } ?: dungeonLevelNames.indexOfFirst {
				it.startsWith(
					dungeonLevel,
					true
				)
			}
			if (masterLevel == null && (l == -1 || null != removePartialPrefix(w, "entrance"))) {
				return "CATACOMBS_ENTRANCE"
			}
			if (l !in dungeonLevelNames.indices) {
				source.sendFeedback(Text.stringifiedTranslatable("firmament.quick-commands.join.unknown-catacombs",
				                                                 kuudraLevel))
				return null
			}
			return "${if (masterLevel != null) "MASTER_" else ""}CATACOMBS_FLOOR_${dungeonLevelNames[l]}"
		}
		return null
	}
}