    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 java.util.ArrayList;
    027import java.util.Arrays;
    028import java.util.Set;
    029import java.util.stream.Collectors;
    030import java.util.stream.Stream;
    031
    032import com.velocitypowered.api.command.CommandSource;
    033import com.velocitypowered.api.proxy.Player;
    034import com.velocitypowered.api.proxy.ProxyServer;
    035
    036import co.aikar.commands.apachecommonslang.ApacheCommonsLangUtil;
    037import net.kyori.text.format.TextColor;
    038import net.kyori.text.format.TextDecoration;
    039import net.kyori.text.format.TextFormat;
    040
    041public class VelocityCommandCompletions extends CommandCompletions<VelocityCommandCompletionContext> {
    042
    043    public VelocityCommandCompletions(ProxyServer server, CommandManager manager) {
    044        super(manager);
    045        registerAsyncCompletion("chatcolors", c -> {
    046            Stream<TextFormat> colors = Stream.of(TextColor.values());
    047            if (!c.hasConfig("colorsonly")) {
    048                colors = Stream.concat(colors, Stream.of(TextDecoration.values()));
    049            }
    050            String filter = c.getConfig("filter");
    051            if (filter != null) {
    052                Set<String> filters = Arrays.stream(ACFPatterns.COLON.split(filter)).map(ACFUtil::simplifyString)
    053                        .collect(Collectors.toSet());
    054
    055                colors = colors.filter(color -> filters.contains(ACFUtil.simplifyString(color.toString())));
    056            }
    057
    058            return colors.map(color -> ACFUtil.simplifyString(color.toString())).collect(Collectors.toList());
    059        });
    060        registerCompletion("players", c -> {
    061            CommandSource sender = c.getSender();
    062            ACFVelocityUtil.validate(sender, "Sender cannot be null");
    063            String input = c.getInput();
    064
    065            ArrayList<String> matchedPlayers = new ArrayList<>();
    066            for (Player player : server.getAllPlayers()) {
    067                String name = player.getUsername();
    068                if (ApacheCommonsLangUtil.startsWithIgnoreCase(name, input)) {
    069                    matchedPlayers.add(name);
    070                }
    071            }
    072
    073            matchedPlayers.sort(String.CASE_INSENSITIVE_ORDER);
    074            return matchedPlayers;
    075        });
    076    }
    077}