CM root and eject commands

Author: robotgryphon

src/main/java/dev/compactmods/machines/command/CMCommandRoot.java

@@ -0,0 +1,15 @@
+package dev.compactmods.machines.command;
+
+import com.mojang.brigadier.CommandDispatcher;
+import com.mojang.brigadier.builder.LiteralArgumentBuilder;
+import dev.compactmods.machines.CompactMachines;
+import net.minecraft.command.CommandSource;
+
+public class CMCommandRoot {
+
+    public static void register(CommandDispatcher<CommandSource> dispatcher) {
+        final LiteralArgumentBuilder<CommandSource> root = LiteralArgumentBuilder.literal(CompactMachines.MOD_ID);
+        root.then(CMEjectSubcommand.register());
+        dispatcher.register(root);
+    }
+}

src/main/java/dev/compactmods/machines/command/CMEjectSubcommand.java

@@ -0,0 +1,42 @@
+package dev.compactmods.machines.command;
+
+import java.util.Collection;
+import com.mojang.brigadier.builder.ArgumentBuilder;
+import com.mojang.brigadier.context.CommandContext;
+import com.mojang.brigadier.exceptions.CommandSyntaxException;
+import dev.compactmods.machines.rooms.capability.CapabilityRoomHistory;
+import dev.compactmods.machines.rooms.capability.IRoomHistory;
+import dev.compactmods.machines.util.PlayerUtil;
+import net.minecraft.command.CommandSource;
+import net.minecraft.command.Commands;
+import net.minecraft.command.arguments.EntityArgument;
+import net.minecraft.entity.player.ServerPlayerEntity;
+
+public class CMEjectSubcommand {
+    public static ArgumentBuilder<CommandSource, ?> register() {
+        return Commands.literal("eject")
+                .requires(cs -> cs.hasPermission(2))
+                .executes(CMEjectSubcommand::execExecutingPlayer)
+                    .then(Commands.argument("player", EntityArgument.player())
+                    .executes(CMEjectSubcommand::execSpecificPlayer));
+    }
+
+    private static int execSpecificPlayer(CommandContext<CommandSource> ctx) throws CommandSyntaxException {
+        Collection<ServerPlayerEntity> ent = EntityArgument.getPlayers(ctx, "player");
+        ent.forEach(player -> {
+            player.getCapability(CapabilityRoomHistory.HISTORY_CAPABILITY).ifPresent(IRoomHistory::clear);
+            PlayerUtil.teleportPlayerToRespawnOrOverworld(ctx.getSource().getServer(), player);
+        });
+
+        return 0;
+    }
+
+    private static int execExecutingPlayer(CommandContext<CommandSource> ctx) throws CommandSyntaxException {
+        final ServerPlayerEntity player = ctx.getSource().getPlayerOrException();
+
+        player.getCapability(CapabilityRoomHistory.HISTORY_CAPABILITY).ifPresent(IRoomHistory::clear);
+        PlayerUtil.teleportPlayerToRespawnOrOverworld(ctx.getSource().getServer(), player);
+
+        return 0;
+    }
+}

src/main/java/dev/compactmods/machines/core/ServerEventHandler.java

@@ -1,7 +1,9 @@
 package dev.compactmods.machines.core;
 
 import dev.compactmods.machines.CompactMachines;
+import dev.compactmods.machines.command.CMCommandRoot;
 import net.minecraft.server.MinecraftServer;
+import net.minecraftforge.event.RegisterCommandsEvent;
 import net.minecraftforge.eventbus.api.SubscribeEvent;
 import net.minecraftforge.fml.common.Mod;
 import net.minecraftforge.fml.event.server.FMLServerStartingEvent;
@@ -14,4 +16,9 @@ public static void onServerStarting(final FMLServerStartingEvent evt) {
         MinecraftServer server = evt.getServer();
         // SavedMachineDataMigrator.migrate(server);
     }
+
+    @SubscribeEvent
+    public static void onCommandsRegister(final RegisterCommandsEvent event) {
+        CMCommandRoot.register(event.getDispatcher());
+    }
 }