    001/*
    002 * Copyright (c) 2016-2017 Daniel Ennis (Aikar) - MIT License
    003 *
    004 *  Permission is hereby granted, free of charge, to any person obtaining
    005 *  a copy of this software and associated documentation files (the
    006 *  "Software"), to deal in the Software without restriction, including
    007 *  without limitation the rights to use, copy, modify, merge, publish,
    008 *  distribute, sublicense, and/or sell copies of the Software, and to
    009 *  permit persons to whom the Software is furnished to do so, subject to
    010 *  the following conditions:
    011 *
    012 *  The above copyright notice and this permission notice shall be
    013 *  included in all copies or substantial portions of the Software.
    014 *
    015 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
    016 *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
    017 *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
    018 *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
    019 *  LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
    020 *  OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
    021 *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
    022 */
    023
    024package co.aikar.commands;
    025
    026import co.aikar.commands.apachecommonslang.ApacheCommonsLangUtil;
    027import net.md_5.bungee.api.ChatColor;
    028import net.md_5.bungee.api.CommandSender;
    029import net.md_5.bungee.api.ProxyServer;
    030import net.md_5.bungee.api.connection.ProxiedPlayer;
    031
    032import java.util.ArrayList;
    033import java.util.Arrays;
    034import java.util.Set;
    035import java.util.stream.Collectors;
    036import java.util.stream.Stream;
    037
    038public class BungeeCommandCompletions extends CommandCompletions<BungeeCommandCompletionContext> {
    039
    040    public BungeeCommandCompletions(CommandManager manager) {
    041        super(manager);
    042        registerAsyncCompletion("chatcolors", c -> {
    043            Stream<ChatColor> colors = Stream.of(ChatColor.values());
    044            if (c.hasConfig("colorsonly")) {
    045                colors = colors.filter(color -> color.ordinal() <= 0xF);
    046            }
    047            String filter = c.getConfig("filter");
    048            if (filter != null) {
    049                Set<String> filters = Arrays.stream(ACFPatterns.COLON.split(filter))
    050                        .map(ACFUtil::simplifyString).collect(Collectors.toSet());
    051
    052                colors = colors.filter(color -> filters.contains(ACFUtil.simplifyString(color.name())));
    053            }
    054
    055            return colors.map(color -> ACFUtil.simplifyString(color.name())).collect(Collectors.toList());
    056        });
    057        registerCompletion("players", c -> {
    058            CommandSender sender = c.getSender();
    059            ACFBungeeUtil.validate(sender, "Sender cannot be null");
    060            String input = c.getInput();
    061
    062            ArrayList<String> matchedPlayers = new ArrayList<>();
    063            for (ProxiedPlayer player : ProxyServer.getInstance().getPlayers()) {
    064                String name = player.getName();
    065                if (ApacheCommonsLangUtil.startsWithIgnoreCase(name, input)) {
    066                    matchedPlayers.add(name);
    067                }
    068            }
    069
    070            matchedPlayers.sort(String.CASE_INSENSITIVE_ORDER);
    071            return matchedPlayers;
    072        });
    073    }
    074}