101 lines
3.7 KiB
C#
101 lines
3.7 KiB
C#
using System.Linq;
|
|
using Content.Server._NF.SectorServices;
|
|
using Content.Server.Administration;
|
|
using Content.Server.Station.Systems;
|
|
using Content.Shared.Administration;
|
|
using JetBrains.Annotations;
|
|
using Robust.Shared.Console;
|
|
|
|
namespace Content.Server.AlertLevel.Commands
|
|
{
|
|
[UsedImplicitly]
|
|
[AdminCommand(AdminFlags.Fun)]
|
|
public sealed class SetAlertLevelCommand : LocalizedCommands
|
|
{
|
|
[Dependency] private readonly IEntitySystemManager _entitySystems = default!;
|
|
|
|
public override string Command => "setalertlevel";
|
|
|
|
public override CompletionResult GetCompletion(IConsoleShell shell, string[] args)
|
|
{
|
|
var levelNames = new string[] {};
|
|
var player = shell.Player;
|
|
if (player?.AttachedEntity != null)
|
|
{
|
|
// Frontier: sector-wide alerts
|
|
levelNames = GetSectorLevelNames();
|
|
// var stationUid = _entitySystems.GetEntitySystem<StationSystem>().GetOwningStation(player.AttachedEntity.Value);
|
|
// if (stationUid != null)
|
|
// {
|
|
// levelNames = GetStationLevelNames(stationUid.Value);
|
|
// }
|
|
// End Frontier
|
|
}
|
|
|
|
return args.Length switch
|
|
{
|
|
1 => CompletionResult.FromHintOptions(levelNames,
|
|
LocalizationManager.GetString("cmd-setalertlevel-hint-1")),
|
|
2 => CompletionResult.FromHintOptions(CompletionHelper.Booleans,
|
|
LocalizationManager.GetString("cmd-setalertlevel-hint-2")),
|
|
_ => CompletionResult.Empty,
|
|
};
|
|
}
|
|
|
|
public override void Execute(IConsoleShell shell, string argStr, string[] args)
|
|
{
|
|
if (args.Length < 1)
|
|
{
|
|
shell.WriteError(LocalizationManager.GetString("shell-wrong-arguments-number"));
|
|
return;
|
|
}
|
|
|
|
var locked = false;
|
|
if (args.Length > 1 && !bool.TryParse(args[1], out locked))
|
|
{
|
|
shell.WriteLine(LocalizationManager.GetString("shell-argument-must-be-boolean"));
|
|
return;
|
|
}
|
|
|
|
var player = shell.Player;
|
|
if (player?.AttachedEntity == null)
|
|
{
|
|
shell.WriteLine(LocalizationManager.GetString("shell-only-players-can-run-this-command"));
|
|
return;
|
|
}
|
|
|
|
var stationUid = _entitySystems.GetEntitySystem<StationSystem>().GetOwningStation(player.AttachedEntity.Value);
|
|
if (stationUid == null)
|
|
{
|
|
shell.WriteLine(LocalizationManager.GetString("cmd-setalertlevel-invalid-grid"));
|
|
return;
|
|
}
|
|
|
|
var level = args[0];
|
|
var levelNames = GetSectorLevelNames();
|
|
if (!levelNames.Contains(level))
|
|
{
|
|
shell.WriteLine(LocalizationManager.GetString("cmd-setalertlevel-invalid-level"));
|
|
return;
|
|
}
|
|
|
|
_entitySystems.GetEntitySystem<AlertLevelSystem>().SetLevel(stationUid.Value, level, true, true, true, locked);
|
|
}
|
|
|
|
// Frontier: sector-wide alert level names
|
|
private string[] GetSectorLevelNames()
|
|
{
|
|
var sectorServiceUid = _entitySystems.GetEntitySystem<SectorServiceSystem>().GetServiceEntity();
|
|
var entityManager = IoCManager.Resolve<IEntityManager>();
|
|
if (!entityManager.TryGetComponent<AlertLevelComponent>(sectorServiceUid, out var alertLevelComp))
|
|
return new string[]{};
|
|
|
|
if (alertLevelComp.AlertLevels == null)
|
|
return new string[]{};
|
|
|
|
return alertLevelComp.AlertLevels.Levels.Keys.ToArray();
|
|
}
|
|
// End Frontier
|
|
}
|
|
}
|