tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor set alert level command (#9794)

Browse Source
This commit is contained in:
Morber
2022-07-17 01:34:43 +03:00
committed by GitHub
parent 058d8371d9
commit 700f4aff57
3 changed files with 60 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
Content.Server/AlertLevel/Commands/SetAlertLevelCommand.cs
View File

@@ -1,6 +1,5 @@
using Content.Server.Administration;
using Content.Server.AlertLevel;
using Content.Server.Players;
using System.Linq;
using Content.Server.Administration;
using Content.Server.Station.Systems;
using Content.Shared.Administration;
using JetBrains.Annotations;
@@ -14,47 +13,82 @@ namespace Content.Server.AlertLevel.Commands
public sealed class SetAlertLevelCommand : IConsoleCommand
{
public string Command => "setalertlevel";
public string Description => "Set current station alert level.";
public string Help => "setalertlevel <level> [locked]";
public string Description => Loc.GetString("cmd-setalertlevel-desc");
public string Help => Loc.GetString("cmd-setalertlevel-help");
public CompletionResult GetCompletion(IConsoleShell shell, string[] args)
{
var levelNames = new string[] {};
var player = shell.Player as IPlayerSession;
if (player?.AttachedEntity != null)
{
var stationUid = EntitySystem.Get<StationSystem>().GetOwningStation(player.AttachedEntity.Value);
if (stationUid != null)
{
levelNames = GetStationLevelNames(stationUid.Value);
}
}
return args.Length switch
{
1 => CompletionResult.FromHintOptions(levelNames,
Loc.GetString("cmd-setalertlevel-hint-1")),
2 => CompletionResult.FromHintOptions(CompletionHelper.Booleans,
Loc.GetString("cmd-setalertlevel-hint-2")),
_ => CompletionResult.Empty,
};
}
public void Execute(IConsoleShell shell, string argStr, string[] args)
{
if (args.Length < 1)
{
shell.WriteError("Incorrect number of arguments. " + Help);
shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
return;
}
var locked = false;
if (args.Length > 1 && !bool.TryParse(args[1], out locked))
{
shell.WriteLine("Invalid boolean flag");
shell.WriteLine(Loc.GetString("shell-argument-must-be-boolean"));
return;
}
var player = shell.Player as IPlayerSession;
if (player == null)
if (player?.AttachedEntity == null)
{
shell.WriteLine("You cannot run this from the server or without an attached entity.");
shell.WriteLine(Loc.GetString("shell-only-players-can-run-this-command"));
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);
var stationUid = EntitySystem.Get<StationSystem>().GetOwningStation(player.AttachedEntity.Value);
if (stationUid == null)
{
shell.WriteLine("You must be on grid of station code that you are going to change.");
shell.WriteLine(Loc.GetString("cmd-setalertlevel-invalid-grid"));
return;
}
var level = args[0];
var levelNames = GetStationLevelNames(stationUid.Value);
if (!levelNames.Contains(level))
{
shell.WriteLine(Loc.GetString("cmd-setalertlevel-invalid-level"));
return;
}
EntitySystem.Get<AlertLevelSystem>().SetLevel(stationUid.Value, level, true, true, true, locked);
}
private string[] GetStationLevelNames(EntityUid station)
{
var entityManager = IoCManager.Resolve<IEntityManager>();
if (!entityManager.TryGetComponent<AlertLevelComponent>(station, out var alertLevelComp))
return new string[]{};
if (alertLevelComp.AlertLevels == null)
return new string[]{};
return alertLevelComp.AlertLevels.Levels.Keys.ToArray();
}
}
}