TriggerOnMeleeHit and more (#39826)

Co-authored-by: iaada <iaada@users.noreply.github.com> Co-authored-by: slarticodefast <161409025+slarticodefast@users.noreply.github.com>
2025-08-31 13:54:15 -05:00
parent 0ba5d036a2
commit 5d25cae33d
4 changed files with 111 additions and 0 deletions
--- a/Content.Shared/Trigger/Systems/MeleeTriggerSystem.cs
+++ b/Content.Shared/Trigger/Systems/MeleeTriggerSystem.cs
@@ -0,0 +1,58 @@
+using Content.Shared.Trigger.Components.Triggers;
+using Content.Shared.Weapons.Melee.Events;
+
+namespace Content.Shared.Trigger.Systems;
+
+/// <summary>
+/// Trigger system for melee related triggers.
+/// </summary>
+public sealed class MeleeTriggerSystem : EntitySystem
+{
+    [Dependency] private readonly TriggerSystem _trigger = default!;
+
+    /// <inheritdoc/>
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<TriggerOnMeleeMissComponent, MeleeHitEvent>(OnMissTrigger);
+        SubscribeLocalEvent<TriggerOnMeleeSwingComponent, MeleeHitEvent>(OnSwingTrigger);
+        SubscribeLocalEvent<TriggerOnMeleeHitComponent, MeleeHitEvent>(OnHitTrigger);
+    }
+
+    private void OnMissTrigger(Entity<TriggerOnMeleeMissComponent> ent, ref MeleeHitEvent args)
+    {
+        if (args.HitEntities.Count == 0)
+            _trigger.Trigger(ent.Owner, args.User, ent.Comp.KeyOut);
+    }
+
+    private void OnSwingTrigger(Entity<TriggerOnMeleeSwingComponent> ent, ref MeleeHitEvent args)
+    {
+        EntityUid? target;
+        if  (args.HitEntities.Count == 0)
+            target = ent.Comp.TargetIsUser ? null : args.User;
+        else
+            target = ent.Comp.TargetIsUser ? args.HitEntities[0] : args.User;
+
+        _trigger.Trigger(ent.Owner, target, ent.Comp.KeyOut);
+    }
+
+    private void OnHitTrigger(Entity<TriggerOnMeleeHitComponent> ent, ref MeleeHitEvent args)
+    {
+        if (args.HitEntities.Count == 0)
+            return;
+
+        if (!ent.Comp.TriggerEveryHit)
+        {
+            var target = ent.Comp.TargetIsUser ? args.HitEntities[0] : args.User;
+            _trigger.Trigger(ent.Owner, target, ent.Comp.KeyOut);
+            return;
+        }
+
+        // if TriggerEveryHit
+        foreach (var target in args.HitEntities)
+        {
+            _trigger.Trigger(ent.Owner, ent.Comp.TargetIsUser ? target : args.User, ent.Comp.KeyOut);
+        }
+    }
+}