tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement ECS alternative for IDragDropOn and fully ECS disposal units (#6380)

Browse Source 
Co-authored-by: metalgearsloth <comedian_vs_clown@hotmail.com>
This commit is contained in:
Julian Giebel
2022-01-30 17:53:22 +01:00
committed by GitHub
parent 1bece1c1ba
commit 305cff2659
10 changed files with 147 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Content.Client/Disposal/Components/DisposalUnitComponent.cs
View File

@@ -17,10 +17,5 @@ namespace Content.Client.Disposal.Components
RecentlyEjected = state.RecentlyEjected; RecentlyEjected = state.RecentlyEjected;
} }
public override bool DragDropOn(DragDropEvent eventArgs)
{
return false;
}
} }
} }

4
Content.Client/DragDrop/DragDropSystem.cs
View File

@@ -30,7 +30,7 @@ namespace Content.Client.DragDrop
/// Handles clientside drag and drop logic /// Handles clientside drag and drop logic
/// </summary> /// </summary>
[UsedImplicitly] [UsedImplicitly]
public class DragDropSystem : EntitySystem public class DragDropSystem : SharedDragDropSystem
{ {
[Dependency] private readonly IStateManager _stateManager = default!; [Dependency] private readonly IStateManager _stateManager = default!;
[Dependency] private readonly IInputManager _inputManager = default!; [Dependency] private readonly IInputManager _inputManager = default!;
@@ -433,7 +433,7 @@ namespace Content.Client.DragDrop
return false; return false;
} }
bool? valid = null; var valid = CheckDragDropOn(eventArgs);
foreach (var comp in EntityManager.GetComponents<IDragDropOn>(eventArgs.Target)) foreach (var comp in EntityManager.GetComponents<IDragDropOn>(eventArgs.Target))
{ {

10
Content.IntegrationTests/Tests/Disposal/DisposalUnitTest.cs
View File

@@ -53,7 +53,7 @@ namespace Content.IntegrationTests.Tests.Disposal
{ {
foreach (var entity in entities) foreach (var entity in entities)
{ {
Assert.That(unit.ContainedEntities.Contains(entity), Is.EqualTo(result)); Assert.That(unit.Container.ContainedEntities.Contains(entity), Is.EqualTo(result));
} }
} }
@@ -65,11 +65,11 @@ namespace Content.IntegrationTests.Tests.Disposal
private void Flush(DisposalUnitComponent unit, bool result, params EntityUid[] entities) private void Flush(DisposalUnitComponent unit, bool result, params EntityUid[] entities)
{ {
Assert.That(unit.ContainedEntities, Is.SupersetOf(entities)); Assert.That(unit.Container.ContainedEntities, Is.SupersetOf(entities));
Assert.That(entities.Length, Is.EqualTo(unit.ContainedEntities.Count)); Assert.That(entities.Length, Is.EqualTo(unit.Container.ContainedEntities.Count));
Assert.That(result, Is.EqualTo(EntitySystem.Get<DisposalUnitSystem>().TryFlush(unit))); Assert.That(result, Is.EqualTo(EntitySystem.Get<DisposalUnitSystem>().TryFlush(unit)));
Assert.That(result || entities.Length == 0, Is.EqualTo(unit.ContainedEntities.Count == 0)); Assert.That(result || entities.Length == 0, Is.EqualTo(unit.Container.ContainedEntities.Count == 0));
} }
private const string Prototypes = @" private const string Prototypes = @"
@@ -194,7 +194,7 @@ namespace Content.IntegrationTests.Tests.Disposal
// Remove power need // Remove power need
Assert.True(entityManager.TryGetComponent(disposalUnit, out ApcPowerReceiverComponent power)); Assert.True(entityManager.TryGetComponent(disposalUnit, out ApcPowerReceiverComponent power));
power!.NeedsPower = false; power!.NeedsPower = false;
Assert.True(unit.Powered); unit.Powered = true;//Power state changed event doesn't get fired smh
// Flush with a mob and an item // Flush with a mob and an item
Flush(unit, true, human, wrench); Flush(unit, true, human, wrench);

2
Content.Server/Disposal/Tube/Components/DisposalEntryComponent.cs
View File

@@ -26,7 +26,7 @@ namespace Content.Server.Disposal.Tube.Components
var holder = _entMan.SpawnEntity(HolderPrototypeId, _entMan.GetComponent<TransformComponent>(Owner).MapPosition); var holder = _entMan.SpawnEntity(HolderPrototypeId, _entMan.GetComponent<TransformComponent>(Owner).MapPosition);
var holderComponent = _entMan.GetComponent<DisposalHolderComponent>(holder); var holderComponent = _entMan.GetComponent<DisposalHolderComponent>(holder);
foreach (var entity in from.ContainedEntities.ToArray()) foreach (var entity in from.Container.ContainedEntities.ToArray())
{ {
holderComponent.TryInsert(entity); holderComponent.TryInsert(entity);
} }

88
Content.Server/Disposal/Unit/Components/DisposalUnitComponent.cs
View File

@@ -2,20 +2,12 @@ using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Threading; using System.Threading;
using Content.Server.Atmos; using Content.Server.Atmos;
using Content.Server.Disposal.Unit.EntitySystems;
using Content.Server.Power.Components;
using Content.Server.UserInterface; using Content.Server.UserInterface;
using Content.Shared.ActionBlocker;
using Content.Shared.Acts;
using Content.Shared.Atmos; using Content.Shared.Atmos;
using Content.Shared.Disposal.Components; using Content.Shared.Disposal.Components;
using Content.Shared.DragDrop;
using Robust.Server.GameObjects; using Robust.Server.GameObjects;
using Robust.Shared.Audio;
using Robust.Shared.Containers; using Robust.Shared.Containers;
using Robust.Shared.GameObjects; using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Player;
using Robust.Shared.Serialization.Manager.Attributes; using Robust.Shared.Serialization.Manager.Attributes;
using Robust.Shared.ViewVariables; using Robust.Shared.ViewVariables;
@@ -23,7 +15,7 @@ namespace Content.Server.Disposal.Unit.Components
{ {
[RegisterComponent] [RegisterComponent]
[ComponentReference(typeof(SharedDisposalUnitComponent))] [ComponentReference(typeof(SharedDisposalUnitComponent))]
public class DisposalUnitComponent : SharedDisposalUnitComponent, IGasMixtureHolder, IDestroyAct public class DisposalUnitComponent : SharedDisposalUnitComponent, IGasMixtureHolder
{ {
/// <summary> /// <summary>
/// Last time that an entity tried to exit this disposal unit. /// Last time that an entity tried to exit this disposal unit.
@@ -41,7 +33,7 @@ namespace Content.Server.Disposal.Unit.Components
[ViewVariables(VVAccess.ReadWrite)] [ViewVariables(VVAccess.ReadWrite)]
[DataField("autoEngageTime")] [DataField("autoEngageTime")]
public readonly TimeSpan _automaticEngageTime = TimeSpan.FromSeconds(30); public readonly TimeSpan AutomaticEngageTime = TimeSpan.FromSeconds(30);
[ViewVariables(VVAccess.ReadWrite)] [ViewVariables(VVAccess.ReadWrite)]
[DataField("flushDelay")] [DataField("flushDelay")]
@@ -71,86 +63,14 @@ namespace Content.Server.Disposal.Unit.Components
/// </summary> /// </summary>
[ViewVariables] public Container Container = default!; [ViewVariables] public Container Container = default!;
[ViewVariables] public IReadOnlyList<EntityUid> ContainedEntities => Container.ContainedEntities; [ViewVariables] public bool Powered = false;
[ViewVariables] [ViewVariables] public PressureState State = PressureState.Ready;
public bool Powered =>
!IoCManager.Resolve<IEntityManager>().TryGetComponent(Owner, out ApcPowerReceiverComponent? receiver) ||
receiver.Powered;
[ViewVariables] public PressureState State => Pressure >= 1 ? PressureState.Ready : PressureState.Pressurizing;
[ViewVariables(VVAccess.ReadWrite)] [ViewVariables(VVAccess.ReadWrite)]
public bool Engaged { get; set; } public bool Engaged { get; set; }
[ViewVariables] public BoundUserInterface? UserInterface => Owner.GetUIOrNull(DisposalUnitUiKey.Key);
[DataField("air")] [DataField("air")]
public GasMixture Air { get; set; } = new(Atmospherics.CellVolume); public GasMixture Air { get; set; } = new(Atmospherics.CellVolume);
private bool PlayerCanUse(EntityUid player)
{
var actionBlocker = EntitySystem.Get<ActionBlockerSystem>();
if (!actionBlocker.CanInteract(player) ||
!actionBlocker.CanUse(player))
{
return false;
}
return true;
}
public void OnUiReceiveMessage(ServerBoundUserInterfaceMessage obj)
{
if (obj.Session.AttachedEntity is not {Valid: true} player)
{
return;
}
if (!PlayerCanUse(player))
{
return;
}
if (obj.Message is not UiButtonPressedMessage message)
{
return;
}
switch (message.Button)
{
case UiButton.Eject:
EntitySystem.Get<DisposalUnitSystem>().TryEjectContents(this);
break;
case UiButton.Engage:
EntitySystem.Get<DisposalUnitSystem>().ToggleEngage(this);
break;
case UiButton.Power:
EntitySystem.Get<DisposalUnitSystem>().TogglePower(this);
SoundSystem.Play(Filter.Pvs(Owner), "/Audio/Machines/machine_switch.ogg", Owner, AudioParams.Default.WithVolume(-2f));
break;
default:
throw new ArgumentOutOfRangeException();
}
}
public override bool CanDragDropOn(DragDropEvent eventArgs)
{
// Base is redundant given this already calls the base CanInsert
// If that changes then update this
return EntitySystem.Get<DisposalUnitSystem>().CanInsert(this, eventArgs.Dragged);
}
public override bool DragDropOn(DragDropEvent eventArgs)
{
EntitySystem.Get<DisposalUnitSystem>().TryInsert(Owner, eventArgs.Dragged, eventArgs.User);
return true;
}
void IDestroyAct.OnDestroy(DestructionEventArgs eventArgs)
{
EntitySystem.Get<DisposalUnitSystem>().TryEjectContents(this);
}
} }
} }

89
Content.Server/Disposal/Unit/EntitySystems/DisposalUnitSystem.cs
View File

@@ -9,10 +9,13 @@ using Content.Server.Disposal.Unit.Components;
using Content.Server.DoAfter; using Content.Server.DoAfter;
using Content.Server.Hands.Components; using Content.Server.Hands.Components;
using Content.Server.Power.Components; using Content.Server.Power.Components;
using Content.Server.UserInterface;
using Content.Shared.ActionBlocker; using Content.Shared.ActionBlocker;
using Content.Shared.Acts;
using Content.Shared.Atmos; using Content.Shared.Atmos;
using Content.Shared.Disposal; using Content.Shared.Disposal;
using Content.Shared.Disposal.Components; using Content.Shared.Disposal.Components;
using Content.Shared.DragDrop;
using Content.Shared.Interaction; using Content.Shared.Interaction;
using Content.Shared.Item; using Content.Shared.Item;
using Content.Shared.Movement; using Content.Shared.Movement;
@@ -20,6 +23,7 @@ using Content.Shared.Popups;
using Content.Shared.Throwing; using Content.Shared.Throwing;
using Content.Shared.Verbs; using Content.Shared.Verbs;
using Robust.Server.GameObjects; using Robust.Server.GameObjects;
using Robust.Shared.Audio;
using Robust.Shared.Containers; using Robust.Shared.Containers;
using Robust.Shared.GameObjects; using Robust.Shared.GameObjects;
using Robust.Shared.IoC; using Robust.Shared.IoC;
@@ -27,6 +31,7 @@ using Robust.Shared.Localization;
using Robust.Shared.Log; using Robust.Shared.Log;
using Robust.Shared.Map; using Robust.Shared.Map;
using Robust.Shared.Maths; using Robust.Shared.Maths;
using Robust.Shared.Player;
using Robust.Shared.Random; using Robust.Shared.Random;
namespace Content.Server.Disposal.Unit.EntitySystems namespace Content.Server.Disposal.Unit.EntitySystems
@@ -60,6 +65,8 @@ namespace Content.Server.Disposal.Unit.EntitySystems
SubscribeLocalEvent<DisposalUnitComponent, ActivateInWorldEvent>(HandleActivate); SubscribeLocalEvent<DisposalUnitComponent, ActivateInWorldEvent>(HandleActivate);
SubscribeLocalEvent<DisposalUnitComponent, InteractHandEvent>(HandleInteractHand); SubscribeLocalEvent<DisposalUnitComponent, InteractHandEvent>(HandleInteractHand);
SubscribeLocalEvent<DisposalUnitComponent, InteractUsingEvent>(HandleInteractUsing); SubscribeLocalEvent<DisposalUnitComponent, InteractUsingEvent>(HandleInteractUsing);
SubscribeLocalEvent<DisposalUnitComponent, DragDropEvent>(HandleDragDropOn);
SubscribeLocalEvent<DisposalUnitComponent, DestructionEventArgs>(HandleDestruction);
// Verbs // Verbs
SubscribeLocalEvent<DisposalUnitComponent, GetAlternativeVerbsEvent>(AddFlushEjectVerbs); SubscribeLocalEvent<DisposalUnitComponent, GetAlternativeVerbsEvent>(AddFlushEjectVerbs);
@@ -67,11 +74,14 @@ namespace Content.Server.Disposal.Unit.EntitySystems
// Units // Units
SubscribeLocalEvent<DoInsertDisposalUnitEvent>(DoInsertDisposalUnit); SubscribeLocalEvent<DoInsertDisposalUnitEvent>(DoInsertDisposalUnit);
//UI
SubscribeLocalEvent<DisposalUnitComponent, SharedDisposalUnitComponent.UiButtonPressedMessage>(OnUiButtonPressed);
} }
private void AddFlushEjectVerbs(EntityUid uid, DisposalUnitComponent component, GetAlternativeVerbsEvent args) private void AddFlushEjectVerbs(EntityUid uid, DisposalUnitComponent component, GetAlternativeVerbsEvent args)
{ {
if (!args.CanAccess || !args.CanInteract || component.ContainedEntities.Count == 0) if (!args.CanAccess || !args.CanInteract || component.Container.ContainedEntities.Count == 0)
return; return;
// Verbs to flush the unit // Verbs to flush the unit
@@ -96,7 +106,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
// unwilling to accept that this is where they belong and don't want to accidentally climb inside. // unwilling to accept that this is where they belong and don't want to accidentally climb inside.
if (!args.CanAccess || if (!args.CanAccess ||
!args.CanInteract || !args.CanInteract ||
component.ContainedEntities.Contains(args.User) || component.Container.ContainedEntities.Contains(args.User) ||
!_actionBlockerSystem.CanMove(args.User)) !_actionBlockerSystem.CanMove(args.User))
return; return;
@@ -142,6 +152,35 @@ namespace Content.Server.Disposal.Unit.EntitySystems
} }
#region UI Handlers #region UI Handlers
private void OnUiButtonPressed(EntityUid uid, DisposalUnitComponent component, SharedDisposalUnitComponent.UiButtonPressedMessage args)
{
if (args.Session.AttachedEntity is not {Valid: true} player)
{
return;
}
if (!_actionBlockerSystem.CanInteract(player) || !_actionBlockerSystem.CanUse(player))
{
return;
}
switch (args.Button)
{
case SharedDisposalUnitComponent.UiButton.Eject:
TryEjectContents(component);
break;
case SharedDisposalUnitComponent.UiButton.Engage:
ToggleEngage(component);
break;
case SharedDisposalUnitComponent.UiButton.Power:
TogglePower(component);
SoundSystem.Play(Filter.Pvs(component.Owner), "/Audio/Machines/machine_switch.ogg", component.Owner, AudioParams.Default.WithVolume(-2f));
break;
default:
throw new ArgumentOutOfRangeException();
}
}
public void ToggleEngage(DisposalUnitComponent component) public void ToggleEngage(DisposalUnitComponent component)
{ {
component.Engaged ^= true; component.Engaged ^= true;
@@ -180,7 +219,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
if (IsValidInteraction(args)) if (IsValidInteraction(args))
{ {
component.UserInterface?.Open(actor.PlayerSession); component.Owner.GetUIOrNull(SharedDisposalUnitComponent.DisposalUnitUiKey.Key)?.Open(actor.PlayerSession);
} }
} }
@@ -191,7 +230,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
// Duplicated code here, not sure how else to get actor inside to make UserInterface happy. // Duplicated code here, not sure how else to get actor inside to make UserInterface happy.
if (!IsValidInteraction(args)) return; if (!IsValidInteraction(args)) return;
component.UserInterface?.Open(actor.PlayerSession); component.Owner.GetUIOrNull(SharedDisposalUnitComponent.DisposalUnitUiKey.Key)?.Open(actor.PlayerSession);
args.Handled = true; args.Handled = true;
} }
@@ -230,11 +269,6 @@ namespace Content.Server.Disposal.Unit.EntitySystems
{ {
component.Container = component.Owner.EnsureContainer<Container>(component.Name); component.Container = component.Owner.EnsureContainer<Container>(component.Name);
if (component.UserInterface != null)
{
component.UserInterface.OnReceiveMessage += component.OnUiReceiveMessage;
}
UpdateInterface(component, component.Powered); UpdateInterface(component, component.Powered);
if (!EntityManager.HasComponent<AnchorableComponent>(component.Owner)) if (!EntityManager.HasComponent<AnchorableComponent>(component.Owner))
@@ -250,7 +284,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
component.Container.ForceRemove(entity); component.Container.ForceRemove(entity);
} }
component.UserInterface?.CloseAll(); component.Owner.GetUIOrNull(SharedDisposalUnitComponent.DisposalUnitUiKey.Key)?.CloseAll();
component.AutomaticEngageToken?.Cancel(); component.AutomaticEngageToken?.Cancel();
component.AutomaticEngageToken = null; component.AutomaticEngageToken = null;
@@ -264,6 +298,8 @@ namespace Content.Server.Disposal.Unit.EntitySystems
if (!component.Running) if (!component.Running)
return; return;
component.Powered = args.Powered;
// TODO: Need to check the other stuff. // TODO: Need to check the other stuff.
if (!args.Powered) if (!args.Powered)
{ {
@@ -318,6 +354,16 @@ namespace Content.Server.Disposal.Unit.EntitySystems
if (!args.Anchored) if (!args.Anchored)
TryEjectContents(component); TryEjectContents(component);
} }
private void HandleDestruction(EntityUid uid, DisposalUnitComponent component, DestructionEventArgs args)
{
TryEjectContents(component);
}
private void HandleDragDropOn(EntityUid uid, DisposalUnitComponent component, DragDropEvent args)
{
args.Handled = TryInsert(component.Owner, args.Dragged, args.User);
}
#endregion #endregion
/// <summary> /// <summary>
@@ -328,6 +374,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
var oldPressure = component.Pressure; var oldPressure = component.Pressure;
component.Pressure = MathF.Min(1.0f, component.Pressure + PressurePerSecond * frameTime); component.Pressure = MathF.Min(1.0f, component.Pressure + PressurePerSecond * frameTime);
component.State = component.Pressure >= 1 ? SharedDisposalUnitComponent.PressureState.Ready : SharedDisposalUnitComponent.PressureState.Pressurizing;
var state = component.State; var state = component.State;
@@ -399,13 +446,13 @@ namespace Content.Server.Disposal.Unit.EntitySystems
return true; return true;
} }
public void TryInsert(EntityUid unitId, EntityUid toInsertId, EntityUid userId, DisposalUnitComponent? unit = null) public bool TryInsert(EntityUid unitId, EntityUid toInsertId, EntityUid userId, DisposalUnitComponent? unit = null)
{ {
if (!Resolve(unitId, ref unit)) if (!Resolve(unitId, ref unit))
return; return false;
if (!CanInsert(unit, toInsertId)) if (!CanInsert(unit, toInsertId))
return; return false;
var delay = userId == toInsertId ? unit.EntryDelay : unit.DraggedEntryDelay; var delay = userId == toInsertId ? unit.EntryDelay : unit.DraggedEntryDelay;
var ev = new DoInsertDisposalUnitEvent(userId, toInsertId, unitId); var ev = new DoInsertDisposalUnitEvent(userId, toInsertId, unitId);
@@ -413,7 +460,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
if (delay <= 0) if (delay <= 0)
{ {
DoInsertDisposalUnit(ev); DoInsertDisposalUnit(ev);
return; return true;
} }
// Can't check if our target AND disposals moves currently so we'll just check target. // Can't check if our target AND disposals moves currently so we'll just check target.
@@ -429,6 +476,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
}; };
_doAfterSystem.DoAfter(doAfterArgs); _doAfterSystem.DoAfter(doAfterArgs);
return true;
} }
@@ -465,6 +513,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
component.AutomaticEngageToken = null; component.AutomaticEngageToken = null;
component.Pressure = 0; component.Pressure = 0;
component.State = component.Pressure >= 1 ? SharedDisposalUnitComponent.PressureState.Ready : SharedDisposalUnitComponent.PressureState.Pressurizing;
component.Engaged = false; component.Engaged = false;
@@ -479,7 +528,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
{ {
var stateString = Loc.GetString($"{component.State}"); var stateString = Loc.GetString($"{component.State}");
var state = new SharedDisposalUnitComponent.DisposalUnitBoundUserInterfaceState(EntityManager.GetComponent<MetaDataComponent>(component.Owner).EntityName, stateString, EstimatedFullPressure(component), powered, component.Engaged); var state = new SharedDisposalUnitComponent.DisposalUnitBoundUserInterfaceState(EntityManager.GetComponent<MetaDataComponent>(component.Owner).EntityName, stateString, EstimatedFullPressure(component), powered, component.Engaged);
component.UserInterface?.SetState(state); component.Owner.GetUIOrNull(SharedDisposalUnitComponent.DisposalUnitUiKey.Key)?.SetState(state);
} }
private TimeSpan EstimatedFullPressure(DisposalUnitComponent component) private TimeSpan EstimatedFullPressure(DisposalUnitComponent component)
@@ -531,7 +580,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
return; return;
} }
if (component.ContainedEntities.Count > 0) if (component.Container.ContainedEntities.Count > 0)
{ {
appearance.SetData(SharedDisposalUnitComponent.Visuals.Light, SharedDisposalUnitComponent.LightState.Full); appearance.SetData(SharedDisposalUnitComponent.Visuals.Light, SharedDisposalUnitComponent.LightState.Full);
return; return;
@@ -546,7 +595,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
{ {
component.Container.Remove(entity); component.Container.Remove(entity);
if (component.ContainedEntities.Count == 0) if (component.Container.ContainedEntities.Count == 0)
{ {
component.AutomaticEngageToken?.Cancel(); component.AutomaticEngageToken?.Cancel();
component.AutomaticEngageToken = null; component.AutomaticEngageToken = null;
@@ -608,14 +657,14 @@ namespace Content.Server.Disposal.Unit.EntitySystems
/// </summary> /// </summary>
public void TryQueueEngage(DisposalUnitComponent component) public void TryQueueEngage(DisposalUnitComponent component)
{ {
if (component.Deleted || !component.Powered && component.ContainedEntities.Count == 0) if (component.Deleted || !component.Powered && component.Container.ContainedEntities.Count == 0)
{ {
return; return;
} }
component.AutomaticEngageToken = new CancellationTokenSource(); component.AutomaticEngageToken = new CancellationTokenSource();
component.Owner.SpawnTimer(component._automaticEngageTime, () => component.Owner.SpawnTimer(component.AutomaticEngageTime, () =>
{ {
if (!TryFlush(component)) if (!TryFlush(component))
{ {
@@ -630,7 +679,7 @@ namespace Content.Server.Disposal.Unit.EntitySystems
if (EntityManager.TryGetComponent(entity, out ActorComponent? actor)) if (EntityManager.TryGetComponent(entity, out ActorComponent? actor))
{ {
component.UserInterface?.Close(actor.PlayerSession); component.Owner.GetUIOrNull(SharedDisposalUnitComponent.DisposalUnitUiKey.Key)?.Close(actor.PlayerSession);
} }
UpdateVisualState(component); UpdateVisualState(component);

10
Content.Shared/Disposal/Components/SharedDisposalUnitComponent.cs
View File

@@ -9,7 +9,7 @@ using Robust.Shared.Serialization;
namespace Content.Shared.Disposal.Components namespace Content.Shared.Disposal.Components
{ {
[NetworkedComponent] [NetworkedComponent]
public abstract class SharedDisposalUnitComponent : Component, IDragDropOn public abstract class SharedDisposalUnitComponent : Component
{ {
public override string Name => "DisposalUnit"; public override string Name => "DisposalUnit";
@@ -134,13 +134,5 @@ namespace Content.Shared.Disposal.Components
{ {
Key Key
} }
// TODO: Unfortunately these aren't really ECS yet so soontm
public virtual bool CanDragDropOn(DragDropEvent eventArgs)
{
return EntitySystem.Get<SharedDisposalUnitSystem>().CanInsert(this, eventArgs.Dragged);
}
public abstract bool DragDropOn(DragDropEvent eventArgs);
} }
} }

14
Content.Shared/Disposal/SharedDisposalUnitSystem.cs
View File

@@ -1,6 +1,7 @@
using System; using System;
using Content.Shared.Body.Components; using Content.Shared.Body.Components;
using Content.Shared.Disposal.Components; using Content.Shared.Disposal.Components;
using Content.Shared.DragDrop;
using Content.Shared.Item; using Content.Shared.Item;
using Content.Shared.MobState.Components; using Content.Shared.MobState.Components;
using Content.Shared.Throwing; using Content.Shared.Throwing;
@@ -26,10 +27,11 @@ namespace Content.Shared.Disposal
public override void Initialize() public override void Initialize()
{ {
base.Initialize(); base.Initialize();
SubscribeLocalEvent<SharedDisposalUnitComponent, PreventCollideEvent>(HandlePreventCollide); SubscribeLocalEvent<SharedDisposalUnitComponent, PreventCollideEvent>(OnPreventCollide);
SubscribeLocalEvent<SharedDisposalUnitComponent, CanDragDropOnEvent>(OnCanDragDropOn);
} }
private void HandlePreventCollide(EntityUid uid, SharedDisposalUnitComponent component, PreventCollideEvent args) private void OnPreventCollide(EntityUid uid, SharedDisposalUnitComponent component, PreventCollideEvent args)
{ {
var otherBody = args.BodyB.Owner; var otherBody = args.BodyB.Owner;
@@ -47,6 +49,14 @@ namespace Content.Shared.Disposal
} }
} }
private void OnCanDragDropOn(EntityUid uid, SharedDisposalUnitComponent component, CanDragDropOnEvent args)
{
if (args.Handled) return;
args.CanDrop = CanInsert(component, args.Dragged);
args.Handled = true;
}
public virtual bool CanInsert(SharedDisposalUnitComponent component, EntityUid entity) public virtual bool CanInsert(SharedDisposalUnitComponent component, EntityUid entity)
{ {
if (!EntityManager.GetComponent<TransformComponent>(component.Owner).Anchored) if (!EntityManager.GetComponent<TransformComponent>(component.Owner).Anchored)

38
Content.Shared/DragDrop/CanDragDropOnEvent.cs Normal file
View File

@@ -0,0 +1,38 @@
using Robust.Shared.GameObjects;
namespace Content.Shared.DragDrop;
/// <summary>
/// Event that gets send to the target of a drag drop action
/// Mark this event as handled to specify that the entity can be dropped on
/// and set CanDrop to true or false, depending on whether dropping the entity onto the target is actually possible.
/// </summary>
public class CanDragDropOnEvent : HandledEntityEventArgs
{
/// <summary>
/// Entity doing the drag and drop.
/// </summary>
public EntityUid User { get; }
/// <summary>
/// Entity that is being dragged.
/// </summary>
public EntityUid Dragged { get; }
/// <summary>
/// Entity that is being dropped on.
/// </summary>
public EntityUid Target { get; }
/// <summary>
/// If the dragged entity can be dropped on the target.
/// </summary>
public bool CanDrop { get; set; } = false;
public CanDragDropOnEvent(EntityUid user, EntityUid dragged, EntityUid target)
{
User = user;
Dragged = dragged;
Target = target;
}
}

15
Content.Shared/DragDrop/SharedDragDropSystem.cs Normal file
View File

@@ -0,0 +1,15 @@
using Robust.Shared.GameObjects;
namespace Content.Shared.DragDrop;
public abstract class SharedDragDropSystem : EntitySystem
{
protected bool? CheckDragDropOn(DragDropEvent eventArgs)
{
var canDragDropOnEvent = new CanDragDropOnEvent(eventArgs.User, eventArgs.Dragged, eventArgs.Target);
RaiseLocalEvent(eventArgs.Target, canDragDropOnEvent, false);
return canDragDropOnEvent.Handled ? canDragDropOnEvent.CanDrop : null;
}
}