Fix linter error (#20334)

dvir001 · Sep 19, 2023 · 39af74e · 39af74e
1 parent b8697d8
commit 39af74e
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/Content.YAMLLinter/Program.cs b/Content.YAMLLinter/Program.cs
@@ -77,7 +77,26 @@ private static async Task<int> Main(string[] _)
 
             await instance.WaitPost(() =>
             {
+                var engineErrors = protoMan.ValidateDirectory(new ResPath("/EnginePrototypes"), out var engPrototypes);
                 yamlErrors = protoMan.ValidateDirectory(new ResPath("/Prototypes"), out var prototypes);
+
+                // Merge engine & content prototypes
+                foreach (var (kind, instances) in engPrototypes)
+                {
+                    if (prototypes.TryGetValue(kind, out var existing))
+                        existing.UnionWith(instances);
+                    else
+                        prototypes[kind] = instances;
+                }
+
+                foreach (var (kind, set) in engineErrors)
+                {
+                    if (yamlErrors.TryGetValue(kind, out var existing))
+                        existing.UnionWith(set);
+                    else
+                        yamlErrors[kind] = set;
+                }
+
                 fieldErrors = protoMan.ValidateFields(prototypes);
             });