SpawnPopulateCommand.java

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

import com.hypixel.hytale.component.RemoveReason;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.server.core.Message;
import com.hypixel.hytale.server.core.asset.type.environment.config.Environment;
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.basecommands.AbstractWorldCommand;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.WorldConfig;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.npc.entities.NPCEntity;
import javax.annotation.Nonnull;

public class SpawnPopulateCommand extends AbstractWorldCommand {
   @Nonnull
   private final OptionalArg<Environment> environmentArg;

   public SpawnPopulateCommand() {
      super("populate", "server.commands.spawning.populate.desc");
      this.environmentArg = this.withOptionalArg("environment", "server.commands.spawning.populate.arg.environment.desc", ArgTypes.ENVIRONMENT_ASSET);
   }

   protected void execute(@Nonnull CommandContext context, @Nonnull World world, @Nonnull Store<EntityStore> store) {
      int environmentIndex = this.environmentArg.provided(context) ? Environment.getAssetMap().getIndex(((Environment)this.environmentArg.get(context)).getId()) : -2147483648;
      store.forEachEntityParallel(NPCEntity.getComponentType(), (index, archetypeChunk, commandBuffer) -> {
         NPCEntity npcComponent = (NPCEntity)archetypeChunk.getComponent(index, NPCEntity.getComponentType());

         assert npcComponent != null;

         int npcEnvironment = npcComponent.getEnvironment();
         if (npcEnvironment >= 0 && (environmentIndex == -2147483648 || environmentIndex == npcEnvironment)) {
            commandBuffer.removeEntity(archetypeChunk.getReferenceTo(index), RemoveReason.REMOVE);
         }

      });
      WorldConfig worldConfig = world.getWorldConfig();
      worldConfig.setSpawningNPC(true);
      worldConfig.markChanged();
      context.sendMessage(Message.translation("server.commands.spawning.populate.success").param("worldName", world.getName()));
   }
}