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