Files
tbd-station-14/Content.Server/Construction/Commands/TileReplaceCommand.cs
2022-10-30 17:48:53 +11:00

86 lines
2.6 KiB
C#

using Content.Server.Administration;
using Content.Shared.Administration;
using Robust.Server.Player;
using Robust.Shared.Console;
using Robust.Shared.Map;
namespace Content.Server.Construction.Commands;
[AdminCommand(AdminFlags.Mapping)]
sealed class TileReplaceCommand : IConsoleCommand
{
// ReSharper disable once StringLiteralTypo
public string Command => "tilereplace";
public string Description => "Replaces one tile with another.";
public string Help => $"Usage: {Command} [<gridId>] <src> <dst>";
public void Execute(IConsoleShell shell, string argStr, string[] args)
{
var player = shell.Player as IPlayerSession;
var entityManager = IoCManager.Resolve<IEntityManager>();
EntityUid? gridId;
string tileIdA;
string tileIdB;
switch (args.Length)
{
case 2:
if (player?.AttachedEntity is not {Valid: true} playerEntity)
{
shell.WriteLine("Only a player can run this command without a grid ID.");
return;
}
gridId = entityManager.GetComponent<TransformComponent>(playerEntity).GridUid;
tileIdA = args[0];
tileIdB = args[1];
break;
case 3:
if (!EntityUid.TryParse(args[0], out var id))
{
shell.WriteLine($"{args[0]} is not a valid entity.");
return;
}
gridId = id;
tileIdA = args[1];
tileIdB = args[2];
break;
default:
shell.WriteLine(Help);
return;
}
var tileDefinitionManager = IoCManager.Resolve<ITileDefinitionManager>();
var tileA = tileDefinitionManager[tileIdA];
var tileB = tileDefinitionManager[tileIdB];
var mapManager = IoCManager.Resolve<IMapManager>();
if (!mapManager.TryGetGrid(gridId, out var grid))
{
shell.WriteLine($"No grid exists with id {gridId}");
return;
}
if (!entityManager.EntityExists(grid.GridEntityId))
{
shell.WriteLine($"Grid {gridId} doesn't have an associated grid entity.");
return;
}
var changed = 0;
foreach (var tile in grid.GetAllTiles())
{
var tileContent = tile.Tile;
if (tileContent.TypeId == tileA.TileId)
{
grid.SetTile(tile.GridIndices, new Tile(tileB.TileId));
changed++;
}
}
shell.WriteLine($"Changed {changed} tiles.");
}
}