granicus.if.org Git - clang/blob - lib/ASTMatchers/Dynamic/Registry.cpp

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