From d3244b604926ec89563c9d6eb05740dde7de36a4 Mon Sep 17 00:00:00 2001
From: Ilya246 <57039557+Ilya246@users.noreply.github.com>
Date: Wed, 9 Aug 2023 22:20:19 +0400
Subject: [PATCH] copy-settings-to-all-similar for scrubbers and vents in air
 alarms (#18363)

---
 .../Monitor/UI/AirAlarmBoundUserInterface.cs  |  6 +++++
 .../Atmos/Monitor/UI/AirAlarmWindow.xaml.cs   |  3 +++
 .../Atmos/Monitor/UI/Widgets/PumpControl.xaml |  5 +++-
 .../Monitor/UI/Widgets/PumpControl.xaml.cs    |  7 +++++
 .../Monitor/UI/Widgets/ScrubberControl.xaml   |  5 +++-
 .../UI/Widgets/ScrubberControl.xaml.cs        |  7 +++++
 .../Atmos/Monitor/Systems/AirAlarmSystem.cs   | 27 +++++++++++++++++++
 .../Components/SharedAirAlarmComponent.cs     | 11 ++++++++
 Resources/Locale/en-US/atmos/air-alarm-ui.ftl |  2 ++
 9 files changed, 71 insertions(+), 2 deletions(-)

diff --git a/Content.Client/Atmos/Monitor/UI/AirAlarmBoundUserInterface.cs b/Content.Client/Atmos/Monitor/UI/AirAlarmBoundUserInterface.cs
index aea0ce41e8..3f7f1e73ee 100644
--- a/Content.Client/Atmos/Monitor/UI/AirAlarmBoundUserInterface.cs
+++ b/Content.Client/Atmos/Monitor/UI/AirAlarmBoundUserInterface.cs
@@ -31,6 +31,7 @@ public sealed class AirAlarmBoundUserInterface : BoundUserInterface
 
         _window.OnClose += Close;
         _window.AtmosDeviceDataChanged += OnDeviceDataChanged;
+		_window.AtmosDeviceDataCopied += OnDeviceDataCopied;
         _window.AtmosAlarmThresholdChanged += OnThresholdChanged;
         _window.AirAlarmModeChanged += OnAirAlarmModeChanged;
         _window.AutoModeChanged += OnAutoModeChanged;
@@ -47,6 +48,11 @@ public sealed class AirAlarmBoundUserInterface : BoundUserInterface
     {
         SendMessage(new AirAlarmUpdateDeviceDataMessage(address, data));
     }
+	
+	private void OnDeviceDataCopied(IAtmosDeviceData data)
+    {
+        SendMessage(new AirAlarmCopyDeviceDataMessage(data));
+    }
 
     private void OnAirAlarmModeChanged(AirAlarmMode mode)
     {
diff --git a/Content.Client/Atmos/Monitor/UI/AirAlarmWindow.xaml.cs b/Content.Client/Atmos/Monitor/UI/AirAlarmWindow.xaml.cs
index 105394b648..de26eb5f19 100644
--- a/Content.Client/Atmos/Monitor/UI/AirAlarmWindow.xaml.cs
+++ b/Content.Client/Atmos/Monitor/UI/AirAlarmWindow.xaml.cs
@@ -18,6 +18,7 @@ namespace Content.Client.Atmos.Monitor.UI;
 public sealed partial class AirAlarmWindow : FancyWindow
 {
     public event Action<string, IAtmosDeviceData>? AtmosDeviceDataChanged;
+	public event Action<IAtmosDeviceData>? AtmosDeviceDataCopied;
     public event Action<string, AtmosMonitorThresholdType, AtmosAlarmThreshold, Gas?>? AtmosAlarmThresholdChanged;
     public event Action<AirAlarmMode>? AirAlarmModeChanged;
     public event Action<bool>? AutoModeChanged;
@@ -137,6 +138,7 @@ public sealed partial class AirAlarmWindow : FancyWindow
                 {
                     var control= new PumpControl(pump, addr);
                     control.PumpDataChanged += AtmosDeviceDataChanged!.Invoke;
+					control.PumpDataCopied += AtmosDeviceDataCopied!.Invoke;
                     _pumps.Add(addr, control);
                     CVentContainer.AddChild(control);
                 }
@@ -151,6 +153,7 @@ public sealed partial class AirAlarmWindow : FancyWindow
                 {
                     var control = new ScrubberControl(scrubber, addr);
                     control.ScrubberDataChanged += AtmosDeviceDataChanged!.Invoke;
+					control.ScrubberDataCopied += AtmosDeviceDataCopied!.Invoke;
                     _scrubbers.Add(addr, control);
                     CScrubberContainer.AddChild(control);
                 }
diff --git a/Content.Client/Atmos/Monitor/UI/Widgets/PumpControl.xaml b/Content.Client/Atmos/Monitor/UI/Widgets/PumpControl.xaml
index ad0e31fbfd..632e44a458 100644
--- a/Content.Client/Atmos/Monitor/UI/Widgets/PumpControl.xaml
+++ b/Content.Client/Atmos/Monitor/UI/Widgets/PumpControl.xaml
@@ -19,7 +19,7 @@
                         <OptionButton Name="CPressureCheck" />
                     </BoxContainer>
                 </BoxContainer>
-                <!-- Lower row: pressure bounds -->
+                <!-- Lower row: pressure bounds, copy settings -->
                 <BoxContainer Orientation="Horizontal" HorizontalExpand="True">
                     <BoxContainer Orientation="Vertical" HorizontalExpand="True">
                         <Label Text="{Loc 'air-alarm-ui-vent-external-bound-label'}" Margin="0 0 0 1" />
@@ -30,6 +30,9 @@
                         <FloatSpinBox Name="CInternalBound" HorizontalExpand="True" />
                     </BoxContainer>
                 </BoxContainer>
+                <BoxContainer Orientation="Horizontal" Margin ="0 0 0 2">
+                    <Button Name="CCopySettings" Text="{Loc 'air-alarm-ui-widget-copy'}" ToolTip="{Loc 'air-alarm-ui-widget-copy-tooltip'}" />
+                </BoxContainer>
             </BoxContainer>
         </CollapsibleBody>
     </Collapsible>
diff --git a/Content.Client/Atmos/Monitor/UI/Widgets/PumpControl.xaml.cs b/Content.Client/Atmos/Monitor/UI/Widgets/PumpControl.xaml.cs
index 94123e0dff..17b03b8468 100644
--- a/Content.Client/Atmos/Monitor/UI/Widgets/PumpControl.xaml.cs
+++ b/Content.Client/Atmos/Monitor/UI/Widgets/PumpControl.xaml.cs
@@ -17,6 +17,7 @@ public sealed partial class PumpControl : BoxContainer
     private string _address;
 
     public event Action<string, IAtmosDeviceData>? PumpDataChanged;
+	public event Action<IAtmosDeviceData>? PumpDataCopied;
 
     private CheckBox _enabled => CEnableDevice;
     private CollapsibleHeading _addressLabel => CAddress;
@@ -24,6 +25,7 @@ public sealed partial class PumpControl : BoxContainer
     private OptionButton _pressureCheck => CPressureCheck;
     private FloatSpinBox _externalBound => CExternalBound;
     private FloatSpinBox _internalBound => CInternalBound;
+	private Button _copySettings => CCopySettings;
 
     public PumpControl(GasVentPumpData data, string address)
     {
@@ -84,6 +86,11 @@ public sealed partial class PumpControl : BoxContainer
             _data.PressureChecks = (VentPressureBound) args.Id;
             PumpDataChanged?.Invoke(_address, _data);
         };
+		
+		_copySettings.OnPressed += _ =>
+		{
+			PumpDataCopied?.Invoke(_data);
+		};
     }
 
     public void ChangeData(GasVentPumpData data)
diff --git a/Content.Client/Atmos/Monitor/UI/Widgets/ScrubberControl.xaml b/Content.Client/Atmos/Monitor/UI/Widgets/ScrubberControl.xaml
index 9a147b29cb..1cb9c9ed5b 100644
--- a/Content.Client/Atmos/Monitor/UI/Widgets/ScrubberControl.xaml
+++ b/Content.Client/Atmos/Monitor/UI/Widgets/ScrubberControl.xaml
@@ -8,7 +8,7 @@
                 <BoxContainer Orientation="Horizontal" Margin="0 0 0 2">
                     <CheckBox Name="CEnableDevice" Text="{Loc 'air-alarm-ui-widget-enable'}" />
                 </BoxContainer>
-                <!-- Upper row: toggle, direction, volume rate, widenet -->
+                <!-- Upper row: toggle, direction, volume rate, widenet, copy settings -->
                 <BoxContainer Orientation="Horizontal" Margin="0 0 0 2" HorizontalExpand="True">
                     <BoxContainer Orientation="Vertical" HorizontalExpand="True">
                         <Label Text="{Loc 'air-alarm-ui-scrubber-pump-direction-label'}" Margin="0 0 0 1"/>
@@ -22,6 +22,9 @@
                 <BoxContainer>
                     <CheckBox Name="CWideNet" Text="{Loc 'air-alarm-ui-scrubber-wide-net-label'}" />
                 </BoxContainer>
+                <BoxContainer Orientation="Horizontal" Margin ="0 0 0 2">
+                    <Button Name="CCopySettings" Text="{Loc 'air-alarm-ui-widget-copy'}" ToolTip="{Loc 'air-alarm-ui-widget-copy-tooltip'}" />
+                </BoxContainer>
                 <!-- Lower row: every single gas -->
                 <Collapsible Orientation="Vertical" Margin="2 2 2 2">
                     <CollapsibleHeading Title="Gas filters" />
diff --git a/Content.Client/Atmos/Monitor/UI/Widgets/ScrubberControl.xaml.cs b/Content.Client/Atmos/Monitor/UI/Widgets/ScrubberControl.xaml.cs
index 650d04a047..f2241bcd8d 100644
--- a/Content.Client/Atmos/Monitor/UI/Widgets/ScrubberControl.xaml.cs
+++ b/Content.Client/Atmos/Monitor/UI/Widgets/ScrubberControl.xaml.cs
@@ -20,12 +20,14 @@ public sealed partial class ScrubberControl : BoxContainer
     private string _address;
 
     public event Action<string, IAtmosDeviceData>? ScrubberDataChanged;
+	public event Action<IAtmosDeviceData>? ScrubberDataCopied;
 
     private CheckBox _enabled => CEnableDevice;
     private CollapsibleHeading _addressLabel => CAddress;
     private OptionButton _pumpDirection => CPumpDirection;
     private FloatSpinBox _volumeRate => CVolumeRate;
     private CheckBox _wideNet => CWideNet;
+	private Button _copySettings => CCopySettings;
 
     private GridContainer _gases => CGasContainer;
     private Dictionary<Gas, Button> _gasControls = new();
@@ -75,6 +77,11 @@ public sealed partial class ScrubberControl : BoxContainer
             _data.PumpDirection = (ScrubberPumpDirection) args.Id;
             ScrubberDataChanged?.Invoke(_address, _data);
         };
+		
+		_copySettings.OnPressed += _ =>
+		{
+			ScrubberDataCopied?.Invoke(_data);
+		};
 
         foreach (var value in Enum.GetValues<Gas>())
         {
diff --git a/Content.Server/Atmos/Monitor/Systems/AirAlarmSystem.cs b/Content.Server/Atmos/Monitor/Systems/AirAlarmSystem.cs
index c05245c81d..9f57510d1f 100644
--- a/Content.Server/Atmos/Monitor/Systems/AirAlarmSystem.cs
+++ b/Content.Server/Atmos/Monitor/Systems/AirAlarmSystem.cs
@@ -161,6 +161,7 @@ public sealed class AirAlarmSystem : EntitySystem
         SubscribeLocalEvent<AirAlarmComponent, AirAlarmUpdateAutoModeMessage>(OnUpdateAutoMode);
         SubscribeLocalEvent<AirAlarmComponent, AirAlarmUpdateAlarmThresholdMessage>(OnUpdateThreshold);
         SubscribeLocalEvent<AirAlarmComponent, AirAlarmUpdateDeviceDataMessage>(OnUpdateDeviceData);
+        SubscribeLocalEvent<AirAlarmComponent, AirAlarmCopyDeviceDataMessage>(OnCopyDeviceData);
         SubscribeLocalEvent<AirAlarmComponent, AirAlarmTabSetMessage>(OnTabChange);
         SubscribeLocalEvent<AirAlarmComponent, DeviceListUpdateEvent>(OnDeviceListUpdate);
         SubscribeLocalEvent<AirAlarmComponent, BoundUIClosedEvent>(OnClose);
@@ -301,6 +302,32 @@ public sealed class AirAlarmSystem : EntitySystem
             UpdateUI(uid, component);
         }
     }
+    
+    private void OnCopyDeviceData(EntityUid uid, AirAlarmComponent component, AirAlarmCopyDeviceDataMessage args)
+    {
+        if (!AccessCheck(uid, args.Session.AttachedEntity, component))
+        {
+           UpdateUI(uid, component);
+            return;       
+        }
+
+        switch (args.Data)
+        {
+            case GasVentPumpData ventData:
+                foreach (string addr in component.VentData.Keys)
+                {
+                    SetData(uid, addr, args.Data);
+                }
+                break;
+                
+            case GasVentScrubberData scrubberData:
+                foreach (string addr in component.ScrubberData.Keys)
+                {
+                    SetData(uid, addr, args.Data);
+                }
+                break;
+        }
+    }
 
     private bool AccessCheck(EntityUid uid, EntityUid? user, AirAlarmComponent? component = null)
     {
diff --git a/Content.Shared/Atmos/Monitor/Components/SharedAirAlarmComponent.cs b/Content.Shared/Atmos/Monitor/Components/SharedAirAlarmComponent.cs
index a31b7624e2..e40099dc72 100644
--- a/Content.Shared/Atmos/Monitor/Components/SharedAirAlarmComponent.cs
+++ b/Content.Shared/Atmos/Monitor/Components/SharedAirAlarmComponent.cs
@@ -117,6 +117,17 @@ public sealed class AirAlarmUpdateDeviceDataMessage : BoundUserInterfaceMessage
     }
 }
 
+[Serializable, NetSerializable]
+public sealed class AirAlarmCopyDeviceDataMessage : BoundUserInterfaceMessage
+{
+    public IAtmosDeviceData Data { get; }
+
+    public AirAlarmCopyDeviceDataMessage(IAtmosDeviceData data)
+    {
+        Data = data;
+    }
+}
+
 [Serializable, NetSerializable]
 public sealed class AirAlarmUpdateAlarmThresholdMessage : BoundUserInterfaceMessage
 {
diff --git a/Resources/Locale/en-US/atmos/air-alarm-ui.ftl b/Resources/Locale/en-US/atmos/air-alarm-ui.ftl
index b4444f75a5..626f30bacb 100644
--- a/Resources/Locale/en-US/atmos/air-alarm-ui.ftl
+++ b/Resources/Locale/en-US/atmos/air-alarm-ui.ftl
@@ -40,6 +40,8 @@ air-alarm-ui-mode-none = None
 ### General
 
 air-alarm-ui-widget-enable = Enabled
+air-alarm-ui-widget-copy = Copy settings to similar devices
+air-alarm-ui-widget-copy-tooltip = Copies the settings of this device to all devices in this air alarm tab.
 air-alarm-ui-widget-ignore = Ignore
 air-alarm-ui-atmos-net-device-label = Address: {$address}