Re-organize all projects (#4166)

Content.Server/Power/Components/PowerProviderComponent.cs

@@ -0,0 +1,174 @@
+#nullable enable
+using System;
+using System.Collections.Generic;
+using Content.Server.APC;
+using Content.Server.APC.Components;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Serialization.Manager.Attributes;
+using Robust.Shared.ViewVariables;
+
+namespace Content.Server.Power.Components
+{
+    /// <summary>
+    ///     Relays <see cref="PowerReceiverComponent"/>s in an area to a <see cref="IApcNet"/> so they can receive power.
+    /// </summary>
+    public interface IPowerProvider
+    {
+        void AddReceiver(PowerReceiverComponent receiver);
+
+        void RemoveReceiver(PowerReceiverComponent receiver);
+
+        void UpdateReceiverLoad(int oldLoad, int newLoad);
+
+        public IEntity? ProviderOwner { get; }
+
+        public bool HasApcPower { get; }
+    }
+
+    [RegisterComponent]
+    public class PowerProviderComponent : BaseApcNetComponent, IPowerProvider
+    {
+        public override string Name => "PowerProvider";
+
+        public IEntity ProviderOwner => Owner;
+
+        [ViewVariables]
+        public bool HasApcPower => Net.Powered;
+
+        /// <summary>
+        ///     The max distance this can transmit power to <see cref="PowerReceiverComponent"/>s from.
+        /// </summary>
+        [ViewVariables(VVAccess.ReadWrite)]
+        public int PowerTransferRange { get => _powerTransferRange; set => SetPowerTransferRange(value); }
+        [DataField("powerTransferRange")]
+        private int _powerTransferRange = 3;
+
+        [ViewVariables]
+        public IReadOnlyList<PowerReceiverComponent> LinkedReceivers => _linkedReceivers;
+        private List<PowerReceiverComponent> _linkedReceivers = new();
+
+        /// <summary>
+        ///     If <see cref="PowerReceiverComponent"/>s should consider connecting to this.
+        /// </summary>
+        [ViewVariables(VVAccess.ReadWrite)]
+        public bool Connectable { get; private set; } = true;
+
+        public static readonly IPowerProvider NullProvider = new NullPowerProvider();
+
+        public void AddReceiver(PowerReceiverComponent receiver)
+        {
+            var oldLoad = GetTotalLoad();
+            _linkedReceivers.Add(receiver);
+            var newLoad = oldLoad + receiver.Load;
+            Net.UpdatePowerProviderReceivers(this, oldLoad, newLoad);
+        }
+
+        public void RemoveReceiver(PowerReceiverComponent receiver)
+        {
+            var oldLoad = GetTotalLoad();
+            _linkedReceivers.Remove(receiver);
+            var newLoad = oldLoad - receiver.Load;
+            Net.UpdatePowerProviderReceivers(this, oldLoad, newLoad);
+        }
+
+        public void UpdateReceiverLoad(int oldLoad, int newLoad)
+        {
+            Net.UpdatePowerProviderReceivers(this, oldLoad, newLoad);
+        }
+
+        protected override void Startup()
+        {
+            base.Startup();
+            foreach (var receiver in FindAvailableReceivers())
+            {
+                receiver.Provider = this;
+            }
+        }
+
+        public override void OnRemove()
+        {
+            Connectable = false;
+            var receivers = _linkedReceivers.ToArray();
+            foreach (var receiver in receivers)
+            {
+                receiver.ClearProvider();
+            }
+            foreach (var receiver in receivers)
+            {
+                receiver.TryFindAndSetProvider();
+            }
+            base.OnRemove();
+        }
+
+        private List<PowerReceiverComponent> FindAvailableReceivers()
+        {
+            var nearbyEntities = IoCManager.Resolve<IEntityLookup>()
+                .GetEntitiesInRange(Owner, PowerTransferRange);
+
+            var receivers = new List<PowerReceiverComponent>();
+
+            foreach (var entity in nearbyEntities)
+            {
+                if (entity.TryGetComponent<PowerReceiverComponent>(out var receiver) &&
+                    receiver.Connectable &&
+                    receiver.NeedsProvider &&
+                    receiver.Owner.Transform.Coordinates.TryDistance(Owner.EntityManager, Owner.Transform.Coordinates, out var distance) &&
+                    distance < Math.Min(PowerTransferRange, receiver.PowerReceptionRange))
+                {
+                    receivers.Add(receiver);
+                }
+            }
+            return receivers;
+        }
+
+        protected override void AddSelfToNet(IApcNet apcNet)
+        {
+            apcNet.AddPowerProvider(this);
+        }
+
+        protected override void RemoveSelfFromNet(IApcNet apcNet)
+        {
+            apcNet.RemovePowerProvider(this);
+        }
+
+        private void SetPowerTransferRange(int newPowerTransferRange)
+        {
+            var receivers = _linkedReceivers.ToArray();
+
+            foreach (var receiver in receivers)
+            {
+                receiver.ClearProvider();
+            }
+            _powerTransferRange = newPowerTransferRange;
+
+            foreach (var receiver in receivers)
+            {
+                receiver.TryFindAndSetProvider();
+            }
+        }
+
+        private int GetTotalLoad()
+        {
+            var load = 0;
+            foreach (var receiver in _linkedReceivers)
+            {
+                load += receiver.Load;
+            }
+            return load;
+        }
+
+        private class NullPowerProvider : IPowerProvider
+        {
+            /// <summary>
+            ///     It is important that this returns false, so <see cref="PowerReceiverComponent"/>s with a <see cref="NullPowerProvider"/> have no power.
+            /// </summary>
+            public bool HasApcPower => false;
+
+            public void AddReceiver(PowerReceiverComponent receiver) { }
+            public void RemoveReceiver(PowerReceiverComponent receiver) { }
+            public void UpdateReceiverLoad(int oldLoad, int newLoad) { }
+            public IEntity? ProviderOwner => default;
+        }
+    }
+}