added project .cs files to reflection

HuaFangYun · May 1, 2019 · a0e8e41 · a0e8e41
1 parent 6b04647
commit a0e8e41
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/NetPrintsVSIX/NetPrintsEditorFactory.cs b/NetPrintsVSIX/NetPrintsEditorFactory.cs
@@ -69,8 +69,8 @@ public int CreateEditorInstance(uint grfCreateDoc, string pszMkDocument, string
                 // Add referenced assemblies
                 var assemblyPaths = package.GetAssemblyReferences().Select(assemblyRef => assemblyRef.AssemblyPath);
 
-                // TODO: Get .cs files
-                var sourcePaths = new string[0];
+                // Get source files in projects
+                var sourcePaths = package.GetSourceDirectoryReferences().SelectMany(sourceRef => sourceRef.SourceFilePaths);
 
                 // Add our own sources
                 var sources = package.GetGeneratedCode();

diff --git a/NetPrintsVSIX/NetPrintsVSIXPackage.cs b/NetPrintsVSIX/NetPrintsVSIXPackage.cs
@@ -118,6 +118,15 @@ public IEnumerable<AssemblyReference> GetAssemblyReferences()
                     .Select(r => new AssemblyReference(r.Path)));
         }
 
+        public IEnumerable<SourceDirectoryReference> GetSourceDirectoryReferences()
+        {
+            ThreadHelper.ThrowIfNotOnUIThread();
+
+            return dte.Solution.Projects
+                .OfType<EnvDTE.Project>()
+                .Select(proj => new SourceDirectoryReference(Path.GetDirectoryName(proj.FileName)));
+        }
+
         public IEnumerable<string> GetGeneratedCode()
         {
             ClassTranslator classTranslator = new ClassTranslator();