Stamina damage (#9230)

2022-07-06 18:06:12 +10:00
parent 305cdc02cb
commit 40eecdd78a
129 changed files with 703 additions and 778 deletions
--- a/Content.Server/Stunnable/Systems/StunbatonSystem.cs
+++ b/Content.Server/Stunnable/Systems/StunbatonSystem.cs
@@ -0,0 +1,134 @@
+using System.Linq;
+using Content.Server.Damage.Components;
+using Content.Server.Damage.Events;
+using Content.Server.Power.Components;
+using Content.Server.Power.Events;
+using Content.Server.Speech.EntitySystems;
+using Content.Server.Stunnable.Components;
+using Content.Server.Weapon.Melee;
+using Content.Shared.Audio;
+using Content.Shared.Examine;
+using Content.Shared.Interaction.Events;
+using Content.Shared.Item;
+using Content.Shared.Jittering;
+using Content.Shared.Popups;
+using Content.Shared.StatusEffect;
+using Content.Shared.Throwing;
+using Robust.Server.GameObjects;
+using Robust.Shared.Audio;
+using Robust.Shared.Player;
+using Robust.Shared.Random;
+
+namespace Content.Server.Stunnable.Systems
+{
+    public sealed class StunbatonSystem : EntitySystem
+    {
+        public override void Initialize()
+        {
+            base.Initialize();
+
+            SubscribeLocalEvent<StunbatonComponent, UseInHandEvent>(OnUseInHand);
+            SubscribeLocalEvent<StunbatonComponent, ExaminedEvent>(OnExamined);
+            SubscribeLocalEvent<StunbatonComponent, StaminaDamageOnHitAttemptEvent>(OnStaminaHitAttempt);
+            SubscribeLocalEvent<StunbatonComponent, MeleeHitEvent>(OnMeleeHit);
+        }
+
+        private void OnMeleeHit(EntityUid uid, StunbatonComponent component, MeleeHitEvent args)
+        {
+            if (!component.Activated) return;
+
+            // Don't apply damage if it's activated; just do stamina damage.
+            args.BonusDamage -= args.BaseDamage;
+        }
+
+        private void OnStaminaHitAttempt(EntityUid uid, StunbatonComponent component, ref StaminaDamageOnHitAttemptEvent args)
+        {
+            if (!component.Activated ||
+                !TryComp<BatteryComponent>(uid, out var battery) || !battery.TryUseCharge(component.EnergyPerUse))
+            {
+                args.Cancelled = true;
+                return;
+            }
+
+            if (battery.CurrentCharge < component.EnergyPerUse)
+            {
+                SoundSystem.Play(component.SparksSound.GetSound(), Filter.Pvs(component.Owner, entityManager: EntityManager), uid, AudioHelpers.WithVariation(0.25f));
+                TurnOff(component);
+            }
+        }
+
+        private void OnUseInHand(EntityUid uid, StunbatonComponent comp, UseInHandEvent args)
+        {
+            if (comp.Activated)
+            {
+                TurnOff(comp);
+            }
+            else
+            {
+                TurnOn(comp, args.User);
+            }
+        }
+
+        private void OnExamined(EntityUid uid, StunbatonComponent comp, ExaminedEvent args)
+        {
+            var msg = comp.Activated
+                ? Loc.GetString("comp-stunbaton-examined-on")
+                : Loc.GetString("comp-stunbaton-examined-off");
+            args.PushMarkup(msg);
+            if(TryComp<BatteryComponent>(uid, out var battery))
+                args.PushMarkup(Loc.GetString("stunbaton-component-on-examine-charge",
+                    ("charge", (int)((battery.CurrentCharge/battery.MaxCharge) * 100))));
+        }
+
+        private void TurnOff(StunbatonComponent comp)
+        {
+            if (!comp.Activated)
+                return;
+
+            // TODO stunbaton visualizer
+            if (TryComp<SpriteComponent>(comp.Owner, out var sprite) &&
+                TryComp<SharedItemComponent>(comp.Owner, out var item))
+            {
+                item.EquippedPrefix = "off";
+                sprite.LayerSetState(0, "stunbaton_off");
+            }
+
+            SoundSystem.Play(comp.SparksSound.GetSound(), Filter.Pvs(comp.Owner), comp.Owner, AudioHelpers.WithVariation(0.25f));
+
+            comp.Activated = false;
+        }
+
+        private void TurnOn(StunbatonComponent comp, EntityUid user)
+        {
+            if (comp.Activated)
+                return;
+
+            var playerFilter = Filter.Pvs(comp.Owner, entityManager: EntityManager);
+            if (!TryComp<BatteryComponent>(comp.Owner, out var battery) || battery.CurrentCharge < comp.EnergyPerUse)
+            {
+                SoundSystem.Play(comp.TurnOnFailSound.GetSound(), playerFilter, comp.Owner, AudioHelpers.WithVariation(0.25f));
+                user.PopupMessage(Loc.GetString("stunbaton-component-low-charge"));
+                return;
+            }
+
+            if (EntityManager.TryGetComponent<SpriteComponent?>(comp.Owner, out var sprite) &&
+                EntityManager.TryGetComponent<SharedItemComponent?>(comp.Owner, out var item))
+            {
+                item.EquippedPrefix = "on";
+                sprite.LayerSetState(0, "stunbaton_on");
+            }
+
+            SoundSystem.Play(comp.SparksSound.GetSound(), playerFilter, comp.Owner, AudioHelpers.WithVariation(0.25f));
+            comp.Activated = true;
+        }
+
+        private void SendPowerPulse(EntityUid target, EntityUid? user, EntityUid used)
+        {
+            RaiseLocalEvent(target, new PowerPulseEvent()
+            {
+                Used = used,
+                User = user
+            }, false);
+        }
+    }
+}