91 lines
2.8 KiB
C#
91 lines
2.8 KiB
C#
using Content.Server.Administration;
|
|
using Content.Shared.Administration;
|
|
using Robust.Server.Player;
|
|
using Robust.Shared.Console;
|
|
using Robust.Shared.Player;
|
|
|
|
namespace Content.Server.Traitor.Uplink.Commands;
|
|
|
|
[AdminCommand(AdminFlags.Admin)]
|
|
public sealed class AddUplinkCommand : LocalizedEntityCommands
|
|
{
|
|
[Dependency] private readonly UplinkSystem _uplinkSystem = default!;
|
|
[Dependency] private readonly IPlayerManager _playerManager = default!;
|
|
|
|
public override string Command => "adduplink";
|
|
|
|
public override void Execute(IConsoleShell shell, string argStr, string[] args)
|
|
{
|
|
if (args.Length > 3)
|
|
{
|
|
shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
|
|
return;
|
|
}
|
|
|
|
ICommonSession? session;
|
|
if (args.Length > 0)
|
|
{
|
|
// Get player entity
|
|
if (!_playerManager.TryGetSessionByUsername(args[0], out session))
|
|
{
|
|
shell.WriteLine(Loc.GetString("shell-target-player-does-not-exist"));
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
session = shell.Player;
|
|
|
|
if (session?.AttachedEntity is not { } user)
|
|
{
|
|
shell.WriteLine(Loc.GetString("add-uplink-command-error-1"));
|
|
return;
|
|
}
|
|
|
|
// Get target item
|
|
EntityUid? uplinkEntity = null;
|
|
if (args.Length >= 2)
|
|
{
|
|
if (!int.TryParse(args[1], out var itemId))
|
|
{
|
|
shell.WriteLine(Loc.GetString("shell-entity-uid-must-be-number"));
|
|
return;
|
|
}
|
|
|
|
var eNet = new NetEntity(itemId);
|
|
|
|
if (!EntityManager.TryGetEntity(eNet, out var eUid))
|
|
{
|
|
shell.WriteLine(Loc.GetString("shell-invalid-entity-id"));
|
|
return;
|
|
}
|
|
|
|
uplinkEntity = eUid;
|
|
}
|
|
|
|
var isDiscounted = false;
|
|
if (args.Length >= 3)
|
|
{
|
|
if (!bool.TryParse(args[2], out isDiscounted))
|
|
{
|
|
shell.WriteLine(Loc.GetString("shell-invalid-bool"));
|
|
return;
|
|
}
|
|
}
|
|
|
|
// Finally add uplink
|
|
if (!_uplinkSystem.AddUplink(user, 20, uplinkEntity: uplinkEntity, giveDiscounts: isDiscounted))
|
|
shell.WriteLine(Loc.GetString("add-uplink-command-error-2"));
|
|
}
|
|
|
|
public override CompletionResult GetCompletion(IConsoleShell shell, string[] args)
|
|
{
|
|
return args.Length switch
|
|
{
|
|
1 => CompletionResult.FromHintOptions(CompletionHelper.SessionNames(), Loc.GetString("add-uplink-command-completion-1")),
|
|
2 => CompletionResult.FromHint(Loc.GetString("add-uplink-command-completion-2")),
|
|
3 => CompletionResult.FromHint(Loc.GetString("add-uplink-command-completion-3")),
|
|
_ => CompletionResult.Empty,
|
|
};
|
|
}
|
|
}
|