class="kw">package com.hypixel.hytale.server.core.command.commands.debug.component.repulsion;

class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.server.core.Message;
class="kw">import com.hypixel.hytale.server.core.command.system.CommandContext;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.system.RequiredArg;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.types.ArgTypes;
class="kw">import com.hypixel.hytale.server.core.command.system.arguments.types.EntityWrappedArg;
class="kw">import com.hypixel.hytale.server.core.command.system.basecommands.AbstractCommandCollection;
class="kw">import com.hypixel.hytale.server.core.command.system.basecommands.AbstractTargetPlayerCommand;
class="kw">import com.hypixel.hytale.server.core.command.system.basecommands.AbstractWorldCommand;
class="kw">import com.hypixel.hytale.server.core.modules.entity.repulsion.Repulsion;
class="kw">import com.hypixel.hytale.server.core.modules.entity.repulsion.RepulsionConfig;
class="kw">import com.hypixel.hytale.server.core.universe.PlayerRef;
class="kw">import com.hypixel.hytale.server.core.universe.world.World;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;

class="kw">public class RepulsionAddCommand class="kw">extends AbstractCommandCollection {
   class="kw">private class="kw">static class="kw">final Message MESSAGE_COMMANDS_ERRORS_TARGET_NOT_IN_WORLD = Message.translation("server.commands.errors.targetNotInWorld");

   class="kw">public RepulsionAddCommand() {
      super("add", "server.commands.repulsion.add.desc");
      this.addSubCommand(new RepulsionAddCommand.RepulsionAddEntityCommand());
      this.addSubCommand(new RepulsionAddCommand.RepulsionAddSelfCommand());
   }

   class="kw">public class="kw">static class RepulsionAddEntityCommand class="kw">extends AbstractWorldCommand {
      @Nonnull
      class="kw">private class="kw">static class="kw">final Message MESSAGE_COMMANDS_REPULSION_ADD_ALREADY_ADDED = Message.translation("server.commands.repulsion.add.alreadyAdded");
      @Nonnull
      class="kw">private class="kw">static class="kw">final Message COMMANDS_REPULSION_ADD_SUCCESS = Message.translation("server.commands.repulsion.add.success");
      @Nonnull
      class="kw">private class="kw">final RequiredArg<RepulsionConfig> repulsionConfigArg = this.withRequiredArg(
         "repulsionConfig", "server.commands.repulsion.add.repulsionConfig.desc", ArgTypes.REPULSION_CONFIG
      );
      @Nonnull
      class="kw">private class="kw">final EntityWrappedArg entityArg = this.withRequiredArg("entity", "server.commands.repulsion.add.entityArg.desc", ArgTypes.ENTITY_ID);

      class="kw">public RepulsionAddEntityCommand() {
         super("entity", "server.commands.repulsion.add.entity.desc");
      }

      @Override
      class="kw">protected void execute(@Nonnull CommandContext var1, @Nonnull World var2, @Nonnull Store<EntityStore> var3) {
         Ref var4 = this.entityArg.get(var3, var1);
         if (var4 != null && var4.isValid()) {
            RepulsionConfig var5 = this.repulsionConfigArg.get(var1);
            if (var3.getArchetype(var4).contains(Repulsion.getComponentType())) {
               var1.sendMessage(MESSAGE_COMMANDS_REPULSION_ADD_ALREADY_ADDED);
            } else {
               var3.addComponent(var4, Repulsion.getComponentType(), new Repulsion(var5));
               var1.sendMessage(COMMANDS_REPULSION_ADD_SUCCESS);
            }
         } else {
            var1.sendMessage(RepulsionAddCommand.MESSAGE_COMMANDS_ERRORS_TARGET_NOT_IN_WORLD);
         }
      }
   }

   class="kw">public class="kw">static class RepulsionAddSelfCommand class="kw">extends AbstractTargetPlayerCommand {
      @Nonnull
      class="kw">private class="kw">static class="kw">final Message MESSAGE_COMMANDS_REPULSION_ADD_ALREADY_ADDED = Message.translation("server.commands.repulsion.add.alreadyAdded");
      @Nonnull
      class="kw">private class="kw">static class="kw">final Message MESSAGE_COMMANDS_REPULSION_ADD_SUCCESS = Message.translation("server.commands.repulsion.add.success");
      @Nonnull
      class="kw">private class="kw">final RequiredArg<RepulsionConfig> repulsionConfigArg = this.withRequiredArg(
         "repulsionConfig", "server.commands.repulsion.add.repulsionConfig.desc", ArgTypes.REPULSION_CONFIG
      );

      class="kw">public RepulsionAddSelfCommand() {
         super("self", "server.commands.repulsion.add.self.desc");
      }

      @Override
      class="kw">protected void execute(
         @Nonnull CommandContext var1,
         @Nullable Ref<EntityStore> var2,
         @Nonnull Ref<EntityStore> var3,
         @Nonnull PlayerRef var4,
         @Nonnull World var5,
         @Nonnull Store<EntityStore> var6
      ) {
         if (var6.getArchetype(var3).contains(Repulsion.getComponentType())) {
            var1.sendMessage(MESSAGE_COMMANDS_REPULSION_ADD_ALREADY_ADDED);
         } else {
            RepulsionConfig var7 = this.repulsionConfigArg.get(var1);
            var6.addComponent(var3, Repulsion.getComponentType(), new Repulsion(var7));
            var1.sendMessage(MESSAGE_COMMANDS_REPULSION_ADD_SUCCESS);
         }
      }
   }
}