ConsumerAction.cpp

class Consumer : public clang::ASTConsumer {
public:
  explicit Consumer(bool RewriteOption, clang::Rewriter &Rewriter)
      : Checker(RewriteOption, Rewriter) {
    using namespace clang::ast_matchers;
    const auto CXXMethodMatcher =
        cxxMethodDecl(unless(isExpansionInSystemHeader())).bind("target");
    Finder.addMatcher(CXXMethodMatcher, &Checker);
  }

  void HandleTranslationUnit(clang::ASTContext &Context) override {
    Finder.matchAST(Context);
  }

private:
  clang::ast_matchers::MatchFinder Finder;
  Checker Checker;
};

class Action : public clang::ASTFrontendAction {
public:
  using ASTConsumerPointer = std::unique_ptr<clang::ASTConsumer>;

  explicit Action(bool RewriteOption) : RewriteOption(RewriteOption) {}

  ASTConsumerPointer CreateASTConsumer(clang::CompilerInstance &Compiler,
                                       llvm::StringRef Filename) override {
    Rewriter.setSourceMgr(Compiler.getSourceManager(), Compiler.getLangOpts());
    return std::make_unique<Consumer>(RewriteOption, Rewriter);
  }

  bool BeginSourceFileAction(clang::CompilerInstance &Compiler) override {
    llvm::errs() << "Processing " << getCurrentFile() << "\n\n";
    return true;
  }

  void EndSourceFileAction() override {
    if (!RewriteOption)
      return;
    const auto File = Rewriter.getSourceMgr().getMainFileID();
    Rewriter.getEditBuffer(File).write(llvm::outs());
  }

private:
  bool RewriteOption;
  clang::Rewriter Rewriter;
};
} // namespace UseOverride