blob: f441befcfbf95b2b179c26a19d123d43c7ca7d2a [file] [log] [blame]
Douglas Gregor2436e712009-09-17 21:32:03 +00001//===---------------- SemaCodeComplete.cpp - Code Completion ----*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file defines the code-completion semantic actions.
11//
12//===----------------------------------------------------------------------===//
John McCall83024632010-08-25 22:03:47 +000013#include "clang/Sema/SemaInternal.h"
John McCallde6836a2010-08-24 07:21:54 +000014#include "clang/AST/DeclObjC.h"
Douglas Gregorf2510672009-09-21 19:57:38 +000015#include "clang/AST/ExprCXX.h"
Douglas Gregor8ce33212009-11-17 17:59:40 +000016#include "clang/AST/ExprObjC.h"
Jordan Rose4938f272013-02-09 10:09:43 +000017#include "clang/Basic/CharInfo.h"
Douglas Gregor07f43572012-01-29 18:15:03 +000018#include "clang/Lex/HeaderSearch.h"
Douglas Gregorf329c7c2009-10-30 16:50:04 +000019#include "clang/Lex/MacroInfo.h"
20#include "clang/Lex/Preprocessor.h"
Chandler Carruth3a022472012-12-04 09:13:33 +000021#include "clang/Sema/CodeCompleteConsumer.h"
22#include "clang/Sema/ExternalSemaSource.h"
23#include "clang/Sema/Lookup.h"
24#include "clang/Sema/Overload.h"
25#include "clang/Sema/Scope.h"
26#include "clang/Sema/ScopeInfo.h"
Douglas Gregor1154e272010-09-16 16:06:31 +000027#include "llvm/ADT/DenseSet.h"
Benjamin Kramere0513cb2012-01-30 16:17:39 +000028#include "llvm/ADT/SmallBitVector.h"
Douglas Gregor3545ff42009-09-21 16:56:56 +000029#include "llvm/ADT/SmallPtrSet.h"
Benjamin Kramer49038022012-02-04 13:45:25 +000030#include "llvm/ADT/SmallString.h"
Douglas Gregore6688e62009-09-28 03:51:44 +000031#include "llvm/ADT/StringExtras.h"
Douglas Gregor9d2ddb22010-04-06 19:22:33 +000032#include "llvm/ADT/StringSwitch.h"
Douglas Gregor67c692c2010-08-26 15:07:07 +000033#include "llvm/ADT/Twine.h"
Douglas Gregor3545ff42009-09-21 16:56:56 +000034#include <list>
35#include <map>
36#include <vector>
Douglas Gregor2436e712009-09-17 21:32:03 +000037
38using namespace clang;
John McCallaab3e412010-08-25 08:40:02 +000039using namespace sema;
Douglas Gregor2436e712009-09-17 21:32:03 +000040
Douglas Gregor3545ff42009-09-21 16:56:56 +000041namespace {
42 /// \brief A container of code-completion results.
43 class ResultBuilder {
44 public:
45 /// \brief The type of a name-lookup filter, which can be provided to the
46 /// name-lookup routines to specify which declarations should be included in
47 /// the result set (when it returns true) and which declarations should be
48 /// filtered out (returns false).
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000049 typedef bool (ResultBuilder::*LookupFilter)(const NamedDecl *) const;
Douglas Gregor3545ff42009-09-21 16:56:56 +000050
John McCall276321a2010-08-25 06:19:51 +000051 typedef CodeCompletionResult Result;
Douglas Gregor3545ff42009-09-21 16:56:56 +000052
53 private:
54 /// \brief The actual results we have found.
55 std::vector<Result> Results;
56
57 /// \brief A record of all of the declarations we have found and placed
58 /// into the result set, used to ensure that no declaration ever gets into
59 /// the result set twice.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000060 llvm::SmallPtrSet<const Decl*, 16> AllDeclsFound;
Douglas Gregor3545ff42009-09-21 16:56:56 +000061
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000062 typedef std::pair<const NamedDecl *, unsigned> DeclIndexPair;
Douglas Gregor05e7ca32009-12-06 20:23:50 +000063
64 /// \brief An entry in the shadow map, which is optimized to store
65 /// a single (declaration, index) mapping (the common case) but
66 /// can also store a list of (declaration, index) mappings.
67 class ShadowMapEntry {
Chris Lattner0e62c1c2011-07-23 10:55:15 +000068 typedef SmallVector<DeclIndexPair, 4> DeclIndexPairVector;
Douglas Gregor05e7ca32009-12-06 20:23:50 +000069
70 /// \brief Contains either the solitary NamedDecl * or a vector
71 /// of (declaration, index) pairs.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000072 llvm::PointerUnion<const NamedDecl *, DeclIndexPairVector*> DeclOrVector;
Douglas Gregor05e7ca32009-12-06 20:23:50 +000073
74 /// \brief When the entry contains a single declaration, this is
75 /// the index associated with that entry.
76 unsigned SingleDeclIndex;
77
78 public:
79 ShadowMapEntry() : DeclOrVector(), SingleDeclIndex(0) { }
80
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000081 void Add(const NamedDecl *ND, unsigned Index) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +000082 if (DeclOrVector.isNull()) {
83 // 0 - > 1 elements: just set the single element information.
84 DeclOrVector = ND;
85 SingleDeclIndex = Index;
86 return;
87 }
88
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000089 if (const NamedDecl *PrevND =
90 DeclOrVector.dyn_cast<const NamedDecl *>()) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +000091 // 1 -> 2 elements: create the vector of results and push in the
92 // existing declaration.
93 DeclIndexPairVector *Vec = new DeclIndexPairVector;
94 Vec->push_back(DeclIndexPair(PrevND, SingleDeclIndex));
95 DeclOrVector = Vec;
96 }
97
98 // Add the new element to the end of the vector.
99 DeclOrVector.get<DeclIndexPairVector*>()->push_back(
100 DeclIndexPair(ND, Index));
101 }
102
103 void Destroy() {
104 if (DeclIndexPairVector *Vec
105 = DeclOrVector.dyn_cast<DeclIndexPairVector *>()) {
106 delete Vec;
Craig Topperc3ec1492014-05-26 06:22:03 +0000107 DeclOrVector = ((NamedDecl *)nullptr);
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000108 }
109 }
110
111 // Iteration.
112 class iterator;
113 iterator begin() const;
114 iterator end() const;
115 };
116
Douglas Gregor3545ff42009-09-21 16:56:56 +0000117 /// \brief A mapping from declaration names to the declarations that have
118 /// this name within a particular scope and their index within the list of
119 /// results.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000120 typedef llvm::DenseMap<DeclarationName, ShadowMapEntry> ShadowMap;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000121
122 /// \brief The semantic analysis object for which results are being
123 /// produced.
124 Sema &SemaRef;
Douglas Gregorb278aaf2011-02-01 19:23:04 +0000125
126 /// \brief The allocator used to allocate new code-completion strings.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +0000127 CodeCompletionAllocator &Allocator;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +0000128
129 CodeCompletionTUInfo &CCTUInfo;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000130
131 /// \brief If non-NULL, a filter function used to remove any code-completion
132 /// results that are not desirable.
133 LookupFilter Filter;
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000134
135 /// \brief Whether we should allow declarations as
136 /// nested-name-specifiers that would otherwise be filtered out.
137 bool AllowNestedNameSpecifiers;
138
Douglas Gregor7aa6b222010-05-30 01:49:25 +0000139 /// \brief If set, the type that we would prefer our resulting value
140 /// declarations to have.
141 ///
142 /// Closely matching the preferred type gives a boost to a result's
143 /// priority.
144 CanQualType PreferredType;
145
Douglas Gregor3545ff42009-09-21 16:56:56 +0000146 /// \brief A list of shadow maps, which is used to model name hiding at
147 /// different levels of, e.g., the inheritance hierarchy.
148 std::list<ShadowMap> ShadowMaps;
149
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000150 /// \brief If we're potentially referring to a C++ member function, the set
151 /// of qualifiers applied to the object type.
152 Qualifiers ObjectTypeQualifiers;
153
154 /// \brief Whether the \p ObjectTypeQualifiers field is active.
155 bool HasObjectTypeQualifiers;
156
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000157 /// \brief The selector that we prefer.
158 Selector PreferredSelector;
159
Douglas Gregor05fcf842010-11-02 20:36:02 +0000160 /// \brief The completion context in which we are gathering results.
Douglas Gregor50832e02010-09-20 22:39:41 +0000161 CodeCompletionContext CompletionContext;
162
James Dennett596e4752012-06-14 03:11:41 +0000163 /// \brief If we are in an instance method definition, the \@implementation
Douglas Gregor05fcf842010-11-02 20:36:02 +0000164 /// object.
165 ObjCImplementationDecl *ObjCImplementation;
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000166
Douglas Gregor50832e02010-09-20 22:39:41 +0000167 void AdjustResultPriorityForDecl(Result &R);
Douglas Gregor95887f92010-07-08 23:20:03 +0000168
Douglas Gregor0212fd72010-09-21 16:06:22 +0000169 void MaybeAddConstructorResults(Result R);
170
Douglas Gregor3545ff42009-09-21 16:56:56 +0000171 public:
Douglas Gregorbcbf46c2011-02-01 22:57:45 +0000172 explicit ResultBuilder(Sema &SemaRef, CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +0000173 CodeCompletionTUInfo &CCTUInfo,
Douglas Gregor0ac41382010-09-23 23:01:17 +0000174 const CodeCompletionContext &CompletionContext,
Craig Topperc3ec1492014-05-26 06:22:03 +0000175 LookupFilter Filter = nullptr)
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +0000176 : SemaRef(SemaRef), Allocator(Allocator), CCTUInfo(CCTUInfo),
177 Filter(Filter),
Douglas Gregorb278aaf2011-02-01 19:23:04 +0000178 AllowNestedNameSpecifiers(false), HasObjectTypeQualifiers(false),
Douglas Gregor05fcf842010-11-02 20:36:02 +0000179 CompletionContext(CompletionContext),
Craig Topperc3ec1492014-05-26 06:22:03 +0000180 ObjCImplementation(nullptr)
Douglas Gregor05fcf842010-11-02 20:36:02 +0000181 {
182 // If this is an Objective-C instance method definition, dig out the
183 // corresponding implementation.
184 switch (CompletionContext.getKind()) {
185 case CodeCompletionContext::CCC_Expression:
186 case CodeCompletionContext::CCC_ObjCMessageReceiver:
187 case CodeCompletionContext::CCC_ParenthesizedExpression:
188 case CodeCompletionContext::CCC_Statement:
189 case CodeCompletionContext::CCC_Recovery:
190 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl())
191 if (Method->isInstanceMethod())
192 if (ObjCInterfaceDecl *Interface = Method->getClassInterface())
193 ObjCImplementation = Interface->getImplementation();
194 break;
195
196 default:
197 break;
198 }
199 }
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000200
201 /// \brief Determine the priority for a reference to the given declaration.
202 unsigned getBasePriority(const NamedDecl *D);
203
Douglas Gregorf64acca2010-05-25 21:41:55 +0000204 /// \brief Whether we should include code patterns in the completion
205 /// results.
206 bool includeCodePatterns() const {
207 return SemaRef.CodeCompleter &&
Douglas Gregorac322ec2010-08-27 21:18:54 +0000208 SemaRef.CodeCompleter->includeCodePatterns();
Douglas Gregorf64acca2010-05-25 21:41:55 +0000209 }
210
Douglas Gregor3545ff42009-09-21 16:56:56 +0000211 /// \brief Set the filter used for code-completion results.
212 void setFilter(LookupFilter Filter) {
213 this->Filter = Filter;
214 }
Craig Topperc3ec1492014-05-26 06:22:03 +0000215
216 Result *data() { return Results.empty()? nullptr : &Results.front(); }
Douglas Gregor3545ff42009-09-21 16:56:56 +0000217 unsigned size() const { return Results.size(); }
218 bool empty() const { return Results.empty(); }
219
Douglas Gregor7aa6b222010-05-30 01:49:25 +0000220 /// \brief Specify the preferred type.
221 void setPreferredType(QualType T) {
222 PreferredType = SemaRef.Context.getCanonicalType(T);
223 }
224
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000225 /// \brief Set the cv-qualifiers on the object type, for us in filtering
226 /// calls to member functions.
227 ///
228 /// When there are qualifiers in this set, they will be used to filter
229 /// out member functions that aren't available (because there will be a
230 /// cv-qualifier mismatch) or prefer functions with an exact qualifier
231 /// match.
232 void setObjectTypeQualifiers(Qualifiers Quals) {
233 ObjectTypeQualifiers = Quals;
234 HasObjectTypeQualifiers = true;
235 }
236
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000237 /// \brief Set the preferred selector.
238 ///
239 /// When an Objective-C method declaration result is added, and that
240 /// method's selector matches this preferred selector, we give that method
241 /// a slight priority boost.
242 void setPreferredSelector(Selector Sel) {
243 PreferredSelector = Sel;
244 }
Douglas Gregor05fcf842010-11-02 20:36:02 +0000245
Douglas Gregor50832e02010-09-20 22:39:41 +0000246 /// \brief Retrieve the code-completion context for which results are
247 /// being collected.
248 const CodeCompletionContext &getCompletionContext() const {
249 return CompletionContext;
250 }
251
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000252 /// \brief Specify whether nested-name-specifiers are allowed.
253 void allowNestedNameSpecifiers(bool Allow = true) {
254 AllowNestedNameSpecifiers = Allow;
255 }
256
Douglas Gregor74661272010-09-21 00:03:25 +0000257 /// \brief Return the semantic analysis object for which we are collecting
258 /// code completion results.
259 Sema &getSema() const { return SemaRef; }
260
Douglas Gregorb278aaf2011-02-01 19:23:04 +0000261 /// \brief Retrieve the allocator used to allocate code completion strings.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +0000262 CodeCompletionAllocator &getAllocator() const { return Allocator; }
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +0000263
264 CodeCompletionTUInfo &getCodeCompletionTUInfo() const { return CCTUInfo; }
Douglas Gregorb278aaf2011-02-01 19:23:04 +0000265
Douglas Gregor7c208612010-01-14 00:20:49 +0000266 /// \brief Determine whether the given declaration is at all interesting
267 /// as a code-completion result.
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000268 ///
269 /// \param ND the declaration that we are inspecting.
270 ///
271 /// \param AsNestedNameSpecifier will be set true if this declaration is
272 /// only interesting when it is a nested-name-specifier.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000273 bool isInterestingDecl(const NamedDecl *ND,
274 bool &AsNestedNameSpecifier) const;
Douglas Gregore0717ab2010-01-14 00:41:07 +0000275
276 /// \brief Check whether the result is hidden by the Hiding declaration.
277 ///
278 /// \returns true if the result is hidden and cannot be found, false if
279 /// the hidden result could still be found. When false, \p R may be
280 /// modified to describe how the result can be found (e.g., via extra
281 /// qualification).
282 bool CheckHiddenResult(Result &R, DeclContext *CurContext,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000283 const NamedDecl *Hiding);
Douglas Gregore0717ab2010-01-14 00:41:07 +0000284
Douglas Gregor3545ff42009-09-21 16:56:56 +0000285 /// \brief Add a new result to this result set (if it isn't already in one
286 /// of the shadow maps), or replace an existing result (for, e.g., a
287 /// redeclaration).
Douglas Gregor2af2f672009-09-21 20:12:40 +0000288 ///
Douglas Gregord8c61782012-02-15 15:34:24 +0000289 /// \param R the result to add (if it is unique).
Douglas Gregor2af2f672009-09-21 20:12:40 +0000290 ///
Douglas Gregord8c61782012-02-15 15:34:24 +0000291 /// \param CurContext the context in which this result will be named.
Craig Topperc3ec1492014-05-26 06:22:03 +0000292 void MaybeAddResult(Result R, DeclContext *CurContext = nullptr);
293
Douglas Gregorc580c522010-01-14 01:09:38 +0000294 /// \brief Add a new result to this result set, where we already know
295 /// the hiding declation (if any).
296 ///
297 /// \param R the result to add (if it is unique).
298 ///
299 /// \param CurContext the context in which this result will be named.
300 ///
301 /// \param Hiding the declaration that hides the result.
Douglas Gregor09bbc652010-01-14 15:47:35 +0000302 ///
303 /// \param InBaseClass whether the result was found in a base
304 /// class of the searched context.
305 void AddResult(Result R, DeclContext *CurContext, NamedDecl *Hiding,
306 bool InBaseClass);
Douglas Gregorc580c522010-01-14 01:09:38 +0000307
Douglas Gregor78a21012010-01-14 16:01:26 +0000308 /// \brief Add a new non-declaration result to this result set.
309 void AddResult(Result R);
310
Douglas Gregor3545ff42009-09-21 16:56:56 +0000311 /// \brief Enter into a new scope.
312 void EnterNewScope();
313
314 /// \brief Exit from the current scope.
315 void ExitScope();
316
Douglas Gregorbaf69612009-11-18 04:19:12 +0000317 /// \brief Ignore this declaration, if it is seen again.
Dmitri Gribenko6cfb1532013-02-14 13:53:30 +0000318 void Ignore(const Decl *D) { AllDeclsFound.insert(D->getCanonicalDecl()); }
Douglas Gregorbaf69612009-11-18 04:19:12 +0000319
Douglas Gregor3545ff42009-09-21 16:56:56 +0000320 /// \name Name lookup predicates
321 ///
322 /// These predicates can be passed to the name lookup functions to filter the
323 /// results of name lookup. All of the predicates have the same type, so that
324 ///
325 //@{
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000326 bool IsOrdinaryName(const NamedDecl *ND) const;
327 bool IsOrdinaryNonTypeName(const NamedDecl *ND) const;
328 bool IsIntegralConstantValue(const NamedDecl *ND) const;
329 bool IsOrdinaryNonValueName(const NamedDecl *ND) const;
330 bool IsNestedNameSpecifier(const NamedDecl *ND) const;
331 bool IsEnum(const NamedDecl *ND) const;
332 bool IsClassOrStruct(const NamedDecl *ND) const;
333 bool IsUnion(const NamedDecl *ND) const;
334 bool IsNamespace(const NamedDecl *ND) const;
335 bool IsNamespaceOrAlias(const NamedDecl *ND) const;
336 bool IsType(const NamedDecl *ND) const;
337 bool IsMember(const NamedDecl *ND) const;
338 bool IsObjCIvar(const NamedDecl *ND) const;
339 bool IsObjCMessageReceiver(const NamedDecl *ND) const;
340 bool IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const;
341 bool IsObjCCollection(const NamedDecl *ND) const;
342 bool IsImpossibleToSatisfy(const NamedDecl *ND) const;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000343 //@}
344 };
345}
346
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000347class ResultBuilder::ShadowMapEntry::iterator {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000348 llvm::PointerUnion<const NamedDecl *, const DeclIndexPair *> DeclOrIterator;
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000349 unsigned SingleDeclIndex;
350
351public:
352 typedef DeclIndexPair value_type;
353 typedef value_type reference;
354 typedef std::ptrdiff_t difference_type;
355 typedef std::input_iterator_tag iterator_category;
356
357 class pointer {
358 DeclIndexPair Value;
359
360 public:
361 pointer(const DeclIndexPair &Value) : Value(Value) { }
362
363 const DeclIndexPair *operator->() const {
364 return &Value;
365 }
366 };
Craig Topperc3ec1492014-05-26 06:22:03 +0000367
368 iterator() : DeclOrIterator((NamedDecl *)nullptr), SingleDeclIndex(0) {}
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000369
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000370 iterator(const NamedDecl *SingleDecl, unsigned Index)
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000371 : DeclOrIterator(SingleDecl), SingleDeclIndex(Index) { }
372
373 iterator(const DeclIndexPair *Iterator)
374 : DeclOrIterator(Iterator), SingleDeclIndex(0) { }
375
376 iterator &operator++() {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000377 if (DeclOrIterator.is<const NamedDecl *>()) {
Craig Topperc3ec1492014-05-26 06:22:03 +0000378 DeclOrIterator = (NamedDecl *)nullptr;
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000379 SingleDeclIndex = 0;
380 return *this;
381 }
382
383 const DeclIndexPair *I = DeclOrIterator.get<const DeclIndexPair*>();
384 ++I;
385 DeclOrIterator = I;
386 return *this;
387 }
388
Chris Lattner9795b392010-09-04 18:12:20 +0000389 /*iterator operator++(int) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000390 iterator tmp(*this);
391 ++(*this);
392 return tmp;
Chris Lattner9795b392010-09-04 18:12:20 +0000393 }*/
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000394
395 reference operator*() const {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000396 if (const NamedDecl *ND = DeclOrIterator.dyn_cast<const NamedDecl *>())
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000397 return reference(ND, SingleDeclIndex);
398
Douglas Gregor94bb5e82009-12-06 21:27:58 +0000399 return *DeclOrIterator.get<const DeclIndexPair*>();
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000400 }
401
402 pointer operator->() const {
403 return pointer(**this);
404 }
405
406 friend bool operator==(const iterator &X, const iterator &Y) {
Douglas Gregor94bb5e82009-12-06 21:27:58 +0000407 return X.DeclOrIterator.getOpaqueValue()
408 == Y.DeclOrIterator.getOpaqueValue() &&
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000409 X.SingleDeclIndex == Y.SingleDeclIndex;
410 }
411
412 friend bool operator!=(const iterator &X, const iterator &Y) {
Douglas Gregor94bb5e82009-12-06 21:27:58 +0000413 return !(X == Y);
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000414 }
415};
416
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000417ResultBuilder::ShadowMapEntry::iterator
418ResultBuilder::ShadowMapEntry::begin() const {
419 if (DeclOrVector.isNull())
420 return iterator();
421
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000422 if (const NamedDecl *ND = DeclOrVector.dyn_cast<const NamedDecl *>())
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000423 return iterator(ND, SingleDeclIndex);
424
425 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->begin());
426}
427
428ResultBuilder::ShadowMapEntry::iterator
429ResultBuilder::ShadowMapEntry::end() const {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000430 if (DeclOrVector.is<const NamedDecl *>() || DeclOrVector.isNull())
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000431 return iterator();
432
433 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->end());
434}
435
Douglas Gregor2af2f672009-09-21 20:12:40 +0000436/// \brief Compute the qualification required to get from the current context
437/// (\p CurContext) to the target context (\p TargetContext).
438///
439/// \param Context the AST context in which the qualification will be used.
440///
441/// \param CurContext the context where an entity is being named, which is
442/// typically based on the current scope.
443///
444/// \param TargetContext the context in which the named entity actually
445/// resides.
446///
447/// \returns a nested name specifier that refers into the target context, or
448/// NULL if no qualification is needed.
449static NestedNameSpecifier *
450getRequiredQualification(ASTContext &Context,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000451 const DeclContext *CurContext,
452 const DeclContext *TargetContext) {
453 SmallVector<const DeclContext *, 4> TargetParents;
Douglas Gregor2af2f672009-09-21 20:12:40 +0000454
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000455 for (const DeclContext *CommonAncestor = TargetContext;
Douglas Gregor2af2f672009-09-21 20:12:40 +0000456 CommonAncestor && !CommonAncestor->Encloses(CurContext);
457 CommonAncestor = CommonAncestor->getLookupParent()) {
458 if (CommonAncestor->isTransparentContext() ||
459 CommonAncestor->isFunctionOrMethod())
460 continue;
461
462 TargetParents.push_back(CommonAncestor);
463 }
Craig Topperc3ec1492014-05-26 06:22:03 +0000464
465 NestedNameSpecifier *Result = nullptr;
Douglas Gregor2af2f672009-09-21 20:12:40 +0000466 while (!TargetParents.empty()) {
Robert Wilhelm25284cc2013-08-23 16:11:15 +0000467 const DeclContext *Parent = TargetParents.pop_back_val();
468
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000469 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Parent)) {
Douglas Gregor68762e72010-08-23 21:17:50 +0000470 if (!Namespace->getIdentifier())
471 continue;
472
Douglas Gregor2af2f672009-09-21 20:12:40 +0000473 Result = NestedNameSpecifier::Create(Context, Result, Namespace);
Douglas Gregor68762e72010-08-23 21:17:50 +0000474 }
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000475 else if (const TagDecl *TD = dyn_cast<TagDecl>(Parent))
Douglas Gregor2af2f672009-09-21 20:12:40 +0000476 Result = NestedNameSpecifier::Create(Context, Result,
477 false,
478 Context.getTypeDeclType(TD).getTypePtr());
Douglas Gregor9eb77012009-11-07 00:00:49 +0000479 }
Douglas Gregor2af2f672009-09-21 20:12:40 +0000480 return Result;
481}
482
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000483bool ResultBuilder::isInterestingDecl(const NamedDecl *ND,
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000484 bool &AsNestedNameSpecifier) const {
485 AsNestedNameSpecifier = false;
486
Douglas Gregor7c208612010-01-14 00:20:49 +0000487 ND = ND->getUnderlyingDecl();
488 unsigned IDNS = ND->getIdentifierNamespace();
Douglas Gregor58acf322009-10-09 22:16:47 +0000489
490 // Skip unnamed entities.
Douglas Gregor7c208612010-01-14 00:20:49 +0000491 if (!ND->getDeclName())
492 return false;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000493
494 // Friend declarations and declarations introduced due to friends are never
495 // added as results.
John McCallbbbbe4e2010-03-11 07:50:04 +0000496 if (IDNS & (Decl::IDNS_OrdinaryFriend | Decl::IDNS_TagFriend))
Douglas Gregor7c208612010-01-14 00:20:49 +0000497 return false;
498
Douglas Gregor99fe2ad2009-12-11 17:31:05 +0000499 // Class template (partial) specializations are never added as results.
Douglas Gregor7c208612010-01-14 00:20:49 +0000500 if (isa<ClassTemplateSpecializationDecl>(ND) ||
501 isa<ClassTemplatePartialSpecializationDecl>(ND))
502 return false;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000503
Douglas Gregor99fe2ad2009-12-11 17:31:05 +0000504 // Using declarations themselves are never added as results.
Douglas Gregor7c208612010-01-14 00:20:49 +0000505 if (isa<UsingDecl>(ND))
506 return false;
507
508 // Some declarations have reserved names that we don't want to ever show.
509 if (const IdentifierInfo *Id = ND->getIdentifier()) {
Douglas Gregor3545ff42009-09-21 16:56:56 +0000510 // __va_list_tag is a freak of nature. Find it and skip it.
511 if (Id->isStr("__va_list_tag") || Id->isStr("__builtin_va_list"))
Douglas Gregor7c208612010-01-14 00:20:49 +0000512 return false;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000513
Douglas Gregor58acf322009-10-09 22:16:47 +0000514 // Filter out names reserved for the implementation (C99 7.1.3,
Douglas Gregor9f1570d2010-07-14 17:44:04 +0000515 // C++ [lib.global.names]) if they come from a system header.
Daniel Dunbar2c422dc92009-10-18 20:26:12 +0000516 //
517 // FIXME: Add predicate for this.
Douglas Gregor58acf322009-10-09 22:16:47 +0000518 if (Id->getLength() >= 2) {
Daniel Dunbar2c422dc92009-10-18 20:26:12 +0000519 const char *Name = Id->getNameStart();
Douglas Gregor58acf322009-10-09 22:16:47 +0000520 if (Name[0] == '_' &&
Douglas Gregor9f1570d2010-07-14 17:44:04 +0000521 (Name[1] == '_' || (Name[1] >= 'A' && Name[1] <= 'Z')) &&
522 (ND->getLocation().isInvalid() ||
523 SemaRef.SourceMgr.isInSystemHeader(
524 SemaRef.SourceMgr.getSpellingLoc(ND->getLocation()))))
Douglas Gregor7c208612010-01-14 00:20:49 +0000525 return false;
Douglas Gregor58acf322009-10-09 22:16:47 +0000526 }
Douglas Gregor3545ff42009-09-21 16:56:56 +0000527 }
Douglas Gregor2927c0c2010-11-09 03:59:40 +0000528
Douglas Gregor59cab552010-08-16 23:05:20 +0000529 if (Filter == &ResultBuilder::IsNestedNameSpecifier ||
530 ((isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND)) &&
531 Filter != &ResultBuilder::IsNamespace &&
Douglas Gregor0ac41382010-09-23 23:01:17 +0000532 Filter != &ResultBuilder::IsNamespaceOrAlias &&
Craig Topperc3ec1492014-05-26 06:22:03 +0000533 Filter != nullptr))
Douglas Gregor59cab552010-08-16 23:05:20 +0000534 AsNestedNameSpecifier = true;
535
Douglas Gregor3545ff42009-09-21 16:56:56 +0000536 // Filter out any unwanted results.
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000537 if (Filter && !(this->*Filter)(ND)) {
538 // Check whether it is interesting as a nested-name-specifier.
David Blaikiebbafb8a2012-03-11 07:00:24 +0000539 if (AllowNestedNameSpecifiers && SemaRef.getLangOpts().CPlusPlus &&
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000540 IsNestedNameSpecifier(ND) &&
541 (Filter != &ResultBuilder::IsMember ||
542 (isa<CXXRecordDecl>(ND) &&
543 cast<CXXRecordDecl>(ND)->isInjectedClassName()))) {
544 AsNestedNameSpecifier = true;
545 return true;
546 }
547
Douglas Gregor7c208612010-01-14 00:20:49 +0000548 return false;
Douglas Gregor59cab552010-08-16 23:05:20 +0000549 }
Douglas Gregor7c208612010-01-14 00:20:49 +0000550 // ... then it must be interesting!
551 return true;
552}
553
Douglas Gregore0717ab2010-01-14 00:41:07 +0000554bool ResultBuilder::CheckHiddenResult(Result &R, DeclContext *CurContext,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000555 const NamedDecl *Hiding) {
Douglas Gregore0717ab2010-01-14 00:41:07 +0000556 // In C, there is no way to refer to a hidden name.
557 // FIXME: This isn't true; we can find a tag name hidden by an ordinary
558 // name if we introduce the tag type.
David Blaikiebbafb8a2012-03-11 07:00:24 +0000559 if (!SemaRef.getLangOpts().CPlusPlus)
Douglas Gregore0717ab2010-01-14 00:41:07 +0000560 return true;
561
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000562 const DeclContext *HiddenCtx =
563 R.Declaration->getDeclContext()->getRedeclContext();
Douglas Gregore0717ab2010-01-14 00:41:07 +0000564
565 // There is no way to qualify a name declared in a function or method.
566 if (HiddenCtx->isFunctionOrMethod())
567 return true;
568
Sebastian Redl50c68252010-08-31 00:36:30 +0000569 if (HiddenCtx == Hiding->getDeclContext()->getRedeclContext())
Douglas Gregore0717ab2010-01-14 00:41:07 +0000570 return true;
571
572 // We can refer to the result with the appropriate qualification. Do it.
573 R.Hidden = true;
574 R.QualifierIsInformative = false;
575
576 if (!R.Qualifier)
577 R.Qualifier = getRequiredQualification(SemaRef.Context,
578 CurContext,
579 R.Declaration->getDeclContext());
580 return false;
581}
582
Douglas Gregor95887f92010-07-08 23:20:03 +0000583/// \brief A simplified classification of types used to determine whether two
584/// types are "similar enough" when adjusting priorities.
Douglas Gregor6e240332010-08-16 16:18:59 +0000585SimplifiedTypeClass clang::getSimplifiedTypeClass(CanQualType T) {
Douglas Gregor95887f92010-07-08 23:20:03 +0000586 switch (T->getTypeClass()) {
587 case Type::Builtin:
588 switch (cast<BuiltinType>(T)->getKind()) {
589 case BuiltinType::Void:
590 return STC_Void;
591
592 case BuiltinType::NullPtr:
593 return STC_Pointer;
594
595 case BuiltinType::Overload:
596 case BuiltinType::Dependent:
Douglas Gregor95887f92010-07-08 23:20:03 +0000597 return STC_Other;
598
599 case BuiltinType::ObjCId:
600 case BuiltinType::ObjCClass:
601 case BuiltinType::ObjCSel:
602 return STC_ObjectiveC;
603
604 default:
605 return STC_Arithmetic;
606 }
David Blaikie8a40f702012-01-17 06:56:22 +0000607
Douglas Gregor95887f92010-07-08 23:20:03 +0000608 case Type::Complex:
609 return STC_Arithmetic;
610
611 case Type::Pointer:
612 return STC_Pointer;
613
614 case Type::BlockPointer:
615 return STC_Block;
616
617 case Type::LValueReference:
618 case Type::RValueReference:
619 return getSimplifiedTypeClass(T->getAs<ReferenceType>()->getPointeeType());
620
621 case Type::ConstantArray:
622 case Type::IncompleteArray:
623 case Type::VariableArray:
624 case Type::DependentSizedArray:
625 return STC_Array;
626
627 case Type::DependentSizedExtVector:
628 case Type::Vector:
629 case Type::ExtVector:
630 return STC_Arithmetic;
631
632 case Type::FunctionProto:
633 case Type::FunctionNoProto:
634 return STC_Function;
635
636 case Type::Record:
637 return STC_Record;
638
639 case Type::Enum:
640 return STC_Arithmetic;
641
642 case Type::ObjCObject:
643 case Type::ObjCInterface:
644 case Type::ObjCObjectPointer:
645 return STC_ObjectiveC;
646
647 default:
648 return STC_Other;
649 }
650}
651
652/// \brief Get the type that a given expression will have if this declaration
653/// is used as an expression in its "typical" code-completion form.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000654QualType clang::getDeclUsageType(ASTContext &C, const NamedDecl *ND) {
Douglas Gregor95887f92010-07-08 23:20:03 +0000655 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
656
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000657 if (const TypeDecl *Type = dyn_cast<TypeDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000658 return C.getTypeDeclType(Type);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000659 if (const ObjCInterfaceDecl *Iface = dyn_cast<ObjCInterfaceDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000660 return C.getObjCInterfaceType(Iface);
661
662 QualType T;
Alp Tokera2794f92014-01-22 07:29:52 +0000663 if (const FunctionDecl *Function = ND->getAsFunction())
Douglas Gregor603d81b2010-07-13 08:18:22 +0000664 T = Function->getCallResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000665 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregor603d81b2010-07-13 08:18:22 +0000666 T = Method->getSendResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000667 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000668 T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext()));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000669 else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000670 T = Property->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000671 else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000672 T = Value->getType();
673 else
674 return QualType();
Douglas Gregoraf670a82011-04-14 20:33:34 +0000675
676 // Dig through references, function pointers, and block pointers to
677 // get down to the likely type of an expression when the entity is
678 // used.
679 do {
680 if (const ReferenceType *Ref = T->getAs<ReferenceType>()) {
681 T = Ref->getPointeeType();
682 continue;
683 }
684
685 if (const PointerType *Pointer = T->getAs<PointerType>()) {
686 if (Pointer->getPointeeType()->isFunctionType()) {
687 T = Pointer->getPointeeType();
688 continue;
689 }
690
691 break;
692 }
693
694 if (const BlockPointerType *Block = T->getAs<BlockPointerType>()) {
695 T = Block->getPointeeType();
696 continue;
697 }
698
699 if (const FunctionType *Function = T->getAs<FunctionType>()) {
Alp Toker314cc812014-01-25 16:55:45 +0000700 T = Function->getReturnType();
Douglas Gregoraf670a82011-04-14 20:33:34 +0000701 continue;
702 }
703
704 break;
705 } while (true);
706
707 return T;
Douglas Gregor95887f92010-07-08 23:20:03 +0000708}
709
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000710unsigned ResultBuilder::getBasePriority(const NamedDecl *ND) {
711 if (!ND)
712 return CCP_Unlikely;
713
714 // Context-based decisions.
Richard Smith541b38b2013-09-20 01:15:31 +0000715 const DeclContext *LexicalDC = ND->getLexicalDeclContext();
716 if (LexicalDC->isFunctionOrMethod()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000717 // _cmd is relatively rare
718 if (const ImplicitParamDecl *ImplicitParam =
719 dyn_cast<ImplicitParamDecl>(ND))
720 if (ImplicitParam->getIdentifier() &&
721 ImplicitParam->getIdentifier()->isStr("_cmd"))
722 return CCP_ObjC_cmd;
723
724 return CCP_LocalDeclaration;
725 }
Richard Smith541b38b2013-09-20 01:15:31 +0000726
727 const DeclContext *DC = ND->getDeclContext()->getRedeclContext();
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000728 if (DC->isRecord() || isa<ObjCContainerDecl>(DC))
729 return CCP_MemberDeclaration;
730
731 // Content-based decisions.
732 if (isa<EnumConstantDecl>(ND))
733 return CCP_Constant;
734
Douglas Gregor52e0de42013-01-31 05:03:46 +0000735 // Use CCP_Type for type declarations unless we're in a statement, Objective-C
736 // message receiver, or parenthesized expression context. There, it's as
737 // likely that the user will want to write a type as other declarations.
738 if ((isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND)) &&
739 !(CompletionContext.getKind() == CodeCompletionContext::CCC_Statement ||
740 CompletionContext.getKind()
741 == CodeCompletionContext::CCC_ObjCMessageReceiver ||
742 CompletionContext.getKind()
743 == CodeCompletionContext::CCC_ParenthesizedExpression))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000744 return CCP_Type;
745
746 return CCP_Declaration;
747}
748
Douglas Gregor50832e02010-09-20 22:39:41 +0000749void ResultBuilder::AdjustResultPriorityForDecl(Result &R) {
750 // If this is an Objective-C method declaration whose selector matches our
751 // preferred selector, give it a priority boost.
752 if (!PreferredSelector.isNull())
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000753 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(R.Declaration))
Douglas Gregor50832e02010-09-20 22:39:41 +0000754 if (PreferredSelector == Method->getSelector())
755 R.Priority += CCD_SelectorMatch;
Douglas Gregor5fb901d2010-09-20 23:11:55 +0000756
Douglas Gregor50832e02010-09-20 22:39:41 +0000757 // If we have a preferred type, adjust the priority for results with exactly-
758 // matching or nearly-matching types.
759 if (!PreferredType.isNull()) {
760 QualType T = getDeclUsageType(SemaRef.Context, R.Declaration);
761 if (!T.isNull()) {
762 CanQualType TC = SemaRef.Context.getCanonicalType(T);
763 // Check for exactly-matching types (modulo qualifiers).
764 if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC))
765 R.Priority /= CCF_ExactTypeMatch;
766 // Check for nearly-matching types, based on classification of each.
767 else if ((getSimplifiedTypeClass(PreferredType)
Douglas Gregor95887f92010-07-08 23:20:03 +0000768 == getSimplifiedTypeClass(TC)) &&
Douglas Gregor50832e02010-09-20 22:39:41 +0000769 !(PreferredType->isEnumeralType() && TC->isEnumeralType()))
770 R.Priority /= CCF_SimilarTypeMatch;
771 }
772 }
Douglas Gregor95887f92010-07-08 23:20:03 +0000773}
774
Douglas Gregor0212fd72010-09-21 16:06:22 +0000775void ResultBuilder::MaybeAddConstructorResults(Result R) {
David Blaikiebbafb8a2012-03-11 07:00:24 +0000776 if (!SemaRef.getLangOpts().CPlusPlus || !R.Declaration ||
Douglas Gregor0212fd72010-09-21 16:06:22 +0000777 !CompletionContext.wantConstructorResults())
778 return;
779
780 ASTContext &Context = SemaRef.Context;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000781 const NamedDecl *D = R.Declaration;
Craig Topperc3ec1492014-05-26 06:22:03 +0000782 const CXXRecordDecl *Record = nullptr;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000783 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D))
Douglas Gregor0212fd72010-09-21 16:06:22 +0000784 Record = ClassTemplate->getTemplatedDecl();
785 else if ((Record = dyn_cast<CXXRecordDecl>(D))) {
786 // Skip specializations and partial specializations.
787 if (isa<ClassTemplateSpecializationDecl>(Record))
788 return;
789 } else {
790 // There are no constructors here.
791 return;
792 }
793
794 Record = Record->getDefinition();
795 if (!Record)
796 return;
797
798
799 QualType RecordTy = Context.getTypeDeclType(Record);
800 DeclarationName ConstructorName
801 = Context.DeclarationNames.getCXXConstructorName(
802 Context.getCanonicalType(RecordTy));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000803 DeclContext::lookup_const_result Ctors = Record->lookup(ConstructorName);
804 for (DeclContext::lookup_const_iterator I = Ctors.begin(),
805 E = Ctors.end();
806 I != E; ++I) {
David Blaikieff7d47a2012-12-19 00:45:41 +0000807 R.Declaration = *I;
Douglas Gregor0212fd72010-09-21 16:06:22 +0000808 R.CursorKind = getCursorKindForDecl(R.Declaration);
809 Results.push_back(R);
810 }
811}
812
Douglas Gregor7c208612010-01-14 00:20:49 +0000813void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) {
814 assert(!ShadowMaps.empty() && "Must enter into a results scope");
815
816 if (R.Kind != Result::RK_Declaration) {
817 // For non-declaration results, just add the result.
818 Results.push_back(R);
819 return;
820 }
821
822 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000823 if (const UsingShadowDecl *Using =
824 dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000825 MaybeAddResult(Result(Using->getTargetDecl(),
826 getBasePriority(Using->getTargetDecl()),
827 R.Qualifier),
828 CurContext);
Douglas Gregor7c208612010-01-14 00:20:49 +0000829 return;
830 }
831
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000832 const Decl *CanonDecl = R.Declaration->getCanonicalDecl();
Douglas Gregor7c208612010-01-14 00:20:49 +0000833 unsigned IDNS = CanonDecl->getIdentifierNamespace();
834
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000835 bool AsNestedNameSpecifier = false;
836 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregor7c208612010-01-14 00:20:49 +0000837 return;
838
Douglas Gregor0212fd72010-09-21 16:06:22 +0000839 // C++ constructors are never found by name lookup.
840 if (isa<CXXConstructorDecl>(R.Declaration))
841 return;
842
Douglas Gregor3545ff42009-09-21 16:56:56 +0000843 ShadowMap &SMap = ShadowMaps.back();
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000844 ShadowMapEntry::iterator I, IEnd;
845 ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName());
846 if (NamePos != SMap.end()) {
847 I = NamePos->second.begin();
848 IEnd = NamePos->second.end();
849 }
850
851 for (; I != IEnd; ++I) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000852 const NamedDecl *ND = I->first;
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000853 unsigned Index = I->second;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000854 if (ND->getCanonicalDecl() == CanonDecl) {
855 // This is a redeclaration. Always pick the newer declaration.
Douglas Gregor3545ff42009-09-21 16:56:56 +0000856 Results[Index].Declaration = R.Declaration;
857
Douglas Gregor3545ff42009-09-21 16:56:56 +0000858 // We're done.
859 return;
860 }
861 }
862
863 // This is a new declaration in this scope. However, check whether this
864 // declaration name is hidden by a similarly-named declaration in an outer
865 // scope.
866 std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end();
867 --SMEnd;
868 for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000869 ShadowMapEntry::iterator I, IEnd;
870 ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName());
871 if (NamePos != SM->end()) {
872 I = NamePos->second.begin();
873 IEnd = NamePos->second.end();
874 }
875 for (; I != IEnd; ++I) {
Douglas Gregor3545ff42009-09-21 16:56:56 +0000876 // A tag declaration does not hide a non-tag declaration.
John McCalle87beb22010-04-23 18:46:30 +0000877 if (I->first->hasTagIdentifierNamespace() &&
Richard Smith541b38b2013-09-20 01:15:31 +0000878 (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary |
879 Decl::IDNS_LocalExtern | Decl::IDNS_ObjCProtocol)))
Douglas Gregor3545ff42009-09-21 16:56:56 +0000880 continue;
881
882 // Protocols are in distinct namespaces from everything else.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000883 if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000884 || (IDNS & Decl::IDNS_ObjCProtocol)) &&
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000885 I->first->getIdentifierNamespace() != IDNS)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000886 continue;
887
888 // The newly-added result is hidden by an entry in the shadow map.
Douglas Gregore0717ab2010-01-14 00:41:07 +0000889 if (CheckHiddenResult(R, CurContext, I->first))
Douglas Gregor3545ff42009-09-21 16:56:56 +0000890 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000891
892 break;
893 }
894 }
895
896 // Make sure that any given declaration only shows up in the result set once.
897 if (!AllDeclsFound.insert(CanonDecl))
898 return;
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000899
Douglas Gregore412a5a2009-09-23 22:26:46 +0000900 // If the filter is for nested-name-specifiers, then this result starts a
901 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000902 if (AsNestedNameSpecifier) {
Douglas Gregore412a5a2009-09-23 22:26:46 +0000903 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000904 R.Priority = CCP_NestedNameSpecifier;
Douglas Gregor50832e02010-09-20 22:39:41 +0000905 } else
906 AdjustResultPriorityForDecl(R);
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000907
Douglas Gregor5bf52692009-09-22 23:15:58 +0000908 // If this result is supposed to have an informative qualifier, add one.
Douglas Gregore412a5a2009-09-23 22:26:46 +0000909 if (R.QualifierIsInformative && !R.Qualifier &&
910 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000911 const DeclContext *Ctx = R.Declaration->getDeclContext();
912 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Craig Topperc3ec1492014-05-26 06:22:03 +0000913 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, nullptr,
914 Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000915 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Craig Topperc3ec1492014-05-26 06:22:03 +0000916 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, nullptr,
917 false, SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregor5bf52692009-09-22 23:15:58 +0000918 else
919 R.QualifierIsInformative = false;
920 }
Douglas Gregore412a5a2009-09-23 22:26:46 +0000921
Douglas Gregor3545ff42009-09-21 16:56:56 +0000922 // Insert this result into the set of results and into the current shadow
923 // map.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000924 SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size());
Douglas Gregor3545ff42009-09-21 16:56:56 +0000925 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +0000926
927 if (!AsNestedNameSpecifier)
928 MaybeAddConstructorResults(R);
Douglas Gregor3545ff42009-09-21 16:56:56 +0000929}
930
Douglas Gregorc580c522010-01-14 01:09:38 +0000931void ResultBuilder::AddResult(Result R, DeclContext *CurContext,
Douglas Gregor09bbc652010-01-14 15:47:35 +0000932 NamedDecl *Hiding, bool InBaseClass = false) {
Douglas Gregor78a21012010-01-14 16:01:26 +0000933 if (R.Kind != Result::RK_Declaration) {
934 // For non-declaration results, just add the result.
935 Results.push_back(R);
936 return;
937 }
938
Douglas Gregorc580c522010-01-14 01:09:38 +0000939 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000940 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000941 AddResult(Result(Using->getTargetDecl(),
942 getBasePriority(Using->getTargetDecl()),
943 R.Qualifier),
944 CurContext, Hiding);
Douglas Gregorc580c522010-01-14 01:09:38 +0000945 return;
946 }
947
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000948 bool AsNestedNameSpecifier = false;
949 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregorc580c522010-01-14 01:09:38 +0000950 return;
951
Douglas Gregor0212fd72010-09-21 16:06:22 +0000952 // C++ constructors are never found by name lookup.
953 if (isa<CXXConstructorDecl>(R.Declaration))
954 return;
955
Douglas Gregorc580c522010-01-14 01:09:38 +0000956 if (Hiding && CheckHiddenResult(R, CurContext, Hiding))
957 return;
Nick Lewyckyc3921482012-04-03 21:44:08 +0000958
Douglas Gregorc580c522010-01-14 01:09:38 +0000959 // Make sure that any given declaration only shows up in the result set once.
960 if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl()))
961 return;
962
963 // If the filter is for nested-name-specifiers, then this result starts a
964 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000965 if (AsNestedNameSpecifier) {
Douglas Gregorc580c522010-01-14 01:09:38 +0000966 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000967 R.Priority = CCP_NestedNameSpecifier;
968 }
Douglas Gregor09bbc652010-01-14 15:47:35 +0000969 else if (Filter == &ResultBuilder::IsMember && !R.Qualifier && InBaseClass &&
970 isa<CXXRecordDecl>(R.Declaration->getDeclContext()
Sebastian Redl50c68252010-08-31 00:36:30 +0000971 ->getRedeclContext()))
Douglas Gregor09bbc652010-01-14 15:47:35 +0000972 R.QualifierIsInformative = true;
973
Douglas Gregorc580c522010-01-14 01:09:38 +0000974 // If this result is supposed to have an informative qualifier, add one.
975 if (R.QualifierIsInformative && !R.Qualifier &&
976 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000977 const DeclContext *Ctx = R.Declaration->getDeclContext();
978 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Craig Topperc3ec1492014-05-26 06:22:03 +0000979 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, nullptr,
980 Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000981 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Craig Topperc3ec1492014-05-26 06:22:03 +0000982 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, nullptr, false,
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000983 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregorc580c522010-01-14 01:09:38 +0000984 else
985 R.QualifierIsInformative = false;
986 }
987
Douglas Gregora2db7932010-05-26 22:00:08 +0000988 // Adjust the priority if this result comes from a base class.
989 if (InBaseClass)
990 R.Priority += CCD_InBaseClass;
991
Douglas Gregor50832e02010-09-20 22:39:41 +0000992 AdjustResultPriorityForDecl(R);
Douglas Gregor7aa6b222010-05-30 01:49:25 +0000993
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000994 if (HasObjectTypeQualifiers)
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000995 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(R.Declaration))
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000996 if (Method->isInstance()) {
997 Qualifiers MethodQuals
998 = Qualifiers::fromCVRMask(Method->getTypeQualifiers());
999 if (ObjectTypeQualifiers == MethodQuals)
1000 R.Priority += CCD_ObjectQualifierMatch;
1001 else if (ObjectTypeQualifiers - MethodQuals) {
1002 // The method cannot be invoked, because doing so would drop
1003 // qualifiers.
1004 return;
1005 }
1006 }
1007
Douglas Gregorc580c522010-01-14 01:09:38 +00001008 // Insert this result into the set of results.
1009 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +00001010
1011 if (!AsNestedNameSpecifier)
1012 MaybeAddConstructorResults(R);
Douglas Gregorc580c522010-01-14 01:09:38 +00001013}
1014
Douglas Gregor78a21012010-01-14 16:01:26 +00001015void ResultBuilder::AddResult(Result R) {
1016 assert(R.Kind != Result::RK_Declaration &&
1017 "Declaration results need more context");
1018 Results.push_back(R);
1019}
1020
Douglas Gregor3545ff42009-09-21 16:56:56 +00001021/// \brief Enter into a new scope.
1022void ResultBuilder::EnterNewScope() {
1023 ShadowMaps.push_back(ShadowMap());
1024}
1025
1026/// \brief Exit from the current scope.
1027void ResultBuilder::ExitScope() {
Douglas Gregor05e7ca32009-12-06 20:23:50 +00001028 for (ShadowMap::iterator E = ShadowMaps.back().begin(),
1029 EEnd = ShadowMaps.back().end();
1030 E != EEnd;
1031 ++E)
1032 E->second.Destroy();
1033
Douglas Gregor3545ff42009-09-21 16:56:56 +00001034 ShadowMaps.pop_back();
1035}
1036
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001037/// \brief Determines whether this given declaration will be found by
1038/// ordinary name lookup.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001039bool ResultBuilder::IsOrdinaryName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001040 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1041
Richard Smith541b38b2013-09-20 01:15:31 +00001042 // If name lookup finds a local extern declaration, then we are in a
1043 // context where it behaves like an ordinary name.
1044 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001045 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9858ed52010-06-15 20:26:51 +00001046 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001047 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor05fcf842010-11-02 20:36:02 +00001048 if (isa<ObjCIvarDecl>(ND))
1049 return true;
Douglas Gregor05fcf842010-11-02 20:36:02 +00001050 }
1051
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001052 return ND->getIdentifierNamespace() & IDNS;
1053}
1054
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001055/// \brief Determines whether this given declaration will be found by
Douglas Gregor70febae2010-05-28 00:49:12 +00001056/// ordinary name lookup but is not a type name.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001057bool ResultBuilder::IsOrdinaryNonTypeName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001058 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1059 if (isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND))
1060 return false;
1061
Richard Smith541b38b2013-09-20 01:15:31 +00001062 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001063 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9858ed52010-06-15 20:26:51 +00001064 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001065 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor05fcf842010-11-02 20:36:02 +00001066 if (isa<ObjCIvarDecl>(ND))
1067 return true;
Douglas Gregor05fcf842010-11-02 20:36:02 +00001068 }
1069
Douglas Gregor70febae2010-05-28 00:49:12 +00001070 return ND->getIdentifierNamespace() & IDNS;
1071}
1072
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001073bool ResultBuilder::IsIntegralConstantValue(const NamedDecl *ND) const {
Douglas Gregor85b50632010-07-28 21:50:18 +00001074 if (!IsOrdinaryNonTypeName(ND))
1075 return 0;
1076
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001077 if (const ValueDecl *VD = dyn_cast<ValueDecl>(ND->getUnderlyingDecl()))
Douglas Gregor85b50632010-07-28 21:50:18 +00001078 if (VD->getType()->isIntegralOrEnumerationType())
1079 return true;
1080
1081 return false;
1082}
1083
Douglas Gregor70febae2010-05-28 00:49:12 +00001084/// \brief Determines whether this given declaration will be found by
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001085/// ordinary name lookup.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001086bool ResultBuilder::IsOrdinaryNonValueName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001087 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1088
Richard Smith541b38b2013-09-20 01:15:31 +00001089 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001090 if (SemaRef.getLangOpts().CPlusPlus)
John McCalle87beb22010-04-23 18:46:30 +00001091 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001092
1093 return (ND->getIdentifierNamespace() & IDNS) &&
Douglas Gregor70febae2010-05-28 00:49:12 +00001094 !isa<ValueDecl>(ND) && !isa<FunctionTemplateDecl>(ND) &&
1095 !isa<ObjCPropertyDecl>(ND);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001096}
1097
Douglas Gregor3545ff42009-09-21 16:56:56 +00001098/// \brief Determines whether the given declaration is suitable as the
1099/// start of a C++ nested-name-specifier, e.g., a class or namespace.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001100bool ResultBuilder::IsNestedNameSpecifier(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001101 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001102 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001103 ND = ClassTemplate->getTemplatedDecl();
1104
1105 return SemaRef.isAcceptableNestedNameSpecifier(ND);
1106}
1107
1108/// \brief Determines whether the given declaration is an enumeration.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001109bool ResultBuilder::IsEnum(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001110 return isa<EnumDecl>(ND);
1111}
1112
1113/// \brief Determines whether the given declaration is a class or struct.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001114bool ResultBuilder::IsClassOrStruct(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001115 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001116 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001117 ND = ClassTemplate->getTemplatedDecl();
Joao Matosdc86f942012-08-31 18:45:21 +00001118
1119 // For purposes of this check, interfaces match too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001120 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara6150c882010-05-11 21:36:43 +00001121 return RD->getTagKind() == TTK_Class ||
Joao Matosdc86f942012-08-31 18:45:21 +00001122 RD->getTagKind() == TTK_Struct ||
1123 RD->getTagKind() == TTK_Interface;
Douglas Gregor3545ff42009-09-21 16:56:56 +00001124
1125 return false;
1126}
1127
1128/// \brief Determines whether the given declaration is a union.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001129bool ResultBuilder::IsUnion(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001130 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001131 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001132 ND = ClassTemplate->getTemplatedDecl();
1133
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001134 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara6150c882010-05-11 21:36:43 +00001135 return RD->getTagKind() == TTK_Union;
Douglas Gregor3545ff42009-09-21 16:56:56 +00001136
1137 return false;
1138}
1139
1140/// \brief Determines whether the given declaration is a namespace.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001141bool ResultBuilder::IsNamespace(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001142 return isa<NamespaceDecl>(ND);
1143}
1144
1145/// \brief Determines whether the given declaration is a namespace or
1146/// namespace alias.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001147bool ResultBuilder::IsNamespaceOrAlias(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001148 return isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND);
1149}
1150
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001151/// \brief Determines whether the given declaration is a type.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001152bool ResultBuilder::IsType(const NamedDecl *ND) const {
1153 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor99fa2642010-08-24 01:06:58 +00001154 ND = Using->getTargetDecl();
1155
1156 return isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND);
Douglas Gregor3545ff42009-09-21 16:56:56 +00001157}
1158
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001159/// \brief Determines which members of a class should be visible via
1160/// "." or "->". Only value declarations, nested name specifiers, and
1161/// using declarations thereof should show up.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001162bool ResultBuilder::IsMember(const NamedDecl *ND) const {
1163 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001164 ND = Using->getTargetDecl();
1165
Douglas Gregor70788392009-12-11 18:14:22 +00001166 return isa<ValueDecl>(ND) || isa<FunctionTemplateDecl>(ND) ||
1167 isa<ObjCPropertyDecl>(ND);
Douglas Gregore412a5a2009-09-23 22:26:46 +00001168}
1169
Douglas Gregora817a192010-05-27 23:06:34 +00001170static bool isObjCReceiverType(ASTContext &C, QualType T) {
1171 T = C.getCanonicalType(T);
1172 switch (T->getTypeClass()) {
1173 case Type::ObjCObject:
1174 case Type::ObjCInterface:
1175 case Type::ObjCObjectPointer:
1176 return true;
1177
1178 case Type::Builtin:
1179 switch (cast<BuiltinType>(T)->getKind()) {
1180 case BuiltinType::ObjCId:
1181 case BuiltinType::ObjCClass:
1182 case BuiltinType::ObjCSel:
1183 return true;
1184
1185 default:
1186 break;
1187 }
1188 return false;
1189
1190 default:
1191 break;
1192 }
1193
David Blaikiebbafb8a2012-03-11 07:00:24 +00001194 if (!C.getLangOpts().CPlusPlus)
Douglas Gregora817a192010-05-27 23:06:34 +00001195 return false;
1196
1197 // FIXME: We could perform more analysis here to determine whether a
1198 // particular class type has any conversions to Objective-C types. For now,
1199 // just accept all class types.
1200 return T->isDependentType() || T->isRecordType();
1201}
1202
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001203bool ResultBuilder::IsObjCMessageReceiver(const NamedDecl *ND) const {
Douglas Gregora817a192010-05-27 23:06:34 +00001204 QualType T = getDeclUsageType(SemaRef.Context, ND);
1205 if (T.isNull())
1206 return false;
1207
1208 T = SemaRef.Context.getBaseElementType(T);
1209 return isObjCReceiverType(SemaRef.Context, T);
1210}
1211
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001212bool ResultBuilder::IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const {
Douglas Gregord8c61782012-02-15 15:34:24 +00001213 if (IsObjCMessageReceiver(ND))
1214 return true;
1215
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001216 const VarDecl *Var = dyn_cast<VarDecl>(ND);
Douglas Gregord8c61782012-02-15 15:34:24 +00001217 if (!Var)
1218 return false;
1219
1220 return Var->hasLocalStorage() && !Var->hasAttr<BlocksAttr>();
1221}
1222
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001223bool ResultBuilder::IsObjCCollection(const NamedDecl *ND) const {
David Blaikiebbafb8a2012-03-11 07:00:24 +00001224 if ((SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryName(ND)) ||
1225 (!SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryNonTypeName(ND)))
Douglas Gregor68762e72010-08-23 21:17:50 +00001226 return false;
1227
1228 QualType T = getDeclUsageType(SemaRef.Context, ND);
1229 if (T.isNull())
1230 return false;
1231
1232 T = SemaRef.Context.getBaseElementType(T);
1233 return T->isObjCObjectType() || T->isObjCObjectPointerType() ||
1234 T->isObjCIdType() ||
David Blaikiebbafb8a2012-03-11 07:00:24 +00001235 (SemaRef.getLangOpts().CPlusPlus && T->isRecordType());
Douglas Gregor68762e72010-08-23 21:17:50 +00001236}
Douglas Gregora817a192010-05-27 23:06:34 +00001237
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001238bool ResultBuilder::IsImpossibleToSatisfy(const NamedDecl *ND) const {
Douglas Gregor0ac41382010-09-23 23:01:17 +00001239 return false;
1240}
1241
James Dennettf1243872012-06-17 05:33:25 +00001242/// \brief Determines whether the given declaration is an Objective-C
Douglas Gregor2b8162b2010-01-14 16:08:12 +00001243/// instance variable.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001244bool ResultBuilder::IsObjCIvar(const NamedDecl *ND) const {
Douglas Gregor2b8162b2010-01-14 16:08:12 +00001245 return isa<ObjCIvarDecl>(ND);
1246}
1247
Douglas Gregorc580c522010-01-14 01:09:38 +00001248namespace {
1249 /// \brief Visible declaration consumer that adds a code-completion result
1250 /// for each visible declaration.
1251 class CodeCompletionDeclConsumer : public VisibleDeclConsumer {
1252 ResultBuilder &Results;
1253 DeclContext *CurContext;
1254
1255 public:
1256 CodeCompletionDeclConsumer(ResultBuilder &Results, DeclContext *CurContext)
1257 : Results(Results), CurContext(CurContext) { }
Craig Toppere14c0f82014-03-12 04:55:44 +00001258
1259 void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
1260 bool InBaseClass) override {
Erik Verbruggen2e657ff2011-10-06 07:27:49 +00001261 bool Accessible = true;
Douglas Gregor03ba1882011-11-03 16:51:37 +00001262 if (Ctx)
1263 Accessible = Results.getSema().IsSimplyAccessible(ND, Ctx);
Craig Topperc3ec1492014-05-26 06:22:03 +00001264
1265 ResultBuilder::Result Result(ND, Results.getBasePriority(ND), nullptr,
1266 false, Accessible);
Erik Verbruggen2e657ff2011-10-06 07:27:49 +00001267 Results.AddResult(Result, CurContext, Hiding, InBaseClass);
Douglas Gregorc580c522010-01-14 01:09:38 +00001268 }
1269 };
1270}
1271
Douglas Gregor3545ff42009-09-21 16:56:56 +00001272/// \brief Add type specifiers for the current language as keyword results.
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001273static void AddTypeSpecifierResults(const LangOptions &LangOpts,
Douglas Gregor3545ff42009-09-21 16:56:56 +00001274 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001275 typedef CodeCompletionResult Result;
Douglas Gregora2db7932010-05-26 22:00:08 +00001276 Results.AddResult(Result("short", CCP_Type));
1277 Results.AddResult(Result("long", CCP_Type));
1278 Results.AddResult(Result("signed", CCP_Type));
1279 Results.AddResult(Result("unsigned", CCP_Type));
1280 Results.AddResult(Result("void", CCP_Type));
1281 Results.AddResult(Result("char", CCP_Type));
1282 Results.AddResult(Result("int", CCP_Type));
1283 Results.AddResult(Result("float", CCP_Type));
1284 Results.AddResult(Result("double", CCP_Type));
1285 Results.AddResult(Result("enum", CCP_Type));
1286 Results.AddResult(Result("struct", CCP_Type));
1287 Results.AddResult(Result("union", CCP_Type));
1288 Results.AddResult(Result("const", CCP_Type));
1289 Results.AddResult(Result("volatile", CCP_Type));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001290
Douglas Gregor3545ff42009-09-21 16:56:56 +00001291 if (LangOpts.C99) {
1292 // C99-specific
Douglas Gregora2db7932010-05-26 22:00:08 +00001293 Results.AddResult(Result("_Complex", CCP_Type));
1294 Results.AddResult(Result("_Imaginary", CCP_Type));
1295 Results.AddResult(Result("_Bool", CCP_Type));
1296 Results.AddResult(Result("restrict", CCP_Type));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001297 }
1298
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001299 CodeCompletionBuilder Builder(Results.getAllocator(),
1300 Results.getCodeCompletionTUInfo());
Douglas Gregor3545ff42009-09-21 16:56:56 +00001301 if (LangOpts.CPlusPlus) {
1302 // C++-specific
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00001303 Results.AddResult(Result("bool", CCP_Type +
1304 (LangOpts.ObjC1? CCD_bool_in_ObjC : 0)));
Douglas Gregora2db7932010-05-26 22:00:08 +00001305 Results.AddResult(Result("class", CCP_Type));
1306 Results.AddResult(Result("wchar_t", CCP_Type));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001307
Douglas Gregorf4c33342010-05-28 00:22:41 +00001308 // typename qualified-id
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001309 Builder.AddTypedTextChunk("typename");
1310 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1311 Builder.AddPlaceholderChunk("qualifier");
1312 Builder.AddTextChunk("::");
1313 Builder.AddPlaceholderChunk("name");
1314 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001315
Richard Smith2bf7fdb2013-01-02 11:42:31 +00001316 if (LangOpts.CPlusPlus11) {
Douglas Gregora2db7932010-05-26 22:00:08 +00001317 Results.AddResult(Result("auto", CCP_Type));
1318 Results.AddResult(Result("char16_t", CCP_Type));
1319 Results.AddResult(Result("char32_t", CCP_Type));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001320
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001321 Builder.AddTypedTextChunk("decltype");
1322 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1323 Builder.AddPlaceholderChunk("expression");
1324 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1325 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001326 }
1327 }
1328
1329 // GNU extensions
1330 if (LangOpts.GNUMode) {
1331 // FIXME: Enable when we actually support decimal floating point.
Douglas Gregor78a21012010-01-14 16:01:26 +00001332 // Results.AddResult(Result("_Decimal32"));
1333 // Results.AddResult(Result("_Decimal64"));
1334 // Results.AddResult(Result("_Decimal128"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001335
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001336 Builder.AddTypedTextChunk("typeof");
1337 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1338 Builder.AddPlaceholderChunk("expression");
1339 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001340
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001341 Builder.AddTypedTextChunk("typeof");
1342 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1343 Builder.AddPlaceholderChunk("type");
1344 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1345 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001346 }
1347}
1348
John McCallfaf5fb42010-08-26 23:41:50 +00001349static void AddStorageSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001350 const LangOptions &LangOpts,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001351 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001352 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001353 // Note: we don't suggest either "auto" or "register", because both
1354 // are pointless as storage specifiers. Elsewhere, we suggest "auto"
1355 // in C++0x as a type specifier.
Douglas Gregor78a21012010-01-14 16:01:26 +00001356 Results.AddResult(Result("extern"));
1357 Results.AddResult(Result("static"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001358}
1359
John McCallfaf5fb42010-08-26 23:41:50 +00001360static void AddFunctionSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001361 const LangOptions &LangOpts,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001362 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001363 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001364 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001365 case Sema::PCC_Class:
1366 case Sema::PCC_MemberTemplate:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001367 if (LangOpts.CPlusPlus) {
Douglas Gregor78a21012010-01-14 16:01:26 +00001368 Results.AddResult(Result("explicit"));
1369 Results.AddResult(Result("friend"));
1370 Results.AddResult(Result("mutable"));
1371 Results.AddResult(Result("virtual"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001372 }
1373 // Fall through
1374
John McCallfaf5fb42010-08-26 23:41:50 +00001375 case Sema::PCC_ObjCInterface:
1376 case Sema::PCC_ObjCImplementation:
1377 case Sema::PCC_Namespace:
1378 case Sema::PCC_Template:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001379 if (LangOpts.CPlusPlus || LangOpts.C99)
Douglas Gregor78a21012010-01-14 16:01:26 +00001380 Results.AddResult(Result("inline"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001381 break;
1382
John McCallfaf5fb42010-08-26 23:41:50 +00001383 case Sema::PCC_ObjCInstanceVariableList:
1384 case Sema::PCC_Expression:
1385 case Sema::PCC_Statement:
1386 case Sema::PCC_ForInit:
1387 case Sema::PCC_Condition:
1388 case Sema::PCC_RecoveryInFunction:
1389 case Sema::PCC_Type:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001390 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor80039242011-02-15 20:33:25 +00001391 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001392 break;
1393 }
1394}
1395
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001396static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt);
1397static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt);
1398static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00001399 ResultBuilder &Results,
1400 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001401static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00001402 ResultBuilder &Results,
1403 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001404static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00001405 ResultBuilder &Results,
1406 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001407static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt);
Douglas Gregorf1934162010-01-13 21:24:21 +00001408
Douglas Gregorf4c33342010-05-28 00:22:41 +00001409static void AddTypedefResult(ResultBuilder &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001410 CodeCompletionBuilder Builder(Results.getAllocator(),
1411 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001412 Builder.AddTypedTextChunk("typedef");
1413 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1414 Builder.AddPlaceholderChunk("type");
1415 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1416 Builder.AddPlaceholderChunk("name");
1417 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001418}
1419
John McCallfaf5fb42010-08-26 23:41:50 +00001420static bool WantTypesInContext(Sema::ParserCompletionContext CCC,
Douglas Gregor70febae2010-05-28 00:49:12 +00001421 const LangOptions &LangOpts) {
Douglas Gregor70febae2010-05-28 00:49:12 +00001422 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001423 case Sema::PCC_Namespace:
1424 case Sema::PCC_Class:
1425 case Sema::PCC_ObjCInstanceVariableList:
1426 case Sema::PCC_Template:
1427 case Sema::PCC_MemberTemplate:
1428 case Sema::PCC_Statement:
1429 case Sema::PCC_RecoveryInFunction:
1430 case Sema::PCC_Type:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001431 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor80039242011-02-15 20:33:25 +00001432 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor70febae2010-05-28 00:49:12 +00001433 return true;
1434
John McCallfaf5fb42010-08-26 23:41:50 +00001435 case Sema::PCC_Expression:
1436 case Sema::PCC_Condition:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001437 return LangOpts.CPlusPlus;
1438
1439 case Sema::PCC_ObjCInterface:
1440 case Sema::PCC_ObjCImplementation:
Douglas Gregor70febae2010-05-28 00:49:12 +00001441 return false;
1442
John McCallfaf5fb42010-08-26 23:41:50 +00001443 case Sema::PCC_ForInit:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001444 return LangOpts.CPlusPlus || LangOpts.ObjC1 || LangOpts.C99;
Douglas Gregor70febae2010-05-28 00:49:12 +00001445 }
David Blaikie8a40f702012-01-17 06:56:22 +00001446
1447 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor70febae2010-05-28 00:49:12 +00001448}
1449
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00001450static PrintingPolicy getCompletionPrintingPolicy(const ASTContext &Context,
1451 const Preprocessor &PP) {
1452 PrintingPolicy Policy = Sema::getPrintingPolicy(Context, PP);
Douglas Gregore5c79d52011-10-18 21:20:17 +00001453 Policy.AnonymousTagLocations = false;
1454 Policy.SuppressStrongLifetime = true;
Douglas Gregor2e10cf92011-11-03 00:16:13 +00001455 Policy.SuppressUnwrittenScope = true;
Douglas Gregore5c79d52011-10-18 21:20:17 +00001456 return Policy;
1457}
1458
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00001459/// \brief Retrieve a printing policy suitable for code completion.
1460static PrintingPolicy getCompletionPrintingPolicy(Sema &S) {
1461 return getCompletionPrintingPolicy(S.Context, S.PP);
1462}
1463
Douglas Gregore5c79d52011-10-18 21:20:17 +00001464/// \brief Retrieve the string representation of the given type as a string
1465/// that has the appropriate lifetime for code completion.
1466///
1467/// This routine provides a fast path where we provide constant strings for
1468/// common type names.
1469static const char *GetCompletionTypeString(QualType T,
1470 ASTContext &Context,
1471 const PrintingPolicy &Policy,
1472 CodeCompletionAllocator &Allocator) {
1473 if (!T.getLocalQualifiers()) {
1474 // Built-in type names are constant strings.
1475 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T))
Argyrios Kyrtzidisbbff3da2012-05-05 04:20:28 +00001476 return BT->getNameAsCString(Policy);
Douglas Gregore5c79d52011-10-18 21:20:17 +00001477
1478 // Anonymous tag types are constant strings.
1479 if (const TagType *TagT = dyn_cast<TagType>(T))
1480 if (TagDecl *Tag = TagT->getDecl())
John McCall5ea95772013-03-09 00:54:27 +00001481 if (!Tag->hasNameForLinkage()) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001482 switch (Tag->getTagKind()) {
1483 case TTK_Struct: return "struct <anonymous>";
Joao Matosdc86f942012-08-31 18:45:21 +00001484 case TTK_Interface: return "__interface <anonymous>";
1485 case TTK_Class: return "class <anonymous>";
Douglas Gregore5c79d52011-10-18 21:20:17 +00001486 case TTK_Union: return "union <anonymous>";
1487 case TTK_Enum: return "enum <anonymous>";
1488 }
1489 }
1490 }
1491
1492 // Slow path: format the type as a string.
1493 std::string Result;
1494 T.getAsStringInternal(Result, Policy);
1495 return Allocator.CopyString(Result);
1496}
1497
Douglas Gregord8c61782012-02-15 15:34:24 +00001498/// \brief Add a completion for "this", if we're in a member function.
1499static void addThisCompletion(Sema &S, ResultBuilder &Results) {
1500 QualType ThisTy = S.getCurrentThisType();
1501 if (ThisTy.isNull())
1502 return;
1503
1504 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001505 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregord8c61782012-02-15 15:34:24 +00001506 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
1507 Builder.AddResultTypeChunk(GetCompletionTypeString(ThisTy,
1508 S.Context,
1509 Policy,
1510 Allocator));
1511 Builder.AddTypedTextChunk("this");
Joao Matosdc86f942012-08-31 18:45:21 +00001512 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregord8c61782012-02-15 15:34:24 +00001513}
1514
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001515/// \brief Add language constructs that show up for "ordinary" names.
John McCallfaf5fb42010-08-26 23:41:50 +00001516static void AddOrdinaryNameResults(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001517 Scope *S,
1518 Sema &SemaRef,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001519 ResultBuilder &Results) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001520 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001521 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregore5c79d52011-10-18 21:20:17 +00001522 PrintingPolicy Policy = getCompletionPrintingPolicy(SemaRef);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001523
John McCall276321a2010-08-25 06:19:51 +00001524 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001525 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001526 case Sema::PCC_Namespace:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001527 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001528 if (Results.includeCodePatterns()) {
1529 // namespace <identifier> { declarations }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001530 Builder.AddTypedTextChunk("namespace");
1531 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1532 Builder.AddPlaceholderChunk("identifier");
1533 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1534 Builder.AddPlaceholderChunk("declarations");
1535 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1536 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1537 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001538 }
1539
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001540 // namespace identifier = identifier ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001541 Builder.AddTypedTextChunk("namespace");
1542 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1543 Builder.AddPlaceholderChunk("name");
1544 Builder.AddChunk(CodeCompletionString::CK_Equal);
1545 Builder.AddPlaceholderChunk("namespace");
1546 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001547
1548 // Using directives
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001549 Builder.AddTypedTextChunk("using");
1550 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1551 Builder.AddTextChunk("namespace");
1552 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1553 Builder.AddPlaceholderChunk("identifier");
1554 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001555
1556 // asm(string-literal)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001557 Builder.AddTypedTextChunk("asm");
1558 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1559 Builder.AddPlaceholderChunk("string-literal");
1560 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1561 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001562
Douglas Gregorf4c33342010-05-28 00:22:41 +00001563 if (Results.includeCodePatterns()) {
1564 // Explicit template instantiation
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001565 Builder.AddTypedTextChunk("template");
1566 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1567 Builder.AddPlaceholderChunk("declaration");
1568 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001569 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001570 }
Douglas Gregorf1934162010-01-13 21:24:21 +00001571
David Blaikiebbafb8a2012-03-11 07:00:24 +00001572 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001573 AddObjCTopLevelResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001574
Douglas Gregorf4c33342010-05-28 00:22:41 +00001575 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001576 // Fall through
1577
John McCallfaf5fb42010-08-26 23:41:50 +00001578 case Sema::PCC_Class:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001579 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001580 // Using declaration
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001581 Builder.AddTypedTextChunk("using");
1582 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1583 Builder.AddPlaceholderChunk("qualifier");
1584 Builder.AddTextChunk("::");
1585 Builder.AddPlaceholderChunk("name");
1586 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001587
Douglas Gregorf4c33342010-05-28 00:22:41 +00001588 // using typename qualifier::name (only in a dependent context)
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001589 if (SemaRef.CurContext->isDependentContext()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001590 Builder.AddTypedTextChunk("using");
1591 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1592 Builder.AddTextChunk("typename");
1593 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1594 Builder.AddPlaceholderChunk("qualifier");
1595 Builder.AddTextChunk("::");
1596 Builder.AddPlaceholderChunk("name");
1597 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001598 }
1599
John McCallfaf5fb42010-08-26 23:41:50 +00001600 if (CCC == Sema::PCC_Class) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001601 AddTypedefResult(Results);
1602
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001603 // public:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001604 Builder.AddTypedTextChunk("public");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001605 if (Results.includeCodePatterns())
1606 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001607 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001608
1609 // protected:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001610 Builder.AddTypedTextChunk("protected");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001611 if (Results.includeCodePatterns())
1612 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001613 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001614
1615 // private:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001616 Builder.AddTypedTextChunk("private");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001617 if (Results.includeCodePatterns())
1618 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001619 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001620 }
1621 }
1622 // Fall through
1623
John McCallfaf5fb42010-08-26 23:41:50 +00001624 case Sema::PCC_Template:
1625 case Sema::PCC_MemberTemplate:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001626 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001627 // template < parameters >
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001628 Builder.AddTypedTextChunk("template");
1629 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1630 Builder.AddPlaceholderChunk("parameters");
1631 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1632 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001633 }
1634
David Blaikiebbafb8a2012-03-11 07:00:24 +00001635 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1636 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001637 break;
1638
John McCallfaf5fb42010-08-26 23:41:50 +00001639 case Sema::PCC_ObjCInterface:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001640 AddObjCInterfaceResults(SemaRef.getLangOpts(), Results, true);
1641 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1642 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001643 break;
1644
John McCallfaf5fb42010-08-26 23:41:50 +00001645 case Sema::PCC_ObjCImplementation:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001646 AddObjCImplementationResults(SemaRef.getLangOpts(), Results, true);
1647 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1648 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001649 break;
1650
John McCallfaf5fb42010-08-26 23:41:50 +00001651 case Sema::PCC_ObjCInstanceVariableList:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001652 AddObjCVisibilityResults(SemaRef.getLangOpts(), Results, true);
Douglas Gregor48d46252010-01-13 21:54:15 +00001653 break;
1654
John McCallfaf5fb42010-08-26 23:41:50 +00001655 case Sema::PCC_RecoveryInFunction:
1656 case Sema::PCC_Statement: {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001657 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001658
David Blaikiebbafb8a2012-03-11 07:00:24 +00001659 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns() &&
1660 SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001661 Builder.AddTypedTextChunk("try");
1662 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1663 Builder.AddPlaceholderChunk("statements");
1664 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1665 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1666 Builder.AddTextChunk("catch");
1667 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1668 Builder.AddPlaceholderChunk("declaration");
1669 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1670 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1671 Builder.AddPlaceholderChunk("statements");
1672 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1673 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1674 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001675 }
David Blaikiebbafb8a2012-03-11 07:00:24 +00001676 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001677 AddObjCStatementResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001678
Douglas Gregorf64acca2010-05-25 21:41:55 +00001679 if (Results.includeCodePatterns()) {
1680 // if (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001681 Builder.AddTypedTextChunk("if");
1682 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001683 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001684 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001685 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001686 Builder.AddPlaceholderChunk("expression");
1687 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1688 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1689 Builder.AddPlaceholderChunk("statements");
1690 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1691 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1692 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001693
Douglas Gregorf64acca2010-05-25 21:41:55 +00001694 // switch (condition) { }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001695 Builder.AddTypedTextChunk("switch");
1696 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001697 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001698 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001699 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001700 Builder.AddPlaceholderChunk("expression");
1701 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1702 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1703 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1704 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1705 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001706 }
1707
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001708 // Switch-specific statements.
John McCallaab3e412010-08-25 08:40:02 +00001709 if (!SemaRef.getCurFunction()->SwitchStack.empty()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001710 // case expression:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001711 Builder.AddTypedTextChunk("case");
1712 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1713 Builder.AddPlaceholderChunk("expression");
1714 Builder.AddChunk(CodeCompletionString::CK_Colon);
1715 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001716
1717 // default:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001718 Builder.AddTypedTextChunk("default");
1719 Builder.AddChunk(CodeCompletionString::CK_Colon);
1720 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001721 }
1722
Douglas Gregorf64acca2010-05-25 21:41:55 +00001723 if (Results.includeCodePatterns()) {
1724 /// while (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001725 Builder.AddTypedTextChunk("while");
1726 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001727 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001728 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001729 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001730 Builder.AddPlaceholderChunk("expression");
1731 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1732 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1733 Builder.AddPlaceholderChunk("statements");
1734 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1735 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1736 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001737
1738 // do { statements } while ( expression );
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001739 Builder.AddTypedTextChunk("do");
1740 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1741 Builder.AddPlaceholderChunk("statements");
1742 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1743 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1744 Builder.AddTextChunk("while");
1745 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1746 Builder.AddPlaceholderChunk("expression");
1747 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1748 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001749
Douglas Gregorf64acca2010-05-25 21:41:55 +00001750 // for ( for-init-statement ; condition ; expression ) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001751 Builder.AddTypedTextChunk("for");
1752 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001753 if (SemaRef.getLangOpts().CPlusPlus || SemaRef.getLangOpts().C99)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001754 Builder.AddPlaceholderChunk("init-statement");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001755 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001756 Builder.AddPlaceholderChunk("init-expression");
1757 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1758 Builder.AddPlaceholderChunk("condition");
1759 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1760 Builder.AddPlaceholderChunk("inc-expression");
1761 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1762 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1763 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1764 Builder.AddPlaceholderChunk("statements");
1765 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1766 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1767 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001768 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001769
1770 if (S->getContinueParent()) {
1771 // continue ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001772 Builder.AddTypedTextChunk("continue");
1773 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001774 }
1775
1776 if (S->getBreakParent()) {
1777 // break ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001778 Builder.AddTypedTextChunk("break");
1779 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001780 }
1781
1782 // "return expression ;" or "return ;", depending on whether we
1783 // know the function is void or not.
1784 bool isVoid = false;
1785 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00001786 isVoid = Function->getReturnType()->isVoidType();
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001787 else if (ObjCMethodDecl *Method
1788 = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00001789 isVoid = Method->getReturnType()->isVoidType();
Douglas Gregor9a28e842010-03-01 23:15:13 +00001790 else if (SemaRef.getCurBlock() &&
1791 !SemaRef.getCurBlock()->ReturnType.isNull())
1792 isVoid = SemaRef.getCurBlock()->ReturnType->isVoidType();
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001793 Builder.AddTypedTextChunk("return");
Douglas Gregor44272ca2010-02-18 04:06:48 +00001794 if (!isVoid) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001795 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1796 Builder.AddPlaceholderChunk("expression");
Douglas Gregor44272ca2010-02-18 04:06:48 +00001797 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001798 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001799
Douglas Gregorf4c33342010-05-28 00:22:41 +00001800 // goto identifier ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001801 Builder.AddTypedTextChunk("goto");
1802 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1803 Builder.AddPlaceholderChunk("label");
1804 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001805
Douglas Gregorf4c33342010-05-28 00:22:41 +00001806 // Using directives
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001807 Builder.AddTypedTextChunk("using");
1808 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1809 Builder.AddTextChunk("namespace");
1810 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1811 Builder.AddPlaceholderChunk("identifier");
1812 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001813 }
1814
1815 // Fall through (for statement expressions).
John McCallfaf5fb42010-08-26 23:41:50 +00001816 case Sema::PCC_ForInit:
1817 case Sema::PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001818 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001819 // Fall through: conditions and statements can have expressions.
1820
Douglas Gregor5e35d592010-09-14 23:59:36 +00001821 case Sema::PCC_ParenthesizedExpression:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001822 if (SemaRef.getLangOpts().ObjCAutoRefCount &&
John McCall31168b02011-06-15 23:02:42 +00001823 CCC == Sema::PCC_ParenthesizedExpression) {
1824 // (__bridge <type>)<expression>
1825 Builder.AddTypedTextChunk("__bridge");
1826 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1827 Builder.AddPlaceholderChunk("type");
1828 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1829 Builder.AddPlaceholderChunk("expression");
1830 Results.AddResult(Result(Builder.TakeString()));
1831
1832 // (__bridge_transfer <Objective-C type>)<expression>
1833 Builder.AddTypedTextChunk("__bridge_transfer");
1834 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1835 Builder.AddPlaceholderChunk("Objective-C type");
1836 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1837 Builder.AddPlaceholderChunk("expression");
1838 Results.AddResult(Result(Builder.TakeString()));
1839
1840 // (__bridge_retained <CF type>)<expression>
1841 Builder.AddTypedTextChunk("__bridge_retained");
1842 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1843 Builder.AddPlaceholderChunk("CF type");
1844 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1845 Builder.AddPlaceholderChunk("expression");
1846 Results.AddResult(Result(Builder.TakeString()));
1847 }
1848 // Fall through
1849
John McCallfaf5fb42010-08-26 23:41:50 +00001850 case Sema::PCC_Expression: {
David Blaikiebbafb8a2012-03-11 07:00:24 +00001851 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001852 // 'this', if we're in a non-static member function.
Douglas Gregord8c61782012-02-15 15:34:24 +00001853 addThisCompletion(SemaRef, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001854
Douglas Gregore5c79d52011-10-18 21:20:17 +00001855 // true
1856 Builder.AddResultTypeChunk("bool");
1857 Builder.AddTypedTextChunk("true");
1858 Results.AddResult(Result(Builder.TakeString()));
1859
1860 // false
1861 Builder.AddResultTypeChunk("bool");
1862 Builder.AddTypedTextChunk("false");
1863 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001864
David Blaikiebbafb8a2012-03-11 07:00:24 +00001865 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001866 // dynamic_cast < type-id > ( expression )
1867 Builder.AddTypedTextChunk("dynamic_cast");
1868 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1869 Builder.AddPlaceholderChunk("type");
1870 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1871 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1872 Builder.AddPlaceholderChunk("expression");
1873 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1874 Results.AddResult(Result(Builder.TakeString()));
1875 }
Douglas Gregorf4c33342010-05-28 00:22:41 +00001876
1877 // static_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001878 Builder.AddTypedTextChunk("static_cast");
1879 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1880 Builder.AddPlaceholderChunk("type");
1881 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1882 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1883 Builder.AddPlaceholderChunk("expression");
1884 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1885 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001886
Douglas Gregorf4c33342010-05-28 00:22:41 +00001887 // reinterpret_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001888 Builder.AddTypedTextChunk("reinterpret_cast");
1889 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1890 Builder.AddPlaceholderChunk("type");
1891 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1892 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1893 Builder.AddPlaceholderChunk("expression");
1894 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1895 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001896
Douglas Gregorf4c33342010-05-28 00:22:41 +00001897 // const_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001898 Builder.AddTypedTextChunk("const_cast");
1899 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1900 Builder.AddPlaceholderChunk("type");
1901 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1902 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1903 Builder.AddPlaceholderChunk("expression");
1904 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1905 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001906
David Blaikiebbafb8a2012-03-11 07:00:24 +00001907 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001908 // typeid ( expression-or-type )
Douglas Gregore5c79d52011-10-18 21:20:17 +00001909 Builder.AddResultTypeChunk("std::type_info");
Douglas Gregorc05f6572011-04-12 02:47:21 +00001910 Builder.AddTypedTextChunk("typeid");
1911 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1912 Builder.AddPlaceholderChunk("expression-or-type");
1913 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1914 Results.AddResult(Result(Builder.TakeString()));
1915 }
1916
Douglas Gregorf4c33342010-05-28 00:22:41 +00001917 // new T ( ... )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001918 Builder.AddTypedTextChunk("new");
1919 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1920 Builder.AddPlaceholderChunk("type");
1921 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1922 Builder.AddPlaceholderChunk("expressions");
1923 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1924 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001925
Douglas Gregorf4c33342010-05-28 00:22:41 +00001926 // new T [ ] ( ... )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001927 Builder.AddTypedTextChunk("new");
1928 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1929 Builder.AddPlaceholderChunk("type");
1930 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1931 Builder.AddPlaceholderChunk("size");
1932 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1933 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1934 Builder.AddPlaceholderChunk("expressions");
1935 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1936 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001937
Douglas Gregorf4c33342010-05-28 00:22:41 +00001938 // delete expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001939 Builder.AddResultTypeChunk("void");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001940 Builder.AddTypedTextChunk("delete");
1941 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1942 Builder.AddPlaceholderChunk("expression");
1943 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001944
Douglas Gregorf4c33342010-05-28 00:22:41 +00001945 // delete [] expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001946 Builder.AddResultTypeChunk("void");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001947 Builder.AddTypedTextChunk("delete");
1948 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1949 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1950 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1951 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1952 Builder.AddPlaceholderChunk("expression");
1953 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001954
David Blaikiebbafb8a2012-03-11 07:00:24 +00001955 if (SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001956 // throw expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001957 Builder.AddResultTypeChunk("void");
Douglas Gregorc05f6572011-04-12 02:47:21 +00001958 Builder.AddTypedTextChunk("throw");
1959 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1960 Builder.AddPlaceholderChunk("expression");
1961 Results.AddResult(Result(Builder.TakeString()));
1962 }
Douglas Gregor4205fef2011-10-18 16:29:03 +00001963
Douglas Gregora2db7932010-05-26 22:00:08 +00001964 // FIXME: Rethrow?
Douglas Gregor4205fef2011-10-18 16:29:03 +00001965
Richard Smith2bf7fdb2013-01-02 11:42:31 +00001966 if (SemaRef.getLangOpts().CPlusPlus11) {
Douglas Gregor4205fef2011-10-18 16:29:03 +00001967 // nullptr
Douglas Gregore5c79d52011-10-18 21:20:17 +00001968 Builder.AddResultTypeChunk("std::nullptr_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001969 Builder.AddTypedTextChunk("nullptr");
1970 Results.AddResult(Result(Builder.TakeString()));
1971
1972 // alignof
Douglas Gregore5c79d52011-10-18 21:20:17 +00001973 Builder.AddResultTypeChunk("size_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001974 Builder.AddTypedTextChunk("alignof");
1975 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1976 Builder.AddPlaceholderChunk("type");
1977 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1978 Results.AddResult(Result(Builder.TakeString()));
1979
1980 // noexcept
Douglas Gregore5c79d52011-10-18 21:20:17 +00001981 Builder.AddResultTypeChunk("bool");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001982 Builder.AddTypedTextChunk("noexcept");
1983 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1984 Builder.AddPlaceholderChunk("expression");
1985 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1986 Results.AddResult(Result(Builder.TakeString()));
1987
1988 // sizeof... expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001989 Builder.AddResultTypeChunk("size_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001990 Builder.AddTypedTextChunk("sizeof...");
1991 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1992 Builder.AddPlaceholderChunk("parameter-pack");
1993 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1994 Results.AddResult(Result(Builder.TakeString()));
1995 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001996 }
1997
David Blaikiebbafb8a2012-03-11 07:00:24 +00001998 if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001999 // Add "super", if we're in an Objective-C class with a superclass.
Ted Kremenek305a0a72010-05-31 21:43:10 +00002000 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) {
2001 // The interface can be NULL.
2002 if (ObjCInterfaceDecl *ID = Method->getClassInterface())
Douglas Gregore5c79d52011-10-18 21:20:17 +00002003 if (ID->getSuperClass()) {
2004 std::string SuperType;
2005 SuperType = ID->getSuperClass()->getNameAsString();
2006 if (Method->isInstanceMethod())
2007 SuperType += " *";
2008
2009 Builder.AddResultTypeChunk(Allocator.CopyString(SuperType));
2010 Builder.AddTypedTextChunk("super");
2011 Results.AddResult(Result(Builder.TakeString()));
2012 }
Ted Kremenek305a0a72010-05-31 21:43:10 +00002013 }
2014
Douglas Gregorf98e6a22010-01-13 23:51:12 +00002015 AddObjCExpressionResults(Results, true);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002016 }
2017
Jordan Rose58d54722012-06-30 21:33:57 +00002018 if (SemaRef.getLangOpts().C11) {
2019 // _Alignof
2020 Builder.AddResultTypeChunk("size_t");
2021 if (SemaRef.getASTContext().Idents.get("alignof").hasMacroDefinition())
2022 Builder.AddTypedTextChunk("alignof");
2023 else
2024 Builder.AddTypedTextChunk("_Alignof");
2025 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2026 Builder.AddPlaceholderChunk("type");
2027 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2028 Results.AddResult(Result(Builder.TakeString()));
2029 }
2030
Douglas Gregorf4c33342010-05-28 00:22:41 +00002031 // sizeof expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00002032 Builder.AddResultTypeChunk("size_t");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002033 Builder.AddTypedTextChunk("sizeof");
2034 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2035 Builder.AddPlaceholderChunk("expression-or-type");
2036 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2037 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002038 break;
2039 }
Douglas Gregor99fa2642010-08-24 01:06:58 +00002040
John McCallfaf5fb42010-08-26 23:41:50 +00002041 case Sema::PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00002042 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor99fa2642010-08-24 01:06:58 +00002043 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002044 }
2045
David Blaikiebbafb8a2012-03-11 07:00:24 +00002046 if (WantTypesInContext(CCC, SemaRef.getLangOpts()))
2047 AddTypeSpecifierResults(SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002048
David Blaikiebbafb8a2012-03-11 07:00:24 +00002049 if (SemaRef.getLangOpts().CPlusPlus && CCC != Sema::PCC_Type)
Douglas Gregor78a21012010-01-14 16:01:26 +00002050 Results.AddResult(Result("operator"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002051}
2052
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002053/// \brief If the given declaration has an associated type, add it as a result
2054/// type chunk.
2055static void AddResultTypeChunk(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002056 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002057 const NamedDecl *ND,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002058 CodeCompletionBuilder &Result) {
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002059 if (!ND)
2060 return;
Douglas Gregor0212fd72010-09-21 16:06:22 +00002061
2062 // Skip constructors and conversion functions, which have their return types
2063 // built into their names.
2064 if (isa<CXXConstructorDecl>(ND) || isa<CXXConversionDecl>(ND))
2065 return;
2066
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002067 // Determine the type of the declaration (if it has a type).
Alp Tokera2794f92014-01-22 07:29:52 +00002068 QualType T;
2069 if (const FunctionDecl *Function = ND->getAsFunction())
Alp Toker314cc812014-01-25 16:55:45 +00002070 T = Function->getReturnType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002071 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Alp Toker314cc812014-01-25 16:55:45 +00002072 T = Method->getReturnType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002073 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002074 T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext()));
2075 else if (isa<UnresolvedUsingValueDecl>(ND)) {
2076 /* Do nothing: ignore unresolved using declarations*/
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002077 } else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND)) {
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002078 T = Value->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002079 } else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002080 T = Property->getType();
2081
2082 if (T.isNull() || Context.hasSameType(T, Context.DependentTy))
2083 return;
2084
Douglas Gregor75acd922011-09-27 23:30:47 +00002085 Result.AddResultTypeChunk(GetCompletionTypeString(T, Context, Policy,
Douglas Gregor304f9b02011-02-01 21:15:40 +00002086 Result.getAllocator()));
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002087}
2088
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002089static void MaybeAddSentinel(ASTContext &Context,
2090 const NamedDecl *FunctionOrMethod,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002091 CodeCompletionBuilder &Result) {
Douglas Gregordbb71db2010-08-23 23:51:41 +00002092 if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>())
2093 if (Sentinel->getSentinel() == 0) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00002094 if (Context.getLangOpts().ObjC1 &&
Douglas Gregordbb71db2010-08-23 23:51:41 +00002095 Context.Idents.get("nil").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002096 Result.AddTextChunk(", nil");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002097 else if (Context.Idents.get("NULL").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002098 Result.AddTextChunk(", NULL");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002099 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002100 Result.AddTextChunk(", (void*)0");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002101 }
2102}
2103
Douglas Gregor8f08d742011-07-30 07:55:26 +00002104static std::string formatObjCParamQualifiers(unsigned ObjCQuals) {
2105 std::string Result;
2106 if (ObjCQuals & Decl::OBJC_TQ_In)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002107 Result += "in ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002108 else if (ObjCQuals & Decl::OBJC_TQ_Inout)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002109 Result += "inout ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002110 else if (ObjCQuals & Decl::OBJC_TQ_Out)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002111 Result += "out ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002112 if (ObjCQuals & Decl::OBJC_TQ_Bycopy)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002113 Result += "bycopy ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002114 else if (ObjCQuals & Decl::OBJC_TQ_Byref)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002115 Result += "byref ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002116 if (ObjCQuals & Decl::OBJC_TQ_Oneway)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002117 Result += "oneway ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002118 return Result;
2119}
2120
Douglas Gregore90dd002010-08-24 16:15:59 +00002121static std::string FormatFunctionParameter(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002122 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002123 const ParmVarDecl *Param,
Douglas Gregord793e7c2011-10-18 04:23:19 +00002124 bool SuppressName = false,
2125 bool SuppressBlock = false) {
Douglas Gregore90dd002010-08-24 16:15:59 +00002126 bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext());
2127 if (Param->getType()->isDependentType() ||
2128 !Param->getType()->isBlockPointerType()) {
2129 // The argument for a dependent or non-block parameter is a placeholder
2130 // containing that parameter's type.
2131 std::string Result;
2132
Douglas Gregor981a0c42010-08-29 19:47:46 +00002133 if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002134 Result = Param->getIdentifier()->getName();
2135
John McCall31168b02011-06-15 23:02:42 +00002136 Param->getType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002137
2138 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002139 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2140 + Result + ")";
Douglas Gregor981a0c42010-08-29 19:47:46 +00002141 if (Param->getIdentifier() && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002142 Result += Param->getIdentifier()->getName();
2143 }
2144 return Result;
2145 }
2146
2147 // The argument for a block pointer parameter is a block literal with
2148 // the appropriate type.
David Blaikie6adc78e2013-02-18 22:06:02 +00002149 FunctionTypeLoc Block;
2150 FunctionProtoTypeLoc BlockProto;
Douglas Gregore90dd002010-08-24 16:15:59 +00002151 TypeLoc TL;
2152 if (TypeSourceInfo *TSInfo = Param->getTypeSourceInfo()) {
2153 TL = TSInfo->getTypeLoc().getUnqualifiedLoc();
2154 while (true) {
2155 // Look through typedefs.
Douglas Gregord793e7c2011-10-18 04:23:19 +00002156 if (!SuppressBlock) {
David Blaikie6adc78e2013-02-18 22:06:02 +00002157 if (TypedefTypeLoc TypedefTL = TL.getAs<TypedefTypeLoc>()) {
2158 if (TypeSourceInfo *InnerTSInfo =
2159 TypedefTL.getTypedefNameDecl()->getTypeSourceInfo()) {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002160 TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc();
2161 continue;
2162 }
2163 }
2164
2165 // Look through qualified types
David Blaikie6adc78e2013-02-18 22:06:02 +00002166 if (QualifiedTypeLoc QualifiedTL = TL.getAs<QualifiedTypeLoc>()) {
2167 TL = QualifiedTL.getUnqualifiedLoc();
Douglas Gregore90dd002010-08-24 16:15:59 +00002168 continue;
2169 }
2170 }
2171
Douglas Gregore90dd002010-08-24 16:15:59 +00002172 // Try to get the function prototype behind the block pointer type,
2173 // then we're done.
David Blaikie6adc78e2013-02-18 22:06:02 +00002174 if (BlockPointerTypeLoc BlockPtr = TL.getAs<BlockPointerTypeLoc>()) {
2175 TL = BlockPtr.getPointeeLoc().IgnoreParens();
2176 Block = TL.getAs<FunctionTypeLoc>();
2177 BlockProto = TL.getAs<FunctionProtoTypeLoc>();
Douglas Gregore90dd002010-08-24 16:15:59 +00002178 }
2179 break;
2180 }
2181 }
2182
2183 if (!Block) {
2184 // We were unable to find a FunctionProtoTypeLoc with parameter names
2185 // for the block; just use the parameter type as a placeholder.
2186 std::string Result;
Douglas Gregord793e7c2011-10-18 04:23:19 +00002187 if (!ObjCMethodParam && Param->getIdentifier())
2188 Result = Param->getIdentifier()->getName();
2189
John McCall31168b02011-06-15 23:02:42 +00002190 Param->getType().getUnqualifiedType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002191
2192 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002193 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2194 + Result + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002195 if (Param->getIdentifier())
2196 Result += Param->getIdentifier()->getName();
2197 }
2198
2199 return Result;
2200 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002201
Douglas Gregore90dd002010-08-24 16:15:59 +00002202 // We have the function prototype behind the block pointer type, as it was
2203 // written in the source.
Douglas Gregor67da50e2010-09-08 22:47:51 +00002204 std::string Result;
Alp Toker314cc812014-01-25 16:55:45 +00002205 QualType ResultType = Block.getTypePtr()->getReturnType();
Douglas Gregord793e7c2011-10-18 04:23:19 +00002206 if (!ResultType->isVoidType() || SuppressBlock)
John McCall31168b02011-06-15 23:02:42 +00002207 ResultType.getAsStringInternal(Result, Policy);
Douglas Gregord793e7c2011-10-18 04:23:19 +00002208
2209 // Format the parameter list.
2210 std::string Params;
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002211 if (!BlockProto || Block.getNumParams() == 0) {
David Blaikie6adc78e2013-02-18 22:06:02 +00002212 if (BlockProto && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002213 Params = "(...)";
Douglas Gregoraf25cfa2010-10-02 23:49:58 +00002214 else
Douglas Gregord793e7c2011-10-18 04:23:19 +00002215 Params = "(void)";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002216 } else {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002217 Params += "(";
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002218 for (unsigned I = 0, N = Block.getNumParams(); I != N; ++I) {
Douglas Gregor67da50e2010-09-08 22:47:51 +00002219 if (I)
Douglas Gregord793e7c2011-10-18 04:23:19 +00002220 Params += ", ";
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002221 Params += FormatFunctionParameter(Context, Policy, Block.getParam(I),
2222 /*SuppressName=*/false,
Douglas Gregord793e7c2011-10-18 04:23:19 +00002223 /*SuppressBlock=*/true);
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002224
David Blaikie6adc78e2013-02-18 22:06:02 +00002225 if (I == N - 1 && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002226 Params += ", ...";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002227 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002228 Params += ")";
Douglas Gregor400f5972010-08-31 05:13:43 +00002229 }
Douglas Gregor67da50e2010-09-08 22:47:51 +00002230
Douglas Gregord793e7c2011-10-18 04:23:19 +00002231 if (SuppressBlock) {
2232 // Format as a parameter.
2233 Result = Result + " (^";
2234 if (Param->getIdentifier())
2235 Result += Param->getIdentifier()->getName();
2236 Result += ")";
2237 Result += Params;
2238 } else {
2239 // Format as a block literal argument.
2240 Result = '^' + Result;
2241 Result += Params;
2242
2243 if (Param->getIdentifier())
2244 Result += Param->getIdentifier()->getName();
2245 }
2246
Douglas Gregore90dd002010-08-24 16:15:59 +00002247 return Result;
2248}
2249
Douglas Gregor3545ff42009-09-21 16:56:56 +00002250/// \brief Add function parameter chunks to the given code completion string.
2251static void AddFunctionParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002252 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002253 const FunctionDecl *Function,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002254 CodeCompletionBuilder &Result,
2255 unsigned Start = 0,
2256 bool InOptional = false) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002257 bool FirstParameter = true;
Douglas Gregor9eb77012009-11-07 00:00:49 +00002258
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002259 for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002260 const ParmVarDecl *Param = Function->getParamDecl(P);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002261
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002262 if (Param->hasDefaultArg() && !InOptional) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002263 // When we see an optional default argument, put that argument and
2264 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002265 CodeCompletionBuilder Opt(Result.getAllocator(),
2266 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002267 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002268 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002269 AddFunctionParameterChunks(Context, Policy, Function, Opt, P, true);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002270 Result.AddOptionalChunk(Opt.TakeString());
2271 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002272 }
2273
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002274 if (FirstParameter)
2275 FirstParameter = false;
2276 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002277 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002278
2279 InOptional = false;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002280
2281 // Format the placeholder string.
Douglas Gregor75acd922011-09-27 23:30:47 +00002282 std::string PlaceholderStr = FormatFunctionParameter(Context, Policy,
2283 Param);
Douglas Gregore90dd002010-08-24 16:15:59 +00002284
Douglas Gregor400f5972010-08-31 05:13:43 +00002285 if (Function->isVariadic() && P == N - 1)
2286 PlaceholderStr += ", ...";
2287
Douglas Gregor3545ff42009-09-21 16:56:56 +00002288 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002289 Result.AddPlaceholderChunk(
2290 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002291 }
Douglas Gregorba449032009-09-22 21:42:17 +00002292
2293 if (const FunctionProtoType *Proto
2294 = Function->getType()->getAs<FunctionProtoType>())
Douglas Gregordbb71db2010-08-23 23:51:41 +00002295 if (Proto->isVariadic()) {
Alp Toker9cacbab2014-01-20 20:26:09 +00002296 if (Proto->getNumParams() == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002297 Result.AddPlaceholderChunk("...");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002298
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002299 MaybeAddSentinel(Context, Function, Result);
Douglas Gregordbb71db2010-08-23 23:51:41 +00002300 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002301}
2302
2303/// \brief Add template parameter chunks to the given code completion string.
2304static void AddTemplateParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002305 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002306 const TemplateDecl *Template,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002307 CodeCompletionBuilder &Result,
2308 unsigned MaxParameters = 0,
2309 unsigned Start = 0,
2310 bool InDefaultArg = false) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002311 bool FirstParameter = true;
2312
2313 TemplateParameterList *Params = Template->getTemplateParameters();
2314 TemplateParameterList::iterator PEnd = Params->end();
2315 if (MaxParameters)
2316 PEnd = Params->begin() + MaxParameters;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002317 for (TemplateParameterList::iterator P = Params->begin() + Start;
2318 P != PEnd; ++P) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002319 bool HasDefaultArg = false;
2320 std::string PlaceholderStr;
2321 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) {
2322 if (TTP->wasDeclaredWithTypename())
2323 PlaceholderStr = "typename";
2324 else
2325 PlaceholderStr = "class";
2326
2327 if (TTP->getIdentifier()) {
2328 PlaceholderStr += ' ';
2329 PlaceholderStr += TTP->getIdentifier()->getName();
2330 }
2331
2332 HasDefaultArg = TTP->hasDefaultArgument();
2333 } else if (NonTypeTemplateParmDecl *NTTP
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002334 = dyn_cast<NonTypeTemplateParmDecl>(*P)) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002335 if (NTTP->getIdentifier())
2336 PlaceholderStr = NTTP->getIdentifier()->getName();
John McCall31168b02011-06-15 23:02:42 +00002337 NTTP->getType().getAsStringInternal(PlaceholderStr, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002338 HasDefaultArg = NTTP->hasDefaultArgument();
2339 } else {
2340 assert(isa<TemplateTemplateParmDecl>(*P));
2341 TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P);
2342
2343 // Since putting the template argument list into the placeholder would
2344 // be very, very long, we just use an abbreviation.
2345 PlaceholderStr = "template<...> class";
2346 if (TTP->getIdentifier()) {
2347 PlaceholderStr += ' ';
2348 PlaceholderStr += TTP->getIdentifier()->getName();
2349 }
2350
2351 HasDefaultArg = TTP->hasDefaultArgument();
2352 }
2353
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002354 if (HasDefaultArg && !InDefaultArg) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002355 // When we see an optional default argument, put that argument and
2356 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002357 CodeCompletionBuilder Opt(Result.getAllocator(),
2358 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002359 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002360 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002361 AddTemplateParameterChunks(Context, Policy, Template, Opt, MaxParameters,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002362 P - Params->begin(), true);
2363 Result.AddOptionalChunk(Opt.TakeString());
2364 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002365 }
2366
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002367 InDefaultArg = false;
2368
Douglas Gregor3545ff42009-09-21 16:56:56 +00002369 if (FirstParameter)
2370 FirstParameter = false;
2371 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002372 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002373
2374 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002375 Result.AddPlaceholderChunk(
2376 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002377 }
2378}
2379
Douglas Gregorf2510672009-09-21 19:57:38 +00002380/// \brief Add a qualifier to the given code-completion string, if the
2381/// provided nested-name-specifier is non-NULL.
Douglas Gregor0f622362009-12-11 18:44:16 +00002382static void
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002383AddQualifierToCompletionString(CodeCompletionBuilder &Result,
Douglas Gregor0f622362009-12-11 18:44:16 +00002384 NestedNameSpecifier *Qualifier,
2385 bool QualifierIsInformative,
Douglas Gregor75acd922011-09-27 23:30:47 +00002386 ASTContext &Context,
2387 const PrintingPolicy &Policy) {
Douglas Gregorf2510672009-09-21 19:57:38 +00002388 if (!Qualifier)
2389 return;
2390
2391 std::string PrintedNNS;
2392 {
2393 llvm::raw_string_ostream OS(PrintedNNS);
Douglas Gregor75acd922011-09-27 23:30:47 +00002394 Qualifier->print(OS, Policy);
Douglas Gregorf2510672009-09-21 19:57:38 +00002395 }
Douglas Gregor5bf52692009-09-22 23:15:58 +00002396 if (QualifierIsInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002397 Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregor5bf52692009-09-22 23:15:58 +00002398 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002399 Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregorf2510672009-09-21 19:57:38 +00002400}
2401
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002402static void
2403AddFunctionTypeQualsToCompletionString(CodeCompletionBuilder &Result,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002404 const FunctionDecl *Function) {
Douglas Gregor0f622362009-12-11 18:44:16 +00002405 const FunctionProtoType *Proto
2406 = Function->getType()->getAs<FunctionProtoType>();
2407 if (!Proto || !Proto->getTypeQuals())
2408 return;
2409
Douglas Gregor304f9b02011-02-01 21:15:40 +00002410 // FIXME: Add ref-qualifier!
2411
2412 // Handle single qualifiers without copying
2413 if (Proto->getTypeQuals() == Qualifiers::Const) {
2414 Result.AddInformativeChunk(" const");
2415 return;
2416 }
2417
2418 if (Proto->getTypeQuals() == Qualifiers::Volatile) {
2419 Result.AddInformativeChunk(" volatile");
2420 return;
2421 }
2422
2423 if (Proto->getTypeQuals() == Qualifiers::Restrict) {
2424 Result.AddInformativeChunk(" restrict");
2425 return;
2426 }
2427
2428 // Handle multiple qualifiers.
Douglas Gregor0f622362009-12-11 18:44:16 +00002429 std::string QualsStr;
David Blaikief5697e52012-08-10 00:55:35 +00002430 if (Proto->isConst())
Douglas Gregor0f622362009-12-11 18:44:16 +00002431 QualsStr += " const";
David Blaikief5697e52012-08-10 00:55:35 +00002432 if (Proto->isVolatile())
Douglas Gregor0f622362009-12-11 18:44:16 +00002433 QualsStr += " volatile";
David Blaikief5697e52012-08-10 00:55:35 +00002434 if (Proto->isRestrict())
Douglas Gregor0f622362009-12-11 18:44:16 +00002435 QualsStr += " restrict";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002436 Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr));
Douglas Gregor0f622362009-12-11 18:44:16 +00002437}
2438
Douglas Gregor0212fd72010-09-21 16:06:22 +00002439/// \brief Add the name of the given declaration
Douglas Gregor75acd922011-09-27 23:30:47 +00002440static void AddTypedNameChunk(ASTContext &Context, const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002441 const NamedDecl *ND,
2442 CodeCompletionBuilder &Result) {
Douglas Gregor0212fd72010-09-21 16:06:22 +00002443 DeclarationName Name = ND->getDeclName();
2444 if (!Name)
2445 return;
2446
2447 switch (Name.getNameKind()) {
Douglas Gregor304f9b02011-02-01 21:15:40 +00002448 case DeclarationName::CXXOperatorName: {
Craig Topperc3ec1492014-05-26 06:22:03 +00002449 const char *OperatorName = nullptr;
Douglas Gregor304f9b02011-02-01 21:15:40 +00002450 switch (Name.getCXXOverloadedOperator()) {
2451 case OO_None:
2452 case OO_Conditional:
2453 case NUM_OVERLOADED_OPERATORS:
2454 OperatorName = "operator";
2455 break;
2456
2457#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
2458 case OO_##Name: OperatorName = "operator" Spelling; break;
2459#define OVERLOADED_OPERATOR_MULTI(Name,Spelling,Unary,Binary,MemberOnly)
2460#include "clang/Basic/OperatorKinds.def"
2461
2462 case OO_New: OperatorName = "operator new"; break;
2463 case OO_Delete: OperatorName = "operator delete"; break;
2464 case OO_Array_New: OperatorName = "operator new[]"; break;
2465 case OO_Array_Delete: OperatorName = "operator delete[]"; break;
2466 case OO_Call: OperatorName = "operator()"; break;
2467 case OO_Subscript: OperatorName = "operator[]"; break;
2468 }
2469 Result.AddTypedTextChunk(OperatorName);
2470 break;
2471 }
2472
Douglas Gregor0212fd72010-09-21 16:06:22 +00002473 case DeclarationName::Identifier:
2474 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor0212fd72010-09-21 16:06:22 +00002475 case DeclarationName::CXXDestructorName:
2476 case DeclarationName::CXXLiteralOperatorName:
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002477 Result.AddTypedTextChunk(
2478 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002479 break;
2480
2481 case DeclarationName::CXXUsingDirective:
2482 case DeclarationName::ObjCZeroArgSelector:
2483 case DeclarationName::ObjCOneArgSelector:
2484 case DeclarationName::ObjCMultiArgSelector:
2485 break;
2486
2487 case DeclarationName::CXXConstructorName: {
Craig Topperc3ec1492014-05-26 06:22:03 +00002488 CXXRecordDecl *Record = nullptr;
Douglas Gregor0212fd72010-09-21 16:06:22 +00002489 QualType Ty = Name.getCXXNameType();
2490 if (const RecordType *RecordTy = Ty->getAs<RecordType>())
2491 Record = cast<CXXRecordDecl>(RecordTy->getDecl());
2492 else if (const InjectedClassNameType *InjectedTy
2493 = Ty->getAs<InjectedClassNameType>())
2494 Record = InjectedTy->getDecl();
2495 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002496 Result.AddTypedTextChunk(
2497 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002498 break;
2499 }
2500
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002501 Result.AddTypedTextChunk(
2502 Result.getAllocator().CopyString(Record->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002503 if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002504 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Douglas Gregor75acd922011-09-27 23:30:47 +00002505 AddTemplateParameterChunks(Context, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002506 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002507 }
2508 break;
2509 }
2510 }
2511}
2512
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002513CodeCompletionString *CodeCompletionResult::CreateCodeCompletionString(Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002514 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002515 CodeCompletionTUInfo &CCTUInfo,
2516 bool IncludeBriefComments) {
2517 return CreateCodeCompletionString(S.Context, S.PP, Allocator, CCTUInfo,
2518 IncludeBriefComments);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002519}
2520
Douglas Gregor3545ff42009-09-21 16:56:56 +00002521/// \brief If possible, create a new code completion string for the given
2522/// result.
2523///
2524/// \returns Either a new, heap-allocated code completion string describing
2525/// how to use this result, or NULL to indicate that the string or name of the
2526/// result is all that is needed.
2527CodeCompletionString *
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002528CodeCompletionResult::CreateCodeCompletionString(ASTContext &Ctx,
2529 Preprocessor &PP,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002530 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002531 CodeCompletionTUInfo &CCTUInfo,
2532 bool IncludeBriefComments) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002533 CodeCompletionBuilder Result(Allocator, CCTUInfo, Priority, Availability);
Douglas Gregor9eb77012009-11-07 00:00:49 +00002534
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002535 PrintingPolicy Policy = getCompletionPrintingPolicy(Ctx, PP);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002536 if (Kind == RK_Pattern) {
2537 Pattern->Priority = Priority;
2538 Pattern->Availability = Availability;
Douglas Gregor78254c82012-03-27 23:34:16 +00002539
2540 if (Declaration) {
2541 Result.addParentContext(Declaration->getDeclContext());
Douglas Gregor78254c82012-03-27 23:34:16 +00002542 Pattern->ParentName = Result.getParentName();
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002543 // Provide code completion comment for self.GetterName where
2544 // GetterName is the getter method for a property with name
2545 // different from the property name (declared via a property
2546 // getter attribute.
2547 const NamedDecl *ND = Declaration;
2548 if (const ObjCMethodDecl *M = dyn_cast<ObjCMethodDecl>(ND))
2549 if (M->isPropertyAccessor())
2550 if (const ObjCPropertyDecl *PDecl = M->findPropertyDecl())
2551 if (PDecl->getGetterName() == M->getSelector() &&
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002552 PDecl->getIdentifier() != M->getIdentifier()) {
2553 if (const RawComment *RC =
2554 Ctx.getRawCommentForAnyRedecl(M)) {
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002555 Result.addBriefComment(RC->getBriefText(Ctx));
2556 Pattern->BriefComment = Result.getBriefComment();
2557 }
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002558 else if (const RawComment *RC =
2559 Ctx.getRawCommentForAnyRedecl(PDecl)) {
2560 Result.addBriefComment(RC->getBriefText(Ctx));
2561 Pattern->BriefComment = Result.getBriefComment();
2562 }
2563 }
Douglas Gregor78254c82012-03-27 23:34:16 +00002564 }
2565
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002566 return Pattern;
2567 }
Douglas Gregorf09935f2009-12-01 05:55:20 +00002568
Douglas Gregorf09935f2009-12-01 05:55:20 +00002569 if (Kind == RK_Keyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002570 Result.AddTypedTextChunk(Keyword);
2571 return Result.TakeString();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002572 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002573
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002574 if (Kind == RK_Macro) {
Argyrios Kyrtzidis09c9e812013-02-20 00:54:57 +00002575 const MacroDirective *MD = PP.getMacroDirectiveHistory(Macro);
2576 assert(MD && "Not a macro?");
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00002577 const MacroInfo *MI = MD->getMacroInfo();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002578
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002579 Result.AddTypedTextChunk(
2580 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregorf09935f2009-12-01 05:55:20 +00002581
2582 if (!MI->isFunctionLike())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002583 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002584
2585 // Format a function-like macro with placeholders for the arguments.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002586 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor0c505312011-07-30 08:17:44 +00002587 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002588
2589 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
2590 if (MI->isC99Varargs()) {
2591 --AEnd;
2592
2593 if (A == AEnd) {
2594 Result.AddPlaceholderChunk("...");
2595 }
Douglas Gregor0c505312011-07-30 08:17:44 +00002596 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002597
Douglas Gregor0c505312011-07-30 08:17:44 +00002598 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002599 if (A != MI->arg_begin())
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002600 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3aa55262012-01-21 00:43:38 +00002601
2602 if (MI->isVariadic() && (A+1) == AEnd) {
Dylan Noblesmith2c1dd272012-02-05 02:13:05 +00002603 SmallString<32> Arg = (*A)->getName();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002604 if (MI->isC99Varargs())
2605 Arg += ", ...";
2606 else
2607 Arg += "...";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002608 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor3aa55262012-01-21 00:43:38 +00002609 break;
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002610 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002611
2612 // Non-variadic macros are simple.
2613 Result.AddPlaceholderChunk(
2614 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregor0c505312011-07-30 08:17:44 +00002615 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002616 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002617 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002618 }
2619
Douglas Gregorf64acca2010-05-25 21:41:55 +00002620 assert(Kind == RK_Declaration && "Missed a result kind?");
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002621 const NamedDecl *ND = Declaration;
Douglas Gregor78254c82012-03-27 23:34:16 +00002622 Result.addParentContext(ND->getDeclContext());
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002623
2624 if (IncludeBriefComments) {
2625 // Add documentation comment, if it exists.
Dmitri Gribenkoa43ec182012-08-11 00:51:43 +00002626 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(ND)) {
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002627 Result.addBriefComment(RC->getBriefText(Ctx));
Fariborz Jahanian15a0b552013-02-28 17:47:14 +00002628 }
2629 else if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2630 if (OMD->isPropertyAccessor())
2631 if (const ObjCPropertyDecl *PDecl = OMD->findPropertyDecl())
2632 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(PDecl))
2633 Result.addBriefComment(RC->getBriefText(Ctx));
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002634 }
2635
Douglas Gregor9eb77012009-11-07 00:00:49 +00002636 if (StartsNestedNameSpecifier) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002637 Result.AddTypedTextChunk(
2638 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002639 Result.AddTextChunk("::");
2640 return Result.TakeString();
Douglas Gregor9eb77012009-11-07 00:00:49 +00002641 }
Erik Verbruggen98ea7f62011-10-14 15:31:08 +00002642
Aaron Ballmanbe22bcb2014-03-10 17:08:28 +00002643 for (const auto *I : ND->specific_attrs<AnnotateAttr>())
2644 Result.AddAnnotation(Result.getAllocator().CopyString(I->getAnnotation()));
Aaron Ballmanb97112e2014-03-08 22:19:01 +00002645
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002646 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002647
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002648 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002649 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002650 Ctx, Policy);
2651 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002652 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002653 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002654 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002655 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002656 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002657 }
2658
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002659 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002660 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002661 Ctx, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002662 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002663 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002664
Douglas Gregor3545ff42009-09-21 16:56:56 +00002665 // Figure out which template parameters are deduced (or have default
2666 // arguments).
Benjamin Kramere0513cb2012-01-30 16:17:39 +00002667 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002668 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002669 unsigned LastDeducibleArgument;
2670 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2671 --LastDeducibleArgument) {
2672 if (!Deduced[LastDeducibleArgument - 1]) {
2673 // C++0x: Figure out if the template argument has a default. If so,
2674 // the user doesn't need to type this argument.
2675 // FIXME: We need to abstract template parameters better!
2676 bool HasDefaultArg = false;
2677 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002678 LastDeducibleArgument - 1);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002679 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2680 HasDefaultArg = TTP->hasDefaultArgument();
2681 else if (NonTypeTemplateParmDecl *NTTP
2682 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2683 HasDefaultArg = NTTP->hasDefaultArgument();
2684 else {
2685 assert(isa<TemplateTemplateParmDecl>(Param));
2686 HasDefaultArg
Douglas Gregor9eb77012009-11-07 00:00:49 +00002687 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002688 }
2689
2690 if (!HasDefaultArg)
2691 break;
2692 }
2693 }
2694
2695 if (LastDeducibleArgument) {
2696 // Some of the function template arguments cannot be deduced from a
2697 // function call, so we introduce an explicit template argument list
2698 // containing all of the arguments up to the first deducible argument.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002699 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002700 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor3545ff42009-09-21 16:56:56 +00002701 LastDeducibleArgument);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002702 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002703 }
2704
2705 // Add the function parameters
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002706 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002707 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002708 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002709 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002710 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002711 }
2712
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002713 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002714 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002715 Ctx, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002716 Result.AddTypedTextChunk(
2717 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002718 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002719 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002720 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002721 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002722 }
2723
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002724 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregord3c5d792009-11-17 16:44:22 +00002725 Selector Sel = Method->getSelector();
2726 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002727 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002728 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002729 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002730 }
2731
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002732 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregor1b605f72009-11-19 01:08:35 +00002733 SelName += ':';
2734 if (StartParameter == 0)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002735 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002736 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002737 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002738
2739 // If there is only one parameter, and we're past it, add an empty
2740 // typed-text chunk since there is nothing to type.
2741 if (Method->param_size() == 1)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002742 Result.AddTypedTextChunk("");
Douglas Gregor1b605f72009-11-19 01:08:35 +00002743 }
Douglas Gregord3c5d792009-11-17 16:44:22 +00002744 unsigned Idx = 0;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002745 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2746 PEnd = Method->param_end();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002747 P != PEnd; (void)++P, ++Idx) {
2748 if (Idx > 0) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00002749 std::string Keyword;
2750 if (Idx > StartParameter)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002751 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregord3c5d792009-11-17 16:44:22 +00002752 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramer632500c2011-07-26 16:59:25 +00002753 Keyword += II->getName();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002754 Keyword += ":";
Douglas Gregor95887f92010-07-08 23:20:03 +00002755 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002756 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002757 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002758 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002759 }
Douglas Gregor1b605f72009-11-19 01:08:35 +00002760
2761 // If we're before the starting parameter, skip the placeholder.
2762 if (Idx < StartParameter)
2763 continue;
Douglas Gregord3c5d792009-11-17 16:44:22 +00002764
2765 std::string Arg;
Douglas Gregore90dd002010-08-24 16:15:59 +00002766
2767 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002768 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregore90dd002010-08-24 16:15:59 +00002769 else {
John McCall31168b02011-06-15 23:02:42 +00002770 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor8f08d742011-07-30 07:55:26 +00002771 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2772 + Arg + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002773 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregor981a0c42010-08-29 19:47:46 +00002774 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramer632500c2011-07-26 16:59:25 +00002775 Arg += II->getName();
Douglas Gregore90dd002010-08-24 16:15:59 +00002776 }
2777
Douglas Gregor400f5972010-08-31 05:13:43 +00002778 if (Method->isVariadic() && (P + 1) == PEnd)
2779 Arg += ", ...";
2780
Douglas Gregor95887f92010-07-08 23:20:03 +00002781 if (DeclaringEntity)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002782 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor95887f92010-07-08 23:20:03 +00002783 else if (AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002784 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8537c52009-11-19 07:41:15 +00002785 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002786 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002787 }
2788
Douglas Gregor04c5f972009-12-23 00:21:46 +00002789 if (Method->isVariadic()) {
Douglas Gregor400f5972010-08-31 05:13:43 +00002790 if (Method->param_size() == 0) {
2791 if (DeclaringEntity)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002792 Result.AddTextChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002793 else if (AllParametersAreInformative)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002794 Result.AddInformativeChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002795 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002796 Result.AddPlaceholderChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002797 }
Douglas Gregordbb71db2010-08-23 23:51:41 +00002798
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002799 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor04c5f972009-12-23 00:21:46 +00002800 }
2801
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002802 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002803 }
2804
Douglas Gregorf09935f2009-12-01 05:55:20 +00002805 if (Qualifier)
Douglas Gregor5bf52692009-09-22 23:15:58 +00002806 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002807 Ctx, Policy);
Douglas Gregorf09935f2009-12-01 05:55:20 +00002808
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002809 Result.AddTypedTextChunk(
2810 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002811 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002812}
2813
Douglas Gregorf0f51982009-09-23 00:34:09 +00002814CodeCompletionString *
2815CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2816 unsigned CurrentArg,
Douglas Gregor36e3b5c2010-10-11 21:37:58 +00002817 Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002818 CodeCompletionAllocator &Allocator,
2819 CodeCompletionTUInfo &CCTUInfo) const {
Douglas Gregor75acd922011-09-27 23:30:47 +00002820 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCall31168b02011-06-15 23:02:42 +00002821
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002822 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002823 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002824 FunctionDecl *FDecl = getFunction();
Douglas Gregor75acd922011-09-27 23:30:47 +00002825 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002826 const FunctionProtoType *Proto
2827 = dyn_cast<FunctionProtoType>(getFunctionType());
2828 if (!FDecl && !Proto) {
2829 // Function without a prototype. Just give the return type and a
2830 // highlighted ellipsis.
2831 const FunctionType *FT = getFunctionType();
Alp Toker314cc812014-01-25 16:55:45 +00002832 Result.AddTextChunk(GetCompletionTypeString(FT->getReturnType(), S.Context,
2833 Policy, Result.getAllocator()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002834 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2835 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2836 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002837 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002838 }
2839
2840 if (FDecl)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002841 Result.AddTextChunk(
2842 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002843 else
Alp Toker314cc812014-01-25 16:55:45 +00002844 Result.AddTextChunk(Result.getAllocator().CopyString(
2845 Proto->getReturnType().getAsString(Policy)));
2846
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002847 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Alp Toker9cacbab2014-01-20 20:26:09 +00002848 unsigned NumParams = FDecl ? FDecl->getNumParams() : Proto->getNumParams();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002849 for (unsigned I = 0; I != NumParams; ++I) {
2850 if (I)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002851 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002852
2853 std::string ArgString;
2854 QualType ArgType;
2855
2856 if (FDecl) {
2857 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2858 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2859 } else {
Alp Toker9cacbab2014-01-20 20:26:09 +00002860 ArgType = Proto->getParamType(I);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002861 }
2862
John McCall31168b02011-06-15 23:02:42 +00002863 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002864
2865 if (I == CurrentArg)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002866 Result.AddChunk(CodeCompletionString::CK_CurrentParameter,
2867 Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002868 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002869 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002870 }
2871
2872 if (Proto && Proto->isVariadic()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002873 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002874 if (CurrentArg < NumParams)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002875 Result.AddTextChunk("...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002876 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002877 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002878 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002879 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002880
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002881 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002882}
2883
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002884unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002885 const LangOptions &LangOpts,
Douglas Gregor6e240332010-08-16 16:18:59 +00002886 bool PreferredTypeIsPointer) {
2887 unsigned Priority = CCP_Macro;
2888
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002889 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2890 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2891 MacroName.equals("Nil")) {
Douglas Gregor6e240332010-08-16 16:18:59 +00002892 Priority = CCP_Constant;
2893 if (PreferredTypeIsPointer)
2894 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002895 }
2896 // Treat "YES", "NO", "true", and "false" as constants.
2897 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2898 MacroName.equals("true") || MacroName.equals("false"))
2899 Priority = CCP_Constant;
2900 // Treat "bool" as a type.
2901 else if (MacroName.equals("bool"))
2902 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2903
Douglas Gregor6e240332010-08-16 16:18:59 +00002904
2905 return Priority;
2906}
2907
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002908CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002909 if (!D)
2910 return CXCursor_UnexposedDecl;
2911
2912 switch (D->getKind()) {
2913 case Decl::Enum: return CXCursor_EnumDecl;
2914 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2915 case Decl::Field: return CXCursor_FieldDecl;
2916 case Decl::Function:
2917 return CXCursor_FunctionDecl;
2918 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2919 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002920 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregordeafd0b2011-12-27 22:43:10 +00002921
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002922 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002923 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2924 case Decl::ObjCMethod:
2925 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2926 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2927 case Decl::CXXMethod: return CXCursor_CXXMethod;
2928 case Decl::CXXConstructor: return CXCursor_Constructor;
2929 case Decl::CXXDestructor: return CXCursor_Destructor;
2930 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2931 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002932 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002933 case Decl::ParmVar: return CXCursor_ParmDecl;
2934 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smithdda56e42011-04-15 14:24:37 +00002935 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002936 case Decl::Var: return CXCursor_VarDecl;
2937 case Decl::Namespace: return CXCursor_Namespace;
2938 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2939 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2940 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2941 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2942 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2943 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis12afd702011-09-30 17:58:23 +00002944 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002945 case Decl::ClassTemplatePartialSpecialization:
2946 return CXCursor_ClassTemplatePartialSpecialization;
2947 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor3e653b32012-04-30 23:41:16 +00002948 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002949
2950 case Decl::Using:
2951 case Decl::UnresolvedUsingValue:
2952 case Decl::UnresolvedUsingTypename:
2953 return CXCursor_UsingDeclaration;
2954
Douglas Gregor4cd65962011-06-03 23:08:58 +00002955 case Decl::ObjCPropertyImpl:
2956 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2957 case ObjCPropertyImplDecl::Dynamic:
2958 return CXCursor_ObjCDynamicDecl;
2959
2960 case ObjCPropertyImplDecl::Synthesize:
2961 return CXCursor_ObjCSynthesizeDecl;
2962 }
Argyrios Kyrtzidis50e5b1d2012-10-05 00:22:24 +00002963
2964 case Decl::Import:
2965 return CXCursor_ModuleImportDecl;
Douglas Gregor4cd65962011-06-03 23:08:58 +00002966
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002967 default:
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002968 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002969 switch (TD->getTagKind()) {
Joao Matosdc86f942012-08-31 18:45:21 +00002970 case TTK_Interface: // fall through
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002971 case TTK_Struct: return CXCursor_StructDecl;
2972 case TTK_Class: return CXCursor_ClassDecl;
2973 case TTK_Union: return CXCursor_UnionDecl;
2974 case TTK_Enum: return CXCursor_EnumDecl;
2975 }
2976 }
2977 }
2978
2979 return CXCursor_UnexposedDecl;
2980}
2981
Douglas Gregor55b037b2010-07-08 20:55:51 +00002982static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor8cb17462012-10-09 16:01:50 +00002983 bool IncludeUndefined,
Douglas Gregor55b037b2010-07-08 20:55:51 +00002984 bool TargetTypeIsPointer = false) {
John McCall276321a2010-08-25 06:19:51 +00002985 typedef CodeCompletionResult Result;
Douglas Gregor55b037b2010-07-08 20:55:51 +00002986
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002987 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002988
Douglas Gregor9eb77012009-11-07 00:00:49 +00002989 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2990 MEnd = PP.macro_end();
Douglas Gregor55b037b2010-07-08 20:55:51 +00002991 M != MEnd; ++M) {
Argyrios Kyrtzidis9ef53ce2014-04-09 18:21:23 +00002992 if (IncludeUndefined || M->first->hasMacroDefinition()) {
2993 if (MacroInfo *MI = M->second->getMacroInfo())
2994 if (MI->isUsedForHeaderGuard())
2995 continue;
2996
Douglas Gregor8cb17462012-10-09 16:01:50 +00002997 Results.AddResult(Result(M->first,
Douglas Gregor6e240332010-08-16 16:18:59 +00002998 getMacroUsagePriority(M->first->getName(),
David Blaikiebbafb8a2012-03-11 07:00:24 +00002999 PP.getLangOpts(),
Douglas Gregor6e240332010-08-16 16:18:59 +00003000 TargetTypeIsPointer)));
Argyrios Kyrtzidis9ef53ce2014-04-09 18:21:23 +00003001 }
Douglas Gregor55b037b2010-07-08 20:55:51 +00003002 }
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003003
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003004 Results.ExitScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003005
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003006}
3007
Douglas Gregorce0e8562010-08-23 21:54:33 +00003008static void AddPrettyFunctionResults(const LangOptions &LangOpts,
3009 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003010 typedef CodeCompletionResult Result;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003011
3012 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003013
Douglas Gregorce0e8562010-08-23 21:54:33 +00003014 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
3015 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith2bf7fdb2013-01-02 11:42:31 +00003016 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregorce0e8562010-08-23 21:54:33 +00003017 Results.AddResult(Result("__func__", CCP_Constant));
3018 Results.ExitScope();
3019}
3020
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003021static void HandleCodeCompleteResults(Sema *S,
3022 CodeCompleteConsumer *CodeCompleter,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003023 CodeCompletionContext Context,
John McCall276321a2010-08-25 06:19:51 +00003024 CodeCompletionResult *Results,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003025 unsigned NumResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00003026 if (CodeCompleter)
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003027 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor3545ff42009-09-21 16:56:56 +00003028}
3029
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003030static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3031 Sema::ParserCompletionContext PCC) {
3032 switch (PCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00003033 case Sema::PCC_Namespace:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003034 return CodeCompletionContext::CCC_TopLevel;
3035
John McCallfaf5fb42010-08-26 23:41:50 +00003036 case Sema::PCC_Class:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003037 return CodeCompletionContext::CCC_ClassStructUnion;
3038
John McCallfaf5fb42010-08-26 23:41:50 +00003039 case Sema::PCC_ObjCInterface:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003040 return CodeCompletionContext::CCC_ObjCInterface;
3041
John McCallfaf5fb42010-08-26 23:41:50 +00003042 case Sema::PCC_ObjCImplementation:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003043 return CodeCompletionContext::CCC_ObjCImplementation;
3044
John McCallfaf5fb42010-08-26 23:41:50 +00003045 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003046 return CodeCompletionContext::CCC_ObjCIvarList;
3047
John McCallfaf5fb42010-08-26 23:41:50 +00003048 case Sema::PCC_Template:
3049 case Sema::PCC_MemberTemplate:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003050 if (S.CurContext->isFileContext())
3051 return CodeCompletionContext::CCC_TopLevel;
David Blaikie8a40f702012-01-17 06:56:22 +00003052 if (S.CurContext->isRecord())
Douglas Gregor0ac41382010-09-23 23:01:17 +00003053 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie8a40f702012-01-17 06:56:22 +00003054 return CodeCompletionContext::CCC_Other;
Douglas Gregor0ac41382010-09-23 23:01:17 +00003055
John McCallfaf5fb42010-08-26 23:41:50 +00003056 case Sema::PCC_RecoveryInFunction:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003057 return CodeCompletionContext::CCC_Recovery;
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003058
John McCallfaf5fb42010-08-26 23:41:50 +00003059 case Sema::PCC_ForInit:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003060 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3061 S.getLangOpts().ObjC1)
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003062 return CodeCompletionContext::CCC_ParenthesizedExpression;
3063 else
3064 return CodeCompletionContext::CCC_Expression;
3065
3066 case Sema::PCC_Expression:
John McCallfaf5fb42010-08-26 23:41:50 +00003067 case Sema::PCC_Condition:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003068 return CodeCompletionContext::CCC_Expression;
3069
John McCallfaf5fb42010-08-26 23:41:50 +00003070 case Sema::PCC_Statement:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003071 return CodeCompletionContext::CCC_Statement;
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003072
John McCallfaf5fb42010-08-26 23:41:50 +00003073 case Sema::PCC_Type:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003074 return CodeCompletionContext::CCC_Type;
Douglas Gregor5e35d592010-09-14 23:59:36 +00003075
3076 case Sema::PCC_ParenthesizedExpression:
3077 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor80039242011-02-15 20:33:25 +00003078
3079 case Sema::PCC_LocalDeclarationSpecifiers:
3080 return CodeCompletionContext::CCC_Type;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003081 }
David Blaikie8a40f702012-01-17 06:56:22 +00003082
3083 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003084}
3085
Douglas Gregorac322ec2010-08-27 21:18:54 +00003086/// \brief If we're in a C++ virtual member function, add completion results
3087/// that invoke the functions we override, since it's common to invoke the
3088/// overridden function as well as adding new functionality.
3089///
3090/// \param S The semantic analysis object for which we are generating results.
3091///
3092/// \param InContext This context in which the nested-name-specifier preceding
3093/// the code-completion point
3094static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3095 ResultBuilder &Results) {
3096 // Look through blocks.
3097 DeclContext *CurContext = S.CurContext;
3098 while (isa<BlockDecl>(CurContext))
3099 CurContext = CurContext->getParent();
3100
3101
3102 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3103 if (!Method || !Method->isVirtual())
3104 return;
3105
3106 // We need to have names for all of the parameters, if we're going to
3107 // generate a forwarding call.
Aaron Ballman43b68be2014-03-07 17:50:17 +00003108 for (auto P : Method->params())
3109 if (!P->getDeclName())
Douglas Gregorac322ec2010-08-27 21:18:54 +00003110 return;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003111
Douglas Gregor75acd922011-09-27 23:30:47 +00003112 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003113 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3114 MEnd = Method->end_overridden_methods();
3115 M != MEnd; ++M) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003116 CodeCompletionBuilder Builder(Results.getAllocator(),
3117 Results.getCodeCompletionTUInfo());
Dmitri Gribenko6cfb1532013-02-14 13:53:30 +00003118 const CXXMethodDecl *Overridden = *M;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003119 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3120 continue;
3121
3122 // If we need a nested-name-specifier, add one now.
3123 if (!InContext) {
3124 NestedNameSpecifier *NNS
3125 = getRequiredQualification(S.Context, CurContext,
3126 Overridden->getDeclContext());
3127 if (NNS) {
3128 std::string Str;
3129 llvm::raw_string_ostream OS(Str);
Douglas Gregor75acd922011-09-27 23:30:47 +00003130 NNS->print(OS, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003131 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003132 }
3133 } else if (!InContext->Equals(Overridden->getDeclContext()))
3134 continue;
3135
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003136 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003137 Overridden->getNameAsString()));
3138 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003139 bool FirstParam = true;
Aaron Ballman43b68be2014-03-07 17:50:17 +00003140 for (auto P : Method->params()) {
Douglas Gregorac322ec2010-08-27 21:18:54 +00003141 if (FirstParam)
3142 FirstParam = false;
3143 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003144 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003145
Aaron Ballman43b68be2014-03-07 17:50:17 +00003146 Builder.AddPlaceholderChunk(
3147 Results.getAllocator().CopyString(P->getIdentifier()->getName()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003148 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003149 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3150 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorac322ec2010-08-27 21:18:54 +00003151 CCP_SuperCompletion,
Douglas Gregor78254c82012-03-27 23:34:16 +00003152 CXCursor_CXXMethod,
3153 CXAvailability_Available,
3154 Overridden));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003155 Results.Ignore(Overridden);
3156 }
3157}
3158
Douglas Gregor07f43572012-01-29 18:15:03 +00003159void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3160 ModuleIdPath Path) {
3161 typedef CodeCompletionResult Result;
3162 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003163 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor07f43572012-01-29 18:15:03 +00003164 CodeCompletionContext::CCC_Other);
3165 Results.EnterNewScope();
3166
3167 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003168 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor07f43572012-01-29 18:15:03 +00003169 typedef CodeCompletionResult Result;
3170 if (Path.empty()) {
3171 // Enumerate all top-level modules.
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003172 SmallVector<Module *, 8> Modules;
Douglas Gregor07f43572012-01-29 18:15:03 +00003173 PP.getHeaderSearchInfo().collectAllModules(Modules);
3174 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3175 Builder.AddTypedTextChunk(
3176 Builder.getAllocator().CopyString(Modules[I]->Name));
3177 Results.AddResult(Result(Builder.TakeString(),
3178 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003179 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003180 Modules[I]->isAvailable()
3181 ? CXAvailability_Available
3182 : CXAvailability_NotAvailable));
3183 }
Daniel Jasper07e6c402013-08-05 20:26:17 +00003184 } else if (getLangOpts().Modules) {
Douglas Gregor07f43572012-01-29 18:15:03 +00003185 // Load the named module.
3186 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3187 Module::AllVisible,
3188 /*IsInclusionDirective=*/false);
3189 // Enumerate submodules.
3190 if (Mod) {
3191 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3192 SubEnd = Mod->submodule_end();
3193 Sub != SubEnd; ++Sub) {
3194
3195 Builder.AddTypedTextChunk(
3196 Builder.getAllocator().CopyString((*Sub)->Name));
3197 Results.AddResult(Result(Builder.TakeString(),
3198 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003199 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003200 (*Sub)->isAvailable()
3201 ? CXAvailability_Available
3202 : CXAvailability_NotAvailable));
3203 }
3204 }
3205 }
3206 Results.ExitScope();
3207 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3208 Results.data(),Results.size());
3209}
3210
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003211void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003212 ParserCompletionContext CompletionContext) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003213 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003214 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003215 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003216 Results.EnterNewScope();
Douglas Gregor50832e02010-09-20 22:39:41 +00003217
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003218 // Determine how to filter results, e.g., so that the names of
3219 // values (functions, enumerators, function templates, etc.) are
3220 // only allowed where we can have an expression.
3221 switch (CompletionContext) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003222 case PCC_Namespace:
3223 case PCC_Class:
3224 case PCC_ObjCInterface:
3225 case PCC_ObjCImplementation:
3226 case PCC_ObjCInstanceVariableList:
3227 case PCC_Template:
3228 case PCC_MemberTemplate:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003229 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003230 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003231 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3232 break;
3233
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003234 case PCC_Statement:
Douglas Gregor5e35d592010-09-14 23:59:36 +00003235 case PCC_ParenthesizedExpression:
Douglas Gregor4d755e82010-08-24 23:58:17 +00003236 case PCC_Expression:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003237 case PCC_ForInit:
3238 case PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003239 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor70febae2010-05-28 00:49:12 +00003240 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3241 else
3242 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003243
David Blaikiebbafb8a2012-03-11 07:00:24 +00003244 if (getLangOpts().CPlusPlus)
Craig Topperc3ec1492014-05-26 06:22:03 +00003245 MaybeAddOverrideCalls(*this, /*InContext=*/nullptr, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003246 break;
Douglas Gregor6da3db42010-05-25 05:58:43 +00003247
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003248 case PCC_RecoveryInFunction:
Douglas Gregor6da3db42010-05-25 05:58:43 +00003249 // Unfiltered
3250 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003251 }
3252
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003253 // If we are in a C++ non-static member function, check the qualifiers on
3254 // the member function to filter/prioritize the results list.
3255 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3256 if (CurMethod->isInstance())
3257 Results.setObjectTypeQualifiers(
3258 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3259
Douglas Gregorc580c522010-01-14 01:09:38 +00003260 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003261 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3262 CodeCompleter->includeGlobals());
Douglas Gregor92253692009-12-07 09:54:55 +00003263
Douglas Gregorf98e6a22010-01-13 23:51:12 +00003264 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor92253692009-12-07 09:54:55 +00003265 Results.ExitScope();
3266
Douglas Gregorce0e8562010-08-23 21:54:33 +00003267 switch (CompletionContext) {
Douglas Gregor5e35d592010-09-14 23:59:36 +00003268 case PCC_ParenthesizedExpression:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003269 case PCC_Expression:
3270 case PCC_Statement:
3271 case PCC_RecoveryInFunction:
3272 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00003273 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003274 break;
3275
3276 case PCC_Namespace:
3277 case PCC_Class:
3278 case PCC_ObjCInterface:
3279 case PCC_ObjCImplementation:
3280 case PCC_ObjCInstanceVariableList:
3281 case PCC_Template:
3282 case PCC_MemberTemplate:
3283 case PCC_ForInit:
3284 case PCC_Condition:
3285 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003286 case PCC_LocalDeclarationSpecifiers:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003287 break;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003288 }
3289
Douglas Gregor9eb77012009-11-07 00:00:49 +00003290 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003291 AddMacroResults(PP, Results, false);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003292
Douglas Gregor50832e02010-09-20 22:39:41 +00003293 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003294 Results.data(),Results.size());
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00003295}
3296
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003297static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3298 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003299 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00003300 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003301 bool IsSuper,
3302 ResultBuilder &Results);
3303
3304void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3305 bool AllowNonIdentifiers,
3306 bool AllowNestedNameSpecifiers) {
John McCall276321a2010-08-25 06:19:51 +00003307 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003308 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003309 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003310 AllowNestedNameSpecifiers
3311 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3312 : CodeCompletionContext::CCC_Name);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003313 Results.EnterNewScope();
3314
3315 // Type qualifiers can come after names.
3316 Results.AddResult(Result("const"));
3317 Results.AddResult(Result("volatile"));
David Blaikiebbafb8a2012-03-11 07:00:24 +00003318 if (getLangOpts().C99)
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003319 Results.AddResult(Result("restrict"));
3320
David Blaikiebbafb8a2012-03-11 07:00:24 +00003321 if (getLangOpts().CPlusPlus) {
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003322 if (AllowNonIdentifiers) {
3323 Results.AddResult(Result("operator"));
3324 }
3325
3326 // Add nested-name-specifiers.
3327 if (AllowNestedNameSpecifiers) {
3328 Results.allowNestedNameSpecifiers();
Douglas Gregor0ac41382010-09-23 23:01:17 +00003329 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003330 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3331 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3332 CodeCompleter->includeGlobals());
Craig Topperc3ec1492014-05-26 06:22:03 +00003333 Results.setFilter(nullptr);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003334 }
3335 }
3336 Results.ExitScope();
3337
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003338 // If we're in a context where we might have an expression (rather than a
3339 // declaration), and what we've seen so far is an Objective-C type that could
3340 // be a receiver of a class message, this may be a class message send with
3341 // the initial opening bracket '[' missing. Add appropriate completions.
3342 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003343 DS.getParsedSpecifiers() == DeclSpec::PQ_TypeSpecifier &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003344 DS.getTypeSpecType() == DeclSpec::TST_typename &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003345 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3346 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003347 !DS.isTypeAltiVecVector() &&
3348 S &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003349 (S->getFlags() & Scope::DeclScope) != 0 &&
3350 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3351 Scope::FunctionPrototypeScope |
3352 Scope::AtCatchScope)) == 0) {
3353 ParsedType T = DS.getRepAsType();
3354 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003355 AddClassMessageCompletions(*this, S, T, None, false, false, Results);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003356 }
3357
Douglas Gregor56ccce02010-08-24 04:59:56 +00003358 // Note that we intentionally suppress macro results here, since we do not
3359 // encourage using macros to produce the names of entities.
3360
Douglas Gregor0ac41382010-09-23 23:01:17 +00003361 HandleCodeCompleteResults(this, CodeCompleter,
3362 Results.getCompletionContext(),
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003363 Results.data(), Results.size());
3364}
3365
Douglas Gregor68762e72010-08-23 21:17:50 +00003366struct Sema::CodeCompleteExpressionData {
3367 CodeCompleteExpressionData(QualType PreferredType = QualType())
3368 : PreferredType(PreferredType), IntegralConstantExpression(false),
3369 ObjCCollection(false) { }
3370
3371 QualType PreferredType;
3372 bool IntegralConstantExpression;
3373 bool ObjCCollection;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003374 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregor68762e72010-08-23 21:17:50 +00003375};
3376
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003377/// \brief Perform code-completion in an expression context when we know what
3378/// type we're looking for.
Douglas Gregor68762e72010-08-23 21:17:50 +00003379void Sema::CodeCompleteExpression(Scope *S,
3380 const CodeCompleteExpressionData &Data) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003381 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003382 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003383 CodeCompletionContext::CCC_Expression);
Douglas Gregor68762e72010-08-23 21:17:50 +00003384 if (Data.ObjCCollection)
3385 Results.setFilter(&ResultBuilder::IsObjCCollection);
3386 else if (Data.IntegralConstantExpression)
Douglas Gregor85b50632010-07-28 21:50:18 +00003387 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003388 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003389 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3390 else
3391 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregor68762e72010-08-23 21:17:50 +00003392
3393 if (!Data.PreferredType.isNull())
3394 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3395
3396 // Ignore any declarations that we were told that we don't care about.
3397 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3398 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003399
3400 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003401 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3402 CodeCompleter->includeGlobals());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003403
3404 Results.EnterNewScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003405 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003406 Results.ExitScope();
3407
Douglas Gregor55b037b2010-07-08 20:55:51 +00003408 bool PreferredTypeIsPointer = false;
Douglas Gregor68762e72010-08-23 21:17:50 +00003409 if (!Data.PreferredType.isNull())
3410 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3411 || Data.PreferredType->isMemberPointerType()
3412 || Data.PreferredType->isBlockPointerType();
Douglas Gregor55b037b2010-07-08 20:55:51 +00003413
Douglas Gregorce0e8562010-08-23 21:54:33 +00003414 if (S->getFnParent() &&
3415 !Data.ObjCCollection &&
3416 !Data.IntegralConstantExpression)
David Blaikiebbafb8a2012-03-11 07:00:24 +00003417 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003418
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003419 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003420 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003421 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor68762e72010-08-23 21:17:50 +00003422 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3423 Data.PreferredType),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003424 Results.data(),Results.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003425}
3426
Douglas Gregoreda7e542010-09-18 01:28:11 +00003427void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3428 if (E.isInvalid())
3429 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003430 else if (getLangOpts().ObjC1)
Nikola Smiljanic01a75982014-05-29 10:55:11 +00003431 CodeCompleteObjCInstanceMessage(S, E.get(), None, false);
Douglas Gregored0b69d2010-09-15 16:23:04 +00003432}
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003433
Douglas Gregorb888acf2010-12-09 23:01:55 +00003434/// \brief The set of properties that have already been added, referenced by
3435/// property name.
3436typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3437
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003438/// \brief Retrieve the container definition, if any?
3439static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3440 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3441 if (Interface->hasDefinition())
3442 return Interface->getDefinition();
3443
3444 return Interface;
3445 }
3446
3447 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3448 if (Protocol->hasDefinition())
3449 return Protocol->getDefinition();
3450
3451 return Protocol;
3452 }
3453 return Container;
3454}
3455
3456static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor5d649882009-11-18 22:32:06 +00003457 bool AllowCategories,
Douglas Gregor95147142011-05-05 15:50:42 +00003458 bool AllowNullaryMethods,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003459 DeclContext *CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003460 AddedPropertiesSet &AddedProperties,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003461 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003462 typedef CodeCompletionResult Result;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003463
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003464 // Retrieve the definition.
3465 Container = getContainerDef(Container);
3466
Douglas Gregor9291bad2009-11-18 01:29:26 +00003467 // Add properties in this container.
Aaron Ballmand174edf2014-03-13 19:11:50 +00003468 for (const auto *P : Container->properties())
Douglas Gregorb888acf2010-12-09 23:01:55 +00003469 if (AddedProperties.insert(P->getIdentifier()))
Craig Topperc3ec1492014-05-26 06:22:03 +00003470 Results.MaybeAddResult(Result(P, Results.getBasePriority(P), nullptr),
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003471 CurContext);
Craig Topperc3ec1492014-05-26 06:22:03 +00003472
Douglas Gregor95147142011-05-05 15:50:42 +00003473 // Add nullary methods
3474 if (AllowNullaryMethods) {
3475 ASTContext &Context = Container->getASTContext();
Douglas Gregor75acd922011-09-27 23:30:47 +00003476 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Aaron Ballmanaff18c02014-03-13 19:03:34 +00003477 for (auto *M : Container->methods()) {
Douglas Gregor95147142011-05-05 15:50:42 +00003478 if (M->getSelector().isUnarySelector())
3479 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3480 if (AddedProperties.insert(Name)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003481 CodeCompletionBuilder Builder(Results.getAllocator(),
3482 Results.getCodeCompletionTUInfo());
Aaron Ballmanaff18c02014-03-13 19:03:34 +00003483 AddResultTypeChunk(Context, Policy, M, Builder);
Douglas Gregor95147142011-05-05 15:50:42 +00003484 Builder.AddTypedTextChunk(
3485 Results.getAllocator().CopyString(Name->getName()));
3486
Aaron Ballmanaff18c02014-03-13 19:03:34 +00003487 Results.MaybeAddResult(Result(Builder.TakeString(), M,
Douglas Gregor78254c82012-03-27 23:34:16 +00003488 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor95147142011-05-05 15:50:42 +00003489 CurContext);
3490 }
3491 }
3492 }
3493
3494
Douglas Gregor9291bad2009-11-18 01:29:26 +00003495 // Add properties in referenced protocols.
3496 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Aaron Ballman0f6e64d2014-03-13 22:58:06 +00003497 for (auto *P : Protocol->protocols())
3498 AddObjCProperties(P, AllowCategories, AllowNullaryMethods, CurContext,
Douglas Gregor95147142011-05-05 15:50:42 +00003499 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003500 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor5d649882009-11-18 22:32:06 +00003501 if (AllowCategories) {
3502 // Look through categories.
Aaron Ballman15063e12014-03-13 21:35:02 +00003503 for (auto *Cat : IFace->known_categories())
3504 AddObjCProperties(Cat, AllowCategories, AllowNullaryMethods, CurContext,
3505 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00003506 }
Aaron Ballman15063e12014-03-13 21:35:02 +00003507
Douglas Gregor9291bad2009-11-18 01:29:26 +00003508 // Look through protocols.
Aaron Ballmana9f49e32014-03-13 20:55:22 +00003509 for (auto *I : IFace->all_referenced_protocols())
3510 AddObjCProperties(I, AllowCategories, AllowNullaryMethods, CurContext,
Douglas Gregor95147142011-05-05 15:50:42 +00003511 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003512
3513 // Look in the superclass.
3514 if (IFace->getSuperClass())
Douglas Gregor95147142011-05-05 15:50:42 +00003515 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3516 AllowNullaryMethods, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003517 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003518 } else if (const ObjCCategoryDecl *Category
3519 = dyn_cast<ObjCCategoryDecl>(Container)) {
3520 // Look through protocols.
Aaron Ballman19a41762014-03-14 12:55:57 +00003521 for (auto *P : Category->protocols())
3522 AddObjCProperties(P, AllowCategories, AllowNullaryMethods, CurContext,
Douglas Gregor95147142011-05-05 15:50:42 +00003523 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003524 }
3525}
3526
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003527void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor2436e712009-09-17 21:32:03 +00003528 SourceLocation OpLoc,
3529 bool IsArrow) {
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003530 if (!Base || !CodeCompleter)
Douglas Gregor2436e712009-09-17 21:32:03 +00003531 return;
3532
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003533 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3534 if (ConvertedBase.isInvalid())
3535 return;
3536 Base = ConvertedBase.get();
3537
John McCall276321a2010-08-25 06:19:51 +00003538 typedef CodeCompletionResult Result;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003539
Douglas Gregor2436e712009-09-17 21:32:03 +00003540 QualType BaseType = Base->getType();
Douglas Gregor3545ff42009-09-21 16:56:56 +00003541
3542 if (IsArrow) {
3543 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3544 BaseType = Ptr->getPointeeType();
3545 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003546 /*Do nothing*/ ;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003547 else
3548 return;
3549 }
3550
Douglas Gregor21325842011-07-07 16:03:39 +00003551 enum CodeCompletionContext::Kind contextKind;
3552
3553 if (IsArrow) {
3554 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3555 }
3556 else {
3557 if (BaseType->isObjCObjectPointerType() ||
3558 BaseType->isObjCObjectOrInterfaceType()) {
3559 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3560 }
3561 else {
3562 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3563 }
3564 }
3565
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003566 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003567 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00003568 CodeCompletionContext(contextKind,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003569 BaseType),
3570 &ResultBuilder::IsMember);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003571 Results.EnterNewScope();
3572 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003573 // Indicate that we are performing a member access, and the cv-qualifiers
3574 // for the base object type.
3575 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3576
Douglas Gregor9291bad2009-11-18 01:29:26 +00003577 // Access to a C/C++ class, struct, or union.
Douglas Gregor6ae4c522010-01-14 03:21:49 +00003578 Results.allowNestedNameSpecifiers();
Douglas Gregor09bbc652010-01-14 15:47:35 +00003579 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003580 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3581 CodeCompleter->includeGlobals());
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003582
David Blaikiebbafb8a2012-03-11 07:00:24 +00003583 if (getLangOpts().CPlusPlus) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003584 if (!Results.empty()) {
3585 // The "template" keyword can follow "->" or "." in the grammar.
3586 // However, we only want to suggest the template keyword if something
3587 // is dependent.
3588 bool IsDependent = BaseType->isDependentType();
3589 if (!IsDependent) {
3590 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
Ted Kremenekc37877d2013-10-08 17:08:03 +00003591 if (DeclContext *Ctx = DepScope->getEntity()) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003592 IsDependent = Ctx->isDependentContext();
3593 break;
3594 }
3595 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003596
Douglas Gregor9291bad2009-11-18 01:29:26 +00003597 if (IsDependent)
Douglas Gregor78a21012010-01-14 16:01:26 +00003598 Results.AddResult(Result("template"));
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003599 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003600 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003601 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3602 // Objective-C property reference.
Douglas Gregorb888acf2010-12-09 23:01:55 +00003603 AddedPropertiesSet AddedProperties;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003604
3605 // Add property results based on our interface.
3606 const ObjCObjectPointerType *ObjCPtr
3607 = BaseType->getAsObjCInterfacePointerType();
3608 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor95147142011-05-05 15:50:42 +00003609 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3610 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003611 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003612
3613 // Add properties from the protocols in a qualified interface.
Aaron Ballman83731462014-03-17 16:14:00 +00003614 for (auto *I : ObjCPtr->quals())
3615 AddObjCProperties(I, true, /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregor95147142011-05-05 15:50:42 +00003616 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003617 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCall8b07ec22010-05-15 11:32:37 +00003618 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003619 // Objective-C instance variable access.
Craig Topperc3ec1492014-05-26 06:22:03 +00003620 ObjCInterfaceDecl *Class = nullptr;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003621 if (const ObjCObjectPointerType *ObjCPtr
3622 = BaseType->getAs<ObjCObjectPointerType>())
3623 Class = ObjCPtr->getInterfaceDecl();
3624 else
John McCall8b07ec22010-05-15 11:32:37 +00003625 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003626
3627 // Add all ivars from this class and its superclasses.
Douglas Gregor2b8162b2010-01-14 16:08:12 +00003628 if (Class) {
3629 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3630 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor39982192010-08-15 06:18:01 +00003631 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3632 CodeCompleter->includeGlobals());
Douglas Gregor9291bad2009-11-18 01:29:26 +00003633 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003634 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003635
3636 // FIXME: How do we cope with isa?
3637
3638 Results.ExitScope();
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003639
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003640 // Hand off the results found for code completion.
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003641 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003642 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003643 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00003644}
3645
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003646void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3647 if (!CodeCompleter)
3648 return;
Craig Topperc3ec1492014-05-26 06:22:03 +00003649
3650 ResultBuilder::LookupFilter Filter = nullptr;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003651 enum CodeCompletionContext::Kind ContextKind
3652 = CodeCompletionContext::CCC_Other;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003653 switch ((DeclSpec::TST)TagSpec) {
3654 case DeclSpec::TST_enum:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003655 Filter = &ResultBuilder::IsEnum;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003656 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003657 break;
3658
3659 case DeclSpec::TST_union:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003660 Filter = &ResultBuilder::IsUnion;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003661 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003662 break;
3663
3664 case DeclSpec::TST_struct:
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003665 case DeclSpec::TST_class:
Joao Matosdc86f942012-08-31 18:45:21 +00003666 case DeclSpec::TST_interface:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003667 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003668 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003669 break;
3670
3671 default:
David Blaikie83d382b2011-09-23 05:06:16 +00003672 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003673 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003674
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003675 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3676 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00003677 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCalle87beb22010-04-23 18:46:30 +00003678
3679 // First pass: look for tags.
3680 Results.setFilter(Filter);
Douglas Gregor39982192010-08-15 06:18:01 +00003681 LookupVisibleDecls(S, LookupTagName, Consumer,
3682 CodeCompleter->includeGlobals());
John McCalle87beb22010-04-23 18:46:30 +00003683
Douglas Gregor39982192010-08-15 06:18:01 +00003684 if (CodeCompleter->includeGlobals()) {
3685 // Second pass: look for nested name specifiers.
3686 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3687 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3688 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003689
Douglas Gregor0ac41382010-09-23 23:01:17 +00003690 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003691 Results.data(),Results.size());
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003692}
3693
Douglas Gregor28c78432010-08-27 17:35:51 +00003694void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003695 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003696 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003697 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor28c78432010-08-27 17:35:51 +00003698 Results.EnterNewScope();
3699 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3700 Results.AddResult("const");
3701 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3702 Results.AddResult("volatile");
David Blaikiebbafb8a2012-03-11 07:00:24 +00003703 if (getLangOpts().C99 &&
Douglas Gregor28c78432010-08-27 17:35:51 +00003704 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3705 Results.AddResult("restrict");
Richard Smith8e1ac332013-03-28 01:55:44 +00003706 if (getLangOpts().C11 &&
3707 !(DS.getTypeQualifiers() & DeclSpec::TQ_atomic))
3708 Results.AddResult("_Atomic");
Douglas Gregor28c78432010-08-27 17:35:51 +00003709 Results.ExitScope();
3710 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003711 Results.getCompletionContext(),
Douglas Gregor28c78432010-08-27 17:35:51 +00003712 Results.data(), Results.size());
3713}
3714
Douglas Gregord328d572009-09-21 18:10:23 +00003715void Sema::CodeCompleteCase(Scope *S) {
John McCallaab3e412010-08-25 08:40:02 +00003716 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregord328d572009-09-21 18:10:23 +00003717 return;
John McCall5939b162011-08-06 07:30:58 +00003718
John McCallaab3e412010-08-25 08:40:02 +00003719 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCall5939b162011-08-06 07:30:58 +00003720 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3721 if (!type->isEnumeralType()) {
3722 CodeCompleteExpressionData Data(type);
Douglas Gregor68762e72010-08-23 21:17:50 +00003723 Data.IntegralConstantExpression = true;
3724 CodeCompleteExpression(S, Data);
Douglas Gregord328d572009-09-21 18:10:23 +00003725 return;
Douglas Gregor85b50632010-07-28 21:50:18 +00003726 }
Douglas Gregord328d572009-09-21 18:10:23 +00003727
3728 // Code-complete the cases of a switch statement over an enumeration type
3729 // by providing the list of
John McCall5939b162011-08-06 07:30:58 +00003730 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003731 if (EnumDecl *Def = Enum->getDefinition())
3732 Enum = Def;
Douglas Gregord328d572009-09-21 18:10:23 +00003733
3734 // Determine which enumerators we have already seen in the switch statement.
3735 // FIXME: Ideally, we would also be able to look *past* the code-completion
3736 // token, in case we are code-completing in the middle of the switch and not
3737 // at the end. However, we aren't able to do so at the moment.
3738 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Craig Topperc3ec1492014-05-26 06:22:03 +00003739 NestedNameSpecifier *Qualifier = nullptr;
Douglas Gregord328d572009-09-21 18:10:23 +00003740 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3741 SC = SC->getNextSwitchCase()) {
3742 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3743 if (!Case)
3744 continue;
3745
3746 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3747 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3748 if (EnumConstantDecl *Enumerator
3749 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3750 // We look into the AST of the case statement to determine which
3751 // enumerator was named. Alternatively, we could compute the value of
3752 // the integral constant expression, then compare it against the
3753 // values of each enumerator. However, value-based approach would not
3754 // work as well with C++ templates where enumerators declared within a
3755 // template are type- and value-dependent.
3756 EnumeratorsSeen.insert(Enumerator);
3757
Douglas Gregorf2510672009-09-21 19:57:38 +00003758 // If this is a qualified-id, keep track of the nested-name-specifier
3759 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregord328d572009-09-21 18:10:23 +00003760 //
3761 // switch (TagD.getKind()) {
3762 // case TagDecl::TK_enum:
3763 // break;
3764 // case XXX
3765 //
Douglas Gregorf2510672009-09-21 19:57:38 +00003766 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregord328d572009-09-21 18:10:23 +00003767 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3768 // TK_struct, and TK_class.
Douglas Gregor4bd90e52009-10-23 18:54:35 +00003769 Qualifier = DRE->getQualifier();
Douglas Gregord328d572009-09-21 18:10:23 +00003770 }
3771 }
3772
David Blaikiebbafb8a2012-03-11 07:00:24 +00003773 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorf2510672009-09-21 19:57:38 +00003774 // If there are no prior enumerators in C++, check whether we have to
3775 // qualify the names of the enumerators that we suggest, because they
3776 // may not be visible in this scope.
Douglas Gregord3cebdb2012-02-01 05:02:47 +00003777 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorf2510672009-09-21 19:57:38 +00003778 }
3779
Douglas Gregord328d572009-09-21 18:10:23 +00003780 // Add any enumerators that have not yet been mentioned.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003781 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003782 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003783 CodeCompletionContext::CCC_Expression);
Douglas Gregord328d572009-09-21 18:10:23 +00003784 Results.EnterNewScope();
Aaron Ballman23a6dcb2014-03-08 18:45:14 +00003785 for (auto *E : Enum->enumerators()) {
3786 if (EnumeratorsSeen.count(E))
Douglas Gregord328d572009-09-21 18:10:23 +00003787 continue;
3788
Aaron Ballman23a6dcb2014-03-08 18:45:14 +00003789 CodeCompletionResult R(E, CCP_EnumInCase, Qualifier);
Craig Topperc3ec1492014-05-26 06:22:03 +00003790 Results.AddResult(R, CurContext, nullptr, false);
Douglas Gregord328d572009-09-21 18:10:23 +00003791 }
3792 Results.ExitScope();
Douglas Gregor285560922010-04-06 20:02:15 +00003793
Douglas Gregor21325842011-07-07 16:03:39 +00003794 //We need to make sure we're setting the right context,
3795 //so only say we include macros if the code completer says we do
3796 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3797 if (CodeCompleter->includeMacros()) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00003798 AddMacroResults(PP, Results, false);
Douglas Gregor21325842011-07-07 16:03:39 +00003799 kind = CodeCompletionContext::CCC_OtherWithMacros;
3800 }
3801
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003802 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00003803 kind,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003804 Results.data(),Results.size());
Douglas Gregord328d572009-09-21 18:10:23 +00003805}
3806
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003807static bool anyNullArguments(ArrayRef<Expr *> Args) {
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003808 if (Args.size() && !Args.data())
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003809 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003810
3811 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003812 if (!Args[I])
3813 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003814
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003815 return false;
3816}
3817
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003818void Sema::CodeCompleteCall(Scope *S, Expr *FnIn, ArrayRef<Expr *> Args) {
Douglas Gregorcabea402009-09-22 15:41:20 +00003819 if (!CodeCompleter)
3820 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003821
3822 // When we're code-completing for a call, we fall back to ordinary
3823 // name code-completion whenever we can't produce specific
3824 // results. We may want to revisit this strategy in the future,
3825 // e.g., by merging the two kinds of results.
3826
Douglas Gregorcabea402009-09-22 15:41:20 +00003827 Expr *Fn = (Expr *)FnIn;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003828
Douglas Gregorcabea402009-09-22 15:41:20 +00003829 // Ignore type-dependent call expressions entirely.
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003830 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3831 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003832 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregorcabea402009-09-22 15:41:20 +00003833 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003834 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003835
John McCall57500772009-12-16 12:17:52 +00003836 // Build an overload candidate set based on the functions we find.
John McCallbc077cf2010-02-08 23:07:23 +00003837 SourceLocation Loc = Fn->getExprLoc();
Richard Smith100b24a2014-04-17 01:52:14 +00003838 OverloadCandidateSet CandidateSet(Loc, OverloadCandidateSet::CSK_Normal);
John McCall57500772009-12-16 12:17:52 +00003839
Douglas Gregorcabea402009-09-22 15:41:20 +00003840 // FIXME: What if we're calling something that isn't a function declaration?
3841 // FIXME: What if we're calling a pseudo-destructor?
3842 // FIXME: What if we're calling a member function?
3843
Douglas Gregorff59f672010-01-21 15:46:19 +00003844 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003845 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorff59f672010-01-21 15:46:19 +00003846
John McCall57500772009-12-16 12:17:52 +00003847 Expr *NakedFn = Fn->IgnoreParenCasts();
3848 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003849 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
John McCall57500772009-12-16 12:17:52 +00003850 /*PartialOverloading=*/ true);
3851 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3852 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorff59f672010-01-21 15:46:19 +00003853 if (FDecl) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00003854 if (!getLangOpts().CPlusPlus ||
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003855 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorff59f672010-01-21 15:46:19 +00003856 Results.push_back(ResultCandidate(FDecl));
3857 else
John McCallb89836b2010-01-26 01:37:31 +00003858 // FIXME: access?
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003859 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), Args,
3860 CandidateSet, false, /*PartialOverloading*/true);
Douglas Gregorff59f672010-01-21 15:46:19 +00003861 }
John McCall57500772009-12-16 12:17:52 +00003862 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003863
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003864 QualType ParamType;
3865
Douglas Gregorff59f672010-01-21 15:46:19 +00003866 if (!CandidateSet.empty()) {
3867 // Sort the overload candidate set by placing the best overloads first.
Benjamin Kramerbbdd7642014-03-01 14:48:57 +00003868 std::stable_sort(
3869 CandidateSet.begin(), CandidateSet.end(),
3870 [&](const OverloadCandidate &X, const OverloadCandidate &Y) {
3871 return isBetterOverloadCandidate(*this, X, Y, Loc);
3872 });
3873
Douglas Gregorff59f672010-01-21 15:46:19 +00003874 // Add the remaining viable overload candidates as code-completion reslults.
3875 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3876 CandEnd = CandidateSet.end();
3877 Cand != CandEnd; ++Cand) {
3878 if (Cand->Viable)
3879 Results.push_back(ResultCandidate(Cand->Function));
3880 }
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003881
3882 // From the viable candidates, try to determine the type of this parameter.
3883 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3884 if (const FunctionType *FType = Results[I].getFunctionType())
3885 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
Alp Toker9cacbab2014-01-20 20:26:09 +00003886 if (Args.size() < Proto->getNumParams()) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003887 if (ParamType.isNull())
Alp Toker9cacbab2014-01-20 20:26:09 +00003888 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003889 else if (!Context.hasSameUnqualifiedType(
Alp Toker9cacbab2014-01-20 20:26:09 +00003890 ParamType.getNonReferenceType(),
3891 Proto->getParamType(Args.size())
3892 .getNonReferenceType())) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003893 ParamType = QualType();
3894 break;
3895 }
3896 }
3897 }
3898 } else {
3899 // Try to determine the parameter type from the type of the expression
3900 // being called.
3901 QualType FunctionType = Fn->getType();
3902 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3903 FunctionType = Ptr->getPointeeType();
3904 else if (const BlockPointerType *BlockPtr
3905 = FunctionType->getAs<BlockPointerType>())
3906 FunctionType = BlockPtr->getPointeeType();
3907 else if (const MemberPointerType *MemPtr
3908 = FunctionType->getAs<MemberPointerType>())
3909 FunctionType = MemPtr->getPointeeType();
3910
3911 if (const FunctionProtoType *Proto
3912 = FunctionType->getAs<FunctionProtoType>()) {
Alp Toker9cacbab2014-01-20 20:26:09 +00003913 if (Args.size() < Proto->getNumParams())
3914 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003915 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003916 }
Douglas Gregor3ef59522009-12-11 19:06:04 +00003917
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003918 if (ParamType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003919 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003920 else
3921 CodeCompleteExpression(S, ParamType);
3922
Douglas Gregorc01890e2010-04-06 20:19:47 +00003923 if (!Results.empty())
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003924 CodeCompleter->ProcessOverloadCandidates(*this, Args.size(), Results.data(),
Douglas Gregor3ef59522009-12-11 19:06:04 +00003925 Results.size());
Douglas Gregorcabea402009-09-22 15:41:20 +00003926}
3927
John McCall48871652010-08-21 09:40:31 +00003928void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3929 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003930 if (!VD) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003931 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003932 return;
3933 }
3934
3935 CodeCompleteExpression(S, VD->getType());
3936}
3937
3938void Sema::CodeCompleteReturn(Scope *S) {
3939 QualType ResultType;
3940 if (isa<BlockDecl>(CurContext)) {
3941 if (BlockScopeInfo *BSI = getCurBlock())
3942 ResultType = BSI->ReturnType;
3943 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00003944 ResultType = Function->getReturnType();
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003945 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00003946 ResultType = Method->getReturnType();
3947
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003948 if (ResultType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003949 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003950 else
3951 CodeCompleteExpression(S, ResultType);
3952}
3953
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003954void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003955 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003956 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003957 mapCodeCompletionContext(*this, PCC_Statement));
3958 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3959 Results.EnterNewScope();
3960
3961 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3962 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3963 CodeCompleter->includeGlobals());
3964
3965 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
3966
3967 // "else" block
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003968 CodeCompletionBuilder Builder(Results.getAllocator(),
3969 Results.getCodeCompletionTUInfo());
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003970 Builder.AddTypedTextChunk("else");
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00003971 if (Results.includeCodePatterns()) {
3972 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3973 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
3974 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3975 Builder.AddPlaceholderChunk("statements");
3976 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3977 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
3978 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003979 Results.AddResult(Builder.TakeString());
3980
3981 // "else if" block
3982 Builder.AddTypedTextChunk("else");
3983 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3984 Builder.AddTextChunk("if");
3985 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3986 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003987 if (getLangOpts().CPlusPlus)
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003988 Builder.AddPlaceholderChunk("condition");
3989 else
3990 Builder.AddPlaceholderChunk("expression");
3991 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00003992 if (Results.includeCodePatterns()) {
3993 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3994 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
3995 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3996 Builder.AddPlaceholderChunk("statements");
3997 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3998 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
3999 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004000 Results.AddResult(Builder.TakeString());
4001
4002 Results.ExitScope();
4003
4004 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004005 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004006
4007 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00004008 AddMacroResults(PP, Results, false);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004009
4010 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4011 Results.data(),Results.size());
4012}
4013
Richard Trieu2bd04012011-09-09 02:00:50 +00004014void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004015 if (LHS)
4016 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4017 else
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004018 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004019}
4020
Jeffrey Yasskinc76498d2010-04-08 16:38:48 +00004021void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor2436e712009-09-17 21:32:03 +00004022 bool EnteringContext) {
4023 if (!SS.getScopeRep() || !CodeCompleter)
4024 return;
4025
Douglas Gregor3545ff42009-09-21 16:56:56 +00004026 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4027 if (!Ctx)
4028 return;
Douglas Gregor800f2f02009-12-11 18:28:39 +00004029
4030 // Try to instantiate any non-dependent declaration contexts before
4031 // we look in them.
John McCall0b66eb32010-05-01 00:40:08 +00004032 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregor800f2f02009-12-11 18:28:39 +00004033 return;
4034
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004035 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004036 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004037 CodeCompletionContext::CCC_Name);
Douglas Gregorac322ec2010-08-27 21:18:54 +00004038 Results.EnterNewScope();
Douglas Gregor0ac41382010-09-23 23:01:17 +00004039
Douglas Gregor3545ff42009-09-21 16:56:56 +00004040 // The "template" keyword can follow "::" in the grammar, but only
4041 // put it into the grammar if the nested-name-specifier is dependent.
Aaron Ballman4a979672014-01-03 13:56:08 +00004042 NestedNameSpecifier *NNS = SS.getScopeRep();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004043 if (!Results.empty() && NNS->isDependent())
Douglas Gregor78a21012010-01-14 16:01:26 +00004044 Results.AddResult("template");
Douglas Gregorac322ec2010-08-27 21:18:54 +00004045
4046 // Add calls to overridden virtual functions, if there are any.
4047 //
4048 // FIXME: This isn't wonderful, because we don't know whether we're actually
4049 // in a context that permits expressions. This is a general issue with
4050 // qualified-id completions.
4051 if (!EnteringContext)
4052 MaybeAddOverrideCalls(*this, Ctx, Results);
4053 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004054
Douglas Gregorac322ec2010-08-27 21:18:54 +00004055 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4056 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4057
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004058 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorc1679ec2011-07-25 17:48:11 +00004059 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004060 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00004061}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004062
4063void Sema::CodeCompleteUsing(Scope *S) {
4064 if (!CodeCompleter)
4065 return;
4066
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004067 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004068 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004069 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4070 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004071 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004072
4073 // If we aren't in class scope, we could see the "namespace" keyword.
4074 if (!S->isClassScope())
John McCall276321a2010-08-25 06:19:51 +00004075 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004076
4077 // After "using", we can see anything that would start a
4078 // nested-name-specifier.
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004079 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004080 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4081 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004082 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004083
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004084 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004085 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004086 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004087}
4088
4089void Sema::CodeCompleteUsingDirective(Scope *S) {
4090 if (!CodeCompleter)
4091 return;
4092
Douglas Gregor3545ff42009-09-21 16:56:56 +00004093 // After "using namespace", we expect to see a namespace name or namespace
4094 // alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004095 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004096 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004097 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004098 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004099 Results.EnterNewScope();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004100 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004101 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4102 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004103 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004104 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004105 CodeCompletionContext::CCC_Namespace,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004106 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004107}
4108
4109void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4110 if (!CodeCompleter)
4111 return;
4112
Ted Kremenekc37877d2013-10-08 17:08:03 +00004113 DeclContext *Ctx = S->getEntity();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004114 if (!S->getParent())
4115 Ctx = Context.getTranslationUnitDecl();
4116
Douglas Gregor0ac41382010-09-23 23:01:17 +00004117 bool SuppressedGlobalResults
4118 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4119
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004120 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004121 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004122 SuppressedGlobalResults
4123 ? CodeCompletionContext::CCC_Namespace
4124 : CodeCompletionContext::CCC_Other,
4125 &ResultBuilder::IsNamespace);
4126
4127 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00004128 // We only want to see those namespaces that have already been defined
4129 // within this scope, because its likely that the user is creating an
4130 // extended namespace declaration. Keep track of the most recent
4131 // definition of each namespace.
4132 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4133 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4134 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4135 NS != NSEnd; ++NS)
David Blaikie40ed2972012-06-06 20:45:41 +00004136 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004137
4138 // Add the most recent definition (or extended definition) of each
4139 // namespace to the list of results.
Douglas Gregor64b12b52009-09-22 23:31:26 +00004140 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004141 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregor78254c82012-03-27 23:34:16 +00004142 NS = OrigToLatest.begin(),
4143 NSEnd = OrigToLatest.end();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004144 NS != NSEnd; ++NS)
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004145 Results.AddResult(CodeCompletionResult(
Craig Topperc3ec1492014-05-26 06:22:03 +00004146 NS->second, Results.getBasePriority(NS->second),
4147 nullptr),
4148 CurContext, nullptr, false);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004149 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004150 }
4151
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004152 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004153 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004154 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004155}
4156
4157void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4158 if (!CodeCompleter)
4159 return;
4160
Douglas Gregor3545ff42009-09-21 16:56:56 +00004161 // After "namespace", we expect to see a namespace or alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004162 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004163 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004164 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004165 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004166 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004167 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4168 CodeCompleter->includeGlobals());
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004169 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004170 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004171 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004172}
4173
Douglas Gregorc811ede2009-09-18 20:05:18 +00004174void Sema::CodeCompleteOperatorName(Scope *S) {
4175 if (!CodeCompleter)
4176 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004177
John McCall276321a2010-08-25 06:19:51 +00004178 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004179 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004180 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004181 CodeCompletionContext::CCC_Type,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004182 &ResultBuilder::IsType);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004183 Results.EnterNewScope();
Douglas Gregorc811ede2009-09-18 20:05:18 +00004184
Douglas Gregor3545ff42009-09-21 16:56:56 +00004185 // Add the names of overloadable operators.
4186#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4187 if (std::strcmp(Spelling, "?")) \
Douglas Gregor78a21012010-01-14 16:01:26 +00004188 Results.AddResult(Result(Spelling));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004189#include "clang/Basic/OperatorKinds.def"
4190
4191 // Add any type names visible from the current scope
Douglas Gregor6ae4c522010-01-14 03:21:49 +00004192 Results.allowNestedNameSpecifiers();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004193 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004194 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4195 CodeCompleter->includeGlobals());
Douglas Gregor3545ff42009-09-21 16:56:56 +00004196
4197 // Add any type specifiers
David Blaikiebbafb8a2012-03-11 07:00:24 +00004198 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004199 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004200
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004201 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004202 CodeCompletionContext::CCC_Type,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004203 Results.data(),Results.size());
Douglas Gregorc811ede2009-09-18 20:05:18 +00004204}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004205
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004206void Sema::CodeCompleteConstructorInitializer(
4207 Decl *ConstructorD,
4208 ArrayRef <CXXCtorInitializer *> Initializers) {
Douglas Gregor75acd922011-09-27 23:30:47 +00004209 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004210 CXXConstructorDecl *Constructor
4211 = static_cast<CXXConstructorDecl *>(ConstructorD);
4212 if (!Constructor)
4213 return;
4214
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004215 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004216 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004217 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004218 Results.EnterNewScope();
4219
4220 // Fill in any already-initialized fields or base classes.
4221 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4222 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004223 for (unsigned I = 0, E = Initializers.size(); I != E; ++I) {
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004224 if (Initializers[I]->isBaseInitializer())
4225 InitializedBases.insert(
4226 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4227 else
Francois Pichetd583da02010-12-04 09:14:42 +00004228 InitializedFields.insert(cast<FieldDecl>(
4229 Initializers[I]->getAnyMember()));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004230 }
4231
4232 // Add completions for base classes.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004233 CodeCompletionBuilder Builder(Results.getAllocator(),
4234 Results.getCodeCompletionTUInfo());
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004235 bool SawLastInitializer = Initializers.empty();
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004236 CXXRecordDecl *ClassDecl = Constructor->getParent();
Aaron Ballman574705e2014-03-13 15:41:46 +00004237 for (const auto &Base : ClassDecl->bases()) {
4238 if (!InitializedBases.insert(Context.getCanonicalType(Base.getType()))) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004239 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004240 = !Initializers.empty() &&
4241 Initializers.back()->isBaseInitializer() &&
Aaron Ballman574705e2014-03-13 15:41:46 +00004242 Context.hasSameUnqualifiedType(Base.getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004243 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004244 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004245 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004246
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004247 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004248 Results.getAllocator().CopyString(
Aaron Ballman574705e2014-03-13 15:41:46 +00004249 Base.getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004250 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4251 Builder.AddPlaceholderChunk("args");
4252 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4253 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004254 SawLastInitializer? CCP_NextInitializer
4255 : CCP_MemberDeclaration));
4256 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004257 }
4258
4259 // Add completions for virtual base classes.
Aaron Ballman445a9392014-03-13 16:15:17 +00004260 for (const auto &Base : ClassDecl->vbases()) {
4261 if (!InitializedBases.insert(Context.getCanonicalType(Base.getType()))) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004262 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004263 = !Initializers.empty() &&
4264 Initializers.back()->isBaseInitializer() &&
Aaron Ballman445a9392014-03-13 16:15:17 +00004265 Context.hasSameUnqualifiedType(Base.getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004266 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004267 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004268 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004269
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004270 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004271 Builder.getAllocator().CopyString(
Aaron Ballman445a9392014-03-13 16:15:17 +00004272 Base.getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004273 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4274 Builder.AddPlaceholderChunk("args");
4275 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4276 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004277 SawLastInitializer? CCP_NextInitializer
4278 : CCP_MemberDeclaration));
4279 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004280 }
4281
4282 // Add completions for members.
Aaron Ballmane8a8bae2014-03-08 20:12:42 +00004283 for (auto *Field : ClassDecl->fields()) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004284 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4285 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004286 = !Initializers.empty() &&
4287 Initializers.back()->isAnyMemberInitializer() &&
Aaron Ballmane8a8bae2014-03-08 20:12:42 +00004288 Initializers.back()->getAnyMember() == Field;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004289 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004290 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004291
4292 if (!Field->getDeclName())
4293 continue;
4294
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004295 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004296 Field->getIdentifier()->getName()));
4297 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4298 Builder.AddPlaceholderChunk("args");
4299 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4300 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004301 SawLastInitializer? CCP_NextInitializer
Douglas Gregorf3af3112010-09-09 21:42:20 +00004302 : CCP_MemberDeclaration,
Douglas Gregor78254c82012-03-27 23:34:16 +00004303 CXCursor_MemberRef,
4304 CXAvailability_Available,
Aaron Ballmane8a8bae2014-03-08 20:12:42 +00004305 Field));
Douglas Gregor99129ef2010-08-29 19:27:27 +00004306 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004307 }
4308 Results.ExitScope();
4309
Douglas Gregor0ac41382010-09-23 23:01:17 +00004310 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004311 Results.data(), Results.size());
4312}
4313
Douglas Gregord8c61782012-02-15 15:34:24 +00004314/// \brief Determine whether this scope denotes a namespace.
4315static bool isNamespaceScope(Scope *S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00004316 DeclContext *DC = S->getEntity();
Douglas Gregord8c61782012-02-15 15:34:24 +00004317 if (!DC)
4318 return false;
4319
4320 return DC->isFileContext();
4321}
4322
4323void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4324 bool AfterAmpersand) {
4325 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004326 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord8c61782012-02-15 15:34:24 +00004327 CodeCompletionContext::CCC_Other);
4328 Results.EnterNewScope();
4329
4330 // Note what has already been captured.
4331 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4332 bool IncludedThis = false;
Benjamin Kramerf3ca26982014-05-10 16:31:55 +00004333 for (const auto &C : Intro.Captures) {
4334 if (C.Kind == LCK_This) {
Douglas Gregord8c61782012-02-15 15:34:24 +00004335 IncludedThis = true;
4336 continue;
4337 }
4338
Benjamin Kramerf3ca26982014-05-10 16:31:55 +00004339 Known.insert(C.Id);
Douglas Gregord8c61782012-02-15 15:34:24 +00004340 }
4341
4342 // Look for other capturable variables.
4343 for (; S && !isNamespaceScope(S); S = S->getParent()) {
Aaron Ballman35c54952014-03-17 16:55:25 +00004344 for (const auto *D : S->decls()) {
4345 const auto *Var = dyn_cast<VarDecl>(D);
Douglas Gregord8c61782012-02-15 15:34:24 +00004346 if (!Var ||
4347 !Var->hasLocalStorage() ||
4348 Var->hasAttr<BlocksAttr>())
4349 continue;
4350
4351 if (Known.insert(Var->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004352 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
Craig Topperc3ec1492014-05-26 06:22:03 +00004353 CurContext, nullptr, false);
Douglas Gregord8c61782012-02-15 15:34:24 +00004354 }
4355 }
4356
4357 // Add 'this', if it would be valid.
4358 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4359 addThisCompletion(*this, Results);
4360
4361 Results.ExitScope();
4362
4363 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4364 Results.data(), Results.size());
4365}
4366
James Dennett596e4752012-06-14 03:11:41 +00004367/// Macro that optionally prepends an "@" to the string literal passed in via
4368/// Keyword, depending on whether NeedAt is true or false.
4369#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4370
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004371static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004372 ResultBuilder &Results,
4373 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004374 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004375 // Since we have an implementation, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004376 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004377
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004378 CodeCompletionBuilder Builder(Results.getAllocator(),
4379 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004380 if (LangOpts.ObjC2) {
4381 // @dynamic
James Dennett596e4752012-06-14 03:11:41 +00004382 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004383 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4384 Builder.AddPlaceholderChunk("property");
4385 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004386
4387 // @synthesize
James Dennett596e4752012-06-14 03:11:41 +00004388 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004389 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4390 Builder.AddPlaceholderChunk("property");
4391 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004392 }
4393}
4394
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004395static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004396 ResultBuilder &Results,
4397 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004398 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004399
4400 // Since we have an interface or protocol, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004401 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004402
4403 if (LangOpts.ObjC2) {
4404 // @property
James Dennett596e4752012-06-14 03:11:41 +00004405 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004406
4407 // @required
James Dennett596e4752012-06-14 03:11:41 +00004408 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004409
4410 // @optional
James Dennett596e4752012-06-14 03:11:41 +00004411 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004412 }
4413}
4414
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004415static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004416 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004417 CodeCompletionBuilder Builder(Results.getAllocator(),
4418 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004419
4420 // @class name ;
James Dennett596e4752012-06-14 03:11:41 +00004421 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004422 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4423 Builder.AddPlaceholderChunk("name");
4424 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004425
Douglas Gregorf4c33342010-05-28 00:22:41 +00004426 if (Results.includeCodePatterns()) {
4427 // @interface name
4428 // FIXME: Could introduce the whole pattern, including superclasses and
4429 // such.
James Dennett596e4752012-06-14 03:11:41 +00004430 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004431 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4432 Builder.AddPlaceholderChunk("class");
4433 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004434
Douglas Gregorf4c33342010-05-28 00:22:41 +00004435 // @protocol name
James Dennett596e4752012-06-14 03:11:41 +00004436 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004437 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4438 Builder.AddPlaceholderChunk("protocol");
4439 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004440
4441 // @implementation name
James Dennett596e4752012-06-14 03:11:41 +00004442 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004443 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4444 Builder.AddPlaceholderChunk("class");
4445 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004446 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004447
4448 // @compatibility_alias name
James Dennett596e4752012-06-14 03:11:41 +00004449 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004450 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4451 Builder.AddPlaceholderChunk("alias");
4452 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4453 Builder.AddPlaceholderChunk("class");
4454 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor61e36812013-03-07 23:26:24 +00004455
4456 if (Results.getSema().getLangOpts().Modules) {
4457 // @import name
4458 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "import"));
4459 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4460 Builder.AddPlaceholderChunk("module");
4461 Results.AddResult(Result(Builder.TakeString()));
4462 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004463}
4464
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004465void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004466 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004467 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004468 CodeCompletionContext::CCC_Other);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004469 Results.EnterNewScope();
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004470 if (isa<ObjCImplDecl>(CurContext))
David Blaikiebbafb8a2012-03-11 07:00:24 +00004471 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004472 else if (CurContext->isObjCContainer())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004473 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorf1934162010-01-13 21:24:21 +00004474 else
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004475 AddObjCTopLevelResults(Results, false);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004476 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004477 HandleCodeCompleteResults(this, CodeCompleter,
4478 CodeCompletionContext::CCC_Other,
4479 Results.data(),Results.size());
Douglas Gregorf48706c2009-12-07 09:27:33 +00004480}
4481
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004482static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004483 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004484 CodeCompletionBuilder Builder(Results.getAllocator(),
4485 Results.getCodeCompletionTUInfo());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004486
4487 // @encode ( type-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004488 const char *EncodeType = "char[]";
David Blaikiebbafb8a2012-03-11 07:00:24 +00004489 if (Results.getSema().getLangOpts().CPlusPlus ||
4490 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose9da05852012-06-15 18:19:56 +00004491 EncodeType = "const char[]";
Douglas Gregore5c79d52011-10-18 21:20:17 +00004492 Builder.AddResultTypeChunk(EncodeType);
James Dennett596e4752012-06-14 03:11:41 +00004493 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004494 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4495 Builder.AddPlaceholderChunk("type-name");
4496 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4497 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004498
4499 // @protocol ( protocol-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004500 Builder.AddResultTypeChunk("Protocol *");
James Dennett596e4752012-06-14 03:11:41 +00004501 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004502 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4503 Builder.AddPlaceholderChunk("protocol-name");
4504 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4505 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004506
4507 // @selector ( selector )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004508 Builder.AddResultTypeChunk("SEL");
James Dennett596e4752012-06-14 03:11:41 +00004509 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004510 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4511 Builder.AddPlaceholderChunk("selector");
4512 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4513 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004514
4515 // @"string"
4516 Builder.AddResultTypeChunk("NSString *");
4517 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4518 Builder.AddPlaceholderChunk("string");
4519 Builder.AddTextChunk("\"");
4520 Results.AddResult(Result(Builder.TakeString()));
4521
Douglas Gregor951de302012-07-17 23:24:47 +00004522 // @[objects, ...]
Jordan Rose9da05852012-06-15 18:19:56 +00004523 Builder.AddResultTypeChunk("NSArray *");
James Dennett596e4752012-06-14 03:11:41 +00004524 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004525 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004526 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4527 Results.AddResult(Result(Builder.TakeString()));
4528
Douglas Gregor951de302012-07-17 23:24:47 +00004529 // @{key : object, ...}
Jordan Rose9da05852012-06-15 18:19:56 +00004530 Builder.AddResultTypeChunk("NSDictionary *");
James Dennett596e4752012-06-14 03:11:41 +00004531 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"{"));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004532 Builder.AddPlaceholderChunk("key");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004533 Builder.AddChunk(CodeCompletionString::CK_Colon);
4534 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4535 Builder.AddPlaceholderChunk("object, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004536 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4537 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004538
Douglas Gregor951de302012-07-17 23:24:47 +00004539 // @(expression)
Jordan Rose9da05852012-06-15 18:19:56 +00004540 Builder.AddResultTypeChunk("id");
4541 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose9da05852012-06-15 18:19:56 +00004542 Builder.AddPlaceholderChunk("expression");
Jordan Rose9da05852012-06-15 18:19:56 +00004543 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4544 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004545}
4546
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004547static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004548 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004549 CodeCompletionBuilder Builder(Results.getAllocator(),
4550 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004551
Douglas Gregorf4c33342010-05-28 00:22:41 +00004552 if (Results.includeCodePatterns()) {
4553 // @try { statements } @catch ( declaration ) { statements } @finally
4554 // { statements }
James Dennett596e4752012-06-14 03:11:41 +00004555 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004556 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4557 Builder.AddPlaceholderChunk("statements");
4558 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4559 Builder.AddTextChunk("@catch");
4560 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4561 Builder.AddPlaceholderChunk("parameter");
4562 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4563 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4564 Builder.AddPlaceholderChunk("statements");
4565 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4566 Builder.AddTextChunk("@finally");
4567 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4568 Builder.AddPlaceholderChunk("statements");
4569 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4570 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004571 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004572
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004573 // @throw
James Dennett596e4752012-06-14 03:11:41 +00004574 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004575 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4576 Builder.AddPlaceholderChunk("expression");
4577 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004578
Douglas Gregorf4c33342010-05-28 00:22:41 +00004579 if (Results.includeCodePatterns()) {
4580 // @synchronized ( expression ) { statements }
James Dennett596e4752012-06-14 03:11:41 +00004581 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004582 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4583 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4584 Builder.AddPlaceholderChunk("expression");
4585 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4586 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4587 Builder.AddPlaceholderChunk("statements");
4588 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4589 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004590 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004591}
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004592
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004593static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00004594 ResultBuilder &Results,
4595 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004596 typedef CodeCompletionResult Result;
James Dennett596e4752012-06-14 03:11:41 +00004597 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4598 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4599 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004600 if (LangOpts.ObjC2)
James Dennett596e4752012-06-14 03:11:41 +00004601 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004602}
4603
4604void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004605 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004606 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004607 CodeCompletionContext::CCC_Other);
Douglas Gregor48d46252010-01-13 21:54:15 +00004608 Results.EnterNewScope();
David Blaikiebbafb8a2012-03-11 07:00:24 +00004609 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregor48d46252010-01-13 21:54:15 +00004610 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004611 HandleCodeCompleteResults(this, CodeCompleter,
4612 CodeCompletionContext::CCC_Other,
4613 Results.data(),Results.size());
Douglas Gregor48d46252010-01-13 21:54:15 +00004614}
4615
4616void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004617 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004618 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004619 CodeCompletionContext::CCC_Other);
Douglas Gregorf1934162010-01-13 21:24:21 +00004620 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004621 AddObjCStatementResults(Results, false);
4622 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004623 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004624 HandleCodeCompleteResults(this, CodeCompleter,
4625 CodeCompletionContext::CCC_Other,
4626 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004627}
4628
4629void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004630 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004631 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004632 CodeCompletionContext::CCC_Other);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004633 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004634 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004635 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004636 HandleCodeCompleteResults(this, CodeCompleter,
4637 CodeCompletionContext::CCC_Other,
4638 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004639}
4640
Douglas Gregore6078da2009-11-19 00:14:45 +00004641/// \brief Determine whether the addition of the given flag to an Objective-C
4642/// property's attributes will cause a conflict.
Bill Wendling44426052012-12-20 19:22:21 +00004643static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregore6078da2009-11-19 00:14:45 +00004644 // Check if we've already added this flag.
Bill Wendling44426052012-12-20 19:22:21 +00004645 if (Attributes & NewFlag)
Douglas Gregore6078da2009-11-19 00:14:45 +00004646 return true;
4647
Bill Wendling44426052012-12-20 19:22:21 +00004648 Attributes |= NewFlag;
Douglas Gregore6078da2009-11-19 00:14:45 +00004649
4650 // Check for collisions with "readonly".
Bill Wendling44426052012-12-20 19:22:21 +00004651 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4652 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregore6078da2009-11-19 00:14:45 +00004653 return true;
4654
Jordan Rose53cb2f32012-08-20 20:01:13 +00004655 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendling44426052012-12-20 19:22:21 +00004656 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCall31168b02011-06-15 23:02:42 +00004657 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregore6078da2009-11-19 00:14:45 +00004658 ObjCDeclSpec::DQ_PR_copy |
Jordan Rose53cb2f32012-08-20 20:01:13 +00004659 ObjCDeclSpec::DQ_PR_retain |
4660 ObjCDeclSpec::DQ_PR_strong |
4661 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregore6078da2009-11-19 00:14:45 +00004662 if (AssignCopyRetMask &&
4663 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCall31168b02011-06-15 23:02:42 +00004664 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregore6078da2009-11-19 00:14:45 +00004665 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCall31168b02011-06-15 23:02:42 +00004666 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rose53cb2f32012-08-20 20:01:13 +00004667 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4668 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregore6078da2009-11-19 00:14:45 +00004669 return true;
4670
4671 return false;
4672}
4673
Douglas Gregor36029f42009-11-18 23:08:07 +00004674void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroff936354c2009-10-08 21:55:05 +00004675 if (!CodeCompleter)
4676 return;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004677
Bill Wendling44426052012-12-20 19:22:21 +00004678 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroff936354c2009-10-08 21:55:05 +00004679
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004680 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004681 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004682 CodeCompletionContext::CCC_Other);
Steve Naroff936354c2009-10-08 21:55:05 +00004683 Results.EnterNewScope();
Bill Wendling44426052012-12-20 19:22:21 +00004684 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall276321a2010-08-25 06:19:51 +00004685 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendling44426052012-12-20 19:22:21 +00004686 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall276321a2010-08-25 06:19:51 +00004687 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendling44426052012-12-20 19:22:21 +00004688 if (!ObjCPropertyFlagConflicts(Attributes,
John McCall31168b02011-06-15 23:02:42 +00004689 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4690 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendling44426052012-12-20 19:22:21 +00004691 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall276321a2010-08-25 06:19:51 +00004692 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendling44426052012-12-20 19:22:21 +00004693 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall276321a2010-08-25 06:19:51 +00004694 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendling44426052012-12-20 19:22:21 +00004695 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCall31168b02011-06-15 23:02:42 +00004696 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendling44426052012-12-20 19:22:21 +00004697 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall276321a2010-08-25 06:19:51 +00004698 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendling44426052012-12-20 19:22:21 +00004699 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall276321a2010-08-25 06:19:51 +00004700 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendling44426052012-12-20 19:22:21 +00004701 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian1c2d29e2011-06-11 17:14:27 +00004702 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rose53cb2f32012-08-20 20:01:13 +00004703
4704 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall3deb1ad2012-08-21 02:47:43 +00004705 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendling44426052012-12-20 19:22:21 +00004706 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rose53cb2f32012-08-20 20:01:13 +00004707 Results.AddResult(CodeCompletionResult("weak"));
4708
Bill Wendling44426052012-12-20 19:22:21 +00004709 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004710 CodeCompletionBuilder Setter(Results.getAllocator(),
4711 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004712 Setter.AddTypedTextChunk("setter");
Argyrios Kyrtzidis7bbb8812014-02-20 07:55:15 +00004713 Setter.AddTextChunk("=");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004714 Setter.AddPlaceholderChunk("method");
4715 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004716 }
Bill Wendling44426052012-12-20 19:22:21 +00004717 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004718 CodeCompletionBuilder Getter(Results.getAllocator(),
4719 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004720 Getter.AddTypedTextChunk("getter");
Argyrios Kyrtzidis7bbb8812014-02-20 07:55:15 +00004721 Getter.AddTextChunk("=");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004722 Getter.AddPlaceholderChunk("method");
4723 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004724 }
Steve Naroff936354c2009-10-08 21:55:05 +00004725 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004726 HandleCodeCompleteResults(this, CodeCompleter,
4727 CodeCompletionContext::CCC_Other,
4728 Results.data(),Results.size());
Steve Naroff936354c2009-10-08 21:55:05 +00004729}
Steve Naroffeae65032009-11-07 02:08:14 +00004730
James Dennettf1243872012-06-17 05:33:25 +00004731/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregorc8537c52009-11-19 07:41:15 +00004732/// via code completion.
4733enum ObjCMethodKind {
Dmitri Gribenko4280e5c2012-06-08 23:13:42 +00004734 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4735 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4736 MK_OneArgSelector ///< One-argument selector.
Douglas Gregorc8537c52009-11-19 07:41:15 +00004737};
4738
Douglas Gregor67c692c2010-08-26 15:07:07 +00004739static bool isAcceptableObjCSelector(Selector Sel,
4740 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004741 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004742 bool AllowSameLength = true) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004743 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor67c692c2010-08-26 15:07:07 +00004744 if (NumSelIdents > Sel.getNumArgs())
4745 return false;
4746
4747 switch (WantKind) {
4748 case MK_Any: break;
4749 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4750 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4751 }
4752
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004753 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4754 return false;
4755
Douglas Gregor67c692c2010-08-26 15:07:07 +00004756 for (unsigned I = 0; I != NumSelIdents; ++I)
4757 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4758 return false;
4759
4760 return true;
4761}
4762
Douglas Gregorc8537c52009-11-19 07:41:15 +00004763static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4764 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004765 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004766 bool AllowSameLength = true) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00004767 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004768 AllowSameLength);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004769}
Douglas Gregor1154e272010-09-16 16:06:31 +00004770
4771namespace {
4772 /// \brief A set of selectors, which is used to avoid introducing multiple
4773 /// completions with the same selector into the result set.
4774 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4775}
4776
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004777/// \brief Add all of the Objective-C methods in the given Objective-C
4778/// container to the set of results.
4779///
4780/// The container will be a class, protocol, category, or implementation of
4781/// any of the above. This mether will recurse to include methods from
4782/// the superclasses of classes along with their categories, protocols, and
4783/// implementations.
4784///
4785/// \param Container the container in which we'll look to find methods.
4786///
James Dennett596e4752012-06-14 03:11:41 +00004787/// \param WantInstanceMethods Whether to add instance methods (only); if
4788/// false, this routine will add factory methods (only).
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004789///
4790/// \param CurContext the context in which we're performing the lookup that
4791/// finds methods.
4792///
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004793/// \param AllowSameLength Whether we allow a method to be added to the list
4794/// when it has the same number of parameters as we have selector identifiers.
4795///
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004796/// \param Results the structure into which we'll add results.
4797static void AddObjCMethods(ObjCContainerDecl *Container,
4798 bool WantInstanceMethods,
Douglas Gregorc8537c52009-11-19 07:41:15 +00004799 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004800 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004801 DeclContext *CurContext,
Douglas Gregor1154e272010-09-16 16:06:31 +00004802 VisitedSelectorSet &Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004803 bool AllowSameLength,
Douglas Gregor416b5752010-08-25 01:08:01 +00004804 ResultBuilder &Results,
4805 bool InOriginalClass = true) {
John McCall276321a2010-08-25 06:19:51 +00004806 typedef CodeCompletionResult Result;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00004807 Container = getContainerDef(Container);
Douglas Gregor41778c32013-01-30 06:58:39 +00004808 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4809 bool isRootClass = IFace && !IFace->getSuperClass();
Aaron Ballmanaff18c02014-03-13 19:03:34 +00004810 for (auto *M : Container->methods()) {
Douglas Gregor41778c32013-01-30 06:58:39 +00004811 // The instance methods on the root class can be messaged via the
4812 // metaclass.
4813 if (M->isInstanceMethod() == WantInstanceMethods ||
4814 (isRootClass && !WantInstanceMethods)) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00004815 // Check whether the selector identifiers we've been given are a
4816 // subset of the identifiers for this particular method.
Aaron Ballmanaff18c02014-03-13 19:03:34 +00004817 if (!isAcceptableObjCMethod(M, WantKind, SelIdents, AllowSameLength))
Douglas Gregor1b605f72009-11-19 01:08:35 +00004818 continue;
Douglas Gregorc8537c52009-11-19 07:41:15 +00004819
David Blaikie2d7c57e2012-04-30 02:36:29 +00004820 if (!Selectors.insert(M->getSelector()))
Douglas Gregor1154e272010-09-16 16:06:31 +00004821 continue;
Craig Topperc3ec1492014-05-26 06:22:03 +00004822
4823 Result R = Result(M, Results.getBasePriority(M), nullptr);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004824 R.StartParameter = SelIdents.size();
Douglas Gregorc8537c52009-11-19 07:41:15 +00004825 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor416b5752010-08-25 01:08:01 +00004826 if (!InOriginalClass)
4827 R.Priority += CCD_InBaseClass;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004828 Results.MaybeAddResult(R, CurContext);
4829 }
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004830 }
4831
Douglas Gregorf37c9492010-09-16 15:34:59 +00004832 // Visit the protocols of protocols.
4833 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregore6e48b12012-01-01 19:29:29 +00004834 if (Protocol->hasDefinition()) {
4835 const ObjCList<ObjCProtocolDecl> &Protocols
4836 = Protocol->getReferencedProtocols();
4837 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4838 E = Protocols.end();
4839 I != E; ++I)
4840 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004841 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregore6e48b12012-01-01 19:29:29 +00004842 }
Douglas Gregorf37c9492010-09-16 15:34:59 +00004843 }
4844
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00004845 if (!IFace || !IFace->hasDefinition())
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004846 return;
4847
4848 // Add methods in protocols.
Aaron Ballmana49c5062014-03-13 20:29:09 +00004849 for (auto *I : IFace->protocols())
4850 AddObjCMethods(I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004851 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004852
4853 // Add methods in categories.
Aaron Ballman15063e12014-03-13 21:35:02 +00004854 for (auto *CatDecl : IFace->known_categories()) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00004855 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004856 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004857 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004858
4859 // Add a categories protocol methods.
4860 const ObjCList<ObjCProtocolDecl> &Protocols
4861 = CatDecl->getReferencedProtocols();
4862 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4863 E = Protocols.end();
4864 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00004865 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004866 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004867 Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004868
4869 // Add methods in category implementations.
4870 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004871 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004872 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004873 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004874 }
4875
4876 // Add methods in superclass.
4877 if (IFace->getSuperClass())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004878 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004879 SelIdents, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004880 AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004881
4882 // Add methods in our implementation, if any.
4883 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004884 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004885 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004886 Results, InOriginalClass);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004887}
4888
4889
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004890void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004891 // Try to find the interface where getters might live.
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004892 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004893 if (!Class) {
4894 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004895 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004896 Class = Category->getClassInterface();
4897
4898 if (!Class)
4899 return;
4900 }
4901
4902 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004903 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004904 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004905 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004906 Results.EnterNewScope();
4907
Douglas Gregor1154e272010-09-16 16:06:31 +00004908 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004909 AddObjCMethods(Class, true, MK_ZeroArgSelector, None, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004910 /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004911 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004912 HandleCodeCompleteResults(this, CodeCompleter,
4913 CodeCompletionContext::CCC_Other,
4914 Results.data(),Results.size());
Douglas Gregorc8537c52009-11-19 07:41:15 +00004915}
4916
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004917void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004918 // Try to find the interface where setters might live.
4919 ObjCInterfaceDecl *Class
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004920 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004921 if (!Class) {
4922 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004923 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004924 Class = Category->getClassInterface();
4925
4926 if (!Class)
4927 return;
4928 }
4929
4930 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004931 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004932 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004933 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004934 Results.EnterNewScope();
4935
Douglas Gregor1154e272010-09-16 16:06:31 +00004936 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004937 AddObjCMethods(Class, true, MK_OneArgSelector, None, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004938 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004939
4940 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004941 HandleCodeCompleteResults(this, CodeCompleter,
4942 CodeCompletionContext::CCC_Other,
4943 Results.data(),Results.size());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004944}
4945
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004946void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
4947 bool IsParameter) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004948 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004949 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004950 CodeCompletionContext::CCC_Type);
Douglas Gregor99fa2642010-08-24 01:06:58 +00004951 Results.EnterNewScope();
4952
4953 // Add context-sensitive, Objective-C parameter-passing keywords.
4954 bool AddedInOut = false;
4955 if ((DS.getObjCDeclQualifier() &
4956 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
4957 Results.AddResult("in");
4958 Results.AddResult("inout");
4959 AddedInOut = true;
4960 }
4961 if ((DS.getObjCDeclQualifier() &
4962 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
4963 Results.AddResult("out");
4964 if (!AddedInOut)
4965 Results.AddResult("inout");
4966 }
4967 if ((DS.getObjCDeclQualifier() &
4968 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
4969 ObjCDeclSpec::DQ_Oneway)) == 0) {
4970 Results.AddResult("bycopy");
4971 Results.AddResult("byref");
4972 Results.AddResult("oneway");
4973 }
4974
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004975 // If we're completing the return type of an Objective-C method and the
4976 // identifier IBAction refers to a macro, provide a completion item for
4977 // an action, e.g.,
4978 // IBAction)<#selector#>:(id)sender
4979 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
4980 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004981 CodeCompletionBuilder Builder(Results.getAllocator(),
4982 Results.getCodeCompletionTUInfo(),
4983 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004984 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00004985 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004986 Builder.AddPlaceholderChunk("selector");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00004987 Builder.AddChunk(CodeCompletionString::CK_Colon);
4988 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004989 Builder.AddTextChunk("id");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00004990 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004991 Builder.AddTextChunk("sender");
4992 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
4993 }
Douglas Gregored1f5972013-01-30 07:11:43 +00004994
4995 // If we're completing the return type, provide 'instancetype'.
4996 if (!IsParameter) {
4997 Results.AddResult(CodeCompletionResult("instancetype"));
4998 }
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004999
Douglas Gregor99fa2642010-08-24 01:06:58 +00005000 // Add various builtin type names and specifiers.
5001 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5002 Results.ExitScope();
5003
5004 // Add the various type names
5005 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5006 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5007 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5008 CodeCompleter->includeGlobals());
5009
5010 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005011 AddMacroResults(PP, Results, false);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005012
5013 HandleCodeCompleteResults(this, CodeCompleter,
5014 CodeCompletionContext::CCC_Type,
5015 Results.data(), Results.size());
5016}
5017
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005018/// \brief When we have an expression with type "id", we may assume
5019/// that it has some more-specific class type based on knowledge of
5020/// common uses of Objective-C. This routine returns that class type,
5021/// or NULL if no better result could be determined.
5022static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregored0b69d2010-09-15 16:23:04 +00005023 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005024 if (!Msg)
Craig Topperc3ec1492014-05-26 06:22:03 +00005025 return nullptr;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005026
5027 Selector Sel = Msg->getSelector();
5028 if (Sel.isNull())
Craig Topperc3ec1492014-05-26 06:22:03 +00005029 return nullptr;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005030
5031 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5032 if (!Id)
Craig Topperc3ec1492014-05-26 06:22:03 +00005033 return nullptr;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005034
5035 ObjCMethodDecl *Method = Msg->getMethodDecl();
5036 if (!Method)
Craig Topperc3ec1492014-05-26 06:22:03 +00005037 return nullptr;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005038
5039 // Determine the class that we're sending the message to.
Craig Topperc3ec1492014-05-26 06:22:03 +00005040 ObjCInterfaceDecl *IFace = nullptr;
Douglas Gregor9a129192010-04-21 00:45:42 +00005041 switch (Msg->getReceiverKind()) {
5042 case ObjCMessageExpr::Class:
John McCall8b07ec22010-05-15 11:32:37 +00005043 if (const ObjCObjectType *ObjType
5044 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5045 IFace = ObjType->getInterface();
Douglas Gregor9a129192010-04-21 00:45:42 +00005046 break;
5047
5048 case ObjCMessageExpr::Instance: {
5049 QualType T = Msg->getInstanceReceiver()->getType();
5050 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5051 IFace = Ptr->getInterfaceDecl();
5052 break;
5053 }
5054
5055 case ObjCMessageExpr::SuperInstance:
5056 case ObjCMessageExpr::SuperClass:
5057 break;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005058 }
5059
5060 if (!IFace)
Craig Topperc3ec1492014-05-26 06:22:03 +00005061 return nullptr;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005062
5063 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5064 if (Method->isInstanceMethod())
5065 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5066 .Case("retain", IFace)
John McCall31168b02011-06-15 23:02:42 +00005067 .Case("strong", IFace)
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005068 .Case("autorelease", IFace)
5069 .Case("copy", IFace)
5070 .Case("copyWithZone", IFace)
5071 .Case("mutableCopy", IFace)
5072 .Case("mutableCopyWithZone", IFace)
5073 .Case("awakeFromCoder", IFace)
5074 .Case("replacementObjectFromCoder", IFace)
5075 .Case("class", IFace)
5076 .Case("classForCoder", IFace)
5077 .Case("superclass", Super)
Craig Topperc3ec1492014-05-26 06:22:03 +00005078 .Default(nullptr);
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005079
5080 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5081 .Case("new", IFace)
5082 .Case("alloc", IFace)
5083 .Case("allocWithZone", IFace)
5084 .Case("class", IFace)
5085 .Case("superclass", Super)
Craig Topperc3ec1492014-05-26 06:22:03 +00005086 .Default(nullptr);
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005087}
5088
Douglas Gregor6fc04132010-08-27 15:10:57 +00005089// Add a special completion for a message send to "super", which fills in the
5090// most likely case of forwarding all of our arguments to the superclass
5091// function.
5092///
5093/// \param S The semantic analysis object.
5094///
Dmitri Gribenkoadba9be2012-08-23 17:58:28 +00005095/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor6fc04132010-08-27 15:10:57 +00005096/// the "super" keyword. Otherwise, we just need to provide the arguments.
5097///
5098/// \param SelIdents The identifiers in the selector that have already been
5099/// provided as arguments for a send to "super".
5100///
Douglas Gregor6fc04132010-08-27 15:10:57 +00005101/// \param Results The set of results to augment.
5102///
5103/// \returns the Objective-C method declaration that would be invoked by
5104/// this "super" completion. If NULL, no completion was added.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005105static ObjCMethodDecl *AddSuperSendCompletion(
5106 Sema &S, bool NeedSuperKeyword,
5107 ArrayRef<IdentifierInfo *> SelIdents,
5108 ResultBuilder &Results) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005109 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5110 if (!CurMethod)
Craig Topperc3ec1492014-05-26 06:22:03 +00005111 return nullptr;
5112
Douglas Gregor6fc04132010-08-27 15:10:57 +00005113 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5114 if (!Class)
Craig Topperc3ec1492014-05-26 06:22:03 +00005115 return nullptr;
5116
Douglas Gregor6fc04132010-08-27 15:10:57 +00005117 // Try to find a superclass method with the same selector.
Craig Topperc3ec1492014-05-26 06:22:03 +00005118 ObjCMethodDecl *SuperMethod = nullptr;
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005119 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5120 // Check in the class
Douglas Gregor6fc04132010-08-27 15:10:57 +00005121 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5122 CurMethod->isInstanceMethod());
5123
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005124 // Check in categories or class extensions.
5125 if (!SuperMethod) {
Aaron Ballman15063e12014-03-13 21:35:02 +00005126 for (const auto *Cat : Class->known_categories()) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005127 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005128 CurMethod->isInstanceMethod())))
5129 break;
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005130 }
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005131 }
5132 }
5133
Douglas Gregor6fc04132010-08-27 15:10:57 +00005134 if (!SuperMethod)
Craig Topperc3ec1492014-05-26 06:22:03 +00005135 return nullptr;
5136
Douglas Gregor6fc04132010-08-27 15:10:57 +00005137 // Check whether the superclass method has the same signature.
5138 if (CurMethod->param_size() != SuperMethod->param_size() ||
5139 CurMethod->isVariadic() != SuperMethod->isVariadic())
Craig Topperc3ec1492014-05-26 06:22:03 +00005140 return nullptr;
5141
Douglas Gregor6fc04132010-08-27 15:10:57 +00005142 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5143 CurPEnd = CurMethod->param_end(),
5144 SuperP = SuperMethod->param_begin();
5145 CurP != CurPEnd; ++CurP, ++SuperP) {
5146 // Make sure the parameter types are compatible.
5147 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5148 (*SuperP)->getType()))
Craig Topperc3ec1492014-05-26 06:22:03 +00005149 return nullptr;
5150
Douglas Gregor6fc04132010-08-27 15:10:57 +00005151 // Make sure we have a parameter name to forward!
5152 if (!(*CurP)->getIdentifier())
Craig Topperc3ec1492014-05-26 06:22:03 +00005153 return nullptr;
Douglas Gregor6fc04132010-08-27 15:10:57 +00005154 }
5155
5156 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005157 CodeCompletionBuilder Builder(Results.getAllocator(),
5158 Results.getCodeCompletionTUInfo());
Douglas Gregor6fc04132010-08-27 15:10:57 +00005159
5160 // Give this completion a return type.
Douglas Gregor75acd922011-09-27 23:30:47 +00005161 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5162 Builder);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005163
5164 // If we need the "super" keyword, add it (plus some spacing).
5165 if (NeedSuperKeyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005166 Builder.AddTypedTextChunk("super");
5167 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005168 }
5169
5170 Selector Sel = CurMethod->getSelector();
5171 if (Sel.isUnarySelector()) {
5172 if (NeedSuperKeyword)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005173 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005174 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005175 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005176 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005177 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005178 } else {
5179 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5180 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005181 if (I > SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005182 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005183
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005184 if (I < SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005185 Builder.AddInformativeChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005186 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005187 Sel.getNameForSlot(I) + ":"));
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005188 else if (NeedSuperKeyword || I > SelIdents.size()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005189 Builder.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005190 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005191 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005192 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005193 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005194 } else {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005195 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005196 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005197 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005198 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005199 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005200 }
5201 }
5202 }
5203
Douglas Gregor78254c82012-03-27 23:34:16 +00005204 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5205 CCP_SuperCompletion));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005206 return SuperMethod;
5207}
5208
Douglas Gregora817a192010-05-27 23:06:34 +00005209void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall276321a2010-08-25 06:19:51 +00005210 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005211 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005212 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005213 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005214 getLangOpts().CPlusPlus11
Douglas Gregord8c61782012-02-15 15:34:24 +00005215 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5216 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregora817a192010-05-27 23:06:34 +00005217
Douglas Gregora817a192010-05-27 23:06:34 +00005218 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5219 Results.EnterNewScope();
Douglas Gregor39982192010-08-15 06:18:01 +00005220 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5221 CodeCompleter->includeGlobals());
Douglas Gregora817a192010-05-27 23:06:34 +00005222
5223 // If we are in an Objective-C method inside a class that has a superclass,
5224 // add "super" as an option.
5225 if (ObjCMethodDecl *Method = getCurMethodDecl())
5226 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor6fc04132010-08-27 15:10:57 +00005227 if (Iface->getSuperClass()) {
Douglas Gregora817a192010-05-27 23:06:34 +00005228 Results.AddResult(Result("super"));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005229
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005230 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, None, Results);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005231 }
Douglas Gregora817a192010-05-27 23:06:34 +00005232
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005233 if (getLangOpts().CPlusPlus11)
Douglas Gregord8c61782012-02-15 15:34:24 +00005234 addThisCompletion(*this, Results);
5235
Douglas Gregora817a192010-05-27 23:06:34 +00005236 Results.ExitScope();
5237
5238 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005239 AddMacroResults(PP, Results, false);
Douglas Gregor50832e02010-09-20 22:39:41 +00005240 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005241 Results.data(), Results.size());
Douglas Gregora817a192010-05-27 23:06:34 +00005242
5243}
5244
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005245void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005246 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005247 bool AtArgumentExpression) {
Craig Topperc3ec1492014-05-26 06:22:03 +00005248 ObjCInterfaceDecl *CDecl = nullptr;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005249 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5250 // Figure out which interface we're in.
5251 CDecl = CurMethod->getClassInterface();
5252 if (!CDecl)
5253 return;
5254
5255 // Find the superclass of this class.
5256 CDecl = CDecl->getSuperClass();
5257 if (!CDecl)
5258 return;
5259
5260 if (CurMethod->isInstanceMethod()) {
5261 // We are inside an instance method, which means that the message
5262 // send [super ...] is actually calling an instance method on the
Douglas Gregor392a84b2010-10-13 21:24:53 +00005263 // current object.
Craig Topperc3ec1492014-05-26 06:22:03 +00005264 return CodeCompleteObjCInstanceMessage(S, nullptr, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005265 AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005266 CDecl);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005267 }
5268
5269 // Fall through to send to the superclass in CDecl.
5270 } else {
5271 // "super" may be the name of a type or variable. Figure out which
5272 // it is.
Argyrios Kyrtzidis3e56dd42013-03-14 22:56:43 +00005273 IdentifierInfo *Super = getSuperIdentifier();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005274 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5275 LookupOrdinaryName);
5276 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5277 // "super" names an interface. Use it.
5278 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCall8b07ec22010-05-15 11:32:37 +00005279 if (const ObjCObjectType *Iface
5280 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5281 CDecl = Iface->getInterface();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005282 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5283 // "super" names an unresolved type; we can't be more specific.
5284 } else {
5285 // Assume that "super" names some kind of value and parse that way.
5286 CXXScopeSpec SS;
Abramo Bagnara7945c982012-01-27 09:46:47 +00005287 SourceLocation TemplateKWLoc;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005288 UnqualifiedId id;
5289 id.setIdentifier(Super, SuperLoc);
Abramo Bagnara7945c982012-01-27 09:46:47 +00005290 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5291 false, false);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005292 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005293 SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005294 AtArgumentExpression);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005295 }
5296
5297 // Fall through
5298 }
5299
John McCallba7bf592010-08-24 05:47:05 +00005300 ParsedType Receiver;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005301 if (CDecl)
John McCallba7bf592010-08-24 05:47:05 +00005302 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005303 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005304 AtArgumentExpression,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005305 /*IsSuper=*/true);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005306}
5307
Douglas Gregor74661272010-09-21 00:03:25 +00005308/// \brief Given a set of code-completion results for the argument of a message
5309/// send, determine the preferred type (if any) for that argument expression.
5310static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5311 unsigned NumSelIdents) {
5312 typedef CodeCompletionResult Result;
5313 ASTContext &Context = Results.getSema().Context;
5314
5315 QualType PreferredType;
5316 unsigned BestPriority = CCP_Unlikely * 2;
5317 Result *ResultsData = Results.data();
5318 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5319 Result &R = ResultsData[I];
5320 if (R.Kind == Result::RK_Declaration &&
5321 isa<ObjCMethodDecl>(R.Declaration)) {
5322 if (R.Priority <= BestPriority) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00005323 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregor74661272010-09-21 00:03:25 +00005324 if (NumSelIdents <= Method->param_size()) {
5325 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
5326 ->getType();
5327 if (R.Priority < BestPriority || PreferredType.isNull()) {
5328 BestPriority = R.Priority;
5329 PreferredType = MyPreferredType;
5330 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5331 MyPreferredType)) {
5332 PreferredType = QualType();
5333 }
5334 }
5335 }
5336 }
5337 }
5338
5339 return PreferredType;
5340}
5341
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005342static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5343 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005344 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005345 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005346 bool IsSuper,
5347 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005348 typedef CodeCompletionResult Result;
Craig Topperc3ec1492014-05-26 06:22:03 +00005349 ObjCInterfaceDecl *CDecl = nullptr;
5350
Douglas Gregor8ce33212009-11-17 17:59:40 +00005351 // If the given name refers to an interface type, retrieve the
5352 // corresponding declaration.
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005353 if (Receiver) {
Craig Topperc3ec1492014-05-26 06:22:03 +00005354 QualType T = SemaRef.GetTypeFromParser(Receiver, nullptr);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005355 if (!T.isNull())
John McCall8b07ec22010-05-15 11:32:37 +00005356 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5357 CDecl = Interface->getInterface();
Douglas Gregor8ce33212009-11-17 17:59:40 +00005358 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005359
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005360 // Add all of the factory methods in this Objective-C class, its protocols,
5361 // superclasses, categories, implementation, etc.
Steve Naroffeae65032009-11-07 02:08:14 +00005362 Results.EnterNewScope();
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005363
Douglas Gregor6fc04132010-08-27 15:10:57 +00005364 // If this is a send-to-super, try to add the special "super" send
5365 // completion.
5366 if (IsSuper) {
5367 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005368 = AddSuperSendCompletion(SemaRef, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005369 Results.Ignore(SuperMethod);
5370 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005371
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005372 // If we're inside an Objective-C method definition, prefer its selector to
5373 // others.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005374 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005375 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005376
Douglas Gregor1154e272010-09-16 16:06:31 +00005377 VisitedSelectorSet Selectors;
Douglas Gregor6285f752010-04-06 16:40:00 +00005378 if (CDecl)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005379 AddObjCMethods(CDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005380 SemaRef.CurContext, Selectors, AtArgumentExpression,
5381 Results);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005382 else {
Douglas Gregor6285f752010-04-06 16:40:00 +00005383 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005384
Douglas Gregord720daf2010-04-06 17:30:22 +00005385 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005386 // pool from the AST file.
Axel Naumanndd433f02012-10-18 19:05:02 +00005387 if (SemaRef.getExternalSource()) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005388 for (uint32_t I = 0,
Axel Naumanndd433f02012-10-18 19:05:02 +00005389 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall75b960e2010-06-01 09:23:16 +00005390 I != N; ++I) {
Axel Naumanndd433f02012-10-18 19:05:02 +00005391 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005392 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005393 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005394
5395 SemaRef.ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005396 }
5397 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005398
5399 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5400 MEnd = SemaRef.MethodPool.end();
Sebastian Redl75d8a322010-08-02 23:18:59 +00005401 M != MEnd; ++M) {
5402 for (ObjCMethodList *MethList = &M->second.second;
5403 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005404 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005405 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005406 continue;
Craig Topperc3ec1492014-05-26 06:22:03 +00005407
5408 Result R(MethList->Method, Results.getBasePriority(MethList->Method),
5409 nullptr);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005410 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005411 R.AllParametersAreInformative = false;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005412 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor6285f752010-04-06 16:40:00 +00005413 }
5414 }
5415 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005416
5417 Results.ExitScope();
5418}
Douglas Gregor6285f752010-04-06 16:40:00 +00005419
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005420void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005421 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005422 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005423 bool IsSuper) {
Douglas Gregor63745d52011-07-21 01:05:26 +00005424
5425 QualType T = this->GetTypeFromParser(Receiver);
5426
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005427 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005428 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005429 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005430 T, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005431
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005432 AddClassMessageCompletions(*this, S, Receiver, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005433 AtArgumentExpression, IsSuper, Results);
Douglas Gregor74661272010-09-21 00:03:25 +00005434
5435 // If we're actually at the argument expression (rather than prior to the
5436 // selector), we're actually performing code completion for an expression.
5437 // Determine whether we have a single, best method. If so, we can
5438 // code-complete the expression using the corresponding parameter type as
5439 // our preferred type, improving completion results.
5440 if (AtArgumentExpression) {
5441 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005442 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005443 if (PreferredType.isNull())
5444 CodeCompleteOrdinaryName(S, PCC_Expression);
5445 else
5446 CodeCompleteExpression(S, PreferredType);
5447 return;
5448 }
5449
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005450 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005451 Results.getCompletionContext(),
Douglas Gregor6fc04132010-08-27 15:10:57 +00005452 Results.data(), Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005453}
5454
Richard Trieu2bd04012011-09-09 02:00:50 +00005455void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005456 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005457 bool AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005458 ObjCInterfaceDecl *Super) {
John McCall276321a2010-08-25 06:19:51 +00005459 typedef CodeCompletionResult Result;
Steve Naroffeae65032009-11-07 02:08:14 +00005460
5461 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffeae65032009-11-07 02:08:14 +00005462
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005463 // If necessary, apply function/array conversion to the receiver.
5464 // C99 6.7.5.3p[7,8].
John Wiegley01296292011-04-08 18:41:53 +00005465 if (RecExpr) {
5466 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5467 if (Conv.isInvalid()) // conversion failed. bail.
5468 return;
Nikola Smiljanic01a75982014-05-29 10:55:11 +00005469 RecExpr = Conv.get();
John Wiegley01296292011-04-08 18:41:53 +00005470 }
Douglas Gregor392a84b2010-10-13 21:24:53 +00005471 QualType ReceiverType = RecExpr? RecExpr->getType()
5472 : Super? Context.getObjCObjectPointerType(
5473 Context.getObjCInterfaceType(Super))
5474 : Context.getObjCIdType();
Steve Naroffeae65032009-11-07 02:08:14 +00005475
Douglas Gregordc520b02010-11-08 21:12:30 +00005476 // If we're messaging an expression with type "id" or "Class", check
5477 // whether we know something special about the receiver that allows
5478 // us to assume a more-specific receiver type.
Anders Carlsson382ba412014-02-28 19:07:22 +00005479 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType()) {
Douglas Gregordc520b02010-11-08 21:12:30 +00005480 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5481 if (ReceiverType->isObjCClassType())
5482 return CodeCompleteObjCClassMessage(S,
5483 ParsedType::make(Context.getObjCInterfaceType(IFace)),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005484 SelIdents,
Douglas Gregordc520b02010-11-08 21:12:30 +00005485 AtArgumentExpression, Super);
5486
5487 ReceiverType = Context.getObjCObjectPointerType(
5488 Context.getObjCInterfaceType(IFace));
5489 }
Anders Carlsson382ba412014-02-28 19:07:22 +00005490 } else if (RecExpr && getLangOpts().CPlusPlus) {
5491 ExprResult Conv = PerformContextuallyConvertToObjCPointer(RecExpr);
5492 if (Conv.isUsable()) {
Nikola Smiljanic01a75982014-05-29 10:55:11 +00005493 RecExpr = Conv.get();
Anders Carlsson382ba412014-02-28 19:07:22 +00005494 ReceiverType = RecExpr->getType();
5495 }
5496 }
Douglas Gregordc520b02010-11-08 21:12:30 +00005497
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005498 // Build the set of methods we can see.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005499 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005500 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005501 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005502 ReceiverType, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005503
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005504 Results.EnterNewScope();
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005505
Douglas Gregor6fc04132010-08-27 15:10:57 +00005506 // If this is a send-to-super, try to add the special "super" send
5507 // completion.
Douglas Gregor392a84b2010-10-13 21:24:53 +00005508 if (Super) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005509 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005510 = AddSuperSendCompletion(*this, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005511 Results.Ignore(SuperMethod);
5512 }
5513
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005514 // If we're inside an Objective-C method definition, prefer its selector to
5515 // others.
5516 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5517 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005518
Douglas Gregor1154e272010-09-16 16:06:31 +00005519 // Keep track of the selectors we've already added.
5520 VisitedSelectorSet Selectors;
5521
Douglas Gregora3329fa2009-11-18 00:06:18 +00005522 // Handle messages to Class. This really isn't a message to an instance
5523 // method, so we treat it the same way we would treat a message send to a
5524 // class method.
5525 if (ReceiverType->isObjCClassType() ||
5526 ReceiverType->isObjCQualifiedClassType()) {
5527 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5528 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005529 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005530 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005531 }
5532 }
5533 // Handle messages to a qualified ID ("id<foo>").
5534 else if (const ObjCObjectPointerType *QualID
5535 = ReceiverType->getAsObjCQualifiedIdType()) {
5536 // Search protocols for instance methods.
Aaron Ballman83731462014-03-17 16:14:00 +00005537 for (auto *I : QualID->quals())
5538 AddObjCMethods(I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005539 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005540 }
5541 // Handle messages to a pointer to interface type.
5542 else if (const ObjCObjectPointerType *IFacePtr
5543 = ReceiverType->getAsObjCInterfacePointerType()) {
5544 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregorc8537c52009-11-19 07:41:15 +00005545 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005546 CurContext, Selectors, AtArgumentExpression,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005547 Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005548
5549 // Search protocols for instance methods.
Aaron Ballman83731462014-03-17 16:14:00 +00005550 for (auto *I : IFacePtr->quals())
5551 AddObjCMethods(I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005552 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005553 }
Douglas Gregor6285f752010-04-06 16:40:00 +00005554 // Handle messages to "id".
5555 else if (ReceiverType->isObjCIdType()) {
Douglas Gregord720daf2010-04-06 17:30:22 +00005556 // We're messaging "id", so provide all instance methods we know
5557 // about as code-completion results.
5558
5559 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005560 // pool from the AST file.
Douglas Gregord720daf2010-04-06 17:30:22 +00005561 if (ExternalSource) {
John McCall75b960e2010-06-01 09:23:16 +00005562 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5563 I != N; ++I) {
5564 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00005565 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005566 continue;
5567
Sebastian Redl75d8a322010-08-02 23:18:59 +00005568 ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005569 }
5570 }
5571
Sebastian Redl75d8a322010-08-02 23:18:59 +00005572 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5573 MEnd = MethodPool.end();
5574 M != MEnd; ++M) {
5575 for (ObjCMethodList *MethList = &M->second.first;
5576 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005577 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005578 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005579 continue;
Douglas Gregor1154e272010-09-16 16:06:31 +00005580
5581 if (!Selectors.insert(MethList->Method->getSelector()))
5582 continue;
Craig Topperc3ec1492014-05-26 06:22:03 +00005583
5584 Result R(MethList->Method, Results.getBasePriority(MethList->Method),
5585 nullptr);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005586 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005587 R.AllParametersAreInformative = false;
5588 Results.MaybeAddResult(R, CurContext);
5589 }
5590 }
5591 }
Steve Naroffeae65032009-11-07 02:08:14 +00005592 Results.ExitScope();
Douglas Gregor74661272010-09-21 00:03:25 +00005593
5594
5595 // If we're actually at the argument expression (rather than prior to the
5596 // selector), we're actually performing code completion for an expression.
5597 // Determine whether we have a single, best method. If so, we can
5598 // code-complete the expression using the corresponding parameter type as
5599 // our preferred type, improving completion results.
5600 if (AtArgumentExpression) {
5601 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005602 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005603 if (PreferredType.isNull())
5604 CodeCompleteOrdinaryName(S, PCC_Expression);
5605 else
5606 CodeCompleteExpression(S, PreferredType);
5607 return;
5608 }
5609
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005610 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005611 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005612 Results.data(),Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005613}
Douglas Gregorbaf69612009-11-18 04:19:12 +00005614
Douglas Gregor68762e72010-08-23 21:17:50 +00005615void Sema::CodeCompleteObjCForCollection(Scope *S,
5616 DeclGroupPtrTy IterationVar) {
5617 CodeCompleteExpressionData Data;
5618 Data.ObjCCollection = true;
5619
5620 if (IterationVar.getAsOpaquePtr()) {
Serge Pavlov9ddb76e2013-08-27 13:15:56 +00005621 DeclGroupRef DG = IterationVar.get();
Douglas Gregor68762e72010-08-23 21:17:50 +00005622 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5623 if (*I)
5624 Data.IgnoreDecls.push_back(*I);
5625 }
5626 }
5627
5628 CodeCompleteExpression(S, Data);
5629}
5630
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005631void Sema::CodeCompleteObjCSelector(Scope *S,
5632 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00005633 // If we have an external source, load the entire class method
5634 // pool from the AST file.
5635 if (ExternalSource) {
5636 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5637 I != N; ++I) {
5638 Selector Sel = ExternalSource->GetExternalSelector(I);
5639 if (Sel.isNull() || MethodPool.count(Sel))
5640 continue;
5641
5642 ReadMethodPool(Sel);
5643 }
5644 }
5645
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005646 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005647 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005648 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor67c692c2010-08-26 15:07:07 +00005649 Results.EnterNewScope();
5650 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5651 MEnd = MethodPool.end();
5652 M != MEnd; ++M) {
5653
5654 Selector Sel = M->first;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005655 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents))
Douglas Gregor67c692c2010-08-26 15:07:07 +00005656 continue;
5657
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005658 CodeCompletionBuilder Builder(Results.getAllocator(),
5659 Results.getCodeCompletionTUInfo());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005660 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005661 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005662 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005663 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005664 continue;
5665 }
5666
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005667 std::string Accumulator;
Douglas Gregor67c692c2010-08-26 15:07:07 +00005668 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005669 if (I == SelIdents.size()) {
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005670 if (!Accumulator.empty()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005671 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005672 Accumulator));
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005673 Accumulator.clear();
5674 }
5675 }
5676
Benjamin Kramer632500c2011-07-26 16:59:25 +00005677 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005678 Accumulator += ':';
Douglas Gregor67c692c2010-08-26 15:07:07 +00005679 }
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005680 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005681 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005682 }
5683 Results.ExitScope();
5684
5685 HandleCodeCompleteResults(this, CodeCompleter,
5686 CodeCompletionContext::CCC_SelectorName,
5687 Results.data(), Results.size());
5688}
5689
Douglas Gregorbaf69612009-11-18 04:19:12 +00005690/// \brief Add all of the protocol declarations that we find in the given
5691/// (translation unit) context.
5692static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005693 bool OnlyForwardDeclarations,
Douglas Gregorbaf69612009-11-18 04:19:12 +00005694 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005695 typedef CodeCompletionResult Result;
Douglas Gregorbaf69612009-11-18 04:19:12 +00005696
Aaron Ballman629afae2014-03-07 19:56:05 +00005697 for (const auto *D : Ctx->decls()) {
Douglas Gregorbaf69612009-11-18 04:19:12 +00005698 // Record any protocols we find.
Aaron Ballman629afae2014-03-07 19:56:05 +00005699 if (const auto *Proto = dyn_cast<ObjCProtocolDecl>(D))
Douglas Gregore6e48b12012-01-01 19:29:29 +00005700 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Craig Topperc3ec1492014-05-26 06:22:03 +00005701 Results.AddResult(Result(Proto, Results.getBasePriority(Proto),nullptr),
5702 CurContext, nullptr, false);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005703 }
5704}
5705
5706void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5707 unsigned NumProtocols) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005708 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005709 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005710 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005711
Douglas Gregora3b23b02010-12-09 21:44:02 +00005712 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5713 Results.EnterNewScope();
5714
5715 // Tell the result set to ignore all of the protocols we have
5716 // already seen.
5717 // FIXME: This doesn't work when caching code-completion results.
5718 for (unsigned I = 0; I != NumProtocols; ++I)
5719 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5720 Protocols[I].second))
5721 Results.Ignore(Protocol);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005722
Douglas Gregora3b23b02010-12-09 21:44:02 +00005723 // Add all protocols.
5724 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5725 Results);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005726
Douglas Gregora3b23b02010-12-09 21:44:02 +00005727 Results.ExitScope();
5728 }
5729
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005730 HandleCodeCompleteResults(this, CodeCompleter,
5731 CodeCompletionContext::CCC_ObjCProtocolName,
5732 Results.data(),Results.size());
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005733}
5734
5735void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005736 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005737 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005738 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005739
Douglas Gregora3b23b02010-12-09 21:44:02 +00005740 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5741 Results.EnterNewScope();
5742
5743 // Add all protocols.
5744 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5745 Results);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005746
Douglas Gregora3b23b02010-12-09 21:44:02 +00005747 Results.ExitScope();
5748 }
5749
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005750 HandleCodeCompleteResults(this, CodeCompleter,
5751 CodeCompletionContext::CCC_ObjCProtocolName,
5752 Results.data(),Results.size());
Douglas Gregorbaf69612009-11-18 04:19:12 +00005753}
Douglas Gregor49c22a72009-11-18 16:26:39 +00005754
5755/// \brief Add all of the Objective-C interface declarations that we find in
5756/// the given (translation unit) context.
5757static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5758 bool OnlyForwardDeclarations,
5759 bool OnlyUnimplemented,
5760 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005761 typedef CodeCompletionResult Result;
Douglas Gregor49c22a72009-11-18 16:26:39 +00005762
Aaron Ballman629afae2014-03-07 19:56:05 +00005763 for (const auto *D : Ctx->decls()) {
Douglas Gregor1c283312010-08-11 12:19:30 +00005764 // Record any interfaces we find.
Aaron Ballman629afae2014-03-07 19:56:05 +00005765 if (const auto *Class = dyn_cast<ObjCInterfaceDecl>(D))
Douglas Gregordc9166c2011-12-15 20:29:51 +00005766 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregor1c283312010-08-11 12:19:30 +00005767 (!OnlyUnimplemented || !Class->getImplementation()))
Craig Topperc3ec1492014-05-26 06:22:03 +00005768 Results.AddResult(Result(Class, Results.getBasePriority(Class),nullptr),
5769 CurContext, nullptr, false);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005770 }
5771}
5772
5773void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005774 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005775 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005776 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005777 Results.EnterNewScope();
5778
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005779 if (CodeCompleter->includeGlobals()) {
5780 // Add all classes.
5781 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5782 false, Results);
5783 }
5784
Douglas Gregor49c22a72009-11-18 16:26:39 +00005785 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005786
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005787 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005788 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005789 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005790}
5791
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005792void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5793 SourceLocation ClassNameLoc) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005794 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005795 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005796 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005797 Results.EnterNewScope();
5798
5799 // Make sure that we ignore the class we're currently defining.
5800 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005801 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005802 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor49c22a72009-11-18 16:26:39 +00005803 Results.Ignore(CurClass);
5804
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005805 if (CodeCompleter->includeGlobals()) {
5806 // Add all classes.
5807 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5808 false, Results);
5809 }
5810
Douglas Gregor49c22a72009-11-18 16:26:39 +00005811 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005812
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005813 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005814 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005815 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005816}
5817
5818void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005819 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005820 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005821 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005822 Results.EnterNewScope();
5823
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005824 if (CodeCompleter->includeGlobals()) {
5825 // Add all unimplemented classes.
5826 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5827 true, Results);
5828 }
5829
Douglas Gregor49c22a72009-11-18 16:26:39 +00005830 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005831
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005832 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005833 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005834 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005835}
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005836
5837void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005838 IdentifierInfo *ClassName,
5839 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005840 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005841
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005842 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005843 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005844 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005845
5846 // Ignore any categories we find that have already been implemented by this
5847 // interface.
5848 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5849 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005850 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005851 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
Aaron Ballman3fe486a2014-03-13 21:23:55 +00005852 for (const auto *Cat : Class->visible_categories())
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005853 CategoryNames.insert(Cat->getIdentifier());
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005854 }
5855
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005856 // Add all of the categories we know about.
5857 Results.EnterNewScope();
5858 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
Aaron Ballman629afae2014-03-07 19:56:05 +00005859 for (const auto *D : TU->decls())
5860 if (const auto *Category = dyn_cast<ObjCCategoryDecl>(D))
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005861 if (CategoryNames.insert(Category->getIdentifier()))
Craig Topperc3ec1492014-05-26 06:22:03 +00005862 Results.AddResult(Result(Category, Results.getBasePriority(Category),
5863 nullptr),
5864 CurContext, nullptr, false);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005865 Results.ExitScope();
5866
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005867 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005868 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005869 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005870}
5871
5872void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005873 IdentifierInfo *ClassName,
5874 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005875 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005876
5877 // Find the corresponding interface. If we couldn't find the interface, the
5878 // program itself is ill-formed. However, we'll try to be helpful still by
5879 // providing the list of all of the categories we know about.
5880 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005881 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005882 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5883 if (!Class)
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005884 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005885
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005886 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005887 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005888 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005889
5890 // Add all of the categories that have have corresponding interface
5891 // declarations in this class and any of its superclasses, except for
5892 // already-implemented categories in the class itself.
5893 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5894 Results.EnterNewScope();
5895 bool IgnoreImplemented = true;
5896 while (Class) {
Aaron Ballman3fe486a2014-03-13 21:23:55 +00005897 for (const auto *Cat : Class->visible_categories()) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005898 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
5899 CategoryNames.insert(Cat->getIdentifier()))
Craig Topperc3ec1492014-05-26 06:22:03 +00005900 Results.AddResult(Result(Cat, Results.getBasePriority(Cat), nullptr),
5901 CurContext, nullptr, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005902 }
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005903
5904 Class = Class->getSuperClass();
5905 IgnoreImplemented = false;
5906 }
5907 Results.ExitScope();
5908
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005909 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005910 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005911 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005912}
Douglas Gregor5d649882009-11-18 22:32:06 +00005913
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005914void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005915 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005916 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005917 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00005918
5919 // Figure out where this @synthesize lives.
5920 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005921 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00005922 if (!Container ||
5923 (!isa<ObjCImplementationDecl>(Container) &&
5924 !isa<ObjCCategoryImplDecl>(Container)))
5925 return;
5926
5927 // Ignore any properties that have already been implemented.
Douglas Gregor9b4f3702012-06-12 13:44:08 +00005928 Container = getContainerDef(Container);
Aaron Ballman629afae2014-03-07 19:56:05 +00005929 for (const auto *D : Container->decls())
5930 if (const auto *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(D))
Douglas Gregor5d649882009-11-18 22:32:06 +00005931 Results.Ignore(PropertyImpl->getPropertyDecl());
5932
5933 // Add any properties that we find.
Douglas Gregorb888acf2010-12-09 23:01:55 +00005934 AddedPropertiesSet AddedProperties;
Douglas Gregor5d649882009-11-18 22:32:06 +00005935 Results.EnterNewScope();
5936 if (ObjCImplementationDecl *ClassImpl
5937 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor95147142011-05-05 15:50:42 +00005938 AddObjCProperties(ClassImpl->getClassInterface(), false,
5939 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00005940 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00005941 else
5942 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor95147142011-05-05 15:50:42 +00005943 false, /*AllowNullaryMethods=*/false, CurContext,
5944 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00005945 Results.ExitScope();
5946
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005947 HandleCodeCompleteResults(this, CodeCompleter,
5948 CodeCompletionContext::CCC_Other,
5949 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00005950}
5951
5952void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005953 IdentifierInfo *PropertyName) {
John McCall276321a2010-08-25 06:19:51 +00005954 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005955 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005956 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005957 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00005958
5959 // Figure out where this @synthesize lives.
5960 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005961 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00005962 if (!Container ||
5963 (!isa<ObjCImplementationDecl>(Container) &&
5964 !isa<ObjCCategoryImplDecl>(Container)))
5965 return;
5966
5967 // Figure out which interface we're looking into.
Craig Topperc3ec1492014-05-26 06:22:03 +00005968 ObjCInterfaceDecl *Class = nullptr;
Douglas Gregor5d649882009-11-18 22:32:06 +00005969 if (ObjCImplementationDecl *ClassImpl
5970 = dyn_cast<ObjCImplementationDecl>(Container))
5971 Class = ClassImpl->getClassInterface();
5972 else
5973 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
5974 ->getClassInterface();
5975
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00005976 // Determine the type of the property we're synthesizing.
5977 QualType PropertyType = Context.getObjCIdType();
5978 if (Class) {
5979 if (ObjCPropertyDecl *Property
5980 = Class->FindPropertyDeclaration(PropertyName)) {
5981 PropertyType
5982 = Property->getType().getNonReferenceType().getUnqualifiedType();
5983
5984 // Give preference to ivars
5985 Results.setPreferredType(PropertyType);
5986 }
5987 }
5988
Douglas Gregor5d649882009-11-18 22:32:06 +00005989 // Add all of the instance variables in this class and its superclasses.
5990 Results.EnterNewScope();
Douglas Gregor331faa02011-04-18 14:13:53 +00005991 bool SawSimilarlyNamedIvar = false;
5992 std::string NameWithPrefix;
5993 NameWithPrefix += '_';
Benjamin Kramer632500c2011-07-26 16:59:25 +00005994 NameWithPrefix += PropertyName->getName();
Douglas Gregor331faa02011-04-18 14:13:53 +00005995 std::string NameWithSuffix = PropertyName->getName().str();
5996 NameWithSuffix += '_';
Douglas Gregor5d649882009-11-18 22:32:06 +00005997 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregor331faa02011-04-18 14:13:53 +00005998 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
5999 Ivar = Ivar->getNextIvar()) {
Craig Topperc3ec1492014-05-26 06:22:03 +00006000 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), nullptr),
6001 CurContext, nullptr, false);
6002
Douglas Gregor331faa02011-04-18 14:13:53 +00006003 // Determine whether we've seen an ivar with a name similar to the
6004 // property.
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006005 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregor331faa02011-04-18 14:13:53 +00006006 NameWithPrefix == Ivar->getName() ||
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006007 NameWithSuffix == Ivar->getName())) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006008 SawSimilarlyNamedIvar = true;
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006009
6010 // Reduce the priority of this result by one, to give it a slight
6011 // advantage over other results whose names don't match so closely.
6012 if (Results.size() &&
6013 Results.data()[Results.size() - 1].Kind
6014 == CodeCompletionResult::RK_Declaration &&
6015 Results.data()[Results.size() - 1].Declaration == Ivar)
6016 Results.data()[Results.size() - 1].Priority--;
6017 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006018 }
Douglas Gregor5d649882009-11-18 22:32:06 +00006019 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006020
6021 if (!SawSimilarlyNamedIvar) {
6022 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006023 // an ivar of the appropriate type.
6024 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregor331faa02011-04-18 14:13:53 +00006025 typedef CodeCompletionResult Result;
6026 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006027 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6028 Priority,CXAvailability_Available);
Douglas Gregor331faa02011-04-18 14:13:53 +00006029
Douglas Gregor75acd922011-09-27 23:30:47 +00006030 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006031 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006032 Policy, Allocator));
Douglas Gregor331faa02011-04-18 14:13:53 +00006033 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6034 Results.AddResult(Result(Builder.TakeString(), Priority,
6035 CXCursor_ObjCIvarDecl));
6036 }
6037
Douglas Gregor5d649882009-11-18 22:32:06 +00006038 Results.ExitScope();
6039
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006040 HandleCodeCompleteResults(this, CodeCompleter,
6041 CodeCompletionContext::CCC_Other,
6042 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006043}
Douglas Gregor636a61e2010-04-07 00:21:17 +00006044
Douglas Gregor416b5752010-08-25 01:08:01 +00006045// Mapping from selectors to the methods that implement that selector, along
6046// with the "in original class" flag.
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006047typedef llvm::DenseMap<
6048 Selector, llvm::PointerIntPair<ObjCMethodDecl *, 1, bool> > KnownMethodsMap;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006049
6050/// \brief Find all of the methods that reside in the given container
6051/// (and its superclasses, protocols, etc.) that meet the given
6052/// criteria. Insert those methods into the map of known methods,
6053/// indexed by selector so they can be easily found.
6054static void FindImplementableMethods(ASTContext &Context,
6055 ObjCContainerDecl *Container,
6056 bool WantInstanceMethods,
6057 QualType ReturnType,
Douglas Gregor416b5752010-08-25 01:08:01 +00006058 KnownMethodsMap &KnownMethods,
6059 bool InOriginalClass = true) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006060 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006061 // Make sure we have a definition; that's what we'll walk.
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006062 if (!IFace->hasDefinition())
6063 return;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006064
6065 IFace = IFace->getDefinition();
6066 Container = IFace;
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006067
Douglas Gregor636a61e2010-04-07 00:21:17 +00006068 const ObjCList<ObjCProtocolDecl> &Protocols
6069 = IFace->getReferencedProtocols();
6070 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006071 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006072 I != E; ++I)
6073 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006074 KnownMethods, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006075
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006076 // Add methods from any class extensions and categories.
Aaron Ballman3fe486a2014-03-13 21:23:55 +00006077 for (auto *Cat : IFace->visible_categories()) {
6078 FindImplementableMethods(Context, Cat, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006079 KnownMethods, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006080 }
6081
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006082 // Visit the superclass.
6083 if (IFace->getSuperClass())
6084 FindImplementableMethods(Context, IFace->getSuperClass(),
6085 WantInstanceMethods, ReturnType,
6086 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006087 }
6088
6089 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6090 // Recurse into protocols.
6091 const ObjCList<ObjCProtocolDecl> &Protocols
6092 = Category->getReferencedProtocols();
6093 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006094 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006095 I != E; ++I)
6096 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006097 KnownMethods, InOriginalClass);
6098
6099 // If this category is the original class, jump to the interface.
6100 if (InOriginalClass && Category->getClassInterface())
6101 FindImplementableMethods(Context, Category->getClassInterface(),
6102 WantInstanceMethods, ReturnType, KnownMethods,
6103 false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006104 }
6105
6106 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006107 // Make sure we have a definition; that's what we'll walk.
6108 if (!Protocol->hasDefinition())
6109 return;
6110 Protocol = Protocol->getDefinition();
6111 Container = Protocol;
6112
6113 // Recurse into protocols.
6114 const ObjCList<ObjCProtocolDecl> &Protocols
6115 = Protocol->getReferencedProtocols();
6116 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6117 E = Protocols.end();
6118 I != E; ++I)
6119 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6120 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006121 }
6122
6123 // Add methods in this container. This operation occurs last because
6124 // we want the methods from this container to override any methods
6125 // we've previously seen with the same selector.
Aaron Ballmanaff18c02014-03-13 19:03:34 +00006126 for (auto *M : Container->methods()) {
David Blaikie2d7c57e2012-04-30 02:36:29 +00006127 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006128 if (!ReturnType.isNull() &&
Alp Toker314cc812014-01-25 16:55:45 +00006129 !Context.hasSameUnqualifiedType(ReturnType, M->getReturnType()))
Douglas Gregor636a61e2010-04-07 00:21:17 +00006130 continue;
6131
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006132 KnownMethods[M->getSelector()] =
Aaron Ballmanaff18c02014-03-13 19:03:34 +00006133 KnownMethodsMap::mapped_type(M, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006134 }
6135 }
6136}
6137
Douglas Gregor669a25a2011-02-17 00:22:45 +00006138/// \brief Add the parenthesized return or parameter type chunk to a code
6139/// completion string.
6140static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor29979142012-04-10 18:35:07 +00006141 unsigned ObjCDeclQuals,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006142 ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006143 const PrintingPolicy &Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006144 CodeCompletionBuilder &Builder) {
6145 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor29979142012-04-10 18:35:07 +00006146 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6147 if (!Quals.empty())
6148 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor75acd922011-09-27 23:30:47 +00006149 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006150 Builder.getAllocator()));
6151 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6152}
6153
6154/// \brief Determine whether the given class is or inherits from a class by
6155/// the given name.
6156static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006157 StringRef Name) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006158 if (!Class)
6159 return false;
6160
6161 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6162 return true;
6163
6164 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6165}
6166
6167/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6168/// Key-Value Observing (KVO).
6169static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6170 bool IsInstanceMethod,
6171 QualType ReturnType,
6172 ASTContext &Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006173 VisitedSelectorSet &KnownSelectors,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006174 ResultBuilder &Results) {
6175 IdentifierInfo *PropName = Property->getIdentifier();
6176 if (!PropName || PropName->getLength() == 0)
6177 return;
6178
Douglas Gregor75acd922011-09-27 23:30:47 +00006179 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6180
Douglas Gregor669a25a2011-02-17 00:22:45 +00006181 // Builder that will create each code completion.
6182 typedef CodeCompletionResult Result;
6183 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006184 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor669a25a2011-02-17 00:22:45 +00006185
6186 // The selector table.
6187 SelectorTable &Selectors = Context.Selectors;
6188
6189 // The property name, copied into the code completion allocation region
6190 // on demand.
6191 struct KeyHolder {
6192 CodeCompletionAllocator &Allocator;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006193 StringRef Key;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006194 const char *CopiedKey;
Craig Topperc3ec1492014-05-26 06:22:03 +00006195
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006196 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Craig Topperc3ec1492014-05-26 06:22:03 +00006197 : Allocator(Allocator), Key(Key), CopiedKey(nullptr) {}
6198
Douglas Gregor669a25a2011-02-17 00:22:45 +00006199 operator const char *() {
6200 if (CopiedKey)
6201 return CopiedKey;
6202
6203 return CopiedKey = Allocator.CopyString(Key);
6204 }
6205 } Key(Allocator, PropName->getName());
6206
6207 // The uppercased name of the property name.
6208 std::string UpperKey = PropName->getName();
6209 if (!UpperKey.empty())
Jordan Rose4938f272013-02-09 10:09:43 +00006210 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006211
6212 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6213 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6214 Property->getType());
6215 bool ReturnTypeMatchesVoid
6216 = ReturnType.isNull() || ReturnType->isVoidType();
6217
6218 // Add the normal accessor -(type)key.
6219 if (IsInstanceMethod &&
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006220 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor669a25a2011-02-17 00:22:45 +00006221 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6222 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006223 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6224 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006225
6226 Builder.AddTypedTextChunk(Key);
6227 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6228 CXCursor_ObjCInstanceMethodDecl));
6229 }
6230
6231 // If we have an integral or boolean property (or the user has provided
6232 // an integral or boolean return type), add the accessor -(type)isKey.
6233 if (IsInstanceMethod &&
6234 ((!ReturnType.isNull() &&
6235 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6236 (ReturnType.isNull() &&
6237 (Property->getType()->isIntegerType() ||
6238 Property->getType()->isBooleanType())))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006239 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006240 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006241 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006242 if (ReturnType.isNull()) {
6243 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6244 Builder.AddTextChunk("BOOL");
6245 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6246 }
6247
6248 Builder.AddTypedTextChunk(
6249 Allocator.CopyString(SelectorId->getName()));
6250 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6251 CXCursor_ObjCInstanceMethodDecl));
6252 }
6253 }
6254
6255 // Add the normal mutator.
6256 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6257 !Property->getSetterMethodDecl()) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006258 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006259 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006260 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006261 if (ReturnType.isNull()) {
6262 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6263 Builder.AddTextChunk("void");
6264 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6265 }
6266
6267 Builder.AddTypedTextChunk(
6268 Allocator.CopyString(SelectorId->getName()));
6269 Builder.AddTypedTextChunk(":");
Douglas Gregor29979142012-04-10 18:35:07 +00006270 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6271 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006272 Builder.AddTextChunk(Key);
6273 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6274 CXCursor_ObjCInstanceMethodDecl));
6275 }
6276 }
6277
6278 // Indexed and unordered accessors
6279 unsigned IndexedGetterPriority = CCP_CodePattern;
6280 unsigned IndexedSetterPriority = CCP_CodePattern;
6281 unsigned UnorderedGetterPriority = CCP_CodePattern;
6282 unsigned UnorderedSetterPriority = CCP_CodePattern;
6283 if (const ObjCObjectPointerType *ObjCPointer
6284 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6285 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6286 // If this interface type is not provably derived from a known
6287 // collection, penalize the corresponding completions.
6288 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6289 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6290 if (!InheritsFromClassNamed(IFace, "NSArray"))
6291 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6292 }
6293
6294 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6295 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6296 if (!InheritsFromClassNamed(IFace, "NSSet"))
6297 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6298 }
6299 }
6300 } else {
6301 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6302 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6303 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6304 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6305 }
6306
6307 // Add -(NSUInteger)countOf<key>
6308 if (IsInstanceMethod &&
6309 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006310 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006311 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006312 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006313 if (ReturnType.isNull()) {
6314 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6315 Builder.AddTextChunk("NSUInteger");
6316 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6317 }
6318
6319 Builder.AddTypedTextChunk(
6320 Allocator.CopyString(SelectorId->getName()));
6321 Results.AddResult(Result(Builder.TakeString(),
6322 std::min(IndexedGetterPriority,
6323 UnorderedGetterPriority),
6324 CXCursor_ObjCInstanceMethodDecl));
6325 }
6326 }
6327
6328 // Indexed getters
6329 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6330 if (IsInstanceMethod &&
6331 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006332 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006333 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006334 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006335 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006336 if (ReturnType.isNull()) {
6337 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6338 Builder.AddTextChunk("id");
6339 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6340 }
6341
6342 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6343 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6344 Builder.AddTextChunk("NSUInteger");
6345 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6346 Builder.AddTextChunk("index");
6347 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6348 CXCursor_ObjCInstanceMethodDecl));
6349 }
6350 }
6351
6352 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6353 if (IsInstanceMethod &&
6354 (ReturnType.isNull() ||
6355 (ReturnType->isObjCObjectPointerType() &&
6356 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6357 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6358 ->getName() == "NSArray"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006359 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006360 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006361 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006362 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006363 if (ReturnType.isNull()) {
6364 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6365 Builder.AddTextChunk("NSArray *");
6366 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6367 }
6368
6369 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6370 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6371 Builder.AddTextChunk("NSIndexSet *");
6372 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6373 Builder.AddTextChunk("indexes");
6374 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6375 CXCursor_ObjCInstanceMethodDecl));
6376 }
6377 }
6378
6379 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6380 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006381 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006382 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006383 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006384 &Context.Idents.get("range")
6385 };
6386
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006387 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006388 if (ReturnType.isNull()) {
6389 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6390 Builder.AddTextChunk("void");
6391 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6392 }
6393
6394 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6395 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6396 Builder.AddPlaceholderChunk("object-type");
6397 Builder.AddTextChunk(" **");
6398 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6399 Builder.AddTextChunk("buffer");
6400 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6401 Builder.AddTypedTextChunk("range:");
6402 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6403 Builder.AddTextChunk("NSRange");
6404 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6405 Builder.AddTextChunk("inRange");
6406 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6407 CXCursor_ObjCInstanceMethodDecl));
6408 }
6409 }
6410
6411 // Mutable indexed accessors
6412
6413 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6414 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006415 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006416 IdentifierInfo *SelectorIds[2] = {
6417 &Context.Idents.get("insertObject"),
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006418 &Context.Idents.get(SelectorName)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006419 };
6420
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006421 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006422 if (ReturnType.isNull()) {
6423 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6424 Builder.AddTextChunk("void");
6425 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6426 }
6427
6428 Builder.AddTypedTextChunk("insertObject:");
6429 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6430 Builder.AddPlaceholderChunk("object-type");
6431 Builder.AddTextChunk(" *");
6432 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6433 Builder.AddTextChunk("object");
6434 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6435 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6436 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6437 Builder.AddPlaceholderChunk("NSUInteger");
6438 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6439 Builder.AddTextChunk("index");
6440 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6441 CXCursor_ObjCInstanceMethodDecl));
6442 }
6443 }
6444
6445 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6446 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006447 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006448 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006449 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006450 &Context.Idents.get("atIndexes")
6451 };
6452
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006453 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006454 if (ReturnType.isNull()) {
6455 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6456 Builder.AddTextChunk("void");
6457 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6458 }
6459
6460 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6461 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6462 Builder.AddTextChunk("NSArray *");
6463 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6464 Builder.AddTextChunk("array");
6465 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6466 Builder.AddTypedTextChunk("atIndexes:");
6467 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6468 Builder.AddPlaceholderChunk("NSIndexSet *");
6469 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6470 Builder.AddTextChunk("indexes");
6471 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6472 CXCursor_ObjCInstanceMethodDecl));
6473 }
6474 }
6475
6476 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6477 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006478 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006479 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006480 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006481 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006482 if (ReturnType.isNull()) {
6483 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6484 Builder.AddTextChunk("void");
6485 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6486 }
6487
6488 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6489 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6490 Builder.AddTextChunk("NSUInteger");
6491 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6492 Builder.AddTextChunk("index");
6493 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6494 CXCursor_ObjCInstanceMethodDecl));
6495 }
6496 }
6497
6498 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6499 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006500 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006501 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006502 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006503 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006504 if (ReturnType.isNull()) {
6505 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6506 Builder.AddTextChunk("void");
6507 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6508 }
6509
6510 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6511 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6512 Builder.AddTextChunk("NSIndexSet *");
6513 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6514 Builder.AddTextChunk("indexes");
6515 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6516 CXCursor_ObjCInstanceMethodDecl));
6517 }
6518 }
6519
6520 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6521 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006522 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006523 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006524 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006525 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006526 &Context.Idents.get("withObject")
6527 };
6528
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006529 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006530 if (ReturnType.isNull()) {
6531 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6532 Builder.AddTextChunk("void");
6533 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6534 }
6535
6536 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6537 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6538 Builder.AddPlaceholderChunk("NSUInteger");
6539 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6540 Builder.AddTextChunk("index");
6541 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6542 Builder.AddTypedTextChunk("withObject:");
6543 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6544 Builder.AddTextChunk("id");
6545 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6546 Builder.AddTextChunk("object");
6547 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6548 CXCursor_ObjCInstanceMethodDecl));
6549 }
6550 }
6551
6552 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6553 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006554 std::string SelectorName1
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006555 = (Twine("replace") + UpperKey + "AtIndexes").str();
6556 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006557 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006558 &Context.Idents.get(SelectorName1),
6559 &Context.Idents.get(SelectorName2)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006560 };
6561
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006562 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006563 if (ReturnType.isNull()) {
6564 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6565 Builder.AddTextChunk("void");
6566 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6567 }
6568
6569 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6570 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6571 Builder.AddPlaceholderChunk("NSIndexSet *");
6572 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6573 Builder.AddTextChunk("indexes");
6574 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6575 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6576 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6577 Builder.AddTextChunk("NSArray *");
6578 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6579 Builder.AddTextChunk("array");
6580 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6581 CXCursor_ObjCInstanceMethodDecl));
6582 }
6583 }
6584
6585 // Unordered getters
6586 // - (NSEnumerator *)enumeratorOfKey
6587 if (IsInstanceMethod &&
6588 (ReturnType.isNull() ||
6589 (ReturnType->isObjCObjectPointerType() &&
6590 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6591 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6592 ->getName() == "NSEnumerator"))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006593 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006594 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006595 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006596 if (ReturnType.isNull()) {
6597 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6598 Builder.AddTextChunk("NSEnumerator *");
6599 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6600 }
6601
6602 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6603 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6604 CXCursor_ObjCInstanceMethodDecl));
6605 }
6606 }
6607
6608 // - (type *)memberOfKey:(type *)object
6609 if (IsInstanceMethod &&
6610 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006611 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006612 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006613 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006614 if (ReturnType.isNull()) {
6615 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6616 Builder.AddPlaceholderChunk("object-type");
6617 Builder.AddTextChunk(" *");
6618 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6619 }
6620
6621 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6622 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6623 if (ReturnType.isNull()) {
6624 Builder.AddPlaceholderChunk("object-type");
6625 Builder.AddTextChunk(" *");
6626 } else {
6627 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006628 Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006629 Builder.getAllocator()));
6630 }
6631 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6632 Builder.AddTextChunk("object");
6633 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6634 CXCursor_ObjCInstanceMethodDecl));
6635 }
6636 }
6637
6638 // Mutable unordered accessors
6639 // - (void)addKeyObject:(type *)object
6640 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006641 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006642 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006643 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006644 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006645 if (ReturnType.isNull()) {
6646 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6647 Builder.AddTextChunk("void");
6648 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6649 }
6650
6651 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6652 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6653 Builder.AddPlaceholderChunk("object-type");
6654 Builder.AddTextChunk(" *");
6655 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6656 Builder.AddTextChunk("object");
6657 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6658 CXCursor_ObjCInstanceMethodDecl));
6659 }
6660 }
6661
6662 // - (void)addKey:(NSSet *)objects
6663 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006664 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006665 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006666 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006667 if (ReturnType.isNull()) {
6668 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6669 Builder.AddTextChunk("void");
6670 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6671 }
6672
6673 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6674 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6675 Builder.AddTextChunk("NSSet *");
6676 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6677 Builder.AddTextChunk("objects");
6678 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6679 CXCursor_ObjCInstanceMethodDecl));
6680 }
6681 }
6682
6683 // - (void)removeKeyObject:(type *)object
6684 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006685 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006686 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006687 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006688 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006689 if (ReturnType.isNull()) {
6690 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6691 Builder.AddTextChunk("void");
6692 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6693 }
6694
6695 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6696 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6697 Builder.AddPlaceholderChunk("object-type");
6698 Builder.AddTextChunk(" *");
6699 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6700 Builder.AddTextChunk("object");
6701 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6702 CXCursor_ObjCInstanceMethodDecl));
6703 }
6704 }
6705
6706 // - (void)removeKey:(NSSet *)objects
6707 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006708 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006709 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006710 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006711 if (ReturnType.isNull()) {
6712 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6713 Builder.AddTextChunk("void");
6714 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6715 }
6716
6717 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6718 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6719 Builder.AddTextChunk("NSSet *");
6720 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6721 Builder.AddTextChunk("objects");
6722 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6723 CXCursor_ObjCInstanceMethodDecl));
6724 }
6725 }
6726
6727 // - (void)intersectKey:(NSSet *)objects
6728 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006729 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006730 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006731 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006732 if (ReturnType.isNull()) {
6733 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6734 Builder.AddTextChunk("void");
6735 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6736 }
6737
6738 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6739 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6740 Builder.AddTextChunk("NSSet *");
6741 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6742 Builder.AddTextChunk("objects");
6743 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6744 CXCursor_ObjCInstanceMethodDecl));
6745 }
6746 }
6747
6748 // Key-Value Observing
6749 // + (NSSet *)keyPathsForValuesAffectingKey
6750 if (!IsInstanceMethod &&
6751 (ReturnType.isNull() ||
6752 (ReturnType->isObjCObjectPointerType() &&
6753 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6754 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6755 ->getName() == "NSSet"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006756 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006757 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006758 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006759 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006760 if (ReturnType.isNull()) {
6761 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6762 Builder.AddTextChunk("NSSet *");
6763 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6764 }
6765
6766 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6767 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor857bcda2011-06-02 04:02:27 +00006768 CXCursor_ObjCClassMethodDecl));
6769 }
6770 }
6771
6772 // + (BOOL)automaticallyNotifiesObserversForKey
6773 if (!IsInstanceMethod &&
6774 (ReturnType.isNull() ||
6775 ReturnType->isIntegerType() ||
6776 ReturnType->isBooleanType())) {
6777 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006778 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor857bcda2011-06-02 04:02:27 +00006779 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6780 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6781 if (ReturnType.isNull()) {
6782 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6783 Builder.AddTextChunk("BOOL");
6784 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6785 }
6786
6787 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6788 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6789 CXCursor_ObjCClassMethodDecl));
Douglas Gregor669a25a2011-02-17 00:22:45 +00006790 }
6791 }
6792}
6793
Douglas Gregor636a61e2010-04-07 00:21:17 +00006794void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6795 bool IsInstanceMethod,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006796 ParsedType ReturnTy) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006797 // Determine the return type of the method we're declaring, if
6798 // provided.
6799 QualType ReturnType = GetTypeFromParser(ReturnTy);
Craig Topperc3ec1492014-05-26 06:22:03 +00006800 Decl *IDecl = nullptr;
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006801 if (CurContext->isObjCContainer()) {
6802 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6803 IDecl = cast<Decl>(OCD);
6804 }
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006805 // Determine where we should start searching for methods.
Craig Topperc3ec1492014-05-26 06:22:03 +00006806 ObjCContainerDecl *SearchDecl = nullptr;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006807 bool IsInImplementation = false;
John McCall48871652010-08-21 09:40:31 +00006808 if (Decl *D = IDecl) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006809 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6810 SearchDecl = Impl->getClassInterface();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006811 IsInImplementation = true;
6812 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006813 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006814 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006815 IsInImplementation = true;
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006816 } else
Douglas Gregor636a61e2010-04-07 00:21:17 +00006817 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006818 }
6819
6820 if (!SearchDecl && S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00006821 if (DeclContext *DC = S->getEntity())
Douglas Gregor636a61e2010-04-07 00:21:17 +00006822 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006823 }
6824
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006825 if (!SearchDecl) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006826 HandleCodeCompleteResults(this, CodeCompleter,
6827 CodeCompletionContext::CCC_Other,
Craig Topperc3ec1492014-05-26 06:22:03 +00006828 nullptr, 0);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006829 return;
6830 }
6831
6832 // Find all of the methods that we could declare/implement here.
6833 KnownMethodsMap KnownMethods;
6834 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006835 ReturnType, KnownMethods);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006836
Douglas Gregor636a61e2010-04-07 00:21:17 +00006837 // Add declarations or definitions for each of the known methods.
John McCall276321a2010-08-25 06:19:51 +00006838 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006839 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006840 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006841 CodeCompletionContext::CCC_Other);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006842 Results.EnterNewScope();
Douglas Gregor75acd922011-09-27 23:30:47 +00006843 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006844 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6845 MEnd = KnownMethods.end();
6846 M != MEnd; ++M) {
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006847 ObjCMethodDecl *Method = M->second.getPointer();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006848 CodeCompletionBuilder Builder(Results.getAllocator(),
6849 Results.getCodeCompletionTUInfo());
Douglas Gregor636a61e2010-04-07 00:21:17 +00006850
6851 // If the result type was not already provided, add it to the
6852 // pattern as (type).
Douglas Gregor669a25a2011-02-17 00:22:45 +00006853 if (ReturnType.isNull())
Alp Toker314cc812014-01-25 16:55:45 +00006854 AddObjCPassingTypeChunk(Method->getReturnType(),
6855 Method->getObjCDeclQualifier(), Context, Policy,
6856 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006857
6858 Selector Sel = Method->getSelector();
6859
6860 // Add the first part of the selector to the pattern.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006861 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006862 Sel.getNameForSlot(0)));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006863
6864 // Add parameters to the pattern.
6865 unsigned I = 0;
6866 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6867 PEnd = Method->param_end();
6868 P != PEnd; (void)++P, ++I) {
6869 // Add the part of the selector name.
6870 if (I == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006871 Builder.AddTypedTextChunk(":");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006872 else if (I < Sel.getNumArgs()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006873 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6874 Builder.AddTypedTextChunk(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006875 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006876 } else
6877 break;
6878
6879 // Add the parameter type.
Douglas Gregor29979142012-04-10 18:35:07 +00006880 AddObjCPassingTypeChunk((*P)->getOriginalType(),
6881 (*P)->getObjCDeclQualifier(),
6882 Context, Policy,
Douglas Gregor75acd922011-09-27 23:30:47 +00006883 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006884
6885 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006886 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006887 }
6888
6889 if (Method->isVariadic()) {
6890 if (Method->param_size() > 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006891 Builder.AddChunk(CodeCompletionString::CK_Comma);
6892 Builder.AddTextChunk("...");
Douglas Gregor400f5972010-08-31 05:13:43 +00006893 }
Douglas Gregor636a61e2010-04-07 00:21:17 +00006894
Douglas Gregord37c59d2010-05-28 00:57:46 +00006895 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006896 // We will be defining the method here, so add a compound statement.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006897 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6898 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6899 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Alp Toker314cc812014-01-25 16:55:45 +00006900 if (!Method->getReturnType()->isVoidType()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006901 // If the result type is not void, add a return clause.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006902 Builder.AddTextChunk("return");
6903 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6904 Builder.AddPlaceholderChunk("expression");
6905 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006906 } else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006907 Builder.AddPlaceholderChunk("statements");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006908
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006909 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6910 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006911 }
6912
Douglas Gregor416b5752010-08-25 01:08:01 +00006913 unsigned Priority = CCP_CodePattern;
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006914 if (!M->second.getInt())
Douglas Gregor416b5752010-08-25 01:08:01 +00006915 Priority += CCD_InBaseClass;
6916
Douglas Gregor78254c82012-03-27 23:34:16 +00006917 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006918 }
6919
Douglas Gregor669a25a2011-02-17 00:22:45 +00006920 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
6921 // the properties in this class and its categories.
David Blaikiebbafb8a2012-03-11 07:00:24 +00006922 if (Context.getLangOpts().ObjC2) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006923 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006924 Containers.push_back(SearchDecl);
6925
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006926 VisitedSelectorSet KnownSelectors;
6927 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6928 MEnd = KnownMethods.end();
6929 M != MEnd; ++M)
6930 KnownSelectors.insert(M->first);
6931
6932
Douglas Gregor669a25a2011-02-17 00:22:45 +00006933 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
6934 if (!IFace)
6935 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
6936 IFace = Category->getClassInterface();
6937
Aaron Ballman3fe486a2014-03-13 21:23:55 +00006938 if (IFace)
6939 for (auto *Cat : IFace->visible_categories())
6940 Containers.push_back(Cat);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006941
Aaron Ballmandc4bea42014-03-13 18:47:37 +00006942 for (unsigned I = 0, N = Containers.size(); I != N; ++I)
Aaron Ballmand174edf2014-03-13 19:11:50 +00006943 for (auto *P : Containers[I]->properties())
Aaron Ballmandc4bea42014-03-13 18:47:37 +00006944 AddObjCKeyValueCompletions(P, IsInstanceMethod, ReturnType, Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006945 KnownSelectors, Results);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006946 }
6947
Douglas Gregor636a61e2010-04-07 00:21:17 +00006948 Results.ExitScope();
6949
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006950 HandleCodeCompleteResults(this, CodeCompleter,
6951 CodeCompletionContext::CCC_Other,
6952 Results.data(),Results.size());
Douglas Gregor636a61e2010-04-07 00:21:17 +00006953}
Douglas Gregor95887f92010-07-08 23:20:03 +00006954
6955void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
6956 bool IsInstanceMethod,
Douglas Gregor45879692010-07-08 23:37:41 +00006957 bool AtParameterName,
John McCallba7bf592010-08-24 05:47:05 +00006958 ParsedType ReturnTy,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00006959 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor95887f92010-07-08 23:20:03 +00006960 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00006961 // pool from the AST file.
Douglas Gregor95887f92010-07-08 23:20:03 +00006962 if (ExternalSource) {
6963 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
6964 I != N; ++I) {
6965 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00006966 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor95887f92010-07-08 23:20:03 +00006967 continue;
Sebastian Redl75d8a322010-08-02 23:18:59 +00006968
6969 ReadMethodPool(Sel);
Douglas Gregor95887f92010-07-08 23:20:03 +00006970 }
6971 }
6972
6973 // Build the set of methods we can see.
John McCall276321a2010-08-25 06:19:51 +00006974 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006975 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006976 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006977 CodeCompletionContext::CCC_Other);
Douglas Gregor95887f92010-07-08 23:20:03 +00006978
6979 if (ReturnTy)
6980 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redl75d8a322010-08-02 23:18:59 +00006981
Douglas Gregor95887f92010-07-08 23:20:03 +00006982 Results.EnterNewScope();
Sebastian Redl75d8a322010-08-02 23:18:59 +00006983 for (GlobalMethodPool::iterator M = MethodPool.begin(),
6984 MEnd = MethodPool.end();
6985 M != MEnd; ++M) {
6986 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
6987 &M->second.second;
6988 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00006989 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00006990 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor95887f92010-07-08 23:20:03 +00006991 continue;
6992
Douglas Gregor45879692010-07-08 23:37:41 +00006993 if (AtParameterName) {
6994 // Suggest parameter names we've seen before.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00006995 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor45879692010-07-08 23:37:41 +00006996 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
6997 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
6998 if (Param->getIdentifier()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006999 CodeCompletionBuilder Builder(Results.getAllocator(),
7000 Results.getCodeCompletionTUInfo());
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007001 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007002 Param->getIdentifier()->getName()));
7003 Results.AddResult(Builder.TakeString());
Douglas Gregor45879692010-07-08 23:37:41 +00007004 }
7005 }
7006
7007 continue;
7008 }
Craig Topperc3ec1492014-05-26 06:22:03 +00007009
7010 Result R(MethList->Method, Results.getBasePriority(MethList->Method),
7011 nullptr);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007012 R.StartParameter = SelIdents.size();
Douglas Gregor95887f92010-07-08 23:20:03 +00007013 R.AllParametersAreInformative = false;
7014 R.DeclaringEntity = true;
7015 Results.MaybeAddResult(R, CurContext);
7016 }
7017 }
7018
7019 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007020 HandleCodeCompleteResults(this, CodeCompleter,
7021 CodeCompletionContext::CCC_Other,
7022 Results.data(),Results.size());
Douglas Gregor95887f92010-07-08 23:20:03 +00007023}
Douglas Gregorb14904c2010-08-13 22:48:40 +00007024
Douglas Gregorec00a262010-08-24 22:20:20 +00007025void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007026 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007027 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007028 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007029 Results.EnterNewScope();
7030
7031 // #if <condition>
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007032 CodeCompletionBuilder Builder(Results.getAllocator(),
7033 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007034 Builder.AddTypedTextChunk("if");
7035 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7036 Builder.AddPlaceholderChunk("condition");
7037 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007038
7039 // #ifdef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007040 Builder.AddTypedTextChunk("ifdef");
7041 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7042 Builder.AddPlaceholderChunk("macro");
7043 Results.AddResult(Builder.TakeString());
7044
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007045 // #ifndef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007046 Builder.AddTypedTextChunk("ifndef");
7047 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7048 Builder.AddPlaceholderChunk("macro");
7049 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007050
7051 if (InConditional) {
7052 // #elif <condition>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007053 Builder.AddTypedTextChunk("elif");
7054 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7055 Builder.AddPlaceholderChunk("condition");
7056 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007057
7058 // #else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007059 Builder.AddTypedTextChunk("else");
7060 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007061
7062 // #endif
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007063 Builder.AddTypedTextChunk("endif");
7064 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007065 }
7066
7067 // #include "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007068 Builder.AddTypedTextChunk("include");
7069 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7070 Builder.AddTextChunk("\"");
7071 Builder.AddPlaceholderChunk("header");
7072 Builder.AddTextChunk("\"");
7073 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007074
7075 // #include <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007076 Builder.AddTypedTextChunk("include");
7077 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7078 Builder.AddTextChunk("<");
7079 Builder.AddPlaceholderChunk("header");
7080 Builder.AddTextChunk(">");
7081 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007082
7083 // #define <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007084 Builder.AddTypedTextChunk("define");
7085 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7086 Builder.AddPlaceholderChunk("macro");
7087 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007088
7089 // #define <macro>(<args>)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007090 Builder.AddTypedTextChunk("define");
7091 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7092 Builder.AddPlaceholderChunk("macro");
7093 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7094 Builder.AddPlaceholderChunk("args");
7095 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7096 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007097
7098 // #undef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007099 Builder.AddTypedTextChunk("undef");
7100 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7101 Builder.AddPlaceholderChunk("macro");
7102 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007103
7104 // #line <number>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007105 Builder.AddTypedTextChunk("line");
7106 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7107 Builder.AddPlaceholderChunk("number");
7108 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007109
7110 // #line <number> "filename"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007111 Builder.AddTypedTextChunk("line");
7112 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7113 Builder.AddPlaceholderChunk("number");
7114 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7115 Builder.AddTextChunk("\"");
7116 Builder.AddPlaceholderChunk("filename");
7117 Builder.AddTextChunk("\"");
7118 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007119
7120 // #error <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007121 Builder.AddTypedTextChunk("error");
7122 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7123 Builder.AddPlaceholderChunk("message");
7124 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007125
7126 // #pragma <arguments>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007127 Builder.AddTypedTextChunk("pragma");
7128 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7129 Builder.AddPlaceholderChunk("arguments");
7130 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007131
David Blaikiebbafb8a2012-03-11 07:00:24 +00007132 if (getLangOpts().ObjC1) {
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007133 // #import "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007134 Builder.AddTypedTextChunk("import");
7135 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7136 Builder.AddTextChunk("\"");
7137 Builder.AddPlaceholderChunk("header");
7138 Builder.AddTextChunk("\"");
7139 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007140
7141 // #import <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007142 Builder.AddTypedTextChunk("import");
7143 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7144 Builder.AddTextChunk("<");
7145 Builder.AddPlaceholderChunk("header");
7146 Builder.AddTextChunk(">");
7147 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007148 }
7149
7150 // #include_next "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007151 Builder.AddTypedTextChunk("include_next");
7152 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7153 Builder.AddTextChunk("\"");
7154 Builder.AddPlaceholderChunk("header");
7155 Builder.AddTextChunk("\"");
7156 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007157
7158 // #include_next <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007159 Builder.AddTypedTextChunk("include_next");
7160 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7161 Builder.AddTextChunk("<");
7162 Builder.AddPlaceholderChunk("header");
7163 Builder.AddTextChunk(">");
7164 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007165
7166 // #warning <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007167 Builder.AddTypedTextChunk("warning");
7168 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7169 Builder.AddPlaceholderChunk("message");
7170 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007171
7172 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7173 // completions for them. And __include_macros is a Clang-internal extension
7174 // that we don't want to encourage anyone to use.
7175
7176 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7177 Results.ExitScope();
7178
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007179 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0de55ce2010-08-25 18:41:16 +00007180 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007181 Results.data(), Results.size());
7182}
7183
7184void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorec00a262010-08-24 22:20:20 +00007185 CodeCompleteOrdinaryName(S,
John McCallfaf5fb42010-08-26 23:41:50 +00007186 S->getFnParent()? Sema::PCC_RecoveryInFunction
7187 : Sema::PCC_Namespace);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007188}
7189
Douglas Gregorec00a262010-08-24 22:20:20 +00007190void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007191 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007192 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007193 IsDefinition? CodeCompletionContext::CCC_MacroName
7194 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor12785102010-08-24 20:21:13 +00007195 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7196 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007197 CodeCompletionBuilder Builder(Results.getAllocator(),
7198 Results.getCodeCompletionTUInfo());
Douglas Gregor12785102010-08-24 20:21:13 +00007199 Results.EnterNewScope();
7200 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7201 MEnd = PP.macro_end();
7202 M != MEnd; ++M) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007203 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007204 M->first->getName()));
Argyrios Kyrtzidis5c8b1cd2012-09-27 00:24:09 +00007205 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7206 CCP_CodePattern,
7207 CXCursor_MacroDefinition));
Douglas Gregor12785102010-08-24 20:21:13 +00007208 }
7209 Results.ExitScope();
7210 } else if (IsDefinition) {
7211 // FIXME: Can we detect when the user just wrote an include guard above?
7212 }
7213
Douglas Gregor0ac41382010-09-23 23:01:17 +00007214 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor12785102010-08-24 20:21:13 +00007215 Results.data(), Results.size());
7216}
7217
Douglas Gregorec00a262010-08-24 22:20:20 +00007218void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007219 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007220 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007221 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorec00a262010-08-24 22:20:20 +00007222
7223 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007224 AddMacroResults(PP, Results, true);
Douglas Gregorec00a262010-08-24 22:20:20 +00007225
7226 // defined (<macro>)
7227 Results.EnterNewScope();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007228 CodeCompletionBuilder Builder(Results.getAllocator(),
7229 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007230 Builder.AddTypedTextChunk("defined");
7231 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7232 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7233 Builder.AddPlaceholderChunk("macro");
7234 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7235 Results.AddResult(Builder.TakeString());
Douglas Gregorec00a262010-08-24 22:20:20 +00007236 Results.ExitScope();
7237
7238 HandleCodeCompleteResults(this, CodeCompleter,
7239 CodeCompletionContext::CCC_PreprocessorExpression,
7240 Results.data(), Results.size());
7241}
7242
7243void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7244 IdentifierInfo *Macro,
7245 MacroInfo *MacroInfo,
7246 unsigned Argument) {
7247 // FIXME: In the future, we could provide "overload" results, much like we
7248 // do for function calls.
7249
Argyrios Kyrtzidis75f6cd22011-08-18 19:41:28 +00007250 // Now just ignore this. There will be another code-completion callback
7251 // for the expanded tokens.
Douglas Gregorec00a262010-08-24 22:20:20 +00007252}
7253
Douglas Gregor11583702010-08-25 17:04:25 +00007254void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor11583702010-08-25 17:04:25 +00007255 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorea736372010-08-25 17:10:00 +00007256 CodeCompletionContext::CCC_NaturalLanguage,
Craig Topperc3ec1492014-05-26 06:22:03 +00007257 nullptr, 0);
Douglas Gregor11583702010-08-25 17:04:25 +00007258}
7259
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007260void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007261 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00007262 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007263 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7264 CodeCompletionContext::CCC_Recovery);
Douglas Gregor39982192010-08-15 06:18:01 +00007265 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7266 CodeCompletionDeclConsumer Consumer(Builder,
7267 Context.getTranslationUnitDecl());
7268 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7269 Consumer);
7270 }
Douglas Gregorb14904c2010-08-13 22:48:40 +00007271
7272 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007273 AddMacroResults(PP, Builder, true);
Douglas Gregorb14904c2010-08-13 22:48:40 +00007274
7275 Results.clear();
7276 Results.insert(Results.end(),
7277 Builder.data(), Builder.data() + Builder.size());
7278}