00e01d51fd
* Removed SnapGridOffset, there is only center now. * SnapGridComponent methods are now static. * Removed SnapGridComponent.OnPositionChanged. * Refactored static functions off SnapGridComponent to MapGrid. Refactored away usages of SnapGridComponent.Position. * Added Transform.Anchored for checking if an entity is a tile entity. More refactoring for static MapGrid functions. * Static snapgrid methods on MapGrid are no longer static. * Add setter to ITransformComponent.Anchored. Removed direct references to SnapGridComponent from content. * Grid functions now deal with EntityUids instead of SnapGridComponents. Began renaming public API functions from SnapGrid to Anchor. * Remove the SnapGridComponent 'Offset' field from all yaml files. This was removed in code previously, so the yaml linter was upset. * Update engine submodule to v0.4.46.
103 lines
3.3 KiB
C#
103 lines
3.3 KiB
C#
#nullable enable
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Content.Server.GameObjects.Components.NodeContainer;
|
|
using Content.Server.GameObjects.Components.NodeContainer.Nodes;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Log;
|
|
using Robust.Shared.Map;
|
|
using Robust.Shared.ViewVariables;
|
|
|
|
namespace Content.Server.GameObjects.Components.Atmos.Piping
|
|
{
|
|
[RegisterComponent]
|
|
public class GasCanisterPortComponent : Component
|
|
{
|
|
public override string Name => "GasCanisterPort";
|
|
|
|
[Dependency] private readonly IMapManager _mapManager = default!;
|
|
|
|
[ViewVariables]
|
|
public GasCanisterComponent? ConnectedCanister { get; private set; }
|
|
|
|
[ViewVariables]
|
|
public bool ConnectedToCanister => ConnectedCanister != null;
|
|
|
|
[ViewVariables]
|
|
private PipeNode? _gasPort;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
Owner.EnsureComponentWarn<PipeNetDeviceComponent>();
|
|
SetGasPort();
|
|
if (Owner.Transform.Anchored)
|
|
{
|
|
var grid = _mapManager.GetGrid(Owner.Transform.GridID);
|
|
var coords = Owner.Transform.Coordinates;
|
|
var entities = grid.GetLocal(coords);
|
|
foreach (var entity in entities)
|
|
{
|
|
if (Owner.EntityManager.ComponentManager.TryGetComponent<GasCanisterComponent>(entity, out var canister) && canister.Anchored && !canister.ConnectedToPort)
|
|
{
|
|
canister.TryConnectToPort();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public override void OnRemove()
|
|
{
|
|
base.OnRemove();
|
|
ConnectedCanister?.DisconnectFromPort();
|
|
}
|
|
|
|
public override void HandleMessage(ComponentMessage message, IComponent? component)
|
|
{
|
|
base.HandleMessage(message, component);
|
|
switch (message)
|
|
{
|
|
case PipeNetUpdateMessage:
|
|
Update();
|
|
break;
|
|
}
|
|
}
|
|
|
|
public void Update()
|
|
{
|
|
if (_gasPort == null || ConnectedCanister == null)
|
|
return;
|
|
|
|
ConnectedCanister.Air.Share(_gasPort.Air, 1);
|
|
ConnectedCanister.AirWasUpdated();
|
|
}
|
|
|
|
public void ConnectGasCanister(GasCanisterComponent gasCanister)
|
|
{
|
|
ConnectedCanister = gasCanister;
|
|
}
|
|
|
|
public void DisconnectGasCanister()
|
|
{
|
|
ConnectedCanister = null;
|
|
}
|
|
|
|
private void SetGasPort()
|
|
{
|
|
if (!Owner.TryGetComponent<NodeContainerComponent>(out var container))
|
|
{
|
|
Logger.Warning($"{nameof(GasCanisterPortComponent)} on {Owner?.Prototype?.ID}, Uid {Owner?.Uid} did not have a {nameof(NodeContainerComponent)}.");
|
|
return;
|
|
}
|
|
_gasPort = container.Nodes.OfType<PipeNode>().FirstOrDefault();
|
|
if (_gasPort == null)
|
|
{
|
|
Logger.Warning($"{nameof(GasCanisterPortComponent)} on {Owner?.Prototype?.ID}, Uid {Owner?.Uid} could not find compatible {nameof(PipeNode)}s on its {nameof(NodeContainerComponent)}.");
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|