ninja 2 electric boogaloo (#15534)

Co-authored-by: deltanedas <@deltanedas:kde.org>

Content.Shared/Ninja/Systems/SharedNinjaSuitSystem.cs

@@ -0,0 +1,139 @@
+using Content.Shared.Actions;
+using Content.Shared.Clothing.Components;
+using Content.Shared.Clothing.EntitySystems;
+using Content.Shared.Inventory.Events;
+using Content.Shared.Ninja.Components;
+using Content.Shared.Timing;
+using Robust.Shared.Audio;
+using Robust.Shared.Network;
+using Robust.Shared.Prototypes;
+using Robust.Shared.Serialization;
+
+namespace Content.Shared.Ninja.Systems;
+
+/// <summary>
+/// Handles (un)equipping and provides some API functions.
+/// </summary>
+public abstract class SharedNinjaSuitSystem : EntitySystem
+{
+    [Dependency] private readonly SharedAudioSystem _audio = default!;
+    [Dependency] private readonly SharedNinjaGlovesSystem _gloves = default!;
+    [Dependency] protected readonly SharedSpaceNinjaSystem _ninja = default!;
+    [Dependency] protected readonly StealthClothingSystem StealthClothing = default!;
+    [Dependency] protected readonly UseDelaySystem UseDelay = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<NinjaSuitComponent, GotEquippedEvent>(OnEquipped);
+        SubscribeLocalEvent<NinjaSuitComponent, GetItemActionsEvent>(OnGetItemActions);
+        SubscribeLocalEvent<NinjaSuitComponent, AddStealthActionEvent>(OnAddStealthAction);
+        SubscribeLocalEvent<NinjaSuitComponent, GotUnequippedEvent>(OnUnequipped);
+    }
+
+    /// <summary>
+    /// Call the shared and serverside code for when a ninja equips the suit.
+    /// </summary>
+    private void OnEquipped(EntityUid uid, NinjaSuitComponent comp, GotEquippedEvent args)
+    {
+        var user = args.Equipee;
+        if (!TryComp<SpaceNinjaComponent>(user, out var ninja))
+            return;
+
+        NinjaEquippedSuit(uid, comp, user, ninja);
+    }
+
+    /// <summary>
+    /// Add all the actions when a suit is equipped by a ninja.
+    /// </summary>
+    private void OnGetItemActions(EntityUid uid, NinjaSuitComponent comp, GetItemActionsEvent args)
+    {
+        if (!HasComp<SpaceNinjaComponent>(args.User))
+            return;
+
+        args.AddAction(ref comp.RecallKatanaActionEntity, comp.RecallKatanaAction);
+        args.AddAction(ref comp.CreateThrowingStarActionEntity, comp.CreateThrowingStarAction);
+        args.AddAction(ref comp.EmpActionEntity, comp.EmpAction);
+    }
+
+    /// <summary>
+    /// Only add stealth clothing's toggle action when equipped by a ninja.
+    /// </summary>
+    private void OnAddStealthAction(EntityUid uid, NinjaSuitComponent comp, AddStealthActionEvent args)
+    {
+        if (!HasComp<SpaceNinjaComponent>(args.User))
+            args.Cancel();
+    }
+
+    /// <summary>
+    /// Call the shared and serverside code for when anyone unequips a suit.
+    /// </summary>
+    private void OnUnequipped(EntityUid uid, NinjaSuitComponent comp, GotUnequippedEvent args)
+    {
+        UserUnequippedSuit(uid, comp, args.Equipee);
+    }
+
+    /// <summary>
+    /// Called when a suit is equipped by a space ninja.
+    /// In the future it might be changed to an explicit activation toggle/verb like gloves are.
+    /// </summary>
+    protected virtual void NinjaEquippedSuit(EntityUid uid, NinjaSuitComponent comp, EntityUid user, SpaceNinjaComponent ninja)
+    {
+        // mark the user as wearing this suit, used when being attacked among other things
+        _ninja.AssignSuit(user, uid, ninja);
+
+        // initialize phase cloak, but keep it off
+        StealthClothing.SetEnabled(uid, user, false);
+    }
+
+    /// <summary>
+    /// Force uncloaks the user and disables suit abilities.
+    /// </summary>
+    public void RevealNinja(EntityUid uid, EntityUid user, NinjaSuitComponent? comp = null, StealthClothingComponent? stealthClothing = null)
+    {
+        if (!Resolve(uid, ref comp, ref stealthClothing))
+            return;
+
+        if (!StealthClothing.SetEnabled(uid, user, false, stealthClothing))
+            return;
+
+        // previously cloaked, disable abilities for a short time
+        _audio.PlayPredicted(comp.RevealSound, uid, user);
+        // all abilities check for a usedelay on the ninja
+        var useDelay = EnsureComp<UseDelayComponent>(user);
+        useDelay.Delay = comp.DisableTime;
+        UseDelay.BeginDelay(user, useDelay);
+    }
+
+    // TODO: modify PowerCellDrain
+    /// <summary>
+    /// Returns the power used by a suit
+    /// </summary>
+    public float SuitWattage(EntityUid uid, NinjaSuitComponent? suit = null)
+    {
+        if (!Resolve(uid, ref suit))
+            return 0f;
+
+        float wattage = suit.PassiveWattage;
+        if (TryComp<StealthClothingComponent>(uid, out var stealthClothing) && stealthClothing.Enabled)
+            wattage += suit.CloakWattage;
+        return wattage;
+    }
+
+    /// <summary>
+    /// Called when a suit is unequipped, not necessarily by a space ninja.
+    /// In the future it might be changed to also have explicit deactivation via toggle.
+    /// </summary>
+    protected virtual void UserUnequippedSuit(EntityUid uid, NinjaSuitComponent comp, EntityUid user)
+    {
+        if (!TryComp<SpaceNinjaComponent>(user, out var ninja))
+            return;
+
+        // mark the user as not wearing a suit
+        _ninja.AssignSuit(user, null, ninja);
+        // disable glove abilities
+        if (ninja.Gloves != null && TryComp<NinjaGlovesComponent>(ninja.Gloves.Value, out var gloves))
+            _gloves.DisableGloves(ninja.Gloves.Value, gloves);
+    }
+}