2019-04-26 08:38:06 +00:00
|
|
|
/*
|
|
|
|
* WorldEdit, a Minecraft world manipulation toolkit
|
|
|
|
* Copyright (C) sk89q <http://www.sk89q.com>
|
|
|
|
* Copyright (C) WorldEdit team and contributors
|
|
|
|
*
|
|
|
|
* This program is free software: you can redistribute it and/or modify it
|
|
|
|
* under the terms of the GNU Lesser General Public License as published by the
|
|
|
|
* Free Software Foundation, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
|
|
|
|
* for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
2019-04-26 08:18:02 +00:00
|
|
|
package com.sk89q.worldedit.command;
|
|
|
|
|
|
|
|
import com.google.common.collect.ImmutableList;
|
|
|
|
import com.google.common.collect.ImmutableSet;
|
|
|
|
import com.sk89q.worldedit.LocalSession;
|
|
|
|
import com.sk89q.worldedit.WorldEditException;
|
|
|
|
import com.sk89q.worldedit.blocks.BaseItem;
|
|
|
|
import com.sk89q.worldedit.command.factory.ItemUseFactory;
|
|
|
|
import com.sk89q.worldedit.command.factory.ReplaceFactory;
|
2019-04-26 08:38:06 +00:00
|
|
|
import com.sk89q.worldedit.command.factory.TreeGeneratorFactory;
|
2019-04-26 08:18:02 +00:00
|
|
|
import com.sk89q.worldedit.command.util.PermissionCondition;
|
|
|
|
import com.sk89q.worldedit.entity.Player;
|
|
|
|
import com.sk89q.worldedit.function.Contextual;
|
|
|
|
import com.sk89q.worldedit.function.RegionFunction;
|
|
|
|
import com.sk89q.worldedit.function.factory.Apply;
|
|
|
|
import com.sk89q.worldedit.function.pattern.Pattern;
|
|
|
|
import com.sk89q.worldedit.regions.factory.RegionFactory;
|
|
|
|
import com.sk89q.worldedit.util.TreeGenerator;
|
2019-04-26 08:38:06 +00:00
|
|
|
import com.sk89q.worldedit.util.command.CommandRegistrationHandler;
|
2019-04-26 08:18:02 +00:00
|
|
|
import com.sk89q.worldedit.util.formatting.text.TextComponent;
|
|
|
|
import com.sk89q.worldedit.util.formatting.text.TranslatableComponent;
|
|
|
|
import org.enginehub.piston.CommandManager;
|
|
|
|
import org.enginehub.piston.CommandParameters;
|
|
|
|
import org.enginehub.piston.DefaultCommandManagerService;
|
|
|
|
import org.enginehub.piston.annotation.Command;
|
|
|
|
import org.enginehub.piston.annotation.CommandContainer;
|
|
|
|
import org.enginehub.piston.annotation.param.Arg;
|
|
|
|
import org.enginehub.piston.inject.Key;
|
|
|
|
import org.enginehub.piston.part.CommandArgument;
|
|
|
|
import org.enginehub.piston.part.SubCommandPart;
|
|
|
|
|
|
|
|
import java.util.stream.Collectors;
|
|
|
|
|
|
|
|
import static java.util.Objects.requireNonNull;
|
|
|
|
import static org.enginehub.piston.part.CommandParts.arg;
|
|
|
|
|
|
|
|
@CommandContainer
|
|
|
|
public class ApplyBrushCommands {
|
|
|
|
|
2019-04-29 00:30:58 +00:00
|
|
|
private static final CommandArgument REGION_FACTORY = arg(TranslatableComponent.of("shape"), TextComponent.of("The shape of the region"))
|
2019-04-26 08:18:02 +00:00
|
|
|
.defaultsTo(ImmutableList.of())
|
|
|
|
.ofTypes(ImmutableList.of(Key.of(RegionFactory.class)))
|
|
|
|
.build();
|
|
|
|
|
|
|
|
private static final CommandArgument RADIUS = arg(TranslatableComponent.of("radius"), TextComponent.of("The size of the brush"))
|
|
|
|
.defaultsTo(ImmutableList.of("5"))
|
|
|
|
.ofTypes(ImmutableList.of(Key.of(double.class)))
|
|
|
|
.build();
|
|
|
|
|
2019-04-26 08:38:06 +00:00
|
|
|
public static void register(CommandManager commandManager, CommandRegistrationHandler registration) {
|
2019-04-26 08:18:02 +00:00
|
|
|
commandManager.register("apply", builder -> {
|
|
|
|
builder.description(TextComponent.of("Apply brush, apply a function to every block"));
|
|
|
|
builder.action(org.enginehub.piston.Command.Action.NULL_ACTION);
|
|
|
|
|
|
|
|
CommandManager manager = DefaultCommandManagerService.getInstance()
|
|
|
|
.newCommandManager();
|
2019-04-26 08:38:06 +00:00
|
|
|
registration.register(
|
2019-04-26 08:18:02 +00:00
|
|
|
manager,
|
|
|
|
ApplyBrushCommandsRegistration.builder(),
|
|
|
|
new ApplyBrushCommands()
|
|
|
|
);
|
|
|
|
|
|
|
|
builder.condition(new PermissionCondition(ImmutableSet.of("worldedit.brush.apply")));
|
|
|
|
|
|
|
|
builder.addParts(REGION_FACTORY, RADIUS);
|
|
|
|
builder.addPart(SubCommandPart.builder(TranslatableComponent.of("type"), TextComponent.of("Type of brush to use"))
|
|
|
|
.withCommands(manager.getAllCommands().collect(Collectors.toList()))
|
2019-04-29 00:27:33 +00:00
|
|
|
.required()
|
2019-04-26 08:18:02 +00:00
|
|
|
.build());
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
private void setApplyBrush(CommandParameters parameters, Player player, LocalSession localSession,
|
|
|
|
Contextual<? extends RegionFunction> generatorFactory) throws WorldEditException {
|
|
|
|
double radius = requireNonNull(RADIUS.value(parameters).asSingle(double.class));
|
|
|
|
RegionFactory regionFactory = REGION_FACTORY.value(parameters).asSingle(RegionFactory.class);
|
|
|
|
BrushCommands.setOperationBasedBrush(player, localSession, radius,
|
|
|
|
new Apply(generatorFactory), regionFactory, "worldedit.brush.apply");
|
|
|
|
}
|
|
|
|
|
|
|
|
@Command(
|
|
|
|
name = "forest",
|
|
|
|
desc = "Plant trees"
|
|
|
|
)
|
|
|
|
public void forest(CommandParameters parameters,
|
|
|
|
Player player, LocalSession localSession,
|
|
|
|
@Arg(desc = "The type of tree to plant")
|
|
|
|
TreeGenerator.TreeType type) throws WorldEditException {
|
|
|
|
setApplyBrush(parameters, player, localSession, new TreeGeneratorFactory(type));
|
|
|
|
}
|
|
|
|
|
|
|
|
@Command(
|
|
|
|
name = "item",
|
|
|
|
desc = "Use an item"
|
|
|
|
)
|
|
|
|
public void item(CommandParameters parameters,
|
|
|
|
Player player, LocalSession localSession,
|
|
|
|
@Arg(desc = "The type of item to use")
|
|
|
|
BaseItem item) throws WorldEditException {
|
|
|
|
setApplyBrush(parameters, player, localSession, new ItemUseFactory(item));
|
|
|
|
}
|
|
|
|
|
|
|
|
@Command(
|
|
|
|
name = "set",
|
|
|
|
desc = "Place a block"
|
|
|
|
)
|
|
|
|
public void set(CommandParameters parameters,
|
|
|
|
Player player, LocalSession localSession,
|
|
|
|
@Arg(desc = "The pattern of blocks to use")
|
|
|
|
Pattern pattern) throws WorldEditException {
|
|
|
|
setApplyBrush(parameters, player, localSession, new ReplaceFactory(pattern));
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|