PrefabEditCreateNewCommand.java

package com.hypixel.hytale.builtin.buildertools.prefabeditor.commands;

import com.hypixel.hytale.builtin.buildertools.BuilderToolsPlugin;
import com.hypixel.hytale.builtin.buildertools.prefabeditor.PrefabEditorCreationSettings;
import com.hypixel.hytale.builtin.buildertools.prefabeditor.enums.PrefabRootDirectory;
import com.hypixel.hytale.builtin.buildertools.prefabeditor.enums.WorldGenType;
import com.hypixel.hytale.codec.validation.Validators;
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.CommandContext;
import com.hypixel.hytale.server.core.command.system.arguments.system.DefaultArg;
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.AbstractAsyncPlayerCommand;
import com.hypixel.hytale.server.core.entity.entities.Player;
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 java.nio.file.Path;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import javax.annotation.Nonnull;

public class PrefabEditCreateNewCommand extends AbstractAsyncPlayerCommand {
   @Nonnull
   private static final Message MESSAGE_COMMANDS_EDIT_PREFAB_NEW_ERRORS_NOT_A_FILE = Message.translation("server.commands.editprefab.new.errors.notAFile");
   @Nonnull
   private final RequiredArg<String> prefabNameArg;
   @Nonnull
   private final DefaultArg<WorldGenType> worldGenTypeArg;
   @Nonnull
   private final DefaultArg<Integer> flatNumBlocksBelowArg;
   @Nonnull
   private final DefaultArg<PrefabRootDirectory> prefabPathArg;

   public PrefabEditCreateNewCommand() {
      super("new", "server.commands.editprefab.new.desc");
      this.prefabNameArg = this.withRequiredArg("prefabName", "server.commands.editprefab.new.name.desc", ArgTypes.STRING);
      this.worldGenTypeArg = this.withDefaultArg("worldgen", "server.commands.editprefab.load.worldGenType.desc", ArgTypes.forEnum("WorldGenType", WorldGenType.class), PrefabEditLoadCommand.DEFAULT_WORLD_GEN_TYPE, "server.commands.editprefab.load.worldGenType.default.desc");
      this.flatNumBlocksBelowArg = (DefaultArg)this.withDefaultArg("numBlocksToSurface", "server.commands.editprefab.load.numBlocksToSurface.desc", ArgTypes.INTEGER, 0, "server.commands.editprefab.load.numBlocksToSurface.default.desc").addValidator(Validators.range(0, 120));
      this.prefabPathArg = this.withDefaultArg("prefabPath", "server.commands.editprefab.load.path.desc", ArgTypes.forEnum("PrefabPath", PrefabRootDirectory.class), PrefabRootDirectory.ASSET, "server.commands.editprefab.load.path.default.desc");
   }

   @Nonnull
   protected CompletableFuture<Void> executeAsync(@Nonnull CommandContext context, @Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> ref, @Nonnull PlayerRef playerRef, @Nonnull World world) {
      Player playerComponent = (Player)store.getComponent(ref, Player.getComponentType());

      assert playerComponent != null;

      Path prefabBaseDirectory = ((PrefabRootDirectory)this.prefabPathArg.get(context)).getPrefabPath();
      String prefabName = (String)this.prefabNameArg.get(context);
      if (!prefabName.endsWith(".prefab.json")) {
         prefabName = prefabName + ".prefab.json";
      }

      Path prefabPath = prefabBaseDirectory.resolve(prefabName);
      if (prefabPath.toString().endsWith("/")) {
         context.sendMessage(MESSAGE_COMMANDS_EDIT_PREFAB_NEW_ERRORS_NOT_A_FILE);
         return CompletableFuture.completedFuture((Object)null);
      } else {
         PrefabEditorCreationSettings prefabEditorLoadCommandSettings = new PrefabEditorCreationSettings((PrefabRootDirectory)this.prefabPathArg.get(context), List.of(prefabName), 55, 15, (WorldGenType)this.worldGenTypeArg.get(context), (Integer)this.flatNumBlocksBelowArg.get(context), PrefabEditLoadCommand.DEFAULT_PREFAB_STACKING_AXIS, PrefabEditLoadCommand.DEFAULT_PREFAB_ALIGNMENT, false, false, true, true, PrefabEditLoadCommand.DEFAULT_ROW_SPLIT_MODE, "Env_Zone1_Plains", "#5B9E28");
         context.sendMessage(Message.translation("server.commands.editprefab.new.success").param("path", prefabPath.toString()));
         return BuilderToolsPlugin.get().getPrefabEditSessionManager().createEditSessionForNewPrefab(ref, playerComponent, prefabEditorLoadCommandSettings, store);
      }
   }
}