82 lines
2.4 KiB
C#
82 lines
2.4 KiB
C#
using System.Linq;
|
|
using Content.Server.Administration;
|
|
using Content.Shared.Administration;
|
|
using Robust.Shared.Console;
|
|
using Robust.Shared.ContentPack;
|
|
using Robust.Shared.EntitySerialization;
|
|
using Robust.Shared.EntitySerialization.Components;
|
|
using Robust.Shared.EntitySerialization.Systems;
|
|
using Robust.Shared.Utility;
|
|
|
|
namespace Content.Server.Maps;
|
|
|
|
/// <summary>
|
|
/// Loads every map and resaves it into the data folder.
|
|
/// </summary>
|
|
[AdminCommand(AdminFlags.Host)]
|
|
public sealed class ResaveCommand : LocalizedCommands
|
|
{
|
|
[Dependency] private readonly IEntityManager _entManager = default!;
|
|
[Dependency] private readonly IResourceManager _res = default!;
|
|
[Dependency] private readonly ILogManager _log = default!;
|
|
|
|
public override string Command => "resave";
|
|
|
|
public override void Execute(IConsoleShell shell, string argStr, string[] args)
|
|
{
|
|
var loader = _entManager.System<MapLoaderSystem>();
|
|
|
|
var opts = MapLoadOptions.Default with
|
|
{
|
|
|
|
DeserializationOptions = DeserializationOptions.Default with
|
|
{
|
|
StoreYamlUids = true,
|
|
LogOrphanedGrids = false
|
|
}
|
|
};
|
|
|
|
var log = _log.GetSawmill(Command);
|
|
var files = _res.ContentFindFiles(new ResPath("/Maps/")).ToList();
|
|
|
|
for (var i = 0; i < files.Count; i++)
|
|
{
|
|
var fn = files[i];
|
|
log.Info($"Re-saving file {i}/{files.Count} : {fn}");
|
|
|
|
if (!loader.TryLoadGeneric(fn, out var result, opts))
|
|
continue;
|
|
|
|
if (result.Maps.Count != 1)
|
|
{
|
|
shell.WriteError(
|
|
$"Multi-map or multi-grid files like {fn} are not yet supported by the {Command} command");
|
|
loader.Delete(result);
|
|
continue;
|
|
}
|
|
|
|
var map = result.Maps.First();
|
|
|
|
// Process deferred component removals.
|
|
_entManager.CullRemovedComponents();
|
|
|
|
if (_entManager.HasComponent<LoadedMapComponent>(map))
|
|
{
|
|
loader.TrySaveMap(map.Comp.MapId, fn);
|
|
}
|
|
else if (result.Grids.Count == 1)
|
|
{
|
|
loader.TrySaveGrid(result.Grids.First(), fn);
|
|
}
|
|
else
|
|
{
|
|
shell.WriteError($"Failed to resave {fn}");
|
|
}
|
|
|
|
loader.Delete(result);
|
|
}
|
|
|
|
shell.WriteLine($"Resaved all maps");
|
|
}
|
|
}
|