Magic staves + wands (#9070)

2022-07-14 19:45:27 -07:00
parent 9608005db0
commit ad7889e8a9
85 changed files with 1299 additions and 34 deletions
--- a/Content.Shared/Weapons/Ranged/Systems/SharedGunSystem.BasicEntity.cs
+++ b/Content.Shared/Weapons/Ranged/Systems/SharedGunSystem.BasicEntity.cs
@@ -0,0 +1,96 @@
+using Content.Shared.Weapons.Ranged.Components;
+using Content.Shared.Weapons.Ranged.Events;
+using Robust.Shared.GameStates;
+
+namespace Content.Shared.Weapons.Ranged.Systems;
+
+public abstract partial class SharedGunSystem
+{
+    protected virtual void InitializeBasicEntity()
+    {
+        SubscribeLocalEvent<BasicEntityAmmoProviderComponent, ComponentInit>(OnBasicEntityInit);
+        SubscribeLocalEvent<BasicEntityAmmoProviderComponent, TakeAmmoEvent>(OnBasicEntityTakeAmmo);
+        SubscribeLocalEvent<BasicEntityAmmoProviderComponent, GetAmmoCountEvent>(OnBasicEntityAmmoCount);
+
+        SubscribeLocalEvent<BasicEntityAmmoProviderComponent, ComponentGetState>(OnBasicEntityGetState);
+        SubscribeLocalEvent<BasicEntityAmmoProviderComponent, ComponentHandleState>(OnBasicEntityHandleState);
+    }
+
+    private void OnBasicEntityGetState(EntityUid uid, BasicEntityAmmoProviderComponent component, ref ComponentGetState args)
+    {
+        args.State = new BasicEntityAmmoProviderComponentState(component.Capacity, component.Count);
+    }
+
+    private void OnBasicEntityHandleState(EntityUid uid, BasicEntityAmmoProviderComponent component, ref ComponentHandleState args)
+    {
+        if (args.Current is BasicEntityAmmoProviderComponentState state)
+        {
+            component.Capacity = state.Capacity;
+            component.Count = state.Count;
+        }
+    }
+
+    private void OnBasicEntityInit(EntityUid uid, BasicEntityAmmoProviderComponent component, ComponentInit args)
+    {
+        if (component.Count is null)
+        {
+            component.Count = component.Capacity;
+            Dirty(component);
+        }
+
+        UpdateBasicEntityAppearance(component);
+    }
+
+    private void OnBasicEntityTakeAmmo(EntityUid uid, BasicEntityAmmoProviderComponent component, TakeAmmoEvent args)
+    {
+        for (int i = 0; i < args.Shots; i++)
+        {
+            if (component.Count <= 0)
+                return;
+
+            if (component.Count != null)
+            {
+                component.Count--;
+            }
+
+            var ent = Spawn(component.Proto, args.Coordinates);
+            args.Ammo.Add(EnsureComp<AmmoComponent>(ent));
+        }
+
+        UpdateBasicEntityAppearance(component);
+        Dirty(component);
+    }
+
+    private void OnBasicEntityAmmoCount(EntityUid uid, BasicEntityAmmoProviderComponent component, ref GetAmmoCountEvent args)
+    {
+        args.Capacity = component.Capacity ?? int.MaxValue;
+        args.Count = component.Count ?? int.MaxValue;
+    }
+
+    private void UpdateBasicEntityAppearance(BasicEntityAmmoProviderComponent component)
+    {
+        if (!Timing.IsFirstTimePredicted || !TryComp<AppearanceComponent>(component.Owner, out var appearance)) return;
+        appearance.SetData(AmmoVisuals.HasAmmo, component.Count != 0);
+        appearance.SetData(AmmoVisuals.AmmoCount, component.Count ?? int.MaxValue);
+        appearance.SetData(AmmoVisuals.AmmoMax, component.Capacity ?? int.MaxValue);
+    }
+
+    #region Public API
+
+    public bool UpdateBasicEntityAmmoCount(EntityUid uid, int count, BasicEntityAmmoProviderComponent? component = null)
+    {
+        if (!Resolve(uid, ref component))
+            return false;
+
+        if (count > component.Capacity)
+            return false;
+
+        component.Count = count;
+        Dirty(component);
+        UpdateBasicEntityAppearance(component);
+
+        return true;
+    }
+
+    #endregion
+}