tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add autocomplete to nukearm (#14819)

Browse Source
This commit is contained in:
metalgearsloth
2023-03-24 13:50:49 +11:00
committed by GitHub
parent 65acae15c0
commit 427e5c95ca
2 changed files with 67 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

106
Content.Server/Nuke/Commands/ToggleNukeCommand.cs
View File

@@ -4,57 +4,73 @@ using Content.Shared.Administration;
using JetBrains.Annotations;
using Robust.Shared.Console;
namespace Content.Server.Nuke.Commands
namespace Content.Server.Nuke.Commands;
[UsedImplicitly]
[AdminCommand(AdminFlags.Fun)]
public sealed class ToggleNukeCommand : LocalizedCommands
{
[UsedImplicitly]
[AdminCommand(AdminFlags.Fun)]
public sealed class ToggleNukeCommand : IConsoleCommand
[Dependency] private readonly IEntityManager _entManager = default!;
public override string Command => "nukearm";
public override void Execute(IConsoleShell shell, string argStr, string[] args)
{
public string Command => "nukearm";
public string Description => "Toggle nuclear bomb timer. You can set timer directly. Uid is optional.";
public string Help => "nukearm <timer> <uid>";
EntityUid bombUid;
NukeComponent? bomb = null;
public void Execute(IConsoleShell shell, string argStr, string[] args)
if (args.Length >= 2)
{
EntityUid bombUid;
NukeComponent? bomb = null;
if (args.Length >= 2)
if (!EntityUid.TryParse(args[1], out bombUid))
{
if (!EntityUid.TryParse(args[1], out bombUid))
{
shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
return;
}
shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
return;
}
else
{
var entManager = IoCManager.Resolve<IEntityManager>();
var bombs = entManager.EntityQuery<NukeComponent>();
bomb = bombs.FirstOrDefault();
if (bomb == null)
{
shell.WriteError("Can't find any entity with a NukeComponent");
return;
}
bombUid = bomb.Owner;
}
var nukeSys = EntitySystem.Get<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);
}
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;
}
}