support sln file as entrypoint

Author: othmane-kinane-nw

Directory.Build.props

@@ -49,7 +49,7 @@
   <PropertyGroup>
     <Product>dotnet-subset</Product>
     <Description>.NET Tool to copy a subset of files from a repository to a directory</Description>
-    <VersionPrefix>0.2.0</VersionPrefix>
+    <VersionPrefix>0.3.0</VersionPrefix>
     <Company>Nimbleways</Company>
     <Authors>$(Company.ToLower())</Authors>
     <Copyright>© $(Company). All rights reserved.</Copyright>

src/dotnet-subset/Program.cs

@@ -9,9 +9,9 @@
 
 MSBuildLocator.RegisterDefaults();
 
-var projectArgument = new Argument<FileInfo>(
-    name: "project",
-    description: "Project to restore.");
+var projectOrSolutionArgument = new Argument<FileInfo>(
+    name: "projectOrSolution",
+    description: "Project or solution to restore.");
 
 var rootDirectoryOption = new Option<DirectoryInfo>(
     name: "--root-directory",
@@ -30,10 +30,10 @@
                 rootDirectoryOption,
                 outputDirectoryOption,
             };
-readCommand.AddArgument(projectArgument);
+readCommand.AddArgument(projectOrSolutionArgument);
 rootCommand.AddCommand(readCommand);
 
-readCommand.SetHandler((project, rootDirectory, outputDirectory) => RestoreSubset.Execute(project.FullName, rootDirectory.FullName, outputDirectory.FullName),
-    projectArgument, rootDirectoryOption, outputDirectoryOption);
+readCommand.SetHandler((projectOrSolution, rootDirectory, outputDirectory) => RestoreSubset.Execute(projectOrSolution.FullName, rootDirectory.FullName, outputDirectory.FullName),
+    projectOrSolutionArgument, rootDirectoryOption, outputDirectoryOption);
 
 return rootCommand.Invoke(args);

src/dotnet-subset/RestoreSubset.cs

@@ -1,31 +1,39 @@
+using Microsoft.Build.Construction;
 using Microsoft.Build.Evaluation;
 
 namespace Nimbleways.Tools.Subset;
 internal static class RestoreSubset
 {
-    public static void Execute(string mainProjectPath, string rootFolder, string destinationFolder)
+    public static void Execute(string projectOrSolution, string rootFolder, string destinationFolder)
     {
-        if (!IsSameOrUnder(rootFolder, mainProjectPath))
+        if (!IsSameOrUnder(rootFolder, projectOrSolution))
         {
-            throw new ArgumentException($"Project '${mainProjectPath}' must be under the root folder '{rootFolder}'");
+            throw new ArgumentException($"Project or solution '${projectOrSolution}' must be under the root folder '{rootFolder}'");
         }
         Directory.CreateDirectory(destinationFolder);
 
         using var projectCollection = new ProjectCollection();
         var projectsByFullPath = new Dictionary<string, Project>();
-        VisitAllProjects(projectCollection, rootFolder, mainProjectPath, projectsByFullPath);
+        foreach (var project in GetRootProjects(projectOrSolution))
+        {
+            VisitAllProjects(projectCollection, rootFolder, project, projectsByFullPath);
+        }
         var projectListAsString = string.Join(Environment.NewLine + " - ", projectsByFullPath.Keys.OrderBy(f => f));
         Console.WriteLine($"Found {projectsByFullPath.Count} project(s) to copy:{Environment.NewLine + " - "}{projectListAsString}");
         var nugetConfigFiles = GetNugetConfigFiles(rootFolder, projectsByFullPath);
         var extraFilesInvolvedInRestore = projectsByFullPath.Values
             .SelectMany(project => GetExtraFilesInvolvedInRestore(rootFolder, project))
             .Concat(nugetConfigFiles)
             .Distinct()
-            .ToArray();
-        if (extraFilesInvolvedInRestore.Length > 0)
+            .ToList();
+        if (IsSolutionFile(projectOrSolution))
+        {
+            extraFilesInvolvedInRestore.Add(projectOrSolution);
+        }
+        if (extraFilesInvolvedInRestore.Count > 0)
         {
             var extraFilesInvolvedInRestoreAsString = string.Join(Environment.NewLine + " - ", extraFilesInvolvedInRestore.OrderBy(f => f));
-            Console.WriteLine($"Found {extraFilesInvolvedInRestore.Length} extra file(s) to copy:{Environment.NewLine + " - "}{extraFilesInvolvedInRestoreAsString}");
+            Console.WriteLine($"Found {extraFilesInvolvedInRestore.Count} extra file(s) to copy:{Environment.NewLine + " - "}{extraFilesInvolvedInRestoreAsString}");
         }
         var allFilesToCopy = projectsByFullPath.Keys.Concat(extraFilesInvolvedInRestore).Distinct();
 
@@ -50,6 +58,22 @@ public static void Execute(string mainProjectPath, string rootFolder, string des
         }
         Console.WriteLine($"Copied {copiedFilesCount} file(s) to '{destinationFolder}'. {allFilesCount - copiedFilesCount} file(s) already exist in destination.");
     }
+    private static IEnumerable<string> GetRootProjects(string projectOrSolution)
+    {
+        if (IsSolutionFile(projectOrSolution))
+        {
+            var solution = SolutionFile.Parse(projectOrSolution);
+            return solution.ProjectsInOrder
+                .Where(p => p.ProjectType != SolutionProjectType.SolutionFolder)
+                .Select(p => p.AbsolutePath);
+        }
+        return new[] { projectOrSolution };
+    }
+
+    private static bool IsSolutionFile(string projectOrSolution)
+    {
+        return ".sln".Equals(Path.GetExtension(projectOrSolution), StringComparison.OrdinalIgnoreCase);
+    }
 
     private static bool IsSameOrUnder(string rootFolder, string path)
     {