Spray Nozzle & Backpack Water Tank (#16133)

2023-05-05 21:50:09 -04:00
parent e80568747d
commit 98481fb9a2
30 changed files with 495 additions and 7 deletions
--- a/Content.Shared/Weapons/Ranged/Systems/SharedGunSystem.Solution.cs
+++ b/Content.Shared/Weapons/Ranged/Systems/SharedGunSystem.Solution.cs
@@ -0,0 +1,60 @@
+using Content.Shared.Chemistry.Components;
+using Content.Shared.Weapons.Ranged.Components;
+using Content.Shared.Weapons.Ranged.Events;
+using Robust.Shared.Map;
+
+namespace Content.Shared.Weapons.Ranged.Systems;
+
+public partial class SharedGunSystem
+{
+    protected virtual void InitializeSolution()
+    {
+        SubscribeLocalEvent<SolutionAmmoProviderComponent, TakeAmmoEvent>(OnSolutionTakeAmmo);
+        SubscribeLocalEvent<SolutionAmmoProviderComponent, GetAmmoCountEvent>(OnSolutionAmmoCount);
+    }
+
+    private void OnSolutionTakeAmmo(EntityUid uid, SolutionAmmoProviderComponent component, TakeAmmoEvent args)
+    {
+        var shots = Math.Min(args.Shots, component.Shots);
+
+        // Don't dirty if it's an empty fire.
+        if (shots == 0)
+            return;
+
+        for (var i = 0; i < shots; i++)
+        {
+            args.Ammo.Add(GetSolutionShot(uid, component, args.Coordinates));
+            component.Shots--;
+        }
+
+        UpdateSolutionShots(uid, component);
+        UpdateSolutionAppearance(uid, component);
+    }
+
+    private void OnSolutionAmmoCount(EntityUid uid, SolutionAmmoProviderComponent component, ref GetAmmoCountEvent args)
+    {
+        args.Count = component.Shots;
+        args.Capacity = component.MaxShots;
+    }
+
+    protected virtual void UpdateSolutionShots(EntityUid uid, SolutionAmmoProviderComponent component, Solution? solution = null)
+    {
+
+    }
+
+    protected virtual (EntityUid Entity, IShootable) GetSolutionShot(EntityUid uid, SolutionAmmoProviderComponent component, EntityCoordinates position)
+    {
+        var ent = Spawn(component.Prototype, position);
+        return (ent, EnsureComp<AmmoComponent>(ent));
+    }
+
+    protected void UpdateSolutionAppearance(EntityUid uid, SolutionAmmoProviderComponent component)
+    {
+        if (!TryComp<AppearanceComponent>(uid, out var appearance))
+            return;
+
+        Appearance.SetData(uid, AmmoVisuals.HasAmmo, component.Shots != 0, appearance);
+        Appearance.SetData(uid, AmmoVisuals.AmmoCount, component.Shots, appearance);
+        Appearance.SetData(uid, AmmoVisuals.AmmoMax, component.MaxShots, appearance);
+    }
+}