54 lines
1.9 KiB
C#
54 lines
1.9 KiB
C#
using Content.Server._DV.StationEvents.Components;
|
|
using Content.Server.Psionics.Glimmer;
|
|
using Content.Server.Station.Systems;
|
|
using Content.Server.StationEvents.Events;
|
|
using Content.Shared.GameTicking.Components;
|
|
using Content.Shared.Psionics.Glimmer;
|
|
using System.Linq;
|
|
|
|
namespace Content.Server._DV.StationEvents.Events;
|
|
|
|
public sealed class LockProbersRule : StationEventSystem<LockProbersRuleComponent>
|
|
{
|
|
[Dependency] private readonly StationSystem _stationSystem = default!;
|
|
[Dependency] private readonly GlimmerReactiveSystem _glimmerReactiveSystem = default!;
|
|
|
|
protected override void Started(EntityUid uid, LockProbersRuleComponent comp, GameRuleComponent gameRule, GameRuleStartedEvent args)
|
|
{
|
|
base.Started(uid, comp, gameRule, args);
|
|
|
|
if (!TryGetRandomStation(out var station))
|
|
return;
|
|
|
|
var probers = GetProbers(station.Value);
|
|
if (probers.Count == 0)
|
|
{
|
|
Log.Info($"{ToPrettyString(uid):rule} found no probers on station {station}!");
|
|
ForceEndSelf(uid, gameRule);
|
|
return;
|
|
}
|
|
RobustRandom.Shuffle(probers);
|
|
int probersToLock = RobustRandom.Next(1, probers.Count); // Lock a random number of probers, at least one
|
|
foreach (var prober in probers.Take(probersToLock))
|
|
{
|
|
_glimmerReactiveSystem.LockProber(prober);
|
|
}
|
|
}
|
|
|
|
private List<EntityUid> GetProbers(EntityUid station)
|
|
{
|
|
var probers = new List<EntityUid>();
|
|
var query = AllEntityQuery<SharedGlimmerReactiveComponent>();
|
|
while (query.MoveNext(out var uid, out var _))
|
|
{
|
|
var xform = Transform(uid);
|
|
if (xform.GridUid == null)
|
|
continue;
|
|
|
|
if (_stationSystem.GetOwningStation(xform.GridUid.Value) == station)
|
|
probers.Add(uid);
|
|
}
|
|
return probers;
|
|
}
|
|
}
|