clang 17.0.0git
SemaCodeComplete.cpp
Go to the documentation of this file.
1//===---------------- SemaCodeComplete.cpp - Code Completion ----*- 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// This file defines the code-completion semantic actions.
10//
11//===----------------------------------------------------------------------===//
13#include "clang/AST/Decl.h"
14#include "clang/AST/DeclBase.h"
15#include "clang/AST/DeclCXX.h"
16#include "clang/AST/DeclObjC.h"
18#include "clang/AST/Expr.h"
19#include "clang/AST/ExprCXX.h"
21#include "clang/AST/ExprObjC.h"
25#include "clang/AST/Type.h"
31#include "clang/Lex/MacroInfo.h"
34#include "clang/Sema/DeclSpec.h"
36#include "clang/Sema/Lookup.h"
37#include "clang/Sema/Overload.h"
40#include "clang/Sema/Scope.h"
42#include "clang/Sema/Sema.h"
44#include "llvm/ADT/ArrayRef.h"
45#include "llvm/ADT/DenseSet.h"
46#include "llvm/ADT/SmallBitVector.h"
47#include "llvm/ADT/SmallPtrSet.h"
48#include "llvm/ADT/SmallString.h"
49#include "llvm/ADT/StringExtras.h"
50#include "llvm/ADT/StringSwitch.h"
51#include "llvm/ADT/Twine.h"
52#include "llvm/ADT/iterator_range.h"
53#include "llvm/Support/Casting.h"
54#include "llvm/Support/Path.h"
55#include "llvm/Support/raw_ostream.h"
56
57#include <list>
58#include <map>
59#include <optional>
60#include <string>
61#include <vector>
62
63using namespace clang;
64using namespace sema;
65
66namespace {
67/// A container of code-completion results.
68class ResultBuilder {
69public:
70 /// The type of a name-lookup filter, which can be provided to the
71 /// name-lookup routines to specify which declarations should be included in
72 /// the result set (when it returns true) and which declarations should be
73 /// filtered out (returns false).
74 typedef bool (ResultBuilder::*LookupFilter)(const NamedDecl *) const;
75
76 typedef CodeCompletionResult Result;
77
78private:
79 /// The actual results we have found.
80 std::vector<Result> Results;
81
82 /// A record of all of the declarations we have found and placed
83 /// into the result set, used to ensure that no declaration ever gets into
84 /// the result set twice.
86
87 typedef std::pair<const NamedDecl *, unsigned> DeclIndexPair;
88
89 /// An entry in the shadow map, which is optimized to store
90 /// a single (declaration, index) mapping (the common case) but
91 /// can also store a list of (declaration, index) mappings.
92 class ShadowMapEntry {
93 typedef SmallVector<DeclIndexPair, 4> DeclIndexPairVector;
94
95 /// Contains either the solitary NamedDecl * or a vector
96 /// of (declaration, index) pairs.
97 llvm::PointerUnion<const NamedDecl *, DeclIndexPairVector *> DeclOrVector;
98
99 /// When the entry contains a single declaration, this is
100 /// the index associated with that entry.
101 unsigned SingleDeclIndex;
102
103 public:
104 ShadowMapEntry() : SingleDeclIndex(0) {}
105 ShadowMapEntry(const ShadowMapEntry &) = delete;
106 ShadowMapEntry(ShadowMapEntry &&Move) { *this = std::move(Move); }
107 ShadowMapEntry &operator=(const ShadowMapEntry &) = delete;
108 ShadowMapEntry &operator=(ShadowMapEntry &&Move) {
109 SingleDeclIndex = Move.SingleDeclIndex;
110 DeclOrVector = Move.DeclOrVector;
111 Move.DeclOrVector = nullptr;
112 return *this;
113 }
114
115 void Add(const NamedDecl *ND, unsigned Index) {
116 if (DeclOrVector.isNull()) {
117 // 0 - > 1 elements: just set the single element information.
118 DeclOrVector = ND;
119 SingleDeclIndex = Index;
120 return;
121 }
122
123 if (const NamedDecl *PrevND =
124 DeclOrVector.dyn_cast<const NamedDecl *>()) {
125 // 1 -> 2 elements: create the vector of results and push in the
126 // existing declaration.
127 DeclIndexPairVector *Vec = new DeclIndexPairVector;
128 Vec->push_back(DeclIndexPair(PrevND, SingleDeclIndex));
129 DeclOrVector = Vec;
130 }
131
132 // Add the new element to the end of the vector.
133 DeclOrVector.get<DeclIndexPairVector *>()->push_back(
134 DeclIndexPair(ND, Index));
135 }
136
137 ~ShadowMapEntry() {
138 if (DeclIndexPairVector *Vec =
139 DeclOrVector.dyn_cast<DeclIndexPairVector *>()) {
140 delete Vec;
141 DeclOrVector = ((NamedDecl *)nullptr);
142 }
143 }
144
145 // Iteration.
146 class iterator;
147 iterator begin() const;
148 iterator end() const;
149 };
150
151 /// A mapping from declaration names to the declarations that have
152 /// this name within a particular scope and their index within the list of
153 /// results.
154 typedef llvm::DenseMap<DeclarationName, ShadowMapEntry> ShadowMap;
155
156 /// The semantic analysis object for which results are being
157 /// produced.
158 Sema &SemaRef;
159
160 /// The allocator used to allocate new code-completion strings.
161 CodeCompletionAllocator &Allocator;
162
163 CodeCompletionTUInfo &CCTUInfo;
164
165 /// If non-NULL, a filter function used to remove any code-completion
166 /// results that are not desirable.
167 LookupFilter Filter;
168
169 /// Whether we should allow declarations as
170 /// nested-name-specifiers that would otherwise be filtered out.
171 bool AllowNestedNameSpecifiers;
172
173 /// If set, the type that we would prefer our resulting value
174 /// declarations to have.
175 ///
176 /// Closely matching the preferred type gives a boost to a result's
177 /// priority.
178 CanQualType PreferredType;
179
180 /// A list of shadow maps, which is used to model name hiding at
181 /// different levels of, e.g., the inheritance hierarchy.
182 std::list<ShadowMap> ShadowMaps;
183
184 /// Overloaded C++ member functions found by SemaLookup.
185 /// Used to determine when one overload is dominated by another.
186 llvm::DenseMap<std::pair<DeclContext *, /*Name*/uintptr_t>, ShadowMapEntry>
187 OverloadMap;
188
189 /// If we're potentially referring to a C++ member function, the set
190 /// of qualifiers applied to the object type.
191 Qualifiers ObjectTypeQualifiers;
192 /// The kind of the object expression, for rvalue/lvalue overloads.
193 ExprValueKind ObjectKind;
194
195 /// Whether the \p ObjectTypeQualifiers field is active.
196 bool HasObjectTypeQualifiers;
197
198 /// The selector that we prefer.
199 Selector PreferredSelector;
200
201 /// The completion context in which we are gathering results.
202 CodeCompletionContext CompletionContext;
203
204 /// If we are in an instance method definition, the \@implementation
205 /// object.
206 ObjCImplementationDecl *ObjCImplementation;
207
208 void AdjustResultPriorityForDecl(Result &R);
209
210 void MaybeAddConstructorResults(Result R);
211
212public:
213 explicit ResultBuilder(Sema &SemaRef, CodeCompletionAllocator &Allocator,
214 CodeCompletionTUInfo &CCTUInfo,
215 const CodeCompletionContext &CompletionContext,
216 LookupFilter Filter = nullptr)
217 : SemaRef(SemaRef), Allocator(Allocator), CCTUInfo(CCTUInfo),
218 Filter(Filter), AllowNestedNameSpecifiers(false),
219 HasObjectTypeQualifiers(false), CompletionContext(CompletionContext),
220 ObjCImplementation(nullptr) {
221 // If this is an Objective-C instance method definition, dig out the
222 // corresponding implementation.
223 switch (CompletionContext.getKind()) {
229 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl())
230 if (Method->isInstanceMethod())
231 if (ObjCInterfaceDecl *Interface = Method->getClassInterface())
232 ObjCImplementation = Interface->getImplementation();
233 break;
234
235 default:
236 break;
237 }
238 }
239
240 /// Determine the priority for a reference to the given declaration.
241 unsigned getBasePriority(const NamedDecl *D);
242
243 /// Whether we should include code patterns in the completion
244 /// results.
245 bool includeCodePatterns() const {
246 return SemaRef.CodeCompleter &&
248 }
249
250 /// Set the filter used for code-completion results.
251 void setFilter(LookupFilter Filter) { this->Filter = Filter; }
252
253 Result *data() { return Results.empty() ? nullptr : &Results.front(); }
254 unsigned size() const { return Results.size(); }
255 bool empty() const { return Results.empty(); }
256
257 /// Specify the preferred type.
258 void setPreferredType(QualType T) {
259 PreferredType = SemaRef.Context.getCanonicalType(T);
260 }
261
262 /// Set the cv-qualifiers on the object type, for us in filtering
263 /// calls to member functions.
264 ///
265 /// When there are qualifiers in this set, they will be used to filter
266 /// out member functions that aren't available (because there will be a
267 /// cv-qualifier mismatch) or prefer functions with an exact qualifier
268 /// match.
269 void setObjectTypeQualifiers(Qualifiers Quals, ExprValueKind Kind) {
270 ObjectTypeQualifiers = Quals;
271 ObjectKind = Kind;
272 HasObjectTypeQualifiers = true;
273 }
274
275 /// Set the preferred selector.
276 ///
277 /// When an Objective-C method declaration result is added, and that
278 /// method's selector matches this preferred selector, we give that method
279 /// a slight priority boost.
280 void setPreferredSelector(Selector Sel) { PreferredSelector = Sel; }
281
282 /// Retrieve the code-completion context for which results are
283 /// being collected.
284 const CodeCompletionContext &getCompletionContext() const {
285 return CompletionContext;
286 }
287
288 /// Specify whether nested-name-specifiers are allowed.
289 void allowNestedNameSpecifiers(bool Allow = true) {
290 AllowNestedNameSpecifiers = Allow;
291 }
292
293 /// Return the semantic analysis object for which we are collecting
294 /// code completion results.
295 Sema &getSema() const { return SemaRef; }
296
297 /// Retrieve the allocator used to allocate code completion strings.
298 CodeCompletionAllocator &getAllocator() const { return Allocator; }
299
300 CodeCompletionTUInfo &getCodeCompletionTUInfo() const { return CCTUInfo; }
301
302 /// Determine whether the given declaration is at all interesting
303 /// as a code-completion result.
304 ///
305 /// \param ND the declaration that we are inspecting.
306 ///
307 /// \param AsNestedNameSpecifier will be set true if this declaration is
308 /// only interesting when it is a nested-name-specifier.
309 bool isInterestingDecl(const NamedDecl *ND,
310 bool &AsNestedNameSpecifier) const;
311
312 /// Check whether the result is hidden by the Hiding declaration.
313 ///
314 /// \returns true if the result is hidden and cannot be found, false if
315 /// the hidden result could still be found. When false, \p R may be
316 /// modified to describe how the result can be found (e.g., via extra
317 /// qualification).
318 bool CheckHiddenResult(Result &R, DeclContext *CurContext,
319 const NamedDecl *Hiding);
320
321 /// Add a new result to this result set (if it isn't already in one
322 /// of the shadow maps), or replace an existing result (for, e.g., a
323 /// redeclaration).
324 ///
325 /// \param R the result to add (if it is unique).
326 ///
327 /// \param CurContext the context in which this result will be named.
328 void MaybeAddResult(Result R, DeclContext *CurContext = nullptr);
329
330 /// Add a new result to this result set, where we already know
331 /// the hiding declaration (if any).
332 ///
333 /// \param R the result to add (if it is unique).
334 ///
335 /// \param CurContext the context in which this result will be named.
336 ///
337 /// \param Hiding the declaration that hides the result.
338 ///
339 /// \param InBaseClass whether the result was found in a base
340 /// class of the searched context.
341 void AddResult(Result R, DeclContext *CurContext, NamedDecl *Hiding,
342 bool InBaseClass);
343
344 /// Add a new non-declaration result to this result set.
345 void AddResult(Result R);
346
347 /// Enter into a new scope.
348 void EnterNewScope();
349
350 /// Exit from the current scope.
351 void ExitScope();
352
353 /// Ignore this declaration, if it is seen again.
354 void Ignore(const Decl *D) { AllDeclsFound.insert(D->getCanonicalDecl()); }
355
356 /// Add a visited context.
357 void addVisitedContext(DeclContext *Ctx) {
358 CompletionContext.addVisitedContext(Ctx);
359 }
360
361 /// \name Name lookup predicates
362 ///
363 /// These predicates can be passed to the name lookup functions to filter the
364 /// results of name lookup. All of the predicates have the same type, so that
365 ///
366 //@{
367 bool IsOrdinaryName(const NamedDecl *ND) const;
368 bool IsOrdinaryNonTypeName(const NamedDecl *ND) const;
369 bool IsIntegralConstantValue(const NamedDecl *ND) const;
370 bool IsOrdinaryNonValueName(const NamedDecl *ND) const;
371 bool IsNestedNameSpecifier(const NamedDecl *ND) const;
372 bool IsEnum(const NamedDecl *ND) const;
373 bool IsClassOrStruct(const NamedDecl *ND) const;
374 bool IsUnion(const NamedDecl *ND) const;
375 bool IsNamespace(const NamedDecl *ND) const;
376 bool IsNamespaceOrAlias(const NamedDecl *ND) const;
377 bool IsType(const NamedDecl *ND) const;
378 bool IsMember(const NamedDecl *ND) const;
379 bool IsObjCIvar(const NamedDecl *ND) const;
380 bool IsObjCMessageReceiver(const NamedDecl *ND) const;
381 bool IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const;
382 bool IsObjCCollection(const NamedDecl *ND) const;
383 bool IsImpossibleToSatisfy(const NamedDecl *ND) const;
384 //@}
385};
386} // namespace
387
389 if (!Enabled)
390 return;
391 if (isa<BlockDecl>(S.CurContext)) {
392 if (sema::BlockScopeInfo *BSI = S.getCurBlock()) {
393 ComputeType = nullptr;
394 Type = BSI->ReturnType;
395 ExpectedLoc = Tok;
396 }
397 } else if (const auto *Function = dyn_cast<FunctionDecl>(S.CurContext)) {
398 ComputeType = nullptr;
399 Type = Function->getReturnType();
400 ExpectedLoc = Tok;
401 } else if (const auto *Method = dyn_cast<ObjCMethodDecl>(S.CurContext)) {
402 ComputeType = nullptr;
403 Type = Method->getReturnType();
404 ExpectedLoc = Tok;
405 }
406}
407
409 if (!Enabled)
410 return;
411 auto *VD = llvm::dyn_cast_or_null<ValueDecl>(D);
412 ComputeType = nullptr;
413 Type = VD ? VD->getType() : QualType();
414 ExpectedLoc = Tok;
415}
416
417static QualType getDesignatedType(QualType BaseType, const Designation &Desig);
418
420 QualType BaseType,
421 const Designation &D) {
422 if (!Enabled)
423 return;
424 ComputeType = nullptr;
425 Type = getDesignatedType(BaseType, D);
426 ExpectedLoc = Tok;
427}
428
430 SourceLocation Tok, llvm::function_ref<QualType()> ComputeType) {
431 if (!Enabled)
432 return;
433 this->ComputeType = ComputeType;
434 Type = QualType();
435 ExpectedLoc = Tok;
436}
437
439 SourceLocation LParLoc) {
440 if (!Enabled)
441 return;
442 // expected type for parenthesized expression does not change.
443 if (ExpectedLoc == LParLoc)
444 ExpectedLoc = Tok;
445}
446
448 tok::TokenKind Op) {
449 if (!LHS)
450 return QualType();
451
452 QualType LHSType = LHS->getType();
453 if (LHSType->isPointerType()) {
454 if (Op == tok::plus || Op == tok::plusequal || Op == tok::minusequal)
456 // Pointer difference is more common than subtracting an int from a pointer.
457 if (Op == tok::minus)
458 return LHSType;
459 }
460
461 switch (Op) {
462 // No way to infer the type of RHS from LHS.
463 case tok::comma:
464 return QualType();
465 // Prefer the type of the left operand for all of these.
466 // Arithmetic operations.
467 case tok::plus:
468 case tok::plusequal:
469 case tok::minus:
470 case tok::minusequal:
471 case tok::percent:
472 case tok::percentequal:
473 case tok::slash:
474 case tok::slashequal:
475 case tok::star:
476 case tok::starequal:
477 // Assignment.
478 case tok::equal:
479 // Comparison operators.
480 case tok::equalequal:
481 case tok::exclaimequal:
482 case tok::less:
483 case tok::lessequal:
484 case tok::greater:
485 case tok::greaterequal:
486 case tok::spaceship:
487 return LHS->getType();
488 // Binary shifts are often overloaded, so don't try to guess those.
489 case tok::greatergreater:
490 case tok::greatergreaterequal:
491 case tok::lessless:
492 case tok::lesslessequal:
493 if (LHSType->isIntegralOrEnumerationType())
494 return S.getASTContext().IntTy;
495 return QualType();
496 // Logical operators, assume we want bool.
497 case tok::ampamp:
498 case tok::pipepipe:
499 case tok::caretcaret:
500 return S.getASTContext().BoolTy;
501 // Operators often used for bit manipulation are typically used with the type
502 // of the left argument.
503 case tok::pipe:
504 case tok::pipeequal:
505 case tok::caret:
506 case tok::caretequal:
507 case tok::amp:
508 case tok::ampequal:
509 if (LHSType->isIntegralOrEnumerationType())
510 return LHSType;
511 return QualType();
512 // RHS should be a pointer to a member of the 'LHS' type, but we can't give
513 // any particular type here.
514 case tok::periodstar:
515 case tok::arrowstar:
516 return QualType();
517 default:
518 // FIXME(ibiryukov): handle the missing op, re-add the assertion.
519 // assert(false && "unhandled binary op");
520 return QualType();
521 }
522}
523
524/// Get preferred type for an argument of an unary expression. \p ContextType is
525/// preferred type of the whole unary expression.
527 tok::TokenKind Op) {
528 switch (Op) {
529 case tok::exclaim:
530 return S.getASTContext().BoolTy;
531 case tok::amp:
532 if (!ContextType.isNull() && ContextType->isPointerType())
533 return ContextType->getPointeeType();
534 return QualType();
535 case tok::star:
536 if (ContextType.isNull())
537 return QualType();
538 return S.getASTContext().getPointerType(ContextType.getNonReferenceType());
539 case tok::plus:
540 case tok::minus:
541 case tok::tilde:
542 case tok::minusminus:
543 case tok::plusplus:
544 if (ContextType.isNull())
545 return S.getASTContext().IntTy;
546 // leave as is, these operators typically return the same type.
547 return ContextType;
548 case tok::kw___real:
549 case tok::kw___imag:
550 return QualType();
551 default:
552 assert(false && "unhandled unary op");
553 return QualType();
554 }
555}
556
558 tok::TokenKind Op) {
559 if (!Enabled)
560 return;
561 ComputeType = nullptr;
562 Type = getPreferredTypeOfBinaryRHS(S, LHS, Op);
563 ExpectedLoc = Tok;
564}
565
567 Expr *Base) {
568 if (!Enabled || !Base)
569 return;
570 // Do we have expected type for Base?
571 if (ExpectedLoc != Base->getBeginLoc())
572 return;
573 // Keep the expected type, only update the location.
574 ExpectedLoc = Tok;
575}
576
578 tok::TokenKind OpKind,
579 SourceLocation OpLoc) {
580 if (!Enabled)
581 return;
582 ComputeType = nullptr;
583 Type = getPreferredTypeOfUnaryArg(S, this->get(OpLoc), OpKind);
584 ExpectedLoc = Tok;
585}
586
588 Expr *LHS) {
589 if (!Enabled)
590 return;
591 ComputeType = nullptr;
593 ExpectedLoc = Tok;
594}
595
598 if (!Enabled)
599 return;
600 ComputeType = nullptr;
601 Type = !CastType.isNull() ? CastType.getCanonicalType() : QualType();
602 ExpectedLoc = Tok;
603}
604
606 if (!Enabled)
607 return;
608 ComputeType = nullptr;
610 ExpectedLoc = Tok;
611}
612
614 llvm::PointerUnion<const NamedDecl *, const DeclIndexPair *> DeclOrIterator;
615 unsigned SingleDeclIndex;
616
617public:
618 typedef DeclIndexPair value_type;
620 typedef std::ptrdiff_t difference_type;
621 typedef std::input_iterator_tag iterator_category;
622
623 class pointer {
624 DeclIndexPair Value;
625
626 public:
627 pointer(const DeclIndexPair &Value) : Value(Value) {}
628
629 const DeclIndexPair *operator->() const { return &Value; }
630 };
631
632 iterator() : DeclOrIterator((NamedDecl *)nullptr), SingleDeclIndex(0) {}
633
634 iterator(const NamedDecl *SingleDecl, unsigned Index)
635 : DeclOrIterator(SingleDecl), SingleDeclIndex(Index) {}
636
637 iterator(const DeclIndexPair *Iterator)
638 : DeclOrIterator(Iterator), SingleDeclIndex(0) {}
639
641 if (DeclOrIterator.is<const NamedDecl *>()) {
642 DeclOrIterator = (NamedDecl *)nullptr;
643 SingleDeclIndex = 0;
644 return *this;
645 }
646
647 const DeclIndexPair *I = DeclOrIterator.get<const DeclIndexPair *>();
648 ++I;
649 DeclOrIterator = I;
650 return *this;
651 }
652
653 /*iterator operator++(int) {
654 iterator tmp(*this);
655 ++(*this);
656 return tmp;
657 }*/
658
660 if (const NamedDecl *ND = DeclOrIterator.dyn_cast<const NamedDecl *>())
661 return reference(ND, SingleDeclIndex);
662
663 return *DeclOrIterator.get<const DeclIndexPair *>();
664 }
665
666 pointer operator->() const { return pointer(**this); }
667
668 friend bool operator==(const iterator &X, const iterator &Y) {
669 return X.DeclOrIterator.getOpaqueValue() ==
670 Y.DeclOrIterator.getOpaqueValue() &&
671 X.SingleDeclIndex == Y.SingleDeclIndex;
672 }
673
674 friend bool operator!=(const iterator &X, const iterator &Y) {
675 return !(X == Y);
676 }
677};
678
680ResultBuilder::ShadowMapEntry::begin() const {
681 if (DeclOrVector.isNull())
682 return iterator();
683
684 if (const NamedDecl *ND = DeclOrVector.dyn_cast<const NamedDecl *>())
685 return iterator(ND, SingleDeclIndex);
686
687 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->begin());
688}
689
691ResultBuilder::ShadowMapEntry::end() const {
692 if (DeclOrVector.is<const NamedDecl *>() || DeclOrVector.isNull())
693 return iterator();
694
695 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->end());
696}
697
698/// Compute the qualification required to get from the current context
699/// (\p CurContext) to the target context (\p TargetContext).
700///
701/// \param Context the AST context in which the qualification will be used.
702///
703/// \param CurContext the context where an entity is being named, which is
704/// typically based on the current scope.
705///
706/// \param TargetContext the context in which the named entity actually
707/// resides.
708///
709/// \returns a nested name specifier that refers into the target context, or
710/// NULL if no qualification is needed.
711static NestedNameSpecifier *
713 const DeclContext *TargetContext) {
715
716 for (const DeclContext *CommonAncestor = TargetContext;
717 CommonAncestor && !CommonAncestor->Encloses(CurContext);
718 CommonAncestor = CommonAncestor->getLookupParent()) {
719 if (CommonAncestor->isTransparentContext() ||
720 CommonAncestor->isFunctionOrMethod())
721 continue;
722
723 TargetParents.push_back(CommonAncestor);
724 }
725
726 NestedNameSpecifier *Result = nullptr;
727 while (!TargetParents.empty()) {
728 const DeclContext *Parent = TargetParents.pop_back_val();
729
730 if (const auto *Namespace = dyn_cast<NamespaceDecl>(Parent)) {
731 if (!Namespace->getIdentifier())
732 continue;
733
734 Result = NestedNameSpecifier::Create(Context, Result, Namespace);
735 } else if (const auto *TD = dyn_cast<TagDecl>(Parent))
737 Context, Result, false, Context.getTypeDeclType(TD).getTypePtr());
738 }
739 return Result;
740}
741
742// Some declarations have reserved names that we don't want to ever show.
743// Filter out names reserved for the implementation if they come from a
744// system header.
745static bool shouldIgnoreDueToReservedName(const NamedDecl *ND, Sema &SemaRef) {
746 ReservedIdentifierStatus Status = ND->isReserved(SemaRef.getLangOpts());
747 // Ignore reserved names for compiler provided decls.
748 if (isReservedInAllContexts(Status) && ND->getLocation().isInvalid())
749 return true;
750
751 // For system headers ignore only double-underscore names.
752 // This allows for system headers providing private symbols with a single
753 // underscore.
756 SemaRef.SourceMgr.getSpellingLoc(ND->getLocation())))
757 return true;
758
759 return false;
760}
761
762bool ResultBuilder::isInterestingDecl(const NamedDecl *ND,
763 bool &AsNestedNameSpecifier) const {
764 AsNestedNameSpecifier = false;
765
766 auto *Named = ND;
767 ND = ND->getUnderlyingDecl();
768
769 // Skip unnamed entities.
770 if (!ND->getDeclName())
771 return false;
772
773 // Friend declarations and declarations introduced due to friends are never
774 // added as results.
776 return false;
777
778 // Class template (partial) specializations are never added as results.
779 if (isa<ClassTemplateSpecializationDecl>(ND) ||
780 isa<ClassTemplatePartialSpecializationDecl>(ND))
781 return false;
782
783 // Using declarations themselves are never added as results.
784 if (isa<UsingDecl>(ND))
785 return false;
786
787 if (shouldIgnoreDueToReservedName(ND, SemaRef))
788 return false;
789
790 if (Filter == &ResultBuilder::IsNestedNameSpecifier ||
791 (isa<NamespaceDecl>(ND) && Filter != &ResultBuilder::IsNamespace &&
792 Filter != &ResultBuilder::IsNamespaceOrAlias && Filter != nullptr))
793 AsNestedNameSpecifier = true;
794
795 // Filter out any unwanted results.
796 if (Filter && !(this->*Filter)(Named)) {
797 // Check whether it is interesting as a nested-name-specifier.
798 if (AllowNestedNameSpecifiers && SemaRef.getLangOpts().CPlusPlus &&
799 IsNestedNameSpecifier(ND) &&
800 (Filter != &ResultBuilder::IsMember ||
801 (isa<CXXRecordDecl>(ND) &&
802 cast<CXXRecordDecl>(ND)->isInjectedClassName()))) {
803 AsNestedNameSpecifier = true;
804 return true;
805 }
806
807 return false;
808 }
809 // ... then it must be interesting!
810 return true;
811}
812
813bool ResultBuilder::CheckHiddenResult(Result &R, DeclContext *CurContext,
814 const NamedDecl *Hiding) {
815 // In C, there is no way to refer to a hidden name.
816 // FIXME: This isn't true; we can find a tag name hidden by an ordinary
817 // name if we introduce the tag type.
818 if (!SemaRef.getLangOpts().CPlusPlus)
819 return true;
820
821 const DeclContext *HiddenCtx =
822 R.Declaration->getDeclContext()->getRedeclContext();
823
824 // There is no way to qualify a name declared in a function or method.
825 if (HiddenCtx->isFunctionOrMethod())
826 return true;
827
828 if (HiddenCtx == Hiding->getDeclContext()->getRedeclContext())
829 return true;
830
831 // We can refer to the result with the appropriate qualification. Do it.
832 R.Hidden = true;
833 R.QualifierIsInformative = false;
834
835 if (!R.Qualifier)
836 R.Qualifier = getRequiredQualification(SemaRef.Context, CurContext,
837 R.Declaration->getDeclContext());
838 return false;
839}
840
841/// A simplified classification of types used to determine whether two
842/// types are "similar enough" when adjusting priorities.
844 switch (T->getTypeClass()) {
845 case Type::Builtin:
846 switch (cast<BuiltinType>(T)->getKind()) {
847 case BuiltinType::Void:
848 return STC_Void;
849
850 case BuiltinType::NullPtr:
851 return STC_Pointer;
852
853 case BuiltinType::Overload:
854 case BuiltinType::Dependent:
855 return STC_Other;
856
857 case BuiltinType::ObjCId:
858 case BuiltinType::ObjCClass:
859 case BuiltinType::ObjCSel:
860 return STC_ObjectiveC;
861
862 default:
863 return STC_Arithmetic;
864 }
865
866 case Type::Complex:
867 return STC_Arithmetic;
868
869 case Type::Pointer:
870 return STC_Pointer;
871
872 case Type::BlockPointer:
873 return STC_Block;
874
875 case Type::LValueReference:
876 case Type::RValueReference:
878
879 case Type::ConstantArray:
880 case Type::IncompleteArray:
881 case Type::VariableArray:
882 case Type::DependentSizedArray:
883 return STC_Array;
884
885 case Type::DependentSizedExtVector:
886 case Type::Vector:
887 case Type::ExtVector:
888 return STC_Arithmetic;
889
890 case Type::FunctionProto:
891 case Type::FunctionNoProto:
892 return STC_Function;
893
894 case Type::Record:
895 return STC_Record;
896
897 case Type::Enum:
898 return STC_Arithmetic;
899
900 case Type::ObjCObject:
901 case Type::ObjCInterface:
902 case Type::ObjCObjectPointer:
903 return STC_ObjectiveC;
904
905 default:
906 return STC_Other;
907 }
908}
909
910/// Get the type that a given expression will have if this declaration
911/// is used as an expression in its "typical" code-completion form.
913 ND = ND->getUnderlyingDecl();
914
915 if (const auto *Type = dyn_cast<TypeDecl>(ND))
916 return C.getTypeDeclType(Type);
917 if (const auto *Iface = dyn_cast<ObjCInterfaceDecl>(ND))
918 return C.getObjCInterfaceType(Iface);
919
920 QualType T;
921 if (const FunctionDecl *Function = ND->getAsFunction())
922 T = Function->getCallResultType();
923 else if (const auto *Method = dyn_cast<ObjCMethodDecl>(ND))
924 T = Method->getSendResultType();
925 else if (const auto *Enumerator = dyn_cast<EnumConstantDecl>(ND))
926 T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext()));
927 else if (const auto *Property = dyn_cast<ObjCPropertyDecl>(ND))
928 T = Property->getType();
929 else if (const auto *Value = dyn_cast<ValueDecl>(ND))
930 T = Value->getType();
931
932 if (T.isNull())
933 return QualType();
934
935 // Dig through references, function pointers, and block pointers to
936 // get down to the likely type of an expression when the entity is
937 // used.
938 do {
939 if (const auto *Ref = T->getAs<ReferenceType>()) {
940 T = Ref->getPointeeType();
941 continue;
942 }
943
944 if (const auto *Pointer = T->getAs<PointerType>()) {
945 if (Pointer->getPointeeType()->isFunctionType()) {
946 T = Pointer->getPointeeType();
947 continue;
948 }
949
950 break;
951 }
952
953 if (const auto *Block = T->getAs<BlockPointerType>()) {
954 T = Block->getPointeeType();
955 continue;
956 }
957
958 if (const auto *Function = T->getAs<FunctionType>()) {
959 T = Function->getReturnType();
960 continue;
961 }
962
963 break;
964 } while (true);
965
966 return T;
967}
968
969unsigned ResultBuilder::getBasePriority(const NamedDecl *ND) {
970 if (!ND)
971 return CCP_Unlikely;
972
973 // Context-based decisions.
974 const DeclContext *LexicalDC = ND->getLexicalDeclContext();
975 if (LexicalDC->isFunctionOrMethod()) {
976 // _cmd is relatively rare
977 if (const auto *ImplicitParam = dyn_cast<ImplicitParamDecl>(ND))
978 if (ImplicitParam->getIdentifier() &&
979 ImplicitParam->getIdentifier()->isStr("_cmd"))
980 return CCP_ObjC_cmd;
981
983 }
984
985 const DeclContext *DC = ND->getDeclContext()->getRedeclContext();
986 if (DC->isRecord() || isa<ObjCContainerDecl>(DC)) {
987 // Explicit destructor calls are very rare.
988 if (isa<CXXDestructorDecl>(ND))
989 return CCP_Unlikely;
990 // Explicit operator and conversion function calls are also very rare.
991 auto DeclNameKind = ND->getDeclName().getNameKind();
992 if (DeclNameKind == DeclarationName::CXXOperatorName ||
995 return CCP_Unlikely;
997 }
998
999 // Content-based decisions.
1000 if (isa<EnumConstantDecl>(ND))
1001 return CCP_Constant;
1002
1003 // Use CCP_Type for type declarations unless we're in a statement, Objective-C
1004 // message receiver, or parenthesized expression context. There, it's as
1005 // likely that the user will want to write a type as other declarations.
1006 if ((isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND)) &&
1007 !(CompletionContext.getKind() == CodeCompletionContext::CCC_Statement ||
1008 CompletionContext.getKind() ==
1010 CompletionContext.getKind() ==
1012 return CCP_Type;
1013
1014 return CCP_Declaration;
1015}
1016
1017void ResultBuilder::AdjustResultPriorityForDecl(Result &R) {
1018 // If this is an Objective-C method declaration whose selector matches our
1019 // preferred selector, give it a priority boost.
1020 if (!PreferredSelector.isNull())
1021 if (const auto *Method = dyn_cast<ObjCMethodDecl>(R.Declaration))
1022 if (PreferredSelector == Method->getSelector())
1023 R.Priority += CCD_SelectorMatch;
1024
1025 // If we have a preferred type, adjust the priority for results with exactly-
1026 // matching or nearly-matching types.
1027 if (!PreferredType.isNull()) {
1028 QualType T = getDeclUsageType(SemaRef.Context, R.Declaration);
1029 if (!T.isNull()) {
1030 CanQualType TC = SemaRef.Context.getCanonicalType(T);
1031 // Check for exactly-matching types (modulo qualifiers).
1032 if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC))
1033 R.Priority /= CCF_ExactTypeMatch;
1034 // Check for nearly-matching types, based on classification of each.
1035 else if ((getSimplifiedTypeClass(PreferredType) ==
1037 !(PreferredType->isEnumeralType() && TC->isEnumeralType()))
1038 R.Priority /= CCF_SimilarTypeMatch;
1039 }
1040 }
1041}
1042
1044 const CXXRecordDecl *Record) {
1045 QualType RecordTy = Context.getTypeDeclType(Record);
1046 DeclarationName ConstructorName =
1048 Context.getCanonicalType(RecordTy));
1049 return Record->lookup(ConstructorName);
1050}
1051
1052void ResultBuilder::MaybeAddConstructorResults(Result R) {
1053 if (!SemaRef.getLangOpts().CPlusPlus || !R.Declaration ||
1054 !CompletionContext.wantConstructorResults())
1055 return;
1056
1057 const NamedDecl *D = R.Declaration;
1058 const CXXRecordDecl *Record = nullptr;
1059 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D))
1060 Record = ClassTemplate->getTemplatedDecl();
1061 else if ((Record = dyn_cast<CXXRecordDecl>(D))) {
1062 // Skip specializations and partial specializations.
1063 if (isa<ClassTemplateSpecializationDecl>(Record))
1064 return;
1065 } else {
1066 // There are no constructors here.
1067 return;
1068 }
1069
1070 Record = Record->getDefinition();
1071 if (!Record)
1072 return;
1073
1074 for (NamedDecl *Ctor : getConstructors(SemaRef.Context, Record)) {
1075 R.Declaration = Ctor;
1076 R.CursorKind = getCursorKindForDecl(R.Declaration);
1077 Results.push_back(R);
1078 }
1079}
1080
1081static bool isConstructor(const Decl *ND) {
1082 if (const auto *Tmpl = dyn_cast<FunctionTemplateDecl>(ND))
1083 ND = Tmpl->getTemplatedDecl();
1084 return isa<CXXConstructorDecl>(ND);
1085}
1086
1087void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) {
1088 assert(!ShadowMaps.empty() && "Must enter into a results scope");
1089
1090 if (R.Kind != Result::RK_Declaration) {
1091 // For non-declaration results, just add the result.
1092 Results.push_back(R);
1093 return;
1094 }
1095
1096 // Look through using declarations.
1097 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
1098 CodeCompletionResult Result(Using->getTargetDecl(),
1099 getBasePriority(Using->getTargetDecl()),
1100 R.Qualifier, false,
1101 (R.Availability == CXAvailability_Available ||
1102 R.Availability == CXAvailability_Deprecated),
1103 std::move(R.FixIts));
1104 Result.ShadowDecl = Using;
1105 MaybeAddResult(Result, CurContext);
1106 return;
1107 }
1108
1109 const Decl *CanonDecl = R.Declaration->getCanonicalDecl();
1110 unsigned IDNS = CanonDecl->getIdentifierNamespace();
1111
1112 bool AsNestedNameSpecifier = false;
1113 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
1114 return;
1115
1116 // C++ constructors are never found by name lookup.
1117 if (isConstructor(R.Declaration))
1118 return;
1119
1120 ShadowMap &SMap = ShadowMaps.back();
1121 ShadowMapEntry::iterator I, IEnd;
1122 ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName());
1123 if (NamePos != SMap.end()) {
1124 I = NamePos->second.begin();
1125 IEnd = NamePos->second.end();
1126 }
1127
1128 for (; I != IEnd; ++I) {
1129 const NamedDecl *ND = I->first;
1130 unsigned Index = I->second;
1131 if (ND->getCanonicalDecl() == CanonDecl) {
1132 // This is a redeclaration. Always pick the newer declaration.
1133 Results[Index].Declaration = R.Declaration;
1134
1135 // We're done.
1136 return;
1137 }
1138 }
1139
1140 // This is a new declaration in this scope. However, check whether this
1141 // declaration name is hidden by a similarly-named declaration in an outer
1142 // scope.
1143 std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end();
1144 --SMEnd;
1145 for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) {
1146 ShadowMapEntry::iterator I, IEnd;
1147 ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName());
1148 if (NamePos != SM->end()) {
1149 I = NamePos->second.begin();
1150 IEnd = NamePos->second.end();
1151 }
1152 for (; I != IEnd; ++I) {
1153 // A tag declaration does not hide a non-tag declaration.
1154 if (I->first->hasTagIdentifierNamespace() &&
1157 continue;
1158
1159 // Protocols are in distinct namespaces from everything else.
1160 if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol) ||
1161 (IDNS & Decl::IDNS_ObjCProtocol)) &&
1162 I->first->getIdentifierNamespace() != IDNS)
1163 continue;
1164
1165 // The newly-added result is hidden by an entry in the shadow map.
1166 if (CheckHiddenResult(R, CurContext, I->first))
1167 return;
1168
1169 break;
1170 }
1171 }
1172
1173 // Make sure that any given declaration only shows up in the result set once.
1174 if (!AllDeclsFound.insert(CanonDecl).second)
1175 return;
1176
1177 // If the filter is for nested-name-specifiers, then this result starts a
1178 // nested-name-specifier.
1179 if (AsNestedNameSpecifier) {
1180 R.StartsNestedNameSpecifier = true;
1181 R.Priority = CCP_NestedNameSpecifier;
1182 } else
1183 AdjustResultPriorityForDecl(R);
1184
1185 // If this result is supposed to have an informative qualifier, add one.
1186 if (R.QualifierIsInformative && !R.Qualifier &&
1187 !R.StartsNestedNameSpecifier) {
1188 const DeclContext *Ctx = R.Declaration->getDeclContext();
1189 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
1190 R.Qualifier =
1191 NestedNameSpecifier::Create(SemaRef.Context, nullptr, Namespace);
1192 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
1193 R.Qualifier = NestedNameSpecifier::Create(
1194 SemaRef.Context, nullptr, false,
1195 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
1196 else
1197 R.QualifierIsInformative = false;
1198 }
1199
1200 // Insert this result into the set of results and into the current shadow
1201 // map.
1202 SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size());
1203 Results.push_back(R);
1204
1205 if (!AsNestedNameSpecifier)
1206 MaybeAddConstructorResults(R);
1207}
1208
1211 R.InBaseClass = true;
1212}
1213
1215// Will Candidate ever be called on the object, when overloaded with Incumbent?
1216// Returns Dominates if Candidate is always called, Dominated if Incumbent is
1217// always called, BothViable if either may be called depending on arguments.
1218// Precondition: must actually be overloads!
1220 const CXXMethodDecl &Incumbent,
1221 const Qualifiers &ObjectQuals,
1222 ExprValueKind ObjectKind) {
1223 // Base/derived shadowing is handled elsewhere.
1224 if (Candidate.getDeclContext() != Incumbent.getDeclContext())
1225 return OverloadCompare::BothViable;
1226 if (Candidate.isVariadic() != Incumbent.isVariadic() ||
1227 Candidate.getNumParams() != Incumbent.getNumParams() ||
1228 Candidate.getMinRequiredArguments() !=
1229 Incumbent.getMinRequiredArguments())
1230 return OverloadCompare::BothViable;
1231 for (unsigned I = 0, E = Candidate.getNumParams(); I != E; ++I)
1232 if (Candidate.parameters()[I]->getType().getCanonicalType() !=
1233 Incumbent.parameters()[I]->getType().getCanonicalType())
1234 return OverloadCompare::BothViable;
1235 if (!Candidate.specific_attrs<EnableIfAttr>().empty() ||
1236 !Incumbent.specific_attrs<EnableIfAttr>().empty())
1237 return OverloadCompare::BothViable;
1238 // At this point, we know calls can't pick one or the other based on
1239 // arguments, so one of the two must win. (Or both fail, handled elsewhere).
1240 RefQualifierKind CandidateRef = Candidate.getRefQualifier();
1241 RefQualifierKind IncumbentRef = Incumbent.getRefQualifier();
1242 if (CandidateRef != IncumbentRef) {
1243 // If the object kind is LValue/RValue, there's one acceptable ref-qualifier
1244 // and it can't be mixed with ref-unqualified overloads (in valid code).
1245
1246 // For xvalue objects, we prefer the rvalue overload even if we have to
1247 // add qualifiers (which is rare, because const&& is rare).
1248 if (ObjectKind == clang::VK_XValue)
1249 return CandidateRef == RQ_RValue ? OverloadCompare::Dominates
1250 : OverloadCompare::Dominated;
1251 }
1252 // Now the ref qualifiers are the same (or we're in some invalid state).
1253 // So make some decision based on the qualifiers.
1254 Qualifiers CandidateQual = Candidate.getMethodQualifiers();
1255 Qualifiers IncumbentQual = Incumbent.getMethodQualifiers();
1256 bool CandidateSuperset = CandidateQual.compatiblyIncludes(IncumbentQual);
1257 bool IncumbentSuperset = IncumbentQual.compatiblyIncludes(CandidateQual);
1258 if (CandidateSuperset == IncumbentSuperset)
1259 return OverloadCompare::BothViable;
1260 return IncumbentSuperset ? OverloadCompare::Dominates
1261 : OverloadCompare::Dominated;
1262}
1263
1264void ResultBuilder::AddResult(Result R, DeclContext *CurContext,
1265 NamedDecl *Hiding, bool InBaseClass = false) {
1266 if (R.Kind != Result::RK_Declaration) {
1267 // For non-declaration results, just add the result.
1268 Results.push_back(R);
1269 return;
1270 }
1271
1272 // Look through using declarations.
1273 if (const auto *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
1274 CodeCompletionResult Result(Using->getTargetDecl(),
1275 getBasePriority(Using->getTargetDecl()),
1276 R.Qualifier, false,
1277 (R.Availability == CXAvailability_Available ||
1278 R.Availability == CXAvailability_Deprecated),
1279 std::move(R.FixIts));
1280 Result.ShadowDecl = Using;
1281 AddResult(Result, CurContext, Hiding);
1282 return;
1283 }
1284
1285 bool AsNestedNameSpecifier = false;
1286 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
1287 return;
1288
1289 // C++ constructors are never found by name lookup.
1290 if (isConstructor(R.Declaration))
1291 return;
1292
1293 if (Hiding && CheckHiddenResult(R, CurContext, Hiding))
1294 return;
1295
1296 // Make sure that any given declaration only shows up in the result set once.
1297 if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl()).second)
1298 return;
1299
1300 // If the filter is for nested-name-specifiers, then this result starts a
1301 // nested-name-specifier.
1302 if (AsNestedNameSpecifier) {
1303 R.StartsNestedNameSpecifier = true;
1304 R.Priority = CCP_NestedNameSpecifier;
1305 } else if (Filter == &ResultBuilder::IsMember && !R.Qualifier &&
1306 InBaseClass &&
1307 isa<CXXRecordDecl>(
1308 R.Declaration->getDeclContext()->getRedeclContext()))
1309 R.QualifierIsInformative = true;
1310
1311 // If this result is supposed to have an informative qualifier, add one.
1312 if (R.QualifierIsInformative && !R.Qualifier &&
1313 !R.StartsNestedNameSpecifier) {
1314 const DeclContext *Ctx = R.Declaration->getDeclContext();
1315 if (const auto *Namespace = dyn_cast<NamespaceDecl>(Ctx))
1316 R.Qualifier =
1317 NestedNameSpecifier::Create(SemaRef.Context, nullptr, Namespace);
1318 else if (const auto *Tag = dyn_cast<TagDecl>(Ctx))
1319 R.Qualifier = NestedNameSpecifier::Create(
1320 SemaRef.Context, nullptr, false,
1321 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
1322 else
1323 R.QualifierIsInformative = false;
1324 }
1325
1326 // Adjust the priority if this result comes from a base class.
1327 if (InBaseClass)
1328 setInBaseClass(R);
1329
1330 AdjustResultPriorityForDecl(R);
1331
1332 if (HasObjectTypeQualifiers)
1333 if (const auto *Method = dyn_cast<CXXMethodDecl>(R.Declaration))
1334 if (Method->isInstance()) {
1335 Qualifiers MethodQuals = Method->getMethodQualifiers();
1336 if (ObjectTypeQualifiers == MethodQuals)
1337 R.Priority += CCD_ObjectQualifierMatch;
1338 else if (ObjectTypeQualifiers - MethodQuals) {
1339 // The method cannot be invoked, because doing so would drop
1340 // qualifiers.
1341 return;
1342 }
1343 // Detect cases where a ref-qualified method cannot be invoked.
1344 switch (Method->getRefQualifier()) {
1345 case RQ_LValue:
1346 if (ObjectKind != VK_LValue && !MethodQuals.hasConst())
1347 return;
1348 break;
1349 case RQ_RValue:
1350 if (ObjectKind == VK_LValue)
1351 return;
1352 break;
1353 case RQ_None:
1354 break;
1355 }
1356
1357 /// Check whether this dominates another overloaded method, which should
1358 /// be suppressed (or vice versa).
1359 /// Motivating case is const_iterator begin() const vs iterator begin().
1360 auto &OverloadSet = OverloadMap[std::make_pair(
1361 CurContext, Method->getDeclName().getAsOpaqueInteger())];
1362 for (const DeclIndexPair Entry : OverloadSet) {
1363 Result &Incumbent = Results[Entry.second];
1364 switch (compareOverloads(*Method,
1365 *cast<CXXMethodDecl>(Incumbent.Declaration),
1366 ObjectTypeQualifiers, ObjectKind)) {
1367 case OverloadCompare::Dominates:
1368 // Replace the dominated overload with this one.
1369 // FIXME: if the overload dominates multiple incumbents then we
1370 // should remove all. But two overloads is by far the common case.
1371 Incumbent = std::move(R);
1372 return;
1373 case OverloadCompare::Dominated:
1374 // This overload can't be called, drop it.
1375 return;
1376 case OverloadCompare::BothViable:
1377 break;
1378 }
1379 }
1380 OverloadSet.Add(Method, Results.size());
1381 }
1382
1383 // When completing a non-static member function (and not via
1384 // dot/arrow member access) and we're not inside that class' scope,
1385 // it can't be a call.
1386 if (CompletionContext.getKind() == clang::CodeCompletionContext::CCC_Symbol) {
1387 const auto *Method = dyn_cast<CXXMethodDecl>(R.getDeclaration());
1388 if (Method && !Method->isStatic()) {
1389 // Find the class scope that we're currently in.
1390 // We could e.g. be inside a lambda, so walk up the DeclContext until we
1391 // find a CXXMethodDecl.
1392 const auto *CurrentClassScope = [&]() -> const CXXRecordDecl * {
1393 for (DeclContext *Ctx = SemaRef.CurContext; Ctx;
1394 Ctx = Ctx->getParent()) {
1395 const auto *CtxMethod = llvm::dyn_cast<CXXMethodDecl>(Ctx);
1396 if (CtxMethod && !CtxMethod->getParent()->isLambda()) {
1397 return CtxMethod->getParent();
1398 }
1399 }
1400 return nullptr;
1401 }();
1402
1403 R.FunctionCanBeCall =
1404 CurrentClassScope &&
1405 (CurrentClassScope == Method->getParent() ||
1406 CurrentClassScope->isDerivedFrom(Method->getParent()));
1407 }
1408 }
1409
1410 // Insert this result into the set of results.
1411 Results.push_back(R);
1412
1413 if (!AsNestedNameSpecifier)
1414 MaybeAddConstructorResults(R);
1415}
1416
1417void ResultBuilder::AddResult(Result R) {
1418 assert(R.Kind != Result::RK_Declaration &&
1419 "Declaration results need more context");
1420 Results.push_back(R);
1421}
1422
1423/// Enter into a new scope.
1424void ResultBuilder::EnterNewScope() { ShadowMaps.emplace_back(); }
1425
1426/// Exit from the current scope.
1427void ResultBuilder::ExitScope() {
1428 ShadowMaps.pop_back();
1429}
1430
1431/// Determines whether this given declaration will be found by
1432/// ordinary name lookup.
1433bool ResultBuilder::IsOrdinaryName(const NamedDecl *ND) const {
1434 ND = ND->getUnderlyingDecl();
1435
1436 // If name lookup finds a local extern declaration, then we are in a
1437 // context where it behaves like an ordinary name.
1439 if (SemaRef.getLangOpts().CPlusPlus)
1441 else if (SemaRef.getLangOpts().ObjC) {
1442 if (isa<ObjCIvarDecl>(ND))
1443 return true;
1444 }
1445
1446 return ND->getIdentifierNamespace() & IDNS;
1447}
1448
1449/// Determines whether this given declaration will be found by
1450/// ordinary name lookup but is not a type name.
1451bool ResultBuilder::IsOrdinaryNonTypeName(const NamedDecl *ND) const {
1452 ND = ND->getUnderlyingDecl();
1453 if (isa<TypeDecl>(ND))
1454 return false;
1455 // Objective-C interfaces names are not filtered by this method because they
1456 // can be used in a class property expression. We can still filter out
1457 // @class declarations though.
1458 if (const auto *ID = dyn_cast<ObjCInterfaceDecl>(ND)) {
1459 if (!ID->getDefinition())
1460 return false;
1461 }
1462
1464 if (SemaRef.getLangOpts().CPlusPlus)
1466 else if (SemaRef.getLangOpts().ObjC) {
1467 if (isa<ObjCIvarDecl>(ND))
1468 return true;
1469 }
1470
1471 return ND->getIdentifierNamespace() & IDNS;
1472}
1473
1474bool ResultBuilder::IsIntegralConstantValue(const NamedDecl *ND) const {
1475 if (!IsOrdinaryNonTypeName(ND))
1476 return false;
1477
1478 if (const auto *VD = dyn_cast<ValueDecl>(ND->getUnderlyingDecl()))
1479 if (VD->getType()->isIntegralOrEnumerationType())
1480 return true;
1481
1482 return false;
1483}
1484
1485/// Determines whether this given declaration will be found by
1486/// ordinary name lookup.
1487bool ResultBuilder::IsOrdinaryNonValueName(const NamedDecl *ND) const {
1488 ND = ND->getUnderlyingDecl();
1489
1491 if (SemaRef.getLangOpts().CPlusPlus)
1493
1494 return (ND->getIdentifierNamespace() & IDNS) && !isa<ValueDecl>(ND) &&
1495 !isa<FunctionTemplateDecl>(ND) && !isa<ObjCPropertyDecl>(ND);
1496}
1497
1498/// Determines whether the given declaration is suitable as the
1499/// start of a C++ nested-name-specifier, e.g., a class or namespace.
1500bool ResultBuilder::IsNestedNameSpecifier(const NamedDecl *ND) const {
1501 // Allow us to find class templates, too.
1502 if (const auto *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
1503 ND = ClassTemplate->getTemplatedDecl();
1504
1505 return SemaRef.isAcceptableNestedNameSpecifier(ND);
1506}
1507
1508/// Determines whether the given declaration is an enumeration.
1509bool ResultBuilder::IsEnum(const NamedDecl *ND) const {
1510 return isa<EnumDecl>(ND);
1511}
1512
1513/// Determines whether the given declaration is a class or struct.
1514bool ResultBuilder::IsClassOrStruct(const NamedDecl *ND) const {
1515 // Allow us to find class templates, too.
1516 if (const auto *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
1517 ND = ClassTemplate->getTemplatedDecl();
1518
1519 // For purposes of this check, interfaces match too.
1520 if (const auto *RD = dyn_cast<RecordDecl>(ND))
1521 return RD->getTagKind() == TTK_Class || RD->getTagKind() == TTK_Struct ||
1522 RD->getTagKind() == TTK_Interface;
1523
1524 return false;
1525}
1526
1527/// Determines whether the given declaration is a union.
1528bool ResultBuilder::IsUnion(const NamedDecl *ND) const {
1529 // Allow us to find class templates, too.
1530 if (const auto *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
1531 ND = ClassTemplate->getTemplatedDecl();
1532
1533 if (const auto *RD = dyn_cast<RecordDecl>(ND))
1534 return RD->getTagKind() == TTK_Union;
1535
1536 return false;
1537}
1538
1539/// Determines whether the given declaration is a namespace.
1540bool ResultBuilder::IsNamespace(const NamedDecl *ND) const {
1541 return isa<NamespaceDecl>(ND);
1542}
1543
1544/// Determines whether the given declaration is a namespace or
1545/// namespace alias.
1546bool ResultBuilder::IsNamespaceOrAlias(const NamedDecl *ND) const {
1547 return isa<NamespaceDecl>(ND->getUnderlyingDecl());
1548}
1549
1550/// Determines whether the given declaration is a type.
1551bool ResultBuilder::IsType(const NamedDecl *ND) const {
1552 ND = ND->getUnderlyingDecl();
1553 return isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND);
1554}
1555
1556/// Determines which members of a class should be visible via
1557/// "." or "->". Only value declarations, nested name specifiers, and
1558/// using declarations thereof should show up.
1559bool ResultBuilder::IsMember(const NamedDecl *ND) const {
1560 ND = ND->getUnderlyingDecl();
1561 return isa<ValueDecl>(ND) || isa<FunctionTemplateDecl>(ND) ||
1562 isa<ObjCPropertyDecl>(ND);
1563}
1564
1566 T = C.getCanonicalType(T);
1567 switch (T->getTypeClass()) {
1568 case Type::ObjCObject:
1569 case Type::ObjCInterface:
1570 case Type::ObjCObjectPointer:
1571 return true;
1572
1573 case Type::Builtin:
1574 switch (cast<BuiltinType>(T)->getKind()) {
1575 case BuiltinType::ObjCId:
1576 case BuiltinType::ObjCClass:
1577 case BuiltinType::ObjCSel:
1578 return true;
1579
1580 default:
1581 break;
1582 }
1583 return false;
1584
1585 default:
1586 break;
1587 }
1588
1589 if (!C.getLangOpts().CPlusPlus)
1590 return false;
1591
1592 // FIXME: We could perform more analysis here to determine whether a
1593 // particular class type has any conversions to Objective-C types. For now,
1594 // just accept all class types.
1595 return T->isDependentType() || T->isRecordType();
1596}
1597
1598bool ResultBuilder::IsObjCMessageReceiver(const NamedDecl *ND) const {
1599 QualType T = getDeclUsageType(SemaRef.Context, ND);
1600 if (T.isNull())
1601 return false;
1602
1603 T = SemaRef.Context.getBaseElementType(T);
1604 return isObjCReceiverType(SemaRef.Context, T);
1605}
1606
1607bool ResultBuilder::IsObjCMessageReceiverOrLambdaCapture(
1608 const NamedDecl *ND) const {
1609 if (IsObjCMessageReceiver(ND))
1610 return true;
1611
1612 const auto *Var = dyn_cast<VarDecl>(ND);
1613 if (!Var)
1614 return false;
1615
1616 return Var->hasLocalStorage() && !Var->hasAttr<BlocksAttr>();
1617}
1618
1619bool ResultBuilder::IsObjCCollection(const NamedDecl *ND) const {
1620 if ((SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryName(ND)) ||
1621 (!SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryNonTypeName(ND)))
1622 return false;
1623
1624 QualType T = getDeclUsageType(SemaRef.Context, ND);
1625 if (T.isNull())
1626 return false;
1627
1628 T = SemaRef.Context.getBaseElementType(T);
1629 return T->isObjCObjectType() || T->isObjCObjectPointerType() ||
1630 T->isObjCIdType() ||
1631 (SemaRef.getLangOpts().CPlusPlus && T->isRecordType());
1632}
1633
1634bool ResultBuilder::IsImpossibleToSatisfy(const NamedDecl *ND) const {
1635 return false;
1636}
1637
1638/// Determines whether the given declaration is an Objective-C
1639/// instance variable.
1640bool ResultBuilder::IsObjCIvar(const NamedDecl *ND) const {
1641 return isa<ObjCIvarDecl>(ND);
1642}
1643
1644namespace {
1645
1646/// Visible declaration consumer that adds a code-completion result
1647/// for each visible declaration.
1648class CodeCompletionDeclConsumer : public VisibleDeclConsumer {
1649 ResultBuilder &Results;
1650 DeclContext *InitialLookupCtx;
1651 // NamingClass and BaseType are used for access-checking. See
1652 // Sema::IsSimplyAccessible for details.
1653 CXXRecordDecl *NamingClass;
1654 QualType BaseType;
1655 std::vector<FixItHint> FixIts;
1656
1657public:
1658 CodeCompletionDeclConsumer(
1659 ResultBuilder &Results, DeclContext *InitialLookupCtx,
1660 QualType BaseType = QualType(),
1661 std::vector<FixItHint> FixIts = std::vector<FixItHint>())
1662 : Results(Results), InitialLookupCtx(InitialLookupCtx),
1663 FixIts(std::move(FixIts)) {
1664 NamingClass = llvm::dyn_cast<CXXRecordDecl>(InitialLookupCtx);
1665 // If BaseType was not provided explicitly, emulate implicit 'this->'.
1666 if (BaseType.isNull()) {
1667 auto ThisType = Results.getSema().getCurrentThisType();
1668 if (!ThisType.isNull()) {
1669 assert(ThisType->isPointerType());
1670 BaseType = ThisType->getPointeeType();
1671 if (!NamingClass)
1672 NamingClass = BaseType->getAsCXXRecordDecl();
1673 }
1674 }
1675 this->BaseType = BaseType;
1676 }
1677
1678 void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
1679 bool InBaseClass) override {
1680 ResultBuilder::Result Result(ND, Results.getBasePriority(ND), nullptr,
1681 false, IsAccessible(ND, Ctx), FixIts);
1682 Results.AddResult(Result, InitialLookupCtx, Hiding, InBaseClass);
1683 }
1684
1685 void EnteredContext(DeclContext *Ctx) override {
1686 Results.addVisitedContext(Ctx);
1687 }
1688
1689private:
1690 bool IsAccessible(NamedDecl *ND, DeclContext *Ctx) {
1691 // Naming class to use for access check. In most cases it was provided
1692 // explicitly (e.g. member access (lhs.foo) or qualified lookup (X::)),
1693 // for unqualified lookup we fallback to the \p Ctx in which we found the
1694 // member.
1695 auto *NamingClass = this->NamingClass;
1696 QualType BaseType = this->BaseType;
1697 if (auto *Cls = llvm::dyn_cast_or_null<CXXRecordDecl>(Ctx)) {
1698 if (!NamingClass)
1699 NamingClass = Cls;
1700 // When we emulate implicit 'this->' in an unqualified lookup, we might
1701 // end up with an invalid naming class. In that case, we avoid emulating
1702 // 'this->' qualifier to satisfy preconditions of the access checking.
1703 if (NamingClass->getCanonicalDecl() != Cls->getCanonicalDecl() &&
1704 !NamingClass->isDerivedFrom(Cls)) {
1705 NamingClass = Cls;
1706 BaseType = QualType();
1707 }
1708 } else {
1709 // The decl was found outside the C++ class, so only ObjC access checks
1710 // apply. Those do not rely on NamingClass and BaseType, so we clear them
1711 // out.
1712 NamingClass = nullptr;
1713 BaseType = QualType();
1714 }
1715 return Results.getSema().IsSimplyAccessible(ND, NamingClass, BaseType);
1716 }
1717};
1718} // namespace
1719
1720/// Add type specifiers for the current language as keyword results.
1721static void AddTypeSpecifierResults(const LangOptions &LangOpts,
1722 ResultBuilder &Results) {
1724 Results.AddResult(Result("short", CCP_Type));
1725 Results.AddResult(Result("long", CCP_Type));
1726 Results.AddResult(Result("signed", CCP_Type));
1727 Results.AddResult(Result("unsigned", CCP_Type));
1728 Results.AddResult(Result("void", CCP_Type));
1729 Results.AddResult(Result("char", CCP_Type));
1730 Results.AddResult(Result("int", CCP_Type));
1731 Results.AddResult(Result("float", CCP_Type));
1732 Results.AddResult(Result("double", CCP_Type));
1733 Results.AddResult(Result("enum", CCP_Type));
1734 Results.AddResult(Result("struct", CCP_Type));
1735 Results.AddResult(Result("union", CCP_Type));
1736 Results.AddResult(Result("const", CCP_Type));
1737 Results.AddResult(Result("volatile", CCP_Type));
1738
1739 if (LangOpts.C99) {
1740 // C99-specific
1741 Results.AddResult(Result("_Complex", CCP_Type));
1742 Results.AddResult(Result("_Imaginary", CCP_Type));
1743 Results.AddResult(Result("_Bool", CCP_Type));
1744 Results.AddResult(Result("restrict", CCP_Type));
1745 }
1746
1747 CodeCompletionBuilder Builder(Results.getAllocator(),
1748 Results.getCodeCompletionTUInfo());
1749 if (LangOpts.CPlusPlus) {
1750 // C++-specific
1751 Results.AddResult(
1752 Result("bool", CCP_Type + (LangOpts.ObjC ? CCD_bool_in_ObjC : 0)));
1753 Results.AddResult(Result("class", CCP_Type));
1754 Results.AddResult(Result("wchar_t", CCP_Type));
1755
1756 // typename name
1757 Builder.AddTypedTextChunk("typename");
1759 Builder.AddPlaceholderChunk("name");
1760 Results.AddResult(Result(Builder.TakeString()));
1761
1762 if (LangOpts.CPlusPlus11) {
1763 Results.AddResult(Result("auto", CCP_Type));
1764 Results.AddResult(Result("char16_t", CCP_Type));
1765 Results.AddResult(Result("char32_t", CCP_Type));
1766
1767 Builder.AddTypedTextChunk("decltype");
1768 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1769 Builder.AddPlaceholderChunk("expression");
1770 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1771 Results.AddResult(Result(Builder.TakeString()));
1772 }
1773 } else
1774 Results.AddResult(Result("__auto_type", CCP_Type));
1775
1776 // GNU keywords
1777 if (LangOpts.GNUKeywords) {
1778 // FIXME: Enable when we actually support decimal floating point.
1779 // Results.AddResult(Result("_Decimal32"));
1780 // Results.AddResult(Result("_Decimal64"));
1781 // Results.AddResult(Result("_Decimal128"));
1782
1783 Builder.AddTypedTextChunk("typeof");
1785 Builder.AddPlaceholderChunk("expression");
1786 Results.AddResult(Result(Builder.TakeString()));
1787
1788 Builder.AddTypedTextChunk("typeof");
1789 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1790 Builder.AddPlaceholderChunk("type");
1791 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1792 Results.AddResult(Result(Builder.TakeString()));
1793 }
1794
1795 // Nullability
1796 Results.AddResult(Result("_Nonnull", CCP_Type));
1797 Results.AddResult(Result("_Null_unspecified", CCP_Type));
1798 Results.AddResult(Result("_Nullable", CCP_Type));
1799}
1800
1802 const LangOptions &LangOpts,
1803 ResultBuilder &Results) {
1805 // Note: we don't suggest either "auto" or "register", because both
1806 // are pointless as storage specifiers. Elsewhere, we suggest "auto"
1807 // in C++0x as a type specifier.
1808 Results.AddResult(Result("extern"));
1809 Results.AddResult(Result("static"));
1810
1811 if (LangOpts.CPlusPlus11) {
1812 CodeCompletionAllocator &Allocator = Results.getAllocator();
1813 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
1814
1815 // alignas
1816 Builder.AddTypedTextChunk("alignas");
1817 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1818 Builder.AddPlaceholderChunk("expression");
1819 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1820 Results.AddResult(Result(Builder.TakeString()));
1821
1822 Results.AddResult(Result("constexpr"));
1823 Results.AddResult(Result("thread_local"));
1824 }
1825}
1826
1828 const LangOptions &LangOpts,
1829 ResultBuilder &Results) {
1831 switch (CCC) {
1832 case Sema::PCC_Class:
1834 if (LangOpts.CPlusPlus) {
1835 Results.AddResult(Result("explicit"));
1836 Results.AddResult(Result("friend"));
1837 Results.AddResult(Result("mutable"));
1838 Results.AddResult(Result("virtual"));
1839 }
1840 [[fallthrough]];
1841
1845 case Sema::PCC_Template:
1846 if (LangOpts.CPlusPlus || LangOpts.C99)
1847 Results.AddResult(Result("inline"));
1848 break;
1849
1853 case Sema::PCC_ForInit:
1856 case Sema::PCC_Type:
1859 break;
1860 }
1861}
1862
1863static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt);
1864static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt);
1865static void AddObjCVisibilityResults(const LangOptions &LangOpts,
1866 ResultBuilder &Results, bool NeedAt);
1867static void AddObjCImplementationResults(const LangOptions &LangOpts,
1868 ResultBuilder &Results, bool NeedAt);
1869static void AddObjCInterfaceResults(const LangOptions &LangOpts,
1870 ResultBuilder &Results, bool NeedAt);
1871static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt);
1872
1873static void AddTypedefResult(ResultBuilder &Results) {
1874 CodeCompletionBuilder Builder(Results.getAllocator(),
1875 Results.getCodeCompletionTUInfo());
1876 Builder.AddTypedTextChunk("typedef");
1878 Builder.AddPlaceholderChunk("type");
1880 Builder.AddPlaceholderChunk("name");
1881 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1882 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
1883}
1884
1885// using name = type
1887 ResultBuilder &Results) {
1888 Builder.AddTypedTextChunk("using");
1890 Builder.AddPlaceholderChunk("name");
1891 Builder.AddChunk(CodeCompletionString::CK_Equal);
1892 Builder.AddPlaceholderChunk("type");
1893 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1894 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
1895}
1896
1898 const LangOptions &LangOpts) {
1899 switch (CCC) {
1901 case Sema::PCC_Class:
1903 case Sema::PCC_Template:
1907 case Sema::PCC_Type:
1910 return true;
1911
1914 return LangOpts.CPlusPlus;
1915
1918 return false;
1919
1920 case Sema::PCC_ForInit:
1921 return LangOpts.CPlusPlus || LangOpts.ObjC || LangOpts.C99;
1922 }
1923
1924 llvm_unreachable("Invalid ParserCompletionContext!");
1925}
1926
1928 const Preprocessor &PP) {
1929 PrintingPolicy Policy = Sema::getPrintingPolicy(Context, PP);
1930 Policy.AnonymousTagLocations = false;
1931 Policy.SuppressStrongLifetime = true;
1932 Policy.SuppressUnwrittenScope = true;
1933 Policy.SuppressScope = true;
1934 Policy.CleanUglifiedParameters = true;
1935 return Policy;
1936}
1937
1938/// Retrieve a printing policy suitable for code completion.
1941}
1942
1943/// Retrieve the string representation of the given type as a string
1944/// that has the appropriate lifetime for code completion.
1945///
1946/// This routine provides a fast path where we provide constant strings for
1947/// common type names.
1948static const char *GetCompletionTypeString(QualType T, ASTContext &Context,
1949 const PrintingPolicy &Policy,
1950 CodeCompletionAllocator &Allocator) {
1951 if (!T.getLocalQualifiers()) {
1952 // Built-in type names are constant strings.
1953 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T))
1954 return BT->getNameAsCString(Policy);
1955
1956 // Anonymous tag types are constant strings.
1957 if (const TagType *TagT = dyn_cast<TagType>(T))
1958 if (TagDecl *Tag = TagT->getDecl())
1959 if (!Tag->hasNameForLinkage()) {
1960 switch (Tag->getTagKind()) {
1961 case TTK_Struct:
1962 return "struct <anonymous>";
1963 case TTK_Interface:
1964 return "__interface <anonymous>";
1965 case TTK_Class:
1966 return "class <anonymous>";
1967 case TTK_Union:
1968 return "union <anonymous>";
1969 case TTK_Enum:
1970 return "enum <anonymous>";
1971 }
1972 }
1973 }
1974
1975 // Slow path: format the type as a string.
1976 std::string Result;
1977 T.getAsStringInternal(Result, Policy);
1978 return Allocator.CopyString(Result);
1979}
1980
1981/// Add a completion for "this", if we're in a member function.
1982static void addThisCompletion(Sema &S, ResultBuilder &Results) {
1983 QualType ThisTy = S.getCurrentThisType();
1984 if (ThisTy.isNull())
1985 return;
1986
1987 CodeCompletionAllocator &Allocator = Results.getAllocator();
1988 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
1990 Builder.AddResultTypeChunk(
1991 GetCompletionTypeString(ThisTy, S.Context, Policy, Allocator));
1992 Builder.AddTypedTextChunk("this");
1993 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
1994}
1995
1997 ResultBuilder &Results,
1998 const LangOptions &LangOpts) {
1999 if (!LangOpts.CPlusPlus11)
2000 return;
2001
2002 Builder.AddTypedTextChunk("static_assert");
2003 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2004 Builder.AddPlaceholderChunk("expression");
2005 Builder.AddChunk(CodeCompletionString::CK_Comma);
2006 Builder.AddPlaceholderChunk("message");
2007 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2008 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2009 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
2010}
2011
2012static void AddOverrideResults(ResultBuilder &Results,
2013 const CodeCompletionContext &CCContext,
2014 CodeCompletionBuilder &Builder) {
2015 Sema &S = Results.getSema();
2016 const auto *CR = llvm::dyn_cast<CXXRecordDecl>(S.CurContext);
2017 // If not inside a class/struct/union return empty.
2018 if (!CR)
2019 return;
2020 // First store overrides within current class.
2021 // These are stored by name to make querying fast in the later step.
2022 llvm::StringMap<std::vector<FunctionDecl *>> Overrides;
2023 for (auto *Method : CR->methods()) {
2024 if (!Method->isVirtual() || !Method->getIdentifier())
2025 continue;
2026 Overrides[Method->getName()].push_back(Method);
2027 }
2028
2029 for (const auto &Base : CR->bases()) {
2030 const auto *BR = Base.getType().getTypePtr()->getAsCXXRecordDecl();
2031 if (!BR)
2032 continue;
2033 for (auto *Method : BR->methods()) {
2034 if (!Method->isVirtual() || !Method->getIdentifier())
2035 continue;
2036 const auto it = Overrides.find(Method->getName());
2037 bool IsOverriden = false;
2038 if (it != Overrides.end()) {
2039 for (auto *MD : it->second) {
2040 // If the method in current body is not an overload of this virtual
2041 // function, then it overrides this one.
2042 if (!S.IsOverload(MD, Method, false)) {
2043 IsOverriden = true;
2044 break;
2045 }
2046 }
2047 }
2048 if (!IsOverriden) {
2049 // Generates a new CodeCompletionResult by taking this function and
2050 // converting it into an override declaration with only one chunk in the
2051 // final CodeCompletionString as a TypedTextChunk.
2052 std::string OverrideSignature;
2053 llvm::raw_string_ostream OS(OverrideSignature);
2054 CodeCompletionResult CCR(Method, 0);
2055 PrintingPolicy Policy =
2058 S.getPreprocessor(), S.getASTContext(), Builder,
2059 /*IncludeBriefComments=*/false, CCContext, Policy);
2060 Results.AddResult(CodeCompletionResult(CCS, Method, CCP_CodePattern));
2061 }
2062 }
2063 }
2064}
2065
2066/// Add language constructs that show up for "ordinary" names.
2068 Sema &SemaRef, ResultBuilder &Results) {
2069 CodeCompletionAllocator &Allocator = Results.getAllocator();
2070 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
2071
2073 switch (CCC) {
2075 if (SemaRef.getLangOpts().CPlusPlus) {
2076 if (Results.includeCodePatterns()) {
2077 // namespace <identifier> { declarations }
2078 Builder.AddTypedTextChunk("namespace");
2080 Builder.AddPlaceholderChunk("identifier");
2082 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
2084 Builder.AddPlaceholderChunk("declarations");
2086 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
2087 Results.AddResult(Result(Builder.TakeString()));
2088 }
2089
2090 // namespace identifier = identifier ;
2091 Builder.AddTypedTextChunk("namespace");
2093 Builder.AddPlaceholderChunk("name");
2094 Builder.AddChunk(CodeCompletionString::CK_Equal);
2095 Builder.AddPlaceholderChunk("namespace");
2096 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2097 Results.AddResult(Result(Builder.TakeString()));
2098
2099 // Using directives
2100 Builder.AddTypedTextChunk("using namespace");
2102 Builder.AddPlaceholderChunk("identifier");
2103 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2104 Results.AddResult(Result(Builder.TakeString()));
2105
2106 // asm(string-literal)
2107 Builder.AddTypedTextChunk("asm");
2108 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2109 Builder.AddPlaceholderChunk("string-literal");
2110 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2111 Results.AddResult(Result(Builder.TakeString()));
2112
2113 if (Results.includeCodePatterns()) {
2114 // Explicit template instantiation
2115 Builder.AddTypedTextChunk("template");
2117 Builder.AddPlaceholderChunk("declaration");
2118 Results.AddResult(Result(Builder.TakeString()));
2119 } else {
2120 Results.AddResult(Result("template", CodeCompletionResult::RK_Keyword));
2121 }
2122 }
2123
2124 if (SemaRef.getLangOpts().ObjC)
2125 AddObjCTopLevelResults(Results, true);
2126
2127 AddTypedefResult(Results);
2128 [[fallthrough]];
2129
2130 case Sema::PCC_Class:
2131 if (SemaRef.getLangOpts().CPlusPlus) {
2132 // Using declaration
2133 Builder.AddTypedTextChunk("using");
2135 Builder.AddPlaceholderChunk("qualifier");
2136 Builder.AddTextChunk("::");
2137 Builder.AddPlaceholderChunk("name");
2138 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2139 Results.AddResult(Result(Builder.TakeString()));
2140
2141 if (SemaRef.getLangOpts().CPlusPlus11)
2142 AddUsingAliasResult(Builder, Results);
2143
2144 // using typename qualifier::name (only in a dependent context)
2145 if (SemaRef.CurContext->isDependentContext()) {
2146 Builder.AddTypedTextChunk("using typename");
2148 Builder.AddPlaceholderChunk("qualifier");
2149 Builder.AddTextChunk("::");
2150 Builder.AddPlaceholderChunk("name");
2151 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2152 Results.AddResult(Result(Builder.TakeString()));
2153 }
2154
2155 AddStaticAssertResult(Builder, Results, SemaRef.getLangOpts());
2156
2157 if (CCC == Sema::PCC_Class) {
2158 AddTypedefResult(Results);
2159
2160 bool IsNotInheritanceScope = !S->isClassInheritanceScope();
2161 // public:
2162 Builder.AddTypedTextChunk("public");
2163 if (IsNotInheritanceScope && Results.includeCodePatterns())
2164 Builder.AddChunk(CodeCompletionString::CK_Colon);
2165 Results.AddResult(Result(Builder.TakeString()));
2166
2167 // protected:
2168 Builder.AddTypedTextChunk("protected");
2169 if (IsNotInheritanceScope && Results.includeCodePatterns())
2170 Builder.AddChunk(CodeCompletionString::CK_Colon);
2171 Results.AddResult(Result(Builder.TakeString()));
2172
2173 // private:
2174 Builder.AddTypedTextChunk("private");
2175 if (IsNotInheritanceScope && Results.includeCodePatterns())
2176 Builder.AddChunk(CodeCompletionString::CK_Colon);
2177 Results.AddResult(Result(Builder.TakeString()));
2178
2179 // FIXME: This adds override results only if we are at the first word of
2180 // the declaration/definition. Also call this from other sides to have
2181 // more use-cases.
2183 Builder);
2184 }
2185 }
2186 [[fallthrough]];
2187
2188 case Sema::PCC_Template:
2190 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns()) {
2191 // template < parameters >
2192 Builder.AddTypedTextChunk("template");
2193 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
2194 Builder.AddPlaceholderChunk("parameters");
2195 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
2196 Results.AddResult(Result(Builder.TakeString()));
2197 } else {
2198 Results.AddResult(Result("template", CodeCompletionResult::RK_Keyword));
2199 }
2200
2201 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
2202 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
2203 break;
2204
2206 AddObjCInterfaceResults(SemaRef.getLangOpts(), Results, true);
2207 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
2208 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
2209 break;
2210
2212 AddObjCImplementationResults(SemaRef.getLangOpts(), Results, true);
2213 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
2214 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
2215 break;
2216
2218 AddObjCVisibilityResults(SemaRef.getLangOpts(), Results, true);
2219 break;
2220
2222 case Sema::PCC_Statement: {
2223 if (SemaRef.getLangOpts().CPlusPlus11)
2224 AddUsingAliasResult(Builder, Results);
2225
2226 AddTypedefResult(Results);
2227
2228 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns() &&
2229 SemaRef.getLangOpts().CXXExceptions) {
2230 Builder.AddTypedTextChunk("try");
2232 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
2234 Builder.AddPlaceholderChunk("statements");
2236 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
2238 Builder.AddTextChunk("catch");
2240 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2241 Builder.AddPlaceholderChunk("declaration");
2242 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2244 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
2246 Builder.AddPlaceholderChunk("statements");
2248 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
2249 Results.AddResult(Result(Builder.TakeString()));
2250 }
2251 if (SemaRef.getLangOpts().ObjC)
2252 AddObjCStatementResults(Results, true);
2253
2254 if (Results.includeCodePatterns()) {
2255 // if (condition) { statements }
2256 Builder.AddTypedTextChunk("if");
2258 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2259 if (SemaRef.getLangOpts().CPlusPlus)
2260 Builder.AddPlaceholderChunk("condition");
2261 else
2262 Builder.AddPlaceholderChunk("expression");
2263 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2265 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
2267 Builder.AddPlaceholderChunk("statements");
2269 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
2270 Results.AddResult(Result(Builder.TakeString()));
2271
2272 // switch (condition) { }
2273 Builder.AddTypedTextChunk("switch");
2275 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2276 if (SemaRef.getLangOpts().CPlusPlus)
2277 Builder.AddPlaceholderChunk("condition");
2278 else
2279 Builder.AddPlaceholderChunk("expression");
2280 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2282 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
2284 Builder.AddPlaceholderChunk("cases");
2286 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
2287 Results.AddResult(Result(Builder.TakeString()));
2288 }
2289
2290 // Switch-specific statements.
2291 if (SemaRef.getCurFunction() &&
2292 !SemaRef.getCurFunction()->SwitchStack.empty()) {
2293 // case expression:
2294 Builder.AddTypedTextChunk("case");
2296 Builder.AddPlaceholderChunk("expression");
2297 Builder.AddChunk(CodeCompletionString::CK_Colon);
2298 Results.AddResult(Result(Builder.TakeString()));
2299
2300 // default:
2301 Builder.AddTypedTextChunk("default");
2302 Builder.AddChunk(CodeCompletionString::CK_Colon);
2303 Results.AddResult(Result(Builder.TakeString()));
2304 }
2305
2306 if (Results.includeCodePatterns()) {
2307 /// while (condition) { statements }
2308 Builder.AddTypedTextChunk("while");
2310 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2311 if (SemaRef.getLangOpts().CPlusPlus)
2312 Builder.AddPlaceholderChunk("condition");
2313 else
2314 Builder.AddPlaceholderChunk("expression");
2315 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2317 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
2319 Builder.AddPlaceholderChunk("statements");
2321 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
2322 Results.AddResult(Result(Builder.TakeString()));
2323
2324 // do { statements } while ( expression );
2325 Builder.AddTypedTextChunk("do");
2327 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
2329 Builder.AddPlaceholderChunk("statements");
2331 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
2332 Builder.AddTextChunk("while");
2334 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2335 Builder.AddPlaceholderChunk("expression");
2336 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2337 Results.AddResult(Result(Builder.TakeString()));
2338
2339 // for ( for-init-statement ; condition ; expression ) { statements }
2340 Builder.AddTypedTextChunk("for");
2342 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2343 if (SemaRef.getLangOpts().CPlusPlus || SemaRef.getLangOpts().C99)
2344 Builder.AddPlaceholderChunk("init-statement");
2345 else
2346 Builder.AddPlaceholderChunk("init-expression");
2347 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2349 Builder.AddPlaceholderChunk("condition");
2350 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2352 Builder.AddPlaceholderChunk("inc-expression");
2353 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2355 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
2357 Builder.AddPlaceholderChunk("statements");
2359 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
2360 Results.AddResult(Result(Builder.TakeString()));
2361
2362 if (SemaRef.getLangOpts().CPlusPlus11 || SemaRef.getLangOpts().ObjC) {
2363 // for ( range_declaration (:|in) range_expression ) { statements }
2364 Builder.AddTypedTextChunk("for");
2366 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2367 Builder.AddPlaceholderChunk("range-declaration");
2369 if (SemaRef.getLangOpts().ObjC)
2370 Builder.AddTextChunk("in");
2371 else
2372 Builder.AddChunk(CodeCompletionString::CK_Colon);
2374 Builder.AddPlaceholderChunk("range-expression");
2375 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2377 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
2379 Builder.AddPlaceholderChunk("statements");
2381 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
2382 Results.AddResult(Result(Builder.TakeString()));
2383 }
2384 }
2385
2386 if (S->getContinueParent()) {
2387 // continue ;
2388 Builder.AddTypedTextChunk("continue");
2389 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2390 Results.AddResult(Result(Builder.TakeString()));
2391 }
2392
2393 if (S->getBreakParent()) {
2394 // break ;
2395 Builder.AddTypedTextChunk("break");
2396 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2397 Results.AddResult(Result(Builder.TakeString()));
2398 }
2399
2400 // "return expression ;" or "return ;", depending on the return type.
2401 QualType ReturnType;
2402 if (const auto *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext))
2403 ReturnType = Function->getReturnType();
2404 else if (const auto *Method = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext))
2405 ReturnType = Method->getReturnType();
2406 else if (SemaRef.getCurBlock() &&
2407 !SemaRef.getCurBlock()->ReturnType.isNull())
2408 ReturnType = SemaRef.getCurBlock()->ReturnType;;
2409 if (ReturnType.isNull() || ReturnType->isVoidType()) {
2410 Builder.AddTypedTextChunk("return");
2411 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2412 Results.AddResult(Result(Builder.TakeString()));
2413 } else {
2414 assert(!ReturnType.isNull());
2415 // "return expression ;"
2416 Builder.AddTypedTextChunk("return");
2418 Builder.AddPlaceholderChunk("expression");
2419 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2420 Results.AddResult(Result(Builder.TakeString()));
2421 // When boolean, also add 'return true;' and 'return false;'.
2422 if (ReturnType->isBooleanType()) {
2423 Builder.AddTypedTextChunk("return true");
2424 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2425 Results.AddResult(Result(Builder.TakeString()));
2426
2427 Builder.AddTypedTextChunk("return false");
2428 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2429 Results.AddResult(Result(Builder.TakeString()));
2430 }
2431 // For pointers, suggest 'return nullptr' in C++.
2432 if (SemaRef.getLangOpts().CPlusPlus11 &&
2433 (ReturnType->isPointerType() || ReturnType->isMemberPointerType())) {
2434 Builder.AddTypedTextChunk("return nullptr");
2435 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2436 Results.AddResult(Result(Builder.TakeString()));
2437 }
2438 }
2439
2440 // goto identifier ;
2441 Builder.AddTypedTextChunk("goto");
2443 Builder.AddPlaceholderChunk("label");
2444 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2445 Results.AddResult(Result(Builder.TakeString()));
2446
2447 // Using directives
2448 Builder.AddTypedTextChunk("using namespace");
2450 Builder.AddPlaceholderChunk("identifier");
2451 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
2452 Results.AddResult(Result(Builder.TakeString()));
2453
2454 AddStaticAssertResult(Builder, Results, SemaRef.getLangOpts());
2455 }
2456 [[fallthrough]];
2457
2458 // Fall through (for statement expressions).
2459 case Sema::PCC_ForInit:
2461 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
2462 // Fall through: conditions and statements can have expressions.
2463 [[fallthrough]];
2464
2466 if (SemaRef.getLangOpts().ObjCAutoRefCount &&
2468 // (__bridge <type>)<expression>
2469 Builder.AddTypedTextChunk("__bridge");
2471 Builder.AddPlaceholderChunk("type");
2472 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2473 Builder.AddPlaceholderChunk("expression");
2474 Results.AddResult(Result(Builder.TakeString()));
2475
2476 // (__bridge_transfer <Objective-C type>)<expression>
2477 Builder.AddTypedTextChunk("__bridge_transfer");
2479 Builder.AddPlaceholderChunk("Objective-C type");
2480 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2481 Builder.AddPlaceholderChunk("expression");
2482 Results.AddResult(Result(Builder.TakeString()));
2483
2484 // (__bridge_retained <CF type>)<expression>
2485 Builder.AddTypedTextChunk("__bridge_retained");
2487 Builder.AddPlaceholderChunk("CF type");
2488 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2489 Builder.AddPlaceholderChunk("expression");
2490 Results.AddResult(Result(Builder.TakeString()));
2491 }
2492 // Fall through
2493 [[fallthrough]];
2494
2495 case Sema::PCC_Expression: {
2496 if (SemaRef.getLangOpts().CPlusPlus) {
2497 // 'this', if we're in a non-static member function.
2498 addThisCompletion(SemaRef, Results);
2499
2500 // true
2501 Builder.AddResultTypeChunk("bool");
2502 Builder.AddTypedTextChunk("true");
2503 Results.AddResult(Result(Builder.TakeString()));
2504
2505 // false
2506 Builder.AddResultTypeChunk("bool");
2507 Builder.AddTypedTextChunk("false");
2508 Results.AddResult(Result(Builder.TakeString()));
2509
2510 if (SemaRef.getLangOpts().RTTI) {
2511 // dynamic_cast < type-id > ( expression )
2512 Builder.AddTypedTextChunk("dynamic_cast");
2513 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
2514 Builder.AddPlaceholderChunk("type");
2515 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
2516 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2517 Builder.AddPlaceholderChunk("expression");
2518 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2519 Results.AddResult(Result(Builder.TakeString()));
2520 }
2521
2522 // static_cast < type-id > ( expression )
2523 Builder.AddTypedTextChunk("static_cast");
2524 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
2525 Builder.AddPlaceholderChunk("type");
2526 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
2527 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2528 Builder.AddPlaceholderChunk("expression");
2529 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2530 Results.AddResult(Result(Builder.TakeString()));
2531
2532 // reinterpret_cast < type-id > ( expression )
2533 Builder.AddTypedTextChunk("reinterpret_cast");
2534 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
2535 Builder.AddPlaceholderChunk("type");
2536 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
2537 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2538 Builder.AddPlaceholderChunk("expression");
2539 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2540 Results.AddResult(Result(Builder.TakeString()));
2541
2542 // const_cast < type-id > ( expression )
2543 Builder.AddTypedTextChunk("const_cast");
2544 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
2545 Builder.AddPlaceholderChunk("type");
2546 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
2547 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2548 Builder.AddPlaceholderChunk("expression");
2549 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2550 Results.AddResult(Result(Builder.TakeString()));
2551
2552 if (SemaRef.getLangOpts().RTTI) {
2553 // typeid ( expression-or-type )
2554 Builder.AddResultTypeChunk("std::type_info");
2555 Builder.AddTypedTextChunk("typeid");
2556 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2557 Builder.AddPlaceholderChunk("expression-or-type");
2558 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2559 Results.AddResult(Result(Builder.TakeString()));
2560 }
2561
2562 // new T ( ... )
2563 Builder.AddTypedTextChunk("new");
2565 Builder.AddPlaceholderChunk("type");
2566 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2567 Builder.AddPlaceholderChunk("expressions");
2568 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2569 Results.AddResult(Result(Builder.TakeString()));
2570
2571 // new T [ ] ( ... )
2572 Builder.AddTypedTextChunk("new");
2574 Builder.AddPlaceholderChunk("type");
2575 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
2576 Builder.AddPlaceholderChunk("size");
2577 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
2578 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2579 Builder.AddPlaceholderChunk("expressions");
2580 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2581 Results.AddResult(Result(Builder.TakeString()));
2582
2583 // delete expression
2584 Builder.AddResultTypeChunk("void");
2585 Builder.AddTypedTextChunk("delete");
2587 Builder.AddPlaceholderChunk("expression");
2588 Results.AddResult(Result(Builder.TakeString()));
2589
2590 // delete [] expression
2591 Builder.AddResultTypeChunk("void");
2592 Builder.AddTypedTextChunk("delete");
2594 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
2595 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
2597 Builder.AddPlaceholderChunk("expression");
2598 Results.AddResult(Result(Builder.TakeString()));
2599
2600 if (SemaRef.getLangOpts().CXXExceptions) {
2601 // throw expression
2602 Builder.AddResultTypeChunk("void");
2603 Builder.AddTypedTextChunk("throw");
2605 Builder.AddPlaceholderChunk("expression");
2606 Results.AddResult(Result(Builder.TakeString()));
2607 }
2608
2609 // FIXME: Rethrow?
2610
2611 if (SemaRef.getLangOpts().CPlusPlus11) {
2612 // nullptr
2613 Builder.AddResultTypeChunk("std::nullptr_t");
2614 Builder.AddTypedTextChunk("nullptr");
2615 Results.AddResult(Result(Builder.TakeString()));
2616
2617 // alignof
2618 Builder.AddResultTypeChunk("size_t");
2619 Builder.AddTypedTextChunk("alignof");
2620 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2621 Builder.AddPlaceholderChunk("type");
2622 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2623 Results.AddResult(Result(Builder.TakeString()));
2624
2625 // noexcept
2626 Builder.AddResultTypeChunk("bool");
2627 Builder.AddTypedTextChunk("noexcept");
2628 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2629 Builder.AddPlaceholderChunk("expression");
2630 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2631 Results.AddResult(Result(Builder.TakeString()));
2632
2633 // sizeof... expression
2634 Builder.AddResultTypeChunk("size_t");
2635 Builder.AddTypedTextChunk("sizeof...");
2636 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2637 Builder.AddPlaceholderChunk("parameter-pack");
2638 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2639 Results.AddResult(Result(Builder.TakeString()));
2640 }
2641 }
2642
2643 if (SemaRef.getLangOpts().ObjC) {
2644 // Add "super", if we're in an Objective-C class with a superclass.
2645 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) {
2646 // The interface can be NULL.
2647 if (ObjCInterfaceDecl *ID = Method->getClassInterface())
2648 if (ID->getSuperClass()) {
2649 std::string SuperType;
2650 SuperType = ID->getSuperClass()->getNameAsString();
2651 if (Method->isInstanceMethod())
2652 SuperType += " *";
2653
2654 Builder.AddResultTypeChunk(Allocator.CopyString(SuperType));
2655 Builder.AddTypedTextChunk("super");
2656 Results.AddResult(Result(Builder.TakeString()));
2657 }
2658 }
2659
2660 AddObjCExpressionResults(Results, true);
2661 }
2662
2663 if (SemaRef.getLangOpts().C11) {
2664 // _Alignof
2665 Builder.AddResultTypeChunk("size_t");
2666 if (SemaRef.PP.isMacroDefined("alignof"))
2667 Builder.AddTypedTextChunk("alignof");
2668 else
2669 Builder.AddTypedTextChunk("_Alignof");
2670 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2671 Builder.AddPlaceholderChunk("type");
2672 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2673 Results.AddResult(Result(Builder.TakeString()));
2674 }
2675
2676 if (SemaRef.getLangOpts().C2x) {
2677 // nullptr
2678 Builder.AddResultTypeChunk("nullptr_t");
2679 Builder.AddTypedTextChunk("nullptr");
2680 Results.AddResult(Result(Builder.TakeString()));
2681 }
2682
2683 // sizeof expression
2684 Builder.AddResultTypeChunk("size_t");
2685 Builder.AddTypedTextChunk("sizeof");
2686 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2687 Builder.AddPlaceholderChunk("expression-or-type");
2688 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2689 Results.AddResult(Result(Builder.TakeString()));
2690 break;
2691 }
2692
2693 case Sema::PCC_Type:
2695 break;
2696 }
2697
2698 if (WantTypesInContext(CCC, SemaRef.getLangOpts()))
2699 AddTypeSpecifierResults(SemaRef.getLangOpts(), Results);
2700
2701 if (SemaRef.getLangOpts().CPlusPlus && CCC != Sema::PCC_Type)
2702 Results.AddResult(Result("operator"));
2703}
2704
2705/// If the given declaration has an associated type, add it as a result
2706/// type chunk.
2707static void AddResultTypeChunk(ASTContext &Context,
2708 const PrintingPolicy &Policy,
2709 const NamedDecl *ND, QualType BaseType,
2711 if (!ND)
2712 return;
2713
2714 // Skip constructors and conversion functions, which have their return types
2715 // built into their names.
2716 if (isConstructor(ND) || isa<CXXConversionDecl>(ND))
2717 return;
2718
2719 // Determine the type of the declaration (if it has a type).
2720 QualType T;
2721 if (const FunctionDecl *Function = ND->getAsFunction())
2722 T = Function->getReturnType();
2723 else if (const auto *Method = dyn_cast<ObjCMethodDecl>(ND)) {
2724 if (!BaseType.isNull())
2725 T = Method->getSendResultType(BaseType);
2726 else
2727 T = Method->getReturnType();
2728 } else if (const auto *Enumerator = dyn_cast<EnumConstantDecl>(ND)) {
2729 T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext()));
2731 } else if (isa<UnresolvedUsingValueDecl>(ND)) {
2732 /* Do nothing: ignore unresolved using declarations*/
2733 } else if (const auto *Ivar = dyn_cast<ObjCIvarDecl>(ND)) {
2734 if (!BaseType.isNull())
2735 T = Ivar->getUsageType(BaseType);
2736 else
2737 T = Ivar->getType();
2738 } else if (const auto *Value = dyn_cast<ValueDecl>(ND)) {
2739 T = Value->getType();
2740 } else if (const auto *Property = dyn_cast<ObjCPropertyDecl>(ND)) {
2741 if (!BaseType.isNull())
2742 T = Property->getUsageType(BaseType);
2743 else
2744 T = Property->getType();
2745 }
2746
2747 if (T.isNull() || Context.hasSameType(T, Context.DependentTy))
2748 return;
2749
2750 Result.AddResultTypeChunk(
2751 GetCompletionTypeString(T, Context, Policy, Result.getAllocator()));
2752}
2753
2755 const NamedDecl *FunctionOrMethod,
2757 if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>())
2758 if (Sentinel->getSentinel() == 0) {
2759 if (PP.getLangOpts().ObjC && PP.isMacroDefined("nil"))
2760 Result.AddTextChunk(", nil");
2761 else if (PP.isMacroDefined("NULL"))
2762 Result.AddTextChunk(", NULL");
2763 else
2764 Result.AddTextChunk(", (void*)0");
2765 }
2766}
2767
2768static std::string formatObjCParamQualifiers(unsigned ObjCQuals,
2769 QualType &Type) {
2770 std::string Result;
2771 if (ObjCQuals & Decl::OBJC_TQ_In)
2772 Result += "in ";
2773 else if (ObjCQuals & Decl::OBJC_TQ_Inout)
2774 Result += "inout ";
2775 else if (ObjCQuals & Decl::OBJC_TQ_Out)
2776 Result += "out ";
2777 if (ObjCQuals & Decl::OBJC_TQ_Bycopy)
2778 Result += "bycopy ";
2779 else if (ObjCQuals & Decl::OBJC_TQ_Byref)
2780 Result += "byref ";
2781 if (ObjCQuals & Decl::OBJC_TQ_Oneway)
2782 Result += "oneway ";
2783 if (ObjCQuals & Decl::OBJC_TQ_CSNullability) {
2784 if (auto nullability = AttributedType::stripOuterNullability(Type)) {
2785 switch (*nullability) {
2787 Result += "nonnull ";
2788 break;
2789
2791 Result += "nullable ";
2792 break;
2793
2795 Result += "null_unspecified ";
2796 break;
2797
2799 llvm_unreachable("Not supported as a context-sensitive keyword!");
2800 break;
2801 }
2802 }
2803 }
2804 return Result;
2805}
2806
2807/// Tries to find the most appropriate type location for an Objective-C
2808/// block placeholder.
2809///
2810/// This function ignores things like typedefs and qualifiers in order to
2811/// present the most relevant and accurate block placeholders in code completion
2812/// results.
2815 FunctionProtoTypeLoc &BlockProto,
2816 bool SuppressBlock = false) {
2817 if (!TSInfo)
2818 return;
2819 TypeLoc TL = TSInfo->getTypeLoc().getUnqualifiedLoc();
2820 while (true) {
2821 // Look through typedefs.
2822 if (!SuppressBlock) {
2823 if (TypedefTypeLoc TypedefTL = TL.getAsAdjusted<TypedefTypeLoc>()) {
2824 if (TypeSourceInfo *InnerTSInfo =
2825 TypedefTL.getTypedefNameDecl()->getTypeSourceInfo()) {
2826 TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc();
2827 continue;
2828 }
2829 }
2830
2831 // Look through qualified types
2832 if (QualifiedTypeLoc QualifiedTL = TL.getAs<QualifiedTypeLoc>()) {
2833 TL = QualifiedTL.getUnqualifiedLoc();
2834 continue;
2835 }
2836
2837 if (AttributedTypeLoc AttrTL = TL.getAs<AttributedTypeLoc>()) {
2838 TL = AttrTL.getModifiedLoc();
2839 continue;
2840 }
2841 }
2842
2843 // Try to get the function prototype behind the block pointer type,
2844 // then we're done.
2845 if (BlockPointerTypeLoc BlockPtr = TL.getAs<BlockPointerTypeLoc>()) {
2846 TL = BlockPtr.getPointeeLoc().IgnoreParens();
2847 Block = TL.getAs<FunctionTypeLoc>();
2848 BlockProto = TL.getAs<FunctionProtoTypeLoc>();
2849 }
2850 break;
2851 }
2852}
2853
2854static std::string formatBlockPlaceholder(
2855 const PrintingPolicy &Policy, const NamedDecl *BlockDecl,
2857 bool SuppressBlockName = false, bool SuppressBlock = false,
2858 std::optional<ArrayRef<QualType>> ObjCSubsts = std::nullopt);
2859
2860static std::string FormatFunctionParameter(
2861 const PrintingPolicy &Policy, const DeclaratorDecl *Param,
2862 bool SuppressName = false, bool SuppressBlock = false,
2863 std::optional<ArrayRef<QualType>> ObjCSubsts = std::nullopt) {
2864 // Params are unavailable in FunctionTypeLoc if the FunctionType is invalid.
2865 // It would be better to pass in the param Type, which is usually available.
2866 // But this case is rare, so just pretend we fell back to int as elsewhere.
2867 if (!Param)
2868 return "int";
2870 if (const auto *PVD = dyn_cast<ParmVarDecl>(Param))
2871 ObjCQual = PVD->getObjCDeclQualifier();
2872 bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext());
2873 if (Param->getType()->isDependentType() ||
2874 !Param->getType()->isBlockPointerType()) {
2875 // The argument for a dependent or non-block parameter is a placeholder
2876 // containing that parameter's type.
2877 std::string Result;
2878
2879 if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName)
2880 Result = std::string(Param->getIdentifier()->deuglifiedName());
2881
2882 QualType Type = Param->getType();
2883 if (ObjCSubsts)
2884 Type = Type.substObjCTypeArgs(Param->getASTContext(), *ObjCSubsts,
2886 if (ObjCMethodParam) {
2887 Result = "(" + formatObjCParamQualifiers(ObjCQual, Type);
2888 Result += Type.getAsString(Policy) + ")";
2889 if (Param->getIdentifier() && !SuppressName)
2890 Result += Param->getIdentifier()->deuglifiedName();
2891 } else {
2892 Type.getAsStringInternal(Result, Policy);
2893 }
2894 return Result;
2895 }
2896
2897 // The argument for a block pointer parameter is a block literal with
2898 // the appropriate type.
2900 FunctionProtoTypeLoc BlockProto;
2902 SuppressBlock);
2903 // Try to retrieve the block type information from the property if this is a
2904 // parameter in a setter.
2905 if (!Block && ObjCMethodParam &&
2906 cast<ObjCMethodDecl>(Param->getDeclContext())->isPropertyAccessor()) {
2907 if (const auto *PD = cast<ObjCMethodDecl>(Param->getDeclContext())
2908 ->findPropertyDecl(/*CheckOverrides=*/false))
2909 findTypeLocationForBlockDecl(PD->getTypeSourceInfo(), Block, BlockProto,
2910 SuppressBlock);
2911 }
2912
2913 if (!Block) {
2914 // We were unable to find a FunctionProtoTypeLoc with parameter names
2915 // for the block; just use the parameter type as a placeholder.
2916 std::string Result;
2917 if (!ObjCMethodParam && Param->getIdentifier())
2918 Result = std::string(Param->getIdentifier()->deuglifiedName());
2919
2921
2922 if (ObjCMethodParam) {
2923 Result = Type.getAsString(Policy);
2924 std::string Quals = formatObjCParamQualifiers(ObjCQual, Type);
2925 if (!Quals.empty())
2926 Result = "(" + Quals + " " + Result + ")";
2927 if (Result.back() != ')')
2928 Result += " ";
2929 if (Param->getIdentifier())
2930 Result += Param->getIdentifier()->deuglifiedName();
2931 } else {
2932 Type.getAsStringInternal(Result, Policy);
2933 }
2934
2935 return Result;
2936 }
2937
2938 // We have the function prototype behind the block pointer type, as it was
2939 // written in the source.
2940 return formatBlockPlaceholder(Policy, Param, Block, BlockProto,
2941 /*SuppressBlockName=*/false, SuppressBlock,
2942 ObjCSubsts);
2943}
2944
2945/// Returns a placeholder string that corresponds to an Objective-C block
2946/// declaration.
2947///
2948/// \param BlockDecl A declaration with an Objective-C block type.
2949///
2950/// \param Block The most relevant type location for that block type.
2951///
2952/// \param SuppressBlockName Determines whether or not the name of the block
2953/// declaration is included in the resulting string.
2954static std::string
2957 bool SuppressBlockName, bool SuppressBlock,
2958 std::optional<ArrayRef<QualType>> ObjCSubsts) {
2959 std::string Result;
2960 QualType ResultType = Block.getTypePtr()->getReturnType();
2961 if (ObjCSubsts)
2962 ResultType =
2963 ResultType.substObjCTypeArgs(BlockDecl->getASTContext(), *ObjCSubsts,
2965 if (!ResultType->isVoidType() || SuppressBlock)
2966 ResultType.getAsStringInternal(Result, Policy);
2967
2968 // Format the parameter list.
2969 std::string Params;
2970 if (!BlockProto || Block.getNumParams() == 0) {
2971 if (BlockProto && BlockProto.getTypePtr()->isVariadic())
2972 Params = "(...)";
2973 else
2974 Params = "(void)";
2975 } else {
2976 Params += "(";
2977 for (unsigned I = 0, N = Block.getNumParams(); I != N; ++I) {
2978 if (I)
2979 Params += ", ";
2980 Params += FormatFunctionParameter(Policy, Block.getParam(I),
2981 /*SuppressName=*/false,
2982 /*SuppressBlock=*/true, ObjCSubsts);
2983
2984 if (I == N - 1 && BlockProto.getTypePtr()->isVariadic())
2985 Params += ", ...";
2986 }
2987 Params += ")";
2988 }
2989
2990 if (SuppressBlock) {
2991 // Format as a parameter.
2992 Result = Result + " (^";
2993 if (!SuppressBlockName && BlockDecl->getIdentifier())
2994 Result += BlockDecl->getIdentifier()->getName();
2995 Result += ")";
2996 Result += Params;
2997 } else {
2998 // Format as a block literal argument.
2999 Result = '^' + Result;
3000 Result += Params;
3001
3002 if (!SuppressBlockName && BlockDecl->getIdentifier())
3003 Result += BlockDecl->getIdentifier()->getName();
3004 }
3005
3006 return Result;
3007}
3008
3009static std::string GetDefaultValueString(const ParmVarDecl *Param,
3010 const SourceManager &SM,
3011 const LangOptions &LangOpts) {
3012 const SourceRange SrcRange = Param->getDefaultArgRange();
3013 CharSourceRange CharSrcRange = CharSourceRange::getTokenRange(SrcRange);
3014 bool Invalid = CharSrcRange.isInvalid();
3015 if (Invalid)
3016 return "";
3017 StringRef srcText =
3018 Lexer::getSourceText(CharSrcRange, SM, LangOpts, &Invalid);
3019 if (Invalid)
3020 return "";
3021
3022 if (srcText.empty() || srcText == "=") {
3023 // Lexer can't determine the value.
3024 // This happens if the code is incorrect (for example class is forward
3025 // declared).
3026 return "";
3027 }
3028 std::string DefValue(srcText.str());
3029 // FIXME: remove this check if the Lexer::getSourceText value is fixed and
3030 // this value always has (or always does not have) '=' in front of it
3031 if (DefValue.at(0) != '=') {
3032 // If we don't have '=' in front of value.
3033 // Lexer returns built-in types values without '=' and user-defined types
3034 // values with it.
3035 return " = " + DefValue;
3036 }
3037 return " " + DefValue;
3038}
3039
3040/// Add function parameter chunks to the given code completion string.
3042 const PrintingPolicy &Policy,
3043 const FunctionDecl *Function,
3045 unsigned Start = 0,
3046 bool InOptional = false) {
3047 bool FirstParameter = true;
3048
3049 for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) {
3050 const ParmVarDecl *Param = Function->getParamDecl(P);
3051
3052 if (Param->hasDefaultArg() && !InOptional) {
3053 // When we see an optional default argument, put that argument and
3054 // the remaining default arguments into a new, optional string.
3055 CodeCompletionBuilder Opt(Result.getAllocator(),
3056 Result.getCodeCompletionTUInfo());
3057 if (!FirstParameter)
3059 AddFunctionParameterChunks(PP, Policy, Function, Opt, P, true);
3060 Result.AddOptionalChunk(Opt.TakeString());
3061 break;
3062 }
3063
3064 if (FirstParameter)
3065 FirstParameter = false;
3066 else
3068
3069 InOptional = false;
3070
3071 // Format the placeholder string.
3072 std::string PlaceholderStr = FormatFunctionParameter(Policy, Param);
3073 if (Param->hasDefaultArg())
3074 PlaceholderStr +=
3076
3077 if (Function->isVariadic() && P == N - 1)
3078 PlaceholderStr += ", ...";
3079
3080 // Add the placeholder string.
3081 Result.AddPlaceholderChunk(
3082 Result.getAllocator().CopyString(PlaceholderStr));
3083 }
3084
3085 if (const auto *Proto = Function->getType()->getAs<FunctionProtoType>())
3086 if (Proto->isVariadic()) {
3087 if (Proto->getNumParams() == 0)
3088 Result.AddPlaceholderChunk("...");
3089
3090 MaybeAddSentinel(PP, Function, Result);
3091 }
3092}
3093
3094/// Add template parameter chunks to the given code completion string.
3096 ASTContext &Context, const PrintingPolicy &Policy,
3097 const TemplateDecl *Template, CodeCompletionBuilder &Result,
3098 unsigned MaxParameters = 0, unsigned Start = 0, bool InDefaultArg = false) {
3099 bool FirstParameter = true;
3100
3101 // Prefer to take the template parameter names from the first declaration of
3102 // the template.
3103 Template = cast<TemplateDecl>(Template->getCanonicalDecl());
3104
3105 TemplateParameterList *Params = Template->getTemplateParameters();
3106 TemplateParameterList::iterator PEnd = Params->end();
3107 if (MaxParameters)
3108 PEnd = Params->begin() + MaxParameters;
3109 for (TemplateParameterList::iterator P = Params->begin() + Start; P != PEnd;
3110 ++P) {
3111 bool HasDefaultArg = false;
3112 std::string PlaceholderStr;
3113 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) {
3114 if (TTP->wasDeclaredWithTypename())
3115 PlaceholderStr = "typename";
3116 else if (const auto *TC = TTP->getTypeConstraint()) {
3117 llvm::raw_string_ostream OS(PlaceholderStr);
3118 TC->print(OS, Policy);
3119 OS.flush();
3120 } else
3121 PlaceholderStr = "class";
3122
3123 if (TTP->getIdentifier()) {
3124 PlaceholderStr += ' ';
3125 PlaceholderStr += TTP->getIdentifier()->deuglifiedName();
3126 }
3127
3128 HasDefaultArg = TTP->hasDefaultArgument();
3129 } else if (NonTypeTemplateParmDecl *NTTP =
3130 dyn_cast<NonTypeTemplateParmDecl>(*P)) {
3131 if (NTTP->getIdentifier())
3132 PlaceholderStr = std::string(NTTP->getIdentifier()->deuglifiedName());
3133 NTTP->getType().getAsStringInternal(PlaceholderStr, Policy);
3134 HasDefaultArg = NTTP->hasDefaultArgument();
3135 } else {
3136 assert(isa<TemplateTemplateParmDecl>(*P));
3137 TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P);
3138
3139 // Since putting the template argument list into the placeholder would
3140 // be very, very long, we just use an abbreviation.
3141 PlaceholderStr = "template<...> class";
3142 if (TTP->getIdentifier()) {
3143 PlaceholderStr += ' ';
3144 PlaceholderStr += TTP->getIdentifier()->deuglifiedName();
3145 }
3146
3147 HasDefaultArg = TTP->hasDefaultArgument();
3148 }
3149
3150 if (HasDefaultArg && !InDefaultArg) {
3151 // When we see an optional default argument, put that argument and
3152 // the remaining default arguments into a new, optional string.
3153 CodeCompletionBuilder Opt(Result.getAllocator(),
3154 Result.getCodeCompletionTUInfo());
3155 if (!FirstParameter)
3157 AddTemplateParameterChunks(Context, Policy, Template, Opt, MaxParameters,
3158 P - Params->begin(), true);
3159 Result.AddOptionalChunk(Opt.TakeString());
3160 break;
3161 }
3162
3163 InDefaultArg = false;
3164
3165 if (FirstParameter)
3166 FirstParameter = false;
3167 else
3169
3170 // Add the placeholder string.
3171 Result.AddPlaceholderChunk(
3172 Result.getAllocator().CopyString(PlaceholderStr));
3173 }
3174}
3175
3176/// Add a qualifier to the given code-completion string, if the
3177/// provided nested-name-specifier is non-NULL.
3179 NestedNameSpecifier *Qualifier,
3180 bool QualifierIsInformative,
3181 ASTContext &Context,
3182 const PrintingPolicy &Policy) {
3183 if (!Qualifier)
3184 return;
3185
3186 std::string PrintedNNS;
3187 {
3188 llvm::raw_string_ostream OS(PrintedNNS);
3189 Qualifier->print(OS, Policy);
3190 }
3191 if (QualifierIsInformative)
3192 Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS));
3193 else
3194 Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS));
3195}
3196
3197static void
3199 const FunctionDecl *Function) {
3200 const auto *Proto = Function->getType()->getAs<FunctionProtoType>();
3201 if (!Proto || !Proto->getMethodQuals())
3202 return;
3203
3204 // FIXME: Add ref-qualifier!
3205
3206 // Handle single qualifiers without copying
3207 if (Proto->getMethodQuals().hasOnlyConst()) {
3208 Result.AddInformativeChunk(" const");
3209 return;
3210 }
3211
3212 if (Proto->getMethodQuals().hasOnlyVolatile()) {
3213 Result.AddInformativeChunk(" volatile");
3214 return;
3215 }
3216
3217 if (Proto->getMethodQuals().hasOnlyRestrict()) {
3218 Result.AddInformativeChunk(" restrict");
3219 return;
3220 }
3221
3222 // Handle multiple qualifiers.
3223 std::string QualsStr;
3224 if (Proto->isConst())
3225 QualsStr += " const";
3226 if (Proto->isVolatile())
3227 QualsStr += " volatile";
3228 if (Proto->isRestrict())
3229 QualsStr += " restrict";
3230 Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr));
3231}
3232
3233/// Add the name of the given declaration
3234static void AddTypedNameChunk(ASTContext &Context, const PrintingPolicy &Policy,
3235 const NamedDecl *ND,
3237 DeclarationName Name = ND->getDeclName();
3238 if (!Name)
3239 return;
3240
3241 switch (Name.getNameKind()) {
3243 const char *OperatorName = nullptr;
3244 switch (Name.getCXXOverloadedOperator()) {
3245 case OO_None:
3246 case OO_Conditional:
3248 OperatorName = "operator";
3249 break;
3250
3251#define OVERLOADED_OPERATOR(Name, Spelling, Token, Unary, Binary, MemberOnly) \
3252 case OO_##Name: \
3253 OperatorName = "operator" Spelling; \
3254 break;
3255#define OVERLOADED_OPERATOR_MULTI(Name, Spelling, Unary, Binary, MemberOnly)
3256#include "clang/Basic/OperatorKinds.def"
3257
3258 case OO_New:
3259 OperatorName = "operator new";
3260 break;
3261 case OO_Delete:
3262 OperatorName = "operator delete";
3263 break;
3264 case OO_Array_New:
3265 OperatorName = "operator new[]";
3266 break;
3267 case OO_Array_Delete:
3268 OperatorName = "operator delete[]";
3269 break;
3270 case OO_Call:
3271 OperatorName = "operator()";
3272 break;
3273 case OO_Subscript:
3274 OperatorName = "operator[]";
3275 break;
3276 }
3277 Result.AddTypedTextChunk(OperatorName);
3278 break;
3279 }
3280
3285 Result.AddTypedTextChunk(
3286 Result.getAllocator().CopyString(ND->getNameAsString()));
3287 break;
3288
3294 break;
3295
3297 CXXRecordDecl *Record = nullptr;
3298 QualType Ty = Name.getCXXNameType();
3299 if (const auto *RecordTy = Ty->getAs<RecordType>())
3300 Record = cast<CXXRecordDecl>(RecordTy->getDecl());
3301 else if (const auto *InjectedTy = Ty->getAs<InjectedClassNameType>())
3302 Record = InjectedTy->getDecl();
3303 else {
3304 Result.AddTypedTextChunk(
3305 Result.getAllocator().CopyString(ND->getNameAsString()));
3306 break;
3307 }
3308
3309 Result.AddTypedTextChunk(
3310 Result.getAllocator().CopyString(Record->getNameAsString()));
3311 if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) {
3313 AddTemplateParameterChunks(Context, Policy, Template, Result);
3315 }
3316 break;
3317 }
3318 }
3319}
3320
3322 Sema &S, const CodeCompletionContext &CCContext,
3323 CodeCompletionAllocator &Allocator, CodeCompletionTUInfo &CCTUInfo,
3324 bool IncludeBriefComments) {
3325 return CreateCodeCompletionString(S.Context, S.PP, CCContext, Allocator,
3326 CCTUInfo, IncludeBriefComments);
3327}
3328
3330 Preprocessor &PP, CodeCompletionAllocator &Allocator,
3331 CodeCompletionTUInfo &CCTUInfo) {
3332 assert(Kind == RK_Macro);
3333 CodeCompletionBuilder Result(Allocator, CCTUInfo, Priority, Availability);
3334 const MacroInfo *MI = PP.getMacroInfo(Macro);
3335 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Macro->getName()));
3336
3337 if (!MI || !MI->isFunctionLike())
3338 return Result.TakeString();
3339
3340 // Format a function-like macro with placeholders for the arguments.
3342 MacroInfo::param_iterator A = MI->param_begin(), AEnd = MI->param_end();
3343
3344 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
3345 if (MI->isC99Varargs()) {
3346 --AEnd;
3347
3348 if (A == AEnd) {
3349 Result.AddPlaceholderChunk("...");
3350 }
3351 }
3352
3353 for (MacroInfo::param_iterator A = MI->param_begin(); A != AEnd; ++A) {
3354 if (A != MI->param_begin())
3356
3357 if (MI->isVariadic() && (A + 1) == AEnd) {
3358 SmallString<32> Arg = (*A)->getName();
3359 if (MI->isC99Varargs())
3360 Arg += ", ...";
3361 else
3362 Arg += "...";
3363 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
3364 break;
3365 }
3366
3367 // Non-variadic macros are simple.
3368 Result.AddPlaceholderChunk(
3369 Result.getAllocator().CopyString((*A)->getName()));
3370 }
3372 return Result.TakeString();
3373}
3374
3375/// If possible, create a new code completion string for the given
3376/// result.
3377///
3378/// \returns Either a new, heap-allocated code completion string describing
3379/// how to use this result, or NULL to indicate that the string or name of the
3380/// result is all that is needed.
3382 ASTContext &Ctx, Preprocessor &PP, const CodeCompletionContext &CCContext,
3383 CodeCompletionAllocator &Allocator, CodeCompletionTUInfo &CCTUInfo,
3384 bool IncludeBriefComments) {
3385 if (Kind == RK_Macro)
3386 return CreateCodeCompletionStringForMacro(PP, Allocator, CCTUInfo);
3387
3388 CodeCompletionBuilder Result(Allocator, CCTUInfo, Priority, Availability);
3389
3391 if (Kind == RK_Pattern) {
3392 Pattern->Priority = Priority;
3393 Pattern->Availability = Availability;
3394
3395 if (Declaration) {
3396 Result.addParentContext(Declaration->getDeclContext());
3397 Pattern->ParentName = Result.getParentName();
3398 if (const RawComment *RC =
3400 Result.addBriefComment(RC->getBriefText(Ctx));
3401 Pattern->BriefComment = Result.getBriefComment();
3402 }
3403 }
3404
3405 return Pattern;
3406 }
3407
3408 if (Kind == RK_Keyword) {
3409 Result.AddTypedTextChunk(Keyword);
3410 return Result.TakeString();
3411 }
3412 assert(Kind == RK_Declaration && "Missed a result kind?");
3414 PP, Ctx, Result, IncludeBriefComments, CCContext, Policy);
3415}
3416
3418 std::string &BeforeName,
3419 std::string &NameAndSignature) {
3420 bool SeenTypedChunk = false;
3421 for (auto &Chunk : CCS) {
3422 if (Chunk.Kind == CodeCompletionString::CK_Optional) {
3423 assert(SeenTypedChunk && "optional parameter before name");
3424 // Note that we put all chunks inside into NameAndSignature.
3425 printOverrideString(*Chunk.Optional, NameAndSignature, NameAndSignature);
3426 continue;
3427 }
3428 SeenTypedChunk |= Chunk.Kind == CodeCompletionString::CK_TypedText;
3429 if (SeenTypedChunk)
3430 NameAndSignature += Chunk.Text;
3431 else
3432 BeforeName += Chunk.Text;
3433 }
3434}
3435
3439 bool IncludeBriefComments, const CodeCompletionContext &CCContext,
3440 PrintingPolicy &Policy) {
3441 auto *CCS = createCodeCompletionStringForDecl(PP, Ctx, Result,
3442 /*IncludeBriefComments=*/false,
3443 CCContext, Policy);
3444 std::string BeforeName;
3445 std::string NameAndSignature;
3446 // For overrides all chunks go into the result, none are informative.
3447 printOverrideString(*CCS, BeforeName, NameAndSignature);
3448 NameAndSignature += " override";
3449
3450 Result.AddTextChunk(Result.getAllocator().CopyString(BeforeName));
3452 Result.AddTypedTextChunk(Result.getAllocator().CopyString(NameAndSignature));
3453 return Result.TakeString();
3454}
3455
3456// FIXME: Right now this works well with lambdas. Add support for other functor
3457// types like std::function.
3459 const auto *VD = dyn_cast<VarDecl>(ND);
3460 if (!VD)
3461 return nullptr;
3462 const auto *RecordDecl = VD->getType()->getAsCXXRecordDecl();
3463 if (!RecordDecl || !RecordDecl->isLambda())
3464 return nullptr;
3465 return RecordDecl->getLambdaCallOperator();
3466}
3467
3470 bool IncludeBriefComments, const CodeCompletionContext &CCContext,
3471 PrintingPolicy &Policy) {
3472 const NamedDecl *ND = Declaration;
3473 Result.addParentContext(ND->getDeclContext());
3474
3475 if (IncludeBriefComments) {
3476 // Add documentation comment, if it exists.
3477 if (const RawComment *RC = getCompletionComment(Ctx, Declaration)) {
3478 Result.addBriefComment(RC->getBriefText(Ctx));
3479 }
3480 }
3481
3483 Result.AddTypedTextChunk(
3484 Result.getAllocator().CopyString(ND->getNameAsString()));
3485 Result.AddTextChunk("::");
3486 return Result.TakeString();
3487 }
3488
3489 for (const auto *I : ND->specific_attrs<AnnotateAttr>())
3490 Result.AddAnnotation(Result.getAllocator().CopyString(I->getAnnotation()));
3491
3492 auto AddFunctionTypeAndResult = [&](const FunctionDecl *Function) {
3493 AddResultTypeChunk(Ctx, Policy, Function, CCContext.getBaseType(), Result);
3495 Ctx, Policy);
3496 AddTypedNameChunk(Ctx, Policy, ND, Result);
3498 AddFunctionParameterChunks(PP, Policy, Function, Result);
3501 };
3502
3503 if (const auto *Function = dyn_cast<FunctionDecl>(ND)) {
3504 AddFunctionTypeAndResult(Function);
3505 return Result.TakeString();
3506 }
3507
3508 if (const auto *CallOperator =
3509 dyn_cast_or_null<FunctionDecl>(extractFunctorCallOperator(ND))) {
3510 AddFunctionTypeAndResult(CallOperator);
3511 return Result.TakeString();
3512 }
3513
3514 AddResultTypeChunk(Ctx, Policy, ND, CCContext.getBaseType(), Result);
3515
3516 if (const FunctionTemplateDecl *FunTmpl =
3517 dyn_cast<FunctionTemplateDecl>(ND)) {
3519 Ctx, Policy);
3520 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
3521 AddTypedNameChunk(Ctx, Policy, Function, Result);
3522
3523 // Figure out which template parameters are deduced (or have default
3524 // arguments).
3525 llvm::SmallBitVector Deduced;
3526 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
3527 unsigned LastDeducibleArgument;
3528 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
3529 --LastDeducibleArgument) {
3530 if (!Deduced[LastDeducibleArgument - 1]) {
3531 // C++0x: Figure out if the template argument has a default. If so,
3532 // the user doesn't need to type this argument.
3533 // FIXME: We need to abstract template parameters better!
3534 bool HasDefaultArg = false;
3535 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
3536 LastDeducibleArgument - 1);
3537 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
3538 HasDefaultArg = TTP->hasDefaultArgument();
3539 else if (NonTypeTemplateParmDecl *NTTP =
3540 dyn_cast<NonTypeTemplateParmDecl>(Param))
3541 HasDefaultArg = NTTP->hasDefaultArgument();
3542 else {
3543 assert(isa<TemplateTemplateParmDecl>(Param));
3544 HasDefaultArg =
3545 cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
3546 }
3547
3548 if (!HasDefaultArg)
3549 break;
3550 }
3551 }
3552
3553 if (LastDeducibleArgument) {
3554 // Some of the function template arguments cannot be deduced from a
3555 // function call, so we introduce an explicit template argument list
3556 // containing all of the arguments up to the first deducible argument.
3558 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
3559 LastDeducibleArgument);
3561 }
3562
3563 // Add the function parameters
3565 AddFunctionParameterChunks(PP, Policy, Function, Result);
3568 return Result.TakeString();
3569 }
3570
3571 if (const auto *Template = dyn_cast<TemplateDecl>(ND)) {
3573 Ctx, Policy);
3574 Result.AddTypedTextChunk(
3575 Result.getAllocator().CopyString(Template->getNameAsString()));
3577 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
3579 return Result.TakeString();
3580 }
3581
3582 if (const auto *Method = dyn_cast<ObjCMethodDecl>(ND)) {
3583 Selector Sel = Method->getSelector();
3584 if (Sel.isUnarySelector()) {
3585 Result.AddTypedTextChunk(
3586 Result.getAllocator().CopyString(Sel.getNameForSlot(0)));
3587 return Result.TakeString();
3588 }
3589
3590 std::string SelName = Sel.getNameForSlot(0).str();
3591 SelName += ':';
3592 if (StartParameter == 0)
3593 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
3594 else {
3595 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
3596
3597 // If there is only one parameter, and we're past it, add an empty
3598 // typed-text chunk since there is nothing to type.
3599 if (Method->param_size() == 1)
3600 Result.AddTypedTextChunk("");
3601 }
3602 unsigned Idx = 0;
3603 // The extra Idx < Sel.getNumArgs() check is needed due to legacy C-style
3604 // method parameters.
3605 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
3606 PEnd = Method->param_end();
3607 P != PEnd && Idx < Sel.getNumArgs(); (void)++P, ++Idx) {
3608 if (Idx > 0) {
3609 std::string Keyword;
3610 if (Idx > StartParameter)
3612 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
3613 Keyword += II->getName();
3614 Keyword += ":";
3616 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
3617 else
3618 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
3619 }
3620
3621 // If we're before the starting parameter, skip the placeholder.
3622 if (Idx < StartParameter)
3623 continue;
3624
3625 std::string Arg;
3626 QualType ParamType = (*P)->getType();
3627 std::optional<ArrayRef<QualType>> ObjCSubsts;
3628 if (!CCContext.getBaseType().isNull())
3629 ObjCSubsts = CCContext.getBaseType()->getObjCSubstitutions(Method);
3630
3631 if (ParamType->isBlockPointerType() && !DeclaringEntity)
3632 Arg = FormatFunctionParameter(Policy, *P, true,
3633 /*SuppressBlock=*/false, ObjCSubsts);
3634 else {
3635 if (ObjCSubsts)
3636 ParamType = ParamType.substObjCTypeArgs(
3637 Ctx, *ObjCSubsts, ObjCSubstitutionContext::Parameter);
3638 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier(),
3639 ParamType);
3640 Arg += ParamType.getAsString(Policy) + ")";
3641 if (IdentifierInfo *II = (*P)->getIdentifier())
3643 Arg += II->getName();
3644 }
3645
3646 if (Method->isVariadic() && (P + 1) == PEnd)
3647 Arg += ", ...";
3648
3649 if (DeclaringEntity)
3650 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
3652 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
3653 else
3654 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
3655 }
3656
3657 if (Method->isVariadic()) {
3658 if (Method->param_size() == 0) {
3659 if (DeclaringEntity)
3660 Result.AddTextChunk(", ...");
3662 Result.AddInformativeChunk(", ...");
3663 else
3664 Result.AddPlaceholderChunk(", ...");
3665 }
3666
3667 MaybeAddSentinel(PP, Method, Result);
3668 }
3669
3670 return Result.TakeString();
3671 }
3672
3673 if (Qualifier)
3675 Ctx, Policy);
3676
3677 Result.AddTypedTextChunk(
3678 Result.getAllocator().CopyString(ND->getNameAsString()));
3679 return Result.TakeString();
3680}
3681
3683 const NamedDecl *ND) {
3684 if (!ND)
3685 return nullptr;
3686 if (auto *RC = Ctx.getRawCommentForAnyRedecl(ND))
3687 return RC;
3688
3689 // Try to find comment from a property for ObjC methods.
3690 const auto *M = dyn_cast<ObjCMethodDecl>(ND);
3691 if (!M)
3692 return nullptr;
3693 const ObjCPropertyDecl *PDecl = M->findPropertyDecl();
3694 if (!PDecl)
3695 return nullptr;
3696
3697 return Ctx.getRawCommentForAnyRedecl(PDecl);
3698}
3699
3701 const NamedDecl *ND) {
3702 const auto *M = dyn_cast_or_null<ObjCMethodDecl>(ND);
3703 if (!M || !M->isPropertyAccessor())
3704 return nullptr;
3705
3706 // Provide code completion comment for self.GetterName where
3707 // GetterName is the getter method for a property with name
3708 // different from the property name (declared via a property
3709 // getter attribute.
3710 const ObjCPropertyDecl *PDecl = M->findPropertyDecl();
3711 if (!PDecl)
3712 return nullptr;
3713 if (PDecl->getGetterName() == M->getSelector() &&
3714 PDecl->getIdentifier() != M->getIdentifier()) {
3715 if (auto *RC = Ctx.getRawCommentForAnyRedecl(M))
3716 return RC;
3717 if (auto *RC = Ctx.getRawCommentForAnyRedecl(PDecl))
3718 return RC;
3719 }
3720 return nullptr;
3721}
3722
3724 const ASTContext &Ctx,
3725 const CodeCompleteConsumer::OverloadCandidate &Result, unsigned ArgIndex) {
3726 auto FDecl = Result.getFunction();
3727 if (!FDecl)
3728 return nullptr;
3729 if (ArgIndex < FDecl->getNumParams())
3730 return Ctx.getRawCommentForAnyRedecl(FDecl->getParamDecl(ArgIndex));
3731 return nullptr;
3732}
3733
3735 const PrintingPolicy &Policy,
3737 unsigned CurrentArg) {
3738 unsigned ChunkIndex = 0;
3739 auto AddChunk = [&](llvm::StringRef Placeholder) {
3740 if (ChunkIndex > 0)
3742 const char *Copy = Result.getAllocator().CopyString(Placeholder);
3743 if (ChunkIndex == CurrentArg)
3744 Result.AddCurrentParameterChunk(Copy);
3745 else
3746 Result.AddPlaceholderChunk(Copy);
3747 ++ChunkIndex;
3748 };
3749 // Aggregate initialization has all bases followed by all fields.
3750 // (Bases are not legal in C++11 but in that case we never get here).
3751 if (auto *CRD = llvm::dyn_cast<CXXRecordDecl>(RD)) {
3752 for (const auto &Base : CRD->bases())
3753 AddChunk(Base.getType().getAsString(Policy));
3754 }
3755 for (const auto &Field : RD->fields())
3756 AddChunk(FormatFunctionParameter(Policy, Field));
3757}
3758
3759/// Add function overload parameter chunks to the given code completion
3760/// string.
3762 ASTContext &Context, const PrintingPolicy &Policy,
3763 const FunctionDecl *Function, const FunctionProtoType *Prototype,
3765 unsigned CurrentArg, unsigned Start = 0, bool InOptional = false) {
3766 if (!Function && !Prototype) {
3768 return;
3769 }
3770
3771 bool FirstParameter = true;
3772 unsigned NumParams =
3773 Function ? Function->getNumParams() : Prototype->getNumParams();
3774
3775 for (unsigned P = Start; P != NumParams; ++P) {
3776 if (Function && Function->getParamDecl(P)->hasDefaultArg() && !InOptional) {
3777 // When we see an optional default argument, put that argument and
3778 // the remaining default arguments into a new, optional string.
3779 CodeCompletionBuilder Opt(Result.getAllocator(),
3780 Result.getCodeCompletionTUInfo());
3781 if (!FirstParameter)
3783 // Optional sections are nested.
3784 AddOverloadParameterChunks(Context, Policy, Function, Prototype,
3785 PrototypeLoc, Opt, CurrentArg, P,
3786 /*InOptional=*/true);
3787 Result.AddOptionalChunk(Opt.TakeString());
3788 return;
3789 }
3790
3791 if (FirstParameter)
3792 FirstParameter = false;
3793 else
3795
3796 InOptional = false;
3797
3798 // Format the placeholder string.
3799 std::string Placeholder;
3800 assert(P < Prototype->getNumParams());
3801 if (Function || PrototypeLoc) {
3802 const ParmVarDecl *Param =
3803 Function ? Function->getParamDecl(P) : PrototypeLoc.getParam(P);
3804 Placeholder = FormatFunctionParameter(Policy, Param);
3805 if (Param->hasDefaultArg())
3806 Placeholder += GetDefaultValueString(Param, Context.getSourceManager(),
3807 Context.getLangOpts());
3808 } else {
3809 Placeholder = Prototype->getParamType(P).getAsString(Policy);
3810 }
3811
3812 if (P == CurrentArg)
3813 Result.AddCurrentParameterChunk(
3814 Result.getAllocator().CopyString(Placeholder));
3815 else
3816 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Placeholder));
3817 }
3818
3819 if (Prototype && Prototype->isVariadic()) {
3820 CodeCompletionBuilder Opt(Result.getAllocator(),
3821 Result.getCodeCompletionTUInfo());
3822 if (!FirstParameter)
3824
3825 if (CurrentArg < NumParams)
3826 Opt.AddPlaceholderChunk("...");
3827 else
3828 Opt.AddCurrentParameterChunk("...");
3829
3830 Result.AddOptionalChunk(Opt.TakeString());
3831 }
3832}
3833
3834static std::string
3836 const PrintingPolicy &Policy) {
3837 if (const auto *Type = dyn_cast<TemplateTypeParmDecl>(Param)) {
3838 Optional = Type->hasDefaultArgument();
3839 } else if (const auto *NonType = dyn_cast<NonTypeTemplateParmDecl>(Param)) {
3840 Optional = NonType->hasDefaultArgument();
3841 } else if (const auto *Template = dyn_cast<TemplateTemplateParmDecl>(Param)) {
3842 Optional = Template->hasDefaultArgument();
3843 }
3844 std::string Result;
3845 llvm::raw_string_ostream OS(Result);
3846 Param->print(OS, Policy);
3847 return Result;
3848}
3849
3850static std::string templateResultType(const TemplateDecl *TD,
3851 const PrintingPolicy &Policy) {
3852 if (const auto *CTD = dyn_cast<ClassTemplateDecl>(TD))
3853 return CTD->getTemplatedDecl()->getKindName().str();
3854 if (const auto *VTD = dyn_cast<VarTemplateDecl>(TD))
3855 return VTD->getTemplatedDecl()->getType().getAsString(Policy);
3856 if (const auto *FTD = dyn_cast<FunctionTemplateDecl>(TD))
3857 return FTD->getTemplatedDecl()->getReturnType().getAsString(Policy);
3858 if (isa<TypeAliasTemplateDecl>(TD))
3859 return "type";
3860 if (isa<TemplateTemplateParmDecl>(TD))
3861 return "class";
3862 if (isa<ConceptDecl>(TD))
3863 return "concept";
3864 return "";
3865}
3866
3868 const TemplateDecl *TD, CodeCompletionBuilder &Builder, unsigned CurrentArg,
3869 const PrintingPolicy &Policy) {
3871 CodeCompletionBuilder OptionalBuilder(Builder.getAllocator(),
3872 Builder.getCodeCompletionTUInfo());
3873 std::string ResultType = templateResultType(TD, Policy);
3874 if (!ResultType.empty())
3875 Builder.AddResultTypeChunk(Builder.getAllocator().CopyString(ResultType));
3876 Builder.AddTextChunk(
3877 Builder.getAllocator().CopyString(TD->getNameAsString()));
3878 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
3879 // Initially we're writing into the main string. Once we see an optional arg
3880 // (with default), we're writing into the nested optional chunk.
3881 CodeCompletionBuilder *Current = &Builder;
3882 for (unsigned I = 0; I < Params.size(); ++I) {
3883 bool Optional = false;
3884 std::string Placeholder =
3885 formatTemplateParameterPlaceholder(Params[I], Optional, Policy);
3886 if (Optional)
3887 Current = &OptionalBuilder;
3888 if (I > 0)
3889 Current->AddChunk(CodeCompletionString::CK_Comma);
3890 Current->AddChunk(I == CurrentArg
3893 Current->getAllocator().CopyString(Placeholder));
3894 }
3895 // Add the optional chunk to the main string if we ever used it.
3896 if (Current == &OptionalBuilder)
3897 Builder.AddOptionalChunk(OptionalBuilder.TakeString());
3898 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
3899 // For function templates, ResultType was the function's return type.
3900 // Give some clue this is a function. (Don't show the possibly-bulky params).
3901 if (isa<FunctionTemplateDecl>(TD))
3902 Builder.AddInformativeChunk("()");
3903 return Builder.TakeString();
3904}
3905
3908 unsigned CurrentArg, Sema &S, CodeCompletionAllocator &Allocator,
3909 CodeCompletionTUInfo &CCTUInfo, bool IncludeBriefComments,
3910 bool Braced) const {
3912 // Show signatures of constructors as they are declared:
3913 // vector(int n) rather than vector<string>(int n)
3914 // This is less noisy without being less clear, and avoids tricky cases.
3916
3917 // FIXME: Set priority, availability appropriately.
3918 CodeCompletionBuilder Result(Allocator, CCTUInfo, 1,
3920
3921 if (getKind() == CK_Template)
3922 return createTemplateSignatureString(getTemplate(), Result, CurrentArg,
3923 Policy);
3924
3925 FunctionDecl *FDecl = getFunction();
3926 const FunctionProtoType *Proto =
3927 dyn_cast_or_null<FunctionProtoType>(getFunctionType());
3928
3929 // First, the name/type of the callee.
3930 if (getKind() == CK_Aggregate) {
3931 Result.AddTextChunk(
3932 Result.getAllocator().CopyString(getAggregate()->getName()));
3933 } else if (FDecl) {
3934 if (IncludeBriefComments) {
3935 if (auto RC = getParameterComment(S.getASTContext(), *this, CurrentArg))
3936 Result.addBriefComment(RC->getBriefText(S.getASTContext()));
3937 }
3938 AddResultTypeChunk(S.Context, Policy, FDecl, QualType(), Result);
3939
3940 std::string Name;
3941 llvm::raw_string_ostream OS(Name);
3942 FDecl->getDeclName().print(OS, Policy);
3943 Result.AddTextChunk(Result.getAllocator().CopyString(OS.str()));
3944 } else {
3945 // Function without a declaration. Just give the return type.
3946 Result.AddResultTypeChunk(Result.getAllocator().CopyString(
3947 getFunctionType()->getReturnType().getAsString(Policy)));
3948 }
3949
3950 // Next, the brackets and parameters.
3953 if (getKind() == CK_Aggregate)
3954 AddOverloadAggregateChunks(getAggregate(), Policy, Result, CurrentArg);
3955 else
3956 AddOverloadParameterChunks(S.getASTContext(), Policy, FDecl, Proto,
3957 getFunctionProtoTypeLoc(), Result, CurrentArg);
3960
3961 return Result.TakeString();
3962}
3963
3964unsigned clang::getMacroUsagePriority(StringRef MacroName,
3965 const LangOptions &LangOpts,
3966 bool PreferredTypeIsPointer) {
3967 unsigned Priority = CCP_Macro;
3968
3969 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
3970 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
3971 MacroName.equals("Nil")) {
3973 if (PreferredTypeIsPointer)
3975 }
3976 // Treat "YES", "NO", "true", and "false" as constants.
3977 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
3978 MacroName.equals("true") || MacroName.equals("false"))
3980 // Treat "bool" as a type.
3981 else if (MacroName.equals("bool"))
3982 Priority = CCP_Type + (LangOpts.ObjC ? CCD_bool_in_ObjC : 0);
3983
3984 return Priority;
3985}
3986
3988 if (!D)
3990
3991 switch (D->getKind()) {
3992 case Decl::Enum:
3993 return CXCursor_EnumDecl;
3994 case Decl::EnumConstant:
3996 case Decl::Field:
3997 return CXCursor_FieldDecl;
3998 case Decl::Function:
3999 return CXCursor_FunctionDecl;
4000 case Decl::ObjCCategory:
4002 case Decl::ObjCCategoryImpl:
4004 case Decl::ObjCImplementation:
4006
4007 case Decl::ObjCInterface:
4009 case Decl::ObjCIvar:
4010 return CXCursor_ObjCIvarDecl;
4011 case Decl::ObjCMethod:
4012 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
4015 case Decl::CXXMethod:
4016 return CXCursor_CXXMethod;
4017 case Decl::CXXConstructor:
4018 return CXCursor_Constructor;
4019 case Decl::CXXDestructor:
4020 return CXCursor_Destructor;
4021 case Decl::CXXConversion:
4023 case Decl::ObjCProperty:
4025 case Decl::ObjCProtocol:
4027 case Decl::ParmVar:
4028 return CXCursor_ParmDecl;
4029 case Decl::Typedef:
4030 return CXCursor_TypedefDecl;
4031 case Decl::TypeAlias:
4033 case Decl::TypeAliasTemplate:
4035 case Decl::Var:
4036 return CXCursor_VarDecl;
4037 case Decl::Namespace:
4038 return CXCursor_Namespace;
4039 case Decl::NamespaceAlias:
4041 case Decl::TemplateTypeParm:
4043 case Decl::NonTypeTemplateParm:
4045 case Decl::TemplateTemplateParm:
4047 case Decl::FunctionTemplate:
4049 case Decl::ClassTemplate:
4051 case Decl::AccessSpec:
4053 case Decl::ClassTemplatePartialSpecialization:
4055 case Decl::UsingDirective:
4057 case Decl::StaticAssert:
4058 return CXCursor_StaticAssert;
4059 case Decl::Friend:
4060 return CXCursor_FriendDecl;
4061 case Decl::TranslationUnit:
4063
4064 case Decl::Using:
4065 case Decl::UnresolvedUsingValue:
4066 case Decl::UnresolvedUsingTypename:
4068
4069 case Decl::UsingEnum:
4070 return CXCursor_EnumDecl;
4071
4072 case Decl::ObjCPropertyImpl:
4073 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
4076
4079 }
4080 llvm_unreachable("Unexpected Kind!");
4081
4082 case Decl::Import:
4084
4085 case Decl::ObjCTypeParam:
4087
4088 case Decl::Concept:
4089 return CXCursor_ConceptDecl;
4090
4091 default:
4092 if (const auto *TD = dyn_cast<TagDecl>(D)) {
4093 switch (TD->getTagKind()) {
4094 case TTK_Interface: // fall through
4095 case TTK_Struct:
4096 return CXCursor_StructDecl;
4097 case TTK_Class:
4098 return CXCursor_ClassDecl;
4099 case TTK_Union:
4100 return CXCursor_UnionDecl;
4101 case TTK_Enum:
4102 return CXCursor_EnumDecl;
4103 }
4104 }
4105 }
4106
4108}
4109
4110static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
4111 bool LoadExternal, bool IncludeUndefined,
4112 bool TargetTypeIsPointer = false) {
4114
4115 Results.EnterNewScope();
4116
4117 for (Preprocessor::macro_iterator M = PP.macro_begin(LoadExternal),
4118 MEnd = PP.macro_end(LoadExternal);
4119 M != MEnd; ++M) {
4120 auto MD = PP.getMacroDefinition(M->first);
4121 if (IncludeUndefined || MD) {
4122 MacroInfo *MI = MD.getMacroInfo();
4123 if (MI && MI->isUsedForHeaderGuard())
4124 continue;
4125
4126 Results.AddResult(
4127 Result(M->first, MI,
4128 getMacroUsagePriority(M->first->getName(), PP.getLangOpts(),
4129 TargetTypeIsPointer)));
4130 }
4131 }
4132
4133 Results.ExitScope();
4134}
4135
4136static void AddPrettyFunctionResults(const LangOptions &LangOpts,
4137 ResultBuilder &Results) {
4139
4140 Results.EnterNewScope();
4141
4142 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
4143 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
4144 if (LangOpts.C99 || LangOpts.CPlusPlus11)
4145 Results.AddResult(Result("__func__", CCP_Constant));
4146 Results.ExitScope();
4147}
4148
4150 CodeCompleteConsumer *CodeCompleter,
4151 CodeCompletionContext Context,
4152 CodeCompletionResult *Results,
4153 unsigned NumResults) {
4154 if (CodeCompleter)
4155 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
4156}
4157
4160 switch (PCC) {
4163
4164 case Sema::PCC_Class:
4166
4169
4172
4175
4176 case Sema::PCC_Template:
4178 if (S.CurContext->isFileContext())
4180 if (S.CurContext->isRecord())
4183
4186
4187 case Sema::PCC_ForInit:
4188 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
4189 S.getLangOpts().ObjC)
4191 else
4193
4198 S.getASTContext().BoolTy);
4199
4202
4203 case Sema::PCC_Type:
4205
4208
4211 }
4212
4213 llvm_unreachable("Invalid ParserCompletionContext!");
4214}
4215
4216/// If we're in a C++ virtual member function, add completion results
4217/// that invoke the functions we override, since it's common to invoke the
4218/// overridden function as well as adding new functionality.
4219///
4220/// \param S The semantic analysis object for which we are generating results.
4221///
4222/// \param InContext This context in which the nested-name-specifier preceding
4223/// the code-completion point
4224static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
4225 ResultBuilder &Results) {
4226 // Look through blocks.
4227 DeclContext *CurContext = S.CurContext;
4228 while (isa<BlockDecl>(CurContext))
4229 CurContext = CurContext->getParent();
4230
4231 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
4232 if (!Method || !Method->isVirtual())
4233 return;
4234
4235 // We need to have names for all of the parameters, if we're going to
4236 // generate a forwarding call.
4237 for (auto *P : Method->parameters())
4238 if (!P->getDeclName())
4239 return;
4240
4242 for (const CXXMethodDecl *Overridden : Method->overridden_methods()) {
4243 CodeCompletionBuilder Builder(Results.getAllocator(),
4244 Results.getCodeCompletionTUInfo());
4245 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
4246 continue;
4247
4248 // If we need a nested-name-specifier, add one now.
4249 if (!InContext) {
4251 S.Context, CurContext, Overridden->getDeclContext());
4252 if (NNS) {
4253 std::string Str;
4254 llvm::raw_string_ostream OS(Str);
4255 NNS->print(OS, Policy);
4256 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
4257 }
4258 } else if (!InContext->Equals(Overridden->getDeclContext()))
4259 continue;
4260
4261 Builder.AddTypedTextChunk(
4262 Results.getAllocator().CopyString(Overridden->getNameAsString()));
4263 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4264 bool FirstParam = true;
4265 for (auto *P : Method->parameters()) {
4266 if (FirstParam)
4267 FirstParam = false;
4268 else
4269 Builder.AddChunk(CodeCompletionString::CK_Comma);
4270
4271 Builder.AddPlaceholderChunk(
4272 Results.getAllocator().CopyString(P->getIdentifier()->getName()));
4273 }
4274 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4275 Results.AddResult(CodeCompletionResult(
4276 Builder.TakeString(), CCP_SuperCompletion, CXCursor_CXXMethod,
4277 CXAvailability_Available, Overridden));
4278 Results.Ignore(Overridden);
4279 }
4280}
4281
4283 ModuleIdPath Path) {
4285 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4286 CodeCompleter->getCodeCompletionTUInfo(),
4288 Results.EnterNewScope();
4289
4290 CodeCompletionAllocator &Allocator = Results.getAllocator();
4291 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
4293 if (Path.empty()) {
4294 // Enumerate all top-level modules.
4296 PP.getHeaderSearchInfo().collectAllModules(Modules);
4297 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
4298 Builder.AddTypedTextChunk(
4299 Builder.getAllocator().CopyString(Modules[I]->Name));
4300 Results.AddResult(Result(
4301 Builder.TakeString(), CCP_Declaration, CXCursor_ModuleImportDecl,
4302 Modules[I]->isAvailable() ? CXAvailability_Available
4304 }
4305 } else if (getLangOpts().Modules) {
4306 // Load the named module.
4307 Module *Mod =
4308 PP.getModuleLoader().loadModule(ImportLoc, Path, Module::AllVisible,
4309 /*IsInclusionDirective=*/false);
4310 // Enumerate submodules.
4311 if (Mod) {
4313 SubEnd = Mod->submodule_end();
4314 Sub != SubEnd; ++Sub) {
4315
4316 Builder.AddTypedTextChunk(
4317 Builder.getAllocator().CopyString((*Sub)->Name));
4318 Results.AddResult(Result(
4319 Builder.TakeString(), CCP_Declaration, CXCursor_ModuleImportDecl,
4320 (*Sub)->isAvailable() ? CXAvailability_Available
4322 }
4323 }
4324 }
4325 Results.ExitScope();
4326 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4327 Results.data(), Results.size());
4328}
4329
4331 ParserCompletionContext CompletionContext) {
4332 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4333 CodeCompleter->getCodeCompletionTUInfo(),
4334 mapCodeCompletionContext(*this, CompletionContext));
4335 Results.EnterNewScope();
4336
4337 // Determine how to filter results, e.g., so that the names of
4338 // values (functions, enumerators, function templates, etc.) are
4339 // only allowed where we can have an expression.
4340 switch (CompletionContext) {
4341 case PCC_Namespace:
4342 case PCC_Class:
4343 case PCC_ObjCInterface:
4344 case PCC_ObjCImplementation:
4345 case PCC_ObjCInstanceVariableList:
4346 case PCC_Template:
4347 case PCC_MemberTemplate:
4348 case PCC_Type:
4349 case PCC_LocalDeclarationSpecifiers:
4350 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
4351 break;
4352
4353 case PCC_Statement:
4354 case PCC_ParenthesizedExpression:
4355 case PCC_Expression:
4356 case PCC_ForInit:
4357 case PCC_Condition:
4358 if (WantTypesInContext(CompletionContext, getLangOpts()))
4359 Results.setFilter(&ResultBuilder::IsOrdinaryName);
4360 else
4361 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
4362
4363 if (getLangOpts().CPlusPlus)
4364 MaybeAddOverrideCalls(*this, /*InContext=*/nullptr, Results);
4365 break;
4366
4367 case PCC_RecoveryInFunction:
4368 // Unfiltered
4369 break;
4370 }
4371
4372 // If we are in a C++ non-static member function, check the qualifiers on
4373 // the member function to filter/prioritize the results list.
4374 auto ThisType = getCurrentThisType();
4375 if (!ThisType.isNull())
4376 Results.setObjectTypeQualifiers(ThisType->getPointeeType().getQualifiers(),
4377 VK_LValue);
4378
4379 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4380 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4381 CodeCompleter->includeGlobals(),
4382 CodeCompleter->loadExternal());
4383
4384 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
4385 Results.ExitScope();
4386
4387 switch (CompletionContext) {
4388 case PCC_ParenthesizedExpression:
4389 case PCC_Expression:
4390 case PCC_Statement:
4391 case PCC_RecoveryInFunction:
4392 if (S->getFnParent())
4393 AddPrettyFunctionResults(getLangOpts(), Results);
4394 break;
4395
4396 case PCC_Namespace:
4397 case PCC_Class:
4398 case PCC_ObjCInterface:
4399 case PCC_ObjCImplementation:
4400 case PCC_ObjCInstanceVariableList:
4401 case PCC_Template:
4402 case PCC_MemberTemplate:
4403 case PCC_ForInit:
4404 case PCC_Condition:
4405 case PCC_Type:
4406 case PCC_LocalDeclarationSpecifiers:
4407 break;
4408 }
4409
4410 if (CodeCompleter->includeMacros())
4411 AddMacroResults(PP, Results, CodeCompleter->loadExternal(), false);
4412
4413 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4414 Results.data(), Results.size());
4415}
4416
4417static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
4418 ParsedType Receiver,
4420 bool AtArgumentExpression, bool IsSuper,
4421 ResultBuilder &Results);
4422
4424 bool AllowNonIdentifiers,
4425 bool AllowNestedNameSpecifiers) {
4427 ResultBuilder Results(
4428 *this, CodeCompleter->getAllocator(),
4429 CodeCompleter->getCodeCompletionTUInfo(),
4430 AllowNestedNameSpecifiers
4431 // FIXME: Try to separate codepath leading here to deduce whether we
4432 // need an existing symbol or a new one.
4435 Results.EnterNewScope();
4436
4437 // Type qualifiers can come after names.
4438 Results.AddResult(Result("const"));
4439 Results.AddResult(Result("volatile"));
4440 if (getLangOpts().C99)
4441 Results.AddResult(Result("restrict"));
4442
4443 if (getLangOpts().CPlusPlus) {
4444 if (getLangOpts().CPlusPlus11 &&
4447 Results.AddResult("final");
4448
4449 if (AllowNonIdentifiers) {
4450 Results.AddResult(Result("operator"));
4451 }
4452
4453 // Add nested-name-specifiers.
4454 if (AllowNestedNameSpecifiers) {
4455 Results.allowNestedNameSpecifiers();
4456 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
4457 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4458 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
4459 CodeCompleter->includeGlobals(),
4460 CodeCompleter->loadExternal());
4461 Results.setFilter(nullptr);
4462 }
4463 }
4464 Results.ExitScope();
4465
4466 // If we're in a context where we might have an expression (rather than a
4467 // declaration), and what we've seen so far is an Objective-C type that could
4468 // be a receiver of a class message, this may be a class message send with
4469 // the initial opening bracket '[' missing. Add appropriate completions.
4470 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
4475 !DS.isTypeAltiVecVector() && S &&
4476 (S->getFlags() & Scope::DeclScope) != 0 &&
4477 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
4479 0) {
4480 ParsedType T = DS.getRepAsType();
4481 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
4482 AddClassMessageCompletions(*this, S, T, std::nullopt, false, false,
4483 Results);
4484 }
4485
4486 // Note that we intentionally suppress macro results here, since we do not
4487 // encourage using macros to produce the names of entities.
4488
4489 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4490 Results.data(), Results.size());
4491}
4492
4493static const char *underscoreAttrScope(llvm::StringRef Scope) {
4494 if (Scope == "clang")
4495 return "_Clang";
4496 if (Scope == "gnu")
4497 return "__gnu__";
4498 return nullptr;
4499}
4500
4501static const char *noUnderscoreAttrScope(llvm::StringRef Scope) {
4502 if (Scope == "_Clang")
4503 return "clang";
4504 if (Scope == "__gnu__")
4505 return "gnu";
4506 return nullptr;
4507}
4508
4510 AttributeCompletion Completion,
4511 const IdentifierInfo *InScope) {
4512 if (Completion == AttributeCompletion::None)
4513 return;
4514 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4515 CodeCompleter->getCodeCompletionTUInfo(),
4517
4518 // We're going to iterate over the normalized spellings of the attribute.
4519 // These don't include "underscore guarding": the normalized spelling is
4520 // clang::foo but you can also write _Clang::__foo__.
4521 //
4522 // (Clang supports a mix like clang::__foo__ but we won't suggest it: either
4523 // you care about clashing with macros or you don't).
4524 //
4525 // So if we're already in a scope, we determine its canonical spellings
4526 // (for comparison with normalized attr spelling) and remember whether it was
4527 // underscore-guarded (so we know how to spell contained attributes).
4528 llvm::StringRef InScopeName;
4529 bool InScopeUnderscore = false;
4530 if (InScope) {
4531 InScopeName = InScope->getName();
4532 if (const char *NoUnderscore = noUnderscoreAttrScope(InScopeName)) {
4533 InScopeName = NoUnderscore;
4534 InScopeUnderscore = true;
4535 }
4536 }
4537 bool SyntaxSupportsGuards = Syntax == AttributeCommonInfo::AS_GNU ||
4540
4542 auto AddCompletions = [&](const ParsedAttrInfo &A) {
4543 if (A.IsTargetSpecific && !A.existsInTarget(Context.getTargetInfo()))
4544 return;
4545 if (!A.acceptsLangOpts(getLangOpts()))
4546 return;
4547 for (const auto &S : A.Spellings) {
4548 if (S.Syntax != Syntax)
4549 continue;
4550 llvm::StringRef Name = S.NormalizedFullName;
4551 llvm::StringRef Scope;
4552 if ((Syntax == AttributeCommonInfo::AS_CXX11 ||
4553 Syntax == AttributeCommonInfo::AS_C2x)) {
4554 std::tie(Scope, Name) = Name.split("::");
4555 if (Name.empty()) // oops, unscoped
4556 std::swap(Name, Scope);
4557 }
4558
4559 // Do we just want a list of scopes rather than attributes?
4560 if (Completion == AttributeCompletion::Scope) {
4561 // Make sure to emit each scope only once.
4562 if (!Scope.empty() && FoundScopes.insert(Scope).second) {
4563 Results.AddResult(
4564 CodeCompletionResult(Results.getAllocator().CopyString(Scope)));
4565 // Include alternate form (__gnu__ instead of gnu).
4566 if (const char *Scope2 = underscoreAttrScope(Scope))
4567 Results.AddResult(CodeCompletionResult(Scope2));
4568 }
4569 continue;
4570 }
4571
4572 // If a scope was specified, it must match but we don't need to print it.
4573 if (!InScopeName.empty()) {
4574 if (Scope != InScopeName)
4575 continue;
4576 Scope = "";
4577 }
4578
4579 auto Add = [&](llvm::StringRef Scope, llvm::StringRef Name,
4580 bool Underscores) {
4581 CodeCompletionBuilder Builder(Results.getAllocator(),
4582 Results.getCodeCompletionTUInfo());
4584 if (!Scope.empty()) {
4585 Text.append(Scope);
4586 Text.append("::");
4587 }
4588 if (Underscores)
4589 Text.append("__");
4590 Text.append(Name);
4591 if (Underscores)
4592 Text.append("__");
4593 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(Text));
4594
4595 if (!A.ArgNames.empty()) {
4596 Builder.AddChunk(CodeCompletionString::CK_LeftParen, "(");
4597 bool First = true;
4598 for (const char *Arg : A.ArgNames) {
4599 if (!First)
4600 Builder.AddChunk(CodeCompletionString::CK_Comma, ", ");
4601 First = false;
4602 Builder.AddPlaceholderChunk(Arg);
4603 }
4604 Builder.AddChunk(CodeCompletionString::CK_RightParen, ")");
4605 }
4606
4607 Results.AddResult(Builder.TakeString());
4608 };
4609
4610 // Generate the non-underscore-guarded result.
4611 // Note this is (a suffix of) the NormalizedFullName, no need to copy.
4612 // If an underscore-guarded scope was specified, only the
4613 // underscore-guarded attribute name is relevant.
4614 if (!InScopeUnderscore)
4615 Add(Scope, Name, /*Underscores=*/false);
4616
4617 // Generate the underscore-guarded version, for syntaxes that support it.
4618 // We skip this if the scope was already spelled and not guarded, or
4619 // we must spell it and can't guard it.
4620 if (!(InScope && !InScopeUnderscore) && SyntaxSupportsGuards) {
4621 llvm::SmallString<32> Guarded;
4622 if (Scope.empty()) {
4623 Add(Scope, Name, /*Underscores=*/true);
4624 } else {
4625 const char *GuardedScope = underscoreAttrScope(Scope);
4626 if (!GuardedScope)
4627 continue;
4628 Add(GuardedScope, Name, /*Underscores=*/true);
4629 }
4630 }
4631
4632 // It may be nice to include the Kind so we can look up the docs later.
4633 }
4634 };
4635
4636 for (const auto *A : ParsedAttrInfo::getAllBuiltin())
4637 AddCompletions(*A);
4638 for (const auto &Entry : ParsedAttrInfoRegistry::entries())
4639 AddCompletions(*Entry.instantiate());
4640
4641 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4642 Results.data(), Results.size());
4643}
4644
4647 bool IsParenthesized = false)
4648 : PreferredType(PreferredType), IntegralConstantExpression(false),
4649 ObjCCollection(false), IsParenthesized(IsParenthesized) {}
4650
4656};
4657
4658namespace {
4659/// Information that allows to avoid completing redundant enumerators.
4660struct CoveredEnumerators {
4662 NestedNameSpecifier *SuggestedQualifier = nullptr;
4663};
4664} // namespace
4665
4666static void AddEnumerators(ResultBuilder &Results, ASTContext &Context,
4667 EnumDecl *Enum, DeclContext *CurContext,
4668 const CoveredEnumerators &Enumerators) {
4669 NestedNameSpecifier *Qualifier = Enumerators.SuggestedQualifier;
4670 if (Context.getLangOpts().CPlusPlus && !Qualifier && Enumerators.Seen.empty()) {
4671 // If there are no prior enumerators in C++, check whether we have to
4672 // qualify the names of the enumerators that we suggest, because they
4673 // may not be visible in this scope.
4674 Qualifier = getRequiredQualification(Context, CurContext, Enum);
4675 }
4676
4677 Results.EnterNewScope();
4678 for (auto *E : Enum->enumerators()) {
4679 if (Enumerators.Seen.count(E))
4680 continue;
4681
4682 CodeCompletionResult R(E, CCP_EnumInCase, Qualifier);
4683 Results.AddResult(R, CurContext, nullptr, false);
4684 }
4685 Results.ExitScope();
4686}
4687
4688/// Try to find a corresponding FunctionProtoType for function-like types (e.g.
4689/// function pointers, std::function, etc).
4691 assert(!T.isNull());
4692 // Try to extract first template argument from std::function<> and similar.
4693 // Note we only handle the sugared types, they closely match what users wrote.
4694 // We explicitly choose to not handle ClassTemplateSpecializationDecl.
4696 if (Specialization->template_arguments().size() != 1)
4697 return nullptr;
4698 const TemplateArgument &Argument = Specialization->template_arguments()[0];
4699 if (Argument.getKind() != TemplateArgument::Type)
4700 return nullptr;
4701 return Argument.getAsType()->getAs<FunctionProtoType>();
4702 }
4703 // Handle other cases.
4704 if (T->isPointerType())
4705 T = T->getPointeeType();
4706 return T->getAs<FunctionProtoType>();
4707}
4708
4709/// Adds a pattern completion for a lambda expression with the specified
4710/// parameter types and placeholders for parameter names.
4711static void AddLambdaCompletion(ResultBuilder &Results,
4712 llvm::ArrayRef<QualType> Parameters,
4713 const LangOptions &LangOpts) {
4714 if (!Results.includeCodePatterns())
4715 return;
4716 CodeCompletionBuilder Completion(Results.getAllocator(),
4717 Results.getCodeCompletionTUInfo());
4718 // [](<parameters>) {}
4720 Completion.AddPlaceholderChunk("=");
4722 if (!Parameters.empty()) {
4724 bool First = true;
4725 for (auto Parameter : Parameters) {
4726 if (!First)
4728 else
4729 First = false;
4730
4731 constexpr llvm::StringLiteral NamePlaceholder = "!#!NAME_GOES_HERE!#!";
4732 std::string Type = std::string(NamePlaceholder);
4733 Parameter.getAsStringInternal(Type, PrintingPolicy(LangOpts));
4734 llvm::StringRef Prefix, Suffix;
4735 std::tie(Prefix, Suffix) = llvm::StringRef(Type).split(NamePlaceholder);
4736 Prefix = Prefix.rtrim();
4737 Suffix = Suffix.ltrim();
4738
4739 Completion.AddTextChunk(Completion.getAllocator().CopyString(Prefix));
4741 Completion.AddPlaceholderChunk("parameter");
4742 Completion.AddTextChunk(Completion.getAllocator().CopyString(Suffix));
4743 };
4745 }
4749 Completion.AddPlaceholderChunk("body");
4752
4753 Results.AddResult(Completion.TakeString());
4754}
4755
4756/// Perform code-completion in an expression context when we know what
4757/// type we're looking for.
4760 ResultBuilder Results(
4761 *this, CodeCompleter->getAllocator(),
4762 CodeCompleter->getCodeCompletionTUInfo(),
4764 Data.IsParenthesized
4767 Data.PreferredType));
4768 auto PCC =
4769 Data.IsParenthesized ? PCC_ParenthesizedExpression : PCC_Expression;
4770 if (Data.ObjCCollection)
4771 Results.setFilter(&ResultBuilder::IsObjCCollection);
4772 else if (Data.IntegralConstantExpression)
4773 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
4774 else if (WantTypesInContext(PCC, getLangOpts()))
4775 Results.setFilter(&ResultBuilder::IsOrdinaryName);
4776 else
4777 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
4778
4779 if (!Data.PreferredType.isNull())
4780 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
4781
4782 // Ignore any declarations that we were told that we don't care about.
4783 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
4784 Results.Ignore(Data.IgnoreDecls[I]);
4785
4786 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4787 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4788 CodeCompleter->includeGlobals(),
4789 CodeCompleter->loadExternal());
4790
4791 Results.EnterNewScope();
4792 AddOrdinaryNameResults(PCC, S, *this, Results);
4793 Results.ExitScope();
4794
4795 bool PreferredTypeIsPointer = false;
4796 if (!Data.PreferredType.isNull()) {
4797 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType() ||
4798 Data.PreferredType->isMemberPointerType() ||
4799 Data.PreferredType->isBlockPointerType();
4800 if (Data.PreferredType->isEnumeralType()) {
4801 EnumDecl *Enum = Data.PreferredType->castAs<EnumType>()->getDecl();
4802 if (auto *Def = Enum->getDefinition())
4803 Enum = Def;
4804 // FIXME: collect covered enumerators in cases like:
4805 // if (x == my_enum::one) { ... } else if (x == ^) {}
4806 AddEnumerators(Results, Context, Enum, CurContext, CoveredEnumerators());
4807 }
4808 }
4809
4810 if (S->getFnParent() && !Data.ObjCCollection &&
4811 !Data.IntegralConstantExpression)
4812 AddPrettyFunctionResults(getLangOpts(), Results);
4813
4814 if (CodeCompleter->includeMacros())
4815 AddMacroResults(PP, Results, CodeCompleter->loadExternal(), false,
4816 PreferredTypeIsPointer);
4817
4818 // Complete a lambda expression when preferred type is a function.
4819 if (!Data.PreferredType.isNull() && getLangOpts().CPlusPlus11) {
4820 if (const FunctionProtoType *F =
4821 TryDeconstructFunctionLike(Data.PreferredType))
4822 AddLambdaCompletion(Results, F->getParamTypes(), getLangOpts());
4823 }
4824
4825 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4826 Results.data(), Results.size());
4827}
4828
4830 bool IsParenthesized) {
4831 return CodeCompleteExpression(
4832 S, CodeCompleteExpressionData(PreferredType, IsParenthesized));
4833}
4834
4836 QualType PreferredType) {
4837 if (E.isInvalid())
4838 CodeCompleteExpression(S, PreferredType);
4839 else if (getLangOpts().ObjC)
4840 CodeCompleteObjCInstanceMessage(S, E.get(), std::nullopt, false);
4841}
4842
4843/// The set of properties that have already been added, referenced by
4844/// property name.
4846
4847/// Retrieve the container definition, if any?
4849 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
4850 if (Interface->hasDefinition())
4851 return Interface->getDefinition();
4852
4853 return Interface;
4854 }
4855
4856 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
4857 if (Protocol->hasDefinition())
4858 return Protocol->getDefinition();
4859
4860 return Protocol;
4861 }
4862 return Container;
4863}
4864
4865/// Adds a block invocation code completion result for the given block
4866/// declaration \p BD.
4867static void AddObjCBlockCall(ASTContext &Context, const PrintingPolicy &Policy,
4868 CodeCompletionBuilder &Builder,
4869 const NamedDecl *BD,
4870 const FunctionTypeLoc &BlockLoc,
4871 const FunctionProtoTypeLoc &BlockProtoLoc) {
4872 Builder.AddResultTypeChunk(
4873 GetCompletionTypeString(BlockLoc.getReturnLoc().getType(), Context,
4874 Policy, Builder.getAllocator()));
4875
4876 AddTypedNameChunk(Context, Policy, BD, Builder);
4877 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4878
4879 if (BlockProtoLoc && BlockProtoLoc.getTypePtr()->isVariadic()) {
4880 Builder.AddPlaceholderChunk("...");
4881 } else {
4882 for (unsigned I = 0, N = BlockLoc.getNumParams(); I != N; ++I) {
4883 if (I)
4884 Builder.AddChunk(CodeCompletionString::CK_Comma);
4885
4886 // Format the placeholder string.
4887 std::string PlaceholderStr =
4888 FormatFunctionParameter(Policy, BlockLoc.getParam(I));
4889
4890 if (I == N - 1 && BlockProtoLoc &&
4891 BlockProtoLoc.getTypePtr()->isVariadic())
4892 PlaceholderStr += ", ...";
4893
4894 // Add the placeholder string.
4895 Builder.AddPlaceholderChunk(
4896 Builder.getAllocator().CopyString(PlaceholderStr));
4897 }
4898 }
4899
4900 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4901}
4902
4903static void
4905 ObjCContainerDecl *Container, bool AllowCategories,
4906 bool AllowNullaryMethods, DeclContext *CurContext,
4907 AddedPropertiesSet &AddedProperties, ResultBuilder &Results,
4908 bool IsBaseExprStatement = false,
4909 bool IsClassProperty = false, bool InOriginalClass = true) {
4911
4912 // Retrieve the definition.
4913 Container = getContainerDef(Container);
4914
4915 // Add properties in this container.
4916 const auto AddProperty = [&](const ObjCPropertyDecl *P) {
4917 if (!AddedProperties.insert(P->getIdentifier()).second)
4918 return;
4919
4920 // FIXME: Provide block invocation completion for non-statement
4921 // expressions.
4922 if (!P->getType().getTypePtr()->isBlockPointerType() ||
4923 !IsBaseExprStatement) {
4924 Result R = Result(P, Results.getBasePriority(P), nullptr);
4925 if (!InOriginalClass)
4926 setInBaseClass(R);
4927 Results.MaybeAddResult(R, CurContext);
4928 return;
4929 }
4930
4931 // Block setter and invocation completion is provided only when we are able
4932 // to find the FunctionProtoTypeLoc with parameter names for the block.
4933 FunctionTypeLoc BlockLoc;
4934 FunctionProtoTypeLoc BlockProtoLoc;
4935 findTypeLocationForBlockDecl(P->getTypeSourceInfo(), BlockLoc,
4936 BlockProtoLoc);
4937 if (!BlockLoc) {
4938 Result R = Result(P, Results.getBasePriority(P), nullptr);
4939 if (!InOriginalClass)
4940 setInBaseClass(R);
4941 Results.MaybeAddResult(R, CurContext);
4942 return;
4943 }
4944
4945 // The default completion result for block properties should be the block
4946 // invocation completion when the base expression is a statement.
4947 CodeCompletionBuilder Builder(Results.getAllocator(),
4948 Results.getCodeCompletionTUInfo());
4949 AddObjCBlockCall(Container->getASTContext(),
4950 getCompletionPrintingPolicy(Results.getSema()), Builder, P,
4951 BlockLoc, BlockProtoLoc);
4952 Result R = Result(Builder.TakeString(), P, Results.getBasePriority(P));
4953 if (!InOriginalClass)
4954 setInBaseClass(R);
4955 Results.MaybeAddResult(R, CurContext);
4956
4957 // Provide additional block setter completion iff the base expression is a
4958 // statement and the block property is mutable.
4959 if (!P->isReadOnly()) {
4960 CodeCompletionBuilder Builder(Results.getAllocator(),
4961 Results.getCodeCompletionTUInfo());
4962 AddResultTypeChunk(Container->getASTContext(),
4963 getCompletionPrintingPolicy(Results.getSema()), P,
4964 CCContext.getBaseType(), Builder);
4965 Builder.AddTypedTextChunk(
4966 Results.getAllocator().CopyString(P->getName()));
4967 Builder.AddChunk(CodeCompletionString::CK_Equal);
4968
4969 std::string PlaceholderStr = formatBlockPlaceholder(
4970 getCompletionPrintingPolicy(Results.getSema()), P, BlockLoc,
4971 BlockProtoLoc, /*SuppressBlockName=*/true);
4972 // Add the placeholder string.
4973 Builder.AddPlaceholderChunk(
4974 Builder.getAllocator().CopyString(PlaceholderStr));
4975
4976 // When completing blocks properties that return void the default
4977 // property completion result should show up before the setter,
4978 // otherwise the setter completion should show up before the default
4979 // property completion, as we normally want to use the result of the
4980 // call.
4981 Result R =
4982 Result(Builder.TakeString(), P,
4983 Results.getBasePriority(P) +
4984 (BlockLoc.getTypePtr()->getReturnType()->