Mapping merge driver: continued (space-wizards#2803)

Co-authored-by: DrSmugleaf <[email protected]>
Co-authored-by: Pieter-Jan Briers <[email protected]>

Author: 3 people

.gitattributes

@@ -34,6 +34,7 @@
 #*.modelproj merge=binary
 #*.sqlproj   merge=binary
 #*.wwaproj   merge=binary
+Resources/Maps/**.yml merge=mapping-merge-driver
 
 ###############################################################################
 # behavior for image files

.gitignore

@@ -284,3 +284,7 @@ BuildFiles/Windows/Godot/*
 # Windows image file caches
 Thumbs.db
 ehthumbs.db
+
+# Merge driver stuff
+Content.Tools/test/out.yml
+

Content.Tools/Content.Tools.csproj

@@ -0,0 +1,16 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+    <PropertyGroup>
+        <OutputType>Exe</OutputType>
+        <TargetFramework>net5.0</TargetFramework>
+    </PropertyGroup>
+
+    <ItemGroup>
+      <PackageReference Include="YamlDotNet" Version="9.1.0" />
+    </ItemGroup>
+
+    <ItemGroup>
+      <ProjectReference Include="..\RobustToolbox\Robust.Shared\Robust.Shared.csproj" />
+    </ItemGroup>
+
+</Project>

Content.Tools/Map.cs

@@ -0,0 +1,83 @@
+using System.IO;
+using System.Collections.Generic;
+using System.Globalization;
+using System.Linq;
+using YamlDotNet.Core;
+using Robust.Shared.Utility;
+using YamlDotNet.RepresentationModel;
+
+namespace Content.Tools
+{
+    public class Map
+    {
+        public Map(string path)
+        {
+            Path = path;
+
+            using var reader = new StreamReader(path);
+            var stream = new YamlStream();
+
+            stream.Load(reader);
+
+            Root = stream.Documents[0].RootNode;
+            TilemapNode = (YamlMappingNode) Root["tilemap"];
+            GridsNode = (YamlSequenceNode) Root["grids"];
+            _entitiesNode = (YamlSequenceNode) Root["entities"];
+
+            foreach (var entity in _entitiesNode)
+            {
+                var uid = uint.Parse(entity["uid"].AsString());
+                if (uid >= NextAvailableEntityId)
+                    NextAvailableEntityId = uid + 1;
+                Entities[uid] = (YamlMappingNode) entity;
+            }
+        }
+
+        // Core
+
+        public string Path { get; }
+
+        public YamlNode Root { get; }
+
+        // Useful
+
+        public YamlMappingNode TilemapNode { get; }
+
+        public YamlSequenceNode GridsNode { get; }
+
+        // Entities lookup
+
+        private YamlSequenceNode _entitiesNode { get; }
+
+        public Dictionary<uint, YamlMappingNode> Entities { get; } = new Dictionary<uint, YamlMappingNode>();
+
+        public uint MaxId => Entities.Max(entry => entry.Key);
+
+        public uint NextAvailableEntityId { get; set; }
+
+        // ----
+
+        public void Save(string fileName)
+        {
+            // Update entities node
+            _entitiesNode.Children.Clear();
+            foreach (var kvp in Entities)
+                _entitiesNode.Add(kvp.Value);
+
+            using var writer = new StreamWriter(fileName);
+            var document = new YamlDocument(Root);
+            var stream = new YamlStream(document);
+            var emitter = new Emitter(writer);
+            var fixer = new TypeTagPreserver(emitter);
+
+            stream.Save(fixer, false);
+
+            writer.Flush();
+        }
+
+        public void Save()
+        {
+            Save(Path);
+        }
+    }
+}

Content.Tools/MappingMergeDriver.cs

@@ -0,0 +1,38 @@
+using System;
+using System.Collections.Generic;
+using System.Globalization;
+using System.Linq;
+using YamlDotNet.Core;
+using YamlDotNet.RepresentationModel;
+
+namespace Content.Tools
+{
+    internal static class MappingMergeDriver
+    {
+        /// %A: Our file
+        /// %O: Origin (common, base) file
+        /// %B: Other file
+        /// %P: Actual filename of the resulting file
+        public static void Main(string[] args)
+        {
+            var ours = new Map(args[0]);
+            var based = new Map(args[1]); // On what?
+            var other = new Map(args[2]);
+
+            if ((ours.GridsNode.Children.Count != 1) || (based.GridsNode.Children.Count != 1) || (other.GridsNode.Children.Count != 1))
+            {
+                Console.WriteLine("one or more files had an amount of grids not equal to 1");
+                Environment.Exit(1);
+            }
+
+            if (!(new Merger(ours, based, other).Merge()))
+            {
+                Console.WriteLine("unable to merge!");
+                Environment.Exit(1);
+            }
+
+            ours.Save();
+            Environment.Exit(0);
+        }
+    }
+}