hytale-server/com/hypixel/hytale/server/npc/commands/NPCDebugCommand.java

package com.hypixel.hytale.server.npc.commands;

import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.server.core.Message;
import com.hypixel.hytale.server.core.command.system.AbstractCommand;
import com.hypixel.hytale.server.core.command.system.CommandContext;
import com.hypixel.hytale.server.core.command.system.arguments.system.OptionalArg;
import com.hypixel.hytale.server.core.command.system.arguments.system.RequiredArg;
import com.hypixel.hytale.server.core.command.system.arguments.types.ArgTypes;
import com.hypixel.hytale.server.core.command.system.basecommands.AbstractCommandCollection;
import com.hypixel.hytale.server.core.entity.nameplate.Nameplate;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.npc.entities.NPCEntity;
import com.hypixel.hytale.server.npc.role.RoleDebugFlags;
import java.util.EnumSet;
import java.util.concurrent.CompletableFuture;
import java.util.function.BiFunction;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class NPCDebugCommand extends AbstractCommandCollection {
   public NPCDebugCommand() {
      super("debug", "server.commands.npc.debug.desc");
      this.addSubCommand(new NPCDebugCommand.ShowCommand());
      this.addSubCommand(new NPCDebugCommand.SetCommand());
      this.addSubCommand(new NPCDebugCommand.ToggleCommand());
      this.addSubCommand(new NPCDebugCommand.DefaultsCommand());
      this.addSubCommand(new NPCDebugCommand.ClearCommand());
      this.addSubCommand(new NPCDebugCommand.PresetsCommand());
   }

   private static void modifyFlags(
      @Nonnull CommandContext context,
      @Nonnull NPCEntity npc,
      @Nonnull Ref<EntityStore> ref,
      @Nonnull EnumSet<RoleDebugFlags> flags,
      @Nonnull BiFunction<EnumSet<RoleDebugFlags>, EnumSet<RoleDebugFlags>, EnumSet<RoleDebugFlags>> flagsModifier,
      @Nonnull Store<EntityStore> store
   ) {
      EnumSet<RoleDebugFlags> newFlags = flagsModifier.apply(npc.getRoleDebugFlags(), flags);
      if (newFlags != null) {
         safeSetRoleDebugFlags(npc, ref, newFlags, store);
         printNewFlags(npc, context, newFlags);
      }
   }

   private static void safeSetRoleDebugFlags(
      @Nonnull NPCEntity npc, @Nonnull Ref<EntityStore> ref, @Nonnull EnumSet<RoleDebugFlags> flags, @Nonnull Store<EntityStore> store
   ) {
      store.tryRemoveComponent(ref, Nameplate.getComponentType());
      npc.setRoleDebugFlags(flags);
   }

   private static void printNewFlags(@Nonnull NPCEntity npc, @Nonnull CommandContext context, @Nonnull EnumSet<RoleDebugFlags> newFlags) {
      String flags = getListOfFlags(newFlags).toString();
      context.sendMessage(
         Message.translation("server.commands.npc.debug.debugFlagsSet").param("role", npc.getRoleName()).param("flags", !flags.isEmpty() ? flags : "<None>")
      );
   }

   @Nonnull
   private static StringBuilder getListOfFlags(@Nonnull EnumSet<RoleDebugFlags> flags) {
      return RoleDebugFlags.getListOfFlags(flags, new StringBuilder());
   }

   public static class ClearCommand extends NPCMultiSelectCommandBase {
      public ClearCommand() {
         super("clear", "server.commands.npc.debug.clear.desc");
      }

      @Override
      protected void execute(
         @Nonnull CommandContext context, @Nonnull NPCEntity npc, @Nonnull World world, @Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> ref
      ) {
         NPCDebugCommand.safeSetRoleDebugFlags(npc, ref, EnumSet.noneOf(RoleDebugFlags.class), store);
      }
   }

   public static class DefaultsCommand extends NPCMultiSelectCommandBase {
      public DefaultsCommand() {
         super("defaults", "server.commands.npc.debug.defaults.desc");
      }

      @Override
      protected void execute(
         @Nonnull CommandContext context, @Nonnull NPCEntity npc, @Nonnull World world, @Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> ref
      ) {
         NPCDebugCommand.safeSetRoleDebugFlags(npc, ref, RoleDebugFlags.getPreset("default"), store);
      }
   }

   public static class PresetsCommand extends AbstractCommand {
      @Nonnull
      private final OptionalArg<String> presetArg = this.withOptionalArg("preset", "server.commands.npc.debug.presets.preset.desc", ArgTypes.STRING);

      public PresetsCommand() {
         super("presets", "server.commands.npc.debug.presets.desc");
      }

      @Nullable
      @Override
      protected CompletableFuture<Void> execute(@Nonnull CommandContext context) {
         if (this.presetArg.provided(context)) {
            String presetName = this.presetArg.get(context);
            if (!presetName.isEmpty() && RoleDebugFlags.havePreset(presetName)) {
               EnumSet<RoleDebugFlags> flags = RoleDebugFlags.getPreset(presetName);
               String flagString = NPCDebugCommand.getListOfFlags(flags).toString();
               context.sendMessage(
                  Message.translation("server.commands.npc.debug.preset.info")
                     .param("preset", presetName)
                     .param("flags", !flagString.isEmpty() ? flagString : "<None>")
               );
               return null;
            } else {
               context.sendMessage(Message.translation("server.commands.errors.npc.unknown_debug_preset").param("preset", presetName));
               return null;
            }
         } else {
            String flags = RoleDebugFlags.getListOfAllFlags(new StringBuilder()).toString();
            String presets = RoleDebugFlags.getListOfAllPresets(new StringBuilder()).toString();
            Message message = Message.translation("server.commands.npc.debug.presets.info").param("flags", flags).param("presets", presets);
            context.sendMessage(message);
            return null;
         }
      }
   }

   public static class SetCommand extends NPCMultiSelectCommandBase {
      @Nonnull
      private final RequiredArg<String> flagsArg = this.withRequiredArg("flags", "server.commands.npc.debug.flags.desc", ArgTypes.STRING);

      public SetCommand() {
         super("set", "server.commands.npc.debug.set.desc");
      }

      @Override
      protected void execute(
         @Nonnull CommandContext context, @Nonnull NPCEntity npc, @Nonnull World world, @Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> ref
      ) {
         String flagsString = this.flagsArg.get(context);
         EnumSet<RoleDebugFlags> flags = RoleDebugFlags.getFlags(flagsString.split(","));
         NPCDebugCommand.modifyFlags(context, npc, ref, flags, (oldFlags, argFlags) -> argFlags, store);
      }
   }

   public static class ShowCommand extends NPCMultiSelectCommandBase {
      public ShowCommand() {
         super("show", "server.commands.npc.debug.show.desc");
      }

      @Override
      protected void execute(
         @Nonnull CommandContext context, @Nonnull NPCEntity npc, @Nonnull World world, @Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> ref
      ) {
         String flags = NPCDebugCommand.getListOfFlags(npc.getRoleDebugFlags()).toString();
         context.sendMessage(
            Message.translation("server.commands.npc.debug.currentFlags").param("role", npc.getRoleName()).param("flags", !flags.isEmpty() ? flags : "<None>")
         );
      }
   }

   public static class ToggleCommand extends NPCMultiSelectCommandBase {
      @Nonnull
      private final RequiredArg<String> flagsArg = this.withRequiredArg("flags", "server.commands.npc.debug.flags.desc", ArgTypes.STRING);

      public ToggleCommand() {
         super("toggle", "server.commands.npc.debug.toggle.desc");
      }

      @Override
      protected void execute(
         @Nonnull CommandContext context, @Nonnull NPCEntity npc, @Nonnull World world, @Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> ref
      ) {
         String flagsString = this.flagsArg.get(context);
         EnumSet<RoleDebugFlags> flags = RoleDebugFlags.getFlags(flagsString.split(","));
         NPCDebugCommand.modifyFlags(context, npc, ref, flags, (oldFlags, argFlags) -> {
            if (argFlags.isEmpty()) {
               return null;
            } else {
               EnumSet<RoleDebugFlags> newFlags = oldFlags.clone();

               for (RoleDebugFlags flag : argFlags) {
                  if (newFlags.contains(flag)) {
                     newFlags.remove(flag);
                  } else {
                     newFlags.add(flag);
                  }
               }

               return newFlags;
            }
         }, store);
      }
   }
}