123 lines
4.2 KiB
C#
123 lines
4.2 KiB
C#
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<VotePlayerListRequestEvent>(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<MindContainerComponent>(attached, out var mind);
|
|
|
|
var jobName = _jobs.MindTryGetJobName(mind?.Mind);
|
|
var playerInfo = $"{Comp<MetaDataComponent>(attached).EntityName} ({jobName})";
|
|
|
|
return playerInfo;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Used to check whether the player initiating a votekick is allowed to do so serverside.
|
|
/// </summary>
|
|
/// <param name="initiator">The session initiating the votekick.</param>
|
|
public async Task<bool> 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;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Used to check whether the player being targetted for a votekick is a valid target.
|
|
/// </summary>
|
|
/// <param name="target">The session being targetted for a votekick.</param>
|
|
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;
|
|
}
|
|
}
|