123 lines
4.0 KiB
C#
123 lines
4.0 KiB
C#
using System.Linq;
|
|
using System.Text;
|
|
using Content.Server.Database;
|
|
using Content.Shared.Administration;
|
|
using Robust.Shared.Console;
|
|
|
|
namespace Content.Server.Administration.Commands;
|
|
|
|
[AdminCommand(AdminFlags.Admin)]
|
|
public sealed class BanExemptionUpdateCommand : LocalizedCommands
|
|
{
|
|
[Dependency] private readonly IServerDbManager _dbManager = default!;
|
|
[Dependency] private readonly IPlayerLocator _playerLocator = default!;
|
|
|
|
public override string Command => "ban_exemption_update";
|
|
|
|
public override async void Execute(IConsoleShell shell, string argStr, string[] args)
|
|
{
|
|
if (args.Length < 2)
|
|
{
|
|
shell.WriteError(LocalizationManager.GetString("cmd-ban_exemption_update-nargs"));
|
|
return;
|
|
}
|
|
|
|
var flags = ServerBanExemptFlags.None;
|
|
for (var i = 1; i < args.Length; i++)
|
|
{
|
|
var arg = args[i];
|
|
if (!Enum.TryParse<ServerBanExemptFlags>(arg, ignoreCase: true, out var flag))
|
|
{
|
|
shell.WriteError(LocalizationManager.GetString("cmd-ban_exemption_update-invalid-flag", ("flag", arg)));
|
|
return;
|
|
}
|
|
|
|
flags |= flag;
|
|
}
|
|
|
|
var player = args[0];
|
|
var playerData = await _playerLocator.LookupIdByNameOrIdAsync(player);
|
|
if (playerData == null)
|
|
{
|
|
shell.WriteError(LocalizationManager.GetString("cmd-ban_exemption_update-locate", ("player", player)));
|
|
return;
|
|
}
|
|
|
|
await _dbManager.UpdateBanExemption(playerData.UserId, flags);
|
|
shell.WriteLine(LocalizationManager.GetString(
|
|
"cmd-ban_exemption_update-success",
|
|
("player", player),
|
|
("uid", playerData.UserId)));
|
|
}
|
|
|
|
public override CompletionResult GetCompletion(IConsoleShell shell, string[] args)
|
|
{
|
|
if (args.Length == 1)
|
|
return CompletionResult.FromHint(LocalizationManager.GetString("cmd-ban_exemption_get-arg-player"));
|
|
|
|
return CompletionResult.FromHintOptions(
|
|
Enum.GetNames<ServerBanExemptFlags>(),
|
|
LocalizationManager.GetString("cmd-ban_exemption_update-arg-flag"));
|
|
}
|
|
}
|
|
|
|
[AdminCommand(AdminFlags.Admin)]
|
|
public sealed class BanExemptionGetCommand : LocalizedCommands
|
|
{
|
|
[Dependency] private readonly IServerDbManager _dbManager = default!;
|
|
[Dependency] private readonly IPlayerLocator _playerLocator = default!;
|
|
|
|
public override string Command => "ban_exemption_get";
|
|
|
|
public override async void Execute(IConsoleShell shell, string argStr, string[] args)
|
|
{
|
|
if (args.Length != 1)
|
|
{
|
|
shell.WriteError(LocalizationManager.GetString("cmd-ban_exemption_get-nargs"));
|
|
return;
|
|
}
|
|
|
|
var player = args[0];
|
|
var playerData = await _playerLocator.LookupIdByNameOrIdAsync(player);
|
|
if (playerData == null)
|
|
{
|
|
shell.WriteError(LocalizationManager.GetString("cmd-ban_exemption_update-locate", ("player", player)));
|
|
return;
|
|
}
|
|
|
|
var flags = await _dbManager.GetBanExemption(playerData.UserId);
|
|
if (flags == ServerBanExemptFlags.None)
|
|
{
|
|
shell.WriteLine(LocalizationManager.GetString("cmd-ban_exemption_get-none"));
|
|
return;
|
|
}
|
|
|
|
var joined = new StringBuilder();
|
|
var first = true;
|
|
for (var i = 0; i < sizeof(ServerBanExemptFlags) * 8; i++)
|
|
{
|
|
var mask = (ServerBanExemptFlags) (1 << i);
|
|
if ((mask & flags) == 0)
|
|
break;
|
|
|
|
if (!first)
|
|
joined.Append(", ");
|
|
first = false;
|
|
|
|
joined.Append(mask.ToString());
|
|
}
|
|
|
|
shell.WriteLine(LocalizationManager.GetString(
|
|
"cmd-ban_exemption_get-show",
|
|
("flags", joined.ToString())));
|
|
}
|
|
|
|
public override CompletionResult GetCompletion(IConsoleShell shell, string[] args)
|
|
{
|
|
if (args.Length == 1)
|
|
return CompletionResult.FromHint(LocalizationManager.GetString("cmd-ban_exemption_get-arg-player"));
|
|
|
|
return CompletionResult.Empty;
|
|
}
|
|
}
|