Delta-v/Content.Server/Nyanotrasen/Construction/Commands/TileWindowsCommand.cs

116 lines
4.0 KiB
C#

using Content.Server.Administration;
using Content.Shared.Administration;
using Content.Shared.Maps;
using Content.Shared.Tag;
using Robust.Server.Player;
using Robust.Shared.Console;
using Robust.Shared.Map;
using Robust.Shared.Map.Components;
using Robust.Shared.Maths;
using Robust.Shared.Player;
using Robust.Shared.Prototypes;
namespace Content.Server.Construction.Commands
{
[AdminCommand(AdminFlags.Mapping)]
sealed class TileWindowsCommand : IConsoleCommand
{
// ReSharper disable once StringLiteralTypo
public string Command => "tilewindows";
public string Description => "Puts a reinforced plating tile below every window on a grid.";
public string Help => $"Usage: {Command} <gridId> | {Command}";
public static readonly ProtoId<ContentTileDefinition> TilePrototypeId = "FloorReinforced";
public static readonly ProtoId<TagPrototype> WindowTag = "Window";
public static readonly ProtoId<TagPrototype> DirectionalTag = "Directional";
public void Execute(IConsoleShell shell, string argStr, string[] args)
{
var player = shell.Player as ICommonSession;
var entityManager = IoCManager.Resolve<IEntityManager>();
EntityUid? gridId;
switch (args.Length)
{
case 0:
if (player?.AttachedEntity is not { Valid: true } playerEntity)
{
shell.WriteLine("Only a player can run this command.");
return;
}
gridId = entityManager.GetComponent<TransformComponent>(playerEntity).GridUid;
break;
case 1:
if (!EntityUid.TryParse(args[0], out var id))
{
shell.WriteLine($"{args[0]} is not a valid entity.");
return;
}
gridId = id;
break;
default:
shell.WriteLine(Help);
return;
}
if (!entityManager.TryGetComponent<MapGridComponent>(gridId, out var grid))
{
shell.WriteLine($"No grid exists with id {gridId}");
return;
}
if (!entityManager.EntityExists(grid.Owner))
{
shell.WriteLine($"Grid {gridId} doesn't have an associated grid entity.");
return;
}
var tileDefinitionManager = IoCManager.Resolve<ITileDefinitionManager>();
var tagSystem = entityManager.EntitySysManager.GetEntitySystem<TagSystem>();
var underplating = tileDefinitionManager[TilePrototypeId];
var underplatingTile = new Tile(underplating.TileId);
var changed = 0;
var children = entityManager.GetComponent<TransformComponent>(grid.Owner).ChildEnumerator;
while (children.MoveNext(out var child))
{
if (!entityManager.EntityExists(child))
{
continue;
}
if (tagSystem.HasTag(child, DirectionalTag))
{
continue;
}
if (!tagSystem.HasTag(child, WindowTag))
{
continue;
}
var childTransform = entityManager.GetComponent<TransformComponent>(child);
if (!childTransform.Anchored)
{
continue;
}
var tile = grid.GetTileRef(childTransform.Coordinates);
var tileDef = (ContentTileDefinition) tileDefinitionManager[tile.Tile.TypeId];
if (tileDef.ID == TilePrototypeId)
{
continue;
}
grid.SetTile(childTransform.Coordinates, underplatingTile);
changed++;
}
shell.WriteLine($"Changed {changed} tiles.");
}
}
}