Examples with CommandSource com.velocitypowered.api.command.CommandSource used on opensource projects

Example 6 with CommandSource

use of com.velocitypowered.api.command.CommandSource in project LibertyBans by A248.

the class CommandHandler method suggest.

@Override
public List<String> suggest(Invocation invocation) {
    CommandSource platformSender = invocation.source();
    String[] args = invocation.arguments();
    return commandHelper.suggest(platformSender, adaptArgs(args, true));
}

Example 7 with CommandSource

use of com.velocitypowered.api.command.CommandSource in project SubServers-2 by ME1312.

the class SubCommand method suggest.

/**
 * Suggest command arguments
 *
 * @param invocation Command invocation
 * @return The validator's response and list of possible arguments
 */
@SuppressWarnings("unchecked")
public List<String> suggest(Invocation invocation) {
    CommandSource sender = invocation.source();
    String[] args = invocation.arguments();
    String Last = (args.length > 0) ? args[args.length - 1] : "";
    String last = Last.toLowerCase();
    ServerInfo pcs = (sender instanceof Player) ? ((Player) sender).getCurrentServer().map(ServerConnection::getServerInfo).orElse(null) : null;
    ServerData pcsd = plugin.getData(pcs);
    if (plugin.api.getSubDataNetwork()[0] == null) {
        if (sender instanceof ConsoleCommandSource)
            new IllegalStateException("SubData is not connected").printStackTrace();
        return Collections.emptyList();
    } else if (sender instanceof Player && (!permitted.containsKey(((Player) sender).getUniqueId()) || !permitted.get(((Player) sender).getUniqueId()).containsKey(pcs) || !permitted.get(((Player) sender).getUniqueId()).get(pcs).value())) {
        if (permitted.containsKey(((Player) sender).getUniqueId()) && permitted.get(((Player) sender).getUniqueId()).containsKey(pcs) && permitted.get(((Player) sender).getUniqueId()).get(pcs).key() == null) {
        // do nothing
        } else if (!permitted.containsKey(((Player) sender).getUniqueId()) || !permitted.get(((Player) sender).getUniqueId()).containsKey(pcs) || Calendar.getInstance().getTime().getTime() - permitted.get(((Player) sender).getUniqueId()).get(pcs).key() >= TimeUnit.MINUTES.toMillis(1)) {
            if (pcsd == null || pcsd.getSubData()[0] == null) {
                HashMap<ServerInfo, Pair<Long, Boolean>> map = (permitted.containsKey(((Player) sender).getUniqueId())) ? permitted.get(((Player) sender).getUniqueId()) : new HashMap<ServerInfo, Pair<Long, Boolean>>();
                map.put(pcs, new ContainedPair<>(Calendar.getInstance().getTime().getTime(), false));
                permitted.put(((Player) sender).getUniqueId(), map);
            } else {
                HashMap<ServerInfo, Pair<Long, Boolean>> map = (permitted.containsKey(((Player) sender).getUniqueId())) ? permitted.get(((Player) sender).getUniqueId()) : new HashMap<ServerInfo, Pair<Long, Boolean>>();
                map.put(pcs, new ContainedPair<>(null, false));
                permitted.put(((Player) sender).getUniqueId(), map);
                ((SubDataSender) pcsd.getSubData()[0]).sendPacket(new PacketCheckPermission(((Player) sender).getUniqueId(), "subservers.command", result -> {
                    map.put(pcs, new ContainedPair<>(Calendar.getInstance().getTime().getTime(), result));
                }));
            }
        }
        return Collections.emptyList();
    } else if (args.length <= 1) {
        List<String> cmds = new ArrayList<>();
        cmds.addAll(Arrays.asList("help", "list", "info", "status", "version", "start", "restart", "stop", "kill", "terminate", "cmd", "command", "create", "update", "upgrade", "restore"));
        if (!(sender instanceof Player))
            cmds.addAll(Arrays.asList("remove", "delete"));
        updateCache();
        List<String> list = new ArrayList<String>();
        for (String cmd : cmds) {
            if (cmd.startsWith(last))
                list.add(Last + cmd.substring(last.length()));
        }
        return list;
    } else {
        if (args[0].equals("info") || args[0].equals("status")) {
            Supplier<Collection<String>> getPlayers = () -> {
                LinkedList<String> names = new LinkedList<String>();
                for (Player player : ExProxy.getInstance().getAllPlayers()) names.add(player.getGameProfile().getName());
                for (CachedPlayer player : plugin.api.getRemotePlayers().values()) if (!names.contains(player.getName()))
                    names.add(player.getName());
                Collections.sort(names);
                return names;
            };
            updateCache();
            if (args.length == 2) {
                List<String> list = new ArrayList<String>();
                List<String> subcommands = new ArrayList<String>();
                subcommands.add("proxy");
                subcommands.add("host");
                subcommands.add("group");
                subcommands.add("server");
                subcommands.add("subserver");
                subcommands.add("player");
                for (String command : subcommands) {
                    if (!list.contains(command) && command.toLowerCase().startsWith(last))
                        list.add(Last + command.substring(last.length()));
                }
                Proxy master = proxyMasterCache;
                if (master != null && !list.contains(master.getName()) && master.getName().toLowerCase().startsWith(last))
                    list.add(Last + master.getName().substring(last.length()));
                for (Proxy proxy : proxyCache.values()) {
                    if (!list.contains(proxy.getName()) && proxy.getName().toLowerCase().startsWith(last))
                        list.add(Last + proxy.getName().substring(last.length()));
                }
                for (Host host : hostCache.values()) {
                    if (!list.contains(host.getName()) && host.getName().toLowerCase().startsWith(last))
                        list.add(Last + host.getName().substring(last.length()));
                }
                for (String group : groupCache.keySet()) {
                    if (!list.contains(group) && group.toLowerCase().startsWith(last))
                        list.add(Last + group.substring(last.length()));
                }
                for (ServerData server : plugin.servers.values()) {
                    if (!list.contains(server.getName()) && server.getName().toLowerCase().startsWith(last))
                        list.add(Last + server.getName().substring(last.length()));
                }
                for (String player : getPlayers.get()) {
                    if (!list.contains(player) && player.toLowerCase().startsWith(last))
                        list.add(Last + player.substring(last.length()));
                }
                return list;
            } else if (args.length == 3) {
                List<String> list = new ArrayList<String>();
                switch(args[1].toLowerCase()) {
                    case "p":
                    case "proxy":
                        Proxy master = proxyMasterCache;
                        if (master != null && master.getName().toLowerCase().startsWith(last))
                            list.add(Last + master.getName().substring(last.length()));
                        for (Proxy proxy : proxyCache.values()) {
                            if (!list.contains(proxy.getName()) && proxy.getName().toLowerCase().startsWith(last))
                                list.add(Last + proxy.getName().substring(last.length()));
                        }
                        break;
                    case "h":
                    case "host":
                        for (Host host : hostCache.values()) {
                            if (host.getName().toLowerCase().startsWith(last))
                                list.add(Last + host.getName().substring(last.length()));
                        }
                        break;
                    case "g":
                    case "group":
                        for (String group : groupCache.keySet()) {
                            if (group.toLowerCase().startsWith(last))
                                list.add(Last + group.substring(last.length()));
                        }
                        break;
                    case "s":
                    case "server":
                    case "subserver":
                        for (ServerData server : plugin.servers.values()) {
                            if ((!args[1].equalsIgnoreCase("subserver") || server instanceof SubServerData) && server.getName().toLowerCase().startsWith(last))
                                list.add(Last + server.getName().substring(last.length()));
                        }
                        break;
                    case "u":
                    case "user":
                    case "player":
                        for (String player : getPlayers.get()) {
                            if (player.toLowerCase().startsWith(last))
                                list.add(Last + player.substring(last.length()));
                        }
                        break;
                }
                return list;
            } else {
                return Collections.emptyList();
            }
        } else if (args[0].equals("start") || args[0].equals("restart") || args[0].equals("stop") || args[0].equals("kill") || args[0].equals("terminate") || args[0].equals("cmd") || args[0].equals("command") || args[0].equals("update") || args[0].equals("upgrade") || args[0].equals("remove") || args[0].equals("del") || args[0].equals("delete")) {
            List<String> list = new ArrayList<String>();
            boolean mode = !args[0].equals("cmd") && !args[0].equals("command");
            RawServerSelection select = selectRawServers(null, args, 1, mode);
            if (select.last != null) {
                if (last.startsWith("::")) {
                    Map<String, Host> hosts = hostCache;
                    if (hosts.size() > 0) {
                        if (Arrays.binarySearch(select.selection, "::*") < 0 && "::*".startsWith(last))
                            list.add("::*");
                        if (sender instanceof Player && Arrays.binarySearch(select.selection, "::.") < 0 && "::.".startsWith(last))
                            list.add("::.");
                        for (Host host : hosts.values()) {
                            String name = "::" + host.getName();
                            if (Arrays.binarySearch(select.selection, name.toLowerCase()) < 0 && name.toLowerCase().startsWith(last))
                                list.add(Last + name.substring(last.length()));
                        }
                    }
                    return list;
                } else if (last.startsWith(":")) {
                    Map<String, List<Server>> groups = groupCache;
                    if (groups.size() > 0) {
                        if (Arrays.binarySearch(select.selection, ":*") < 0 && ":*".startsWith(last))
                            list.add(":*");
                        if (sender instanceof Player && Arrays.binarySearch(select.selection, ":.") < 0 && ":.".startsWith(last))
                            list.add(":.");
                        for (String group : groups.keySet()) {
                            group = ":" + group;
                            if (Arrays.binarySearch(select.selection, group.toLowerCase()) < 0 && group.toLowerCase().startsWith(last))
                                list.add(Last + group.substring(last.length()));
                        }
                    }
                    return list;
                } else {
                    Map<ServerInfo, ServerData> subservers = plugin.servers;
                    if (subservers.size() > 0) {
                        if (Arrays.binarySearch(select.selection, "*") < 0 && "*".startsWith(last))
                            list.add("*");
                        if (sender instanceof Player && Arrays.binarySearch(select.selection, ".") < 0 && ".".startsWith(last))
                            list.add(".");
                        for (ServerData server : subservers.values()) {
                            if ((!mode || server instanceof SubServerData) && Arrays.binarySearch(select.selection, server.getName().toLowerCase()) < 0 && server.getName().toLowerCase().startsWith(last))
                                list.add(Last + server.getName().substring(last.length()));
                        }
                    }
                    return list;
                }
            } else if (args[0].equals("cmd") || args[0].equals("command")) {
                if (select.args.length == 3) {
                    return Collections.singletonList("<Command>");
                } else {
                    return Collections.singletonList("[Args...]");
                }
            } else if (args[0].equals("update") || args[0].equals("upgrade")) {
                if (select.args.length == 3) {
                    return Arrays.asList("[Template]", "[Version]");
                } else if (select.args.length == 4) {
                    return Collections.singletonList("<Version>");
                }
            }
            return Collections.emptyList();
        } else if (args[0].equals("create")) {
            updateCache();
            if (args.length == 2) {
                return Collections.singletonList("<Name>");
            } else if (args.length == 3) {
                List<String> list = new ArrayList<String>();
                for (Host host : hostCache.values()) {
                    if (host.getName().toLowerCase().startsWith(last))
                        list.add(Last + host.getName().substring(last.length()));
                }
                return list;
            } else if (args.length == 4) {
                List<String> list = new ArrayList<String>();
                Map<String, Host> hosts = hostCache;
                if (!hosts.containsKey(args[2].toLowerCase())) {
                    list.add("<Template>");
                } else {
                    for (SubCreator.ServerTemplate template : hosts.get(args[2].toLowerCase()).getCreator().getTemplates().values()) {
                        if (template.getName().toLowerCase().startsWith(last))
                            list.add(Last + template.getName().substring(last.length()));
                    }
                }
                return list;
            } else if (args.length == 5) {
                return Collections.singletonList("[Version]");
            } else if (args.length == 6) {
                if (last.length() > 0) {
                    if (!Try.all.run(() -> Integer.parseInt(last)) || Integer.parseInt(last) <= 0 || Integer.parseInt(last) > 65535) {
                        return Collections.emptyList();
                    }
                }
                return Collections.singletonList("[Port]");
            } else {
                return Collections.emptyList();
            }
        } else if (args[0].equals("restore")) {
            if (args.length == 2) {
                return Collections.singletonList("<Subserver>");
            } else {
                return Collections.emptyList();
            }
        } else if (sender instanceof Player && (args[0].equals("tp") || args[0].equals("teleport"))) {
            if (args.length == 2 || args.length == 3) {
                List<String> list = new ArrayList<String>();
                if (args.length == 2) {
                    list.add("@p");
                    list.add("@a");
                    list.add("@r");
                    list.add("@s");
                    List<UUID> used = new ArrayList<UUID>();
                    Optional<ServerConnection> server = ((Player) sender).getCurrentServer();
                    if (server.isPresent()) {
                        for (Player player : server.get().getServer().getPlayersConnected()) {
                            if (player.getGameProfile().getName().toLowerCase().startsWith(last))
                                list.add(Last + player.getGameProfile().getName().substring(last.length()));
                            used.add(player.getUniqueId());
                        }
                        for (CachedPlayer player : SubAPI.getInstance().getRemotePlayers(server.get().getServerInfo()).values()) {
                            if (!used.contains(player.getUniqueId())) {
                                if (player.getName().toLowerCase().startsWith(last))
                                    list.add(Last + player.getName().substring(last.length()));
                                used.add(player.getUniqueId());
                            }
                        }
                    }
                }
                for (ServerData server : plugin.servers.values()) {
                    if (server.getName().toLowerCase().startsWith(last))
                        list.add(Last + server.getName().substring(last.length()));
                }
                return list;
            } else {
                return Collections.emptyList();
            }
        } else {
            return Collections.emptyList();
        }
    }
}

Example 8 with CommandSource

use of com.velocitypowered.api.command.CommandSource in project SubServers-2 by ME1312.

the class SubCommand method selectServers.

private void selectServers(CommandSource sender, String[] rargs, int index, boolean mode, Consumer<ServerSelection> callback) {
    StackTraceElement[] origin = new Throwable().getStackTrace();
    LinkedList<Component> msgs = new LinkedList<Component>();
    LinkedList<String> args = new LinkedList<String>();
    LinkedList<String> selection = new LinkedList<String>();
    LinkedList<Server> select = new LinkedList<Server>();
    Value<String> last = new Container<String>(null);
    // Step 1
    Value<Integer> ic = new Container<Integer>(0);
    while (ic.value() < index) {
        args.add(rargs[ic.value()]);
        ic.value(ic.value() + 1);
    }
    // Step 3
    StringBuilder completed = new StringBuilder();
    Runnable finished = () -> {
        args.add(completed.toString());
        int i = ic.value();
        while (i < rargs.length) {
            args.add(rargs[i]);
            last.value(null);
            i++;
        }
        LinkedList<Server> servers = new LinkedList<Server>();
        LinkedList<SubServer> subservers = new LinkedList<SubServer>();
        for (Server server : select) {
            if (!servers.contains(server)) {
                servers.add(server);
                if (server instanceof SubServer)
                    subservers.add((SubServer) server);
            }
        }
        if ((!mode && servers.size() <= 0) || (mode && subservers.size() <= 0)) {
            Component msg = Component.text("SubServers > No " + ((mode) ? "sub" : "") + "servers were selected");
            if (sender != null)
                sender.sendMessage(msg);
            msgs.add(msg);
        }
        try {
            callback.accept(new ServerSelection(msgs, selection, servers, subservers, args, last.value()));
        } catch (Throwable e) {
            Throwable ew = new InvocationTargetException(e);
            ew.setStackTrace(origin);
            ew.printStackTrace();
        }
    };
    // Step 2
    AsyncConsolidator merge = new AsyncConsolidator(finished);
    for (boolean run = true; run && ic.value() < rargs.length; ic.value(ic.value() + 1)) {
        String current = rargs[ic.value()];
        last.value(current);
        completed.append(current);
        if (current.endsWith(",")) {
            current = current.substring(0, current.length() - 1);
            completed.append(' ');
        } else
            run = false;
        selection.add(current.toLowerCase());
        if (current.length() > 0) {
            merge.reserve();
            if (current.startsWith("::") && current.length() > 2) {
                current = current.substring(2);
                if (current.equals("*")) {
                    plugin.api.getHosts(hostMap -> {
                        for (Host host : hostMap.values()) {
                            select.addAll(host.getSubServers().values());
                        }
                        merge.release();
                    });
                } else {
                    final String fcurrent = current;
                    plugin.api.getHost(current, host -> {
                        if (host != null) {
                            if (!select.addAll(host.getSubServers().values())) {
                                Component msg = Component.text("SubServers > There are no " + ((mode) ? "sub" : "") + "servers on host: " + host.getName());
                                if (sender != null)
                                    sender.sendMessage(msg);
                                msgs.add(msg);
                            }
                        } else {
                            Component msg = Component.text("SubServers > There is no host with name: " + fcurrent);
                            if (sender != null)
                                sender.sendMessage(msg);
                            msgs.add(msg);
                        }
                        merge.release();
                    });
                }
            } else if (current.startsWith(":") && current.length() > 1) {
                current = current.substring(1);
                if (current.equals("*")) {
                    plugin.api.getGroups(groupMap -> {
                        for (List<Server> group : groupMap.values()) for (Server server : group) {
                            if (!mode || server instanceof SubServer)
                                select.add(server);
                        }
                        merge.release();
                    });
                } else {
                    final String fcurrent = current;
                    plugin.api.getGroup(current, group -> {
                        if (group != null) {
                            int i = 0;
                            for (Server server : group.value()) {
                                if (!mode || server instanceof SubServer) {
                                    select.add(server);
                                    i++;
                                }
                            }
                            if (i <= 0) {
                                Component msg = Component.text("SubServers > There are no " + ((mode) ? "sub" : "") + "servers in group: " + group.key());
                                if (sender != null)
                                    sender.sendMessage(msg);
                                msgs.add(msg);
                            }
                        } else {
                            Component msg = Component.text("SubServers > There is no group with name: " + fcurrent);
                            if (sender != null)
                                sender.sendMessage(msg);
                            msgs.add(msg);
                        }
                        merge.release();
                    });
                }
            } else {
                if (current.equals("*")) {
                    plugin.api.getServers(serverMap -> {
                        for (Server server : serverMap.values()) {
                            if (!mode || server instanceof SubServer)
                                select.add(server);
                        }
                        merge.release();
                    });
                } else {
                    final String fcurrent = current;
                    plugin.api.getServer(current, server -> {
                        if (server != null) {
                            select.add(server);
                        } else {
                            Component msg = Component.text("SubServers > There is no " + ((mode) ? "sub" : "") + "server with name: " + fcurrent);
                            if (sender != null)
                                sender.sendMessage(msg);
                            msgs.add(msg);
                        }
                        merge.release();
                    });
                }
            }
        }
    }
}

Example 9 with CommandSource

use of com.velocitypowered.api.command.CommandSource in project LimboFilter by Elytrium.

the class LimboFilterCommand method suggest.

@Override
public List<String> suggest(SimpleCommand.Invocation invocation) {
    CommandSource source = invocation.source();
    String[] args = invocation.arguments();
    if (args.length == 0) {
        return this.getSubCommands().filter(cmd -> source.hasPermission("limbofilter." + cmd)).collect(Collectors.toList());
    } else if (args.length == 1) {
        return this.getSubCommands().filter(cmd -> source.hasPermission("limbofilter." + cmd)).filter(str -> str.regionMatches(true, 0, args[0], 0, args[0].length())).collect(Collectors.toList());
    }
    return ImmutableList.of();
}

Example 10 with CommandSource

use of com.velocitypowered.api.command.CommandSource in project Lamp by Revxrsal.

the class VelocitySimpleCommand method execute.

@Override
public void execute(Invocation invocation) {
    CommandSource source = invocation.source();
    VelocityCommandActor actor = new VelocityActor(source, handler.getServer(), handler);
    try {
        ArgumentStack arguments = ArgumentStack.of(invocation.arguments());
        arguments.addFirst(invocation.alias());
        handler.dispatch(actor, arguments);
    } catch (Throwable t) {
        handler.getExceptionHandler().handleException(t, actor);
    }
}