Add UpdateAmmoCount

SerbiaStrong-220 · Aug 15, 2024 · 8cdd111 · 8cdd111
1 parent 387c6fc
commit 8cdd111
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Content.Client/Weapons/Ranged/Systems/GunSystem.Battery.cs b/Content.Client/Weapons/Ranged/Systems/GunSystem.Battery.cs
@@ -1,4 +1,5 @@
 using Content.Shared.Weapons.Ranged.Components;
+using Robust.Client.GameObjects;
 
 namespace Content.Client.Weapons.Ranged.Systems;
 
@@ -10,10 +11,12 @@ protected override void InitializeBattery()
         // Hitscan
         SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, AmmoCounterControlEvent>(OnControl);
         SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, UpdateAmmoCounterEvent>(OnAmmoCountUpdate);
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, AppearanceChangeEvent>(OnAppearanceChange); //SS220 Add Multifaze gun
 
         // Projectile
         SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, AmmoCounterControlEvent>(OnControl);
         SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, UpdateAmmoCounterEvent>(OnAmmoCountUpdate);
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, AppearanceChangeEvent>(OnAppearanceChange); //SS220 Add Multifaze gun
     }
 
     private void OnAmmoCountUpdate(EntityUid uid, BatteryAmmoProviderComponent component, UpdateAmmoCounterEvent args)
@@ -27,4 +30,11 @@ private void OnControl(EntityUid uid, BatteryAmmoProviderComponent component, Am
     {
         args.Control = new BoxesStatusControl();
     }
+
+    //SS220 Add Multifaze gun begin
+    private void OnAppearanceChange(EntityUid uid, BatteryAmmoProviderComponent component, ref AppearanceChangeEvent args)
+    {
+        UpdateAmmoCount(uid);
+    }
+    //SS220 Add Multifaze gun end
 }