feat: added /ftbteams nbtedit command

Direct viewing and editing of team NBT, use with caution...

Author: desht

common/src/main/java/dev/ftb/mods/ftbteams/FTBTeams.java

@@ -8,18 +8,23 @@
 import dev.architectury.event.events.common.PlayerEvent;
 import dev.architectury.utils.Env;
 import dev.architectury.utils.EnvExecutor;
+import dev.ftb.mods.ftblibrary.nbtedit.NBTEditResponseHandlers;
 import dev.ftb.mods.ftbteams.api.FTBTeamsAPI;
+import dev.ftb.mods.ftbteams.api.Team;
 import dev.ftb.mods.ftbteams.api.event.TeamCollectPropertiesEvent;
 import dev.ftb.mods.ftbteams.api.event.TeamEvent;
 import dev.ftb.mods.ftbteams.api.event.TeamManagerEvent;
 import dev.ftb.mods.ftbteams.api.property.TeamProperties;
 import dev.ftb.mods.ftbteams.client.FTBTeamsClient;
+import dev.ftb.mods.ftbteams.data.AbstractTeam;
 import dev.ftb.mods.ftbteams.data.FTBTeamsCommands;
 import dev.ftb.mods.ftbteams.data.TeamManagerImpl;
+import dev.ftb.mods.ftbteams.data.TeamType;
 import dev.ftb.mods.ftbteams.net.FTBTeamsNet;
 import net.minecraft.commands.CommandBuildContext;
 import net.minecraft.commands.CommandSourceStack;
 import net.minecraft.commands.Commands;
+import net.minecraft.nbt.CompoundTag;
 import net.minecraft.network.chat.Component;
 import net.minecraft.server.MinecraftServer;
 import net.minecraft.server.level.ServerLevel;
@@ -35,6 +40,7 @@ public FTBTeams() {
 		FTBTeamsAPI._init(FTBTeamsAPIImpl.INSTANCE);
 
 		LifecycleEvent.SERVER_BEFORE_START.register(this::serverAboutToStart);
+		LifecycleEvent.SERVER_STARTED.register(this::serverStarted);
 		CommandRegistrationEvent.EVENT.register(this::registerCommands);
 		LifecycleEvent.SERVER_STOPPED.register(this::serverStopped);
 		LifecycleEvent.SERVER_LEVEL_SAVE.register(this::worldSaved);
@@ -48,6 +54,17 @@ public FTBTeams() {
 		FTBTeamsNet.register();
 	}
 
+	private void serverStarted(MinecraftServer server) {
+		NBTEditResponseHandlers.INSTANCE.registerHandler("ftbteams:team", (serverPlayer, info, data) -> {
+			TeamManagerImpl.INSTANCE.getTeamByID(info.getUUID("id")).ifPresent(team -> {
+				if (team instanceof AbstractTeam abstractTeam) {
+					abstractTeam.deserializeNBT(data, server.registryAccess());
+					abstractTeam.markDirty();
+				}
+			});
+        });
+	}
+
 	private void serverAboutToStart(MinecraftServer server) {
 		TeamManagerImpl.INSTANCE = new TeamManagerImpl(server);
 		TeamManagerEvent.CREATED.invoker().accept(new TeamManagerEvent(TeamManagerImpl.INSTANCE));

common/src/main/java/dev/ftb/mods/ftbteams/data/FTBTeamsCommands.java

@@ -8,6 +8,9 @@
 import com.mojang.brigadier.context.CommandContext;
 import com.mojang.brigadier.exceptions.CommandSyntaxException;
 import dev.architectury.platform.Platform;
+import dev.ftb.mods.ftblibrary.FTBLibraryCommands;
+import dev.ftb.mods.ftblibrary.net.EditNBTPacket;
+import dev.ftb.mods.ftblibrary.util.NetworkHelper;
 import dev.ftb.mods.ftbteams.FTBTeamsAPIImpl;
 import dev.ftb.mods.ftbteams.api.FTBTeamsAPI;
 import dev.ftb.mods.ftbteams.api.Team;
@@ -17,9 +20,11 @@
 import dev.ftb.mods.ftbteams.api.event.TeamInfoEvent;
 import dev.ftb.mods.ftbteams.api.property.TeamPropertyArgument;
 import net.minecraft.ChatFormatting;
+import net.minecraft.Util;
 import net.minecraft.commands.CommandSourceStack;
 import net.minecraft.commands.Commands;
 import net.minecraft.commands.arguments.GameProfileArgument;
+import net.minecraft.nbt.CompoundTag;
 import net.minecraft.network.chat.Component;
 import net.minecraft.server.level.ServerPlayer;
 
@@ -196,6 +201,13 @@ public void register(CommandDispatcher<CommandSourceStack> dispatcher) {
 										.executes(FTBTeamsCommands::forceRemovePlayers))
 						)
 				)
+				.then(Commands.literal("nbtedit")
+						.requires(requiresOPorSP())
+						.executes(this::editPlayerTeamNBT)
+						.then(createTeamArg()
+								.executes(FTBTeamsCommands::editTeamNBT)
+						)
+				)
 		);
 
 		if (Platform.isDevelopmentEnvironment()) {
@@ -359,4 +371,33 @@ private int addFakePlayer(Collection<GameProfile> profiles) {
 
 		return Command.SINGLE_SUCCESS;
 	}
+
+	private int editPlayerTeamNBT(CommandContext<CommandSourceStack> ctx) throws CommandSyntaxException {
+		return doTeamEdit(ctx, ctx.getSource().getPlayerOrException(), getTeam(ctx));
+	}
+
+	private static int editTeamNBT(CommandContext<CommandSourceStack> ctx) throws CommandSyntaxException {
+		return doTeamEdit(ctx, ctx.getSource().getPlayerOrException(), TeamArgument.get(ctx, "team"));
+	}
+
+	private static int doTeamEdit(CommandContext<CommandSourceStack> ctx, ServerPlayer editor, Team team) {
+		if (team instanceof AbstractTeam abstractTeam) {
+			CompoundTag info = Util.make(new CompoundTag(), t -> {
+				t.putString("title", Component.Serializer.toJson(abstractTeam.getColoredName(), editor.registryAccess()));
+				t.putString("type", "ftbteams:team");
+				t.putUUID("id", team.getTeamId());
+				t.putString("team_type", abstractTeam.getType().getSerializedName());
+				t.put("text", FTBLibraryCommands.InfoBuilder.create(ctx)
+						.add("Team Type", Component.translatable(team.getTypeTranslationKey()))
+						.add("Owner", Component.literal(team.getOwner().toString()))
+						.add("Members", Component.literal(String.valueOf(team.getMembers().size())))
+						.build()
+				);
+			});
+			CompoundTag tag = abstractTeam.serializeNBT(ctx.getSource().getServer().registryAccess());
+			NetworkHelper.sendTo(editor, new EditNBTPacket(info, tag));
+			return Command.SINGLE_SUCCESS;
+		}
+		return 0;
+	}
 }