Re-organize all projects (#4166)
This commit is contained in:
DrSmugleaf
GitHub
@@ -0,0 +1,191 @@
|
||||
#nullable enable
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using Content.Server.Atmos;
|
||||
using Content.Server.Disposal.Tube.Components;
|
||||
using Content.Server.Interfaces;
|
||||
using Content.Server.Items;
|
||||
using Content.Shared.Atmos;
|
||||
using Content.Shared.Body.Components;
|
||||
using Robust.Shared.Containers;
|
||||
using Robust.Shared.GameObjects;
|
||||
using Robust.Shared.Maths;
|
||||
using Robust.Shared.Physics;
|
||||
using Robust.Shared.Serialization.Manager.Attributes;
|
||||
using Robust.Shared.ViewVariables;
|
||||
|
||||
namespace Content.Server.Disposal.Unit.Components
|
||||
{
|
||||
// TODO: Add gas
|
||||
[RegisterComponent]
|
||||
public class DisposalHolderComponent : Component, IGasMixtureHolder
|
||||
{
|
||||
public override string Name => "DisposalHolder";
|
||||
|
||||
private bool _deletionRequested = false;
|
||||
private Container _contents = null!;
|
||||
|
||||
/// <summary>
|
||||
/// The total amount of time that it will take for this entity to
|
||||
/// be pushed to the next tube
|
||||
/// </summary>
|
||||
[ViewVariables]
|
||||
private float StartingTime { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// Time left until the entity is pushed to the next tube
|
||||
/// </summary>
|
||||
[ViewVariables]
|
||||
private float TimeLeft { get; set; }
|
||||
|
||||
[ViewVariables]
|
||||
public IDisposalTubeComponent? PreviousTube { get; set; }
|
||||
|
||||
[ViewVariables]
|
||||
public IDisposalTubeComponent? CurrentTube { get; private set; }
|
||||
|
||||
[ViewVariables]
|
||||
public IDisposalTubeComponent? NextTube { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// A list of tags attached to the content, used for sorting
|
||||
/// </summary>
|
||||
[ViewVariables]
|
||||
public HashSet<string> Tags { get; set; } = new();
|
||||
|
||||
[ViewVariables]
|
||||
[DataField("air")]
|
||||
public GasMixture Air { get; set; } = new GasMixture(Atmospherics.CellVolume);
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
_contents = ContainerHelpers.EnsureContainer<Container>(Owner, nameof(DisposalHolderComponent));
|
||||
}
|
||||
|
||||
public override void OnRemove()
|
||||
{
|
||||
base.OnRemove();
|
||||
ExitDisposals();
|
||||
}
|
||||
|
||||
private bool CanInsert(IEntity entity)
|
||||
{
|
||||
if (!_contents.CanInsert(entity))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
return entity.HasComponent<ItemComponent>() ||
|
||||
entity.HasComponent<IBody>();
|
||||
}
|
||||
|
||||
public bool TryInsert(IEntity entity)
|
||||
{
|
||||
if (!CanInsert(entity) || !_contents.Insert(entity))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
if (entity.TryGetComponent(out IPhysBody? physics))
|
||||
{
|
||||
physics.CanCollide = false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public void EnterTube(IDisposalTubeComponent tube)
|
||||
{
|
||||
if (CurrentTube != null)
|
||||
{
|
||||
PreviousTube = CurrentTube;
|
||||
}
|
||||
|
||||
Owner.Transform.Coordinates = tube.Owner.Transform.Coordinates;
|
||||
CurrentTube = tube;
|
||||
NextTube = tube.NextTube(this);
|
||||
StartingTime = 0.1f;
|
||||
TimeLeft = 0.1f;
|
||||
}
|
||||
|
||||
public void ExitDisposals()
|
||||
{
|
||||
if (_deletionRequested || Deleted)
|
||||
return;
|
||||
|
||||
PreviousTube = null;
|
||||
CurrentTube = null;
|
||||
NextTube = null;
|
||||
StartingTime = 0;
|
||||
TimeLeft = 0;
|
||||
|
||||
foreach (var entity in _contents.ContainedEntities.ToArray())
|
||||
{
|
||||
if (entity.TryGetComponent(out IPhysBody? physics))
|
||||
{
|
||||
physics.CanCollide = true;
|
||||
}
|
||||
|
||||
_contents.ForceRemove(entity);
|
||||
|
||||
if (entity.Transform.Parent == Owner.Transform)
|
||||
{
|
||||
entity.Transform.AttachParentToContainerOrGrid();
|
||||
}
|
||||
}
|
||||
|
||||
if (Owner.Transform.Coordinates.TryGetTileAtmosphere(out var tileAtmos) &&
|
||||
tileAtmos.Air != null)
|
||||
{
|
||||
tileAtmos.AssumeAir(Air);
|
||||
Air.Clear();
|
||||
}
|
||||
|
||||
// FIXME: This is a workaround for https://github.com/space-wizards/RobustToolbox/issues/1646
|
||||
Owner.SpawnTimer(TimeSpan.Zero, () => Owner.Delete());
|
||||
_deletionRequested = true;
|
||||
}
|
||||
|
||||
public void Update(float frameTime)
|
||||
{
|
||||
while (frameTime > 0)
|
||||
{
|
||||
var time = frameTime;
|
||||
if (time > TimeLeft)
|
||||
{
|
||||
time = TimeLeft;
|
||||
}
|
||||
|
||||
TimeLeft -= time;
|
||||
frameTime -= time;
|
||||
|
||||
if (CurrentTube == null)
|
||||
{
|
||||
ExitDisposals();
|
||||
break;
|
||||
}
|
||||
|
||||
if (TimeLeft > 0)
|
||||
{
|
||||
var progress = 1 - TimeLeft / StartingTime;
|
||||
var origin = CurrentTube.Owner.Transform.WorldPosition;
|
||||
var destination = CurrentTube.NextDirection(this).ToVec();
|
||||
var newPosition = destination * progress;
|
||||
|
||||
Owner.Transform.WorldPosition = origin + newPosition;
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (NextTube == null || !CurrentTube.TransferTo(this, NextTube))
|
||||
{
|
||||
CurrentTube.Remove(this);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
691
Content.Server/Disposal/Unit/Components/DisposalUnitComponent.cs
Normal file
691
Content.Server/Disposal/Unit/Components/DisposalUnitComponent.cs
Normal file
@@ -0,0 +1,691 @@
|
||||
#nullable enable
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using Content.Server.Anchor;
|
||||
using Content.Server.Atmos;
|
||||
using Content.Server.Disposal.Tube.Components;
|
||||
using Content.Server.DoAfter;
|
||||
using Content.Server.GameObjects.EntitySystems;
|
||||
using Content.Server.Hands.Components;
|
||||
using Content.Server.Interfaces;
|
||||
using Content.Server.Power.Components;
|
||||
using Content.Server.UserInterface;
|
||||
using Content.Shared.ActionBlocker;
|
||||
using Content.Shared.Atmos;
|
||||
using Content.Shared.Disposal.Components;
|
||||
using Content.Shared.DragDrop;
|
||||
using Content.Shared.Interaction;
|
||||
using Content.Shared.Notification;
|
||||
using Content.Shared.Throwing;
|
||||
using Content.Shared.Verbs;
|
||||
using Robust.Server.GameObjects;
|
||||
using Robust.Shared.Audio;
|
||||
using Robust.Shared.Containers;
|
||||
using Robust.Shared.GameObjects;
|
||||
using Robust.Shared.IoC;
|
||||
using Robust.Shared.Localization;
|
||||
using Robust.Shared.Log;
|
||||
using Robust.Shared.Map;
|
||||
using Robust.Shared.Player;
|
||||
using Robust.Shared.Random;
|
||||
using Robust.Shared.Serialization.Manager.Attributes;
|
||||
using Robust.Shared.Timing;
|
||||
using Robust.Shared.ViewVariables;
|
||||
|
||||
namespace Content.Server.Disposal.Unit.Components
|
||||
{
|
||||
[RegisterComponent]
|
||||
[ComponentReference(typeof(SharedDisposalUnitComponent))]
|
||||
[ComponentReference(typeof(IActivate))]
|
||||
[ComponentReference(typeof(IInteractUsing))]
|
||||
public class DisposalUnitComponent : SharedDisposalUnitComponent, IInteractHand, IActivate, IInteractUsing, IThrowCollide, IGasMixtureHolder
|
||||
{
|
||||
[Dependency] private readonly IGameTiming _gameTiming = default!;
|
||||
[Dependency] private readonly IMapManager _mapManager = 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>
|
||||
[ViewVariables]
|
||||
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>
|
||||
[ViewVariables]
|
||||
[DataField("pressure")]
|
||||
private float _pressure;
|
||||
|
||||
private bool _engaged;
|
||||
|
||||
[ViewVariables(VVAccess.ReadWrite)]
|
||||
[DataField("autoEngageTime")]
|
||||
private readonly TimeSpan _automaticEngageTime = TimeSpan.FromSeconds(30);
|
||||
|
||||
[ViewVariables(VVAccess.ReadWrite)]
|
||||
[DataField("flushDelay")]
|
||||
private readonly TimeSpan _flushDelay = TimeSpan.FromSeconds(3);
|
||||
|
||||
/// <summary>
|
||||
/// Delay from trying to enter disposals ourselves.
|
||||
/// </summary>
|
||||
[ViewVariables(VVAccess.ReadWrite)]
|
||||
[DataField("entryDelay")]
|
||||
private float _entryDelay = 0.5f;
|
||||
|
||||
/// <summary>
|
||||
/// Delay from trying to shove someone else into disposals.
|
||||
/// </summary>
|
||||
[ViewVariables(VVAccess.ReadWrite)]
|
||||
private float _draggedEntryDelay = 0.5f;
|
||||
|
||||
/// <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]
|
||||
private PressureState State => _pressure >= 1 ? PressureState.Ready : PressureState.Pressurizing;
|
||||
|
||||
[ViewVariables(VVAccess.ReadWrite)]
|
||||
private bool Engaged
|
||||
{
|
||||
get => _engaged;
|
||||
set
|
||||
{
|
||||
var oldEngaged = _engaged;
|
||||
_engaged = value;
|
||||
|
||||
if (oldEngaged == value)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
UpdateVisualState();
|
||||
}
|
||||
}
|
||||
|
||||
[ViewVariables] private BoundUserInterface? UserInterface => Owner.GetUIOrNull(DisposalUnitUiKey.Key);
|
||||
|
||||
[DataField("air")]
|
||||
public GasMixture Air { get; set; } = new GasMixture(Atmospherics.CellVolume);
|
||||
|
||||
public override bool CanInsert(IEntity entity)
|
||||
{
|
||||
if (!base.CanInsert(entity))
|
||||
return false;
|
||||
|
||||
return _container.CanInsert(entity);
|
||||
}
|
||||
|
||||
private void TryQueueEngage()
|
||||
{
|
||||
if (!Powered && ContainedEntities.Count == 0)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
_automaticEngageToken = new CancellationTokenSource();
|
||||
|
||||
Owner.SpawnTimer(_automaticEngageTime, () =>
|
||||
{
|
||||
if (!TryFlush())
|
||||
{
|
||||
TryQueueEngage();
|
||||
}
|
||||
}, _automaticEngageToken.Token);
|
||||
}
|
||||
|
||||
private void AfterInsert(IEntity entity)
|
||||
{
|
||||
TryQueueEngage();
|
||||
|
||||
if (entity.TryGetComponent(out ActorComponent? actor))
|
||||
{
|
||||
UserInterface?.Close(actor.PlayerSession);
|
||||
}
|
||||
|
||||
UpdateVisualState();
|
||||
}
|
||||
|
||||
public async Task<bool> TryInsert(IEntity entity, IEntity? user = default)
|
||||
{
|
||||
if (!CanInsert(entity))
|
||||
return false;
|
||||
|
||||
var delay = user == entity ? _entryDelay : _draggedEntryDelay;
|
||||
|
||||
if (user != null && delay > 0.0f)
|
||||
{
|
||||
var doAfterSystem = EntitySystem.Get<DoAfterSystem>();
|
||||
|
||||
// Can't check if our target AND disposals moves currently so we'll just check target.
|
||||
// if you really want to check if disposals moves then add a predicate.
|
||||
var doAfterArgs = new DoAfterEventArgs(user, delay, default, entity)
|
||||
{
|
||||
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())
|
||||
{
|
||||
Owner.SpawnTimer(_flushDelay, () => TryFlush());
|
||||
}
|
||||
}
|
||||
|
||||
public bool TryFlush()
|
||||
{
|
||||
if (!CanFlush())
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
var grid = _mapManager.GetGrid(Owner.Transform.GridID);
|
||||
var coords = Owner.Transform.Coordinates;
|
||||
var entry = grid.GetLocal(coords)
|
||||
.FirstOrDefault(entity => Owner.EntityManager.ComponentManager.HasComponent<DisposalEntryComponent>(entity));
|
||||
|
||||
if (entry == default)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
var entryComponent = Owner.EntityManager.ComponentManager.GetComponent<DisposalEntryComponent>(entry);
|
||||
|
||||
if (Owner.Transform.Coordinates.TryGetTileAtmosphere(out var tileAtmos) &&
|
||||
tileAtmos.Air != null &&
|
||||
tileAtmos.Air.Temperature > 0)
|
||||
{
|
||||
var tileAir = tileAtmos.Air;
|
||||
var transferMoles = 0.1f * (0.05f * Atmospherics.OneAtmosphere * 1.01f - Air.Pressure) * Air.Volume / (tileAir.Temperature * Atmospherics.R);
|
||||
|
||||
Air = tileAir.Remove(transferMoles);
|
||||
|
||||
var atmosSystem = EntitySystem.Get<AtmosphereSystem>();
|
||||
atmosSystem
|
||||
.GetGridAtmosphere(Owner.Transform.Coordinates)?
|
||||
.Invalidate(tileAtmos.GridIndices);
|
||||
}
|
||||
|
||||
entryComponent.TryInsert(this);
|
||||
|
||||
_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 void UpdateInterface()
|
||||
{
|
||||
string stateString;
|
||||
|
||||
stateString = Loc.GetString($"{State}");
|
||||
var state = new DisposalUnitBoundUserInterfaceState(Owner.Name, stateString, _pressure, Powered, Engaged);
|
||||
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 not UiButtonPressedMessage message)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
switch (message.Button)
|
||||
{
|
||||
case UiButton.Eject:
|
||||
TryEjectContents();
|
||||
break;
|
||||
case UiButton.Engage:
|
||||
ToggleEngage();
|
||||
break;
|
||||
case UiButton.Power:
|
||||
TogglePower();
|
||||
SoundSystem.Play(Filter.Pvs(Owner), "/Audio/Machines/machine_switch.ogg", Owner, AudioParams.Default.WithVolume(-2f));
|
||||
break;
|
||||
default:
|
||||
throw new ArgumentOutOfRangeException();
|
||||
}
|
||||
}
|
||||
|
||||
public 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();
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: Ideally we'd just send the start and end and client could lerp as the bandwidth would be way lower
|
||||
if (_pressure < 1.0f || oldPressure < 1.0f && _pressure >= 1.0f)
|
||||
{
|
||||
UpdateInterface();
|
||||
}
|
||||
}
|
||||
|
||||
private void PowerStateChanged(PowerChangedMessage args)
|
||||
{
|
||||
if (!args.Powered)
|
||||
{
|
||||
_automaticEngageToken?.Cancel();
|
||||
_automaticEngageToken = null;
|
||||
}
|
||||
|
||||
UpdateVisualState();
|
||||
|
||||
if (Engaged && !TryFlush())
|
||||
{
|
||||
TryQueueEngage();
|
||||
}
|
||||
}
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
_container = ContainerHelpers.EnsureContainer<Container>(Owner, Name);
|
||||
|
||||
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");
|
||||
}
|
||||
|
||||
UpdateVisualState();
|
||||
UpdateInterface();
|
||||
}
|
||||
|
||||
public override void OnRemove()
|
||||
{
|
||||
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;
|
||||
|
||||
case PowerChangedMessage powerChanged:
|
||||
PowerStateChanged(powerChanged);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
bool IsValidInteraction(ITargetedInteractEventArgs eventArgs)
|
||||
{
|
||||
if (!ActionBlockerSystem.CanInteract(eventArgs.User))
|
||||
{
|
||||
Owner.PopupMessage(eventArgs.User, Loc.GetString("You can't do that!"));
|
||||
return false;
|
||||
}
|
||||
|
||||
if (eventArgs.User.IsInContainer())
|
||||
{
|
||||
Owner.PopupMessage(eventArgs.User, Loc.GetString("You can't reach there!"));
|
||||
return false;
|
||||
}
|
||||
// This popup message doesn't appear on clicks, even when code was seperate. Unsure why.
|
||||
|
||||
if (!eventArgs.User.HasComponent<IHandsComponent>())
|
||||
{
|
||||
Owner.PopupMessage(eventArgs.User, Loc.GetString("You have no hands!"));
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
bool IInteractHand.InteractHand(InteractHandEventArgs eventArgs)
|
||||
{
|
||||
if (!eventArgs.User.TryGetComponent(out ActorComponent? actor))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
// Duplicated code here, not sure how else to get actor inside to make UserInterface happy.
|
||||
|
||||
if (IsValidInteraction(eventArgs))
|
||||
{
|
||||
UserInterface?.Open(actor.PlayerSession);
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
void IActivate.Activate(ActivateEventArgs eventArgs)
|
||||
{
|
||||
if (!eventArgs.User.TryGetComponent(out ActorComponent? actor))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (IsValidInteraction(eventArgs))
|
||||
{
|
||||
UserInterface?.Open(actor.PlayerSession);
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
async Task<bool> IInteractUsing.InteractUsing(InteractUsingEventArgs eventArgs)
|
||||
{
|
||||
return TryDrop(eventArgs.User, eventArgs.Using);
|
||||
}
|
||||
|
||||
public override bool CanDragDropOn(DragDropEvent eventArgs)
|
||||
{
|
||||
// Base is redundant given this already calls the base CanInsert
|
||||
// If that changes then update this
|
||||
return CanInsert(eventArgs.Dragged);
|
||||
}
|
||||
|
||||
public override bool DragDropOn(DragDropEvent eventArgs)
|
||||
{
|
||||
_ = TryInsert(eventArgs.Dragged, eventArgs.User);
|
||||
return true;
|
||||
}
|
||||
|
||||
void IThrowCollide.HitBy(ThrowCollideEventArgs eventArgs)
|
||||
{
|
||||
if (!CanInsert(eventArgs.Thrown) ||
|
||||
IoCManager.Resolve<IRobustRandom>().NextDouble() > 0.75 ||
|
||||
!_container.Insert(eventArgs.Thrown))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
AfterInsert(eventArgs.Thrown);
|
||||
}
|
||||
|
||||
[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");
|
||||
data.IconTexture = "/Textures/Interface/VerbIcons/eject.svg.192dpi.png";
|
||||
}
|
||||
|
||||
protected override void Activate(IEntity user, DisposalUnitComponent component)
|
||||
{
|
||||
component.Engaged = true;
|
||||
component.TryFlush();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user