clang-tools  8.0.0
Query.cpp
Go to the documentation of this file.
1 //===---- Query.cpp - clang-query query -----------------------------------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "Query.h"
11 #include "QuerySession.h"
12 #include "clang/ASTMatchers/ASTMatchFinder.h"
13 #include "clang/Frontend/ASTUnit.h"
14 #include "clang/Frontend/TextDiagnostic.h"
15 #include "llvm/Support/raw_ostream.h"
16 
17 using namespace clang::ast_matchers;
18 using namespace clang::ast_matchers::dynamic;
19 
20 namespace clang {
21 namespace query {
22 
23 Query::~Query() {}
24 
25 bool InvalidQuery::run(llvm::raw_ostream &OS, QuerySession &QS) const {
26  OS << ErrStr << "\n";
27  return false;
28 }
29 
30 bool NoOpQuery::run(llvm::raw_ostream &OS, QuerySession &QS) const {
31  return true;
32 }
33 
34 bool HelpQuery::run(llvm::raw_ostream &OS, QuerySession &QS) const {
35  OS << "Available commands:\n\n"
36  " match MATCHER, m MATCHER "
37  "Match the loaded ASTs against the given matcher.\n"
38  " let NAME MATCHER, l NAME MATCHER "
39  "Give a matcher expression a name, to be used later\n"
40  " "
41  "as part of other expressions.\n"
42  " set bind-root (true|false) "
43  "Set whether to bind the root matcher to \"root\".\n"
44  " set print-matcher (true|false) "
45  "Set whether to print the current matcher,\n"
46  " set output <feature> "
47  "Set whether to output only <feature> content.\n"
48  " enable output <feature> "
49  "Enable <feature> content non-exclusively.\n"
50  " disable output <feature> "
51  "Disable <feature> content non-exclusively.\n"
52  " quit, q "
53  "Terminates the query session.\n\n"
54  "Several commands accept a <feature> parameter. The available features "
55  "are:\n\n"
56  " print "
57  "Pretty-print bound nodes.\n"
58  " diag "
59  "Diagnostic location for bound nodes.\n"
60  " detailed-ast "
61  "Detailed AST output for bound nodes.\n"
62  " dump "
63  "Detailed AST output for bound nodes (alias of detailed-ast).\n\n";
64  return true;
65 }
66 
67 bool QuitQuery::run(llvm::raw_ostream &OS, QuerySession &QS) const {
68  QS.Terminate = true;
69  return true;
70 }
71 
72 namespace {
73 
74 struct CollectBoundNodes : MatchFinder::MatchCallback {
75  std::vector<BoundNodes> &Bindings;
76  CollectBoundNodes(std::vector<BoundNodes> &Bindings) : Bindings(Bindings) {}
77  void run(const MatchFinder::MatchResult &Result) override {
78  Bindings.push_back(Result.Nodes);
79  }
80 };
81 
82 } // namespace
83 
84 bool MatchQuery::run(llvm::raw_ostream &OS, QuerySession &QS) const {
85  unsigned MatchCount = 0;
86 
87  for (auto &AST : QS.ASTs) {
88  MatchFinder Finder;
89  std::vector<BoundNodes> Matches;
90  DynTypedMatcher MaybeBoundMatcher = Matcher;
91  if (QS.BindRoot) {
92  llvm::Optional<DynTypedMatcher> M = Matcher.tryBind("root");
93  if (M)
94  MaybeBoundMatcher = *M;
95  }
96  CollectBoundNodes Collect(Matches);
97  if (!Finder.addDynamicMatcher(MaybeBoundMatcher, &Collect)) {
98  OS << "Not a valid top-level matcher.\n";
99  return false;
100  }
101  Finder.matchAST(AST->getASTContext());
102 
103  if (QS.PrintMatcher) {
104  std::string prefixText = "Matcher: ";
105  OS << "\n " << prefixText << Source << "\n";
106  OS << " " << std::string(prefixText.size() + Source.size(), '=') << '\n';
107  }
108 
109  for (auto MI = Matches.begin(), ME = Matches.end(); MI != ME; ++MI) {
110  OS << "\nMatch #" << ++MatchCount << ":\n\n";
111 
112  for (auto BI = MI->getMap().begin(), BE = MI->getMap().end(); BI != BE;
113  ++BI) {
114  if (QS.DiagOutput) {
115  clang::SourceRange R = BI->second.getSourceRange();
116  if (R.isValid()) {
117  TextDiagnostic TD(OS, AST->getASTContext().getLangOpts(),
118  &AST->getDiagnostics().getDiagnosticOptions());
119  TD.emitDiagnostic(
120  FullSourceLoc(R.getBegin(), AST->getSourceManager()),
121  DiagnosticsEngine::Note, "\"" + BI->first + "\" binds here",
122  CharSourceRange::getTokenRange(R), None);
123  }
124  }
125  if (QS.PrintOutput) {
126  OS << "Binding for \"" << BI->first << "\":\n";
127  BI->second.print(OS, AST->getASTContext().getPrintingPolicy());
128  OS << "\n";
129  }
130  if (QS.DetailedASTOutput) {
131  OS << "Binding for \"" << BI->first << "\":\n";
132  BI->second.dump(OS, AST->getSourceManager());
133  OS << "\n";
134  }
135  }
136 
137  if (MI->getMap().empty())
138  OS << "No bindings.\n";
139  }
140  }
141 
142  OS << MatchCount << (MatchCount == 1 ? " match.\n" : " matches.\n");
143  return true;
144 }
145 
146 bool LetQuery::run(llvm::raw_ostream &OS, QuerySession &QS) const {
147  if (Value) {
148  QS.NamedValues[Name] = Value;
149  } else {
150  QS.NamedValues.erase(Name);
151  }
152  return true;
153 }
154 
155 #ifndef _MSC_VER
158 #endif
159 
160 } // namespace query
161 } // namespace clang
Represents the state for a particular clang-query session.
Definition: QuerySession.h:24
llvm::ArrayRef< std::unique_ptr< ASTUnit > > ASTs
Definition: QuerySession.h:31
llvm::StringMap< ast_matchers::dynamic::VariantValue > NamedValues
Definition: QuerySession.h:40
static constexpr llvm::StringLiteral Name
llvm::Optional< llvm::Expected< tooling::AtomicChanges > > Result
===– Representation.cpp - ClangDoc Representation --------—*- C++ -*-===//
std::vector< BoundNodes > & Bindings
Definition: Query.cpp:75