Melee refactor (#10897)

Co-authored-by: metalgearsloth <metalgearsloth@gmail.com>
2022-09-29 15:51:59 +10:00
parent c583b7b361
commit f51248ecaa
140 changed files with 2440 additions and 1824 deletions
--- a/Content.Server/Weapons/Ranged/Systems/GunSystem.Battery.cs
+++ b/Content.Server/Weapons/Ranged/Systems/GunSystem.Battery.cs
@@ -0,0 +1,133 @@
+using Content.Server.Power.Components;
+using Content.Server.Projectiles.Components;
+using Content.Shared.Damage;
+using Content.Shared.FixedPoint;
+using Content.Shared.Verbs;
+using Content.Shared.Weapons.Ranged;
+using Content.Shared.Weapons.Ranged.Components;
+using Robust.Shared.Prototypes;
+
+namespace Content.Server.Weapons.Ranged.Systems;
+
+public sealed partial class GunSystem
+{
+    protected override void InitializeBattery()
+    {
+        base.InitializeBattery();
+
+        // Hitscan
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, ComponentStartup>(OnBatteryStartup);
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, ChargeChangedEvent>(OnBatteryChargeChange);
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, GetVerbsEvent<ExamineVerb>>(OnBatteryExaminableVerb);
+
+        // Projectile
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, ComponentStartup>(OnBatteryStartup);
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, ChargeChangedEvent>(OnBatteryChargeChange);
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, GetVerbsEvent<ExamineVerb>>(OnBatteryExaminableVerb);
+    }
+
+    private void OnBatteryStartup(EntityUid uid, BatteryAmmoProviderComponent component, ComponentStartup args)
+    {
+        UpdateShots(uid, component);
+    }
+
+    private void OnBatteryChargeChange(EntityUid uid, BatteryAmmoProviderComponent component, ChargeChangedEvent args)
+    {
+        UpdateShots(uid, component);
+    }
+
+    private void UpdateShots(EntityUid uid, BatteryAmmoProviderComponent component)
+    {
+        if (!TryComp<BatteryComponent>(uid, out var battery)) return;
+        UpdateShots(component, battery);
+    }
+
+    private void UpdateShots(BatteryAmmoProviderComponent component, BatteryComponent battery)
+    {
+        var shots = (int) (battery.CurrentCharge / component.FireCost);
+        var maxShots = (int) (battery.MaxCharge / component.FireCost);
+
+        if (component.Shots != shots || component.Capacity != maxShots)
+        {
+            Dirty(component);
+        }
+
+        component.Shots = shots;
+        component.Capacity = maxShots;
+        UpdateBatteryAppearance(component.Owner, component);
+    }
+
+    private void OnBatteryExaminableVerb(EntityUid uid, BatteryAmmoProviderComponent component, GetVerbsEvent<ExamineVerb> args)
+    {
+        if (!args.CanInteract || !args.CanAccess)
+            return;
+
+        var damageSpec = GetDamage(component);
+
+        if (damageSpec == null)
+            return;
+
+        string damageType;
+
+        switch (component)
+        {
+            case HitscanBatteryAmmoProviderComponent:
+                damageType = Loc.GetString("damage-hitscan");
+                break;
+            case ProjectileBatteryAmmoProviderComponent:
+                damageType = Loc.GetString("damage-projectile");
+                break;
+            default:
+                throw new ArgumentOutOfRangeException();
+        }
+
+        var verb = new ExamineVerb()
+        {
+            Act = () =>
+            {
+                var markup = Damageable.GetDamageExamine(damageSpec, damageType);
+                Examine.SendExamineTooltip(args.User, uid, markup, false, false);
+            },
+            Text = Loc.GetString("damage-examinable-verb-text"),
+            Message = Loc.GetString("damage-examinable-verb-message"),
+            Category = VerbCategory.Examine,
+            IconTexture = "/Textures/Interface/VerbIcons/smite.svg.192dpi.png"
+        };
+
+        args.Verbs.Add(verb);
+    }
+
+    private DamageSpecifier? GetDamage(BatteryAmmoProviderComponent component)
+    {
+        if (component is ProjectileBatteryAmmoProviderComponent battery)
+        {
+            if (ProtoManager.Index<EntityPrototype>(battery.Prototype).Components
+                .TryGetValue(_factory.GetComponentName(typeof(ProjectileComponent)), out var projectile))
+            {
+                var p = (ProjectileComponent) projectile.Component;
+
+                if (p.Damage.Total > FixedPoint2.Zero)
+                {
+                    return p.Damage;
+                }
+            }
+
+            return null;
+        }
+
+        if (component is HitscanBatteryAmmoProviderComponent hitscan)
+        {
+            return ProtoManager.Index<HitscanPrototype>(hitscan.Prototype).Damage;
+        }
+
+        return null;
+    }
+
+    protected override void TakeCharge(EntityUid uid, BatteryAmmoProviderComponent component)
+    {
+        if (!TryComp<BatteryComponent>(uid, out var battery)) return;
+
+        battery.CurrentCharge -= component.FireCost;
+        UpdateShots(component, battery);
+    }
+}