Implement commands

2025-11-09 13:49:28 +01:00
parent d655f6a25d
commit 28a0327363
4 changed files with 130 additions and 13 deletions
@@ -0,0 +1,107 @@
+package net.fretux.ascend.command;
+
+import com.mojang.brigadier.CommandDispatcher;
+import com.mojang.brigadier.arguments.IntegerArgumentType;
+import com.mojang.brigadier.arguments.StringArgumentType;
+import com.mojang.brigadier.context.CommandContext;
+import com.mojang.brigadier.exceptions.CommandSyntaxException;
+import net.fretux.ascend.player.PlayerStats;
+import net.fretux.ascend.player.PlayerStatsProvider;
+import net.fretux.ascend.player.StatEffects;
+import net.fretux.ascend.network.PacketHandler;
+import net.fretux.ascend.network.ClientboundSyncStatsPacket;
+import net.minecraft.commands.CommandSourceStack;
+import net.minecraft.commands.Commands;
+import net.minecraft.commands.arguments.EntityArgument;
+import net.minecraft.nbt.CompoundTag;
+import net.minecraft.network.chat.Component;
+import net.minecraft.server.level.ServerPlayer;
+import net.minecraftforge.network.PacketDistributor;
+
+public class AscendCommand {
+    public static void register(CommandDispatcher<CommandSourceStack> dispatcher) {
+        dispatcher.register(
+                Commands.literal("ascend")
+                        .requires(src -> src.hasPermission(2))
+                        .then(Commands.literal("get")
+                                .then(Commands.argument("player", EntityArgument.player())
+                                        .executes(AscendCommand::getStats)))
+                        .then(Commands.literal("set")
+                                .then(Commands.argument("player", EntityArgument.player())
+                                        .then(Commands.argument("attribute", StringArgumentType.string())
+                                                .then(Commands.argument("value", IntegerArgumentType.integer(0))
+                                                        .executes(AscendCommand::setAttribute)))))
+                        .then(Commands.literal("addxp")
+                                .then(Commands.argument("player", EntityArgument.player())
+                                        .then(Commands.argument("amount", IntegerArgumentType.integer(1))
+                                                .executes(AscendCommand::addXP))))
+                        .then(Commands.literal("reset")
+                                .then(Commands.argument("player", EntityArgument.player())
+                                        .executes(AscendCommand::resetStats)))
+        );
+    }
+
+    private static int getStats(CommandContext<CommandSourceStack> ctx) throws CommandSyntaxException {
+        ServerPlayer target = EntityArgument.getPlayer(ctx, "player");
+        target.getCapability(PlayerStatsProvider.PLAYER_STATS).ifPresent(stats -> {
+            ctx.getSource().sendSuccess(() -> Component.literal("=== Ascend Stats for " + target.getName().getString() + " ==="), false);
+            ctx.getSource().sendSuccess(() ->
+                    Component.literal("Level: " + stats.getAscendLevel() + " | XP: " + stats.getAscendXP() + "/" + stats.getXPToNextAscendLevel()), false);
+            ctx.getSource().sendSuccess(() ->
+                    Component.literal("Unspent Points: " + stats.getUnspentPoints()), false);
+            CompoundTag attrTag = stats.serializeNBT().getCompound("Attributes");
+            for (String key : attrTag.getAllKeys()) {
+                int val = attrTag.getInt(key);
+                ctx.getSource().sendSuccess(() -> Component.literal(" - " + key + ": " + val), false);
+            }
+        });
+        return 1;
+    }
+
+    private static int setAttribute(CommandContext<CommandSourceStack> ctx) throws CommandSyntaxException {
+        ServerPlayer target = EntityArgument.getPlayer(ctx, "player");
+        String attr = StringArgumentType.getString(ctx, "attribute");
+        int value = IntegerArgumentType.getInteger(ctx, "value");
+        target.getCapability(PlayerStatsProvider.PLAYER_STATS).ifPresent(stats -> {
+            stats.setAttributeLevel(attr, value);
+            StatEffects.applyAll(target);
+            sync(target, stats);
+            ctx.getSource().sendSuccess(() ->
+                    Component.literal("Set " + target.getName().getString() + "'s " + attr + " to " + value), true);
+        });
+        return 1;
+    }
+
+    private static int addXP(CommandContext<CommandSourceStack> ctx) throws CommandSyntaxException {
+        ServerPlayer target = EntityArgument.getPlayer(ctx, "player");
+        int amount = IntegerArgumentType.getInteger(ctx, "amount");
+        target.getCapability(PlayerStatsProvider.PLAYER_STATS).ifPresent(stats -> {
+            stats.addAscendXP(amount);
+            StatEffects.applyAll(target);
+            sync(target, stats);
+            ctx.getSource().sendSuccess(() ->
+                    Component.literal("Added " + amount + " Ascend XP to " + target.getName().getString()), true);
+        });
+        return 1;
+    }
+
+    private static int resetStats(CommandContext<CommandSourceStack> ctx) throws CommandSyntaxException {
+        ServerPlayer target = EntityArgument.getPlayer(ctx, "player");
+        target.getCapability(PlayerStatsProvider.PLAYER_STATS).ifPresent(stats -> {
+            PlayerStats fresh = new PlayerStats();
+            stats.deserializeNBT(fresh.serializeNBT());
+            StatEffects.applyAll(target);
+            sync(target, stats);
+            ctx.getSource().sendSuccess(() ->
+                    Component.literal("Reset " + target.getName().getString() + "'s Ascend stats."), true);
+        });
+        return 1;
+    }
+
+    private static void sync(ServerPlayer player, PlayerStats stats) {
+        PacketHandler.INSTANCE.send(
+                PacketDistributor.PLAYER.with(() -> player),
+                new ClientboundSyncStatsPacket(stats.serializeNBT())
+        );
+    }
+}