clang-tools  10.0.0git
IncludeFixerPlugin.cpp
Go to the documentation of this file.
1 //===- IncludeFixerPlugin.cpp - clang-include-fixer as a clang plugin -----===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "../IncludeFixer.h"
10 #include "../YamlSymbolIndex.h"
11 #include "clang/Frontend/CompilerInstance.h"
12 #include "clang/Frontend/FrontendPluginRegistry.h"
13 #include "clang/Parse/ParseAST.h"
14 #include "clang/Sema/Sema.h"
15 #include "llvm/Support/Path.h"
16 
17 namespace clang {
18 namespace include_fixer {
19 
20 /// The core include fixer plugin action. This just provides the AST consumer
21 /// and command line flag parsing for using include fixer as a clang plugin.
23  /// ASTConsumer to keep the symbol index alive. We don't really need an
24  /// ASTConsumer for this plugin (everything is funneled on the side through
25  /// Sema) but we have to keep the symbol index alive until sema is done.
26  struct ASTConsumerManagerWrapper : public ASTConsumer {
27  ASTConsumerManagerWrapper(std::shared_ptr<SymbolIndexManager> SIM)
28  : SymbolIndexMgr(std::move(SIM)) {}
29  std::shared_ptr<SymbolIndexManager> SymbolIndexMgr;
30  };
31 
32 public:
34  : SymbolIndexMgr(std::make_shared<SymbolIndexManager>()),
35  SemaSource(new IncludeFixerSemaSource(*SymbolIndexMgr,
36  /*MinimizeIncludePaths=*/true,
37  /*GenerateDiagnostics=*/true)) {}
38 
39  std::unique_ptr<clang::ASTConsumer>
40  CreateASTConsumer(clang::CompilerInstance &CI, StringRef InFile) override {
41  CI.setExternalSemaSource(SemaSource);
42  SemaSource->setFilePath(InFile);
43  SemaSource->setCompilerInstance(&CI);
44  return std::make_unique<ASTConsumerManagerWrapper>(SymbolIndexMgr);
45  }
46 
47  void ExecuteAction() override {} // Do nothing.
48 
49  bool ParseArgs(const CompilerInstance &CI,
50  const std::vector<std::string> &Args) override {
51  StringRef DB = "yaml";
52  StringRef Input;
53 
54  // Parse the extra command line args.
55  // FIXME: This is very limited at the moment.
56  for (StringRef Arg : Args) {
57  if (Arg.startswith("-db="))
58  DB = Arg.substr(strlen("-db="));
59  else if (Arg.startswith("-input="))
60  Input = Arg.substr(strlen("-input="));
61  }
62 
63  std::string InputFile = CI.getFrontendOpts().Inputs[0].getFile();
64  auto CreateYamlIdx = [=]() -> std::unique_ptr<include_fixer::SymbolIndex> {
65  llvm::ErrorOr<std::unique_ptr<include_fixer::YamlSymbolIndex>> SymbolIdx(
66  nullptr);
67  if (DB == "yaml") {
68  if (!Input.empty()) {
70  } else {
71  // If we don't have any input file, look in the directory of the first
72  // file and its parents.
73  SmallString<128> AbsolutePath(tooling::getAbsolutePath(InputFile));
74  StringRef Directory = llvm::sys::path::parent_path(AbsolutePath);
76  Directory, "find_all_symbols_db.yaml");
77  }
78  }
79  return std::move(*SymbolIdx);
80  };
81 
82  SymbolIndexMgr->addSymbolIndex(std::move(CreateYamlIdx));
83  return true;
84  }
85 
86 private:
87  std::shared_ptr<SymbolIndexManager> SymbolIndexMgr;
88  IntrusiveRefCntPtr<IncludeFixerSemaSource> SemaSource;
89 };
90 } // namespace include_fixer
91 } // namespace clang
92 
93 // This anchor is used to force the linker to link in the generated object file
94 // and thus register the include fixer plugin.
96 
97 static clang::FrontendPluginRegistry::Add<
99  X("clang-include-fixer", "clang-include-fixer");
Handles callbacks from sema, does the include lookup and turns it into an IncludeFixerContext.
static llvm::ErrorOr< std::unique_ptr< YamlSymbolIndex > > createFromDirectory(llvm::StringRef Directory, llvm::StringRef Name)
Look for a file called Name in Directory and all parent directories.
This class provides an interface for finding the header files corresponding to an identifier in the s...
llvm::StringRef Directory
std::unique_ptr< clang::ASTConsumer > CreateASTConsumer(clang::CompilerInstance &CI, StringRef InFile) override
===– Representation.cpp - ClangDoc Representation --------—*- C++ -*-===//
static llvm::ErrorOr< std::unique_ptr< YamlSymbolIndex > > createFromFile(llvm::StringRef FilePath)
Create a new Yaml db from a file.
static clang::FrontendPluginRegistry::Add< clang::include_fixer::ClangIncludeFixerPluginAction > X("clang-include-fixer", "clang-include-fixer")
volatile int ClangIncludeFixerPluginAnchorSource
bool ParseArgs(const CompilerInstance &CI, const std::vector< std::string > &Args) override