studio.magemonkey.blueprint.commands.LoadSubCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of blueprint Show documentation
Let NPCs build your schematics and structures block by block with Citizens
The newest version!
package studio.magemonkey.blueprint.commands;

import studio.magemonkey.blueprint.Blueprint;
import studio.magemonkey.blueprint.hooks.citizens.BuilderTrait;
import studio.magemonkey.blueprint.schematic.Schematic;
import org.bukkit.ChatColor;
import org.bukkit.command.Command;
import org.bukkit.command.CommandSender;
import org.bukkit.scheduler.BukkitRunnable;
import org.bukkit.util.StringUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;

public class LoadSubCommand extends AbstractCommand {
    public LoadSubCommand(@Nullable SchematicBuilderCommand parent) {
        super("load", "Loads a schematic file", parent);
        this.permission = "schematicbuilder.load";
    }

    @Override
    public List getArguments(CommandSender sender) {
        List files = new ArrayList<>();
        try {
            Path dir = new File(Blueprint.getInstance().config().getSchematicsFolder()).toPath();
            Files.walk(dir).forEach(path -> {
                String stringPath = path.toString();
                if (path.toFile().isFile() && (stringPath.endsWith(".schem") || stringPath.endsWith(".nbt")
                        || stringPath.endsWith(".yml"))) {
                    files.add(dir.relativize(path).toString());
                }
            });
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        return files;
    }

    @Override
    public List getUsages(CommandSender sender) {
        List list = new ArrayList<>();
        list.add('/' + getFullCommand() + " ");
        return list;
    }

    @Override
    @Nullable
    public List onTabComplete(@NotNull CommandSender sender,
                                      @NotNull Command command,
                                      @NotNull String label,
                                      @NotNull String[] args) {
        if (args.length == 0) {
            return getArguments(sender);
        } else {
            StringBuilder stringBuilder = new StringBuilder(args[0]);
            for (int i = 1; i < args.length; i++) {
                stringBuilder.append(" ").append(args[i]);
            }
            String       arg         = stringBuilder.toString();
            List completions = new ArrayList<>();
            StringUtil.copyPartialMatches(stringBuilder.toString().trim(), getArguments(sender), completions);
            completions.removeIf(arg::startsWith);
            return completions;
        }
    }

    @Override
    public void execute(@NotNull CommandSender sender,
                        @NotNull Command command,
                        @NotNull String label,
                        @NotNull List args) {
        if (args.size() == 0) {
            sendUsage(sender);
            return;
        }

        StringBuilder stringBuilder = new StringBuilder(args.get(0));
        for (int i = 1; i < args.size(); i++) {
            stringBuilder.append(" ").append(args.get(i));
        }
        LoadSubCommand.execute(stringBuilder.toString().trim(), sender, null);
    }

    public static void execute(String path, CommandSender sender, @Nullable Runnable onComplete) {
        BuilderTrait builder = getSelectedBuilder(sender);
        if (builder == null) {
            sender.sendMessage(ChatColor.RED + "No builder is selected.");
            return;
        }
        if (builder.getState() != BuilderTrait.BuilderState.IDLE) {
            sender.sendMessage(ChatColor.RED + "Please cancel current build before loading new schematic.");
            return;
        }
        new BukkitRunnable() {
            @Override
            public void run() {
                try {
                    Schematic schematic = Blueprint.getSchematic(path);
                    new BukkitRunnable() {
                        @Override
                        public void run() {
                            if (schematic == null) {
                                sender.sendMessage(ChatColor.RED + "no such file " + path);
                            } else {
                                builder.setSchematic(schematic);
                                sender.sendMessage(ChatColor.GREEN + "Loaded Sucessfully");
                                sender.sendMessage(schematic.getInfo());
                            }
                        }
                    }.runTask(Blueprint.getInstance());
                } catch (Exception e) {
                    new BukkitRunnable() {
                        @Override
                        public void run() {
                            sender.sendMessage(ChatColor.RED + "Failed to load schematic " + path
                                    + ", check console for more details.");
                            Blueprint.getInstance()
                                    .getLogger()
                                    .log(Level.WARNING, "Failed to load schematic: " + path);
                            e.printStackTrace();
                        }
                    }.runTask(Blueprint.getInstance());
                }
                if (onComplete != null) {
                    new BukkitRunnable() {
                        @Override
                        public void run() {onComplete.run();}
                    }.runTask(Blueprint.getInstance());
                }
            }
        }.runTaskAsynchronously(Blueprint.getInstance());
    }
}