BetterAmongUs/Commands/SetRoleCommand.cs

#if DEBUG
using BetterAmongUs.Helpers;
using BetterAmongUs.Modules;
using BetterAmongUs.Patches;

namespace BetterAmongUs.Commands;

public class SetRoleCommand : BaseCommand
{
    public override CommandType Type => CommandType.Debug;
    public override string Name => "setrole";
    public override string Description => "Set another players role for the next game";

    public SetRoleCommand()
    {
        _arguments = new Lazy<BaseArgument[]>(() => new BaseArgument[]
        {
            new PlayerArgument(this),
            new StringArgument(this, "{role}"),
        }); ;
        roleArgument.GetArgSuggestions = () => { return RoleManager.Instance.AllRoles.Select(role => role.NiceName.ToLower()).ToArray(); };
    }
    private readonly Lazy<BaseArgument[]> _arguments;
    public override BaseArgument[]? Arguments => _arguments.Value;

    private PlayerArgument? playerArgument => (PlayerArgument)Arguments[0];
    private StringArgument? roleArgument => (StringArgument)Arguments[1];

    public override bool ShowCommand() => GameState.IsHost && Main.MyData.HasAll() && Main.MyData.IsVerified();

    public override void Run()
    {
        var player = playerArgument.TryGetTarget();
        if (player.IsLocalPlayer())
        {
            CommandErrorText("Unable to use /setrole for self, use /role!");
            return;
        }
        var role = RoleManager.Instance.AllRoles.FirstOrDefault(r => r.NiceName.StartsWith(roleArgument.Arg));
        if (role != null)
        {
            Utils.AddChatPrivate($"Set role to <color={Utils.GetTeamHexColor(role.TeamType)}>{role.NiceName}</color> for the next game!");
            RoleManagerPatch.SetPlayerRole[PlayerControl.LocalPlayer] = role.Role;
        }
        else
        {
            CommandErrorText("Unable to find role");
        }
    }
}
#endif