DoAfter Refactor (#13225)
Co-authored-by: DrSmugleaf <drsmugleaf@gmail.com>
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
using Content.Server.Anomaly.Components;
|
||||
using Content.Server.DoAfter;
|
||||
using Content.Shared.Anomaly;
|
||||
using Content.Shared.Anomaly.Components;
|
||||
using Content.Shared.DoAfter;
|
||||
using Content.Shared.Interaction;
|
||||
using Robust.Server.GameObjects;
|
||||
using Robust.Shared.Utility;
|
||||
@@ -17,8 +17,7 @@ public sealed partial class AnomalySystem
|
||||
{
|
||||
SubscribeLocalEvent<AnomalyScannerComponent, BoundUIOpenedEvent>(OnScannerUiOpened);
|
||||
SubscribeLocalEvent<AnomalyScannerComponent, AfterInteractEvent>(OnScannerAfterInteract);
|
||||
SubscribeLocalEvent<AnomalyScannerComponent, AnomalyScanFinishedEvent>(OnScannerDoAfterFinished);
|
||||
SubscribeLocalEvent<AnomalyScannerComponent, AnomalyScanCancelledEvent>(OnScannerDoAfterCancelled);
|
||||
SubscribeLocalEvent<AnomalyScannerComponent, DoAfterEvent>(OnDoAfter);
|
||||
|
||||
SubscribeLocalEvent<AnomalyShutdownEvent>(OnScannerAnomalyShutdown);
|
||||
SubscribeLocalEvent<AnomalySeverityChangedEvent>(OnScannerAnomalySeverityChanged);
|
||||
@@ -73,38 +72,29 @@ public sealed partial class AnomalySystem
|
||||
|
||||
private void OnScannerAfterInteract(EntityUid uid, AnomalyScannerComponent component, AfterInteractEvent args)
|
||||
{
|
||||
if (component.TokenSource != null)
|
||||
return;
|
||||
|
||||
if (args.Target is not { } target)
|
||||
return;
|
||||
if (!HasComp<AnomalyComponent>(target))
|
||||
return;
|
||||
|
||||
component.TokenSource = new();
|
||||
_doAfter.DoAfter(new DoAfterEventArgs(args.User, component.ScanDoAfterDuration, component.TokenSource.Token, target, uid)
|
||||
_doAfter.DoAfter(new DoAfterEventArgs(args.User, component.ScanDoAfterDuration, target:target, used:uid)
|
||||
{
|
||||
DistanceThreshold = 2f,
|
||||
UsedFinishedEvent = new AnomalyScanFinishedEvent(target, args.User),
|
||||
UsedCancelledEvent = new AnomalyScanCancelledEvent()
|
||||
DistanceThreshold = 2f
|
||||
});
|
||||
}
|
||||
|
||||
private void OnScannerDoAfterFinished(EntityUid uid, AnomalyScannerComponent component, AnomalyScanFinishedEvent args)
|
||||
private void OnDoAfter(EntityUid uid, AnomalyScannerComponent component, DoAfterEvent args)
|
||||
{
|
||||
component.TokenSource = null;
|
||||
if (args.Cancelled || args.Handled || args.Args.Target == null)
|
||||
return;
|
||||
|
||||
Audio.PlayPvs(component.CompleteSound, uid);
|
||||
Popup.PopupEntity(Loc.GetString("anomaly-scanner-component-scan-complete"), uid);
|
||||
UpdateScannerWithNewAnomaly(uid, args.Anomaly, component);
|
||||
UpdateScannerWithNewAnomaly(uid, args.Args.Target.Value, component);
|
||||
|
||||
if (TryComp<ActorComponent>(args.User, out var actor))
|
||||
_ui.TryOpen(uid, AnomalyScannerUiKey.Key, actor.PlayerSession);
|
||||
}
|
||||
if (TryComp<ActorComponent>(args.Args.User, out var actor)) _ui.TryOpen(uid, AnomalyScannerUiKey.Key, actor.PlayerSession);
|
||||
|
||||
private void OnScannerDoAfterCancelled(EntityUid uid, AnomalyScannerComponent component, AnomalyScanCancelledEvent args)
|
||||
{
|
||||
component.TokenSource = null;
|
||||
args.Handled = true;
|
||||
}
|
||||
|
||||
public void UpdateScannerUi(EntityUid uid, AnomalyScannerComponent? component = null)
|
||||
|
||||
Reference in New Issue
Block a user