add gateway for admeme (#17587)

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

Content.Server/Gateway/Systems/GatewaySystem.cs

@@ -0,0 +1,181 @@
+using Content.Server.Gateway.Components;
+using Content.Shared.Gateway;
+using Content.Shared.Teleportation.Components;
+using Content.Shared.Teleportation.Systems;
+using Robust.Server.GameObjects;
+using Robust.Shared.Audio;
+using Robust.Shared.GameObjects;
+using Robust.Shared.Timing;
+using System.Diagnostics.CodeAnalysis;
+using System.Linq;
+
+namespace Content.Server.Gateway.Systems;
+
+public sealed class GatewaySystem : EntitySystem
+{
+    [Dependency] private readonly IGameTiming _timing = default!;
+    [Dependency] private readonly LinkedEntitySystem _linkedEntity = default!;
+    [Dependency] private readonly SharedAppearanceSystem _appearance = default!;
+    [Dependency] private readonly SharedAudioSystem _audio = default!;
+    [Dependency] private readonly UserInterfaceSystem _ui = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<GatewayComponent, ComponentStartup>(OnStartup);
+        SubscribeLocalEvent<GatewayComponent, BoundUIOpenedEvent>(UpdateUserInterface);
+        SubscribeLocalEvent<GatewayComponent, GatewayOpenPortalMessage>(OnOpenPortal);
+
+        SubscribeLocalEvent<GatewayDestinationComponent, ComponentStartup>(OnDestinationStartup);
+        SubscribeLocalEvent<GatewayDestinationComponent, ComponentShutdown>(OnDestinationShutdown);
+    }
+
+    public override void Update(float frameTime)
+    {
+        base.Update(frameTime);
+
+        // close portals after theyve been open long enough
+        var query = EntityQueryEnumerator<GatewayComponent, PortalComponent>();
+        while (query.MoveNext(out var uid, out var comp, out var _))
+        {
+            if (_timing.CurTime < comp.NextClose)
+                continue;
+
+            ClosePortal(uid, comp);
+        }
+    }
+
+    private void OnStartup(EntityUid uid, GatewayComponent comp, ComponentStartup args)
+    {
+        // add existing destinations
+        var query = EntityQueryEnumerator<GatewayDestinationComponent>();
+        while (query.MoveNext(out var dest, out _))
+        {
+            comp.Destinations.Add(dest);
+        }
+
+        // no need to update ui since its just been created, just do portal
+        UpdateAppearance(uid);
+    }
+
+    private void UpdateUserInterface<T>(EntityUid uid, GatewayComponent comp, T args)
+    {
+        UpdateUserInterface(uid, comp);
+    }
+
+    private void UpdateUserInterface(EntityUid uid, GatewayComponent comp)
+    {
+        var destinations = new List<(EntityUid, String, TimeSpan, bool)>();
+        foreach (var destUid in comp.Destinations)
+        {
+            var dest = Comp<GatewayDestinationComponent>(destUid);
+            if (!dest.Enabled)
+                continue;
+
+            destinations.Add((destUid, dest.Name, dest.NextReady, HasComp<PortalComponent>(destUid)));
+        }
+
+        GetDestination(uid, out var current);
+        var state = new GatewayBoundUserInterfaceState(destinations, current, comp.NextClose, comp.LastOpen);
+        _ui.TrySetUiState(uid, GatewayUiKey.Key, state);
+    }
+
+    private void UpdateAppearance(EntityUid uid)
+    {
+        _appearance.SetData(uid, GatewayVisuals.Active, HasComp<PortalComponent>(uid));
+    }
+
+    private void OnOpenPortal(EntityUid uid, GatewayComponent comp, GatewayOpenPortalMessage args)
+    {
+        // can't link if portal is already open on either side, the destination is invalid or on cooldown
+        if (HasComp<PortalComponent>(uid) ||
+            HasComp<PortalComponent>(args.Destination) ||
+            !TryComp<GatewayDestinationComponent>(args.Destination, out var dest) ||
+            !dest.Enabled ||
+            _timing.CurTime < dest.NextReady)
+            return;
+
+        // TODO: admin log???
+        OpenPortal(uid, comp, args.Destination, dest);
+    }
+
+    private void OpenPortal(EntityUid uid, GatewayComponent comp, EntityUid dest, GatewayDestinationComponent destComp)
+    {
+        _linkedEntity.TryLink(uid, dest);
+        EnsureComp<PortalComponent>(uid);
+        EnsureComp<PortalComponent>(dest);
+
+        // for ui
+        comp.LastOpen = _timing.CurTime;
+        // close automatically after time is up
+        comp.NextClose = comp.LastOpen + destComp.OpenTime;
+
+        _audio.PlayPvs(comp.PortalSound, uid);
+        _audio.PlayPvs(comp.PortalSound, dest);
+
+        UpdateUserInterface(uid, comp);
+        UpdateAppearance(uid);
+        UpdateAppearance(dest);
+    }
+
+    private void ClosePortal(EntityUid uid, GatewayComponent comp)
+    {
+        RemComp<PortalComponent>(uid);
+        if (!GetDestination(uid, out var dest))
+            return;
+
+        if (TryComp<GatewayDestinationComponent>(dest, out var destComp))
+        {
+            // portals closed, put it on cooldown and let it eventually be opened again
+            destComp.NextReady = _timing.CurTime + destComp.Cooldown;
+        }
+
+        _audio.PlayPvs(comp.PortalSound, uid);
+        _audio.PlayPvs(comp.PortalSound, dest.Value);
+
+        _linkedEntity.TryUnlink(uid, dest.Value);
+        RemComp<PortalComponent>(dest.Value);
+        UpdateUserInterface(uid, comp);
+        UpdateAppearance(uid);
+        UpdateAppearance(dest.Value);
+    }
+
+    private bool GetDestination(EntityUid uid, [NotNullWhen(true)] out EntityUid? dest)
+    {
+        dest = null;
+        if (TryComp<LinkedEntityComponent>(uid, out var linked))
+        {
+            var first = linked.LinkedEntities.FirstOrDefault();
+            if (first != EntityUid.Invalid)
+            {
+                dest = first;
+                return true;
+            }
+        }
+
+        return false;
+    }
+
+    private void OnDestinationStartup(EntityUid uid, GatewayDestinationComponent comp, ComponentStartup args)
+    {
+        var query = EntityQueryEnumerator<GatewayComponent>();
+        while (query.MoveNext(out var gatewayUid, out var gateway))
+        {
+            gateway.Destinations.Add(uid);
+            UpdateUserInterface(gatewayUid, gateway);
+        }
+
+        UpdateAppearance(uid);
+    }
+
+    private void OnDestinationShutdown(EntityUid uid, GatewayDestinationComponent comp, ComponentShutdown args)
+    {
+        var query = EntityQueryEnumerator<GatewayComponent>();
+        while (query.MoveNext(out var gatewayUid, out var gateway))
+        {
+            gateway.Destinations.Remove(uid);
+            UpdateUserInterface(gatewayUid, gateway);
+        }
+    }
+}