Command resolve killing and LEC conversions batch 2 (#38367)

commit progress
This commit is contained in:
Kyle Tyo
2025-06-17 14:01:28 -04:00
committed by GitHub
parent 57babe15ee
commit 0e1ff2644f
18 changed files with 130 additions and 126 deletions

View File

@@ -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"));
}
}
}