Resolves #2: Extend SolutionHelper to find similar project items for shared projects and add RetrieveFileCodeModel to CodeModelBuilder that will conditionally resolve the FileCodeModel through a similar project item.

Author: codecadwallader

CodeMaid/Helpers/SolutionHelper.cs

@@ -97,6 +97,19 @@ internal static IEnumerable<ProjectItem> GetSelectedProjectItemsRecursively(Code
             return selectedProjectItems;
         }
 
+        /// <summary>
+        /// Gets an enumerable set of similar project items compared by file name, useful for shared projects.
+        /// </summary>
+        /// <param name="package">The hosting package.</param>
+        /// <param name="projectItem">The project item to match.</param>
+        /// <returns>The enumerable set of similar project items.</returns>
+        internal static IEnumerable<ProjectItem> GetSimilarProjectItems(CodeMaidPackage package, ProjectItem projectItem)
+        {
+            var allItems = GetAllItemsInSolution<ProjectItem>(package.IDE.Solution);
+
+            return allItems.Where(x => x.Name == projectItem.Name && x.Kind == projectItem.Kind && x.Document.FullName == projectItem.Document.FullName);
+        }
+
         #endregion Internal Methods
 
         #region Private Methods

CodeMaid/Model/CodeModelBuilder.cs

@@ -72,10 +72,8 @@ internal SetCodeItems RetrieveAllCodeItems(Document document)
         {
             var codeItems = new SetCodeItems();
 
-            if (document.ProjectItem != null)
-            {
-                RetrieveCodeItems(codeItems, document.ProjectItem.FileCodeModel);
-            }
+            var fileCodeModel = RetrieveFileCodeModel(document.ProjectItem);
+            RetrieveCodeItems(codeItems, fileCodeModel);
 
             codeItems.AddRange(_codeModelHelper.RetrieveCodeRegions(document.GetTextDocument()));
 
@@ -86,6 +84,39 @@ internal SetCodeItems RetrieveAllCodeItems(Document document)
 
         #region Private Methods
 
+        /// <summary>
+        /// Attempts to return the FileCodeModel associated with the specified project item.
+        /// </summary>
+        /// <param name="projectItem">The project item.</param>
+        /// <returns>The associated FileCodeModel, otherwise null.</returns>
+        private FileCodeModel RetrieveFileCodeModel(ProjectItem projectItem)
+        {
+            if (projectItem == null)
+            {
+                return null;
+            }
+
+            if (projectItem.FileCodeModel != null)
+            {
+                return projectItem.FileCodeModel;
+            }
+
+            // If this project item is part of a shared project, retrieve the FileCodeModel via a similar platform project item.
+            const string sharedProjectTypeGUID = "{d954291e-2a0b-460d-934e-dc6b0785db48}";
+            var containingProject = projectItem.ContainingProject;
+
+            if (containingProject != null && containingProject.Kind != null &&
+                containingProject.Kind.ToLowerInvariant() == sharedProjectTypeGUID)
+            {
+                var similarProjectItems = SolutionHelper.GetSimilarProjectItems(_package, projectItem);
+                var fileCodeModel = similarProjectItems.Select(x => x.FileCodeModel).FirstOrDefault(y => y != null);
+
+                return fileCodeModel;
+            }
+
+            return null;
+        }
+
         /// <summary>
         /// Walks the given FileCodeModel, turning CodeElements into code items within the specified
         /// code items set.