BreakerFlipRule.cs 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. using Content.Server.Power.Components;
  2. using Content.Server.Power.EntitySystems;
  3. using Content.Server.StationEvents.Components;
  4. using Content.Shared.GameTicking.Components;
  5. using Content.Shared.Station.Components;
  6. using JetBrains.Annotations;
  7. namespace Content.Server.StationEvents.Events;
  8. [UsedImplicitly]
  9. public sealed class BreakerFlipRule : StationEventSystem<BreakerFlipRuleComponent>
  10. {
  11. [Dependency] private readonly ApcSystem _apcSystem = default!;
  12. protected override void Added(EntityUid uid, BreakerFlipRuleComponent component, GameRuleComponent gameRule, GameRuleAddedEvent args)
  13. {
  14. if (!TryComp<StationEventComponent>(uid, out var stationEvent))
  15. return;
  16. var str = Loc.GetString("station-event-breaker-flip-announcement", ("data", Loc.GetString(Loc.GetString($"random-sentience-event-data-{RobustRandom.Next(1, 6)}"))));
  17. stationEvent.StartAnnouncement = str;
  18. base.Added(uid, component, gameRule, args);
  19. }
  20. protected override void Started(EntityUid uid, BreakerFlipRuleComponent component, GameRuleComponent gameRule, GameRuleStartedEvent args)
  21. {
  22. base.Started(uid, component, gameRule, args);
  23. if (!TryGetRandomStation(out var chosenStation))
  24. return;
  25. var stationApcs = new List<Entity<ApcComponent>>();
  26. var query = EntityQueryEnumerator<ApcComponent, TransformComponent>();
  27. while (query.MoveNext(out var apcUid, out var apc, out var xform))
  28. {
  29. if (apc.MainBreakerEnabled && CompOrNull<StationMemberComponent>(xform.GridUid)?.Station == chosenStation)
  30. {
  31. stationApcs.Add((apcUid, apc));
  32. }
  33. }
  34. var toDisable = Math.Min(RobustRandom.Next(3, 7), stationApcs.Count);
  35. if (toDisable == 0)
  36. return;
  37. RobustRandom.Shuffle(stationApcs);
  38. for (var i = 0; i < toDisable; i++)
  39. {
  40. _apcSystem.ApcToggleBreaker(stationApcs[i], stationApcs[i]);
  41. }
  42. }
  43. }