Add CaptureDiagsKind to buildASTFromCodeWithArgs

Author: MythreyaK

clang/include/clang/Tooling/Tooling.h

@@ -32,6 +32,7 @@
 #include "clang/AST/ASTConsumer.h"
 #include "clang/Basic/FileManager.h"
 #include "clang/Basic/LLVM.h"
+#include "clang/Frontend/ASTUnit.h"
 #include "clang/Frontend/FrontendAction.h"
 #include "clang/Frontend/PCHContainerOperations.h"
 #include "clang/Tooling/ArgumentsAdjusters.h"
@@ -238,6 +239,7 @@ std::unique_ptr<ASTUnit> buildASTFromCodeWithArgs(
     ArgumentsAdjuster Adjuster = getClangStripDependencyFileAdjuster(),
     const FileContentMappings &VirtualMappedFiles = FileContentMappings(),
     DiagnosticConsumer *DiagConsumer = nullptr,
+    CaptureDiagsKind CaptureKind = CaptureDiagsKind::None,
     IntrusiveRefCntPtr<llvm::vfs::FileSystem> BaseFS =
         llvm::vfs::getRealFileSystem());
 

clang/lib/Tooling/Tooling.cpp

@@ -692,7 +692,7 @@ std::unique_ptr<ASTUnit> buildASTFromCodeWithArgs(
     StringRef Code, const std::vector<std::string> &Args, StringRef FileName,
     StringRef ToolName, std::shared_ptr<PCHContainerOperations> PCHContainerOps,
     ArgumentsAdjuster Adjuster, const FileContentMappings &VirtualMappedFiles,
-    DiagnosticConsumer *DiagConsumer,
+    DiagnosticConsumer *DiagConsumer, CaptureDiagsKind CaptureKind,
     IntrusiveRefCntPtr<llvm::vfs::FileSystem> BaseFS) {
   std::vector<std::unique_ptr<ASTUnit>> ASTs;
   ASTBuilderAction Action(ASTs);

clang/unittests/Tooling/ToolingTest.cpp

@@ -161,7 +161,7 @@ TEST(buildASTFromCode, FileSystem) {
       R"(#include "included_file.h")", {}, "input.cc", "clang-tool",
       std::make_shared<PCHContainerOperations>(),
       getClangStripDependencyFileAdjuster(), FileContentMappings(), nullptr,
-      InMemoryFileSystem);
+      CaptureDiagsKind::None, InMemoryFileSystem);
   ASSERT_TRUE(AST.get());
   EXPECT_TRUE(FindClassDeclX(AST.get()));
 }