56 lines
1.8 KiB
C#
56 lines
1.8 KiB
C#
using Content.Server.Commands;
|
|
using Content.Server.Station.Components;
|
|
using Content.Server.Station.Systems;
|
|
using Content.Shared.Administration;
|
|
using Robust.Shared.Console;
|
|
|
|
namespace Content.Server.Administration.Commands.Station;
|
|
|
|
[AdminCommand(AdminFlags.Admin)]
|
|
public sealed class ListStationJobsCommand : IConsoleCommand
|
|
{
|
|
[Dependency] private readonly IEntityManager _entityManager = default!;
|
|
[Dependency] private readonly IEntitySystemManager _entSysManager = default!;
|
|
|
|
public string Command => "lsstationjobs";
|
|
|
|
public string Description => "Lists all jobs on the given station.";
|
|
|
|
public string Help => "lsstationjobs <station id>";
|
|
|
|
public void Execute(IConsoleShell shell, string argStr, string[] args)
|
|
{
|
|
if (args.Length != 1)
|
|
{
|
|
shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
|
|
return;
|
|
}
|
|
|
|
var stationSystem = _entSysManager.GetEntitySystem<StationSystem>();
|
|
var stationJobs = _entSysManager.GetEntitySystem<StationJobsSystem>();
|
|
|
|
if (!EntityUid.TryParse(args[0], out var station) || !_entityManager.HasComponent<StationJobsComponent>(station))
|
|
{
|
|
shell.WriteError(Loc.GetString("shell-argument-station-id-invalid", ("index", 1)));
|
|
return;
|
|
}
|
|
|
|
foreach (var (job, amount) in stationJobs.GetJobs(station))
|
|
{
|
|
var amountText = amount is null ? "Infinite" : amount.ToString();
|
|
shell.WriteLine($"{job}: {amountText}");
|
|
}
|
|
}
|
|
|
|
public CompletionResult GetCompletion(IConsoleShell shell, string[] args)
|
|
{
|
|
if (args.Length == 1)
|
|
{
|
|
var options = ContentCompletionHelper.StationIds(_entityManager);
|
|
return CompletionResult.FromHintOptions(options, "<station id>");
|
|
}
|
|
|
|
return CompletionResult.Empty;
|
|
}
|
|
}
|