SuggestionHelper.java

package com.sk89q.commandbook.util.suggestion;

import com.sk89q.commandbook.CommandBook;
import com.sk89q.commandbook.util.InputUtil;
import com.sk89q.minecraft.util.commands.CommandException;
import org.bukkit.ChatColor;
import org.bukkit.entity.Player;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

public class SuggestionHelper {
    private SuggestionHelper() { }

    public static void trialAddPlayerSuggestion(List<String> suggestions, String trailText) {
        try {
            InputUtil.PlayerParser.matchPlayers(CommandBook.server().getConsoleSender(), trailText);
            suggestions.add(trailText);
        } catch (CommandException ignored) { }
    }

    public static void addPlayerNameSuggestions(List<String> suggestions, String input) {
        Collection<? extends Player> players = CommandBook.server().getOnlinePlayers();
        boolean useDisplayNames = CommandBook.inst().lookupWithDisplayNames;

        List<String> secondarySuggestions = new ArrayList<>();

        for (Player player : players) {
            String playerName = player.getName();

            if (playerName.startsWith(input)) {
                suggestions.add(playerName);
            } else {
                secondarySuggestions.add(playerName);
            }

            String displayName = ChatColor.stripColor(player.getDisplayName());
            if (useDisplayNames && !playerName.equals(displayName)) {
                if (displayName.startsWith(input)) {
                    suggestions.add(displayName);
                } else {
                    secondarySuggestions.add(displayName);
                }
            }
        }

        suggestions.addAll(secondarySuggestions);
    }
}