using CounterStrikeSharp.API.Modules.Commands;
using GangsAPI.Data;
using GangsAPI.Data.Command;
using GangsAPI.Services.Commands;

namespace Mock;

public class MockCommandManager : ICommandManager {
  private readonly Dictionary<string, ICommand> commands = new();

  public virtual bool RegisterCommand(ICommand command) {
    return command.Aliases.All(alias => commands.TryAdd(alias, command));
  }

  public bool UnregisterCommand(ICommand command) {
    return command.Aliases.All(alias => commands.Remove(alias));
  }

  public async Task<CommandResult> ProcessCommand(PlayerWrapper? executor,
    CommandInfo sourceInfo) {
    var info = new CommandInfoWrapper(sourceInfo);
    return await ProcessCommand(executor, info);
  }

  public async Task<CommandResult> ProcessCommand(PlayerWrapper? executor,
    CommandInfoWrapper sourceInfo) {
    if (sourceInfo.ArgCount == 0) return CommandResult.FAILURE;
    var result = CommandResult.FAILURE;

    if (!commands.TryGetValue(sourceInfo[0], out var command)) {
      sourceInfo.ReplySync("Unknown command: " + sourceInfo[0]);
      return CommandResult.UNKNOWN_COMMAND;
    }

    if (!command.CanExecute(executor)) return CommandResult.NO_PERMISSION;


    await Task.Run(async () => {
      result = await command.Execute(executor, sourceInfo);
    });

    if (result == CommandResult.PLAYER_ONLY)
      sourceInfo.ReplySync("This command can only be executed by a player");

    return result;
  }

  public async Task<CommandResult> ProcessCommand(PlayerWrapper? executor,
    params string[] args) {
    var info = new CommandInfoWrapper(executor, 0, args);
    return await ProcessCommand(executor, info);
  }
}