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