PlaceholderAPI/src/main/java/me/clip/placeholderapi/commands/PlaceholderCommandRouter.java

package me.clip.placeholderapi.commands;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import me.clip.placeholderapi.PlaceholderAPIPlugin;
import me.clip.placeholderapi.commands.impl.cloud.CommandECloud;
import me.clip.placeholderapi.commands.impl.local.CommandExpansionRegister;
import me.clip.placeholderapi.commands.impl.local.CommandExpansionUnregister;
import me.clip.placeholderapi.commands.impl.local.CommandHelp;
import me.clip.placeholderapi.commands.impl.local.CommandInfo;
import me.clip.placeholderapi.commands.impl.local.CommandList;
import me.clip.placeholderapi.commands.impl.local.CommandParse;
import me.clip.placeholderapi.commands.impl.local.CommandReload;
import me.clip.placeholderapi.commands.impl.local.CommandVersion;
import me.clip.placeholderapi.util.Msg;
import org.bukkit.command.Command;
import org.bukkit.command.CommandExecutor;
import org.bukkit.command.CommandSender;
import org.bukkit.command.TabCompleter;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Unmodifiable;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.stream.Stream;

public final class PlaceholderCommandRouter implements CommandExecutor, TabCompleter
{

	@Unmodifiable
	private static final List<PlaceholderCommand> COMMANDS = ImmutableList.of(new CommandHelp(),
																			  new CommandInfo(),
																			  new CommandList(),
																			  new CommandECloud(),
																			  new CommandParse(),
																			  new CommandReload(),
																			  new CommandVersion(),
																			  new CommandExpansionRegister(),
																			  new CommandExpansionUnregister());


	@NotNull
	private final PlaceholderAPIPlugin            plugin;
	@NotNull
	@Unmodifiable
	private final Map<String, PlaceholderCommand> commands;


	public PlaceholderCommandRouter(@NotNull final PlaceholderAPIPlugin plugin)
	{
		this.plugin = plugin;

		final ImmutableMap.Builder<String, PlaceholderCommand> commands = ImmutableMap.builder();

		for (final PlaceholderCommand command : COMMANDS)
		{
			command.getLabels().forEach(label -> commands.put(label, command));
		}

		this.commands = commands.build();
	}


	@Override
	public boolean onCommand(@NotNull final CommandSender sender, @NotNull final Command command, @NotNull final String alias, @NotNull final String[] args)
	{
		if (args.length == 0)
		{
			final PlaceholderCommand fallback = commands.get("version");
			if (fallback != null)
			{
				fallback.evaluate(plugin, sender, "", Collections.emptyList());
			}

			return true;
		}

		final String             search = args[0].toLowerCase();
		final PlaceholderCommand target = commands.get(search);

		if (target == null)
		{
			Msg.msg(sender, "&cUnknown command &7" + search);
			return true;
		}

		final String permission = target.getPermission();
		if (permission != null && !permission.isEmpty() && !sender.hasPermission(permission))
		{
			Msg.msg(sender, "&cYou do not have permission to do this!");
			return true;
		}

		target.evaluate(plugin, sender, search, Arrays.asList(Arrays.copyOfRange(args, 1, args.length)));

		return true;
	}

	@Override
	public List<String> onTabComplete(@NotNull final CommandSender sender, @NotNull final Command command, @NotNull final String alias, @NotNull final String[] args)
	{
		final List<String> suggestions = new ArrayList<>();

		if (args.length > 1)
		{
			final PlaceholderCommand target = this.commands.get(args[0].toLowerCase());

			if (target != null)
			{
				target.complete(plugin, sender, args[0].toLowerCase(), Arrays.asList(Arrays.copyOfRange(args, 1, args.length)), suggestions);
			}

			return suggestions;
		}

		final Stream<String> targets = PlaceholderCommand.filterByPermission(sender, commands.values().stream()).map(PlaceholderCommand::getLabels).flatMap(Collection::stream);
		PlaceholderCommand.suggestByParameter(targets, suggestions, args.length == 0 ? null : args[0]);

		return suggestions;
	}

}