tbd-station-14/Content.Server/Administration/Commands/WarpCommand.cs

using System.Linq;
using Content.Server.Warps;
using Content.Shared.Administration;
using Robust.Server.Player;
using Robust.Shared.Console;
using Robust.Shared.Enums;
using Robust.Shared.Map;
using Robust.Shared.Physics;

namespace Content.Server.Administration.Commands
{
    [AdminCommand(AdminFlags.Admin)]
    public sealed class WarpCommand : IConsoleCommand
    {
        public string Command => "warp";
        public string Description => "Teleports you to predefined areas on the map.";

        public string Help =>
            "warp <location>\nLocations you can teleport to are predefined by the map. " +
            "You can specify '?' as location to get a list of valid locations.";

        public void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            var player = shell.Player as IPlayerSession;
            if (player == null)
            {
                shell.WriteLine("Only players can use this command");
                return;
            }

            if (args.Length != 1)
            {
                shell.WriteLine("Expected a single argument.");
                return;
            }

            var entMan = IoCManager.Resolve<IEntityManager>();
            var location = args[0];
            if (location == "?")
            {
                var locations = string.Join(", ", GetWarpPointNames(entMan));

                shell.WriteLine(locations);
            }
            else
            {
                if (player.Status != SessionStatus.InGame || player.AttachedEntity is not {Valid: true} playerEntity)
                {
                    shell.WriteLine("You are not in-game!");
                    return;
                }

                var mapManager = IoCManager.Resolve<IMapManager>();
                var currentMap = entMan.GetComponent<TransformComponent>(playerEntity).MapID;
                var currentGrid = entMan.GetComponent<TransformComponent>(playerEntity).GridUid;

                var found = entMan.EntityQuery<WarpPointComponent>(true)
                    .Where(p => p.Location == location)
                    .Select(p => entMan.GetComponent<TransformComponent>(p.Owner).Coordinates)
                    .OrderBy(p => p, Comparer<EntityCoordinates>.Create((a, b) =>
                    {
                        // Sort so that warp points on the same grid/map are first.
                        // So if you have two maps loaded with the same warp points,
                        // it will prefer the warp points on the map you're currently on.
                        var aGrid = a.GetGridUid(entMan);
                        var bGrid = b.GetGridUid(entMan);

                        if (aGrid == bGrid)
                        {
                            return 0;
                        }

                        if (aGrid == currentGrid)
                        {
                            return -1;
                        }

                        if (bGrid == currentGrid)
                        {
                            return 1;
                        }

                        var mapA = a.GetMapId(entMan);
                        var mapB = a.GetMapId(entMan);

                        if (mapA == mapB)
                        {
                            return 0;
                        }

                        if (mapA == currentMap)
                        {
                            return -1;
                        }

                        if (mapB == currentMap)
                        {
                            return 1;
                        }

                        return 0;
                    }))
                    .FirstOrDefault();

                var entityUid = found.GetGridUid(entMan);
                if (entityUid != EntityUid.Invalid)
                {
                    entMan.GetComponent<TransformComponent>(playerEntity).Coordinates = found;
                    if (entMan.TryGetComponent(playerEntity, out IPhysBody? physics))
                    {
                        physics.LinearVelocity = Vector2.Zero;
                    }
                }
                else
                {
                    shell.WriteLine("That location does not exist!");
                }
            }
        }

        private static IEnumerable<string> GetWarpPointNames(IEntityManager entMan)
        {
            return entMan.EntityQuery<WarpPointComponent>(true)
                .Select(p => p.Location)
                .Where(p => p != null)
                .OrderBy(p => p)
                .Distinct()!;
        }

        public CompletionResult GetCompletion(IConsoleShell shell, string[] args)
        {
            if (args.Length == 1)
            {
                var ent = IoCManager.Resolve<IEntityManager>();
                var options = new[] { "?" }.Concat(GetWarpPointNames(ent));

                return CompletionResult.FromHintOptions(options, "<warp point | ?>");
            }

            return CompletionResult.Empty;
        }
    }
}