Pow3r: stage 1 (#4208)

Co-authored-by: 20kdc <asdd2808@gmail.com>

Content.Server/Power/Components/BatteryDischargerComponent.cs

@@ -1,79 +1,21 @@
-#nullable enable
-using Content.Server.Battery.Components;
+using Content.Server.Power.NodeGroups;
 using Robust.Shared.GameObjects;
-using Robust.Shared.Serialization.Manager.Attributes;
-using Robust.Shared.ViewVariables;
 
 namespace Content.Server.Power.Components
 {
-    /// <summary>
-    ///     Uses charge from a <see cref="BatteryComponent"/> to supply power via a <see cref="PowerSupplierComponent"/>.
-    /// </summary>
     [RegisterComponent]
-    public class BatteryDischargerComponent : Component
+    public class BatteryDischargerComponent : BasePowerNetComponent
     {
         public override string Name => "BatteryDischarger";
 
-        [ViewVariables]
-        [ComponentDependency] private BatteryComponent? _battery = default!;
-
-        [ViewVariables]
-        [ComponentDependency] private PowerSupplierComponent? _supplier = default!;
-
-        [ViewVariables(VVAccess.ReadWrite)]
-        public int ActiveSupplyRate { get => _activeSupplyRate; set => SetActiveSupplyRate(value); }
-
-        [DataField("activeSupplyRate")]
-        private int _activeSupplyRate = 50;
-
-        protected override void Initialize()
+        protected override void AddSelfToNet(IPowerNet net)
         {
-            base.Initialize();
-            Owner.EnsureComponentWarn<PowerSupplierComponent>();
-            UpdateSupplyRate();
+            net.AddDischarger(this);
         }
 
-        public void Update(float frameTime)
+        protected override void RemoveSelfFromNet(IPowerNet net)
         {
-            if (_battery == null)
-                return;
-
-            //Simplified implementation - if the battery is empty, and charge is being added to the battery
-            //at a lower rate that this is using it, the charge is used without creating power supply.
-            _battery.CurrentCharge -= ActiveSupplyRate * frameTime;
-            UpdateSupplyRate();
-        }
-
-        private void UpdateSupplyRate()
-        {
-            if (_battery == null)
-                return;
-
-            if (_battery.BatteryState == BatteryState.Empty)
-            {
-                SetSupplierSupplyRate(0);
-            }
-            else
-            {
-                SetSupplierSupplyRate(ActiveSupplyRate);
-            }
-        }
-
-        private void SetSupplierSupplyRate(int newSupplierSupplyRate)
-        {
-            if (_supplier == null)
-                return;
-
-            if (_supplier.SupplyRate != newSupplierSupplyRate)
-            {
-                _supplier.SupplyRate = newSupplierSupplyRate;
-            }
-        }
-
-        private void SetActiveSupplyRate(int newEnabledSupplyRate)
-        {
-            _activeSupplyRate = newEnabledSupplyRate;
-            UpdateSupplyRate();
+            net.RemoveDischarger(this);
         }
     }
 }