48 lines
2.3 KiB
Java
48 lines
2.3 KiB
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 = this.withOptionalArg(
|
|
"environment", "server.commands.spawning.populate.arg.environment.desc", ArgTypes.ENVIRONMENT_ASSET
|
|
);
|
|
|
|
public SpawnPopulateCommand() {
|
|
super("populate", "server.commands.spawning.populate.desc");
|
|
}
|
|
|
|
@Override
|
|
protected void execute(@Nonnull CommandContext context, @Nonnull World world, @Nonnull Store<EntityStore> store) {
|
|
int environmentIndex = this.environmentArg.provided(context)
|
|
? Environment.getAssetMap().getIndex(this.environmentArg.get(context).getId())
|
|
: Integer.MIN_VALUE;
|
|
store.forEachEntityParallel(NPCEntity.getComponentType(), (index, archetypeChunk, commandBuffer) -> {
|
|
NPCEntity npcComponent = archetypeChunk.getComponent(index, NPCEntity.getComponentType());
|
|
|
|
assert npcComponent != null;
|
|
|
|
int npcEnvironment = npcComponent.getEnvironment();
|
|
if (npcEnvironment >= 0 && (environmentIndex == Integer.MIN_VALUE || 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()));
|
|
}
|
|
}
|