using Content.Server.Administration.Managers; using Content.Server.Database; using Content.Server.Ghost; using Content.Server.Roles.Jobs; using Content.Shared.CCVar; using Content.Shared.Ghost; using Content.Shared.Mind.Components; using Content.Shared.Voting; using Robust.Server.Player; using Robust.Shared.Configuration; using Robust.Shared.Network; using Robust.Shared.Player; using Robust.Shared.Timing; using System.Threading.Tasks; namespace Content.Server.Voting; public sealed class VotingSystem : EntitySystem { [Dependency] private readonly IPlayerManager _playerManager = default!; [Dependency] private readonly IAdminManager _adminManager = default!; [Dependency] private readonly IServerDbManager _dbManager = default!; [Dependency] private readonly IGameTiming _gameTiming = default!; [Dependency] private readonly IConfigurationManager _cfg = default!; [Dependency] private readonly JobSystem _jobs = default!; public override void Initialize() { base.Initialize(); SubscribeNetworkEvent(OnVotePlayerListRequestEvent); } private async void OnVotePlayerListRequestEvent(VotePlayerListRequestEvent msg, EntitySessionEventArgs args) { if (args.SenderSession.AttachedEntity is not { Valid: true } entity || !await CheckVotekickInitEligibility(args.SenderSession)) { var deniedResponse = new VotePlayerListResponseEvent(new (NetUserId, NetEntity, string)[0], true); RaiseNetworkEvent(deniedResponse, args.SenderSession.Channel); return; } List<(NetUserId, NetEntity, string)> players = new(); foreach (var player in _playerManager.Sessions) { if (player.AttachedEntity is not { Valid: true } attached) continue; if (attached == entity) continue; if (_adminManager.IsAdmin(player, false)) continue; var playerName = GetPlayerVoteListName(attached); var netEntity = GetNetEntity(attached); players.Add((player.UserId, netEntity, playerName)); } var response = new VotePlayerListResponseEvent(players.ToArray(), false); RaiseNetworkEvent(response, args.SenderSession.Channel); } public string GetPlayerVoteListName(EntityUid attached) { TryComp(attached, out var mind); var jobName = _jobs.MindTryGetJobName(mind?.Mind); var playerInfo = $"{Comp(attached).EntityName} ({jobName})"; return playerInfo; } /// /// Used to check whether the player initiating a votekick is allowed to do so serverside. /// /// The session initiating the votekick. public async Task CheckVotekickInitEligibility(ICommonSession? initiator) { if (initiator == null) return false; // Being an admin overrides the votekick eligibility if (initiator.AttachedEntity != null && _adminManager.IsAdmin(initiator.AttachedEntity.Value, false)) return true; if (_cfg.GetCVar(CCVars.VotekickInitiatorGhostRequirement)) { // Must be ghost if (!TryComp(initiator.AttachedEntity, out GhostComponent? ghostComp)) return false; // Must have been dead for x seconds if ((int)_gameTiming.RealTime.Subtract(ghostComp.TimeOfDeath).TotalSeconds < _cfg.GetCVar(CCVars.VotekickEligibleVoterDeathtime)) return false; } // Must be whitelisted if (!await _dbManager.GetWhitelistStatusAsync(initiator.UserId)) return false; return true; } /// /// Used to check whether the player being targetted for a votekick is a valid target. /// /// The session being targetted for a votekick. public bool CheckVotekickTargetEligibility(ICommonSession? target) { if (target == null) return false; // Admins can't be votekicked if (target.AttachedEntity != null && _adminManager.IsAdmin(target.AttachedEntity.Value)) return false; return true; } }