LogItTabCompleter.java

package io.github.lucaseasedup.logit.command;

import io.github.lucaseasedup.logit.LogItCoreObject;
import io.github.lucaseasedup.logit.account.Account;
import io.github.lucaseasedup.logit.command.hub.HubCommand;
import io.github.lucaseasedup.logit.command.hub.HubCommands;
import io.github.lucaseasedup.logit.storage.Infix;
import io.github.lucaseasedup.logit.storage.SelectorCondition;
import io.github.lucaseasedup.logit.util.Utils;
import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;

public final class LogItTabCompleter extends LogItCoreObject
{
    public List<String> completeHubCommand(String stub)
    {
        if (stub == null)
            throw new IllegalArgumentException();
        
        List<String> suggestions = new ArrayList<>();
        Iterator<HubCommand> commandIt = HubCommands.iterator();
        
        while (commandIt.hasNext() && suggestions.size() < MAX_SUGGESTIONS)
        {
            HubCommand command = commandIt.next();
            
            if (command.getSubcommand().startsWith(stub))
            {
                String[] stubWords = Utils.getWords(stub);
                String[] commandWords = Utils.getWords(command.getSubcommand());
                String suggestion;
                
                // Complete next word.
                if (stubWords.length == 0 || stub.endsWith(" "))
                {
                    suggestion = commandWords[stubWords.length];
                }
                // Complete current word.
                else
                {
                    suggestion = commandWords[stubWords.length - 1];
                }
                
                if (!suggestions.contains(suggestion))
                {
                    suggestions.add(suggestion);
                }
            }
        }
        
        return suggestions;
    }
    
    public List<String> completeUsername(String stub)
    {
        if (stub == null)
            throw new IllegalArgumentException();
        
        List<Account> accounts = getAccountManager().selectAccounts(
                Arrays.asList(keys().username()),
                new SelectorCondition(
                        keys().username(),
                        Infix.STARTS_WITH,
                        stub
                )
        );
        
        if (accounts == null)
            return null;
        
        List<String> suggestions = new ArrayList<>();
        
        for (Account account : accounts)
        {
            suggestions.add(account.getUsername());
        }
        
        Collections.sort(suggestions);
        
        return suggestions.subList(0,
                (suggestions.size() < MAX_SUGGESTIONS)
                        ? suggestions.size() : MAX_SUGGESTIONS);
    }
    
    public List<String> completeBackupFilename(String stub)
    {
        if (stub == null)
            throw new IllegalArgumentException();
        
        File[] backups = getBackupManager().getBackups();
        List<String> suggestions = new ArrayList<>();
        
        for (File backup : backups)
        {
            if (backup.getName().startsWith(stub))
            {
                suggestions.add(backup.getName());
            }
            
            if (suggestions.size() >= MAX_SUGGESTIONS)
            {
                break;
            }
        }
        
        return suggestions;
    }
    
    public List<String> completeConfigProperty(String stub)
    {
        if (stub == null)
            throw new IllegalArgumentException();
        
        List<String> suggestions = new ArrayList<>();
        
        for (String property : getConfig("config.yml").getProperties().keySet())
        {
            if (property.startsWith(stub))
            {
                suggestions.add(property);
                
                if (suggestions.size() >= MAX_SUGGESTIONS)
                {
                    break;
                }
            }
        }
        
        return suggestions;
    }
    
    private static final int MAX_SUGGESTIONS = 16;
}