Fix nuke command exception, add autocomplete (#11828)
* fix nuke command exception, add autocomplete * caches entity manager
This commit is contained in:
@@ -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 JetBrains.Annotations;
|
||||
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 Help => "nukecodes [station EntityUid]";
|
||||
|
||||
[Dependency] private readonly IEntityManager _entityManager = default!;
|
||||
|
||||
public SendNukeCodesCommand()
|
||||
{
|
||||
IoCManager.InjectDependencies(this);
|
||||
}
|
||||
|
||||
public void Execute(IConsoleShell shell, string argStr, string[] args)
|
||||
{
|
||||
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))
|
||||
{
|
||||
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);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user