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

Fix nuke command exception, add autocomplete (#11828)

Browse Source 
* fix nuke command exception, add autocomplete

* caches entity manager
This commit is contained in:
Flipp Syder
2022-10-10 22:35:34 -07:00
committed by GitHub
parent 9f9691511a
commit ad64d0330c
1 changed files with 34 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
Content.Server/Nuke/Commands/SendNukeCodesCommand.cs
View File

@@ -1,4 +1,6 @@
using Content.Server.Administration; using System.Linq;
using Content.Server.Administration;
using Content.Server.Station.Systems;
using Content.Shared.Administration; using Content.Shared.Administration;
using JetBrains.Annotations; using JetBrains.Annotations;
using Robust.Shared.Console; using Robust.Shared.Console;
@@ -13,19 +15,48 @@ namespace Content.Server.Nuke.Commands
public string Description => "Send nuke codes to a station's communication consoles"; public string Description => "Send nuke codes to a station's communication consoles";
public string Help => "nukecodes [station EntityUid]"; public string Help => "nukecodes [station EntityUid]";
[Dependency] private readonly IEntityManager _entityManager = default!;
public SendNukeCodesCommand()
{
IoCManager.InjectDependencies(this);
}
public void Execute(IConsoleShell shell, string argStr, string[] args) public void Execute(IConsoleShell shell, string argStr, string[] args)
{ {
if (args.Length != 1) if (args.Length != 1)
{ {
shell.WriteError("shell-need-exactly-one-argument"); shell.WriteError(Loc.GetString("shell-need-exactly-one-argument"));
return;
} }
if (!EntityUid.TryParse(args[0], out var uid)) if (!EntityUid.TryParse(args[0], out var uid))
{ {
shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number")); shell.WriteError(Loc.GetString("shell-entity-uid-must-be-number"));
return;
} }
IoCManager.Resolve<EntityManager>().System<NukeCodePaperSystem>().SendNukeCodes(uid); _entityManager.System<NukeCodePaperSystem>().SendNukeCodes(uid);
}
public CompletionResult GetCompletion(IConsoleShell shell, string[] args)
{
if (args.Length != 1)
{
return CompletionResult.Empty;
}
var stations = _entityManager
.System<StationSystem>()
.Stations
.Select(station =>
{
var meta = _entityManager.GetComponent<MetaDataComponent>(station);
return new CompletionOption(station.ToString(), meta.EntityName);
});
return CompletionResult.FromHintOptions(stations, null);
} }
} }
} }