RobustToolbox/Robust.Server/Console/Commands/PlayerCommands.cs

using System;
using System.Diagnostics.CodeAnalysis;
using System.Linq;
using System.Text;
using Robust.Server.Player;
using Robust.Shared.Console;
using Robust.Shared.Enums;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Network;

namespace Robust.Server.Console.Commands
{
    public sealed class TeleportToCommand : LocalizedCommands
    {
        [Dependency] private readonly IPlayerManager _players = default!;
        [Dependency] private readonly IEntityManager _entities = default!;

        public override string Command => "tpto";

        public override void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            if (args.Length == 0)
                return;

            var target = args[^1];

            if (!TryGetTransformFromUidOrUsername(target, shell, _entities, _players, out _, out var targetTransform))
                return;

            var transformSystem = _entities.System<SharedTransformSystem>();
            var targetCoords = targetTransform.Coordinates;

            if (args.Length == 1)
            {
                var player = shell.Player as IPlayerSession;
                if (player?.Status != SessionStatus.InGame)
                {
                    shell.WriteError("You need to be in game to teleport to an entity.");
                    return;
                }

                if (!_entities.TryGetComponent(player.AttachedEntity, out TransformComponent? playerTransform))
                {
                    shell.WriteError("You don't have an entity.");
                    return;
                }

                transformSystem.SetCoordinates(player.AttachedEntity!.Value, targetCoords);
                playerTransform.AttachToGridOrMap();
            }
            else
            {
                foreach (var victim in args)
                {
                    if (victim == target)
                        continue;

                    if (!TryGetTransformFromUidOrUsername(victim, shell, _entities, _players, out var uid, out var victimTransform))
                        return;

                    transformSystem.SetCoordinates(uid.Value, targetCoords);
                    victimTransform.AttachToGridOrMap();
                }
            }
        }

        private static bool TryGetTransformFromUidOrUsername(
            string str,
            IConsoleShell shell,
            IEntityManager entMan,
            IPlayerManager playerMan,
            [NotNullWhen(true)] out EntityUid? victimUid,
            [NotNullWhen(true)] out TransformComponent? transform)
        {
            if (EntityUid.TryParse(str, out var uid) && entMan.TryGetComponent(uid, out transform))
            {
                victimUid = uid;
                return true;
            }

            if (playerMan.TryGetSessionByUsername(str, out var session)
                && entMan.TryGetComponent(session.AttachedEntity, out transform))
            {
                victimUid = session.AttachedEntity;
                return true;
            }

            if (session == null)
                shell.WriteError("Can't find username/id: " + str);
            else
                shell.WriteError(str + " does not have an entity.");

            transform = null;
            victimUid = default;
            return false;
        }
    }

    public sealed class ListPlayers : LocalizedCommands
    {
        [Dependency] private readonly IPlayerManager _players = default!;

        public override string Command => "listplayers";
        public override void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            // Player: number of people connected and their byond keys
            // Admin: read a byond variable which shows their ip, byond version, ckey, attached entity and hardware id
            // EDIT: inb4 sued by MSO for AGPLv3 violation.

            var sb = new StringBuilder();

            var players = _players.ServerSessions;
            sb.AppendLine($"{"Player Name",20} {"Status",12} {"Playing Time",14} {"Ping",9} {"IP EndPoint",20}");
            sb.AppendLine("-------------------------------------------------------------------------------");

            foreach (var p in players)
            {
                sb.AppendLine(string.Format("{4,20} {1,12} {2,14:hh\\:mm\\:ss} {3,9} {0,20}",
                    p.ConnectedClient.RemoteEndPoint,
                    p.Status.ToString(),
                    DateTime.UtcNow - p.ConnectedTime,
                    p.ConnectedClient.Ping + "ms",
                    p.Name));
            }

            shell.WriteLine(sb.ToString());
        }
    }

    internal sealed class KickCommand : LocalizedCommands
    {
        [Dependency] private readonly IPlayerManager _players = default!;
        [Dependency] private readonly IServerNetManager _netManager = default!;

        public override string Command => "kick";

        public override void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            if (args.Length < 1)
            {
                var player = shell.Player as IPlayerSession;
                var toKickPlayer = player ?? _players.ServerSessions.FirstOrDefault();
                if (toKickPlayer == null)
                {
                    shell.WriteLine("You need to provide a player to kick.");
                    return;
                }
                shell.WriteLine($"You need to provide a player to kick. Try running 'kick {toKickPlayer.Name}' as an example.");
                return;
            }

            var name = args[0];

            if (_players.TryGetSessionByUsername(name, out var target))
            {
                string reason;
                if (args.Length >= 2)
                    reason = $"Kicked by console: {string.Join(' ', args[1..])}";
                else
                    reason = "Kicked by console";

                _netManager.DisconnectChannel(target.ConnectedClient, reason);
            }
        }

        public override CompletionResult GetCompletion(IConsoleShell shell, string[] args)
        {
            if (args.Length == 1)
            {
                var options = _players.ServerSessions.OrderBy(c => c.Name).Select(c => c.Name).ToArray();

                return CompletionResult.FromHintOptions(options, "<PlayerIndex>");
            }

            if (args.Length > 1)
            {
                return CompletionResult.FromHint("[<Reason>]");
            }

            return CompletionResult.Empty;
        }
    }
}