Deal with more complicated solution structures

Author: GrahamTheCoder

Vsix/SolutionProjectExtensions.cs

@@ -0,0 +1,28 @@
+using System.Collections.Generic;
+using System.Linq;
+using EnvDTE;
+using EnvDTE80;
+
+namespace CodeConverter.VsExtension
+{
+    /// <summary>
+    /// http://www.wwwlicious.com/2011/03/29/envdte-getting-all-projects-html/#comment-2557459433
+    /// </summary>
+    public static class SolutionProjectExtensions
+    {
+        public static IEnumerable<Project> GetAllProjects(this Solution sln)
+        {
+            return sln.Projects.Cast<Project>().SelectMany(GetProjects);
+        }
+
+        public static IEnumerable<Project> GetProjects(this Project project)
+        {
+            if (project.Kind == ProjectKinds.vsProjectKindSolutionFolder) {
+                return project.ProjectItems.Cast<ProjectItem>()
+                    .Select(x => x.SubProject).Where(x => x != null)
+                    .SelectMany(GetProjects);
+            }
+            return new[] { project };
+        }
+    }
+}

Vsix/VisualStudioInteraction.cs

@@ -68,6 +68,11 @@ private static IEnumerable<T> GetSelectedSolutionExplorerItems<T>() where T: cla
             var selectedObjects = (IEnumerable<object>) Dte.ToolWindows.SolutionExplorer.SelectedItems;
             var selectedItems = selectedObjects.Cast<UIHierarchyItem>().ToList();
 
+            return ObjectOfType<T>(selectedItems);
+        }
+
+        private static IEnumerable<T> ObjectOfType<T>(IReadOnlyCollection<UIHierarchyItem> selectedItems) where T : class
+        {
             var returnType = typeof(T);
             return selectedItems.Select(item => item.Object).Where(returnType.IsInstanceOfType).Cast<T>();
         }
@@ -86,13 +91,12 @@ public static void SelectAll(this Window window)
 
         public static IReadOnlyCollection<Project> GetSelectedProjects(string projectExtension)
         {
-            var items = GetSelectedSolutionExplorerItems<Solution>()
-                .SelectMany(s => s.Projects.Cast<Project>())
-                .Concat(GetSelectedSolutionExplorerItems<Project>())
-                .Concat(GetSelectedSolutionExplorerItems<ProjectItem>().Where(p => p.SubProject != null).Select(p => p.SubProject))
+            var projects = GetSelectedSolutionExplorerItems<Solution>().SelectMany(s => s.GetAllProjects())
+                .Concat(GetSelectedSolutionExplorerItems<Project>().SelectMany(p => p.GetProjects()))
+                .Concat(GetSelectedSolutionExplorerItems<ProjectItem>().Where(p => p.SubProject != null).SelectMany(p => p.SubProject.GetProjects()))
                 .Where(project => project.FullName.EndsWith(projectExtension, StringComparison.InvariantCultureIgnoreCase))
-                .ToList();
-            return items;
+                .Distinct().ToList();
+            return projects;
         }
 
         public static IWpfTextViewHost GetCurrentViewHost(IServiceProvider serviceProvider)

Vsix/Vsix.csproj

@@ -245,6 +245,7 @@
     </Compile>
     <Compile Include="ConvertVBToCSCommand.cs" />
     <Compile Include="REConverterPackage.cs" />
+    <Compile Include="SolutionProjectExtensions.cs" />
     <Compile Include="VisualStudioInteraction.cs" />
     <Compile Include="VsDocument.cs" />
   </ItemGroup>