38 lines
1.3 KiB
C#
38 lines
1.3 KiB
C#
#nullable enable
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Content.Server.NodeContainer.Nodes;
|
|
using Content.Server.Power.Components;
|
|
|
|
namespace Content.Server.NodeContainer.NodeGroups
|
|
{
|
|
public abstract class BaseNetConnectorNodeGroup<TNetConnector, TNetType> : BaseNodeGroup where TNetConnector : BaseNetConnectorComponent<TNetType>
|
|
{
|
|
private readonly Dictionary<Node, List<TNetConnector>> _netConnectorComponents = new();
|
|
|
|
protected override void OnAddNode(Node node)
|
|
{
|
|
var newNetConnectorComponents = node.Owner
|
|
.GetAllComponents<TNetConnector>()
|
|
.Where(powerComp => (NodeGroupID) powerComp.Voltage == node.NodeGroupID)
|
|
.ToList();
|
|
_netConnectorComponents[node] = newNetConnectorComponents;
|
|
foreach (var netConnectorComponent in newNetConnectorComponents)
|
|
{
|
|
SetNetConnectorNet(netConnectorComponent);
|
|
}
|
|
}
|
|
|
|
protected abstract void SetNetConnectorNet(TNetConnector netConnectorComponent);
|
|
|
|
protected override void OnRemoveNode(Node node)
|
|
{
|
|
foreach (var netConnectorComponent in _netConnectorComponents[node])
|
|
{
|
|
netConnectorComponent.ClearNet();
|
|
}
|
|
_netConnectorComponents.Remove(node);
|
|
}
|
|
}
|
|
}
|