Disposal mailing (#2194)

* Implement device networking * Implement device configuration menu * Fix device network * Implement disposal mailing unit * Implement base network connection Implement wired and wireless network connection Implement device network metadata * Fix dereference null error * Fix wired network null checks * Change BaseNetworks enum to NetworkUtils class Add PingResponse function to NetworkUtils Change device network file structure * Add doc comments * Apply suggestions from code review Co-authored-by: DrSmugleaf <DrSmugleaf@users.noreply.github.com> * Add tag validation to disposal mailing unit * Add tag validation to the mailing unit component * Address reviews Change WiredNetwork can connect check Change device networking string literals to constants * Address reviews Revert changes to PowerProvider and PowerReceiver Add new NodeGroup WELP * Fix recursive access to Owner property * Integrate suggested changes * Fix TryGetWireNet acting on NullPowerProvider Fix network connections not checking if their owner has been deleted * Close device network connection when the owning entity got deleted Fix mailing unit not closing the device network connection on remove * Remove GetWireNet from NullPowerProvider Co-authored-by: Julian Giebel <j.giebel@netrocks.info> Co-authored-by: DrSmugleaf <DrSmugleaf@users.noreply.github.com>
2020-10-30 01:16:26 +01:00
parent 1b3cbd4d3a
commit 45b610f933
31 changed files with 2450 additions and 13 deletions
--- a/Content.Server/DeviceNetwork/NetworkConnections/WiredNetworkConnection.cs
+++ b/Content.Server/DeviceNetwork/NetworkConnections/WiredNetworkConnection.cs
@@ -0,0 +1,84 @@
+using Content.Server.GameObjects.Components.NodeContainer;
+using Content.Server.GameObjects.Components.NodeContainer.NodeGroups;
+using Content.Server.GameObjects.Components.Power.ApcNetComponents;
+using Robust.Shared.Interfaces.GameObjects;
+using System.Collections.Generic;
+
+namespace Content.Server.GameObjects.EntitySystems.DeviceNetwork
+{
+    public class WiredNetworkConnection : BaseNetworkConnection
+    {
+        public const string WIRENET = "powernet";
+
+        private readonly IEntity _owner;
+
+        public WiredNetworkConnection(OnReceiveNetMessage onReceive, bool receiveAll, IEntity owner) : base(NetworkUtils.WIRED, 0, onReceive, receiveAll)
+        {
+            _owner = owner;
+        }
+
+        protected override bool CanReceive(int frequency, string sender, IReadOnlyDictionary<string, string> payload, Metadata metadata, bool broadcast)
+        {
+            if (_owner.Deleted)
+            {
+                Connection.Close();
+                return false;
+            }
+
+            if (_owner.TryGetComponent<PowerReceiverComponent>(out var powerReceiver)
+                && TryGetWireNet(powerReceiver, out var ownNet)
+                && metadata.TryParseMetadata<INodeGroup>(WIRENET, out var senderNet))
+            {
+                return ownNet.Equals(senderNet);
+            }
+
+            return false;
+        }
+
+        protected override Metadata GetMetadata()
+        {
+            if (_owner.Deleted)
+            {
+                Connection.Close();
+                return new Metadata();
+            }
+
+            if (_owner.TryGetComponent<PowerReceiverComponent>(out var powerReceiver)
+                && TryGetWireNet(powerReceiver, out var net))
+            {
+                var metadata = new Metadata
+                {
+                    {WIRENET, net }
+                };
+
+                return metadata;
+            }
+
+            return new Metadata();
+        }
+
+        protected override Dictionary<string, string> ManipulatePayload(Dictionary<string, string> payload)
+        {
+            return payload;
+        }
+
+        private bool TryGetWireNet(PowerReceiverComponent powerReceiver, out INodeGroup net)
+        {
+            if (powerReceiver.Provider is PowerProviderComponent && powerReceiver.Provider.ProviderOwner.TryGetComponent<NodeContainerComponent>(out var nodeContainer))
+            {
+                var nodes = nodeContainer.Nodes;
+                for (var index = 0; index < nodes.Count; index++)
+                {
+                    if (nodes[index].NodeGroupID == NodeGroupID.WireNet)
+                    {
+                        net = nodes[index].NodeGroup;
+                        return true;
+                    }
+                }
+
+            }
+            net = default;
+            return false;
+        }
+    }
+}