private void ExtractSelectionAndIncludes(EnvDTE.Document document, TrialAndErrorRemovalOptionsPage settings,
out ITextBuffer textBuffer, out Formatter.IncludeLineInfo[] includeLinesArray)
{
// Parsing.
document.Activate();
var documentTextView = VSUtils.GetCurrentTextViewHost();
textBuffer = documentTextView.TextView.TextBuffer;
string documentText = documentTextView.TextView.TextSnapshot.GetText();
IEnumerable <Formatter.IncludeLineInfo> includeLines = Formatter.IncludeLineInfo.ParseIncludes(documentText, Formatter.ParseOptions.IgnoreIncludesInPreprocessorConditionals | Formatter.ParseOptions.KeepOnlyValidIncludes);
// Optionally skip top most include.
if (settings.IgnoreFirstInclude)
{
includeLines = includeLines.Skip(1);
}
// Skip everything with preserve flag.
includeLines = includeLines.Where(x => !x.ShouldBePreserved);
// Apply filter ignore regex.
{
string documentName = Path.GetFileNameWithoutExtension(document.FullName);
string[] ignoreRegexList = RegexUtils.FixupRegexes(settings.IgnoreList, documentName);
includeLines = includeLines.Where(line => !ignoreRegexList.Any(regexPattern =>
new System.Text.RegularExpressions.Regex(regexPattern).Match(line.IncludeContent).Success));
}
// Reverse order if necessary.
if (settings.RemovalOrder == TrialAndErrorRemovalOptionsPage.IncludeRemovalOrder.BottomToTop)
{
includeLines = includeLines.Reverse();
}
includeLinesArray = includeLines.ToArray();
}
private void SubscribeBuildEvents()
{
ThreadHelper.ThrowIfNotOnUIThread();
buildEvents = VSUtils.GetDTE().Events.BuildEvents;
buildEvents.OnBuildDone += OnBuildFinished;
buildEvents.OnBuildProjConfigDone += OnBuildConfigFinished;
}
public static EnvDTE.Window OpenFileAndShowDocument(string filePath)
{
if (filePath == null)
{
return(null);
}
var dte = VSUtils.GetDTE();
EnvDTE.Window fileWindow = dte.ItemOperations.OpenFile(filePath);
if (fileWindow == null)
{
Output.Instance.WriteLine("Failed to open File {0}", filePath);
return(null);
}
fileWindow.Activate();
fileWindow.Visible = true;
return(fileWindow);
}
private void SubscribeBuildEvents()
{
buildEvents = VSUtils.GetDTE().Events.BuildEvents;
buildEvents.OnBuildDone += OnBuildFinished;
buildEvents.OnBuildProjConfigDone += OnBuildConfigFinished;
}
public string RunIncludeWhatYouUse(string fullFileName, EnvDTE.Project project, IncludeWhatYouUseOptionsPage settings)
{
string reasonForFailure;
string preprocessorDefintions = VSUtils.VCUtils.GetCompilerSetting_PreprocessorDefinitions(project, out reasonForFailure);
if (preprocessorDefintions == null)
{
Output.Instance.ErrorMsg("Can't run IWYU: {0}", reasonForFailure);
return(null);
}
string output = "";
using (var process = new System.Diagnostics.Process())
{
process.StartInfo.UseShellExecute = false;
process.StartInfo.CreateNoWindow = true;
process.StartInfo.RedirectStandardOutput = true;
process.StartInfo.RedirectStandardError = true;
process.StartInfo.FileName = Path.Combine(Path.GetDirectoryName(GetType().Assembly.Location), "include-what-you-use.exe");
process.StartInfo.Arguments = "";
// Includes and Preprocessor.
var includeEntries = VSUtils.GetProjectIncludeDirectories(project, false);
process.StartInfo.Arguments = includeEntries.Aggregate("", (current, inc) => current + ("-I \"" + inc + "\" "));
process.StartInfo.Arguments = preprocessorDefintions.
Split(new char[] { ';' }, StringSplitOptions.RemoveEmptyEntries).
Aggregate(process.StartInfo.Arguments, (current, def) => current + ("-D" + def + " "));
process.StartInfo.Arguments += " -DM_X64 -DM_AMD64 ";// TODO!!!
// Clang options
process.StartInfo.Arguments += "-w -x c++ -std=c++14 -fcxx-exceptions -fexceptions -fms-compatibility -fms-extensions -fmsc-version=1900 -ferror-limit=0 -Wno-invalid-token-paste "; // todo fmsc-version!
// icwyu options
{
process.StartInfo.Arguments += "-Xiwyu --verbose=" + settings.LogVerbosity + " ";
for (int i = 0; i < settings.MappingFiles.Length; ++i)
{
process.StartInfo.Arguments += "-Xiwyu --mapping_file=" + settings.MappingFiles[i] + " ";
}
if (settings.NoDefaultMappings)
{
process.StartInfo.Arguments += "-Xiwyu --no_default_mappings ";
}
if (settings.PCHInCode)
{
process.StartInfo.Arguments += "-Xiwyu --pch_in_code ";
}
switch (settings.PrefixHeaderIncludes)
{
case IncludeWhatYouUseOptionsPage.PrefixHeaderMode.Add:
process.StartInfo.Arguments += "-Xiwyu --prefix_header_includes=add ";
break;
case IncludeWhatYouUseOptionsPage.PrefixHeaderMode.Remove:
process.StartInfo.Arguments += "-Xiwyu --prefix_header_includes=remove ";
break;
case IncludeWhatYouUseOptionsPage.PrefixHeaderMode.Keep:
process.StartInfo.Arguments += "-Xiwyu --prefix_header_includes=keep ";
break;
}
if (settings.TransitiveIncludesOnly)
{
process.StartInfo.Arguments += "-Xiwyu --transitive_includes_only ";
}
}
// Finally, the file itself.
process.StartInfo.Arguments += "\"";
process.StartInfo.Arguments += fullFileName;
process.StartInfo.Arguments += "\"";
Output.Instance.Write("Running command '{0}' with following arguments:\n{1}\n\n", process.StartInfo.FileName, process.StartInfo.Arguments);
// Start the child process.
process.EnableRaisingEvents = true;
process.OutputDataReceived += (s, args) =>
{
Output.Instance.WriteLine(args.Data);
output += args.Data + "\n";
};
process.ErrorDataReceived += (s, args) =>
{
Output.Instance.WriteLine(args.Data);
output += args.Data + "\n";
};
process.Start();
process.BeginOutputReadLine();
process.BeginErrorReadLine();
process.WaitForExit();
process.CancelOutputRead();
process.CancelErrorRead();
}
return(output);
}
private void ApplyTasks(Dictionary <string, FormatTask> tasks)
{
var dte = VSUtils.GetDTE();
foreach (KeyValuePair <string, FormatTask> entry in tasks)
{
string filename = entry.Key.Replace('/', '\\'); // Classy. But Necessary.
EnvDTE.Window fileWindow = dte.ItemOperations.OpenFile(filename);
if (fileWindow == null)
{
Output.Instance.ErrorMsg("Failed to open File {0}", filename);
continue;
}
fileWindow.Activate();
var viewHost = VSUtils.GetCurrentTextViewHost();
using (var edit = viewHost.TextView.TextBuffer.CreateEdit())
{
var originalLines = edit.Snapshot.Lines.ToArray();
// Add lines.
{
// Find last include.
// Will find even if commented out, but we don't care.
int lastIncludeLine = -1;
for (int line = originalLines.Length - 1; line >= 0; --line)
{
if (originalLines[line].GetText().Contains("#include"))
{
lastIncludeLine = line;
break;
}
}
// Insert lines.
int insertPosition = 0;
if (lastIncludeLine >= 0 && lastIncludeLine < originalLines.Length)
{
insertPosition = originalLines[lastIncludeLine].EndIncludingLineBreak;
}
StringBuilder stringToInsert = new StringBuilder();
foreach (string lineToAdd in entry.Value.linesToAdd)
{
stringToInsert.Clear();
stringToInsert.Append(lineToAdd);
stringToInsert.Append("\n"); // todo: Consistent new lines?
edit.Insert(insertPosition, stringToInsert.ToString());
}
}
// Remove lines.
// It should safe to do that last since we added includes at the bottom, this way there is no confusion with the text snapshot.
{
foreach (int lineToRemove in entry.Value.linesToRemove.Reverse())
{
edit.Delete(originalLines[lineToRemove].ExtentIncludingLineBreak);
}
}
edit.Apply();
}
// For Debugging:
//Output.Instance.WriteLine("");
//Output.Instance.WriteLine(entry.Key);
//Output.Instance.WriteLine(entry.Value.ToString());
}
}
