xref: /llvm-project/clang/lib/ASTMatchers/Dynamic/Registry.cpp (revision fa1ea6977c37f4d6ac89907215dfa04b50f7ab8b)
1 //===- Registry.cpp - Matcher registry ------------------------------------===//
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 /// \file
11 /// Registry map populated at static initialization time.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #include "clang/ASTMatchers/Dynamic/Registry.h"
16 #include "Marshallers.h"
17 #include "clang/AST/ASTTypeTraits.h"
18 #include "clang/ASTMatchers/ASTMatchers.h"
19 #include "clang/ASTMatchers/Dynamic/Diagnostics.h"
20 #include "clang/ASTMatchers/Dynamic/VariantValue.h"
21 #include "llvm/ADT/Optional.h"
22 #include "llvm/ADT/STLExtras.h"
23 #include "llvm/ADT/StringMap.h"
24 #include "llvm/ADT/StringRef.h"
25 #include "llvm/Support/ManagedStatic.h"
26 #include "llvm/Support/raw_ostream.h"
27 #include <cassert>
28 #include <iterator>
29 #include <memory>
30 #include <set>
31 #include <string>
32 #include <utility>
33 #include <vector>
34 
35 using namespace clang::ast_type_traits;
36 
37 namespace clang {
38 namespace ast_matchers {
39 namespace dynamic {
40 
41 namespace {
42 
43 using internal::MatcherDescriptor;
44 
45 using ConstructorMap = llvm::StringMap<std::unique_ptr<const MatcherDescriptor>>;
46 
47 class RegistryMaps {
48 public:
49   RegistryMaps();
50   ~RegistryMaps();
51 
52   const ConstructorMap &constructors() const { return Constructors; }
53 
54 private:
55   void registerMatcher(StringRef MatcherName,
56                        std::unique_ptr<MatcherDescriptor> Callback);
57 
58   ConstructorMap Constructors;
59 };
60 
61 } // namespace
62 
63 void RegistryMaps::registerMatcher(
64     StringRef MatcherName, std::unique_ptr<MatcherDescriptor> Callback) {
65   assert(Constructors.find(MatcherName) == Constructors.end());
66   Constructors[MatcherName] = std::move(Callback);
67 }
68 
69 #define REGISTER_MATCHER(name)                                                 \
70   registerMatcher(#name, internal::makeMatcherAutoMarshall(                    \
71                              ::clang::ast_matchers::name, #name));
72 
73 #define REGISTER_MATCHER_OVERLOAD(name)                                        \
74   registerMatcher(#name,                                                       \
75       llvm::make_unique<internal::OverloadedMatcherDescriptor>(name##Callbacks))
76 
77 #define SPECIFIC_MATCHER_OVERLOAD(name, Id)                                    \
78   static_cast<::clang::ast_matchers::name##_Type##Id>(                         \
79       ::clang::ast_matchers::name)
80 
81 #define MATCHER_OVERLOAD_ENTRY(name, Id)                                       \
82         internal::makeMatcherAutoMarshall(SPECIFIC_MATCHER_OVERLOAD(name, Id), \
83                                           #name)
84 
85 #define REGISTER_OVERLOADED_2(name)                                            \
86   do {                                                                         \
87     std::unique_ptr<MatcherDescriptor> name##Callbacks[] = {                   \
88         MATCHER_OVERLOAD_ENTRY(name, 0),                                       \
89         MATCHER_OVERLOAD_ENTRY(name, 1)};                                      \
90     REGISTER_MATCHER_OVERLOAD(name);                                           \
91   } while (false)
92 
93 /// Generate a registry map with all the known matchers.
94 RegistryMaps::RegistryMaps() {
95   // TODO: Here is the list of the missing matchers, grouped by reason.
96   //
97   // Need Variant/Parser fixes:
98   // ofKind
99   //
100   // Polymorphic + argument overload:
101   // findAll
102   //
103   // Other:
104   // equalsNode
105 
106   REGISTER_OVERLOADED_2(callee);
107   REGISTER_OVERLOADED_2(hasPrefix);
108   REGISTER_OVERLOADED_2(hasType);
109   REGISTER_OVERLOADED_2(isDerivedFrom);
110   REGISTER_OVERLOADED_2(isSameOrDerivedFrom);
111   REGISTER_OVERLOADED_2(loc);
112   REGISTER_OVERLOADED_2(pointsTo);
113   REGISTER_OVERLOADED_2(references);
114   REGISTER_OVERLOADED_2(thisPointerType);
115 
116   std::unique_ptr<MatcherDescriptor> equalsCallbacks[] = {
117       MATCHER_OVERLOAD_ENTRY(equals, 0),
118       MATCHER_OVERLOAD_ENTRY(equals, 1),
119       MATCHER_OVERLOAD_ENTRY(equals, 2),
120   };
121   REGISTER_MATCHER_OVERLOAD(equals);
122 
123   REGISTER_MATCHER(accessSpecDecl);
124   REGISTER_MATCHER(addrLabelExpr);
125   REGISTER_MATCHER(alignOfExpr);
126   REGISTER_MATCHER(allOf);
127   REGISTER_MATCHER(anyOf);
128   REGISTER_MATCHER(anything);
129   REGISTER_MATCHER(argumentCountIs);
130   REGISTER_MATCHER(arraySubscriptExpr);
131   REGISTER_MATCHER(arrayType);
132   REGISTER_MATCHER(asmStmt);
133   REGISTER_MATCHER(asString);
134   REGISTER_MATCHER(atomicExpr);
135   REGISTER_MATCHER(atomicType);
136   REGISTER_MATCHER(autoType);
137   REGISTER_MATCHER(autoreleasePoolStmt)
138   REGISTER_MATCHER(binaryOperator);
139   REGISTER_MATCHER(binaryConditionalOperator);
140   REGISTER_MATCHER(blockDecl);
141   REGISTER_MATCHER(blockPointerType);
142   REGISTER_MATCHER(booleanType);
143   REGISTER_MATCHER(breakStmt);
144   REGISTER_MATCHER(builtinType);
145   REGISTER_MATCHER(callExpr);
146   REGISTER_MATCHER(caseStmt);
147   REGISTER_MATCHER(castExpr);
148   REGISTER_MATCHER(characterLiteral);
149   REGISTER_MATCHER(classTemplateDecl);
150   REGISTER_MATCHER(classTemplateSpecializationDecl);
151   REGISTER_MATCHER(complexType);
152   REGISTER_MATCHER(compoundLiteralExpr);
153   REGISTER_MATCHER(compoundStmt);
154   REGISTER_MATCHER(conditionalOperator);
155   REGISTER_MATCHER(constantArrayType);
156   REGISTER_MATCHER(containsDeclaration);
157   REGISTER_MATCHER(continueStmt);
158   REGISTER_MATCHER(cStyleCastExpr);
159   REGISTER_MATCHER(cudaKernelCallExpr);
160   REGISTER_MATCHER(cxxBindTemporaryExpr);
161   REGISTER_MATCHER(cxxBoolLiteral);
162   REGISTER_MATCHER(cxxCatchStmt);
163   REGISTER_MATCHER(cxxConstCastExpr);
164   REGISTER_MATCHER(cxxConstructExpr);
165   REGISTER_MATCHER(cxxConstructorDecl);
166   REGISTER_MATCHER(cxxConversionDecl);
167   REGISTER_MATCHER(cxxCtorInitializer);
168   REGISTER_MATCHER(cxxDefaultArgExpr);
169   REGISTER_MATCHER(cxxDeleteExpr);
170   REGISTER_MATCHER(cxxDestructorDecl);
171   REGISTER_MATCHER(cxxDynamicCastExpr);
172   REGISTER_MATCHER(cxxForRangeStmt);
173   REGISTER_MATCHER(cxxFunctionalCastExpr);
174   REGISTER_MATCHER(cxxMemberCallExpr);
175   REGISTER_MATCHER(cxxMethodDecl);
176   REGISTER_MATCHER(cxxNewExpr);
177   REGISTER_MATCHER(cxxNullPtrLiteralExpr);
178   REGISTER_MATCHER(cxxOperatorCallExpr);
179   REGISTER_MATCHER(cxxRecordDecl);
180   REGISTER_MATCHER(cxxReinterpretCastExpr);
181   REGISTER_MATCHER(cxxStaticCastExpr);
182   REGISTER_MATCHER(cxxStdInitializerListExpr);
183   REGISTER_MATCHER(cxxTemporaryObjectExpr);
184   REGISTER_MATCHER(cxxThisExpr);
185   REGISTER_MATCHER(cxxThrowExpr);
186   REGISTER_MATCHER(cxxTryStmt);
187   REGISTER_MATCHER(cxxUnresolvedConstructExpr);
188   REGISTER_MATCHER(decayedType);
189   REGISTER_MATCHER(decl);
190   REGISTER_MATCHER(declaratorDecl);
191   REGISTER_MATCHER(decltypeType);
192   REGISTER_MATCHER(declCountIs);
193   REGISTER_MATCHER(declRefExpr);
194   REGISTER_MATCHER(declStmt);
195   REGISTER_MATCHER(defaultStmt);
196   REGISTER_MATCHER(dependentSizedArrayType);
197   REGISTER_MATCHER(designatedInitExpr);
198   REGISTER_MATCHER(designatorCountIs);
199   REGISTER_MATCHER(doStmt);
200   REGISTER_MATCHER(eachOf);
201   REGISTER_MATCHER(elaboratedType);
202   REGISTER_MATCHER(enumConstantDecl);
203   REGISTER_MATCHER(enumDecl);
204   REGISTER_MATCHER(enumType);
205   REGISTER_MATCHER(equalsBoundNode);
206   REGISTER_MATCHER(equalsIntegralValue);
207   REGISTER_MATCHER(explicitCastExpr);
208   REGISTER_MATCHER(expr);
209   REGISTER_MATCHER(exprWithCleanups);
210   REGISTER_MATCHER(fieldDecl);
211   REGISTER_MATCHER(floatLiteral);
212   REGISTER_MATCHER(forEach);
213   REGISTER_MATCHER(forEachArgumentWithParam);
214   REGISTER_MATCHER(forEachConstructorInitializer);
215   REGISTER_MATCHER(forEachDescendant);
216   REGISTER_MATCHER(forEachOverridden);
217   REGISTER_MATCHER(forEachSwitchCase);
218   REGISTER_MATCHER(forField);
219   REGISTER_MATCHER(forFunction);
220   REGISTER_MATCHER(forStmt);
221   REGISTER_MATCHER(friendDecl);
222   REGISTER_MATCHER(functionDecl);
223   REGISTER_MATCHER(functionProtoType);
224   REGISTER_MATCHER(functionTemplateDecl);
225   REGISTER_MATCHER(functionType);
226   REGISTER_MATCHER(gnuNullExpr);
227   REGISTER_MATCHER(gotoStmt);
228   REGISTER_MATCHER(has);
229   REGISTER_MATCHER(hasAncestor);
230   REGISTER_MATCHER(hasAnyArgument);
231   REGISTER_MATCHER(hasAnyConstructorInitializer);
232   REGISTER_MATCHER(hasAnyDeclaration);
233   REGISTER_MATCHER(hasAnyName);
234   REGISTER_MATCHER(hasAnyParameter);
235   REGISTER_MATCHER(hasAnySubstatement);
236   REGISTER_MATCHER(hasAnyTemplateArgument);
237   REGISTER_MATCHER(hasAnyUsingShadowDecl);
238   REGISTER_MATCHER(hasArgument);
239   REGISTER_MATCHER(hasArgumentOfType);
240   REGISTER_MATCHER(hasArraySize);
241   REGISTER_MATCHER(hasAttr);
242   REGISTER_MATCHER(hasAutomaticStorageDuration);
243   REGISTER_MATCHER(hasBase);
244   REGISTER_MATCHER(hasBitWidth);
245   REGISTER_MATCHER(hasBody);
246   REGISTER_MATCHER(hasCanonicalType);
247   REGISTER_MATCHER(hasCaseConstant);
248   REGISTER_MATCHER(hasCastKind);
249   REGISTER_MATCHER(hasCondition);
250   REGISTER_MATCHER(hasConditionVariableStatement);
251   REGISTER_MATCHER(hasDecayedType);
252   REGISTER_MATCHER(hasDeclaration);
253   REGISTER_MATCHER(hasDeclContext);
254   REGISTER_MATCHER(hasDeducedType);
255   REGISTER_MATCHER(hasDefaultArgument);
256   REGISTER_MATCHER(hasDefinition);
257   REGISTER_MATCHER(hasDescendant);
258   REGISTER_MATCHER(hasDestinationType);
259   REGISTER_MATCHER(hasDynamicExceptionSpec);
260   REGISTER_MATCHER(hasEitherOperand);
261   REGISTER_MATCHER(hasElementType);
262   REGISTER_MATCHER(hasElse);
263   REGISTER_MATCHER(hasExternalFormalLinkage);
264   REGISTER_MATCHER(hasFalseExpression);
265   REGISTER_MATCHER(hasGlobalStorage);
266   REGISTER_MATCHER(hasImplicitDestinationType);
267   REGISTER_MATCHER(hasInClassInitializer);
268   REGISTER_MATCHER(hasIncrement);
269   REGISTER_MATCHER(hasIndex);
270   REGISTER_MATCHER(hasInitializer);
271   REGISTER_MATCHER(hasKeywordSelector);
272   REGISTER_MATCHER(hasLHS);
273   REGISTER_MATCHER(hasLocalQualifiers);
274   REGISTER_MATCHER(hasLocalStorage);
275   REGISTER_MATCHER(hasLoopInit);
276   REGISTER_MATCHER(hasLoopVariable);
277   REGISTER_MATCHER(hasMethod);
278   REGISTER_MATCHER(hasName);
279   REGISTER_MATCHER(hasNullSelector);
280   REGISTER_MATCHER(hasObjectExpression);
281   REGISTER_MATCHER(hasOperatorName);
282   REGISTER_MATCHER(hasOverloadedOperatorName);
283   REGISTER_MATCHER(hasParameter);
284   REGISTER_MATCHER(hasParent);
285   REGISTER_MATCHER(hasQualifier);
286   REGISTER_MATCHER(hasRangeInit);
287   REGISTER_MATCHER(hasReceiver);
288   REGISTER_MATCHER(hasReceiverType);
289   REGISTER_MATCHER(hasReplacementType);
290   REGISTER_MATCHER(hasReturnValue);
291   REGISTER_MATCHER(hasRHS);
292   REGISTER_MATCHER(hasSelector);
293   REGISTER_MATCHER(hasAnySelector);
294   REGISTER_MATCHER(hasSingleDecl);
295   REGISTER_MATCHER(hasSize);
296   REGISTER_MATCHER(hasSizeExpr);
297   REGISTER_MATCHER(hasSourceExpression);
298   REGISTER_MATCHER(hasStaticStorageDuration);
299   REGISTER_MATCHER(hasSyntacticForm);
300   REGISTER_MATCHER(hasTargetDecl);
301   REGISTER_MATCHER(hasTemplateArgument);
302   REGISTER_MATCHER(hasThen);
303   REGISTER_MATCHER(hasThreadStorageDuration);
304   REGISTER_MATCHER(hasTrailingReturn);
305   REGISTER_MATCHER(hasTrueExpression);
306   REGISTER_MATCHER(hasTypeLoc);
307   REGISTER_MATCHER(hasUnaryOperand);
308   REGISTER_MATCHER(hasUnarySelector);
309   REGISTER_MATCHER(hasUnderlyingDecl);
310   REGISTER_MATCHER(hasUnqualifiedDesugaredType);
311   REGISTER_MATCHER(hasValueType);
312   REGISTER_MATCHER(ifStmt);
313   REGISTER_MATCHER(ignoringImplicit);
314   REGISTER_MATCHER(ignoringImpCasts);
315   REGISTER_MATCHER(ignoringParenCasts);
316   REGISTER_MATCHER(ignoringParenImpCasts);
317   REGISTER_MATCHER(ignoringParens);
318   REGISTER_MATCHER(implicitCastExpr);
319   REGISTER_MATCHER(implicitValueInitExpr);
320   REGISTER_MATCHER(incompleteArrayType);
321   REGISTER_MATCHER(initListExpr);
322   REGISTER_MATCHER(injectedClassNameType);
323   REGISTER_MATCHER(innerType);
324   REGISTER_MATCHER(integerLiteral);
325   REGISTER_MATCHER(isAnonymous);
326   REGISTER_MATCHER(isAnyCharacter);
327   REGISTER_MATCHER(isAnyPointer);
328   REGISTER_MATCHER(isArray);
329   REGISTER_MATCHER(isArrow);
330   REGISTER_MATCHER(isAssignmentOperator);
331   REGISTER_MATCHER(isBaseInitializer);
332   REGISTER_MATCHER(isBitField);
333   REGISTER_MATCHER(isCatchAll);
334   REGISTER_MATCHER(isClass);
335   REGISTER_MATCHER(isConst);
336   REGISTER_MATCHER(isConstexpr);
337   REGISTER_MATCHER(isConstQualified);
338   REGISTER_MATCHER(isCopyAssignmentOperator);
339   REGISTER_MATCHER(isCopyConstructor);
340   REGISTER_MATCHER(isDefaultConstructor);
341   REGISTER_MATCHER(isDefaulted);
342   REGISTER_MATCHER(isDefinition);
343   REGISTER_MATCHER(isDeleted);
344   REGISTER_MATCHER(isExceptionVariable);
345   REGISTER_MATCHER(isExplicit);
346   REGISTER_MATCHER(isExplicitTemplateSpecialization);
347   REGISTER_MATCHER(isExpr);
348   REGISTER_MATCHER(isExternC);
349   REGISTER_MATCHER(isFinal);
350   REGISTER_MATCHER(isInline);
351   REGISTER_MATCHER(isImplicit);
352   REGISTER_MATCHER(isExpansionInFileMatching);
353   REGISTER_MATCHER(isExpansionInMainFile);
354   REGISTER_MATCHER(isInstanceMessage);
355   REGISTER_MATCHER(isInstantiated);
356   REGISTER_MATCHER(isExpansionInSystemHeader);
357   REGISTER_MATCHER(isInteger);
358   REGISTER_MATCHER(isIntegral);
359   REGISTER_MATCHER(isInTemplateInstantiation);
360   REGISTER_MATCHER(isLambda);
361   REGISTER_MATCHER(isListInitialization);
362   REGISTER_MATCHER(isMemberInitializer);
363   REGISTER_MATCHER(isMoveAssignmentOperator);
364   REGISTER_MATCHER(isMoveConstructor);
365   REGISTER_MATCHER(isNoReturn);
366   REGISTER_MATCHER(isNoThrow);
367   REGISTER_MATCHER(isOverride);
368   REGISTER_MATCHER(isPrivate);
369   REGISTER_MATCHER(isProtected);
370   REGISTER_MATCHER(isPublic);
371   REGISTER_MATCHER(isPure);
372   REGISTER_MATCHER(isScoped);
373   REGISTER_MATCHER(isSignedInteger);
374   REGISTER_MATCHER(isStaticStorageClass);
375   REGISTER_MATCHER(isStruct);
376   REGISTER_MATCHER(isTemplateInstantiation);
377   REGISTER_MATCHER(isUnion);
378   REGISTER_MATCHER(isUnsignedInteger);
379   REGISTER_MATCHER(isVariadic);
380   REGISTER_MATCHER(isVirtual);
381   REGISTER_MATCHER(isVirtualAsWritten);
382   REGISTER_MATCHER(isVolatileQualified);
383   REGISTER_MATCHER(isWritten);
384   REGISTER_MATCHER(labelDecl);
385   REGISTER_MATCHER(labelStmt);
386   REGISTER_MATCHER(lambdaExpr);
387   REGISTER_MATCHER(linkageSpecDecl);
388   REGISTER_MATCHER(lValueReferenceType);
389   REGISTER_MATCHER(matchesName);
390   REGISTER_MATCHER(matchesSelector);
391   REGISTER_MATCHER(materializeTemporaryExpr);
392   REGISTER_MATCHER(member);
393   REGISTER_MATCHER(memberExpr);
394   REGISTER_MATCHER(memberPointerType);
395   REGISTER_MATCHER(namedDecl);
396   REGISTER_MATCHER(namespaceAliasDecl);
397   REGISTER_MATCHER(namespaceDecl);
398   REGISTER_MATCHER(namesType);
399   REGISTER_MATCHER(nestedNameSpecifier);
400   REGISTER_MATCHER(nestedNameSpecifierLoc);
401   REGISTER_MATCHER(nullPointerConstant);
402   REGISTER_MATCHER(nullStmt);
403   REGISTER_MATCHER(numSelectorArgs);
404   REGISTER_MATCHER(ofClass);
405   REGISTER_MATCHER(objcCatchStmt);
406   REGISTER_MATCHER(objcCategoryDecl);
407   REGISTER_MATCHER(objcCategoryImplDecl);
408   REGISTER_MATCHER(objcFinallyStmt);
409   REGISTER_MATCHER(objcImplementationDecl);
410   REGISTER_MATCHER(objcInterfaceDecl);
411   REGISTER_MATCHER(objcIvarDecl);
412   REGISTER_MATCHER(objcMessageExpr);
413   REGISTER_MATCHER(objcMethodDecl);
414   REGISTER_MATCHER(objcObjectPointerType);
415   REGISTER_MATCHER(objcPropertyDecl);
416   REGISTER_MATCHER(objcProtocolDecl);
417   REGISTER_MATCHER(objcThrowStmt);
418   REGISTER_MATCHER(objcTryStmt);
419   REGISTER_MATCHER(on);
420   REGISTER_MATCHER(onImplicitObjectArgument);
421   REGISTER_MATCHER(opaqueValueExpr);
422   REGISTER_MATCHER(parameterCountIs);
423   REGISTER_MATCHER(parenExpr);
424   REGISTER_MATCHER(parenListExpr);
425   REGISTER_MATCHER(parenType);
426   REGISTER_MATCHER(parmVarDecl);
427   REGISTER_MATCHER(pointee);
428   REGISTER_MATCHER(pointerType);
429   REGISTER_MATCHER(predefinedExpr);
430   REGISTER_MATCHER(qualType);
431   REGISTER_MATCHER(realFloatingPointType);
432   REGISTER_MATCHER(recordDecl);
433   REGISTER_MATCHER(recordType);
434   REGISTER_MATCHER(referenceType);
435   REGISTER_MATCHER(refersToDeclaration);
436   REGISTER_MATCHER(refersToIntegralType);
437   REGISTER_MATCHER(refersToType);
438   REGISTER_MATCHER(requiresZeroInitialization);
439   REGISTER_MATCHER(returns);
440   REGISTER_MATCHER(returnStmt);
441   REGISTER_MATCHER(rValueReferenceType);
442   REGISTER_MATCHER(sizeOfExpr);
443   REGISTER_MATCHER(specifiesNamespace);
444   REGISTER_MATCHER(specifiesType);
445   REGISTER_MATCHER(specifiesTypeLoc);
446   REGISTER_MATCHER(statementCountIs);
447   REGISTER_MATCHER(staticAssertDecl);
448   REGISTER_MATCHER(stmt);
449   REGISTER_MATCHER(stmtExpr);
450   REGISTER_MATCHER(stringLiteral);
451   REGISTER_MATCHER(substNonTypeTemplateParmExpr);
452   REGISTER_MATCHER(substTemplateTypeParmType);
453   REGISTER_MATCHER(switchCase);
454   REGISTER_MATCHER(switchStmt);
455   REGISTER_MATCHER(templateArgument);
456   REGISTER_MATCHER(templateName);
457   REGISTER_MATCHER(templateArgumentCountIs);
458   REGISTER_MATCHER(templateSpecializationType);
459   REGISTER_MATCHER(templateTypeParmDecl);
460   REGISTER_MATCHER(templateTypeParmType);
461   REGISTER_MATCHER(throughUsingDecl);
462   REGISTER_MATCHER(to);
463   REGISTER_MATCHER(translationUnitDecl);
464   REGISTER_MATCHER(type);
465   REGISTER_MATCHER(typedefDecl);
466   REGISTER_MATCHER(typedefNameDecl);
467   REGISTER_MATCHER(typedefType);
468   REGISTER_MATCHER(typeAliasDecl);
469   REGISTER_MATCHER(typeAliasTemplateDecl);
470   REGISTER_MATCHER(typeLoc);
471   REGISTER_MATCHER(unaryExprOrTypeTraitExpr);
472   REGISTER_MATCHER(unaryOperator);
473   REGISTER_MATCHER(unaryTransformType);
474   REGISTER_MATCHER(unless);
475   REGISTER_MATCHER(unresolvedLookupExpr);
476   REGISTER_MATCHER(unresolvedUsingTypenameDecl);
477   REGISTER_MATCHER(unresolvedUsingValueDecl);
478   REGISTER_MATCHER(userDefinedLiteral);
479   REGISTER_MATCHER(usingDecl);
480   REGISTER_MATCHER(usingDirectiveDecl);
481   REGISTER_MATCHER(valueDecl);
482   REGISTER_MATCHER(varDecl);
483   REGISTER_MATCHER(variableArrayType);
484   REGISTER_MATCHER(voidType);
485   REGISTER_MATCHER(whileStmt);
486   REGISTER_MATCHER(withInitializer);
487 }
488 
489 RegistryMaps::~RegistryMaps() = default;
490 
491 static llvm::ManagedStatic<RegistryMaps> RegistryData;
492 
493 // static
494 llvm::Optional<MatcherCtor> Registry::lookupMatcherCtor(StringRef MatcherName) {
495   auto it = RegistryData->constructors().find(MatcherName);
496   return it == RegistryData->constructors().end()
497              ? llvm::Optional<MatcherCtor>()
498              : it->second.get();
499 }
500 
501 static llvm::raw_ostream &operator<<(llvm::raw_ostream &OS,
502                                      const std::set<ASTNodeKind> &KS) {
503   unsigned Count = 0;
504   for (std::set<ASTNodeKind>::const_iterator I = KS.begin(), E = KS.end();
505        I != E; ++I) {
506     if (I != KS.begin())
507       OS << "|";
508     if (Count++ == 3) {
509       OS << "...";
510       break;
511     }
512     OS << *I;
513   }
514   return OS;
515 }
516 
517 std::vector<ArgKind> Registry::getAcceptedCompletionTypes(
518     ArrayRef<std::pair<MatcherCtor, unsigned>> Context) {
519   ASTNodeKind InitialTypes[] = {
520       ASTNodeKind::getFromNodeKind<Decl>(),
521       ASTNodeKind::getFromNodeKind<QualType>(),
522       ASTNodeKind::getFromNodeKind<Type>(),
523       ASTNodeKind::getFromNodeKind<Stmt>(),
524       ASTNodeKind::getFromNodeKind<NestedNameSpecifier>(),
525       ASTNodeKind::getFromNodeKind<NestedNameSpecifierLoc>(),
526       ASTNodeKind::getFromNodeKind<TypeLoc>()};
527 
528   // Starting with the above seed of acceptable top-level matcher types, compute
529   // the acceptable type set for the argument indicated by each context element.
530   std::set<ArgKind> TypeSet(std::begin(InitialTypes), std::end(InitialTypes));
531   for (const auto &CtxEntry : Context) {
532     MatcherCtor Ctor = CtxEntry.first;
533     unsigned ArgNumber = CtxEntry.second;
534     std::vector<ArgKind> NextTypeSet;
535     for (const ArgKind &Kind : TypeSet) {
536       if (Kind.getArgKind() == Kind.AK_Matcher &&
537           Ctor->isConvertibleTo(Kind.getMatcherKind()) &&
538           (Ctor->isVariadic() || ArgNumber < Ctor->getNumArgs()))
539         Ctor->getArgKinds(Kind.getMatcherKind(), ArgNumber, NextTypeSet);
540     }
541     TypeSet.clear();
542     TypeSet.insert(NextTypeSet.begin(), NextTypeSet.end());
543   }
544   return std::vector<ArgKind>(TypeSet.begin(), TypeSet.end());
545 }
546 
547 std::vector<MatcherCompletion>
548 Registry::getMatcherCompletions(ArrayRef<ArgKind> AcceptedTypes) {
549   std::vector<MatcherCompletion> Completions;
550 
551   // Search the registry for acceptable matchers.
552   for (const auto &M : RegistryData->constructors()) {
553     const MatcherDescriptor& Matcher = *M.getValue();
554     StringRef Name = M.getKey();
555 
556     std::set<ASTNodeKind> RetKinds;
557     unsigned NumArgs = Matcher.isVariadic() ? 1 : Matcher.getNumArgs();
558     bool IsPolymorphic = Matcher.isPolymorphic();
559     std::vector<std::vector<ArgKind>> ArgsKinds(NumArgs);
560     unsigned MaxSpecificity = 0;
561     for (const ArgKind& Kind : AcceptedTypes) {
562       if (Kind.getArgKind() != Kind.AK_Matcher)
563         continue;
564       unsigned Specificity;
565       ASTNodeKind LeastDerivedKind;
566       if (Matcher.isConvertibleTo(Kind.getMatcherKind(), &Specificity,
567                                   &LeastDerivedKind)) {
568         if (MaxSpecificity < Specificity)
569           MaxSpecificity = Specificity;
570         RetKinds.insert(LeastDerivedKind);
571         for (unsigned Arg = 0; Arg != NumArgs; ++Arg)
572           Matcher.getArgKinds(Kind.getMatcherKind(), Arg, ArgsKinds[Arg]);
573         if (IsPolymorphic)
574           break;
575       }
576     }
577 
578     if (!RetKinds.empty() && MaxSpecificity > 0) {
579       std::string Decl;
580       llvm::raw_string_ostream OS(Decl);
581 
582       if (IsPolymorphic) {
583         OS << "Matcher<T> " << Name << "(Matcher<T>";
584       } else {
585         OS << "Matcher<" << RetKinds << "> " << Name << "(";
586         for (const std::vector<ArgKind> &Arg : ArgsKinds) {
587           if (&Arg != &ArgsKinds[0])
588             OS << ", ";
589 
590           bool FirstArgKind = true;
591           std::set<ASTNodeKind> MatcherKinds;
592           // Two steps. First all non-matchers, then matchers only.
593           for (const ArgKind &AK : Arg) {
594             if (AK.getArgKind() == ArgKind::AK_Matcher) {
595               MatcherKinds.insert(AK.getMatcherKind());
596             } else {
597               if (!FirstArgKind) OS << "|";
598               FirstArgKind = false;
599               OS << AK.asString();
600             }
601           }
602           if (!MatcherKinds.empty()) {
603             if (!FirstArgKind) OS << "|";
604             OS << "Matcher<" << MatcherKinds << ">";
605           }
606         }
607       }
608       if (Matcher.isVariadic())
609         OS << "...";
610       OS << ")";
611 
612       std::string TypedText = Name;
613       TypedText += "(";
614       if (ArgsKinds.empty())
615         TypedText += ")";
616       else if (ArgsKinds[0][0].getArgKind() == ArgKind::AK_String)
617         TypedText += "\"";
618 
619       Completions.emplace_back(TypedText, OS.str(), MaxSpecificity);
620     }
621   }
622 
623   return Completions;
624 }
625 
626 VariantMatcher Registry::constructMatcher(MatcherCtor Ctor,
627                                           SourceRange NameRange,
628                                           ArrayRef<ParserValue> Args,
629                                           Diagnostics *Error) {
630   return Ctor->create(NameRange, Args, Error);
631 }
632 
633 VariantMatcher Registry::constructBoundMatcher(MatcherCtor Ctor,
634                                                SourceRange NameRange,
635                                                StringRef BindID,
636                                                ArrayRef<ParserValue> Args,
637                                                Diagnostics *Error) {
638   VariantMatcher Out = constructMatcher(Ctor, NameRange, Args, Error);
639   if (Out.isNull()) return Out;
640 
641   llvm::Optional<DynTypedMatcher> Result = Out.getSingleMatcher();
642   if (Result.hasValue()) {
643     llvm::Optional<DynTypedMatcher> Bound = Result->tryBind(BindID);
644     if (Bound.hasValue()) {
645       return VariantMatcher::SingleMatcher(*Bound);
646     }
647   }
648   Error->addError(NameRange, Error->ET_RegistryNotBindable);
649   return VariantMatcher();
650 }
651 
652 } // namespace dynamic
653 } // namespace ast_matchers
654 } // namespace clang
655