77 lines
2.0 KiB
C#
77 lines
2.0 KiB
C#
using System.Linq;
|
|
using Content.Server.Administration;
|
|
using Content.Shared.Administration;
|
|
using JetBrains.Annotations;
|
|
using Robust.Shared.Console;
|
|
|
|
namespace Content.Server.Nuke.Commands;
|
|
|
|
[UsedImplicitly]
|
|
[AdminCommand(AdminFlags.Fun)]
|
|
public sealed class ToggleNukeCommand : LocalizedCommands
|
|
{
|
|
[Dependency] private readonly IEntityManager _entManager = default!;
|
|
|
|
public override string Command => "nukearm";
|
|
|
|
public override void Execute(IConsoleShell shell, string argStr, string[] args)
|
|
{
|
|
EntityUid bombUid;
|
|
NukeComponent? bomb = null;
|
|
|
|
if (args.Length >= 2)
|
|
{
|
|
if (!EntityUid.TryParse(args[1], out bombUid))
|
|
{
|
|
shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
var query = _entManager.EntityQueryEnumerator<NukeComponent>();
|
|
|
|
while (query.MoveNext(out bombUid, out bomb))
|
|
{
|
|
break;
|
|
}
|
|
|
|
if (bomb == null)
|
|
{
|
|
shell.WriteError(Loc.GetString("cmd-nukearm-not-found"));
|
|
return;
|
|
}
|
|
}
|
|
|
|
var nukeSys = _entManager.System<NukeSystem>();
|
|
|
|
if (args.Length >= 1)
|
|
{
|
|
if (!float.TryParse(args[0], out var timer))
|
|
{
|
|
shell.WriteError("shell-argument-must-be-number");
|
|
return;
|
|
}
|
|
|
|
nukeSys.SetRemainingTime(bombUid, timer, bomb);
|
|
}
|
|
|
|
nukeSys.ToggleBomb(bombUid, bomb);
|
|
}
|
|
|
|
public override CompletionResult GetCompletion(IConsoleShell shell, string[] args)
|
|
{
|
|
if (args.Length == 1)
|
|
{
|
|
return CompletionResult.FromHint(Loc.GetString(Loc.GetString("cmd-nukearm-1-help")));
|
|
}
|
|
|
|
if (args.Length == 2)
|
|
{
|
|
return CompletionResult.FromHintOptions(CompletionHelper.Components<NukeComponent>(args[1]), Loc.GetString("cmd-nukearm-2-help"));
|
|
}
|
|
|
|
return CompletionResult.Empty;
|
|
}
|
|
}
|