85 lines
2.7 KiB
C#
85 lines
2.7 KiB
C#
using Content.Server.Commands;
|
|
using Content.Server.Station.Components;
|
|
using Content.Server.Station.Systems;
|
|
using Content.Shared.Administration;
|
|
using Content.Shared.Roles;
|
|
using Robust.Shared.Console;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Server.Administration.Commands.Station;
|
|
|
|
[AdminCommand(AdminFlags.Round)]
|
|
public sealed class AdjustStationJobCommand : IConsoleCommand
|
|
{
|
|
[Dependency] private readonly IEntityManager _entityManager = default!;
|
|
[Dependency] private readonly IEntitySystemManager _entSysManager = default!;
|
|
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
|
|
|
public string Command => "adjstationjob";
|
|
|
|
public string Description => "Adjust the job manifest on a station.";
|
|
|
|
public string Help => "adjstationjob <station id> <job id> <amount>";
|
|
|
|
public void Execute(IConsoleShell shell, string argStr, string[] args)
|
|
{
|
|
if (args.Length != 3)
|
|
{
|
|
shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
|
|
return;
|
|
}
|
|
|
|
var stationJobs = _entSysManager.GetEntitySystem<StationJobsSystem>();
|
|
|
|
if (!EntityUid.TryParse(args[0], out var station) || !_entityManager.HasComponent<StationDataComponent>(station))
|
|
{
|
|
shell.WriteError(Loc.GetString("shell-argument-station-id-invalid", ("index", 1)));
|
|
return;
|
|
}
|
|
|
|
if (!_prototypeManager.TryIndex<JobPrototype>(args[1], out var job))
|
|
{
|
|
shell.WriteError(Loc.GetString("shell-argument-must-be-prototype",
|
|
("index", 2), ("prototypeName", nameof(JobPrototype))));
|
|
return;
|
|
}
|
|
|
|
if (!int.TryParse(args[2], out var amount) || amount < -1)
|
|
{
|
|
shell.WriteError(Loc.GetString("shell-argument-number-must-be-between",
|
|
("index", 3), ("lower", -1), ("upper", int.MaxValue)));
|
|
return;
|
|
}
|
|
|
|
if (amount == -1)
|
|
{
|
|
stationJobs.MakeJobUnlimited(station, job);
|
|
return;
|
|
}
|
|
|
|
stationJobs.TrySetJobSlot(station, job, amount, true);
|
|
}
|
|
|
|
public CompletionResult GetCompletion(IConsoleShell shell, string[] args)
|
|
{
|
|
if (args.Length == 1)
|
|
{
|
|
var options = ContentCompletionHelper.StationIds(_entityManager);
|
|
return CompletionResult.FromHintOptions(options, "<station id>");
|
|
}
|
|
|
|
if (args.Length == 2)
|
|
{
|
|
var options = CompletionHelper.PrototypeIDs<JobPrototype>();
|
|
return CompletionResult.FromHintOptions(options, "<job id>");
|
|
}
|
|
|
|
if (args.Length == 3)
|
|
{
|
|
return CompletionResult.FromHint("<amount>");
|
|
}
|
|
|
|
return CompletionResult.Empty;
|
|
}
|
|
}
|