Gun refactor (#8301)

Co-authored-by: Kara <lunarautomaton6@gmail.com> Co-authored-by: T-Stalker <le0nel_1van@hotmail.com> Co-authored-by: T-Stalker <43253663+DogZeroX@users.noreply.github.com> Co-authored-by: ElectroJr <leonsfriedrich@gmail.com> Co-authored-by: metalgearsloth <metalgearsloth@gmail.com>
2022-06-01 19:59:58 +10:00
parent 1ced3c5002
commit fb943a61dc
1051 changed files with 8230 additions and 99090 deletions
--- a/Content.Shared/Weapons/Ranged/Systems/SharedGunSystem.Battery.cs
+++ b/Content.Shared/Weapons/Ranged/Systems/SharedGunSystem.Battery.cs
@@ -0,0 +1,111 @@
+using Content.Shared.Examine;
+using Content.Shared.Weapons.Ranged.Components;
+using Content.Shared.Weapons.Ranged.Events;
+using Robust.Shared.GameStates;
+using Robust.Shared.Map;
+using Robust.Shared.Serialization;
+
+namespace Content.Shared.Weapons.Ranged.Systems;
+
+public abstract partial class SharedGunSystem
+{
+    protected virtual void InitializeBattery()
+    {
+        // Trying to dump comp references hence the below
+        // Hitscan
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, ComponentGetState>(OnBatteryGetState);
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, ComponentHandleState>(OnBatteryHandleState);
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, TakeAmmoEvent>(OnBatteryTakeAmmo);
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, GetAmmoCountEvent>(OnBatteryAmmoCount);
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, ExaminedEvent>(OnBatteryExamine);
+
+        // Projectile
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, ComponentGetState>(OnBatteryGetState);
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, ComponentHandleState>(OnBatteryHandleState);
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, TakeAmmoEvent>(OnBatteryTakeAmmo);
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, GetAmmoCountEvent>(OnBatteryAmmoCount);
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, ExaminedEvent>(OnBatteryExamine);
+    }
+
+    private void OnBatteryHandleState(EntityUid uid, BatteryAmmoProviderComponent component, ref ComponentHandleState args)
+    {
+        if (args.Current is not BatteryAmmoProviderComponentState state) return;
+
+        component.Shots = state.Shots;
+        component.Capacity = state.MaxShots;
+        component.FireCost = state.FireCost;
+    }
+
+    private void OnBatteryGetState(EntityUid uid, BatteryAmmoProviderComponent component, ref ComponentGetState args)
+    {
+        args.State = new BatteryAmmoProviderComponentState()
+        {
+            Shots = component.Shots,
+            MaxShots = component.Capacity,
+            FireCost = component.FireCost,
+        };
+    }
+
+    private void OnBatteryExamine(EntityUid uid, BatteryAmmoProviderComponent component, ExaminedEvent args)
+    {
+        args.PushMarkup(Loc.GetString("gun-battery-examine", ("color", AmmoExamineColor), ("count", component.Shots)));
+    }
+
+    private void OnBatteryTakeAmmo(EntityUid uid, BatteryAmmoProviderComponent 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(GetShootable(component, args.Coordinates));
+            component.Shots--;
+        }
+
+        TakeCharge(uid, component);
+        UpdateBatteryAppearance(uid, component);
+        Dirty(component);
+    }
+
+    private void OnBatteryAmmoCount(EntityUid uid, BatteryAmmoProviderComponent component, ref GetAmmoCountEvent args)
+    {
+        args.Count = component.Shots;
+        args.Capacity = component.Capacity;
+    }
+
+    /// <summary>
+    /// Update the battery (server-only) whenever fired.
+    /// </summary>
+    protected virtual void TakeCharge(EntityUid uid, BatteryAmmoProviderComponent component) {}
+
+    protected void UpdateBatteryAppearance(EntityUid uid, BatteryAmmoProviderComponent component)
+    {
+        if (!TryComp<AppearanceComponent>(uid, out var appearance)) return;
+        appearance.SetData(AmmoVisuals.AmmoCount, component.Shots);
+        appearance.SetData(AmmoVisuals.AmmoMax, component.Capacity);
+    }
+
+    private IShootable GetShootable(BatteryAmmoProviderComponent component, EntityCoordinates coordinates)
+    {
+        switch (component)
+        {
+            case ProjectileBatteryAmmoProviderComponent proj:
+                var ent = Spawn(proj.Prototype, coordinates);
+                return EnsureComp<AmmoComponent>(ent);
+            case HitscanBatteryAmmoProviderComponent hitscan:
+                return ProtoManager.Index<HitscanPrototype>(hitscan.Prototype);
+            default:
+                throw new ArgumentOutOfRangeException();
+        }
+    }
+
+    [Serializable, NetSerializable]
+    private sealed class BatteryAmmoProviderComponentState : ComponentState
+    {
+        public int Shots;
+        public int MaxShots;
+        public float FireCost;
+    }
+}