public static Document?GetRoslynDocumentFromVisualStudioDocument(this VisualStudioWorkspace workspace, EnvDTE.Document visualStudioDocument)
{
if (workspace.CurrentSolution == null)
{
return(null);
}
if (visualStudioDocument?.ProjectItem == null)
{
return(null);
}
var documentsProject = workspace.GetRoslynProjectFromVisualStudioProject(visualStudioDocument.ProjectItem.ContainingProject);
return(documentsProject?
.Documents
.FirstOrDefault(document => document.FilePath == visualStudioDocument.FullName));
}
public static IReadOnlyCollection <Document> GetRoslynDocumentsFromVisualStudioProjectItems(this VisualStudioWorkspace workspace, IEnumerable <ProjectItem> visualStudioProjectItems)
{
// TODO-PERF: It's crazy what this method does for such a trivial(?) task
// of mapping "this to that"!!
// Is there any better way to do this?
// (We expect the number of selected items to always be small, but still.)
if (workspace.CurrentSolution == null)
{
return(Array.Empty <Document>());
}
if (visualStudioProjectItems == null)
{
return(Array.Empty <Document>());
}
var roslynProjects = new Dictionary <string, Project>();
var result = new HashSet <Document>();
foreach (var projectItem in visualStudioProjectItems)
{
var roslynProject = GetRoslynProjectOfProjectItem(projectItem);
if (roslynProject == null)
{
continue;
}
var selectedItemFileName = GetProjectItemFileName(projectItem);
if (selectedItemFileName == null)
{
continue;
}
var roslynDocument = roslynProject
.Documents
.FirstOrDefault(document => document.FilePath == selectedItemFileName);
if (roslynDocument != null)
{
result.Add(roslynDocument);
}
}
return(result);
string?GetProjectItemFileName(ProjectItem projectItem)
{
if (projectItem?.FileCount != 1)
{
return(null);
}
// Some items report a file count > 0 but don't return a file name!
// See: https://github.com/tom-englert/Wax/blob/210b1038b0c282f3ae7c399178ae17bc5bf8fcd8/Wax.Model/VisualStudio/DteExtensions.cs#L181
try
{
return(projectItem.FileNames[0]);
}
catch
{
return(null);
}
}
Project?GetRoslynProjectOfProjectItem(ProjectItem projectItem)
{
var visualStudioProject = projectItem?.ContainingProject;
var projectUniqueName = visualStudioProject?.UniqueName;
if (projectUniqueName == null)
{
return(null);
}
if (!roslynProjects.ContainsKey(projectUniqueName))
{
roslynProjects.Add(projectUniqueName, workspace.GetRoslynProjectFromVisualStudioProject(visualStudioProject));
}
return(roslynProjects[projectUniqueName]);
}
}
