Pet internals (#12376)

* toggle other crew and pet internals

* oops

Content.Server/Body/Systems/InternalsSystem.cs

@@ -7,6 +7,12 @@ using Content.Shared.Atmos;
 using Content.Shared.Inventory;
 using Robust.Shared.Containers;
 using Robust.Shared.Prototypes;
+using Content.Shared.Verbs;
+using Robust.Server.GameObjects;
+using Content.Server.Popups;
+using Robust.Shared.Player;
+using Content.Server.DoAfter;
+using System.Threading;
 
 namespace Content.Server.Body.Systems;
 
@@ -18,6 +24,8 @@ public sealed class InternalsSystem : EntitySystem
     [Dependency] private readonly GasTankSystem _gasTank = default!;
     [Dependency] private readonly HandsSystem _hands = default!;
     [Dependency] private readonly InventorySystem _inventory = default!;
+    [Dependency] private readonly PopupSystem _popupSystem = default!;
+    [Dependency] private readonly DoAfterSystem _doAfter = default!;
 
     public override void Initialize()
     {
@@ -26,6 +34,93 @@ public sealed class InternalsSystem : EntitySystem
         SubscribeLocalEvent<InternalsComponent, InhaleLocationEvent>(OnInhaleLocation);
         SubscribeLocalEvent<InternalsComponent, ComponentStartup>(OnInternalsStartup);
         SubscribeLocalEvent<InternalsComponent, ComponentShutdown>(OnInternalsShutdown);
+        SubscribeLocalEvent<InternalsComponent, GetVerbsEvent<InteractionVerb>>(OnGetInteractionVerbs);
+        SubscribeLocalEvent<ToggleOtherInternalsCompleteEvent>(OnToggleOtherInternalsComplete);
+        SubscribeLocalEvent<ToggleOtherInternalsCancelledEvent>(OnToggleOtherInternalCanceled);
+    }
+
+    private void OnGetInteractionVerbs(EntityUid uid, InternalsComponent component, GetVerbsEvent<InteractionVerb> args)
+    {
+        if (!args.CanAccess || !args.CanInteract || args.Hands == null)
+            return;
+
+        if (!EntityManager.TryGetComponent<ActorComponent?>(args.User, out var actor))
+            return;
+
+        InteractionVerb verb = new()
+        {
+            Act = () =>
+            {
+                var entManager = IoCManager.Resolve<IEntityManager>();
+                if (!entManager.TryGetComponent<InternalsComponent>(uid, out var internals)) return;
+
+                // Toggle off if they're on
+                if (AreInternalsWorking(component))
+                {
+                    DisconnectTank(component);
+                    return;
+                }
+
+                // If they're not on then check if we have a mask to use
+                if (component.BreathToolEntity == null)
+                {
+                    _popupSystem.PopupEntity(Loc.GetString("internals-no-breath-tool"), args.User, Filter.Entities(args.User));
+                    return;
+                }
+
+                var tank = FindBestGasTank(component);
+                if (tank == null)
+                {
+                    _popupSystem.PopupEntity(Loc.GetString("internals-no-tank"), args.User, Filter.Entities(args.User));
+                    return;
+                }
+
+                // Is the target not you? If yes, use a do-after to give them time to respond.
+                if (args.User != args.Target)
+                {
+                    component.CancelToken = new CancellationTokenSource();
+                    _doAfter.DoAfter(new DoAfterEventArgs(args.User, component.Delay, component.CancelToken.Token, args.Target)
+                    {
+                        BreakOnUserMove = true,
+                        BreakOnDamage = true,
+                        BreakOnStun = true,
+                        BreakOnTargetMove = true,
+                        MovementThreshold = 0.1f,
+                        BroadcastFinishedEvent = new ToggleOtherInternalsCompleteEvent()
+                        {
+                            Component = component,
+                            User = args.User,
+                            Target = args.Target,
+                            Tank = tank
+                        },
+                        BroadcastCancelledEvent = new ToggleOtherInternalsCancelledEvent()
+                        {
+                            Component = component,
+                        }
+                    });
+                    return;
+                }
+                entManager.EntitySysManager.GetEntitySystem<GasTankSystem>().ConnectToInternals(tank);
+            },
+            Message = Loc.GetString("action-description-internals-toggle"),
+            IconTexture = "/Textures/Interface/VerbIcons/dot.svg.192dpi.png",
+            Text = Loc.GetString("action-name-internals-toggle"),
+        };
+
+        args.Verbs.Add(verb);
+    }
+
+    private void OnToggleOtherInternalsComplete(ToggleOtherInternalsCompleteEvent ev)
+    {
+        ev.Component.CancelToken = null;
+        var entManager = IoCManager.Resolve<IEntityManager>();
+        if (ev.Tank != null)
+            entManager.EntitySysManager.GetEntitySystem<GasTankSystem>().ConnectToInternals(ev.Tank);
+    }
+
+    private static void OnToggleOtherInternalCanceled(ToggleOtherInternalsCancelledEvent ev)
+    {
+        ev.Component.CancelToken = null;
     }
 
     private void OnInternalsStartup(EntityUid uid, InternalsComponent component, ComponentStartup args)
@@ -174,4 +269,16 @@ public sealed class InternalsSystem : EntitySystem
 
         return null;
     }
+    private sealed class ToggleOtherInternalsCompleteEvent : EntityEventArgs
+    {
+        public InternalsComponent Component = default!;
+        public EntityUid User { get; init; }
+        public EntityUid Target { get; init; }
+        public GasTankComponent? Tank { get; init; }
+    }
+
+    private sealed class ToggleOtherInternalsCancelledEvent : EntityEventArgs
+    {
+        public InternalsComponent Component = default!;
+    }
 }