hytale-server/com/hypixel/hytale/builtin/buildertools/prefabeditor/commands/PrefabEditCreateNewCommand....

104 lines
4.7 KiB
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 = this.withRequiredArg("prefabName", "server.commands.editprefab.new.name.desc", ArgTypes.STRING);
@Nonnull
private final DefaultArg<WorldGenType> 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"
);
@Nonnull
private final DefaultArg<Integer> flatNumBlocksBelowArg = this.withDefaultArg(
"numBlocksToSurface",
"server.commands.editprefab.load.numBlocksToSurface.desc",
ArgTypes.INTEGER,
0,
"server.commands.editprefab.load.numBlocksToSurface.default.desc"
)
.addValidator(Validators.range(0, 120));
@Nonnull
private final DefaultArg<PrefabRootDirectory> prefabPathArg = this.withDefaultArg(
"prefabPath",
"server.commands.editprefab.load.path.desc",
ArgTypes.forEnum("PrefabPath", PrefabRootDirectory.class),
PrefabRootDirectory.ASSET,
"server.commands.editprefab.load.path.default.desc"
);
public PrefabEditCreateNewCommand() {
super("new", "server.commands.editprefab.new.desc");
}
@Nonnull
@Override
protected CompletableFuture<Void> executeAsync(
@Nonnull CommandContext context, @Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> ref, @Nonnull PlayerRef playerRef, @Nonnull World world
) {
Player playerComponent = store.getComponent(ref, Player.getComponentType());
assert playerComponent != null;
Path prefabBaseDirectory = this.prefabPathArg.get(context).getPrefabPath();
String prefabName = 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(null);
} else {
PrefabEditorCreationSettings prefabEditorLoadCommandSettings = new PrefabEditorCreationSettings(
this.prefabPathArg.get(context),
List.of(prefabName),
55,
15,
this.worldGenTypeArg.get(context),
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);
}
}
}