blob: ccde6f0893482facbc8f23621fc9882a6822bf9f (
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
|
package moe.ymc.acron.mixin;
import com.mojang.brigadier.CommandDispatcher;
import moe.ymc.acron.cmd.CmdSrc;
import moe.ymc.acron.config.ConfigReloadCmd;
import net.minecraft.server.command.CommandManager;
import net.minecraft.server.command.ServerCommandSource;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.spongepowered.asm.mixin.Final;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
import static net.minecraft.server.command.CommandManager.literal;
@Mixin(CommandManager.class)
public abstract class CommandManagerMixin {
private static final Logger AC_LOGGER = LogManager.getLogger();
@Shadow
@Final
private CommandDispatcher<ServerCommandSource> dispatcher;
@Inject(method = "<init>", at = @At("RETURN"))
private void onRegister(CommandManager.RegistrationEnvironment arg, CallbackInfo ci) {
AC_LOGGER.debug("onRegister");
dispatcher.register(
literal("acron").requires(player -> player.hasPermissionLevel(4)).then(
literal("rule").then(
literal("update")
.executes(new ConfigReloadCmd()))
)
);
}
@Inject(at = @At(value = "RETURN"),
method = "execute")
// Unfortunately we cannot inject into the catch blocks, so
// a cursed approach would be always sending the result if we
// haven't.
public void execute(ServerCommandSource commandSource,
String command,
CallbackInfoReturnable<Integer> cir) {
if (commandSource instanceof CmdSrc src) {
src.sendResultIfNot();
}
}
}
|