BetterAmongUs/Commands/SetRoleCommand.cs

#if DEBUG
using BetterAmongUs.Helpers;
using BetterAmongUs.Items.Attributes;
using BetterAmongUs.Modules;
using BetterAmongUs.Patches.Managers;

namespace BetterAmongUs.Commands;

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

    internal SetRoleCommand()
    {
        playerArgument = new PlayerArgument(this);
        roleArgument = new StringArgument(this, "{role}")
        {
            GetArgSuggestions = () => { return RoleManager.Instance.AllRoles.ToArray().Select(role => role.NiceName.ToLower()).ToArray(); }
        };
        Arguments = [playerArgument, roleArgument];
    }
    private PlayerArgument playerArgument { get; }
    private StringArgument roleArgument { get; }

    internal override bool ShowCommand() => GameState.IsHost && BAUPlugin.MyData.HasAll() && BAUPlugin.MyData.IsVerified();

    internal 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.ToArray().FirstOrDefault(r => r.NiceName.ToLower().StartsWith(roleArgument.Arg.ToLower()));
        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