109 lines
3.9 KiB
C#
109 lines
3.9 KiB
C#
using Content.Server.Administration;
|
|
using Content.Server.Power.Components;
|
|
using Content.Shared.Administration;
|
|
using Content.Shared.Construction;
|
|
using Content.Shared.Tag;
|
|
using Robust.Server.Player;
|
|
using Robust.Shared.Console;
|
|
using Robust.Shared.Map;
|
|
|
|
namespace Content.Server.Construction.Commands
|
|
{
|
|
[AdminCommand(AdminFlags.Mapping)]
|
|
public sealed class FixRotationsCommand : IConsoleCommand
|
|
{
|
|
[Dependency] private readonly IEntityManager _entManager = default!;
|
|
[Dependency] private readonly IMapManager _mapManager = default!;
|
|
|
|
// ReSharper disable once StringLiteralTypo
|
|
public string Command => "fixrotations";
|
|
public string Description => "Sets the rotation of all occluders, low walls and windows to south.";
|
|
public string Help => $"Usage: {Command} <gridId> | {Command}";
|
|
|
|
public void Execute(IConsoleShell shell, string argsOther, string[] args)
|
|
{
|
|
var player = shell.Player as IPlayerSession;
|
|
EntityUid? gridId;
|
|
var xformQuery = _entManager.GetEntityQuery<TransformComponent>();
|
|
|
|
switch (args.Length)
|
|
{
|
|
case 0:
|
|
if (player?.AttachedEntity is not {Valid: true} playerEntity)
|
|
{
|
|
shell.WriteError("Only a player can run this command.");
|
|
return;
|
|
}
|
|
|
|
gridId = xformQuery.GetComponent(playerEntity).GridUid;
|
|
break;
|
|
case 1:
|
|
if (!NetEntity.TryParse(args[0], out var idNet) || !_entManager.TryGetEntity(idNet, out var id))
|
|
{
|
|
shell.WriteError($"{args[0]} is not a valid entity.");
|
|
return;
|
|
}
|
|
|
|
gridId = id;
|
|
break;
|
|
default:
|
|
shell.WriteLine(Help);
|
|
return;
|
|
}
|
|
|
|
if (!_mapManager.TryGetGrid(gridId, out var grid))
|
|
{
|
|
shell.WriteError($"No grid exists with id {gridId}");
|
|
return;
|
|
}
|
|
|
|
if (!_entManager.EntityExists(grid.Owner))
|
|
{
|
|
shell.WriteError($"Grid {gridId} doesn't have an associated grid entity.");
|
|
return;
|
|
}
|
|
|
|
var changed = 0;
|
|
var tagSystem = _entManager.EntitySysManager.GetEntitySystem<TagSystem>();
|
|
|
|
foreach (var child in xformQuery.GetComponent(grid.Owner).ChildEntities)
|
|
{
|
|
if (!_entManager.EntityExists(child))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
var valid = false;
|
|
|
|
// Occluders should only count if the state of it right now is enabled.
|
|
// This prevents issues with edge firelocks.
|
|
if (_entManager.TryGetComponent<OccluderComponent>(child, out var occluder))
|
|
{
|
|
valid |= occluder.Enabled;
|
|
}
|
|
// low walls & grilles
|
|
valid |= _entManager.HasComponent<SharedCanBuildWindowOnTopComponent>(child);
|
|
// cables
|
|
valid |= _entManager.HasComponent<CableComponent>(child);
|
|
// anything else that might need this forced
|
|
valid |= tagSystem.HasTag(child, "ForceFixRotations");
|
|
// override
|
|
valid &= !tagSystem.HasTag(child, "ForceNoFixRotations");
|
|
|
|
if (!valid)
|
|
continue;
|
|
|
|
var childXform = xformQuery.GetComponent(child);
|
|
|
|
if (childXform.LocalRotation != Angle.Zero)
|
|
{
|
|
childXform.LocalRotation = Angle.Zero;
|
|
changed++;
|
|
}
|
|
}
|
|
|
|
shell.WriteLine($"Changed {changed} entities. If things seem wrong, reconnect.");
|
|
}
|
|
}
|
|
}
|