[Port]Game presets can now have cooldowns to prevent back-to-back modes

Content.Server/GameTicking/Presets/GamePresetPrototype.cs

@@ -33,6 +33,15 @@ public sealed partial class GamePresetPrototype : IPrototype
         [DataField("maxPlayers")]
         public int? MaxPlayers;
 
+        // Begin Imp
+        /// <summary>
+        /// Ensures that this gamemode does not get selected for a number of rounds
+        /// by something like Secret. This is not considered when the preset is forced.
+        /// </summary>
+        [DataField]
+        public int Cooldown = 0;
+        // End Imp
+
         [DataField("rules", customTypeSerializer: typeof(PrototypeIdListSerializer<EntityPrototype>))]
         public IReadOnlyList<string> Rules { get; private set; } = Array.Empty<string>();
 

Content.Server/GameTicking/Rules/SecretRuleSystem.cs

@@ -22,6 +22,11 @@ public sealed class SecretRuleSystem : GameRuleSystem<SecretRuleComponent>
     [Dependency] private readonly IConfigurationManager _configurationManager = default!;
     [Dependency] private readonly IAdminLogManager _adminLogger = default!;
     [Dependency] private readonly IComponentFactory _compFact = default!;
+    [Dependency] private readonly GameTicker _ticker = default!; // begin Imp
+
+    // Dictionary that contains the minimum round number for certain preset
+    // prototypes to be allowed to roll again
+    private static Dictionary<ProtoId<GamePresetPrototype>, int> _nextRoundAllowed = new(); // end Imp
 
     private string _ruleCompName = default!;
 
@@ -46,6 +51,12 @@ protected override void Added(EntityUid uid, SecretRuleComponent component, Game
         Log.Info($"Selected {preset.ID} as the secret preset.");
         _adminLogger.Add(LogType.EventStarted, $"Selected {preset.ID} as the secret preset.");
 
+        if (preset.Cooldown > 0) // Begin Imp
+        {
+            _nextRoundAllowed[preset.ID] = _ticker.RoundId + preset.Cooldown + 1;
+            Log.Info($"{preset.ID} is now on cooldown until {_nextRoundAllowed[preset.ID]}");
+        } // End Imp
+
         foreach (var rule in preset.Rules)
         {
             EntityUid ruleEnt;
@@ -166,6 +177,11 @@ private bool CanPick([NotNullWhen(true)] GamePresetPrototype? selected, int play
             if (ruleComp.MinPlayers > players && ruleComp.CancelPresetOnTooFewPlayers)
                 return false;
         }
+        if (_nextRoundAllowed.ContainsKey(selected.ID) && _nextRoundAllowed[selected.ID] > _ticker.RoundId) // Begin Imp
+        {
+            Log.Info($"Skipping preset {selected.ID} (Not available until round {_nextRoundAllowed[selected.ID]}");
+            return false;
+        } // End Imp
 
         return true;
     }

Resources/Prototypes/game_presets.yml

@@ -5,6 +5,7 @@
   name: survival-title
   showInVote: true # secret # DeltaV - Me when the survival. Used for periapsis.
   description: survival-description
+  cooldown: 2 # Imp - Can't occur thrice
   rules:
     - MeteorSwarmScheduler
     - RampingStationEventScheduler
@@ -185,6 +186,7 @@
   name: nukeops-title
   description: nukeops-description
   showInVote: false
+  cooldown: 1 # Imp - Can't occur back to back
   rules:
     - Nukeops
     - SubGamemodesRule
@@ -222,6 +224,7 @@
   - zomber
   name: zombie-title
   description: zombie-description
+  cooldown: 2 # Imp - Can't occur thrice
   showInVote: false
   rules:
   - Zombie