Command resolve killing and LEC conversions batch 2 (#38367)
commit progress
This commit is contained in:
@@ -12,57 +12,53 @@ using Robust.Shared.Console;
|
||||
namespace Content.Server.Singularity
|
||||
{
|
||||
[AdminCommand(AdminFlags.Admin)]
|
||||
public sealed class StartSingularityEngineCommand : IConsoleCommand
|
||||
public sealed class StartSingularityEngineCommand : LocalizedEntityCommands
|
||||
{
|
||||
public string Command => "startsingularityengine";
|
||||
public string Description => "Automatically turns on the particle accelerator and containment field emitters.";
|
||||
public string Help => $"{Command}";
|
||||
[Dependency] private readonly EmitterSystem _emitterSystem = default!;
|
||||
[Dependency] private readonly MultipartMachineSystem _multipartSystem = default!;
|
||||
[Dependency] private readonly ParticleAcceleratorSystem _paSystem = default!;
|
||||
[Dependency] private readonly RadiationCollectorSystem _radCollectorSystem = default!;
|
||||
|
||||
public void Execute(IConsoleShell shell, string argStr, string[] args)
|
||||
public override string Command => "startsingularityengine";
|
||||
|
||||
public override void Execute(IConsoleShell shell, string argStr, string[] args)
|
||||
{
|
||||
if (args.Length != 0)
|
||||
{
|
||||
shell.WriteLine($"Invalid amount of arguments: {args.Length}.\n{Help}");
|
||||
shell.WriteLine(Loc.GetString($"shell-need-exactly-zero-arguments"));
|
||||
return;
|
||||
}
|
||||
|
||||
var entityManager = IoCManager.Resolve<IEntityManager>();
|
||||
var entitySystemManager = IoCManager.Resolve<IEntitySystemManager>();
|
||||
|
||||
// Turn on emitters
|
||||
var emitterQuery = entityManager.EntityQueryEnumerator<EmitterComponent>();
|
||||
var emitterSystem = entitySystemManager.GetEntitySystem<EmitterSystem>();
|
||||
var emitterQuery = EntityManager.EntityQueryEnumerator<EmitterComponent>();
|
||||
while (emitterQuery.MoveNext(out var uid, out var emitterComponent))
|
||||
{
|
||||
//FIXME: This turns on ALL emitters, including APEs. It should only turn on the containment field emitters.
|
||||
emitterSystem.SwitchOn(uid, emitterComponent);
|
||||
_emitterSystem.SwitchOn(uid, emitterComponent);
|
||||
}
|
||||
|
||||
// Turn on radiation collectors
|
||||
var radiationCollectorQuery = entityManager.EntityQueryEnumerator<RadiationCollectorComponent>();
|
||||
var radiationCollectorSystem = entitySystemManager.GetEntitySystem<RadiationCollectorSystem>();
|
||||
var radiationCollectorQuery = EntityManager.EntityQueryEnumerator<RadiationCollectorComponent>();
|
||||
while (radiationCollectorQuery.MoveNext(out var uid, out var radiationCollectorComponent))
|
||||
{
|
||||
radiationCollectorSystem.SetCollectorEnabled(uid, enabled: true, user: null, radiationCollectorComponent);
|
||||
_radCollectorSystem.SetCollectorEnabled(uid, enabled: true, user: null, radiationCollectorComponent);
|
||||
}
|
||||
|
||||
// Setup PA
|
||||
var multipartMachineManager = entitySystemManager.GetEntitySystem<MultipartMachineSystem>();
|
||||
var paSystem = entitySystemManager.GetEntitySystem<ParticleAcceleratorSystem>();
|
||||
var paQuery = entityManager.EntityQueryEnumerator<ParticleAcceleratorControlBoxComponent>();
|
||||
var paQuery = EntityManager.EntityQueryEnumerator<ParticleAcceleratorControlBoxComponent>();
|
||||
while (paQuery.MoveNext(out var paId, out var paControl))
|
||||
{
|
||||
if (!entityManager.TryGetComponent<MultipartMachineComponent>(paId, out var machine))
|
||||
if (!EntityManager.TryGetComponent<MultipartMachineComponent>(paId, out var machine))
|
||||
continue;
|
||||
|
||||
if (!multipartMachineManager.Rescan((paId, machine)))
|
||||
if (!_multipartSystem.Rescan((paId, machine)))
|
||||
continue;
|
||||
|
||||
paSystem.SetStrength(paId, ParticleAcceleratorPowerState.Level0, comp: paControl);
|
||||
paSystem.SwitchOn(paId, comp: paControl);
|
||||
_paSystem.SetStrength(paId, ParticleAcceleratorPowerState.Level0, comp: paControl);
|
||||
_paSystem.SwitchOn(paId, comp: paControl);
|
||||
}
|
||||
|
||||
shell.WriteLine("Done!");
|
||||
shell.WriteLine(Loc.GetString($"shell-command-success"));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user