diff --git a/Content.Server/_CorvaxNext/AutoVote/AutoVoteSystem.cs b/Content.Server/_CorvaxNext/AutoVote/AutoVoteSystem.cs new file mode 100644 index 00000000000..12b5a1712fd --- /dev/null +++ b/Content.Server/_CorvaxNext/AutoVote/AutoVoteSystem.cs @@ -0,0 +1,54 @@ +using Robust.Shared.Configuration; +using Content.Server.Voting.Managers; +using Content.Shared.GameTicking; +using Content.Shared.Voting; +using Content.Shared._CorvaxNext.NextVars; +using Robust.Server.Player; +using Content.Server.GameTicking; + +namespace Content.Server._CorvaxNext.AutoVote; + +public sealed class AutoVoteSystem : EntitySystem +{ + [Dependency] private readonly IConfigurationManager _cfg = default!; + [Dependency] public readonly IVoteManager _voteManager = default!; + [Dependency] public readonly IPlayerManager _playerManager = default!; + + public bool _shouldVoteNextJoin = false; + + public override void Initialize() + { + base.Initialize(); + + SubscribeLocalEvent(OnReturnedToLobby); + SubscribeLocalEvent(OnPlayerJoinedLobby); + } + + public void OnReturnedToLobby(RoundRestartCleanupEvent ev) => CallAutovote(); + + public void OnPlayerJoinedLobby(PlayerJoinedLobbyEvent ev) + { + if (!_shouldVoteNextJoin) + return; + + CallAutovote(); + _shouldVoteNextJoin = false; + } + + private void CallAutovote() + { + if (!_cfg.GetCVar(NextVars.AutoVoteEnabled)) + return; + + if (_playerManager.PlayerCount == 0) + { + _shouldVoteNextJoin = true; + return; + } + + if (_cfg.GetCVar(NextVars.MapAutoVoteEnabled)) + _voteManager.CreateStandardVote(null, StandardVoteType.Map); + if (_cfg.GetCVar(NextVars.PresetAutoVoteEnabled)) + _voteManager.CreateStandardVote(null, StandardVoteType.Preset); + } +} diff --git a/Content.Shared/_CorvaxNext/NextVars.cs b/Content.Shared/_CorvaxNext/NextVars.cs index 9f355adbfec..216dd19b518 100644 --- a/Content.Shared/_CorvaxNext/NextVars.cs +++ b/Content.Shared/_CorvaxNext/NextVars.cs @@ -3,7 +3,7 @@ namespace Content.Shared._CorvaxNext.NextVars; /// -/// Corvax modules console variables +/// Corvax modules console variables /// [CVarDefs] // ReSharper disable once InconsistentNaming @@ -15,8 +15,35 @@ public sealed class NextVars public static readonly CVarDef OfferModeIndicatorsPointShow = CVarDef.Create("hud.offer_mode_indicators_point_show", true, CVar.ARCHIVE | CVar.CLIENTONLY); + /* + * AUTOVOTE SYSTEM + */ + #region Autovote + + /// + /// Enables the automatic voting system. + /// + public static readonly CVarDef AutoVoteEnabled = + CVarDef.Create("vote.autovote_enabled", false, CVar.SERVERONLY); + + /// + /// Automatically starts a map vote when returning to the lobby. + /// Requires auto voting to be enabled. + /// + public static readonly CVarDef MapAutoVoteEnabled = + CVarDef.Create("vote.map_autovote_enabled", true, CVar.SERVERONLY); + + /// + /// Automatically starts a gamemode vote when returning to the lobby. + /// Requires auto voting to be enabled. + /// + public static readonly CVarDef PresetAutoVoteEnabled = + CVarDef.Create("vote.preset_autovote_enabled", true, CVar.SERVERONLY); + + #endregion + /// - /// CorvaxNext Surgery cvars + /// _CorvaxNext Surgery cvars /// #region Surgery @@ -38,7 +65,4 @@ public sealed class NextVars /// public static readonly CVarDef CrawlUnderTables = CVarDef.Create("laying.crawlundertables", false, CVar.REPLICATED); - - // public static readonly CVarDef OfferModeIndicatorsPointShow = - // CVarDef.Create("hud.offer_mode_indicators_point_show", true, CVar.ARCHIVE | CVar.CLIENTONLY); }