113 lines
3.0 KiB
C#
113 lines
3.0 KiB
C#
using System.Diagnostics.CodeAnalysis;
|
|
using System.Linq;
|
|
using Content.Server.NodeContainer;
|
|
using Content.Server.NodeContainer.NodeGroups;
|
|
|
|
namespace Content.Server.Power.Components
|
|
{
|
|
public interface IBaseNetConnectorComponent<in TNetType>
|
|
{
|
|
public TNetType? Net { set; }
|
|
public Voltage Voltage { get; }
|
|
public string? NodeId { get; }
|
|
}
|
|
|
|
public abstract class BaseNetConnectorComponent<TNetType> : Component, IBaseNetConnectorComponent<TNetType>
|
|
{
|
|
[Dependency] private readonly IEntityManager _entMan = default!;
|
|
|
|
[ViewVariables(VVAccess.ReadWrite)]
|
|
public Voltage Voltage { get => _voltage; set => SetVoltage(value); }
|
|
[DataField("voltage")]
|
|
private Voltage _voltage = Voltage.High;
|
|
|
|
[ViewVariables]
|
|
public TNetType? Net { get => _net; set => SetNet(value); }
|
|
private TNetType? _net;
|
|
|
|
[ViewVariables]
|
|
private bool _needsNet => _net != null;
|
|
|
|
[DataField("node")] [ViewVariables] public string? NodeId { get; set; }
|
|
|
|
protected override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
if (_needsNet)
|
|
{
|
|
TryFindAndSetNet();
|
|
}
|
|
}
|
|
|
|
protected override void OnRemove()
|
|
{
|
|
ClearNet();
|
|
base.OnRemove();
|
|
}
|
|
|
|
public void TryFindAndSetNet()
|
|
{
|
|
if (TryFindNet(out var net))
|
|
{
|
|
Net = net;
|
|
}
|
|
}
|
|
|
|
public void ClearNet()
|
|
{
|
|
if (_net != null)
|
|
RemoveSelfFromNet(_net);
|
|
}
|
|
|
|
protected abstract void AddSelfToNet(TNetType net);
|
|
|
|
protected abstract void RemoveSelfFromNet(TNetType net);
|
|
|
|
private bool TryFindNet([NotNullWhen(true)] out TNetType? foundNet)
|
|
{
|
|
if (_entMan.TryGetComponent<NodeContainerComponent?>(Owner, out var container))
|
|
{
|
|
var compatibleNet = container.Nodes.Values
|
|
.Where(node => (NodeId == null || NodeId == node.Name) && node.NodeGroupID == (NodeGroupID) Voltage)
|
|
.Select(node => node.NodeGroup)
|
|
.OfType<TNetType>()
|
|
.FirstOrDefault();
|
|
|
|
if (compatibleNet != null)
|
|
{
|
|
foundNet = compatibleNet;
|
|
return true;
|
|
}
|
|
}
|
|
foundNet = default;
|
|
return false;
|
|
}
|
|
|
|
private void SetNet(TNetType? newNet)
|
|
{
|
|
if (_net != null)
|
|
RemoveSelfFromNet(_net);
|
|
|
|
if (newNet != null)
|
|
AddSelfToNet(newNet);
|
|
|
|
_net = newNet;
|
|
}
|
|
|
|
private void SetVoltage(Voltage newVoltage)
|
|
{
|
|
ClearNet();
|
|
_voltage = newVoltage;
|
|
TryFindAndSetNet();
|
|
}
|
|
}
|
|
|
|
public enum Voltage
|
|
{
|
|
High = NodeGroupID.HVPower,
|
|
Medium = NodeGroupID.MVPower,
|
|
Apc = NodeGroupID.Apc,
|
|
}
|
|
}
|