diff --git a/clang-tools-extra/clangd/FeatureModule.h b/clang-tools-extra/clangd/FeatureModule.h --- a/clang-tools-extra/clangd/FeatureModule.h +++ b/clang-tools-extra/clangd/FeatureModule.h @@ -20,6 +20,7 @@ #include namespace clang { +class CompilerInstance; namespace clangd { struct Diag; class LSPBinder; @@ -105,6 +106,10 @@ /// Listeners are destroyed once the AST is built. virtual ~ASTListener() = default; + // Called before the preamble build. Allows modules to modify the + // CompilerInvocation, prefetch required files, etc. + virtual void beforeExecute(CompilerInstance &CI) {} + /// Called everytime a diagnostic is encountered. Modules can use this /// modify the final diagnostic, or store some information to surface code /// actions later on. diff --git a/clang-tools-extra/clangd/ParsedAST.cpp b/clang-tools-extra/clangd/ParsedAST.cpp --- a/clang-tools-extra/clangd/ParsedAST.cpp +++ b/clang-tools-extra/clangd/ParsedAST.cpp @@ -550,6 +550,9 @@ // Collect tokens of the main file. syntax::TokenCollector CollectTokens(Clang->getPreprocessor()); + for (const auto &L : ASTListeners) + L->beforeExecute(*Clang); + if (llvm::Error Err = Action->Execute()) log("Execute() failed when building AST for {0}: {1}", MainInput.getFile(), toString(std::move(Err))); diff --git a/clang-tools-extra/clangd/Preamble.cpp b/clang-tools-extra/clangd/Preamble.cpp --- a/clang-tools-extra/clangd/Preamble.cpp +++ b/clang-tools-extra/clangd/Preamble.cpp @@ -64,9 +64,12 @@ class CppFilePreambleCallbacks : public PreambleCallbacks { public: - CppFilePreambleCallbacks(PathRef File, PreambleParsedCallback ParsedCallback, - PreambleBuildStats *Stats) - : File(File), ParsedCallback(ParsedCallback), Stats(Stats) {} + CppFilePreambleCallbacks( + PathRef File, PreambleParsedCallback ParsedCallback, + PreambleBuildStats *Stats, + std::function BeforeExecuteCallback) + : File(File), ParsedCallback(ParsedCallback), Stats(Stats), + BeforeExecuteCallback(BeforeExecuteCallback) {} IncludeStructure takeIncludes() { return std::move(Includes); } @@ -111,6 +114,8 @@ } void BeforeExecute(CompilerInstance &CI) override { + if (BeforeExecuteCallback) + BeforeExecuteCallback(CI); CanonIncludes.addSystemHeadersMapping(CI.getLangOpts()); LangOpts = &CI.getLangOpts(); SourceMgr = &CI.getSourceManager(); @@ -156,6 +161,7 @@ const clang::LangOptions *LangOpts = nullptr; const SourceManager *SourceMgr = nullptr; PreambleBuildStats *Stats; + std::function BeforeExecuteCallback; }; // Represents directives other than includes, where basic textual information is @@ -477,7 +483,12 @@ // to read back. We rely on dynamic index for the comments instead. CI.getPreprocessorOpts().WriteCommentListToPCH = false; - CppFilePreambleCallbacks CapturedInfo(FileName, PreambleCallback, Stats); + CppFilePreambleCallbacks CapturedInfo(FileName, PreambleCallback, Stats, + [&ASTListeners](CompilerInstance &CI) { + for (const auto &L : ASTListeners) { + L->beforeExecute(CI); + } + }); auto VFS = Inputs.TFS->view(Inputs.CompileCommand.Directory); llvm::SmallString<32> AbsFileName(FileName); VFS->makeAbsolute(AbsFileName); @@ -716,5 +727,6 @@ } return Loc; } + } // namespace clangd } // namespace clang diff --git a/clang-tools-extra/clangd/unittests/FeatureModulesTests.cpp b/clang-tools-extra/clangd/unittests/FeatureModulesTests.cpp --- a/clang-tools-extra/clangd/unittests/FeatureModulesTests.cpp +++ b/clang-tools-extra/clangd/unittests/FeatureModulesTests.cpp @@ -12,6 +12,7 @@ #include "TestTU.h" #include "refactor/Tweak.h" #include "support/Logger.h" +#include "clang/Lex/PreprocessorOptions.h" #include "llvm/Support/Error.h" #include "gmock/gmock.h" #include "gtest/gtest.h" @@ -85,6 +86,41 @@ } } +TEST(FeatureModulesTest, BeforeExecute) { + struct BeforeExecuteModule final : public FeatureModule { + struct Listener : public FeatureModule::ASTListener { + void beforeExecute(CompilerInstance &CI) override { + CI.getPreprocessor().SetSuppressIncludeNotFoundError(true); + } + }; + std::unique_ptr astListeners() override { + return std::make_unique(); + }; + }; + FeatureModuleSet FMS; + FMS.add(std::make_unique()); + + TestTU TU = TestTU::withCode(R"cpp( + /*error-ok*/ + #include "not_found.h" + + void foo() { + #include "not_found_not_preamble.h" + } + )cpp"); + + { + auto AST = TU.build(); + EXPECT_THAT(*AST.getDiagnostics(), testing::Not(testing::IsEmpty())); + } + + TU.FeatureModules = &FMS; + { + auto AST = TU.build(); + EXPECT_THAT(*AST.getDiagnostics(), testing::IsEmpty()); + } +} + } // namespace } // namespace clangd } // namespace clang