CatStation/Content.Server/GameTicking/Commands/JoinGameCommand.cs

using Content.Server.Administration.Managers;
using Content.Server.Station.Systems;
using Content.Shared.Administration;
using Content.Shared.CCVar;
using Content.Shared.GameTicking;
using Content.Shared.Roles;
using Robust.Shared.Configuration;
using Robust.Shared.Console;
using Robust.Shared.Prototypes;

namespace Content.Server.GameTicking.Commands
{
    [AnyCommand]
    sealed partial class JoinGameCommand : IConsoleCommand
    {
        [Dependency] private IEntityManager _entManager = default!;
        [Dependency] private IPrototypeManager _prototypeManager = default!;
        [Dependency] private IAdminManager _adminManager = default!;
        [Dependency] private IConfigurationManager _cfg = default!;
        [Dependency] private ILogManager _logManager = default!;

        private readonly ISawmill _sawmill;

        public string Command => "joingame";
        public string Description => "";
        public string Help => "";

        public JoinGameCommand()
        {
            IoCManager.InjectDependencies(this);

            _sawmill = _logManager.GetSawmill("security");
        }

        public void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            if (args.Length != 2)
            {
                shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
                return;
            }

            var player = shell.Player;

            if (player == null)
            {
                return;
            }

            var ticker = _entManager.System<ServerGameTicker>();
            var stationJobs = _entManager.System<ServerStationJobsSystem>();

            if (ticker.PlayerGameStatuses.TryGetValue(player.UserId, out var status) && status == PlayerGameStatus.JoinedGame)
            {
                _sawmill.Info($"{player.Name} ({player.UserId}) attempted to latejoin while in-game.");
                shell.WriteError($"{player.Name} is not in the lobby. This incident will be reported.");
                return;
            }

            if (ticker.RunLevel == GameRunLevel.PreRoundLobby)
            {
                shell.WriteLine("Round has not started.");
                return;
            }

            var id = args[0];
            if (!int.TryParse(args[1], out var sid))
            {
                shell.WriteError(Loc.GetString("shell-argument-must-be-number"));
            }

            var station = _entManager.GetEntity(new NetEntity(sid));
            var jobPrototype = _prototypeManager.Index<JobPrototype>(id);
            if (stationJobs.TryGetJobSlot(station, jobPrototype, out var slots) == false || slots == 0)
            {
                shell.WriteLine($"{jobPrototype.LocalizedName} has no available slots.");
                return;
            }

            if (_adminManager.IsAdmin(player) && _cfg.GetCVar(CCVars.AdminDeadminOnJoin))
            {
                _adminManager.DeAdmin(player);
            }

            ticker.MakeJoinGame(player, station, id);
        }
    }
}