TF-Marriage/src/main/java/com/lenis0012/bukkit/marriage2/internal/MarriageCommandExecutor.java

package com.lenis0012.bukkit.marriage2.internal;

import java.util.Collection;
import java.util.Map;
import java.util.logging.Level;

import com.google.common.collect.Maps;
import org.bukkit.ChatColor;
import org.bukkit.command.CommandExecutor;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Player;

import com.lenis0012.bukkit.marriage2.Marriage;
import com.lenis0012.bukkit.marriage2.commands.Command;

public class MarriageCommandExecutor implements CommandExecutor {
	private final MarriageCore core;
	private final Map<String, Command> commands = Maps.newHashMap();
	
	public MarriageCommandExecutor(MarriageBase core) {
		this.core = (MarriageCore) core;
	}
	
	@Override
	public boolean onCommand(CommandSender sender, org.bukkit.command.Command cmd, String label, String[] args) {
		String subCommand = args.length > 0 ? args[0] : "help";
		Command command = commands.get(subCommand.toLowerCase());
		if(command == null) {
			command = commands.get("marry");
		}
		
		// Assuming that the command is not null now, if it is, then that is a mistake on my side.
		if(args.length >= command.getMinArgs()) {
			if(command.getPermission() == null || sender.hasPermission(command.getPermission())) {
				if(command.isAllowConsole() || sender instanceof Player) {
					command.prepare(sender, args);
					command.execute();
				} else {
					sender.sendMessage(ChatColor.RED + "You must be a player to execute this command.");
				}
			} else {
				sender.sendMessage(ChatColor.RED + "You are not permitted to use this command.");
			}
		} else {
			sender.sendMessage(ChatColor.RED + "You have not specified enough arguments for this command.");
		}
		
		return true;
	}
	
	public void regster(Class<? extends Command> commandClass) {
		try {
			Command command = commandClass.getConstructor(Marriage.class).newInstance(core);
			for(String alias : command.getAliases()) {
				commands.put(alias.toLowerCase(), command);
			}
		} catch(Exception e) {
			core.getLogger().log(Level.SEVERE, "Failed to register sub command", e);
		}
	}
	
	public Collection<Command> getSubCommands() {
		return commands.values();
	}
}