Set alert level command (#8507)

* Add force mode for setting alert level

* Add console command for changing alert level

* Filename typo

* Return delta disableSelection property

* Make locked a component property

* Not lock green level after nuke disarm

Content.Server/AlertLevel/AlertLevelComponent.cs

@@ -22,6 +22,11 @@ public sealed class AlertLevelComponent : Component
     /// </summary>
     [ViewVariables(VVAccess.ReadWrite)] public string CurrentLevel = string.Empty;
 
+    /// <summary>
+    /// Is current station level can be changed by crew.
+    /// </summary>
+    [ViewVariables(VVAccess.ReadWrite)] public bool IsLevelLocked = false;
+
     [ViewVariables] public const float Delay = 300;
     [ViewVariables] public float CurrentDelay = 0;
     [ViewVariables] public bool ActiveDelay;
@@ -40,7 +45,7 @@ public sealed class AlertLevelComponent : Component
                 return false;
             }
 
-            return level.Selectable && !level.DisableSelection;
+            return level.Selectable && !level.DisableSelection && !IsLevelLocked;
         }
     }
 }

Content.Server/AlertLevel/AlertLevelPrototype.cs

@@ -20,7 +20,7 @@ public sealed class AlertLevelPrototype : IPrototype
     /// Default level that the station is on upon initialization.
     /// If this isn't in the dictionary, this will default to whatever .First() gives.
     /// </summary>
-    [DataField("defaultLevel")] public string DefaultLevel { get; }= default!;
+    [DataField("defaultLevel")] public string DefaultLevel { get; } = default!;
 }
 
 /// <summary>

Content.Server/AlertLevel/AlertLevelSystem.cs

@@ -85,8 +85,9 @@ public sealed class AlertLevelSystem : EntitySystem
     /// <param name="playSound">Play the alert level's sound.</param>
     /// <param name="announce">Say the alert level's announcement.</param>
     /// <param name="force">Force the alert change. This applies if the alert level is not selectable or not.</param>
+    /// <param name="locked">Will it be possible to change level by crew.</param>
     public void SetLevel(EntityUid station, string level, bool playSound, bool announce, bool force = false,
-        MetaDataComponent? dataComponent = null, AlertLevelComponent? component = null)
+        bool locked = false, MetaDataComponent? dataComponent = null, AlertLevelComponent? component = null)
     {
         if (!Resolve(station, ref component, ref dataComponent)
             || component.AlertLevels == null
@@ -99,7 +100,8 @@ public sealed class AlertLevelSystem : EntitySystem
         if (!force)
         {
             if (!detail.Selectable
-                || component.CurrentDelay > 0)
+                || component.CurrentDelay > 0
+                || component.IsLevelLocked)
             {
                 return;
             }
@@ -109,6 +111,7 @@ public sealed class AlertLevelSystem : EntitySystem
         }
 
         component.CurrentLevel = level;
+        component.IsLevelLocked = locked;
 
         var stationName = dataComponent.EntityName;
 

Content.Server/AlertLevel/Commands/SetAlertLevelCommand.cs

@@ -0,0 +1,60 @@
+using Content.Server.Administration;
+using Content.Server.AlertLevel;
+using Content.Server.Players;
+using Content.Server.Station.Systems;
+using Content.Shared.Administration;
+using JetBrains.Annotations;
+using Robust.Server.Player;
+using Robust.Shared.Console;
+
+namespace Content.Server.AlertLevel.Commands
+{
+    [UsedImplicitly]
+    [AdminCommand(AdminFlags.Fun)]
+    public sealed class SetAlertLevelCommand : IConsoleCommand
+    {
+        public string Command => "setalertlevel";
+        public string Description => "Set current station alert level.";
+        public string Help => "setalertlevel <level> [locked]";
+
+        public void Execute(IConsoleShell shell, string argStr, string[] args)
+        {
+            if (args.Length < 1)
+            {
+                shell.WriteError("Incorrect number of arguments. " + Help);
+                return;
+            }
+
+            var locked = false;
+            if (args.Length > 1 && !bool.TryParse(args[1], out locked))
+            {
+                shell.WriteLine("Invalid boolean flag");
+                return;
+            }
+
+            var player = shell.Player as IPlayerSession;
+            if (player == null)
+            {
+                shell.WriteLine("You cannot run this from the server or without an attached entity.");
+                return;
+            }
+
+            var playerEntityUid = player.AttachedEntity;
+            if (playerEntityUid == null)
+            {
+                shell.WriteLine("You cannot run this from the server or without an attached entity.");
+                return;
+            }
+
+            var stationUid = EntitySystem.Get<StationSystem>().GetOwningStation(playerEntityUid.Value);
+            if (stationUid == null)
+            {
+                shell.WriteLine("You must be on grid of station code that you are going to change.");
+                return;
+            }
+
+            var level = args[0];
+            EntitySystem.Get<AlertLevelSystem>().SetLevel(stationUid.Value, level, true, true, true, locked);
+        }
+    }
+}

Content.Server/Nuke/NukeSystem.cs

@@ -333,7 +333,7 @@ namespace Content.Server.Nuke
             // Otherwise, you could set every station to whatever AlertLevelOnActivate is.
             if (stationUid != null)
             {
-                _alertLevel.SetLevel(stationUid.Value, component.AlertLevelOnActivate, true, true, true);
+                _alertLevel.SetLevel(stationUid.Value, component.AlertLevelOnActivate, true, true, true, true);
             }
 
             // warn a crew