net.badbird5907.blib.command.BukkitCompleter Maven / Gradle / Ivy

Go to download
package net.badbird5907.blib.command;

import net.badbird5907.blib.bLib;
import org.bukkit.Bukkit;
import org.bukkit.command.Command;
import org.bukkit.command.CommandSender;
import org.bukkit.command.TabCompleter;
import org.bukkit.entity.Player;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.*;
import java.util.Map.Entry;

/**
 * Command Framework - BukkitCompleter 

 * An implementation of the TabCompleter class allowing for multiple tab
 * completers per command
 * 
 * @author minnymin3
 * 
 */
public class BukkitCompleter implements TabCompleter {

	private Map> completers = new HashMap>();

	public void addCompleter(String label, Method m, Object obj) {
		completers.put(label, new AbstractMap.SimpleEntry(m, obj));
	}

	@SuppressWarnings("unchecked")
	@Override
	public List onTabComplete(CommandSender sender, Command command, String label, String[] args) {
		for (int i = args.length; i >= 0; i--) {
			StringBuffer buffer = new StringBuffer();
			buffer.append(label.toLowerCase());
			for (int x = 0; x < i; x++) {
				if (!args[x].equals("") && !args[x].equals(" ")) {
					buffer.append("." + args[x].toLowerCase());
				}
			}
			String cmdLabel = buffer.toString();
			if (completers.containsKey(cmdLabel)) {
				Entry entry = completers.get(cmdLabel);
				try {
					//return (List) entry.getKey().invoke(entry.getValue(), new CommandArgs(sender, command, label, args, cmdLabel.split("\\.").length - 1));
					return (List) entry.getKey().invoke(entry.getValue(),new Sender(sender), args);
				} catch (IllegalArgumentException e) {
					e.printStackTrace();
				} catch (IllegalAccessException e) {
					e.printStackTrace();
				} catch (InvocationTargetException e) {
					e.printStackTrace();
				}
			}else if (bLib.isAutoCompleteCommandsFromUsage() && CommandFramework.getInstance().getOtherMap().containsKey(cmdLabel.toLowerCase())){
				net.badbird5907.blib.command.Command command1 = CommandFramework.getInstance().getOtherMap().get(cmdLabel.toLowerCase());
				if (!command1.usage().isEmpty()){
					String[] options = command1.usage().split("\\s+");
					int already = args.length;
					List list = new ArrayList<>();
					int index = 0;
					for (String option : options) {
						index++;
						if (index < already)
							continue;
						if (option.startsWith("<") && option.endsWith(">")){
							String[] options1 = option.replace("<","").replace(">","").split("/");
							for (String s : options1) {
								if (s.equalsIgnoreCase("player")){
									for (Player onlinePlayer : Bukkit.getOnlinePlayers()) {
										list.add(onlinePlayer.getName());
									}
								}else list.add(s);
							}
						}else{
							if (option.startsWith("[") && option.endsWith("]")){
								String[] options1 = option.replace("[","").replace("]","").split("/");
								for (String s : options1) {
									if (s.equalsIgnoreCase("player")){
										for (Player onlinePlayer : Bukkit.getOnlinePlayers()) {
											list.add(onlinePlayer.getName());
										}
									}else list.add(s);
								}
							}

						}
					}
					return list;
				}
			}
		}
		return null;
	}

}