1 //===-- FindAllSymbolsAction.cpp - find all symbols action --------*- C++ -*-===//
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 "FindAllSymbolsAction.h"
10 #include "FindAllMacros.h"
11 #include "clang/Lex/PPCallbacks.h"
12 #include "clang/Lex/Preprocessor.h"
13 #include "llvm/ADT/STLExtras.h"
14 
15 namespace clang {
16 namespace find_all_symbols {
17 
FindAllSymbolsAction(SymbolReporter * Reporter,const HeaderMapCollector::RegexHeaderMap * RegexHeaderMap)18 FindAllSymbolsAction::FindAllSymbolsAction(
19     SymbolReporter *Reporter,
20     const HeaderMapCollector::RegexHeaderMap *RegexHeaderMap)
21     : Reporter(Reporter), Collector(RegexHeaderMap), Handler(&Collector),
22       Matcher(Reporter, &Collector) {
23   Matcher.registerMatchers(&MatchFinder);
24 }
25 
26 std::unique_ptr<ASTConsumer>
CreateASTConsumer(CompilerInstance & Compiler,StringRef InFile)27 FindAllSymbolsAction::CreateASTConsumer(CompilerInstance &Compiler,
28                                         StringRef InFile) {
29   Compiler.getPreprocessor().addCommentHandler(&Handler);
30   Compiler.getPreprocessor().addPPCallbacks(std::make_unique<FindAllMacros>(
31       Reporter, &Compiler.getSourceManager(), &Collector));
32   return MatchFinder.newASTConsumer();
33 }
34 
35 } // namespace find_all_symbols
36 } // namespace clang
37