709 lines
20 KiB
C#
709 lines
20 KiB
C#
#nullable enable
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Content.Server.GameObjects.Components.GUI;
|
|
using Content.Server.GameObjects.Components.Items.Storage;
|
|
using Content.Server.GameObjects.Components.Power.ApcNetComponents;
|
|
using Content.Server.GameObjects.EntitySystems.DoAfter;
|
|
using Content.Server.Interfaces;
|
|
using Content.Server.Interfaces.GameObjects.Components.Items;
|
|
using Content.Server.Utility;
|
|
using Content.Shared.GameObjects.Components.Body;
|
|
using Content.Shared.GameObjects.Components.Disposal;
|
|
using Content.Shared.GameObjects.EntitySystems;
|
|
using Content.Shared.GameObjects.Verbs;
|
|
using Content.Shared.Interfaces.GameObjects.Components;
|
|
using Robust.Server.GameObjects;
|
|
using Robust.Server.GameObjects.Components.Container;
|
|
using Robust.Server.GameObjects.Components.UserInterface;
|
|
using Robust.Server.GameObjects.EntitySystems;
|
|
using Robust.Server.Interfaces.GameObjects;
|
|
using Robust.Shared.Audio;
|
|
using Robust.Shared.Containers;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.GameObjects.Components;
|
|
using Robust.Shared.GameObjects.Components.Transform;
|
|
using Robust.Shared.GameObjects.Systems;
|
|
using Robust.Shared.Interfaces.GameObjects;
|
|
using Robust.Shared.Interfaces.Timing;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Localization;
|
|
using Robust.Shared.Log;
|
|
using Robust.Shared.Serialization;
|
|
using Robust.Shared.ViewVariables;
|
|
using Timer = Robust.Shared.Timers.Timer;
|
|
|
|
namespace Content.Server.GameObjects.Components.Disposal
|
|
{
|
|
[RegisterComponent]
|
|
[ComponentReference(typeof(SharedDisposalUnitComponent))]
|
|
[ComponentReference(typeof(IInteractUsing))]
|
|
public class DisposalUnitComponent : SharedDisposalUnitComponent, IInteractHand, IInteractUsing, IDragDropOn
|
|
{
|
|
[Dependency] private readonly IServerNotifyManager _notifyManager = default!;
|
|
[Dependency] private readonly IGameTiming _gameTiming = default!;
|
|
|
|
public override string Name => "DisposalUnit";
|
|
|
|
/// <summary>
|
|
/// The delay for an entity trying to move out of this unit.
|
|
/// </summary>
|
|
private static readonly TimeSpan ExitAttemptDelay = TimeSpan.FromSeconds(0.5);
|
|
|
|
/// <summary>
|
|
/// Last time that an entity tried to exit this disposal unit.
|
|
/// </summary>
|
|
private TimeSpan _lastExitAttempt;
|
|
|
|
/// <summary>
|
|
/// The current pressure of this disposal unit.
|
|
/// Prevents it from flushing if it is not equal to or bigger than 1.
|
|
/// </summary>
|
|
private float _pressure;
|
|
|
|
private bool _engaged;
|
|
|
|
[ViewVariables]
|
|
private TimeSpan _automaticEngageTime;
|
|
|
|
[ViewVariables]
|
|
private TimeSpan _flushDelay;
|
|
|
|
[ViewVariables]
|
|
private float _entryDelay;
|
|
|
|
/// <summary>
|
|
/// Token used to cancel the automatic engage of a disposal unit
|
|
/// after an entity enters it.
|
|
/// </summary>
|
|
private CancellationTokenSource? _automaticEngageToken;
|
|
|
|
/// <summary>
|
|
/// Container of entities inside this disposal unit.
|
|
/// </summary>
|
|
[ViewVariables]
|
|
private Container _container = default!;
|
|
|
|
[ViewVariables] public IReadOnlyList<IEntity> ContainedEntities => _container.ContainedEntities;
|
|
|
|
[ViewVariables]
|
|
public bool Powered =>
|
|
!Owner.TryGetComponent(out PowerReceiverComponent? receiver) ||
|
|
receiver.Powered;
|
|
|
|
[ViewVariables]
|
|
public bool Anchored =>
|
|
!Owner.TryGetComponent(out CollidableComponent? collidable) ||
|
|
collidable.Anchored;
|
|
|
|
[ViewVariables]
|
|
private PressureState State => _pressure >= 1 ? PressureState.Ready : PressureState.Pressurizing;
|
|
|
|
[ViewVariables]
|
|
private bool Engaged
|
|
{
|
|
get => _engaged;
|
|
set
|
|
{
|
|
var oldEngaged = _engaged;
|
|
_engaged = value;
|
|
|
|
if (oldEngaged == value)
|
|
{
|
|
return;
|
|
}
|
|
|
|
UpdateVisualState();
|
|
}
|
|
}
|
|
|
|
[ViewVariables] private BoundUserInterface? UserInterface => Owner.GetUIOrNull(DisposalUnitUiKey.Key);
|
|
|
|
private DisposalUnitBoundUserInterfaceState? _lastUiState;
|
|
|
|
/// <summary>
|
|
/// Store the translated state.
|
|
/// </summary>
|
|
private (PressureState State, string Localized) _locState;
|
|
|
|
public bool CanInsert(IEntity entity)
|
|
{
|
|
if (!Anchored)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (!entity.TryGetComponent(out ICollidableComponent? collidable) ||
|
|
!collidable.CanCollide)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (!entity.HasComponent<ItemComponent>() &&
|
|
!entity.HasComponent<ISharedBodyManagerComponent>())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return _container.CanInsert(entity);
|
|
}
|
|
|
|
private void TryQueueEngage()
|
|
{
|
|
if (!Powered && ContainedEntities.Count == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
_automaticEngageToken = new CancellationTokenSource();
|
|
|
|
Timer.Spawn(_automaticEngageTime, () =>
|
|
{
|
|
if (!TryFlush())
|
|
{
|
|
TryQueueEngage();
|
|
}
|
|
}, _automaticEngageToken.Token);
|
|
}
|
|
|
|
private void AfterInsert(IEntity entity)
|
|
{
|
|
TryQueueEngage();
|
|
|
|
if (entity.TryGetComponent(out IActorComponent? actor))
|
|
{
|
|
UserInterface?.Close(actor.playerSession);
|
|
}
|
|
|
|
UpdateVisualState();
|
|
}
|
|
|
|
public async Task<bool> TryInsert(IEntity entity, IEntity? user = default)
|
|
{
|
|
if (!CanInsert(entity))
|
|
return false;
|
|
|
|
if (user != null && _entryDelay > 0f)
|
|
{
|
|
var doAfterSystem = EntitySystem.Get<DoAfterSystem>();
|
|
|
|
var doAfterArgs = new DoAfterEventArgs(user, _entryDelay, default, Owner)
|
|
{
|
|
BreakOnDamage = true,
|
|
BreakOnStun = true,
|
|
BreakOnTargetMove = true,
|
|
BreakOnUserMove = true,
|
|
NeedHand = false,
|
|
};
|
|
|
|
var result = await doAfterSystem.DoAfter(doAfterArgs);
|
|
|
|
if (result == DoAfterStatus.Cancelled)
|
|
return false;
|
|
|
|
}
|
|
|
|
if (!_container.Insert(entity))
|
|
return false;
|
|
|
|
AfterInsert(entity);
|
|
|
|
return true;
|
|
}
|
|
|
|
private bool TryDrop(IEntity user, IEntity entity)
|
|
{
|
|
if (!user.TryGetComponent(out HandsComponent? hands))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (!CanInsert(entity) || !hands.Drop(entity, _container))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
AfterInsert(entity);
|
|
|
|
return true;
|
|
}
|
|
|
|
private void Remove(IEntity entity)
|
|
{
|
|
_container.Remove(entity);
|
|
|
|
if (ContainedEntities.Count == 0)
|
|
{
|
|
_automaticEngageToken?.Cancel();
|
|
_automaticEngageToken = null;
|
|
}
|
|
|
|
UpdateVisualState();
|
|
}
|
|
|
|
private bool CanFlush()
|
|
{
|
|
return _pressure >= 1 && Powered && Anchored;
|
|
}
|
|
|
|
private void ToggleEngage()
|
|
{
|
|
Engaged ^= true;
|
|
|
|
if (Engaged && CanFlush())
|
|
{
|
|
Timer.Spawn(_flushDelay, () => TryFlush());
|
|
}
|
|
}
|
|
|
|
public bool TryFlush()
|
|
{
|
|
if (!CanFlush())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
var snapGrid = Owner.GetComponent<SnapGridComponent>();
|
|
var entry = snapGrid
|
|
.GetLocal()
|
|
.FirstOrDefault(entity => entity.HasComponent<DisposalEntryComponent>());
|
|
|
|
if (entry == null)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
var entryComponent = entry.GetComponent<DisposalEntryComponent>();
|
|
var entities = _container.ContainedEntities.ToList();
|
|
foreach (var entity in _container.ContainedEntities.ToList())
|
|
{
|
|
_container.Remove(entity);
|
|
}
|
|
|
|
entryComponent.TryInsert(entities);
|
|
|
|
_automaticEngageToken?.Cancel();
|
|
_automaticEngageToken = null;
|
|
|
|
_pressure = 0;
|
|
|
|
Engaged = false;
|
|
|
|
UpdateVisualState(true);
|
|
UpdateInterface();
|
|
|
|
return true;
|
|
}
|
|
|
|
private void TryEjectContents()
|
|
{
|
|
foreach (var entity in _container.ContainedEntities.ToArray())
|
|
{
|
|
Remove(entity);
|
|
}
|
|
}
|
|
|
|
private void TogglePower()
|
|
{
|
|
if (!Owner.TryGetComponent(out PowerReceiverComponent? receiver))
|
|
{
|
|
return;
|
|
}
|
|
|
|
receiver.PowerDisabled = !receiver.PowerDisabled;
|
|
UpdateInterface();
|
|
}
|
|
|
|
private DisposalUnitBoundUserInterfaceState GetInterfaceState()
|
|
{
|
|
string stateString;
|
|
|
|
if (_locState.State != State)
|
|
{
|
|
stateString = Loc.GetString($"{State}");
|
|
_locState = (State, stateString);
|
|
}
|
|
else
|
|
{
|
|
stateString = _locState.Localized;
|
|
}
|
|
|
|
return new DisposalUnitBoundUserInterfaceState(Owner.Name, stateString, _pressure, Powered, Engaged);
|
|
}
|
|
|
|
private void UpdateInterface()
|
|
{
|
|
var state = GetInterfaceState();
|
|
|
|
if (_lastUiState != null && _lastUiState.Equals(state))
|
|
{
|
|
return;
|
|
}
|
|
|
|
_lastUiState = state;
|
|
UserInterface?.SetState(state);
|
|
}
|
|
|
|
private bool PlayerCanUse(IEntity? player)
|
|
{
|
|
if (player == null)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (!ActionBlockerSystem.CanInteract(player) ||
|
|
!ActionBlockerSystem.CanUse(player))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
private void OnUiReceiveMessage(ServerBoundUserInterfaceMessage obj)
|
|
{
|
|
if (obj.Session.AttachedEntity == null)
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (!PlayerCanUse(obj.Session.AttachedEntity))
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (!(obj.Message is UiButtonPressedMessage message))
|
|
{
|
|
return;
|
|
}
|
|
|
|
switch (message.Button)
|
|
{
|
|
case UiButton.Eject:
|
|
TryEjectContents();
|
|
break;
|
|
case UiButton.Engage:
|
|
ToggleEngage();
|
|
break;
|
|
case UiButton.Power:
|
|
TogglePower();
|
|
EntitySystem.Get<AudioSystem>().PlayFromEntity("/Audio/Machines/machine_switch.ogg", Owner, AudioParams.Default.WithVolume(-2f));
|
|
|
|
break;
|
|
default:
|
|
throw new ArgumentOutOfRangeException();
|
|
}
|
|
}
|
|
|
|
private void UpdateVisualState()
|
|
{
|
|
UpdateVisualState(false);
|
|
}
|
|
|
|
private void UpdateVisualState(bool flush)
|
|
{
|
|
if (!Owner.TryGetComponent(out AppearanceComponent? appearance))
|
|
{
|
|
return;
|
|
}
|
|
|
|
|
|
if (!Anchored)
|
|
{
|
|
appearance.SetData(Visuals.VisualState, VisualState.UnAnchored);
|
|
appearance.SetData(Visuals.Handle, HandleState.Normal);
|
|
appearance.SetData(Visuals.Light, LightState.Off);
|
|
return;
|
|
}
|
|
else if (_pressure < 1)
|
|
{
|
|
appearance.SetData(Visuals.VisualState, VisualState.Charging);
|
|
}
|
|
else
|
|
{
|
|
appearance.SetData(Visuals.VisualState, VisualState.Anchored);
|
|
}
|
|
|
|
appearance.SetData(Visuals.Handle, Engaged
|
|
? HandleState.Engaged
|
|
: HandleState.Normal);
|
|
|
|
if (!Powered)
|
|
{
|
|
appearance.SetData(Visuals.Light, LightState.Off);
|
|
return;
|
|
}
|
|
|
|
if (flush)
|
|
{
|
|
appearance.SetData(Visuals.VisualState, VisualState.Flushing);
|
|
appearance.SetData(Visuals.Light, LightState.Off);
|
|
return;
|
|
}
|
|
|
|
if (ContainedEntities.Count > 0)
|
|
{
|
|
appearance.SetData(Visuals.Light, LightState.Full);
|
|
return;
|
|
}
|
|
|
|
appearance.SetData(Visuals.Light, _pressure < 1
|
|
? LightState.Charging
|
|
: LightState.Ready);
|
|
}
|
|
|
|
public override void Update(float frameTime)
|
|
{
|
|
base.Update(frameTime);
|
|
if (!Powered)
|
|
{
|
|
return;
|
|
}
|
|
|
|
var oldPressure = _pressure;
|
|
|
|
_pressure = _pressure + frameTime > 1
|
|
? 1
|
|
: _pressure + 0.05f * frameTime;
|
|
|
|
if (oldPressure < 1 && _pressure >= 1)
|
|
{
|
|
UpdateVisualState();
|
|
|
|
if (Engaged)
|
|
{
|
|
TryFlush();
|
|
}
|
|
}
|
|
|
|
UpdateInterface();
|
|
}
|
|
|
|
private void PowerStateChanged(object? sender, PowerStateEventArgs args)
|
|
{
|
|
if (!args.Powered)
|
|
{
|
|
_automaticEngageToken?.Cancel();
|
|
_automaticEngageToken = null;
|
|
}
|
|
|
|
UpdateVisualState();
|
|
|
|
if (Engaged && !TryFlush())
|
|
{
|
|
TryQueueEngage();
|
|
}
|
|
}
|
|
|
|
public override void ExposeData(ObjectSerializer serializer)
|
|
{
|
|
base.ExposeData(serializer);
|
|
|
|
serializer.DataReadWriteFunction(
|
|
"pressure",
|
|
1.0f,
|
|
pressure => _pressure = pressure,
|
|
() => _pressure);
|
|
|
|
serializer.DataReadWriteFunction(
|
|
"automaticEngageTime",
|
|
30,
|
|
seconds => _automaticEngageTime = TimeSpan.FromSeconds(seconds),
|
|
() => (int) _automaticEngageTime.TotalSeconds);
|
|
|
|
serializer.DataReadWriteFunction(
|
|
"flushDelay",
|
|
3,
|
|
seconds => _flushDelay = TimeSpan.FromSeconds(seconds),
|
|
() => (int) _flushDelay.TotalSeconds);
|
|
|
|
serializer.DataReadWriteFunction(
|
|
"entryDelay",
|
|
0.5f,
|
|
seconds => _entryDelay = seconds,
|
|
() => (int) _entryDelay);
|
|
}
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
_container = ContainerManagerComponent.Ensure<Container>(Name, Owner);
|
|
|
|
if (UserInterface != null)
|
|
{
|
|
UserInterface.OnReceiveMessage += OnUiReceiveMessage;
|
|
}
|
|
|
|
UpdateInterface();
|
|
}
|
|
|
|
protected override void Startup()
|
|
{
|
|
base.Startup();
|
|
|
|
if(!Owner.HasComponent<AnchorableComponent>())
|
|
{
|
|
Logger.WarningS("VitalComponentMissing", $"Disposal unit {Owner.Uid} is missing an anchorable component");
|
|
}
|
|
|
|
if (Owner.TryGetComponent(out CollidableComponent? collidable))
|
|
{
|
|
collidable.AnchoredChanged += UpdateVisualState;
|
|
}
|
|
|
|
if (Owner.TryGetComponent(out PowerReceiverComponent? receiver))
|
|
{
|
|
receiver.OnPowerStateChanged += PowerStateChanged;
|
|
}
|
|
|
|
UpdateVisualState();
|
|
}
|
|
|
|
public override void OnRemove()
|
|
{
|
|
if (Owner.TryGetComponent(out ICollidableComponent? collidable))
|
|
{
|
|
collidable.AnchoredChanged -= UpdateVisualState;
|
|
}
|
|
|
|
if (Owner.TryGetComponent(out PowerReceiverComponent? receiver))
|
|
{
|
|
receiver.OnPowerStateChanged -= PowerStateChanged;
|
|
}
|
|
|
|
foreach (var entity in _container.ContainedEntities.ToArray())
|
|
{
|
|
_container.ForceRemove(entity);
|
|
}
|
|
|
|
UserInterface?.CloseAll();
|
|
|
|
_automaticEngageToken?.Cancel();
|
|
_automaticEngageToken = null;
|
|
|
|
_container = null!;
|
|
|
|
base.OnRemove();
|
|
}
|
|
|
|
public override void HandleMessage(ComponentMessage message, IComponent? component)
|
|
{
|
|
base.HandleMessage(message, component);
|
|
|
|
switch (message)
|
|
{
|
|
case RelayMovementEntityMessage msg:
|
|
if (!msg.Entity.TryGetComponent(out HandsComponent? hands) ||
|
|
hands.Count == 0 ||
|
|
_gameTiming.CurTime < _lastExitAttempt + ExitAttemptDelay)
|
|
{
|
|
break;
|
|
}
|
|
|
|
_lastExitAttempt = _gameTiming.CurTime;
|
|
Remove(msg.Entity);
|
|
break;
|
|
}
|
|
}
|
|
|
|
bool IInteractHand.InteractHand(InteractHandEventArgs eventArgs)
|
|
{
|
|
if (!ActionBlockerSystem.CanInteract(eventArgs.User))
|
|
{
|
|
_notifyManager.PopupMessage(Owner.Transform.GridPosition, eventArgs.User,
|
|
Loc.GetString("You can't do that!"));
|
|
return false;
|
|
}
|
|
|
|
if (ContainerHelpers.IsInContainer(eventArgs.User))
|
|
{
|
|
_notifyManager.PopupMessage(Owner.Transform.GridPosition, eventArgs.User,
|
|
Loc.GetString("You can't reach there!"));
|
|
return false;
|
|
}
|
|
|
|
if (!eventArgs.User.TryGetComponent(out IActorComponent? actor))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (!eventArgs.User.HasComponent<IHandsComponent>())
|
|
{
|
|
_notifyManager.PopupMessage(Owner.Transform.GridPosition, eventArgs.User,
|
|
Loc.GetString("You have no hands!"));
|
|
return false;
|
|
}
|
|
|
|
UserInterface?.Open(actor.playerSession);
|
|
return true;
|
|
}
|
|
|
|
async Task<bool> IInteractUsing.InteractUsing(InteractUsingEventArgs eventArgs)
|
|
{
|
|
return TryDrop(eventArgs.User, eventArgs.Using);
|
|
}
|
|
|
|
bool IDragDropOn.CanDragDropOn(DragDropEventArgs eventArgs)
|
|
{
|
|
return CanInsert(eventArgs.Dropped);
|
|
}
|
|
|
|
bool IDragDropOn.DragDropOn(DragDropEventArgs eventArgs)
|
|
{
|
|
_ = TryInsert(eventArgs.Dropped, eventArgs.User);
|
|
return true;
|
|
}
|
|
|
|
[Verb]
|
|
private sealed class SelfInsertVerb : Verb<DisposalUnitComponent>
|
|
{
|
|
protected override void GetData(IEntity user, DisposalUnitComponent component, VerbData data)
|
|
{
|
|
data.Visibility = VerbVisibility.Invisible;
|
|
|
|
if (!ActionBlockerSystem.CanInteract(user) ||
|
|
component.ContainedEntities.Contains(user))
|
|
{
|
|
return;
|
|
}
|
|
|
|
data.Visibility = VerbVisibility.Visible;
|
|
data.Text = Loc.GetString("Jump inside");
|
|
}
|
|
|
|
protected override void Activate(IEntity user, DisposalUnitComponent component)
|
|
{
|
|
_ = component.TryInsert(user, user);
|
|
}
|
|
}
|
|
|
|
[Verb]
|
|
private sealed class FlushVerb : Verb<DisposalUnitComponent>
|
|
{
|
|
protected override void GetData(IEntity user, DisposalUnitComponent component, VerbData data)
|
|
{
|
|
data.Visibility = VerbVisibility.Invisible;
|
|
|
|
if (!ActionBlockerSystem.CanInteract(user) ||
|
|
component.ContainedEntities.Contains(user))
|
|
{
|
|
return;
|
|
}
|
|
|
|
data.Visibility = VerbVisibility.Visible;
|
|
data.Text = Loc.GetString("Flush");
|
|
}
|
|
|
|
protected override void Activate(IEntity user, DisposalUnitComponent component)
|
|
{
|
|
component.Engaged = true;
|
|
component.TryFlush();
|
|
}
|
|
}
|
|
}
|
|
}
|