AbstractTargetEntityCommand.java

package com.hypixel.hytale.server.core.command.system.basecommands;

import com.hypixel.hytale.codec.validation.Validators;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.spatial.SpatialResource;
import com.hypixel.hytale.math.vector.Vector3d;
import com.hypixel.hytale.server.core.Message;
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.types.ArgTypes;
import com.hypixel.hytale.server.core.command.system.arguments.types.EntityWrappedArg;
import com.hypixel.hytale.server.core.modules.entity.EntityModule;
import com.hypixel.hytale.server.core.modules.entity.component.TransformComponent;
import com.hypixel.hytale.server.core.universe.PlayerRef;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.core.util.TargetUtil;
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
import it.unimi.dsi.fastutil.objects.ObjectList;
import it.unimi.dsi.fastutil.objects.ObjectLists;
import java.util.concurrent.CompletableFuture;
import javax.annotation.Nonnull;

public abstract class AbstractTargetEntityCommand extends AbstractAsyncCommand {
   private static final Message MESSAGE_COMMANDS_ERRORS_PLAYER_NOT_IN_WORLD = Message.translation("server.commands.errors.playerNotInWorld");
   private static final Message MESSAGE_GENERAL_NO_ENTITY_IN_VIEW = Message.translation("server.general.noEntityInView");
   @Nonnull
   private final OptionalArg<World> worldArg;
   @Nonnull
   private final OptionalArg<Double> radiusArg;
   @Nonnull
   private final OptionalArg<PlayerRef> playerArg;
   @Nonnull
   private final EntityWrappedArg entityArg;

   public AbstractTargetEntityCommand(@Nonnull String name, @Nonnull String description) {
      super(name, description);
      this.worldArg = this.withOptionalArg("world", "server.commands.worldthread.arg.desc", ArgTypes.WORLD);
      this.radiusArg = (OptionalArg)this.withOptionalArg("radius", "server.commands.entity.radius.desc", ArgTypes.DOUBLE).addValidator(Validators.greaterThan(0.0));
      this.playerArg = this.withOptionalArg("player", "server.commands.argtype.player.desc", ArgTypes.PLAYER_REF);
      this.entityArg = (EntityWrappedArg)this.withOptionalArg("entity", "server.commands.entity.entity.desc", ArgTypes.ENTITY_ID);
   }

   public AbstractTargetEntityCommand(@Nonnull String name, @Nonnull String description, boolean requiresConfirmation) {
      super(name, description, requiresConfirmation);
      this.worldArg = this.withOptionalArg("world", "server.commands.worldthread.arg.desc", ArgTypes.WORLD);
      this.radiusArg = (OptionalArg)this.withOptionalArg("radius", "server.commands.entity.radius.desc", ArgTypes.DOUBLE).addValidator(Validators.greaterThan(0.0));
      this.playerArg = this.withOptionalArg("player", "server.commands.argtype.player.desc", ArgTypes.PLAYER_REF);
      this.entityArg = (EntityWrappedArg)this.withOptionalArg("entity", "server.commands.entity.entity.desc", ArgTypes.ENTITY_ID);
   }

   public AbstractTargetEntityCommand(@Nonnull String description) {
      super(description);
      this.worldArg = this.withOptionalArg("world", "server.commands.worldthread.arg.desc", ArgTypes.WORLD);
      this.radiusArg = (OptionalArg)this.withOptionalArg("radius", "server.commands.entity.radius.desc", ArgTypes.DOUBLE).addValidator(Validators.greaterThan(0.0));
      this.playerArg = this.withOptionalArg("player", "server.commands.argtype.player.desc", ArgTypes.PLAYER_REF);
      this.entityArg = (EntityWrappedArg)this.withOptionalArg("entity", "server.commands.entity.entity.desc", ArgTypes.ENTITY_ID);
   }

   @Nonnull
   protected final CompletableFuture<Void> executeAsync(@Nonnull CommandContext context) {
      World world;
      if (this.worldArg.provided(context)) {
         world = (World)this.worldArg.get(context);
      } else {
         if (!context.isPlayer()) {
            context.sendMessage(Message.translation("server.commands.errors.playerOrArg").param("option", "world"));
            return CompletableFuture.completedFuture((Object)null);
         }

         Ref<EntityStore> ref = context.senderAsPlayerRef();
         if (ref == null || !ref.isValid()) {
            context.sendMessage(MESSAGE_COMMANDS_ERRORS_PLAYER_NOT_IN_WORLD);
            return CompletableFuture.completedFuture((Object)null);
         }

         world = ((EntityStore)ref.getStore().getExternalData()).getWorld();
      }

      Store<EntityStore> store = world.getEntityStore().getStore();
      return this.runAsync(context, () -> {
         ObjectList<Ref<EntityStore>> entitiesToOperateOn;
         if (this.radiusArg.provided(context)) {
            if (!context.isPlayer()) {
               context.sendMessage(Message.translation("server.commands.errors.playerOrArg").param("option", "radius"));
               return;
            }

            Ref<EntityStore> playerRef = context.senderAsPlayerRef();
            if (playerRef == null || !playerRef.isValid()) {
               context.sendMessage(MESSAGE_COMMANDS_ERRORS_PLAYER_NOT_IN_WORLD);
               return;
            }

            TransformComponent transformComponent = (TransformComponent)store.getComponent(playerRef, TransformComponent.getComponentType());
            if (transformComponent == null) {
               context.sendMessage(MESSAGE_COMMANDS_ERRORS_PLAYER_NOT_IN_WORLD);
               return;
            }

            double radius = (Double)this.radiusArg.get(context);
            Vector3d position = transformComponent.getPosition();
            entitiesToOperateOn = new ObjectArrayList<Ref<EntityStore>>();
            ObjectList<Ref<EntityStore>> results = SpatialResource.getThreadLocalReferenceList();
            SpatialResource<Ref<EntityStore>, EntityStore> entitySpatialResource = (SpatialResource)store.getResource(EntityModule.get().getEntitySpatialResourceType());
            entitySpatialResource.getSpatialStructure().collect(position, radius, results);
            entitiesToOperateOn.addAll(results);
            SpatialResource<Ref<EntityStore>, EntityStore> playerSpatialResource = (SpatialResource)store.getResource(EntityModule.get().getPlayerSpatialResourceType());
            playerSpatialResource.getSpatialStructure().collect(position, radius, results);
            entitiesToOperateOn.addAll(results);
         } else if (this.playerArg.provided(context)) {
            PlayerRef targetPlayerRef = (PlayerRef)this.playerArg.get(context);
            Ref<EntityStore> targetRef = targetPlayerRef.getReference();
            if (targetRef == null || !targetRef.isValid()) {
               context.sendMessage(MESSAGE_COMMANDS_ERRORS_PLAYER_NOT_IN_WORLD);
               return;
            }

            entitiesToOperateOn = ObjectLists.<Ref<EntityStore>>singleton(targetRef);
         } else if (this.entityArg.provided(context)) {
            Ref<EntityStore> entityRef = this.entityArg.get(store, context);
            if (entityRef == null || !entityRef.isValid()) {
               context.sendMessage(MESSAGE_GENERAL_NO_ENTITY_IN_VIEW);
               return;
            }

            entitiesToOperateOn = ObjectLists.<Ref<EntityStore>>singleton(entityRef);
         } else {
            if (!context.isPlayer()) {
               context.sendMessage(Message.translation("server.commands.errors.playerOrArg").param("option", "entity"));
               return;
            }

            Ref<EntityStore> playerRef = context.senderAsPlayerRef();
            if (playerRef == null || !playerRef.isValid()) {
               context.sendMessage(MESSAGE_COMMANDS_ERRORS_PLAYER_NOT_IN_WORLD);
               return;
            }

            Ref<EntityStore> entityRef = TargetUtil.getTargetEntity(playerRef, store);
            if (entityRef == null) {
               context.sendMessage(MESSAGE_GENERAL_NO_ENTITY_IN_VIEW);
               return;
            }

            entitiesToOperateOn = ObjectLists.<Ref<EntityStore>>singleton(entityRef);
         }

         this.execute(context, entitiesToOperateOn, world, store);
      }, world);
   }

   protected abstract void execute(@Nonnull CommandContext var1, @Nonnull ObjectList<Ref<EntityStore>> var2, @Nonnull World var3, @Nonnull Store<EntityStore> var4);
}