mirror of
https://github.com/corvax-team/ss14-wl.git
synced 2026-09-15 14:52:26 +02:00
239 lines
8.2 KiB
C#
239 lines
8.2 KiB
C#
using System.Diagnostics.CodeAnalysis;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using Robust.Shared.ContentPack;
|
|
using Robust.Shared.EntitySerialization.Systems;
|
|
using Robust.Shared.Map.Events;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Serialization.Markdown;
|
|
using Robust.Shared.Serialization.Markdown.Mapping;
|
|
using Robust.Shared.Serialization.Markdown.Value;
|
|
using Robust.Shared.Utility;
|
|
|
|
namespace Content.Server.Maps;
|
|
|
|
/// <summary>
|
|
/// Performs basic map migration operations by listening for engine <see cref="MapLoaderSystem"/> events.
|
|
/// </summary>
|
|
public sealed partial class MapMigrationSystem : EntitySystem
|
|
{
|
|
#if DEBUG
|
|
[Dependency] private IPrototypeManager _protoMan = default!;
|
|
#endif
|
|
[Dependency] private IResourceManager _resMan = default!;
|
|
|
|
private const string MigrationFile = "/migration.yml";
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
SubscribeLocalEvent<BeforeEntityReadEvent>(OnBeforeReadEvent);
|
|
|
|
#if DEBUG
|
|
if (!TryReadFile(out var mappings))
|
|
return;
|
|
|
|
// Verify that all of the entries map to valid entity prototypes.
|
|
foreach (var node in mappings.Children.Values)
|
|
{
|
|
var newId = ((ValueDataNode) node).Value;
|
|
if (!string.IsNullOrEmpty(newId) && newId != "null")
|
|
DebugTools.Assert(_protoMan.HasIndex<EntityPrototype>(newId), $"{newId} is not an entity prototype.");
|
|
}
|
|
#endif
|
|
}
|
|
|
|
private bool TryReadFile([NotNullWhen(true)] out MappingDataNode? mappings)
|
|
{
|
|
mappings = null;
|
|
var path = new ResPath(MigrationFile);
|
|
if (!_resMan.TryContentFileRead(path, out var stream))
|
|
return false;
|
|
|
|
using var reader = new StreamReader(stream, EncodingHelpers.UTF8);
|
|
var documents = DataNodeParser.ParseYamlStream(reader).FirstOrDefault();
|
|
|
|
if (documents == null)
|
|
return false;
|
|
|
|
mappings = (MappingDataNode) documents.Root;
|
|
return true;
|
|
}
|
|
|
|
private void OnBeforeReadEvent(BeforeEntityReadEvent ev)
|
|
{
|
|
if (!TryReadFile(out var mappings))
|
|
return;
|
|
|
|
foreach (var (key, value) in mappings)
|
|
{
|
|
if (value is not ValueDataNode valueNode)
|
|
continue;
|
|
|
|
if (string.IsNullOrWhiteSpace(valueNode.Value) || valueNode.Value == "null")
|
|
ev.DeletedPrototypes.Add(key);
|
|
else
|
|
ev.RenamedPrototypes.Add(key, valueNode.Value);
|
|
}
|
|
}
|
|
}
|
|
|
|
//This is too fuck to fix it
|
|
// Uncommit, when it's need
|
|
//
|
|
// using System.Diagnostics.CodeAnalysis;
|
|
// using System.IO;
|
|
// using System.Linq;
|
|
// using Robust.Server.GameObjects;
|
|
// using Robust.Server.Maps;
|
|
// using Robust.Shared.ContentPack;
|
|
// using Robust.Shared.Map.Events;
|
|
// using Robust.Shared.Prototypes;
|
|
// using Robust.Shared.Serialization.Markdown;
|
|
// using Robust.Shared.Serialization.Markdown.Mapping;
|
|
// using Robust.Shared.Serialization.Markdown.Value;
|
|
// using Robust.Shared.Utility;
|
|
|
|
// namespace Content.Server.Maps;
|
|
|
|
// /// <summary>
|
|
// /// Performs basic map migration operations by listening for engine <see cref="MapLoaderSystem"/> events.
|
|
// /// </summary>
|
|
// public sealed class MapMigrationSystem : EntitySystem
|
|
// {
|
|
// [Dependency] private IPrototypeManager _protoMan = default!;
|
|
// [Dependency] private IResourceManager _resMan = default!;
|
|
|
|
// private const string MigrationFile = "/migration.yml";
|
|
|
|
// //WL-Changes-start
|
|
// private const string WLMigrationFile = "/wl-migration.yml";
|
|
// //WL-Changes-end
|
|
|
|
// public override void Initialize()
|
|
// {
|
|
// base.Initialize();
|
|
// SubscribeLocalEvent<BeforeEntityReadEvent>(OnBeforeReadEvent);
|
|
|
|
// #if DEBUG
|
|
// if (!TryReadFile(out var mappings))
|
|
// return;
|
|
|
|
// // Verify that all of the entries map to valid entity prototypes.
|
|
// foreach (var node in mappings.Value.CorvaxMigration.Values)
|
|
// {
|
|
// var newId = ((ValueDataNode) node).Value;
|
|
// if (!string.IsNullOrEmpty(newId) && newId != "null")
|
|
// DebugTools.Assert(_protoMan.HasIndex<EntityPrototype>(newId), $"{newId} is not an entity prototype.");
|
|
// }
|
|
|
|
// //WL-Changes-start
|
|
// // Verify that all of the entries map to valid entity prototypes.
|
|
// foreach (var node in mappings.Value.WLMigration.Values)
|
|
// {
|
|
// var newId = ((ValueDataNode)node).Value;
|
|
// if (!string.IsNullOrEmpty(newId) && newId != "null")
|
|
// DebugTools.Assert(_protoMan.HasIndex<EntityPrototype>(newId), $"{newId} is not an entity prototype.");
|
|
// }
|
|
// //WL-Changes-end
|
|
// #endif
|
|
// }
|
|
|
|
// private bool TryReadFile(
|
|
// [NotNullWhen(true)] out /*WL-Changes-start*/(MappingDataNode CorvaxMigration, MappingDataNode WLMigration)?/*WL-Changes-start*/ mappings)
|
|
// {
|
|
// mappings = null;
|
|
// var path = new ResPath(MigrationFile);
|
|
|
|
// //WL-Changes-start
|
|
// var wl_path = new ResPath(WLMigrationFile);
|
|
// //WL-Changes-end
|
|
|
|
// if (!_resMan.TryContentFileRead(path, out var stream))
|
|
// return false;
|
|
|
|
// //WL-Changes-start
|
|
// if (!_resMan.TryContentFileRead(wl_path, out var wl_stream))
|
|
// return false;
|
|
// //WL-Changes-end
|
|
|
|
// using var reader = new StreamReader(stream, EncodingHelpers.UTF8);
|
|
// var documents = DataNodeParser.ParseYamlStream(reader).FirstOrDefault();
|
|
|
|
// //WL-Changes-start
|
|
// using var wl_reader = new StreamReader(wl_stream, EncodingHelpers.UTF8);
|
|
// var wl_documents = DataNodeParser.ParseYamlStream(wl_reader).FirstOrDefault();
|
|
// //WL-Changes-end
|
|
|
|
// if (documents == null /*WL-Changes-start*/|| wl_documents == null/*WL-Changes-end*/)
|
|
// return false;
|
|
|
|
// mappings = ((MappingDataNode)documents.Root, (MappingDataNode)wl_documents.Root);
|
|
// return true;
|
|
// }
|
|
|
|
// private void OnBeforeReadEvent(BeforeEntityReadEvent ev)
|
|
// {
|
|
// if (!TryReadFile(out var mappings_nullable))
|
|
// return;
|
|
|
|
// //WL-Changes-start
|
|
// var dict = new Dictionary<string, string>();
|
|
|
|
// var mappings = mappings_nullable.Value;
|
|
// var corvax_mappings = mappings.CorvaxMigration
|
|
// .Select(c_m =>
|
|
// {
|
|
// if (c_m.Key is not ValueDataNode keyNode || c_m.Value is not ValueDataNode valueNode)
|
|
// return ((string, string)?)null;
|
|
|
|
// return (keyNode.Value, valueNode.Value);
|
|
// })
|
|
// .Where(c_m => c_m != null);
|
|
|
|
// var wl_mappings = mappings.WLMigration
|
|
// .Select(w_m =>
|
|
// {
|
|
// if (w_m.Key is not ValueDataNode keyNode || w_m.Value is not ValueDataNode valueNode)
|
|
// return ((string, string)?)null;
|
|
|
|
// return (keyNode.Value, valueNode.Value);
|
|
// })
|
|
// .Where(w_m => w_m != null);
|
|
|
|
// // Именно в таком порядке: сначала миграция основы, только потом наша, вдруг наша перезаписывает что-то на основе.
|
|
// foreach (var c_m in corvax_mappings)
|
|
// {
|
|
// if (c_m == null)
|
|
// continue;
|
|
|
|
// var key = c_m.Value.Item1;
|
|
// var value = c_m.Value.Item2;
|
|
|
|
// dict[key] = value;
|
|
// }
|
|
|
|
// foreach (var w_m in wl_mappings)
|
|
// {
|
|
// if (w_m == null)
|
|
// continue;
|
|
|
|
// var key = w_m.Value.Item1;
|
|
// var value = w_m.Value.Item2;
|
|
|
|
// dict[key] = value;
|
|
// }
|
|
|
|
// //WL-Changes-end
|
|
|
|
|
|
// foreach (var (key, value) in /*WL-Changes-start*/dict/*WL-Changes-end*/)
|
|
// {
|
|
// if (string.IsNullOrWhiteSpace(/*WL-Changes-start*/value/*WL-Changes-end*/) || /*WL-Changes-start*/value/*WL-Changes-end*/ == "null")
|
|
// ev.DeletedPrototypes.Add(/*WL-Changes-start*/key/*WL-Changes-end*/);
|
|
// else
|
|
// ev.RenamedPrototypes.Add(/*WL-Changes-start*/key/*WL-Changes-end*/, /*WL-Changes-start*/value/*WL-Changes-end*//*WL-Changes-end*/);
|
|
// }
|
|
// }
|
|
// }
|