Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===- Registry.h - Matcher registry ----------------------------*- 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 | /// \file |
||
| 10 | /// Registry of all known matchers. |
||
| 11 | /// |
||
| 12 | /// The registry provides a generic interface to construct any matcher by name. |
||
| 13 | // |
||
| 14 | //===----------------------------------------------------------------------===// |
||
| 15 | |||
| 16 | #ifndef LLVM_CLANG_ASTMATCHERS_DYNAMIC_REGISTRY_H |
||
| 17 | #define LLVM_CLANG_ASTMATCHERS_DYNAMIC_REGISTRY_H |
||
| 18 | |||
| 19 | #include "clang/ASTMatchers/Dynamic/Diagnostics.h" |
||
| 20 | #include "clang/ASTMatchers/Dynamic/VariantValue.h" |
||
| 21 | #include "llvm/ADT/ArrayRef.h" |
||
| 22 | #include "llvm/ADT/StringRef.h" |
||
| 23 | #include <optional> |
||
| 24 | #include <string> |
||
| 25 | #include <utility> |
||
| 26 | #include <vector> |
||
| 27 | |||
| 28 | namespace clang { |
||
| 29 | namespace ast_matchers { |
||
| 30 | namespace dynamic { |
||
| 31 | |||
| 32 | namespace internal { |
||
| 33 | |||
| 34 | class MatcherDescriptor; |
||
| 35 | |||
| 36 | /// A smart (owning) pointer for MatcherDescriptor. We can't use unique_ptr |
||
| 37 | /// because MatcherDescriptor is forward declared |
||
| 38 | class MatcherDescriptorPtr { |
||
| 39 | public: |
||
| 40 | explicit MatcherDescriptorPtr(MatcherDescriptor *); |
||
| 41 | ~MatcherDescriptorPtr(); |
||
| 42 | MatcherDescriptorPtr(MatcherDescriptorPtr &&) = default; |
||
| 43 | MatcherDescriptorPtr &operator=(MatcherDescriptorPtr &&) = default; |
||
| 44 | MatcherDescriptorPtr(const MatcherDescriptorPtr &) = delete; |
||
| 45 | MatcherDescriptorPtr &operator=(const MatcherDescriptorPtr &) = delete; |
||
| 46 | |||
| 47 | MatcherDescriptor *get() { return Ptr; } |
||
| 48 | |||
| 49 | private: |
||
| 50 | MatcherDescriptor *Ptr; |
||
| 51 | }; |
||
| 52 | |||
| 53 | } // namespace internal |
||
| 54 | |||
| 55 | using MatcherCtor = const internal::MatcherDescriptor *; |
||
| 56 | |||
| 57 | struct MatcherCompletion { |
||
| 58 | MatcherCompletion() = default; |
||
| 59 | MatcherCompletion(StringRef TypedText, StringRef MatcherDecl, |
||
| 60 | unsigned Specificity) |
||
| 61 | : TypedText(TypedText), MatcherDecl(MatcherDecl), |
||
| 62 | Specificity(Specificity) {} |
||
| 63 | |||
| 64 | bool operator==(const MatcherCompletion &Other) const { |
||
| 65 | return TypedText == Other.TypedText && MatcherDecl == Other.MatcherDecl; |
||
| 66 | } |
||
| 67 | |||
| 68 | /// The text to type to select this matcher. |
||
| 69 | std::string TypedText; |
||
| 70 | |||
| 71 | /// The "declaration" of the matcher, with type information. |
||
| 72 | std::string MatcherDecl; |
||
| 73 | |||
| 74 | /// Value corresponding to the "specificity" of the converted matcher. |
||
| 75 | /// |
||
| 76 | /// Zero specificity indicates that this conversion would produce a trivial |
||
| 77 | /// matcher that will either always or never match. |
||
| 78 | /// Such matchers are excluded from code completion results. |
||
| 79 | unsigned Specificity; |
||
| 80 | }; |
||
| 81 | |||
| 82 | class Registry { |
||
| 83 | public: |
||
| 84 | Registry() = delete; |
||
| 85 | |||
| 86 | static ASTNodeKind nodeMatcherType(MatcherCtor); |
||
| 87 | |||
| 88 | static bool isBuilderMatcher(MatcherCtor Ctor); |
||
| 89 | |||
| 90 | static internal::MatcherDescriptorPtr |
||
| 91 | buildMatcherCtor(MatcherCtor, SourceRange NameRange, |
||
| 92 | ArrayRef<ParserValue> Args, Diagnostics *Error); |
||
| 93 | |||
| 94 | /// Look up a matcher in the registry by name, |
||
| 95 | /// |
||
| 96 | /// \return An opaque value which may be used to refer to the matcher |
||
| 97 | /// constructor, or std::optional<MatcherCtor>() if not found. |
||
| 98 | static std::optional<MatcherCtor> lookupMatcherCtor(StringRef MatcherName); |
||
| 99 | |||
| 100 | /// Compute the list of completion types for \p Context. |
||
| 101 | /// |
||
| 102 | /// Each element of \p Context represents a matcher invocation, going from |
||
| 103 | /// outermost to innermost. Elements are pairs consisting of a reference to |
||
| 104 | /// the matcher constructor and the index of the next element in the |
||
| 105 | /// argument list of that matcher (or for the last element, the index of |
||
| 106 | /// the completion point in the argument list). An empty list requests |
||
| 107 | /// completion for the root matcher. |
||
| 108 | static std::vector<ArgKind> getAcceptedCompletionTypes( |
||
| 109 | llvm::ArrayRef<std::pair<MatcherCtor, unsigned>> Context); |
||
| 110 | |||
| 111 | /// Compute the list of completions that match any of |
||
| 112 | /// \p AcceptedTypes. |
||
| 113 | /// |
||
| 114 | /// \param AcceptedTypes All types accepted for this completion. |
||
| 115 | /// |
||
| 116 | /// \return All completions for the specified types. |
||
| 117 | /// Completions should be valid when used in \c lookupMatcherCtor(). |
||
| 118 | /// The matcher constructed from the return of \c lookupMatcherCtor() |
||
| 119 | /// should be convertible to some type in \p AcceptedTypes. |
||
| 120 | static std::vector<MatcherCompletion> |
||
| 121 | getMatcherCompletions(ArrayRef<ArgKind> AcceptedTypes); |
||
| 122 | |||
| 123 | /// Construct a matcher from the registry. |
||
| 124 | /// |
||
| 125 | /// \param Ctor The matcher constructor to instantiate. |
||
| 126 | /// |
||
| 127 | /// \param NameRange The location of the name in the matcher source. |
||
| 128 | /// Useful for error reporting. |
||
| 129 | /// |
||
| 130 | /// \param Args The argument list for the matcher. The number and types of the |
||
| 131 | /// values must be valid for the matcher requested. Otherwise, the function |
||
| 132 | /// will return an error. |
||
| 133 | /// |
||
| 134 | /// \return The matcher object constructed if no error was found. |
||
| 135 | /// A null matcher if the number of arguments or argument types do not match |
||
| 136 | /// the signature. In that case \c Error will contain the description of |
||
| 137 | /// the error. |
||
| 138 | static VariantMatcher constructMatcher(MatcherCtor Ctor, |
||
| 139 | SourceRange NameRange, |
||
| 140 | ArrayRef<ParserValue> Args, |
||
| 141 | Diagnostics *Error); |
||
| 142 | |||
| 143 | /// Construct a matcher from the registry and bind it. |
||
| 144 | /// |
||
| 145 | /// Similar the \c constructMatcher() above, but it then tries to bind the |
||
| 146 | /// matcher to the specified \c BindID. |
||
| 147 | /// If the matcher is not bindable, it sets an error in \c Error and returns |
||
| 148 | /// a null matcher. |
||
| 149 | static VariantMatcher constructBoundMatcher(MatcherCtor Ctor, |
||
| 150 | SourceRange NameRange, |
||
| 151 | StringRef BindID, |
||
| 152 | ArrayRef<ParserValue> Args, |
||
| 153 | Diagnostics *Error); |
||
| 154 | }; |
||
| 155 | |||
| 156 | } // namespace dynamic |
||
| 157 | } // namespace ast_matchers |
||
| 158 | } // namespace clang |
||
| 159 | |||
| 160 | #endif // LLVM_CLANG_ASTMATCHERS_DYNAMIC_REGISTRY_H |