Add references of 'step projects' to test projects

Author: seveneleven

src/Moryx.Cli.Commands/AddStep.cs

@@ -1,5 +1,7 @@
-using Moryx.Cli.Commands.Extensions;
+using Moryx.Cli.Commands.Components;
+using Moryx.Cli.Commands.Extensions;
 using Moryx.Cli.Templates;
+using Moryx.Cli.Templates.Models;
 
 namespace Moryx.Cli.Commands
 {
@@ -20,23 +22,48 @@ public static CommandResult Exec(Template template, string step)
 
                     { $"{Template.AppPlaceholder}.Resources", $"{template.AppName}.Resources.{step}" },
                     { $"{template.AppName}.Resources", $"{template.AppName}.Resources.{step}" }
-                };  
+                };
                 var replacements = new StringReplacements(addConfig)
                     .AddFileNamePatterns(namespacePlaceholder)
                     .AddContentPatterns(namespacePlaceholder)
                     ;
-                
+
                 return AddThing.Exec(
                     template,
                     addConfig,
                     template.Step(step),
                     (createdFiles) =>
                     {
                         createdFiles.AddProjectsToSolution(template.Settings);
+                        AddProjectsTests(createdFiles, template.Settings);
                     },
                     replacements
                 );
             });
         }
+
+        private static void AddProjectsTests(IEnumerable<string> createdFiles, TemplateSettings settings)
+        {
+            var projectFiles = createdFiles
+               .Where(f => f.EndsWith(".csproj"))
+               .ToList();
+            var testFiles = createdFiles
+                .Where(f => f.ToLower().EndsWith("tests.cs") || f.ToLower().EndsWith("test.cs"))
+                .Select(Path.GetFullPath)
+                .Distinct()
+                .Select(p => Directory.GetFiles(Path.Combine(Path.GetDirectoryName(p) ?? "")))
+                .SelectMany(p => p)
+                .Where(f => f.EndsWith(".csproj"))
+                .ToList()
+                ?? [];
+
+            foreach (var testFile in testFiles)
+            {
+                foreach (var project in projectFiles)
+                {
+                    ProjectFileManipulation.AddProjectReference(testFile, project);
+                }
+            }
+        }
     }
 }

src/Moryx.Cli.Commands/Components/ProjectFileManipulation.cs

@@ -0,0 +1,62 @@
+using Castle.MicroKernel.Registration;
+using Microsoft.AspNetCore.Routing.Constraints;
+using Moryx.Container;
+using System.Xml;
+using System.Xml.Linq;
+
+namespace Moryx.Cli.Commands.Components
+{
+    internal class ProjectFileManipulation
+    {
+        private const string ProjectReference = "ProjectReference";
+        private const string ItemGroup = "ItemGroup";
+
+        public static void AddProjectReference(string targetProjectFileName, string referenceProjectFileName)
+        {
+
+            var referencePath = Path.GetRelativePath(targetProjectFileName, referenceProjectFileName);
+
+            var projectFile = LoadXml(targetProjectFileName);
+
+            if (projectFile.Root == null)
+                return;
+
+            var itemGroup = projectFile.Root
+                .Elements(ItemGroup)
+                .FirstOrDefault(i => i.Elements(ProjectReference).Any())
+                ;
+
+            if (itemGroup == null)
+            {
+                itemGroup = new XElement(ItemGroup);
+                projectFile.Root.Add(itemGroup);
+            }
+
+            var nodes = itemGroup.Nodes().Where(n => n.NodeType == XmlNodeType.Element).ToList();
+
+            var projectReference = new XElement(ProjectReference, new XAttribute("Include", referencePath));
+            nodes.Add(projectReference);
+
+            itemGroup.RemoveAll();
+            foreach (var node in nodes)
+            {
+                itemGroup.Add(node);
+                node.AddBeforeSelf(new XText($"{Environment.NewLine}    "));
+            }
+            itemGroup.LastNode?.AddAfterSelf(new XText($"{Environment.NewLine}  "));
+            
+            SaveXml(projectFile, targetProjectFileName);
+        }
+
+        private static XDocument LoadXml(string fileName)
+        {
+            using var reader = XmlReader.Create(fileName);
+            return XDocument.Load(reader, LoadOptions.PreserveWhitespace);
+        }
+
+        private static void SaveXml(XDocument xmlDocument, string fileName)
+        {
+            xmlDocument.Save(fileName);
+        }
+    }
+}