ecc969d8f4
* add low pressure data field * resolve CheckSeverity bug, replace logic * add configurable low pressure threshhold
285 lines
11 KiB
C#
285 lines
11 KiB
C#
using Content.Server.Atmos.Components;
|
|
using Content.Server.Atmos.EntitySystems;
|
|
using Content.Server.Body.Components;
|
|
using Content.Server.Hands.Systems;
|
|
using Content.Shared.Alert;
|
|
using Content.Shared.Atmos;
|
|
using Content.Shared.Inventory;
|
|
using Robust.Shared.Containers;
|
|
using Robust.Shared.Prototypes;
|
|
using Content.Shared.Verbs;
|
|
using Robust.Server.GameObjects;
|
|
using Content.Server.Popups;
|
|
using Robust.Shared.Player;
|
|
using Content.Server.DoAfter;
|
|
using System.Threading;
|
|
|
|
namespace Content.Server.Body.Systems;
|
|
|
|
public sealed class InternalsSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IPrototypeManager _protoManager = default!;
|
|
[Dependency] private readonly AlertsSystem _alerts = default!;
|
|
[Dependency] private readonly AtmosphereSystem _atmos = default!;
|
|
[Dependency] private readonly GasTankSystem _gasTank = default!;
|
|
[Dependency] private readonly HandsSystem _hands = default!;
|
|
[Dependency] private readonly InventorySystem _inventory = default!;
|
|
[Dependency] private readonly PopupSystem _popupSystem = default!;
|
|
[Dependency] private readonly DoAfterSystem _doAfter = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<InternalsComponent, InhaleLocationEvent>(OnInhaleLocation);
|
|
SubscribeLocalEvent<InternalsComponent, ComponentStartup>(OnInternalsStartup);
|
|
SubscribeLocalEvent<InternalsComponent, ComponentShutdown>(OnInternalsShutdown);
|
|
SubscribeLocalEvent<InternalsComponent, GetVerbsEvent<InteractionVerb>>(OnGetInteractionVerbs);
|
|
SubscribeLocalEvent<ToggleOtherInternalsCompleteEvent>(OnToggleOtherInternalsComplete);
|
|
SubscribeLocalEvent<ToggleOtherInternalsCancelledEvent>(OnToggleOtherInternalCanceled);
|
|
}
|
|
|
|
private void OnGetInteractionVerbs(EntityUid uid, InternalsComponent component, GetVerbsEvent<InteractionVerb> args)
|
|
{
|
|
if (!args.CanAccess || !args.CanInteract || args.Hands == null)
|
|
return;
|
|
|
|
if (!EntityManager.TryGetComponent<ActorComponent?>(args.User, out var actor))
|
|
return;
|
|
|
|
InteractionVerb verb = new()
|
|
{
|
|
Act = () =>
|
|
{
|
|
var entManager = IoCManager.Resolve<IEntityManager>();
|
|
if (!entManager.TryGetComponent<InternalsComponent>(uid, out var internals)) return;
|
|
|
|
// Toggle off if they're on
|
|
if (AreInternalsWorking(component))
|
|
{
|
|
DisconnectTank(component);
|
|
return;
|
|
}
|
|
|
|
// If they're not on then check if we have a mask to use
|
|
if (component.BreathToolEntity == null)
|
|
{
|
|
_popupSystem.PopupEntity(Loc.GetString("internals-no-breath-tool"), args.User, Filter.Entities(args.User));
|
|
return;
|
|
}
|
|
|
|
var tank = FindBestGasTank(component);
|
|
if (tank == null)
|
|
{
|
|
_popupSystem.PopupEntity(Loc.GetString("internals-no-tank"), args.User, Filter.Entities(args.User));
|
|
return;
|
|
}
|
|
|
|
// Is the target not you? If yes, use a do-after to give them time to respond.
|
|
if (args.User != args.Target)
|
|
{
|
|
component.CancelToken = new CancellationTokenSource();
|
|
_doAfter.DoAfter(new DoAfterEventArgs(args.User, component.Delay, component.CancelToken.Token, args.Target)
|
|
{
|
|
BreakOnUserMove = true,
|
|
BreakOnDamage = true,
|
|
BreakOnStun = true,
|
|
BreakOnTargetMove = true,
|
|
MovementThreshold = 0.1f,
|
|
BroadcastFinishedEvent = new ToggleOtherInternalsCompleteEvent()
|
|
{
|
|
Component = component,
|
|
User = args.User,
|
|
Target = args.Target,
|
|
Tank = tank
|
|
},
|
|
BroadcastCancelledEvent = new ToggleOtherInternalsCancelledEvent()
|
|
{
|
|
Component = component,
|
|
}
|
|
});
|
|
return;
|
|
}
|
|
entManager.EntitySysManager.GetEntitySystem<GasTankSystem>().ConnectToInternals(tank);
|
|
},
|
|
Message = Loc.GetString("action-description-internals-toggle"),
|
|
IconTexture = "/Textures/Interface/VerbIcons/dot.svg.192dpi.png",
|
|
Text = Loc.GetString("action-name-internals-toggle"),
|
|
};
|
|
|
|
args.Verbs.Add(verb);
|
|
}
|
|
|
|
private void OnToggleOtherInternalsComplete(ToggleOtherInternalsCompleteEvent ev)
|
|
{
|
|
ev.Component.CancelToken = null;
|
|
var entManager = IoCManager.Resolve<IEntityManager>();
|
|
if (ev.Tank != null)
|
|
entManager.EntitySysManager.GetEntitySystem<GasTankSystem>().ConnectToInternals(ev.Tank);
|
|
}
|
|
|
|
private static void OnToggleOtherInternalCanceled(ToggleOtherInternalsCancelledEvent ev)
|
|
{
|
|
ev.Component.CancelToken = null;
|
|
}
|
|
|
|
private void OnInternalsStartup(EntityUid uid, InternalsComponent component, ComponentStartup args)
|
|
{
|
|
_alerts.ShowAlert(component.Owner, AlertType.Internals, GetSeverity(component));
|
|
}
|
|
|
|
private void OnInternalsShutdown(EntityUid uid, InternalsComponent component, ComponentShutdown args)
|
|
{
|
|
_alerts.ClearAlert(uid, AlertType.Internals);
|
|
}
|
|
|
|
private void OnInhaleLocation(EntityUid uid, InternalsComponent component, InhaleLocationEvent args)
|
|
{
|
|
if (AreInternalsWorking(component))
|
|
{
|
|
var gasTank = Comp<GasTankComponent>(component.GasTankEntity!.Value);
|
|
args.Gas = _gasTank.RemoveAirVolume(gasTank, Atmospherics.BreathVolume);
|
|
// TODO: Should listen to gas tank updates instead I guess?
|
|
_alerts.ShowAlert(component.Owner, AlertType.Internals, GetSeverity(component));
|
|
}
|
|
}
|
|
public void DisconnectBreathTool(InternalsComponent component)
|
|
{
|
|
var old = component.BreathToolEntity;
|
|
component.BreathToolEntity = null;
|
|
|
|
if (TryComp(old, out BreathToolComponent? breathTool) )
|
|
{
|
|
_atmos.DisconnectInternals(breathTool);
|
|
DisconnectTank(component);
|
|
}
|
|
|
|
_alerts.ShowAlert(component.Owner, AlertType.Internals, GetSeverity(component));
|
|
}
|
|
|
|
public void ConnectBreathTool(InternalsComponent component, EntityUid toolEntity)
|
|
{
|
|
if (TryComp(component.BreathToolEntity, out BreathToolComponent? tool))
|
|
{
|
|
_atmos.DisconnectInternals(tool);
|
|
}
|
|
|
|
component.BreathToolEntity = toolEntity;
|
|
_alerts.ShowAlert(component.Owner, AlertType.Internals, GetSeverity(component));
|
|
}
|
|
|
|
public void DisconnectTank(InternalsComponent? component)
|
|
{
|
|
if (component == null) return;
|
|
|
|
if (TryComp(component.GasTankEntity, out GasTankComponent? tank))
|
|
{
|
|
_gasTank.DisconnectFromInternals(tank);
|
|
}
|
|
|
|
component.GasTankEntity = null;
|
|
_alerts.ShowAlert(component.Owner, AlertType.Internals, GetSeverity(component));
|
|
}
|
|
|
|
public bool TryConnectTank(InternalsComponent component, EntityUid tankEntity)
|
|
{
|
|
if (component.BreathToolEntity == null)
|
|
return false;
|
|
|
|
if (TryComp(component.GasTankEntity, out GasTankComponent? tank))
|
|
{
|
|
_gasTank.DisconnectFromInternals(tank);
|
|
}
|
|
|
|
component.GasTankEntity = tankEntity;
|
|
_alerts.ShowAlert(component.Owner, AlertType.Internals, GetSeverity(component));
|
|
return true;
|
|
}
|
|
|
|
public bool AreInternalsWorking(InternalsComponent component)
|
|
{
|
|
return TryComp(component.BreathToolEntity, out BreathToolComponent? breathTool) &&
|
|
breathTool.IsFunctional &&
|
|
TryComp(component.GasTankEntity, out GasTankComponent? gasTank) &&
|
|
gasTank.Air != null;
|
|
}
|
|
|
|
private short GetSeverity(InternalsComponent component)
|
|
{
|
|
if (component.BreathToolEntity == null || !AreInternalsWorking(component)) return 2;
|
|
|
|
// If pressure in the tank is below low pressure threshhold, flash warning on internals UI
|
|
if (TryComp<GasTankComponent>(component.GasTankEntity, out var gasTank)
|
|
&& gasTank.IsLowPressure)
|
|
return 0;
|
|
|
|
return 1;
|
|
}
|
|
|
|
public GasTankComponent? FindBestGasTank(InternalsComponent component)
|
|
{
|
|
// Prioritise
|
|
// 1. exo-slot tanks
|
|
// 2. in-hand tanks
|
|
// 3. pocket/belt tanks
|
|
InventoryComponent? inventory = null;
|
|
ContainerManagerComponent? containerManager = null;
|
|
|
|
if (_inventory.TryGetSlotEntity(component.Owner, "suitstorage", out var entity, inventory, containerManager) &&
|
|
TryComp<GasTankComponent>(entity, out var gasTank) &&
|
|
_gasTank.CanConnectToInternals(gasTank))
|
|
{
|
|
return gasTank;
|
|
}
|
|
|
|
var tanks = new List<GasTankComponent>();
|
|
|
|
foreach (var hand in _hands.EnumerateHands(component.Owner))
|
|
{
|
|
if (TryComp(hand.HeldEntity, out gasTank) && _gasTank.CanConnectToInternals(gasTank))
|
|
{
|
|
tanks.Add(gasTank);
|
|
}
|
|
}
|
|
|
|
if (tanks.Count > 0)
|
|
{
|
|
tanks.Sort((x, y) => y.Air.TotalMoles.CompareTo(x.Air.TotalMoles));
|
|
return tanks[0];
|
|
}
|
|
|
|
if (Resolve(component.Owner, ref inventory, false))
|
|
{
|
|
var enumerator = new InventorySystem.ContainerSlotEnumerator(component.Owner, inventory.TemplateId, _protoManager, _inventory, SlotFlags.POCKET | SlotFlags.BELT);
|
|
|
|
while (enumerator.MoveNext(out var container))
|
|
{
|
|
if (TryComp(container.ContainedEntity, out gasTank) && _gasTank.CanConnectToInternals(gasTank))
|
|
{
|
|
tanks.Add(gasTank);
|
|
}
|
|
}
|
|
|
|
if (tanks.Count > 0)
|
|
{
|
|
tanks.Sort((x, y) => y.Air.TotalMoles.CompareTo(x.Air.TotalMoles));
|
|
return tanks[0];
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
private sealed class ToggleOtherInternalsCompleteEvent : EntityEventArgs
|
|
{
|
|
public InternalsComponent Component = default!;
|
|
public EntityUid User { get; init; }
|
|
public EntityUid Target { get; init; }
|
|
public GasTankComponent? Tank { get; init; }
|
|
}
|
|
|
|
private sealed class ToggleOtherInternalsCancelledEvent : EntityEventArgs
|
|
{
|
|
public InternalsComponent Component = default!;
|
|
}
|
|
}
|