blob: 73a758db6434908b95729dedcdfaed24a6ea9420 [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;
107 DeclOrVector = ((NamedDecl *)0);
108 }
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,
175 LookupFilter Filter = 0)
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),
180 ObjCImplementation(0)
181 {
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 }
215
Douglas Gregor3545ff42009-09-21 16:56:56 +0000216 Result *data() { return Results.empty()? 0 : &Results.front(); }
217 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.
Douglas Gregor2af2f672009-09-21 20:12:40 +0000292 void MaybeAddResult(Result R, DeclContext *CurContext = 0);
Douglas Gregor3545ff42009-09-21 16:56:56 +0000293
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 };
367
368 iterator() : DeclOrIterator((NamedDecl *)0), SingleDeclIndex(0) { }
369
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 *>()) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000378 DeclOrIterator = (NamedDecl *)0;
379 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 }
464
465 NestedNameSpecifier *Result = 0;
466 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 &&
533 Filter != 0))
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;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000663 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND))
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 FunctionTemplateDecl *FunTmpl =
668 dyn_cast<FunctionTemplateDecl>(ND))
Douglas Gregor603d81b2010-07-13 08:18:22 +0000669 T = FunTmpl->getTemplatedDecl()->getCallResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000670 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000671 T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext()));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000672 else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000673 T = Property->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000674 else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000675 T = Value->getType();
676 else
677 return QualType();
Douglas Gregoraf670a82011-04-14 20:33:34 +0000678
679 // Dig through references, function pointers, and block pointers to
680 // get down to the likely type of an expression when the entity is
681 // used.
682 do {
683 if (const ReferenceType *Ref = T->getAs<ReferenceType>()) {
684 T = Ref->getPointeeType();
685 continue;
686 }
687
688 if (const PointerType *Pointer = T->getAs<PointerType>()) {
689 if (Pointer->getPointeeType()->isFunctionType()) {
690 T = Pointer->getPointeeType();
691 continue;
692 }
693
694 break;
695 }
696
697 if (const BlockPointerType *Block = T->getAs<BlockPointerType>()) {
698 T = Block->getPointeeType();
699 continue;
700 }
701
702 if (const FunctionType *Function = T->getAs<FunctionType>()) {
703 T = Function->getResultType();
704 continue;
705 }
706
707 break;
708 } while (true);
709
710 return T;
Douglas Gregor95887f92010-07-08 23:20:03 +0000711}
712
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000713unsigned ResultBuilder::getBasePriority(const NamedDecl *ND) {
714 if (!ND)
715 return CCP_Unlikely;
716
717 // Context-based decisions.
Richard Smith541b38b2013-09-20 01:15:31 +0000718 const DeclContext *LexicalDC = ND->getLexicalDeclContext();
719 if (LexicalDC->isFunctionOrMethod()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000720 // _cmd is relatively rare
721 if (const ImplicitParamDecl *ImplicitParam =
722 dyn_cast<ImplicitParamDecl>(ND))
723 if (ImplicitParam->getIdentifier() &&
724 ImplicitParam->getIdentifier()->isStr("_cmd"))
725 return CCP_ObjC_cmd;
726
727 return CCP_LocalDeclaration;
728 }
Richard Smith541b38b2013-09-20 01:15:31 +0000729
730 const DeclContext *DC = ND->getDeclContext()->getRedeclContext();
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000731 if (DC->isRecord() || isa<ObjCContainerDecl>(DC))
732 return CCP_MemberDeclaration;
733
734 // Content-based decisions.
735 if (isa<EnumConstantDecl>(ND))
736 return CCP_Constant;
737
Douglas Gregor52e0de42013-01-31 05:03:46 +0000738 // Use CCP_Type for type declarations unless we're in a statement, Objective-C
739 // message receiver, or parenthesized expression context. There, it's as
740 // likely that the user will want to write a type as other declarations.
741 if ((isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND)) &&
742 !(CompletionContext.getKind() == CodeCompletionContext::CCC_Statement ||
743 CompletionContext.getKind()
744 == CodeCompletionContext::CCC_ObjCMessageReceiver ||
745 CompletionContext.getKind()
746 == CodeCompletionContext::CCC_ParenthesizedExpression))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000747 return CCP_Type;
748
749 return CCP_Declaration;
750}
751
Douglas Gregor50832e02010-09-20 22:39:41 +0000752void ResultBuilder::AdjustResultPriorityForDecl(Result &R) {
753 // If this is an Objective-C method declaration whose selector matches our
754 // preferred selector, give it a priority boost.
755 if (!PreferredSelector.isNull())
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000756 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(R.Declaration))
Douglas Gregor50832e02010-09-20 22:39:41 +0000757 if (PreferredSelector == Method->getSelector())
758 R.Priority += CCD_SelectorMatch;
Douglas Gregor5fb901d2010-09-20 23:11:55 +0000759
Douglas Gregor50832e02010-09-20 22:39:41 +0000760 // If we have a preferred type, adjust the priority for results with exactly-
761 // matching or nearly-matching types.
762 if (!PreferredType.isNull()) {
763 QualType T = getDeclUsageType(SemaRef.Context, R.Declaration);
764 if (!T.isNull()) {
765 CanQualType TC = SemaRef.Context.getCanonicalType(T);
766 // Check for exactly-matching types (modulo qualifiers).
767 if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC))
768 R.Priority /= CCF_ExactTypeMatch;
769 // Check for nearly-matching types, based on classification of each.
770 else if ((getSimplifiedTypeClass(PreferredType)
Douglas Gregor95887f92010-07-08 23:20:03 +0000771 == getSimplifiedTypeClass(TC)) &&
Douglas Gregor50832e02010-09-20 22:39:41 +0000772 !(PreferredType->isEnumeralType() && TC->isEnumeralType()))
773 R.Priority /= CCF_SimilarTypeMatch;
774 }
775 }
Douglas Gregor95887f92010-07-08 23:20:03 +0000776}
777
Douglas Gregor0212fd72010-09-21 16:06:22 +0000778void ResultBuilder::MaybeAddConstructorResults(Result R) {
David Blaikiebbafb8a2012-03-11 07:00:24 +0000779 if (!SemaRef.getLangOpts().CPlusPlus || !R.Declaration ||
Douglas Gregor0212fd72010-09-21 16:06:22 +0000780 !CompletionContext.wantConstructorResults())
781 return;
782
783 ASTContext &Context = SemaRef.Context;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000784 const NamedDecl *D = R.Declaration;
785 const CXXRecordDecl *Record = 0;
786 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D))
Douglas Gregor0212fd72010-09-21 16:06:22 +0000787 Record = ClassTemplate->getTemplatedDecl();
788 else if ((Record = dyn_cast<CXXRecordDecl>(D))) {
789 // Skip specializations and partial specializations.
790 if (isa<ClassTemplateSpecializationDecl>(Record))
791 return;
792 } else {
793 // There are no constructors here.
794 return;
795 }
796
797 Record = Record->getDefinition();
798 if (!Record)
799 return;
800
801
802 QualType RecordTy = Context.getTypeDeclType(Record);
803 DeclarationName ConstructorName
804 = Context.DeclarationNames.getCXXConstructorName(
805 Context.getCanonicalType(RecordTy));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000806 DeclContext::lookup_const_result Ctors = Record->lookup(ConstructorName);
807 for (DeclContext::lookup_const_iterator I = Ctors.begin(),
808 E = Ctors.end();
809 I != E; ++I) {
David Blaikieff7d47a2012-12-19 00:45:41 +0000810 R.Declaration = *I;
Douglas Gregor0212fd72010-09-21 16:06:22 +0000811 R.CursorKind = getCursorKindForDecl(R.Declaration);
812 Results.push_back(R);
813 }
814}
815
Douglas Gregor7c208612010-01-14 00:20:49 +0000816void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) {
817 assert(!ShadowMaps.empty() && "Must enter into a results scope");
818
819 if (R.Kind != Result::RK_Declaration) {
820 // For non-declaration results, just add the result.
821 Results.push_back(R);
822 return;
823 }
824
825 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000826 if (const UsingShadowDecl *Using =
827 dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000828 MaybeAddResult(Result(Using->getTargetDecl(),
829 getBasePriority(Using->getTargetDecl()),
830 R.Qualifier),
831 CurContext);
Douglas Gregor7c208612010-01-14 00:20:49 +0000832 return;
833 }
834
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000835 const Decl *CanonDecl = R.Declaration->getCanonicalDecl();
Douglas Gregor7c208612010-01-14 00:20:49 +0000836 unsigned IDNS = CanonDecl->getIdentifierNamespace();
837
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000838 bool AsNestedNameSpecifier = false;
839 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregor7c208612010-01-14 00:20:49 +0000840 return;
841
Douglas Gregor0212fd72010-09-21 16:06:22 +0000842 // C++ constructors are never found by name lookup.
843 if (isa<CXXConstructorDecl>(R.Declaration))
844 return;
845
Douglas Gregor3545ff42009-09-21 16:56:56 +0000846 ShadowMap &SMap = ShadowMaps.back();
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000847 ShadowMapEntry::iterator I, IEnd;
848 ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName());
849 if (NamePos != SMap.end()) {
850 I = NamePos->second.begin();
851 IEnd = NamePos->second.end();
852 }
853
854 for (; I != IEnd; ++I) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000855 const NamedDecl *ND = I->first;
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000856 unsigned Index = I->second;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000857 if (ND->getCanonicalDecl() == CanonDecl) {
858 // This is a redeclaration. Always pick the newer declaration.
Douglas Gregor3545ff42009-09-21 16:56:56 +0000859 Results[Index].Declaration = R.Declaration;
860
Douglas Gregor3545ff42009-09-21 16:56:56 +0000861 // We're done.
862 return;
863 }
864 }
865
866 // This is a new declaration in this scope. However, check whether this
867 // declaration name is hidden by a similarly-named declaration in an outer
868 // scope.
869 std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end();
870 --SMEnd;
871 for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000872 ShadowMapEntry::iterator I, IEnd;
873 ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName());
874 if (NamePos != SM->end()) {
875 I = NamePos->second.begin();
876 IEnd = NamePos->second.end();
877 }
878 for (; I != IEnd; ++I) {
Douglas Gregor3545ff42009-09-21 16:56:56 +0000879 // A tag declaration does not hide a non-tag declaration.
John McCalle87beb22010-04-23 18:46:30 +0000880 if (I->first->hasTagIdentifierNamespace() &&
Richard Smith541b38b2013-09-20 01:15:31 +0000881 (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary |
882 Decl::IDNS_LocalExtern | Decl::IDNS_ObjCProtocol)))
Douglas Gregor3545ff42009-09-21 16:56:56 +0000883 continue;
884
885 // Protocols are in distinct namespaces from everything else.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000886 if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000887 || (IDNS & Decl::IDNS_ObjCProtocol)) &&
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000888 I->first->getIdentifierNamespace() != IDNS)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000889 continue;
890
891 // The newly-added result is hidden by an entry in the shadow map.
Douglas Gregore0717ab2010-01-14 00:41:07 +0000892 if (CheckHiddenResult(R, CurContext, I->first))
Douglas Gregor3545ff42009-09-21 16:56:56 +0000893 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000894
895 break;
896 }
897 }
898
899 // Make sure that any given declaration only shows up in the result set once.
900 if (!AllDeclsFound.insert(CanonDecl))
901 return;
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000902
Douglas Gregore412a5a2009-09-23 22:26:46 +0000903 // If the filter is for nested-name-specifiers, then this result starts a
904 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000905 if (AsNestedNameSpecifier) {
Douglas Gregore412a5a2009-09-23 22:26:46 +0000906 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000907 R.Priority = CCP_NestedNameSpecifier;
Douglas Gregor50832e02010-09-20 22:39:41 +0000908 } else
909 AdjustResultPriorityForDecl(R);
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000910
Douglas Gregor5bf52692009-09-22 23:15:58 +0000911 // If this result is supposed to have an informative qualifier, add one.
Douglas Gregore412a5a2009-09-23 22:26:46 +0000912 if (R.QualifierIsInformative && !R.Qualifier &&
913 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000914 const DeclContext *Ctx = R.Declaration->getDeclContext();
915 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregor5bf52692009-09-22 23:15:58 +0000916 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000917 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregor5bf52692009-09-22 23:15:58 +0000918 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
919 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
920 else
921 R.QualifierIsInformative = false;
922 }
Douglas Gregore412a5a2009-09-23 22:26:46 +0000923
Douglas Gregor3545ff42009-09-21 16:56:56 +0000924 // Insert this result into the set of results and into the current shadow
925 // map.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000926 SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size());
Douglas Gregor3545ff42009-09-21 16:56:56 +0000927 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +0000928
929 if (!AsNestedNameSpecifier)
930 MaybeAddConstructorResults(R);
Douglas Gregor3545ff42009-09-21 16:56:56 +0000931}
932
Douglas Gregorc580c522010-01-14 01:09:38 +0000933void ResultBuilder::AddResult(Result R, DeclContext *CurContext,
Douglas Gregor09bbc652010-01-14 15:47:35 +0000934 NamedDecl *Hiding, bool InBaseClass = false) {
Douglas Gregor78a21012010-01-14 16:01:26 +0000935 if (R.Kind != Result::RK_Declaration) {
936 // For non-declaration results, just add the result.
937 Results.push_back(R);
938 return;
939 }
940
Douglas Gregorc580c522010-01-14 01:09:38 +0000941 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000942 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000943 AddResult(Result(Using->getTargetDecl(),
944 getBasePriority(Using->getTargetDecl()),
945 R.Qualifier),
946 CurContext, Hiding);
Douglas Gregorc580c522010-01-14 01:09:38 +0000947 return;
948 }
949
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000950 bool AsNestedNameSpecifier = false;
951 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregorc580c522010-01-14 01:09:38 +0000952 return;
953
Douglas Gregor0212fd72010-09-21 16:06:22 +0000954 // C++ constructors are never found by name lookup.
955 if (isa<CXXConstructorDecl>(R.Declaration))
956 return;
957
Douglas Gregorc580c522010-01-14 01:09:38 +0000958 if (Hiding && CheckHiddenResult(R, CurContext, Hiding))
959 return;
Nick Lewyckyc3921482012-04-03 21:44:08 +0000960
Douglas Gregorc580c522010-01-14 01:09:38 +0000961 // Make sure that any given declaration only shows up in the result set once.
962 if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl()))
963 return;
964
965 // If the filter is for nested-name-specifiers, then this result starts a
966 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000967 if (AsNestedNameSpecifier) {
Douglas Gregorc580c522010-01-14 01:09:38 +0000968 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000969 R.Priority = CCP_NestedNameSpecifier;
970 }
Douglas Gregor09bbc652010-01-14 15:47:35 +0000971 else if (Filter == &ResultBuilder::IsMember && !R.Qualifier && InBaseClass &&
972 isa<CXXRecordDecl>(R.Declaration->getDeclContext()
Sebastian Redl50c68252010-08-31 00:36:30 +0000973 ->getRedeclContext()))
Douglas Gregor09bbc652010-01-14 15:47:35 +0000974 R.QualifierIsInformative = true;
975
Douglas Gregorc580c522010-01-14 01:09:38 +0000976 // If this result is supposed to have an informative qualifier, add one.
977 if (R.QualifierIsInformative && !R.Qualifier &&
978 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000979 const DeclContext *Ctx = R.Declaration->getDeclContext();
980 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregorc580c522010-01-14 01:09:38 +0000981 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000982 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregorc580c522010-01-14 01:09:38 +0000983 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000984 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregorc580c522010-01-14 01:09:38 +0000985 else
986 R.QualifierIsInformative = false;
987 }
988
Douglas Gregora2db7932010-05-26 22:00:08 +0000989 // Adjust the priority if this result comes from a base class.
990 if (InBaseClass)
991 R.Priority += CCD_InBaseClass;
992
Douglas Gregor50832e02010-09-20 22:39:41 +0000993 AdjustResultPriorityForDecl(R);
Douglas Gregor7aa6b222010-05-30 01:49:25 +0000994
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000995 if (HasObjectTypeQualifiers)
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000996 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(R.Declaration))
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000997 if (Method->isInstance()) {
998 Qualifiers MethodQuals
999 = Qualifiers::fromCVRMask(Method->getTypeQualifiers());
1000 if (ObjectTypeQualifiers == MethodQuals)
1001 R.Priority += CCD_ObjectQualifierMatch;
1002 else if (ObjectTypeQualifiers - MethodQuals) {
1003 // The method cannot be invoked, because doing so would drop
1004 // qualifiers.
1005 return;
1006 }
1007 }
1008
Douglas Gregorc580c522010-01-14 01:09:38 +00001009 // Insert this result into the set of results.
1010 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +00001011
1012 if (!AsNestedNameSpecifier)
1013 MaybeAddConstructorResults(R);
Douglas Gregorc580c522010-01-14 01:09:38 +00001014}
1015
Douglas Gregor78a21012010-01-14 16:01:26 +00001016void ResultBuilder::AddResult(Result R) {
1017 assert(R.Kind != Result::RK_Declaration &&
1018 "Declaration results need more context");
1019 Results.push_back(R);
1020}
1021
Douglas Gregor3545ff42009-09-21 16:56:56 +00001022/// \brief Enter into a new scope.
1023void ResultBuilder::EnterNewScope() {
1024 ShadowMaps.push_back(ShadowMap());
1025}
1026
1027/// \brief Exit from the current scope.
1028void ResultBuilder::ExitScope() {
Douglas Gregor05e7ca32009-12-06 20:23:50 +00001029 for (ShadowMap::iterator E = ShadowMaps.back().begin(),
1030 EEnd = ShadowMaps.back().end();
1031 E != EEnd;
1032 ++E)
1033 E->second.Destroy();
1034
Douglas Gregor3545ff42009-09-21 16:56:56 +00001035 ShadowMaps.pop_back();
1036}
1037
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001038/// \brief Determines whether this given declaration will be found by
1039/// ordinary name lookup.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001040bool ResultBuilder::IsOrdinaryName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001041 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1042
Richard Smith541b38b2013-09-20 01:15:31 +00001043 // If name lookup finds a local extern declaration, then we are in a
1044 // context where it behaves like an ordinary name.
1045 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001046 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9858ed52010-06-15 20:26:51 +00001047 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001048 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor05fcf842010-11-02 20:36:02 +00001049 if (isa<ObjCIvarDecl>(ND))
1050 return true;
Douglas Gregor05fcf842010-11-02 20:36:02 +00001051 }
1052
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001053 return ND->getIdentifierNamespace() & IDNS;
1054}
1055
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001056/// \brief Determines whether this given declaration will be found by
Douglas Gregor70febae2010-05-28 00:49:12 +00001057/// ordinary name lookup but is not a type name.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001058bool ResultBuilder::IsOrdinaryNonTypeName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001059 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1060 if (isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND))
1061 return false;
1062
Richard Smith541b38b2013-09-20 01:15:31 +00001063 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001064 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9858ed52010-06-15 20:26:51 +00001065 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001066 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor05fcf842010-11-02 20:36:02 +00001067 if (isa<ObjCIvarDecl>(ND))
1068 return true;
Douglas Gregor05fcf842010-11-02 20:36:02 +00001069 }
1070
Douglas Gregor70febae2010-05-28 00:49:12 +00001071 return ND->getIdentifierNamespace() & IDNS;
1072}
1073
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001074bool ResultBuilder::IsIntegralConstantValue(const NamedDecl *ND) const {
Douglas Gregor85b50632010-07-28 21:50:18 +00001075 if (!IsOrdinaryNonTypeName(ND))
1076 return 0;
1077
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001078 if (const ValueDecl *VD = dyn_cast<ValueDecl>(ND->getUnderlyingDecl()))
Douglas Gregor85b50632010-07-28 21:50:18 +00001079 if (VD->getType()->isIntegralOrEnumerationType())
1080 return true;
1081
1082 return false;
1083}
1084
Douglas Gregor70febae2010-05-28 00:49:12 +00001085/// \brief Determines whether this given declaration will be found by
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001086/// ordinary name lookup.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001087bool ResultBuilder::IsOrdinaryNonValueName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001088 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1089
Richard Smith541b38b2013-09-20 01:15:31 +00001090 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001091 if (SemaRef.getLangOpts().CPlusPlus)
John McCalle87beb22010-04-23 18:46:30 +00001092 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001093
1094 return (ND->getIdentifierNamespace() & IDNS) &&
Douglas Gregor70febae2010-05-28 00:49:12 +00001095 !isa<ValueDecl>(ND) && !isa<FunctionTemplateDecl>(ND) &&
1096 !isa<ObjCPropertyDecl>(ND);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001097}
1098
Douglas Gregor3545ff42009-09-21 16:56:56 +00001099/// \brief Determines whether the given declaration is suitable as the
1100/// start of a C++ nested-name-specifier, e.g., a class or namespace.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001101bool ResultBuilder::IsNestedNameSpecifier(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001102 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001103 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001104 ND = ClassTemplate->getTemplatedDecl();
1105
1106 return SemaRef.isAcceptableNestedNameSpecifier(ND);
1107}
1108
1109/// \brief Determines whether the given declaration is an enumeration.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001110bool ResultBuilder::IsEnum(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001111 return isa<EnumDecl>(ND);
1112}
1113
1114/// \brief Determines whether the given declaration is a class or struct.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001115bool ResultBuilder::IsClassOrStruct(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001116 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001117 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001118 ND = ClassTemplate->getTemplatedDecl();
Joao Matosdc86f942012-08-31 18:45:21 +00001119
1120 // For purposes of this check, interfaces match too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001121 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara6150c882010-05-11 21:36:43 +00001122 return RD->getTagKind() == TTK_Class ||
Joao Matosdc86f942012-08-31 18:45:21 +00001123 RD->getTagKind() == TTK_Struct ||
1124 RD->getTagKind() == TTK_Interface;
Douglas Gregor3545ff42009-09-21 16:56:56 +00001125
1126 return false;
1127}
1128
1129/// \brief Determines whether the given declaration is a union.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001130bool ResultBuilder::IsUnion(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001131 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001132 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001133 ND = ClassTemplate->getTemplatedDecl();
1134
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001135 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara6150c882010-05-11 21:36:43 +00001136 return RD->getTagKind() == TTK_Union;
Douglas Gregor3545ff42009-09-21 16:56:56 +00001137
1138 return false;
1139}
1140
1141/// \brief Determines whether the given declaration is a namespace.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001142bool ResultBuilder::IsNamespace(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001143 return isa<NamespaceDecl>(ND);
1144}
1145
1146/// \brief Determines whether the given declaration is a namespace or
1147/// namespace alias.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001148bool ResultBuilder::IsNamespaceOrAlias(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001149 return isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND);
1150}
1151
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001152/// \brief Determines whether the given declaration is a type.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001153bool ResultBuilder::IsType(const NamedDecl *ND) const {
1154 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor99fa2642010-08-24 01:06:58 +00001155 ND = Using->getTargetDecl();
1156
1157 return isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND);
Douglas Gregor3545ff42009-09-21 16:56:56 +00001158}
1159
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001160/// \brief Determines which members of a class should be visible via
1161/// "." or "->". Only value declarations, nested name specifiers, and
1162/// using declarations thereof should show up.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001163bool ResultBuilder::IsMember(const NamedDecl *ND) const {
1164 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001165 ND = Using->getTargetDecl();
1166
Douglas Gregor70788392009-12-11 18:14:22 +00001167 return isa<ValueDecl>(ND) || isa<FunctionTemplateDecl>(ND) ||
1168 isa<ObjCPropertyDecl>(ND);
Douglas Gregore412a5a2009-09-23 22:26:46 +00001169}
1170
Douglas Gregora817a192010-05-27 23:06:34 +00001171static bool isObjCReceiverType(ASTContext &C, QualType T) {
1172 T = C.getCanonicalType(T);
1173 switch (T->getTypeClass()) {
1174 case Type::ObjCObject:
1175 case Type::ObjCInterface:
1176 case Type::ObjCObjectPointer:
1177 return true;
1178
1179 case Type::Builtin:
1180 switch (cast<BuiltinType>(T)->getKind()) {
1181 case BuiltinType::ObjCId:
1182 case BuiltinType::ObjCClass:
1183 case BuiltinType::ObjCSel:
1184 return true;
1185
1186 default:
1187 break;
1188 }
1189 return false;
1190
1191 default:
1192 break;
1193 }
1194
David Blaikiebbafb8a2012-03-11 07:00:24 +00001195 if (!C.getLangOpts().CPlusPlus)
Douglas Gregora817a192010-05-27 23:06:34 +00001196 return false;
1197
1198 // FIXME: We could perform more analysis here to determine whether a
1199 // particular class type has any conversions to Objective-C types. For now,
1200 // just accept all class types.
1201 return T->isDependentType() || T->isRecordType();
1202}
1203
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001204bool ResultBuilder::IsObjCMessageReceiver(const NamedDecl *ND) const {
Douglas Gregora817a192010-05-27 23:06:34 +00001205 QualType T = getDeclUsageType(SemaRef.Context, ND);
1206 if (T.isNull())
1207 return false;
1208
1209 T = SemaRef.Context.getBaseElementType(T);
1210 return isObjCReceiverType(SemaRef.Context, T);
1211}
1212
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001213bool ResultBuilder::IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const {
Douglas Gregord8c61782012-02-15 15:34:24 +00001214 if (IsObjCMessageReceiver(ND))
1215 return true;
1216
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001217 const VarDecl *Var = dyn_cast<VarDecl>(ND);
Douglas Gregord8c61782012-02-15 15:34:24 +00001218 if (!Var)
1219 return false;
1220
1221 return Var->hasLocalStorage() && !Var->hasAttr<BlocksAttr>();
1222}
1223
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001224bool ResultBuilder::IsObjCCollection(const NamedDecl *ND) const {
David Blaikiebbafb8a2012-03-11 07:00:24 +00001225 if ((SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryName(ND)) ||
1226 (!SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryNonTypeName(ND)))
Douglas Gregor68762e72010-08-23 21:17:50 +00001227 return false;
1228
1229 QualType T = getDeclUsageType(SemaRef.Context, ND);
1230 if (T.isNull())
1231 return false;
1232
1233 T = SemaRef.Context.getBaseElementType(T);
1234 return T->isObjCObjectType() || T->isObjCObjectPointerType() ||
1235 T->isObjCIdType() ||
David Blaikiebbafb8a2012-03-11 07:00:24 +00001236 (SemaRef.getLangOpts().CPlusPlus && T->isRecordType());
Douglas Gregor68762e72010-08-23 21:17:50 +00001237}
Douglas Gregora817a192010-05-27 23:06:34 +00001238
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001239bool ResultBuilder::IsImpossibleToSatisfy(const NamedDecl *ND) const {
Douglas Gregor0ac41382010-09-23 23:01:17 +00001240 return false;
1241}
1242
James Dennettf1243872012-06-17 05:33:25 +00001243/// \brief Determines whether the given declaration is an Objective-C
Douglas Gregor2b8162b2010-01-14 16:08:12 +00001244/// instance variable.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001245bool ResultBuilder::IsObjCIvar(const NamedDecl *ND) const {
Douglas Gregor2b8162b2010-01-14 16:08:12 +00001246 return isa<ObjCIvarDecl>(ND);
1247}
1248
Douglas Gregorc580c522010-01-14 01:09:38 +00001249namespace {
1250 /// \brief Visible declaration consumer that adds a code-completion result
1251 /// for each visible declaration.
1252 class CodeCompletionDeclConsumer : public VisibleDeclConsumer {
1253 ResultBuilder &Results;
1254 DeclContext *CurContext;
1255
1256 public:
1257 CodeCompletionDeclConsumer(ResultBuilder &Results, DeclContext *CurContext)
1258 : Results(Results), CurContext(CurContext) { }
1259
Erik Verbruggen2e657ff2011-10-06 07:27:49 +00001260 virtual void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
1261 bool InBaseClass) {
1262 bool Accessible = true;
Douglas Gregor03ba1882011-11-03 16:51:37 +00001263 if (Ctx)
1264 Accessible = Results.getSema().IsSimplyAccessible(ND, Ctx);
1265
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00001266 ResultBuilder::Result Result(ND, Results.getBasePriority(ND), 0, false,
1267 Accessible);
Erik Verbruggen2e657ff2011-10-06 07:27:49 +00001268 Results.AddResult(Result, CurContext, Hiding, InBaseClass);
Douglas Gregorc580c522010-01-14 01:09:38 +00001269 }
1270 };
1271}
1272
Douglas Gregor3545ff42009-09-21 16:56:56 +00001273/// \brief Add type specifiers for the current language as keyword results.
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001274static void AddTypeSpecifierResults(const LangOptions &LangOpts,
Douglas Gregor3545ff42009-09-21 16:56:56 +00001275 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001276 typedef CodeCompletionResult Result;
Douglas Gregora2db7932010-05-26 22:00:08 +00001277 Results.AddResult(Result("short", CCP_Type));
1278 Results.AddResult(Result("long", CCP_Type));
1279 Results.AddResult(Result("signed", CCP_Type));
1280 Results.AddResult(Result("unsigned", CCP_Type));
1281 Results.AddResult(Result("void", CCP_Type));
1282 Results.AddResult(Result("char", CCP_Type));
1283 Results.AddResult(Result("int", CCP_Type));
1284 Results.AddResult(Result("float", CCP_Type));
1285 Results.AddResult(Result("double", CCP_Type));
1286 Results.AddResult(Result("enum", CCP_Type));
1287 Results.AddResult(Result("struct", CCP_Type));
1288 Results.AddResult(Result("union", CCP_Type));
1289 Results.AddResult(Result("const", CCP_Type));
1290 Results.AddResult(Result("volatile", CCP_Type));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001291
Douglas Gregor3545ff42009-09-21 16:56:56 +00001292 if (LangOpts.C99) {
1293 // C99-specific
Douglas Gregora2db7932010-05-26 22:00:08 +00001294 Results.AddResult(Result("_Complex", CCP_Type));
1295 Results.AddResult(Result("_Imaginary", CCP_Type));
1296 Results.AddResult(Result("_Bool", CCP_Type));
1297 Results.AddResult(Result("restrict", CCP_Type));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001298 }
1299
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001300 CodeCompletionBuilder Builder(Results.getAllocator(),
1301 Results.getCodeCompletionTUInfo());
Douglas Gregor3545ff42009-09-21 16:56:56 +00001302 if (LangOpts.CPlusPlus) {
1303 // C++-specific
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00001304 Results.AddResult(Result("bool", CCP_Type +
1305 (LangOpts.ObjC1? CCD_bool_in_ObjC : 0)));
Douglas Gregora2db7932010-05-26 22:00:08 +00001306 Results.AddResult(Result("class", CCP_Type));
1307 Results.AddResult(Result("wchar_t", CCP_Type));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001308
Douglas Gregorf4c33342010-05-28 00:22:41 +00001309 // typename qualified-id
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001310 Builder.AddTypedTextChunk("typename");
1311 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1312 Builder.AddPlaceholderChunk("qualifier");
1313 Builder.AddTextChunk("::");
1314 Builder.AddPlaceholderChunk("name");
1315 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001316
Richard Smith2bf7fdb2013-01-02 11:42:31 +00001317 if (LangOpts.CPlusPlus11) {
Douglas Gregora2db7932010-05-26 22:00:08 +00001318 Results.AddResult(Result("auto", CCP_Type));
1319 Results.AddResult(Result("char16_t", CCP_Type));
1320 Results.AddResult(Result("char32_t", CCP_Type));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001321
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001322 Builder.AddTypedTextChunk("decltype");
1323 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1324 Builder.AddPlaceholderChunk("expression");
1325 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1326 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001327 }
1328 }
1329
1330 // GNU extensions
1331 if (LangOpts.GNUMode) {
1332 // FIXME: Enable when we actually support decimal floating point.
Douglas Gregor78a21012010-01-14 16:01:26 +00001333 // Results.AddResult(Result("_Decimal32"));
1334 // Results.AddResult(Result("_Decimal64"));
1335 // Results.AddResult(Result("_Decimal128"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001336
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001337 Builder.AddTypedTextChunk("typeof");
1338 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1339 Builder.AddPlaceholderChunk("expression");
1340 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001341
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001342 Builder.AddTypedTextChunk("typeof");
1343 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1344 Builder.AddPlaceholderChunk("type");
1345 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1346 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001347 }
1348}
1349
John McCallfaf5fb42010-08-26 23:41:50 +00001350static void AddStorageSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001351 const LangOptions &LangOpts,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001352 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001353 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001354 // Note: we don't suggest either "auto" or "register", because both
1355 // are pointless as storage specifiers. Elsewhere, we suggest "auto"
1356 // in C++0x as a type specifier.
Douglas Gregor78a21012010-01-14 16:01:26 +00001357 Results.AddResult(Result("extern"));
1358 Results.AddResult(Result("static"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001359}
1360
John McCallfaf5fb42010-08-26 23:41:50 +00001361static void AddFunctionSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001362 const LangOptions &LangOpts,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001363 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001364 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001365 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001366 case Sema::PCC_Class:
1367 case Sema::PCC_MemberTemplate:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001368 if (LangOpts.CPlusPlus) {
Douglas Gregor78a21012010-01-14 16:01:26 +00001369 Results.AddResult(Result("explicit"));
1370 Results.AddResult(Result("friend"));
1371 Results.AddResult(Result("mutable"));
1372 Results.AddResult(Result("virtual"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001373 }
1374 // Fall through
1375
John McCallfaf5fb42010-08-26 23:41:50 +00001376 case Sema::PCC_ObjCInterface:
1377 case Sema::PCC_ObjCImplementation:
1378 case Sema::PCC_Namespace:
1379 case Sema::PCC_Template:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001380 if (LangOpts.CPlusPlus || LangOpts.C99)
Douglas Gregor78a21012010-01-14 16:01:26 +00001381 Results.AddResult(Result("inline"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001382 break;
1383
John McCallfaf5fb42010-08-26 23:41:50 +00001384 case Sema::PCC_ObjCInstanceVariableList:
1385 case Sema::PCC_Expression:
1386 case Sema::PCC_Statement:
1387 case Sema::PCC_ForInit:
1388 case Sema::PCC_Condition:
1389 case Sema::PCC_RecoveryInFunction:
1390 case Sema::PCC_Type:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001391 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor80039242011-02-15 20:33:25 +00001392 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001393 break;
1394 }
1395}
1396
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001397static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt);
1398static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt);
1399static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00001400 ResultBuilder &Results,
1401 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001402static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00001403 ResultBuilder &Results,
1404 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001405static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00001406 ResultBuilder &Results,
1407 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001408static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt);
Douglas Gregorf1934162010-01-13 21:24:21 +00001409
Douglas Gregorf4c33342010-05-28 00:22:41 +00001410static void AddTypedefResult(ResultBuilder &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001411 CodeCompletionBuilder Builder(Results.getAllocator(),
1412 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001413 Builder.AddTypedTextChunk("typedef");
1414 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1415 Builder.AddPlaceholderChunk("type");
1416 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1417 Builder.AddPlaceholderChunk("name");
1418 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001419}
1420
John McCallfaf5fb42010-08-26 23:41:50 +00001421static bool WantTypesInContext(Sema::ParserCompletionContext CCC,
Douglas Gregor70febae2010-05-28 00:49:12 +00001422 const LangOptions &LangOpts) {
Douglas Gregor70febae2010-05-28 00:49:12 +00001423 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001424 case Sema::PCC_Namespace:
1425 case Sema::PCC_Class:
1426 case Sema::PCC_ObjCInstanceVariableList:
1427 case Sema::PCC_Template:
1428 case Sema::PCC_MemberTemplate:
1429 case Sema::PCC_Statement:
1430 case Sema::PCC_RecoveryInFunction:
1431 case Sema::PCC_Type:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001432 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor80039242011-02-15 20:33:25 +00001433 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor70febae2010-05-28 00:49:12 +00001434 return true;
1435
John McCallfaf5fb42010-08-26 23:41:50 +00001436 case Sema::PCC_Expression:
1437 case Sema::PCC_Condition:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001438 return LangOpts.CPlusPlus;
1439
1440 case Sema::PCC_ObjCInterface:
1441 case Sema::PCC_ObjCImplementation:
Douglas Gregor70febae2010-05-28 00:49:12 +00001442 return false;
1443
John McCallfaf5fb42010-08-26 23:41:50 +00001444 case Sema::PCC_ForInit:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001445 return LangOpts.CPlusPlus || LangOpts.ObjC1 || LangOpts.C99;
Douglas Gregor70febae2010-05-28 00:49:12 +00001446 }
David Blaikie8a40f702012-01-17 06:56:22 +00001447
1448 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor70febae2010-05-28 00:49:12 +00001449}
1450
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00001451static PrintingPolicy getCompletionPrintingPolicy(const ASTContext &Context,
1452 const Preprocessor &PP) {
1453 PrintingPolicy Policy = Sema::getPrintingPolicy(Context, PP);
Douglas Gregore5c79d52011-10-18 21:20:17 +00001454 Policy.AnonymousTagLocations = false;
1455 Policy.SuppressStrongLifetime = true;
Douglas Gregor2e10cf92011-11-03 00:16:13 +00001456 Policy.SuppressUnwrittenScope = true;
Douglas Gregore5c79d52011-10-18 21:20:17 +00001457 return Policy;
1458}
1459
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00001460/// \brief Retrieve a printing policy suitable for code completion.
1461static PrintingPolicy getCompletionPrintingPolicy(Sema &S) {
1462 return getCompletionPrintingPolicy(S.Context, S.PP);
1463}
1464
Douglas Gregore5c79d52011-10-18 21:20:17 +00001465/// \brief Retrieve the string representation of the given type as a string
1466/// that has the appropriate lifetime for code completion.
1467///
1468/// This routine provides a fast path where we provide constant strings for
1469/// common type names.
1470static const char *GetCompletionTypeString(QualType T,
1471 ASTContext &Context,
1472 const PrintingPolicy &Policy,
1473 CodeCompletionAllocator &Allocator) {
1474 if (!T.getLocalQualifiers()) {
1475 // Built-in type names are constant strings.
1476 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T))
Argyrios Kyrtzidisbbff3da2012-05-05 04:20:28 +00001477 return BT->getNameAsCString(Policy);
Douglas Gregore5c79d52011-10-18 21:20:17 +00001478
1479 // Anonymous tag types are constant strings.
1480 if (const TagType *TagT = dyn_cast<TagType>(T))
1481 if (TagDecl *Tag = TagT->getDecl())
John McCall5ea95772013-03-09 00:54:27 +00001482 if (!Tag->hasNameForLinkage()) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001483 switch (Tag->getTagKind()) {
1484 case TTK_Struct: return "struct <anonymous>";
Joao Matosdc86f942012-08-31 18:45:21 +00001485 case TTK_Interface: return "__interface <anonymous>";
1486 case TTK_Class: return "class <anonymous>";
Douglas Gregore5c79d52011-10-18 21:20:17 +00001487 case TTK_Union: return "union <anonymous>";
1488 case TTK_Enum: return "enum <anonymous>";
1489 }
1490 }
1491 }
1492
1493 // Slow path: format the type as a string.
1494 std::string Result;
1495 T.getAsStringInternal(Result, Policy);
1496 return Allocator.CopyString(Result);
1497}
1498
Douglas Gregord8c61782012-02-15 15:34:24 +00001499/// \brief Add a completion for "this", if we're in a member function.
1500static void addThisCompletion(Sema &S, ResultBuilder &Results) {
1501 QualType ThisTy = S.getCurrentThisType();
1502 if (ThisTy.isNull())
1503 return;
1504
1505 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001506 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregord8c61782012-02-15 15:34:24 +00001507 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
1508 Builder.AddResultTypeChunk(GetCompletionTypeString(ThisTy,
1509 S.Context,
1510 Policy,
1511 Allocator));
1512 Builder.AddTypedTextChunk("this");
Joao Matosdc86f942012-08-31 18:45:21 +00001513 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregord8c61782012-02-15 15:34:24 +00001514}
1515
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001516/// \brief Add language constructs that show up for "ordinary" names.
John McCallfaf5fb42010-08-26 23:41:50 +00001517static void AddOrdinaryNameResults(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001518 Scope *S,
1519 Sema &SemaRef,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001520 ResultBuilder &Results) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001521 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001522 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregore5c79d52011-10-18 21:20:17 +00001523 PrintingPolicy Policy = getCompletionPrintingPolicy(SemaRef);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001524
John McCall276321a2010-08-25 06:19:51 +00001525 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001526 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001527 case Sema::PCC_Namespace:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001528 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001529 if (Results.includeCodePatterns()) {
1530 // namespace <identifier> { declarations }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001531 Builder.AddTypedTextChunk("namespace");
1532 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1533 Builder.AddPlaceholderChunk("identifier");
1534 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1535 Builder.AddPlaceholderChunk("declarations");
1536 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1537 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1538 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001539 }
1540
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001541 // namespace identifier = identifier ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001542 Builder.AddTypedTextChunk("namespace");
1543 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1544 Builder.AddPlaceholderChunk("name");
1545 Builder.AddChunk(CodeCompletionString::CK_Equal);
1546 Builder.AddPlaceholderChunk("namespace");
1547 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001548
1549 // Using directives
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001550 Builder.AddTypedTextChunk("using");
1551 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1552 Builder.AddTextChunk("namespace");
1553 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1554 Builder.AddPlaceholderChunk("identifier");
1555 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001556
1557 // asm(string-literal)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001558 Builder.AddTypedTextChunk("asm");
1559 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1560 Builder.AddPlaceholderChunk("string-literal");
1561 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1562 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001563
Douglas Gregorf4c33342010-05-28 00:22:41 +00001564 if (Results.includeCodePatterns()) {
1565 // Explicit template instantiation
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001566 Builder.AddTypedTextChunk("template");
1567 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1568 Builder.AddPlaceholderChunk("declaration");
1569 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001570 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001571 }
Douglas Gregorf1934162010-01-13 21:24:21 +00001572
David Blaikiebbafb8a2012-03-11 07:00:24 +00001573 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001574 AddObjCTopLevelResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001575
Douglas Gregorf4c33342010-05-28 00:22:41 +00001576 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001577 // Fall through
1578
John McCallfaf5fb42010-08-26 23:41:50 +00001579 case Sema::PCC_Class:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001580 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001581 // Using declaration
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001582 Builder.AddTypedTextChunk("using");
1583 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1584 Builder.AddPlaceholderChunk("qualifier");
1585 Builder.AddTextChunk("::");
1586 Builder.AddPlaceholderChunk("name");
1587 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001588
Douglas Gregorf4c33342010-05-28 00:22:41 +00001589 // using typename qualifier::name (only in a dependent context)
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001590 if (SemaRef.CurContext->isDependentContext()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001591 Builder.AddTypedTextChunk("using");
1592 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1593 Builder.AddTextChunk("typename");
1594 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1595 Builder.AddPlaceholderChunk("qualifier");
1596 Builder.AddTextChunk("::");
1597 Builder.AddPlaceholderChunk("name");
1598 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001599 }
1600
John McCallfaf5fb42010-08-26 23:41:50 +00001601 if (CCC == Sema::PCC_Class) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001602 AddTypedefResult(Results);
1603
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001604 // public:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001605 Builder.AddTypedTextChunk("public");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001606 if (Results.includeCodePatterns())
1607 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001608 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001609
1610 // protected:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001611 Builder.AddTypedTextChunk("protected");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001612 if (Results.includeCodePatterns())
1613 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001614 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001615
1616 // private:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001617 Builder.AddTypedTextChunk("private");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001618 if (Results.includeCodePatterns())
1619 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001620 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001621 }
1622 }
1623 // Fall through
1624
John McCallfaf5fb42010-08-26 23:41:50 +00001625 case Sema::PCC_Template:
1626 case Sema::PCC_MemberTemplate:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001627 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001628 // template < parameters >
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001629 Builder.AddTypedTextChunk("template");
1630 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1631 Builder.AddPlaceholderChunk("parameters");
1632 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1633 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001634 }
1635
David Blaikiebbafb8a2012-03-11 07:00:24 +00001636 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1637 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001638 break;
1639
John McCallfaf5fb42010-08-26 23:41:50 +00001640 case Sema::PCC_ObjCInterface:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001641 AddObjCInterfaceResults(SemaRef.getLangOpts(), Results, true);
1642 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1643 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001644 break;
1645
John McCallfaf5fb42010-08-26 23:41:50 +00001646 case Sema::PCC_ObjCImplementation:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001647 AddObjCImplementationResults(SemaRef.getLangOpts(), Results, true);
1648 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1649 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001650 break;
1651
John McCallfaf5fb42010-08-26 23:41:50 +00001652 case Sema::PCC_ObjCInstanceVariableList:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001653 AddObjCVisibilityResults(SemaRef.getLangOpts(), Results, true);
Douglas Gregor48d46252010-01-13 21:54:15 +00001654 break;
1655
John McCallfaf5fb42010-08-26 23:41:50 +00001656 case Sema::PCC_RecoveryInFunction:
1657 case Sema::PCC_Statement: {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001658 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001659
David Blaikiebbafb8a2012-03-11 07:00:24 +00001660 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns() &&
1661 SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001662 Builder.AddTypedTextChunk("try");
1663 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1664 Builder.AddPlaceholderChunk("statements");
1665 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1666 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1667 Builder.AddTextChunk("catch");
1668 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1669 Builder.AddPlaceholderChunk("declaration");
1670 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1671 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1672 Builder.AddPlaceholderChunk("statements");
1673 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1674 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1675 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001676 }
David Blaikiebbafb8a2012-03-11 07:00:24 +00001677 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001678 AddObjCStatementResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001679
Douglas Gregorf64acca2010-05-25 21:41:55 +00001680 if (Results.includeCodePatterns()) {
1681 // if (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001682 Builder.AddTypedTextChunk("if");
1683 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001684 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001685 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001686 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001687 Builder.AddPlaceholderChunk("expression");
1688 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1689 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1690 Builder.AddPlaceholderChunk("statements");
1691 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1692 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1693 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001694
Douglas Gregorf64acca2010-05-25 21:41:55 +00001695 // switch (condition) { }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001696 Builder.AddTypedTextChunk("switch");
1697 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001698 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001699 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001700 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001701 Builder.AddPlaceholderChunk("expression");
1702 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1703 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1704 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1705 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1706 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001707 }
1708
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001709 // Switch-specific statements.
John McCallaab3e412010-08-25 08:40:02 +00001710 if (!SemaRef.getCurFunction()->SwitchStack.empty()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001711 // case expression:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001712 Builder.AddTypedTextChunk("case");
1713 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1714 Builder.AddPlaceholderChunk("expression");
1715 Builder.AddChunk(CodeCompletionString::CK_Colon);
1716 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001717
1718 // default:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001719 Builder.AddTypedTextChunk("default");
1720 Builder.AddChunk(CodeCompletionString::CK_Colon);
1721 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001722 }
1723
Douglas Gregorf64acca2010-05-25 21:41:55 +00001724 if (Results.includeCodePatterns()) {
1725 /// while (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001726 Builder.AddTypedTextChunk("while");
1727 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001728 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001729 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001730 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001731 Builder.AddPlaceholderChunk("expression");
1732 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1733 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1734 Builder.AddPlaceholderChunk("statements");
1735 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1736 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1737 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001738
1739 // do { statements } while ( expression );
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001740 Builder.AddTypedTextChunk("do");
1741 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1742 Builder.AddPlaceholderChunk("statements");
1743 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1744 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1745 Builder.AddTextChunk("while");
1746 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1747 Builder.AddPlaceholderChunk("expression");
1748 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1749 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001750
Douglas Gregorf64acca2010-05-25 21:41:55 +00001751 // for ( for-init-statement ; condition ; expression ) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001752 Builder.AddTypedTextChunk("for");
1753 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001754 if (SemaRef.getLangOpts().CPlusPlus || SemaRef.getLangOpts().C99)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001755 Builder.AddPlaceholderChunk("init-statement");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001756 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001757 Builder.AddPlaceholderChunk("init-expression");
1758 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1759 Builder.AddPlaceholderChunk("condition");
1760 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1761 Builder.AddPlaceholderChunk("inc-expression");
1762 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1763 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1764 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1765 Builder.AddPlaceholderChunk("statements");
1766 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1767 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1768 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001769 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001770
1771 if (S->getContinueParent()) {
1772 // continue ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001773 Builder.AddTypedTextChunk("continue");
1774 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001775 }
1776
1777 if (S->getBreakParent()) {
1778 // break ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001779 Builder.AddTypedTextChunk("break");
1780 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001781 }
1782
1783 // "return expression ;" or "return ;", depending on whether we
1784 // know the function is void or not.
1785 bool isVoid = false;
1786 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext))
1787 isVoid = Function->getResultType()->isVoidType();
1788 else if (ObjCMethodDecl *Method
1789 = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext))
1790 isVoid = Method->getResultType()->isVoidType();
Douglas Gregor9a28e842010-03-01 23:15:13 +00001791 else if (SemaRef.getCurBlock() &&
1792 !SemaRef.getCurBlock()->ReturnType.isNull())
1793 isVoid = SemaRef.getCurBlock()->ReturnType->isVoidType();
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001794 Builder.AddTypedTextChunk("return");
Douglas Gregor44272ca2010-02-18 04:06:48 +00001795 if (!isVoid) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001796 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1797 Builder.AddPlaceholderChunk("expression");
Douglas Gregor44272ca2010-02-18 04:06:48 +00001798 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001799 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001800
Douglas Gregorf4c33342010-05-28 00:22:41 +00001801 // goto identifier ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001802 Builder.AddTypedTextChunk("goto");
1803 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1804 Builder.AddPlaceholderChunk("label");
1805 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001806
Douglas Gregorf4c33342010-05-28 00:22:41 +00001807 // Using directives
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001808 Builder.AddTypedTextChunk("using");
1809 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1810 Builder.AddTextChunk("namespace");
1811 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1812 Builder.AddPlaceholderChunk("identifier");
1813 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001814 }
1815
1816 // Fall through (for statement expressions).
John McCallfaf5fb42010-08-26 23:41:50 +00001817 case Sema::PCC_ForInit:
1818 case Sema::PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001819 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001820 // Fall through: conditions and statements can have expressions.
1821
Douglas Gregor5e35d592010-09-14 23:59:36 +00001822 case Sema::PCC_ParenthesizedExpression:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001823 if (SemaRef.getLangOpts().ObjCAutoRefCount &&
John McCall31168b02011-06-15 23:02:42 +00001824 CCC == Sema::PCC_ParenthesizedExpression) {
1825 // (__bridge <type>)<expression>
1826 Builder.AddTypedTextChunk("__bridge");
1827 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1828 Builder.AddPlaceholderChunk("type");
1829 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1830 Builder.AddPlaceholderChunk("expression");
1831 Results.AddResult(Result(Builder.TakeString()));
1832
1833 // (__bridge_transfer <Objective-C type>)<expression>
1834 Builder.AddTypedTextChunk("__bridge_transfer");
1835 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1836 Builder.AddPlaceholderChunk("Objective-C type");
1837 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1838 Builder.AddPlaceholderChunk("expression");
1839 Results.AddResult(Result(Builder.TakeString()));
1840
1841 // (__bridge_retained <CF type>)<expression>
1842 Builder.AddTypedTextChunk("__bridge_retained");
1843 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1844 Builder.AddPlaceholderChunk("CF type");
1845 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1846 Builder.AddPlaceholderChunk("expression");
1847 Results.AddResult(Result(Builder.TakeString()));
1848 }
1849 // Fall through
1850
John McCallfaf5fb42010-08-26 23:41:50 +00001851 case Sema::PCC_Expression: {
David Blaikiebbafb8a2012-03-11 07:00:24 +00001852 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001853 // 'this', if we're in a non-static member function.
Douglas Gregord8c61782012-02-15 15:34:24 +00001854 addThisCompletion(SemaRef, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001855
Douglas Gregore5c79d52011-10-18 21:20:17 +00001856 // true
1857 Builder.AddResultTypeChunk("bool");
1858 Builder.AddTypedTextChunk("true");
1859 Results.AddResult(Result(Builder.TakeString()));
1860
1861 // false
1862 Builder.AddResultTypeChunk("bool");
1863 Builder.AddTypedTextChunk("false");
1864 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001865
David Blaikiebbafb8a2012-03-11 07:00:24 +00001866 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001867 // dynamic_cast < type-id > ( expression )
1868 Builder.AddTypedTextChunk("dynamic_cast");
1869 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1870 Builder.AddPlaceholderChunk("type");
1871 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1872 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1873 Builder.AddPlaceholderChunk("expression");
1874 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1875 Results.AddResult(Result(Builder.TakeString()));
1876 }
Douglas Gregorf4c33342010-05-28 00:22:41 +00001877
1878 // static_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001879 Builder.AddTypedTextChunk("static_cast");
1880 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1881 Builder.AddPlaceholderChunk("type");
1882 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1883 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1884 Builder.AddPlaceholderChunk("expression");
1885 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1886 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001887
Douglas Gregorf4c33342010-05-28 00:22:41 +00001888 // reinterpret_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001889 Builder.AddTypedTextChunk("reinterpret_cast");
1890 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1891 Builder.AddPlaceholderChunk("type");
1892 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1893 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1894 Builder.AddPlaceholderChunk("expression");
1895 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1896 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001897
Douglas Gregorf4c33342010-05-28 00:22:41 +00001898 // const_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001899 Builder.AddTypedTextChunk("const_cast");
1900 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1901 Builder.AddPlaceholderChunk("type");
1902 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1903 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1904 Builder.AddPlaceholderChunk("expression");
1905 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1906 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001907
David Blaikiebbafb8a2012-03-11 07:00:24 +00001908 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001909 // typeid ( expression-or-type )
Douglas Gregore5c79d52011-10-18 21:20:17 +00001910 Builder.AddResultTypeChunk("std::type_info");
Douglas Gregorc05f6572011-04-12 02:47:21 +00001911 Builder.AddTypedTextChunk("typeid");
1912 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1913 Builder.AddPlaceholderChunk("expression-or-type");
1914 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1915 Results.AddResult(Result(Builder.TakeString()));
1916 }
1917
Douglas Gregorf4c33342010-05-28 00:22:41 +00001918 // new T ( ... )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001919 Builder.AddTypedTextChunk("new");
1920 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1921 Builder.AddPlaceholderChunk("type");
1922 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1923 Builder.AddPlaceholderChunk("expressions");
1924 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1925 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001926
Douglas Gregorf4c33342010-05-28 00:22:41 +00001927 // new T [ ] ( ... )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001928 Builder.AddTypedTextChunk("new");
1929 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1930 Builder.AddPlaceholderChunk("type");
1931 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1932 Builder.AddPlaceholderChunk("size");
1933 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1934 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1935 Builder.AddPlaceholderChunk("expressions");
1936 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1937 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001938
Douglas Gregorf4c33342010-05-28 00:22:41 +00001939 // delete expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001940 Builder.AddResultTypeChunk("void");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001941 Builder.AddTypedTextChunk("delete");
1942 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1943 Builder.AddPlaceholderChunk("expression");
1944 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001945
Douglas Gregorf4c33342010-05-28 00:22:41 +00001946 // delete [] expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001947 Builder.AddResultTypeChunk("void");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001948 Builder.AddTypedTextChunk("delete");
1949 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1950 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1951 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1952 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1953 Builder.AddPlaceholderChunk("expression");
1954 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001955
David Blaikiebbafb8a2012-03-11 07:00:24 +00001956 if (SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001957 // throw expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001958 Builder.AddResultTypeChunk("void");
Douglas Gregorc05f6572011-04-12 02:47:21 +00001959 Builder.AddTypedTextChunk("throw");
1960 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1961 Builder.AddPlaceholderChunk("expression");
1962 Results.AddResult(Result(Builder.TakeString()));
1963 }
Douglas Gregor4205fef2011-10-18 16:29:03 +00001964
Douglas Gregora2db7932010-05-26 22:00:08 +00001965 // FIXME: Rethrow?
Douglas Gregor4205fef2011-10-18 16:29:03 +00001966
Richard Smith2bf7fdb2013-01-02 11:42:31 +00001967 if (SemaRef.getLangOpts().CPlusPlus11) {
Douglas Gregor4205fef2011-10-18 16:29:03 +00001968 // nullptr
Douglas Gregore5c79d52011-10-18 21:20:17 +00001969 Builder.AddResultTypeChunk("std::nullptr_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001970 Builder.AddTypedTextChunk("nullptr");
1971 Results.AddResult(Result(Builder.TakeString()));
1972
1973 // alignof
Douglas Gregore5c79d52011-10-18 21:20:17 +00001974 Builder.AddResultTypeChunk("size_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001975 Builder.AddTypedTextChunk("alignof");
1976 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1977 Builder.AddPlaceholderChunk("type");
1978 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1979 Results.AddResult(Result(Builder.TakeString()));
1980
1981 // noexcept
Douglas Gregore5c79d52011-10-18 21:20:17 +00001982 Builder.AddResultTypeChunk("bool");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001983 Builder.AddTypedTextChunk("noexcept");
1984 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1985 Builder.AddPlaceholderChunk("expression");
1986 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1987 Results.AddResult(Result(Builder.TakeString()));
1988
1989 // sizeof... expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001990 Builder.AddResultTypeChunk("size_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001991 Builder.AddTypedTextChunk("sizeof...");
1992 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1993 Builder.AddPlaceholderChunk("parameter-pack");
1994 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1995 Results.AddResult(Result(Builder.TakeString()));
1996 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001997 }
1998
David Blaikiebbafb8a2012-03-11 07:00:24 +00001999 if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002000 // Add "super", if we're in an Objective-C class with a superclass.
Ted Kremenek305a0a72010-05-31 21:43:10 +00002001 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) {
2002 // The interface can be NULL.
2003 if (ObjCInterfaceDecl *ID = Method->getClassInterface())
Douglas Gregore5c79d52011-10-18 21:20:17 +00002004 if (ID->getSuperClass()) {
2005 std::string SuperType;
2006 SuperType = ID->getSuperClass()->getNameAsString();
2007 if (Method->isInstanceMethod())
2008 SuperType += " *";
2009
2010 Builder.AddResultTypeChunk(Allocator.CopyString(SuperType));
2011 Builder.AddTypedTextChunk("super");
2012 Results.AddResult(Result(Builder.TakeString()));
2013 }
Ted Kremenek305a0a72010-05-31 21:43:10 +00002014 }
2015
Douglas Gregorf98e6a22010-01-13 23:51:12 +00002016 AddObjCExpressionResults(Results, true);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002017 }
2018
Jordan Rose58d54722012-06-30 21:33:57 +00002019 if (SemaRef.getLangOpts().C11) {
2020 // _Alignof
2021 Builder.AddResultTypeChunk("size_t");
2022 if (SemaRef.getASTContext().Idents.get("alignof").hasMacroDefinition())
2023 Builder.AddTypedTextChunk("alignof");
2024 else
2025 Builder.AddTypedTextChunk("_Alignof");
2026 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2027 Builder.AddPlaceholderChunk("type");
2028 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2029 Results.AddResult(Result(Builder.TakeString()));
2030 }
2031
Douglas Gregorf4c33342010-05-28 00:22:41 +00002032 // sizeof expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00002033 Builder.AddResultTypeChunk("size_t");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002034 Builder.AddTypedTextChunk("sizeof");
2035 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2036 Builder.AddPlaceholderChunk("expression-or-type");
2037 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2038 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002039 break;
2040 }
Douglas Gregor99fa2642010-08-24 01:06:58 +00002041
John McCallfaf5fb42010-08-26 23:41:50 +00002042 case Sema::PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00002043 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor99fa2642010-08-24 01:06:58 +00002044 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002045 }
2046
David Blaikiebbafb8a2012-03-11 07:00:24 +00002047 if (WantTypesInContext(CCC, SemaRef.getLangOpts()))
2048 AddTypeSpecifierResults(SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002049
David Blaikiebbafb8a2012-03-11 07:00:24 +00002050 if (SemaRef.getLangOpts().CPlusPlus && CCC != Sema::PCC_Type)
Douglas Gregor78a21012010-01-14 16:01:26 +00002051 Results.AddResult(Result("operator"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002052}
2053
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002054/// \brief If the given declaration has an associated type, add it as a result
2055/// type chunk.
2056static void AddResultTypeChunk(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002057 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002058 const NamedDecl *ND,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002059 CodeCompletionBuilder &Result) {
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002060 if (!ND)
2061 return;
Douglas Gregor0212fd72010-09-21 16:06:22 +00002062
2063 // Skip constructors and conversion functions, which have their return types
2064 // built into their names.
2065 if (isa<CXXConstructorDecl>(ND) || isa<CXXConversionDecl>(ND))
2066 return;
2067
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002068 // Determine the type of the declaration (if it has a type).
Douglas Gregor0212fd72010-09-21 16:06:22 +00002069 QualType T;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002070 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002071 T = Function->getResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002072 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002073 T = Method->getResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002074 else if (const FunctionTemplateDecl *FunTmpl =
2075 dyn_cast<FunctionTemplateDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002076 T = FunTmpl->getTemplatedDecl()->getResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002077 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002078 T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext()));
2079 else if (isa<UnresolvedUsingValueDecl>(ND)) {
2080 /* Do nothing: ignore unresolved using declarations*/
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002081 } else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND)) {
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002082 T = Value->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002083 } else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002084 T = Property->getType();
2085
2086 if (T.isNull() || Context.hasSameType(T, Context.DependentTy))
2087 return;
2088
Douglas Gregor75acd922011-09-27 23:30:47 +00002089 Result.AddResultTypeChunk(GetCompletionTypeString(T, Context, Policy,
Douglas Gregor304f9b02011-02-01 21:15:40 +00002090 Result.getAllocator()));
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002091}
2092
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002093static void MaybeAddSentinel(ASTContext &Context,
2094 const NamedDecl *FunctionOrMethod,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002095 CodeCompletionBuilder &Result) {
Douglas Gregordbb71db2010-08-23 23:51:41 +00002096 if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>())
2097 if (Sentinel->getSentinel() == 0) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00002098 if (Context.getLangOpts().ObjC1 &&
Douglas Gregordbb71db2010-08-23 23:51:41 +00002099 Context.Idents.get("nil").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002100 Result.AddTextChunk(", nil");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002101 else if (Context.Idents.get("NULL").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002102 Result.AddTextChunk(", NULL");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002103 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002104 Result.AddTextChunk(", (void*)0");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002105 }
2106}
2107
Douglas Gregor8f08d742011-07-30 07:55:26 +00002108static std::string formatObjCParamQualifiers(unsigned ObjCQuals) {
2109 std::string Result;
2110 if (ObjCQuals & Decl::OBJC_TQ_In)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002111 Result += "in ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002112 else if (ObjCQuals & Decl::OBJC_TQ_Inout)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002113 Result += "inout ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002114 else if (ObjCQuals & Decl::OBJC_TQ_Out)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002115 Result += "out ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002116 if (ObjCQuals & Decl::OBJC_TQ_Bycopy)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002117 Result += "bycopy ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002118 else if (ObjCQuals & Decl::OBJC_TQ_Byref)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002119 Result += "byref ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002120 if (ObjCQuals & Decl::OBJC_TQ_Oneway)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002121 Result += "oneway ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002122 return Result;
2123}
2124
Douglas Gregore90dd002010-08-24 16:15:59 +00002125static std::string FormatFunctionParameter(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002126 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002127 const ParmVarDecl *Param,
Douglas Gregord793e7c2011-10-18 04:23:19 +00002128 bool SuppressName = false,
2129 bool SuppressBlock = false) {
Douglas Gregore90dd002010-08-24 16:15:59 +00002130 bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext());
2131 if (Param->getType()->isDependentType() ||
2132 !Param->getType()->isBlockPointerType()) {
2133 // The argument for a dependent or non-block parameter is a placeholder
2134 // containing that parameter's type.
2135 std::string Result;
2136
Douglas Gregor981a0c42010-08-29 19:47:46 +00002137 if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002138 Result = Param->getIdentifier()->getName();
2139
John McCall31168b02011-06-15 23:02:42 +00002140 Param->getType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002141
2142 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002143 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2144 + Result + ")";
Douglas Gregor981a0c42010-08-29 19:47:46 +00002145 if (Param->getIdentifier() && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002146 Result += Param->getIdentifier()->getName();
2147 }
2148 return Result;
2149 }
2150
2151 // The argument for a block pointer parameter is a block literal with
2152 // the appropriate type.
David Blaikie6adc78e2013-02-18 22:06:02 +00002153 FunctionTypeLoc Block;
2154 FunctionProtoTypeLoc BlockProto;
Douglas Gregore90dd002010-08-24 16:15:59 +00002155 TypeLoc TL;
2156 if (TypeSourceInfo *TSInfo = Param->getTypeSourceInfo()) {
2157 TL = TSInfo->getTypeLoc().getUnqualifiedLoc();
2158 while (true) {
2159 // Look through typedefs.
Douglas Gregord793e7c2011-10-18 04:23:19 +00002160 if (!SuppressBlock) {
David Blaikie6adc78e2013-02-18 22:06:02 +00002161 if (TypedefTypeLoc TypedefTL = TL.getAs<TypedefTypeLoc>()) {
2162 if (TypeSourceInfo *InnerTSInfo =
2163 TypedefTL.getTypedefNameDecl()->getTypeSourceInfo()) {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002164 TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc();
2165 continue;
2166 }
2167 }
2168
2169 // Look through qualified types
David Blaikie6adc78e2013-02-18 22:06:02 +00002170 if (QualifiedTypeLoc QualifiedTL = TL.getAs<QualifiedTypeLoc>()) {
2171 TL = QualifiedTL.getUnqualifiedLoc();
Douglas Gregore90dd002010-08-24 16:15:59 +00002172 continue;
2173 }
2174 }
2175
Douglas Gregore90dd002010-08-24 16:15:59 +00002176 // Try to get the function prototype behind the block pointer type,
2177 // then we're done.
David Blaikie6adc78e2013-02-18 22:06:02 +00002178 if (BlockPointerTypeLoc BlockPtr = TL.getAs<BlockPointerTypeLoc>()) {
2179 TL = BlockPtr.getPointeeLoc().IgnoreParens();
2180 Block = TL.getAs<FunctionTypeLoc>();
2181 BlockProto = TL.getAs<FunctionProtoTypeLoc>();
Douglas Gregore90dd002010-08-24 16:15:59 +00002182 }
2183 break;
2184 }
2185 }
2186
2187 if (!Block) {
2188 // We were unable to find a FunctionProtoTypeLoc with parameter names
2189 // for the block; just use the parameter type as a placeholder.
2190 std::string Result;
Douglas Gregord793e7c2011-10-18 04:23:19 +00002191 if (!ObjCMethodParam && Param->getIdentifier())
2192 Result = Param->getIdentifier()->getName();
2193
John McCall31168b02011-06-15 23:02:42 +00002194 Param->getType().getUnqualifiedType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002195
2196 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002197 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2198 + Result + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002199 if (Param->getIdentifier())
2200 Result += Param->getIdentifier()->getName();
2201 }
2202
2203 return Result;
2204 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002205
Douglas Gregore90dd002010-08-24 16:15:59 +00002206 // We have the function prototype behind the block pointer type, as it was
2207 // written in the source.
Douglas Gregor67da50e2010-09-08 22:47:51 +00002208 std::string Result;
David Blaikie6adc78e2013-02-18 22:06:02 +00002209 QualType ResultType = Block.getTypePtr()->getResultType();
Douglas Gregord793e7c2011-10-18 04:23:19 +00002210 if (!ResultType->isVoidType() || SuppressBlock)
John McCall31168b02011-06-15 23:02:42 +00002211 ResultType.getAsStringInternal(Result, Policy);
Douglas Gregord793e7c2011-10-18 04:23:19 +00002212
2213 // Format the parameter list.
2214 std::string Params;
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002215 if (!BlockProto || Block.getNumParams() == 0) {
David Blaikie6adc78e2013-02-18 22:06:02 +00002216 if (BlockProto && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002217 Params = "(...)";
Douglas Gregoraf25cfa2010-10-02 23:49:58 +00002218 else
Douglas Gregord793e7c2011-10-18 04:23:19 +00002219 Params = "(void)";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002220 } else {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002221 Params += "(";
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002222 for (unsigned I = 0, N = Block.getNumParams(); I != N; ++I) {
Douglas Gregor67da50e2010-09-08 22:47:51 +00002223 if (I)
Douglas Gregord793e7c2011-10-18 04:23:19 +00002224 Params += ", ";
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002225 Params += FormatFunctionParameter(Context, Policy, Block.getParam(I),
2226 /*SuppressName=*/false,
Douglas Gregord793e7c2011-10-18 04:23:19 +00002227 /*SuppressBlock=*/true);
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002228
David Blaikie6adc78e2013-02-18 22:06:02 +00002229 if (I == N - 1 && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002230 Params += ", ...";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002231 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002232 Params += ")";
Douglas Gregor400f5972010-08-31 05:13:43 +00002233 }
Douglas Gregor67da50e2010-09-08 22:47:51 +00002234
Douglas Gregord793e7c2011-10-18 04:23:19 +00002235 if (SuppressBlock) {
2236 // Format as a parameter.
2237 Result = Result + " (^";
2238 if (Param->getIdentifier())
2239 Result += Param->getIdentifier()->getName();
2240 Result += ")";
2241 Result += Params;
2242 } else {
2243 // Format as a block literal argument.
2244 Result = '^' + Result;
2245 Result += Params;
2246
2247 if (Param->getIdentifier())
2248 Result += Param->getIdentifier()->getName();
2249 }
2250
Douglas Gregore90dd002010-08-24 16:15:59 +00002251 return Result;
2252}
2253
Douglas Gregor3545ff42009-09-21 16:56:56 +00002254/// \brief Add function parameter chunks to the given code completion string.
2255static void AddFunctionParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002256 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002257 const FunctionDecl *Function,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002258 CodeCompletionBuilder &Result,
2259 unsigned Start = 0,
2260 bool InOptional = false) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002261 bool FirstParameter = true;
Douglas Gregor9eb77012009-11-07 00:00:49 +00002262
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002263 for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002264 const ParmVarDecl *Param = Function->getParamDecl(P);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002265
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002266 if (Param->hasDefaultArg() && !InOptional) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002267 // When we see an optional default argument, put that argument and
2268 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002269 CodeCompletionBuilder Opt(Result.getAllocator(),
2270 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002271 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002272 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002273 AddFunctionParameterChunks(Context, Policy, Function, Opt, P, true);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002274 Result.AddOptionalChunk(Opt.TakeString());
2275 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002276 }
2277
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002278 if (FirstParameter)
2279 FirstParameter = false;
2280 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002281 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002282
2283 InOptional = false;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002284
2285 // Format the placeholder string.
Douglas Gregor75acd922011-09-27 23:30:47 +00002286 std::string PlaceholderStr = FormatFunctionParameter(Context, Policy,
2287 Param);
Douglas Gregore90dd002010-08-24 16:15:59 +00002288
Douglas Gregor400f5972010-08-31 05:13:43 +00002289 if (Function->isVariadic() && P == N - 1)
2290 PlaceholderStr += ", ...";
2291
Douglas Gregor3545ff42009-09-21 16:56:56 +00002292 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002293 Result.AddPlaceholderChunk(
2294 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002295 }
Douglas Gregorba449032009-09-22 21:42:17 +00002296
2297 if (const FunctionProtoType *Proto
2298 = Function->getType()->getAs<FunctionProtoType>())
Douglas Gregordbb71db2010-08-23 23:51:41 +00002299 if (Proto->isVariadic()) {
Alp Toker9cacbab2014-01-20 20:26:09 +00002300 if (Proto->getNumParams() == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002301 Result.AddPlaceholderChunk("...");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002302
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002303 MaybeAddSentinel(Context, Function, Result);
Douglas Gregordbb71db2010-08-23 23:51:41 +00002304 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002305}
2306
2307/// \brief Add template parameter chunks to the given code completion string.
2308static void AddTemplateParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002309 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002310 const TemplateDecl *Template,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002311 CodeCompletionBuilder &Result,
2312 unsigned MaxParameters = 0,
2313 unsigned Start = 0,
2314 bool InDefaultArg = false) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002315 bool FirstParameter = true;
2316
2317 TemplateParameterList *Params = Template->getTemplateParameters();
2318 TemplateParameterList::iterator PEnd = Params->end();
2319 if (MaxParameters)
2320 PEnd = Params->begin() + MaxParameters;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002321 for (TemplateParameterList::iterator P = Params->begin() + Start;
2322 P != PEnd; ++P) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002323 bool HasDefaultArg = false;
2324 std::string PlaceholderStr;
2325 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) {
2326 if (TTP->wasDeclaredWithTypename())
2327 PlaceholderStr = "typename";
2328 else
2329 PlaceholderStr = "class";
2330
2331 if (TTP->getIdentifier()) {
2332 PlaceholderStr += ' ';
2333 PlaceholderStr += TTP->getIdentifier()->getName();
2334 }
2335
2336 HasDefaultArg = TTP->hasDefaultArgument();
2337 } else if (NonTypeTemplateParmDecl *NTTP
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002338 = dyn_cast<NonTypeTemplateParmDecl>(*P)) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002339 if (NTTP->getIdentifier())
2340 PlaceholderStr = NTTP->getIdentifier()->getName();
John McCall31168b02011-06-15 23:02:42 +00002341 NTTP->getType().getAsStringInternal(PlaceholderStr, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002342 HasDefaultArg = NTTP->hasDefaultArgument();
2343 } else {
2344 assert(isa<TemplateTemplateParmDecl>(*P));
2345 TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P);
2346
2347 // Since putting the template argument list into the placeholder would
2348 // be very, very long, we just use an abbreviation.
2349 PlaceholderStr = "template<...> class";
2350 if (TTP->getIdentifier()) {
2351 PlaceholderStr += ' ';
2352 PlaceholderStr += TTP->getIdentifier()->getName();
2353 }
2354
2355 HasDefaultArg = TTP->hasDefaultArgument();
2356 }
2357
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002358 if (HasDefaultArg && !InDefaultArg) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002359 // When we see an optional default argument, put that argument and
2360 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002361 CodeCompletionBuilder Opt(Result.getAllocator(),
2362 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002363 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002364 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002365 AddTemplateParameterChunks(Context, Policy, Template, Opt, MaxParameters,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002366 P - Params->begin(), true);
2367 Result.AddOptionalChunk(Opt.TakeString());
2368 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002369 }
2370
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002371 InDefaultArg = false;
2372
Douglas Gregor3545ff42009-09-21 16:56:56 +00002373 if (FirstParameter)
2374 FirstParameter = false;
2375 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002376 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002377
2378 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002379 Result.AddPlaceholderChunk(
2380 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002381 }
2382}
2383
Douglas Gregorf2510672009-09-21 19:57:38 +00002384/// \brief Add a qualifier to the given code-completion string, if the
2385/// provided nested-name-specifier is non-NULL.
Douglas Gregor0f622362009-12-11 18:44:16 +00002386static void
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002387AddQualifierToCompletionString(CodeCompletionBuilder &Result,
Douglas Gregor0f622362009-12-11 18:44:16 +00002388 NestedNameSpecifier *Qualifier,
2389 bool QualifierIsInformative,
Douglas Gregor75acd922011-09-27 23:30:47 +00002390 ASTContext &Context,
2391 const PrintingPolicy &Policy) {
Douglas Gregorf2510672009-09-21 19:57:38 +00002392 if (!Qualifier)
2393 return;
2394
2395 std::string PrintedNNS;
2396 {
2397 llvm::raw_string_ostream OS(PrintedNNS);
Douglas Gregor75acd922011-09-27 23:30:47 +00002398 Qualifier->print(OS, Policy);
Douglas Gregorf2510672009-09-21 19:57:38 +00002399 }
Douglas Gregor5bf52692009-09-22 23:15:58 +00002400 if (QualifierIsInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002401 Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregor5bf52692009-09-22 23:15:58 +00002402 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002403 Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregorf2510672009-09-21 19:57:38 +00002404}
2405
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002406static void
2407AddFunctionTypeQualsToCompletionString(CodeCompletionBuilder &Result,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002408 const FunctionDecl *Function) {
Douglas Gregor0f622362009-12-11 18:44:16 +00002409 const FunctionProtoType *Proto
2410 = Function->getType()->getAs<FunctionProtoType>();
2411 if (!Proto || !Proto->getTypeQuals())
2412 return;
2413
Douglas Gregor304f9b02011-02-01 21:15:40 +00002414 // FIXME: Add ref-qualifier!
2415
2416 // Handle single qualifiers without copying
2417 if (Proto->getTypeQuals() == Qualifiers::Const) {
2418 Result.AddInformativeChunk(" const");
2419 return;
2420 }
2421
2422 if (Proto->getTypeQuals() == Qualifiers::Volatile) {
2423 Result.AddInformativeChunk(" volatile");
2424 return;
2425 }
2426
2427 if (Proto->getTypeQuals() == Qualifiers::Restrict) {
2428 Result.AddInformativeChunk(" restrict");
2429 return;
2430 }
2431
2432 // Handle multiple qualifiers.
Douglas Gregor0f622362009-12-11 18:44:16 +00002433 std::string QualsStr;
David Blaikief5697e52012-08-10 00:55:35 +00002434 if (Proto->isConst())
Douglas Gregor0f622362009-12-11 18:44:16 +00002435 QualsStr += " const";
David Blaikief5697e52012-08-10 00:55:35 +00002436 if (Proto->isVolatile())
Douglas Gregor0f622362009-12-11 18:44:16 +00002437 QualsStr += " volatile";
David Blaikief5697e52012-08-10 00:55:35 +00002438 if (Proto->isRestrict())
Douglas Gregor0f622362009-12-11 18:44:16 +00002439 QualsStr += " restrict";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002440 Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr));
Douglas Gregor0f622362009-12-11 18:44:16 +00002441}
2442
Douglas Gregor0212fd72010-09-21 16:06:22 +00002443/// \brief Add the name of the given declaration
Douglas Gregor75acd922011-09-27 23:30:47 +00002444static void AddTypedNameChunk(ASTContext &Context, const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002445 const NamedDecl *ND,
2446 CodeCompletionBuilder &Result) {
Douglas Gregor0212fd72010-09-21 16:06:22 +00002447 DeclarationName Name = ND->getDeclName();
2448 if (!Name)
2449 return;
2450
2451 switch (Name.getNameKind()) {
Douglas Gregor304f9b02011-02-01 21:15:40 +00002452 case DeclarationName::CXXOperatorName: {
2453 const char *OperatorName = 0;
2454 switch (Name.getCXXOverloadedOperator()) {
2455 case OO_None:
2456 case OO_Conditional:
2457 case NUM_OVERLOADED_OPERATORS:
2458 OperatorName = "operator";
2459 break;
2460
2461#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
2462 case OO_##Name: OperatorName = "operator" Spelling; break;
2463#define OVERLOADED_OPERATOR_MULTI(Name,Spelling,Unary,Binary,MemberOnly)
2464#include "clang/Basic/OperatorKinds.def"
2465
2466 case OO_New: OperatorName = "operator new"; break;
2467 case OO_Delete: OperatorName = "operator delete"; break;
2468 case OO_Array_New: OperatorName = "operator new[]"; break;
2469 case OO_Array_Delete: OperatorName = "operator delete[]"; break;
2470 case OO_Call: OperatorName = "operator()"; break;
2471 case OO_Subscript: OperatorName = "operator[]"; break;
2472 }
2473 Result.AddTypedTextChunk(OperatorName);
2474 break;
2475 }
2476
Douglas Gregor0212fd72010-09-21 16:06:22 +00002477 case DeclarationName::Identifier:
2478 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor0212fd72010-09-21 16:06:22 +00002479 case DeclarationName::CXXDestructorName:
2480 case DeclarationName::CXXLiteralOperatorName:
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002481 Result.AddTypedTextChunk(
2482 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002483 break;
2484
2485 case DeclarationName::CXXUsingDirective:
2486 case DeclarationName::ObjCZeroArgSelector:
2487 case DeclarationName::ObjCOneArgSelector:
2488 case DeclarationName::ObjCMultiArgSelector:
2489 break;
2490
2491 case DeclarationName::CXXConstructorName: {
2492 CXXRecordDecl *Record = 0;
2493 QualType Ty = Name.getCXXNameType();
2494 if (const RecordType *RecordTy = Ty->getAs<RecordType>())
2495 Record = cast<CXXRecordDecl>(RecordTy->getDecl());
2496 else if (const InjectedClassNameType *InjectedTy
2497 = Ty->getAs<InjectedClassNameType>())
2498 Record = InjectedTy->getDecl();
2499 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002500 Result.AddTypedTextChunk(
2501 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002502 break;
2503 }
2504
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002505 Result.AddTypedTextChunk(
2506 Result.getAllocator().CopyString(Record->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002507 if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002508 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Douglas Gregor75acd922011-09-27 23:30:47 +00002509 AddTemplateParameterChunks(Context, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002510 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002511 }
2512 break;
2513 }
2514 }
2515}
2516
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002517CodeCompletionString *CodeCompletionResult::CreateCodeCompletionString(Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002518 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002519 CodeCompletionTUInfo &CCTUInfo,
2520 bool IncludeBriefComments) {
2521 return CreateCodeCompletionString(S.Context, S.PP, Allocator, CCTUInfo,
2522 IncludeBriefComments);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002523}
2524
Douglas Gregor3545ff42009-09-21 16:56:56 +00002525/// \brief If possible, create a new code completion string for the given
2526/// result.
2527///
2528/// \returns Either a new, heap-allocated code completion string describing
2529/// how to use this result, or NULL to indicate that the string or name of the
2530/// result is all that is needed.
2531CodeCompletionString *
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002532CodeCompletionResult::CreateCodeCompletionString(ASTContext &Ctx,
2533 Preprocessor &PP,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002534 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002535 CodeCompletionTUInfo &CCTUInfo,
2536 bool IncludeBriefComments) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002537 CodeCompletionBuilder Result(Allocator, CCTUInfo, Priority, Availability);
Douglas Gregor9eb77012009-11-07 00:00:49 +00002538
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002539 PrintingPolicy Policy = getCompletionPrintingPolicy(Ctx, PP);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002540 if (Kind == RK_Pattern) {
2541 Pattern->Priority = Priority;
2542 Pattern->Availability = Availability;
Douglas Gregor78254c82012-03-27 23:34:16 +00002543
2544 if (Declaration) {
2545 Result.addParentContext(Declaration->getDeclContext());
Douglas Gregor78254c82012-03-27 23:34:16 +00002546 Pattern->ParentName = Result.getParentName();
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002547 // Provide code completion comment for self.GetterName where
2548 // GetterName is the getter method for a property with name
2549 // different from the property name (declared via a property
2550 // getter attribute.
2551 const NamedDecl *ND = Declaration;
2552 if (const ObjCMethodDecl *M = dyn_cast<ObjCMethodDecl>(ND))
2553 if (M->isPropertyAccessor())
2554 if (const ObjCPropertyDecl *PDecl = M->findPropertyDecl())
2555 if (PDecl->getGetterName() == M->getSelector() &&
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002556 PDecl->getIdentifier() != M->getIdentifier()) {
2557 if (const RawComment *RC =
2558 Ctx.getRawCommentForAnyRedecl(M)) {
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002559 Result.addBriefComment(RC->getBriefText(Ctx));
2560 Pattern->BriefComment = Result.getBriefComment();
2561 }
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002562 else if (const RawComment *RC =
2563 Ctx.getRawCommentForAnyRedecl(PDecl)) {
2564 Result.addBriefComment(RC->getBriefText(Ctx));
2565 Pattern->BriefComment = Result.getBriefComment();
2566 }
2567 }
Douglas Gregor78254c82012-03-27 23:34:16 +00002568 }
2569
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002570 return Pattern;
2571 }
Douglas Gregorf09935f2009-12-01 05:55:20 +00002572
Douglas Gregorf09935f2009-12-01 05:55:20 +00002573 if (Kind == RK_Keyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002574 Result.AddTypedTextChunk(Keyword);
2575 return Result.TakeString();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002576 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002577
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002578 if (Kind == RK_Macro) {
Argyrios Kyrtzidis09c9e812013-02-20 00:54:57 +00002579 const MacroDirective *MD = PP.getMacroDirectiveHistory(Macro);
2580 assert(MD && "Not a macro?");
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00002581 const MacroInfo *MI = MD->getMacroInfo();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002582
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002583 Result.AddTypedTextChunk(
2584 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregorf09935f2009-12-01 05:55:20 +00002585
2586 if (!MI->isFunctionLike())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002587 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002588
2589 // Format a function-like macro with placeholders for the arguments.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002590 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor0c505312011-07-30 08:17:44 +00002591 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002592
2593 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
2594 if (MI->isC99Varargs()) {
2595 --AEnd;
2596
2597 if (A == AEnd) {
2598 Result.AddPlaceholderChunk("...");
2599 }
Douglas Gregor0c505312011-07-30 08:17:44 +00002600 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002601
Douglas Gregor0c505312011-07-30 08:17:44 +00002602 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002603 if (A != MI->arg_begin())
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002604 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3aa55262012-01-21 00:43:38 +00002605
2606 if (MI->isVariadic() && (A+1) == AEnd) {
Dylan Noblesmith2c1dd272012-02-05 02:13:05 +00002607 SmallString<32> Arg = (*A)->getName();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002608 if (MI->isC99Varargs())
2609 Arg += ", ...";
2610 else
2611 Arg += "...";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002612 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor3aa55262012-01-21 00:43:38 +00002613 break;
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002614 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002615
2616 // Non-variadic macros are simple.
2617 Result.AddPlaceholderChunk(
2618 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregor0c505312011-07-30 08:17:44 +00002619 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002620 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002621 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002622 }
2623
Douglas Gregorf64acca2010-05-25 21:41:55 +00002624 assert(Kind == RK_Declaration && "Missed a result kind?");
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002625 const NamedDecl *ND = Declaration;
Douglas Gregor78254c82012-03-27 23:34:16 +00002626 Result.addParentContext(ND->getDeclContext());
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002627
2628 if (IncludeBriefComments) {
2629 // Add documentation comment, if it exists.
Dmitri Gribenkoa43ec182012-08-11 00:51:43 +00002630 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(ND)) {
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002631 Result.addBriefComment(RC->getBriefText(Ctx));
Fariborz Jahanian15a0b552013-02-28 17:47:14 +00002632 }
2633 else if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2634 if (OMD->isPropertyAccessor())
2635 if (const ObjCPropertyDecl *PDecl = OMD->findPropertyDecl())
2636 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(PDecl))
2637 Result.addBriefComment(RC->getBriefText(Ctx));
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002638 }
2639
Douglas Gregor9eb77012009-11-07 00:00:49 +00002640 if (StartsNestedNameSpecifier) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002641 Result.AddTypedTextChunk(
2642 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002643 Result.AddTextChunk("::");
2644 return Result.TakeString();
Douglas Gregor9eb77012009-11-07 00:00:49 +00002645 }
Erik Verbruggen98ea7f62011-10-14 15:31:08 +00002646
2647 for (Decl::attr_iterator i = ND->attr_begin(); i != ND->attr_end(); ++i) {
2648 if (AnnotateAttr *Attr = dyn_cast_or_null<AnnotateAttr>(*i)) {
2649 Result.AddAnnotation(Result.getAllocator().CopyString(Attr->getAnnotation()));
2650 }
2651 }
Douglas Gregor9eb77012009-11-07 00:00:49 +00002652
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002653 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002654
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002655 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002656 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002657 Ctx, Policy);
2658 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002659 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002660 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002661 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002662 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002663 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002664 }
2665
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002666 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002667 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002668 Ctx, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002669 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002670 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002671
Douglas Gregor3545ff42009-09-21 16:56:56 +00002672 // Figure out which template parameters are deduced (or have default
2673 // arguments).
Benjamin Kramere0513cb2012-01-30 16:17:39 +00002674 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002675 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002676 unsigned LastDeducibleArgument;
2677 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2678 --LastDeducibleArgument) {
2679 if (!Deduced[LastDeducibleArgument - 1]) {
2680 // C++0x: Figure out if the template argument has a default. If so,
2681 // the user doesn't need to type this argument.
2682 // FIXME: We need to abstract template parameters better!
2683 bool HasDefaultArg = false;
2684 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002685 LastDeducibleArgument - 1);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002686 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2687 HasDefaultArg = TTP->hasDefaultArgument();
2688 else if (NonTypeTemplateParmDecl *NTTP
2689 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2690 HasDefaultArg = NTTP->hasDefaultArgument();
2691 else {
2692 assert(isa<TemplateTemplateParmDecl>(Param));
2693 HasDefaultArg
Douglas Gregor9eb77012009-11-07 00:00:49 +00002694 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002695 }
2696
2697 if (!HasDefaultArg)
2698 break;
2699 }
2700 }
2701
2702 if (LastDeducibleArgument) {
2703 // Some of the function template arguments cannot be deduced from a
2704 // function call, so we introduce an explicit template argument list
2705 // containing all of the arguments up to the first deducible argument.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002706 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002707 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor3545ff42009-09-21 16:56:56 +00002708 LastDeducibleArgument);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002709 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002710 }
2711
2712 // Add the function parameters
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002713 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002714 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002715 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002716 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002717 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002718 }
2719
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002720 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002721 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002722 Ctx, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002723 Result.AddTypedTextChunk(
2724 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002725 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002726 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002727 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002728 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002729 }
2730
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002731 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregord3c5d792009-11-17 16:44:22 +00002732 Selector Sel = Method->getSelector();
2733 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002734 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002735 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002736 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002737 }
2738
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002739 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregor1b605f72009-11-19 01:08:35 +00002740 SelName += ':';
2741 if (StartParameter == 0)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002742 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002743 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002744 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002745
2746 // If there is only one parameter, and we're past it, add an empty
2747 // typed-text chunk since there is nothing to type.
2748 if (Method->param_size() == 1)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002749 Result.AddTypedTextChunk("");
Douglas Gregor1b605f72009-11-19 01:08:35 +00002750 }
Douglas Gregord3c5d792009-11-17 16:44:22 +00002751 unsigned Idx = 0;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002752 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2753 PEnd = Method->param_end();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002754 P != PEnd; (void)++P, ++Idx) {
2755 if (Idx > 0) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00002756 std::string Keyword;
2757 if (Idx > StartParameter)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002758 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregord3c5d792009-11-17 16:44:22 +00002759 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramer632500c2011-07-26 16:59:25 +00002760 Keyword += II->getName();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002761 Keyword += ":";
Douglas Gregor95887f92010-07-08 23:20:03 +00002762 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002763 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002764 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002765 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002766 }
Douglas Gregor1b605f72009-11-19 01:08:35 +00002767
2768 // If we're before the starting parameter, skip the placeholder.
2769 if (Idx < StartParameter)
2770 continue;
Douglas Gregord3c5d792009-11-17 16:44:22 +00002771
2772 std::string Arg;
Douglas Gregore90dd002010-08-24 16:15:59 +00002773
2774 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002775 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregore90dd002010-08-24 16:15:59 +00002776 else {
John McCall31168b02011-06-15 23:02:42 +00002777 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor8f08d742011-07-30 07:55:26 +00002778 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2779 + Arg + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002780 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregor981a0c42010-08-29 19:47:46 +00002781 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramer632500c2011-07-26 16:59:25 +00002782 Arg += II->getName();
Douglas Gregore90dd002010-08-24 16:15:59 +00002783 }
2784
Douglas Gregor400f5972010-08-31 05:13:43 +00002785 if (Method->isVariadic() && (P + 1) == PEnd)
2786 Arg += ", ...";
2787
Douglas Gregor95887f92010-07-08 23:20:03 +00002788 if (DeclaringEntity)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002789 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor95887f92010-07-08 23:20:03 +00002790 else if (AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002791 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8537c52009-11-19 07:41:15 +00002792 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002793 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002794 }
2795
Douglas Gregor04c5f972009-12-23 00:21:46 +00002796 if (Method->isVariadic()) {
Douglas Gregor400f5972010-08-31 05:13:43 +00002797 if (Method->param_size() == 0) {
2798 if (DeclaringEntity)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002799 Result.AddTextChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002800 else if (AllParametersAreInformative)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002801 Result.AddInformativeChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002802 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002803 Result.AddPlaceholderChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002804 }
Douglas Gregordbb71db2010-08-23 23:51:41 +00002805
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002806 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor04c5f972009-12-23 00:21:46 +00002807 }
2808
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002809 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002810 }
2811
Douglas Gregorf09935f2009-12-01 05:55:20 +00002812 if (Qualifier)
Douglas Gregor5bf52692009-09-22 23:15:58 +00002813 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002814 Ctx, Policy);
Douglas Gregorf09935f2009-12-01 05:55:20 +00002815
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002816 Result.AddTypedTextChunk(
2817 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002818 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002819}
2820
Douglas Gregorf0f51982009-09-23 00:34:09 +00002821CodeCompletionString *
2822CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2823 unsigned CurrentArg,
Douglas Gregor36e3b5c2010-10-11 21:37:58 +00002824 Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002825 CodeCompletionAllocator &Allocator,
2826 CodeCompletionTUInfo &CCTUInfo) const {
Douglas Gregor75acd922011-09-27 23:30:47 +00002827 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCall31168b02011-06-15 23:02:42 +00002828
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002829 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002830 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002831 FunctionDecl *FDecl = getFunction();
Douglas Gregor75acd922011-09-27 23:30:47 +00002832 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002833 const FunctionProtoType *Proto
2834 = dyn_cast<FunctionProtoType>(getFunctionType());
2835 if (!FDecl && !Proto) {
2836 // Function without a prototype. Just give the return type and a
2837 // highlighted ellipsis.
2838 const FunctionType *FT = getFunctionType();
Douglas Gregor304f9b02011-02-01 21:15:40 +00002839 Result.AddTextChunk(GetCompletionTypeString(FT->getResultType(),
Douglas Gregor75acd922011-09-27 23:30:47 +00002840 S.Context, Policy,
Douglas Gregor304f9b02011-02-01 21:15:40 +00002841 Result.getAllocator()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002842 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2843 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2844 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002845 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002846 }
2847
2848 if (FDecl)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002849 Result.AddTextChunk(
2850 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002851 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002852 Result.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002853 Result.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00002854 Proto->getResultType().getAsString(Policy)));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002855
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002856 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Alp Toker9cacbab2014-01-20 20:26:09 +00002857 unsigned NumParams = FDecl ? FDecl->getNumParams() : Proto->getNumParams();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002858 for (unsigned I = 0; I != NumParams; ++I) {
2859 if (I)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002860 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002861
2862 std::string ArgString;
2863 QualType ArgType;
2864
2865 if (FDecl) {
2866 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2867 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2868 } else {
Alp Toker9cacbab2014-01-20 20:26:09 +00002869 ArgType = Proto->getParamType(I);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002870 }
2871
John McCall31168b02011-06-15 23:02:42 +00002872 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002873
2874 if (I == CurrentArg)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002875 Result.AddChunk(CodeCompletionString::CK_CurrentParameter,
2876 Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002877 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002878 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002879 }
2880
2881 if (Proto && Proto->isVariadic()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002882 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002883 if (CurrentArg < NumParams)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002884 Result.AddTextChunk("...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002885 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002886 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002887 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002888 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002889
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002890 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002891}
2892
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002893unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002894 const LangOptions &LangOpts,
Douglas Gregor6e240332010-08-16 16:18:59 +00002895 bool PreferredTypeIsPointer) {
2896 unsigned Priority = CCP_Macro;
2897
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002898 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2899 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2900 MacroName.equals("Nil")) {
Douglas Gregor6e240332010-08-16 16:18:59 +00002901 Priority = CCP_Constant;
2902 if (PreferredTypeIsPointer)
2903 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002904 }
2905 // Treat "YES", "NO", "true", and "false" as constants.
2906 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2907 MacroName.equals("true") || MacroName.equals("false"))
2908 Priority = CCP_Constant;
2909 // Treat "bool" as a type.
2910 else if (MacroName.equals("bool"))
2911 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2912
Douglas Gregor6e240332010-08-16 16:18:59 +00002913
2914 return Priority;
2915}
2916
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002917CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002918 if (!D)
2919 return CXCursor_UnexposedDecl;
2920
2921 switch (D->getKind()) {
2922 case Decl::Enum: return CXCursor_EnumDecl;
2923 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2924 case Decl::Field: return CXCursor_FieldDecl;
2925 case Decl::Function:
2926 return CXCursor_FunctionDecl;
2927 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2928 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002929 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregordeafd0b2011-12-27 22:43:10 +00002930
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002931 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002932 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2933 case Decl::ObjCMethod:
2934 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2935 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2936 case Decl::CXXMethod: return CXCursor_CXXMethod;
2937 case Decl::CXXConstructor: return CXCursor_Constructor;
2938 case Decl::CXXDestructor: return CXCursor_Destructor;
2939 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2940 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002941 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002942 case Decl::ParmVar: return CXCursor_ParmDecl;
2943 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smithdda56e42011-04-15 14:24:37 +00002944 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002945 case Decl::Var: return CXCursor_VarDecl;
2946 case Decl::Namespace: return CXCursor_Namespace;
2947 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2948 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2949 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2950 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2951 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2952 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis12afd702011-09-30 17:58:23 +00002953 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002954 case Decl::ClassTemplatePartialSpecialization:
2955 return CXCursor_ClassTemplatePartialSpecialization;
2956 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor3e653b32012-04-30 23:41:16 +00002957 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002958
2959 case Decl::Using:
2960 case Decl::UnresolvedUsingValue:
2961 case Decl::UnresolvedUsingTypename:
2962 return CXCursor_UsingDeclaration;
2963
Douglas Gregor4cd65962011-06-03 23:08:58 +00002964 case Decl::ObjCPropertyImpl:
2965 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2966 case ObjCPropertyImplDecl::Dynamic:
2967 return CXCursor_ObjCDynamicDecl;
2968
2969 case ObjCPropertyImplDecl::Synthesize:
2970 return CXCursor_ObjCSynthesizeDecl;
2971 }
Argyrios Kyrtzidis50e5b1d2012-10-05 00:22:24 +00002972
2973 case Decl::Import:
2974 return CXCursor_ModuleImportDecl;
Douglas Gregor4cd65962011-06-03 23:08:58 +00002975
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002976 default:
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002977 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002978 switch (TD->getTagKind()) {
Joao Matosdc86f942012-08-31 18:45:21 +00002979 case TTK_Interface: // fall through
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002980 case TTK_Struct: return CXCursor_StructDecl;
2981 case TTK_Class: return CXCursor_ClassDecl;
2982 case TTK_Union: return CXCursor_UnionDecl;
2983 case TTK_Enum: return CXCursor_EnumDecl;
2984 }
2985 }
2986 }
2987
2988 return CXCursor_UnexposedDecl;
2989}
2990
Douglas Gregor55b037b2010-07-08 20:55:51 +00002991static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor8cb17462012-10-09 16:01:50 +00002992 bool IncludeUndefined,
Douglas Gregor55b037b2010-07-08 20:55:51 +00002993 bool TargetTypeIsPointer = false) {
John McCall276321a2010-08-25 06:19:51 +00002994 typedef CodeCompletionResult Result;
Douglas Gregor55b037b2010-07-08 20:55:51 +00002995
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002996 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002997
Douglas Gregor9eb77012009-11-07 00:00:49 +00002998 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2999 MEnd = PP.macro_end();
Douglas Gregor55b037b2010-07-08 20:55:51 +00003000 M != MEnd; ++M) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00003001 if (IncludeUndefined || M->first->hasMacroDefinition())
3002 Results.AddResult(Result(M->first,
Douglas Gregor6e240332010-08-16 16:18:59 +00003003 getMacroUsagePriority(M->first->getName(),
David Blaikiebbafb8a2012-03-11 07:00:24 +00003004 PP.getLangOpts(),
Douglas Gregor6e240332010-08-16 16:18:59 +00003005 TargetTypeIsPointer)));
Douglas Gregor55b037b2010-07-08 20:55:51 +00003006 }
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003007
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003008 Results.ExitScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003009
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003010}
3011
Douglas Gregorce0e8562010-08-23 21:54:33 +00003012static void AddPrettyFunctionResults(const LangOptions &LangOpts,
3013 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003014 typedef CodeCompletionResult Result;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003015
3016 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003017
Douglas Gregorce0e8562010-08-23 21:54:33 +00003018 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
3019 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith2bf7fdb2013-01-02 11:42:31 +00003020 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregorce0e8562010-08-23 21:54:33 +00003021 Results.AddResult(Result("__func__", CCP_Constant));
3022 Results.ExitScope();
3023}
3024
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003025static void HandleCodeCompleteResults(Sema *S,
3026 CodeCompleteConsumer *CodeCompleter,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003027 CodeCompletionContext Context,
John McCall276321a2010-08-25 06:19:51 +00003028 CodeCompletionResult *Results,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003029 unsigned NumResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00003030 if (CodeCompleter)
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003031 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor3545ff42009-09-21 16:56:56 +00003032}
3033
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003034static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3035 Sema::ParserCompletionContext PCC) {
3036 switch (PCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00003037 case Sema::PCC_Namespace:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003038 return CodeCompletionContext::CCC_TopLevel;
3039
John McCallfaf5fb42010-08-26 23:41:50 +00003040 case Sema::PCC_Class:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003041 return CodeCompletionContext::CCC_ClassStructUnion;
3042
John McCallfaf5fb42010-08-26 23:41:50 +00003043 case Sema::PCC_ObjCInterface:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003044 return CodeCompletionContext::CCC_ObjCInterface;
3045
John McCallfaf5fb42010-08-26 23:41:50 +00003046 case Sema::PCC_ObjCImplementation:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003047 return CodeCompletionContext::CCC_ObjCImplementation;
3048
John McCallfaf5fb42010-08-26 23:41:50 +00003049 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003050 return CodeCompletionContext::CCC_ObjCIvarList;
3051
John McCallfaf5fb42010-08-26 23:41:50 +00003052 case Sema::PCC_Template:
3053 case Sema::PCC_MemberTemplate:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003054 if (S.CurContext->isFileContext())
3055 return CodeCompletionContext::CCC_TopLevel;
David Blaikie8a40f702012-01-17 06:56:22 +00003056 if (S.CurContext->isRecord())
Douglas Gregor0ac41382010-09-23 23:01:17 +00003057 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie8a40f702012-01-17 06:56:22 +00003058 return CodeCompletionContext::CCC_Other;
Douglas Gregor0ac41382010-09-23 23:01:17 +00003059
John McCallfaf5fb42010-08-26 23:41:50 +00003060 case Sema::PCC_RecoveryInFunction:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003061 return CodeCompletionContext::CCC_Recovery;
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003062
John McCallfaf5fb42010-08-26 23:41:50 +00003063 case Sema::PCC_ForInit:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003064 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3065 S.getLangOpts().ObjC1)
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003066 return CodeCompletionContext::CCC_ParenthesizedExpression;
3067 else
3068 return CodeCompletionContext::CCC_Expression;
3069
3070 case Sema::PCC_Expression:
John McCallfaf5fb42010-08-26 23:41:50 +00003071 case Sema::PCC_Condition:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003072 return CodeCompletionContext::CCC_Expression;
3073
John McCallfaf5fb42010-08-26 23:41:50 +00003074 case Sema::PCC_Statement:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003075 return CodeCompletionContext::CCC_Statement;
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003076
John McCallfaf5fb42010-08-26 23:41:50 +00003077 case Sema::PCC_Type:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003078 return CodeCompletionContext::CCC_Type;
Douglas Gregor5e35d592010-09-14 23:59:36 +00003079
3080 case Sema::PCC_ParenthesizedExpression:
3081 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor80039242011-02-15 20:33:25 +00003082
3083 case Sema::PCC_LocalDeclarationSpecifiers:
3084 return CodeCompletionContext::CCC_Type;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003085 }
David Blaikie8a40f702012-01-17 06:56:22 +00003086
3087 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003088}
3089
Douglas Gregorac322ec2010-08-27 21:18:54 +00003090/// \brief If we're in a C++ virtual member function, add completion results
3091/// that invoke the functions we override, since it's common to invoke the
3092/// overridden function as well as adding new functionality.
3093///
3094/// \param S The semantic analysis object for which we are generating results.
3095///
3096/// \param InContext This context in which the nested-name-specifier preceding
3097/// the code-completion point
3098static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3099 ResultBuilder &Results) {
3100 // Look through blocks.
3101 DeclContext *CurContext = S.CurContext;
3102 while (isa<BlockDecl>(CurContext))
3103 CurContext = CurContext->getParent();
3104
3105
3106 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3107 if (!Method || !Method->isVirtual())
3108 return;
3109
3110 // We need to have names for all of the parameters, if we're going to
3111 // generate a forwarding call.
3112 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3113 PEnd = Method->param_end();
3114 P != PEnd;
3115 ++P) {
3116 if (!(*P)->getDeclName())
3117 return;
3118 }
3119
Douglas Gregor75acd922011-09-27 23:30:47 +00003120 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003121 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3122 MEnd = Method->end_overridden_methods();
3123 M != MEnd; ++M) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003124 CodeCompletionBuilder Builder(Results.getAllocator(),
3125 Results.getCodeCompletionTUInfo());
Dmitri Gribenko6cfb1532013-02-14 13:53:30 +00003126 const CXXMethodDecl *Overridden = *M;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003127 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3128 continue;
3129
3130 // If we need a nested-name-specifier, add one now.
3131 if (!InContext) {
3132 NestedNameSpecifier *NNS
3133 = getRequiredQualification(S.Context, CurContext,
3134 Overridden->getDeclContext());
3135 if (NNS) {
3136 std::string Str;
3137 llvm::raw_string_ostream OS(Str);
Douglas Gregor75acd922011-09-27 23:30:47 +00003138 NNS->print(OS, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003139 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003140 }
3141 } else if (!InContext->Equals(Overridden->getDeclContext()))
3142 continue;
3143
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003144 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003145 Overridden->getNameAsString()));
3146 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003147 bool FirstParam = true;
3148 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3149 PEnd = Method->param_end();
3150 P != PEnd; ++P) {
3151 if (FirstParam)
3152 FirstParam = false;
3153 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003154 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003155
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003156 Builder.AddPlaceholderChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003157 (*P)->getIdentifier()->getName()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003158 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003159 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3160 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorac322ec2010-08-27 21:18:54 +00003161 CCP_SuperCompletion,
Douglas Gregor78254c82012-03-27 23:34:16 +00003162 CXCursor_CXXMethod,
3163 CXAvailability_Available,
3164 Overridden));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003165 Results.Ignore(Overridden);
3166 }
3167}
3168
Douglas Gregor07f43572012-01-29 18:15:03 +00003169void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3170 ModuleIdPath Path) {
3171 typedef CodeCompletionResult Result;
3172 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003173 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor07f43572012-01-29 18:15:03 +00003174 CodeCompletionContext::CCC_Other);
3175 Results.EnterNewScope();
3176
3177 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003178 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor07f43572012-01-29 18:15:03 +00003179 typedef CodeCompletionResult Result;
3180 if (Path.empty()) {
3181 // Enumerate all top-level modules.
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003182 SmallVector<Module *, 8> Modules;
Douglas Gregor07f43572012-01-29 18:15:03 +00003183 PP.getHeaderSearchInfo().collectAllModules(Modules);
3184 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3185 Builder.AddTypedTextChunk(
3186 Builder.getAllocator().CopyString(Modules[I]->Name));
3187 Results.AddResult(Result(Builder.TakeString(),
3188 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003189 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003190 Modules[I]->isAvailable()
3191 ? CXAvailability_Available
3192 : CXAvailability_NotAvailable));
3193 }
Daniel Jasper07e6c402013-08-05 20:26:17 +00003194 } else if (getLangOpts().Modules) {
Douglas Gregor07f43572012-01-29 18:15:03 +00003195 // Load the named module.
3196 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3197 Module::AllVisible,
3198 /*IsInclusionDirective=*/false);
3199 // Enumerate submodules.
3200 if (Mod) {
3201 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3202 SubEnd = Mod->submodule_end();
3203 Sub != SubEnd; ++Sub) {
3204
3205 Builder.AddTypedTextChunk(
3206 Builder.getAllocator().CopyString((*Sub)->Name));
3207 Results.AddResult(Result(Builder.TakeString(),
3208 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003209 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003210 (*Sub)->isAvailable()
3211 ? CXAvailability_Available
3212 : CXAvailability_NotAvailable));
3213 }
3214 }
3215 }
3216 Results.ExitScope();
3217 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3218 Results.data(),Results.size());
3219}
3220
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003221void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003222 ParserCompletionContext CompletionContext) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003223 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003224 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003225 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003226 Results.EnterNewScope();
Douglas Gregor50832e02010-09-20 22:39:41 +00003227
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003228 // Determine how to filter results, e.g., so that the names of
3229 // values (functions, enumerators, function templates, etc.) are
3230 // only allowed where we can have an expression.
3231 switch (CompletionContext) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003232 case PCC_Namespace:
3233 case PCC_Class:
3234 case PCC_ObjCInterface:
3235 case PCC_ObjCImplementation:
3236 case PCC_ObjCInstanceVariableList:
3237 case PCC_Template:
3238 case PCC_MemberTemplate:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003239 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003240 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003241 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3242 break;
3243
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003244 case PCC_Statement:
Douglas Gregor5e35d592010-09-14 23:59:36 +00003245 case PCC_ParenthesizedExpression:
Douglas Gregor4d755e82010-08-24 23:58:17 +00003246 case PCC_Expression:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003247 case PCC_ForInit:
3248 case PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003249 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor70febae2010-05-28 00:49:12 +00003250 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3251 else
3252 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003253
David Blaikiebbafb8a2012-03-11 07:00:24 +00003254 if (getLangOpts().CPlusPlus)
Douglas Gregorac322ec2010-08-27 21:18:54 +00003255 MaybeAddOverrideCalls(*this, /*InContext=*/0, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003256 break;
Douglas Gregor6da3db42010-05-25 05:58:43 +00003257
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003258 case PCC_RecoveryInFunction:
Douglas Gregor6da3db42010-05-25 05:58:43 +00003259 // Unfiltered
3260 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003261 }
3262
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003263 // If we are in a C++ non-static member function, check the qualifiers on
3264 // the member function to filter/prioritize the results list.
3265 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3266 if (CurMethod->isInstance())
3267 Results.setObjectTypeQualifiers(
3268 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3269
Douglas Gregorc580c522010-01-14 01:09:38 +00003270 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003271 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3272 CodeCompleter->includeGlobals());
Douglas Gregor92253692009-12-07 09:54:55 +00003273
Douglas Gregorf98e6a22010-01-13 23:51:12 +00003274 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor92253692009-12-07 09:54:55 +00003275 Results.ExitScope();
3276
Douglas Gregorce0e8562010-08-23 21:54:33 +00003277 switch (CompletionContext) {
Douglas Gregor5e35d592010-09-14 23:59:36 +00003278 case PCC_ParenthesizedExpression:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003279 case PCC_Expression:
3280 case PCC_Statement:
3281 case PCC_RecoveryInFunction:
3282 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00003283 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003284 break;
3285
3286 case PCC_Namespace:
3287 case PCC_Class:
3288 case PCC_ObjCInterface:
3289 case PCC_ObjCImplementation:
3290 case PCC_ObjCInstanceVariableList:
3291 case PCC_Template:
3292 case PCC_MemberTemplate:
3293 case PCC_ForInit:
3294 case PCC_Condition:
3295 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003296 case PCC_LocalDeclarationSpecifiers:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003297 break;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003298 }
3299
Douglas Gregor9eb77012009-11-07 00:00:49 +00003300 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003301 AddMacroResults(PP, Results, false);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003302
Douglas Gregor50832e02010-09-20 22:39:41 +00003303 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003304 Results.data(),Results.size());
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00003305}
3306
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003307static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3308 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003309 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00003310 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003311 bool IsSuper,
3312 ResultBuilder &Results);
3313
3314void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3315 bool AllowNonIdentifiers,
3316 bool AllowNestedNameSpecifiers) {
John McCall276321a2010-08-25 06:19:51 +00003317 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003318 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003319 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003320 AllowNestedNameSpecifiers
3321 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3322 : CodeCompletionContext::CCC_Name);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003323 Results.EnterNewScope();
3324
3325 // Type qualifiers can come after names.
3326 Results.AddResult(Result("const"));
3327 Results.AddResult(Result("volatile"));
David Blaikiebbafb8a2012-03-11 07:00:24 +00003328 if (getLangOpts().C99)
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003329 Results.AddResult(Result("restrict"));
3330
David Blaikiebbafb8a2012-03-11 07:00:24 +00003331 if (getLangOpts().CPlusPlus) {
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003332 if (AllowNonIdentifiers) {
3333 Results.AddResult(Result("operator"));
3334 }
3335
3336 // Add nested-name-specifiers.
3337 if (AllowNestedNameSpecifiers) {
3338 Results.allowNestedNameSpecifiers();
Douglas Gregor0ac41382010-09-23 23:01:17 +00003339 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003340 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3341 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3342 CodeCompleter->includeGlobals());
Douglas Gregor0ac41382010-09-23 23:01:17 +00003343 Results.setFilter(0);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003344 }
3345 }
3346 Results.ExitScope();
3347
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003348 // If we're in a context where we might have an expression (rather than a
3349 // declaration), and what we've seen so far is an Objective-C type that could
3350 // be a receiver of a class message, this may be a class message send with
3351 // the initial opening bracket '[' missing. Add appropriate completions.
3352 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003353 DS.getParsedSpecifiers() == DeclSpec::PQ_TypeSpecifier &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003354 DS.getTypeSpecType() == DeclSpec::TST_typename &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003355 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3356 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003357 !DS.isTypeAltiVecVector() &&
3358 S &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003359 (S->getFlags() & Scope::DeclScope) != 0 &&
3360 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3361 Scope::FunctionPrototypeScope |
3362 Scope::AtCatchScope)) == 0) {
3363 ParsedType T = DS.getRepAsType();
3364 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003365 AddClassMessageCompletions(*this, S, T, None, false, false, Results);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003366 }
3367
Douglas Gregor56ccce02010-08-24 04:59:56 +00003368 // Note that we intentionally suppress macro results here, since we do not
3369 // encourage using macros to produce the names of entities.
3370
Douglas Gregor0ac41382010-09-23 23:01:17 +00003371 HandleCodeCompleteResults(this, CodeCompleter,
3372 Results.getCompletionContext(),
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003373 Results.data(), Results.size());
3374}
3375
Douglas Gregor68762e72010-08-23 21:17:50 +00003376struct Sema::CodeCompleteExpressionData {
3377 CodeCompleteExpressionData(QualType PreferredType = QualType())
3378 : PreferredType(PreferredType), IntegralConstantExpression(false),
3379 ObjCCollection(false) { }
3380
3381 QualType PreferredType;
3382 bool IntegralConstantExpression;
3383 bool ObjCCollection;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003384 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregor68762e72010-08-23 21:17:50 +00003385};
3386
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003387/// \brief Perform code-completion in an expression context when we know what
3388/// type we're looking for.
Douglas Gregor68762e72010-08-23 21:17:50 +00003389void Sema::CodeCompleteExpression(Scope *S,
3390 const CodeCompleteExpressionData &Data) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003391 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003392 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003393 CodeCompletionContext::CCC_Expression);
Douglas Gregor68762e72010-08-23 21:17:50 +00003394 if (Data.ObjCCollection)
3395 Results.setFilter(&ResultBuilder::IsObjCCollection);
3396 else if (Data.IntegralConstantExpression)
Douglas Gregor85b50632010-07-28 21:50:18 +00003397 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003398 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003399 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3400 else
3401 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregor68762e72010-08-23 21:17:50 +00003402
3403 if (!Data.PreferredType.isNull())
3404 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3405
3406 // Ignore any declarations that we were told that we don't care about.
3407 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3408 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003409
3410 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003411 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3412 CodeCompleter->includeGlobals());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003413
3414 Results.EnterNewScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003415 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003416 Results.ExitScope();
3417
Douglas Gregor55b037b2010-07-08 20:55:51 +00003418 bool PreferredTypeIsPointer = false;
Douglas Gregor68762e72010-08-23 21:17:50 +00003419 if (!Data.PreferredType.isNull())
3420 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3421 || Data.PreferredType->isMemberPointerType()
3422 || Data.PreferredType->isBlockPointerType();
Douglas Gregor55b037b2010-07-08 20:55:51 +00003423
Douglas Gregorce0e8562010-08-23 21:54:33 +00003424 if (S->getFnParent() &&
3425 !Data.ObjCCollection &&
3426 !Data.IntegralConstantExpression)
David Blaikiebbafb8a2012-03-11 07:00:24 +00003427 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003428
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003429 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003430 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003431 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor68762e72010-08-23 21:17:50 +00003432 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3433 Data.PreferredType),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003434 Results.data(),Results.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003435}
3436
Douglas Gregoreda7e542010-09-18 01:28:11 +00003437void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3438 if (E.isInvalid())
3439 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003440 else if (getLangOpts().ObjC1)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003441 CodeCompleteObjCInstanceMessage(S, E.take(), None, false);
Douglas Gregored0b69d2010-09-15 16:23:04 +00003442}
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003443
Douglas Gregorb888acf2010-12-09 23:01:55 +00003444/// \brief The set of properties that have already been added, referenced by
3445/// property name.
3446typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3447
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003448/// \brief Retrieve the container definition, if any?
3449static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3450 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3451 if (Interface->hasDefinition())
3452 return Interface->getDefinition();
3453
3454 return Interface;
3455 }
3456
3457 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3458 if (Protocol->hasDefinition())
3459 return Protocol->getDefinition();
3460
3461 return Protocol;
3462 }
3463 return Container;
3464}
3465
3466static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor5d649882009-11-18 22:32:06 +00003467 bool AllowCategories,
Douglas Gregor95147142011-05-05 15:50:42 +00003468 bool AllowNullaryMethods,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003469 DeclContext *CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003470 AddedPropertiesSet &AddedProperties,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003471 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003472 typedef CodeCompletionResult Result;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003473
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003474 // Retrieve the definition.
3475 Container = getContainerDef(Container);
3476
Douglas Gregor9291bad2009-11-18 01:29:26 +00003477 // Add properties in this container.
3478 for (ObjCContainerDecl::prop_iterator P = Container->prop_begin(),
3479 PEnd = Container->prop_end();
3480 P != PEnd;
Douglas Gregorb888acf2010-12-09 23:01:55 +00003481 ++P) {
3482 if (AddedProperties.insert(P->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003483 Results.MaybeAddResult(Result(*P, Results.getBasePriority(*P), 0),
3484 CurContext);
Douglas Gregorb888acf2010-12-09 23:01:55 +00003485 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003486
Douglas Gregor95147142011-05-05 15:50:42 +00003487 // Add nullary methods
3488 if (AllowNullaryMethods) {
3489 ASTContext &Context = Container->getASTContext();
Douglas Gregor75acd922011-09-27 23:30:47 +00003490 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Douglas Gregor95147142011-05-05 15:50:42 +00003491 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
3492 MEnd = Container->meth_end();
3493 M != MEnd; ++M) {
3494 if (M->getSelector().isUnarySelector())
3495 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3496 if (AddedProperties.insert(Name)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003497 CodeCompletionBuilder Builder(Results.getAllocator(),
3498 Results.getCodeCompletionTUInfo());
David Blaikie40ed2972012-06-06 20:45:41 +00003499 AddResultTypeChunk(Context, Policy, *M, Builder);
Douglas Gregor95147142011-05-05 15:50:42 +00003500 Builder.AddTypedTextChunk(
3501 Results.getAllocator().CopyString(Name->getName()));
3502
David Blaikie40ed2972012-06-06 20:45:41 +00003503 Results.MaybeAddResult(Result(Builder.TakeString(), *M,
Douglas Gregor78254c82012-03-27 23:34:16 +00003504 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor95147142011-05-05 15:50:42 +00003505 CurContext);
3506 }
3507 }
3508 }
3509
3510
Douglas Gregor9291bad2009-11-18 01:29:26 +00003511 // Add properties in referenced protocols.
3512 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3513 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
3514 PEnd = Protocol->protocol_end();
3515 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003516 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3517 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003518 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor5d649882009-11-18 22:32:06 +00003519 if (AllowCategories) {
3520 // Look through categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00003521 for (ObjCInterfaceDecl::known_categories_iterator
3522 Cat = IFace->known_categories_begin(),
3523 CatEnd = IFace->known_categories_end();
3524 Cat != CatEnd; ++Cat)
3525 AddObjCProperties(*Cat, AllowCategories, AllowNullaryMethods,
Douglas Gregor95147142011-05-05 15:50:42 +00003526 CurContext, AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00003527 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003528
3529 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003530 for (ObjCInterfaceDecl::all_protocol_iterator
3531 I = IFace->all_referenced_protocol_begin(),
3532 E = IFace->all_referenced_protocol_end(); I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003533 AddObjCProperties(*I, AllowCategories, AllowNullaryMethods, CurContext,
3534 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003535
3536 // Look in the superclass.
3537 if (IFace->getSuperClass())
Douglas Gregor95147142011-05-05 15:50:42 +00003538 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3539 AllowNullaryMethods, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003540 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003541 } else if (const ObjCCategoryDecl *Category
3542 = dyn_cast<ObjCCategoryDecl>(Container)) {
3543 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003544 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
3545 PEnd = Category->protocol_end();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003546 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003547 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3548 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003549 }
3550}
3551
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003552void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor2436e712009-09-17 21:32:03 +00003553 SourceLocation OpLoc,
3554 bool IsArrow) {
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003555 if (!Base || !CodeCompleter)
Douglas Gregor2436e712009-09-17 21:32:03 +00003556 return;
3557
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003558 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3559 if (ConvertedBase.isInvalid())
3560 return;
3561 Base = ConvertedBase.get();
3562
John McCall276321a2010-08-25 06:19:51 +00003563 typedef CodeCompletionResult Result;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003564
Douglas Gregor2436e712009-09-17 21:32:03 +00003565 QualType BaseType = Base->getType();
Douglas Gregor3545ff42009-09-21 16:56:56 +00003566
3567 if (IsArrow) {
3568 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3569 BaseType = Ptr->getPointeeType();
3570 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003571 /*Do nothing*/ ;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003572 else
3573 return;
3574 }
3575
Douglas Gregor21325842011-07-07 16:03:39 +00003576 enum CodeCompletionContext::Kind contextKind;
3577
3578 if (IsArrow) {
3579 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3580 }
3581 else {
3582 if (BaseType->isObjCObjectPointerType() ||
3583 BaseType->isObjCObjectOrInterfaceType()) {
3584 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3585 }
3586 else {
3587 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3588 }
3589 }
3590
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003591 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003592 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00003593 CodeCompletionContext(contextKind,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003594 BaseType),
3595 &ResultBuilder::IsMember);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003596 Results.EnterNewScope();
3597 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003598 // Indicate that we are performing a member access, and the cv-qualifiers
3599 // for the base object type.
3600 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3601
Douglas Gregor9291bad2009-11-18 01:29:26 +00003602 // Access to a C/C++ class, struct, or union.
Douglas Gregor6ae4c522010-01-14 03:21:49 +00003603 Results.allowNestedNameSpecifiers();
Douglas Gregor09bbc652010-01-14 15:47:35 +00003604 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003605 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3606 CodeCompleter->includeGlobals());
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003607
David Blaikiebbafb8a2012-03-11 07:00:24 +00003608 if (getLangOpts().CPlusPlus) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003609 if (!Results.empty()) {
3610 // The "template" keyword can follow "->" or "." in the grammar.
3611 // However, we only want to suggest the template keyword if something
3612 // is dependent.
3613 bool IsDependent = BaseType->isDependentType();
3614 if (!IsDependent) {
3615 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
Ted Kremenekc37877d2013-10-08 17:08:03 +00003616 if (DeclContext *Ctx = DepScope->getEntity()) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003617 IsDependent = Ctx->isDependentContext();
3618 break;
3619 }
3620 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003621
Douglas Gregor9291bad2009-11-18 01:29:26 +00003622 if (IsDependent)
Douglas Gregor78a21012010-01-14 16:01:26 +00003623 Results.AddResult(Result("template"));
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003624 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003625 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003626 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3627 // Objective-C property reference.
Douglas Gregorb888acf2010-12-09 23:01:55 +00003628 AddedPropertiesSet AddedProperties;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003629
3630 // Add property results based on our interface.
3631 const ObjCObjectPointerType *ObjCPtr
3632 = BaseType->getAsObjCInterfacePointerType();
3633 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor95147142011-05-05 15:50:42 +00003634 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3635 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003636 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003637
3638 // Add properties from the protocols in a qualified interface.
3639 for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(),
3640 E = ObjCPtr->qual_end();
3641 I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003642 AddObjCProperties(*I, true, /*AllowNullaryMethods=*/true, CurContext,
3643 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003644 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCall8b07ec22010-05-15 11:32:37 +00003645 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003646 // Objective-C instance variable access.
3647 ObjCInterfaceDecl *Class = 0;
3648 if (const ObjCObjectPointerType *ObjCPtr
3649 = BaseType->getAs<ObjCObjectPointerType>())
3650 Class = ObjCPtr->getInterfaceDecl();
3651 else
John McCall8b07ec22010-05-15 11:32:37 +00003652 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003653
3654 // Add all ivars from this class and its superclasses.
Douglas Gregor2b8162b2010-01-14 16:08:12 +00003655 if (Class) {
3656 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3657 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor39982192010-08-15 06:18:01 +00003658 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3659 CodeCompleter->includeGlobals());
Douglas Gregor9291bad2009-11-18 01:29:26 +00003660 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003661 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003662
3663 // FIXME: How do we cope with isa?
3664
3665 Results.ExitScope();
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003666
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003667 // Hand off the results found for code completion.
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003668 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003669 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003670 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00003671}
3672
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003673void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3674 if (!CodeCompleter)
3675 return;
3676
Douglas Gregor3545ff42009-09-21 16:56:56 +00003677 ResultBuilder::LookupFilter Filter = 0;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003678 enum CodeCompletionContext::Kind ContextKind
3679 = CodeCompletionContext::CCC_Other;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003680 switch ((DeclSpec::TST)TagSpec) {
3681 case DeclSpec::TST_enum:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003682 Filter = &ResultBuilder::IsEnum;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003683 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003684 break;
3685
3686 case DeclSpec::TST_union:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003687 Filter = &ResultBuilder::IsUnion;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003688 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003689 break;
3690
3691 case DeclSpec::TST_struct:
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003692 case DeclSpec::TST_class:
Joao Matosdc86f942012-08-31 18:45:21 +00003693 case DeclSpec::TST_interface:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003694 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003695 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003696 break;
3697
3698 default:
David Blaikie83d382b2011-09-23 05:06:16 +00003699 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003700 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003701
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003702 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3703 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00003704 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCalle87beb22010-04-23 18:46:30 +00003705
3706 // First pass: look for tags.
3707 Results.setFilter(Filter);
Douglas Gregor39982192010-08-15 06:18:01 +00003708 LookupVisibleDecls(S, LookupTagName, Consumer,
3709 CodeCompleter->includeGlobals());
John McCalle87beb22010-04-23 18:46:30 +00003710
Douglas Gregor39982192010-08-15 06:18:01 +00003711 if (CodeCompleter->includeGlobals()) {
3712 // Second pass: look for nested name specifiers.
3713 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3714 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3715 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003716
Douglas Gregor0ac41382010-09-23 23:01:17 +00003717 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003718 Results.data(),Results.size());
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003719}
3720
Douglas Gregor28c78432010-08-27 17:35:51 +00003721void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003722 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003723 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003724 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor28c78432010-08-27 17:35:51 +00003725 Results.EnterNewScope();
3726 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3727 Results.AddResult("const");
3728 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3729 Results.AddResult("volatile");
David Blaikiebbafb8a2012-03-11 07:00:24 +00003730 if (getLangOpts().C99 &&
Douglas Gregor28c78432010-08-27 17:35:51 +00003731 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3732 Results.AddResult("restrict");
Richard Smith8e1ac332013-03-28 01:55:44 +00003733 if (getLangOpts().C11 &&
3734 !(DS.getTypeQualifiers() & DeclSpec::TQ_atomic))
3735 Results.AddResult("_Atomic");
Douglas Gregor28c78432010-08-27 17:35:51 +00003736 Results.ExitScope();
3737 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003738 Results.getCompletionContext(),
Douglas Gregor28c78432010-08-27 17:35:51 +00003739 Results.data(), Results.size());
3740}
3741
Douglas Gregord328d572009-09-21 18:10:23 +00003742void Sema::CodeCompleteCase(Scope *S) {
John McCallaab3e412010-08-25 08:40:02 +00003743 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregord328d572009-09-21 18:10:23 +00003744 return;
John McCall5939b162011-08-06 07:30:58 +00003745
John McCallaab3e412010-08-25 08:40:02 +00003746 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCall5939b162011-08-06 07:30:58 +00003747 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3748 if (!type->isEnumeralType()) {
3749 CodeCompleteExpressionData Data(type);
Douglas Gregor68762e72010-08-23 21:17:50 +00003750 Data.IntegralConstantExpression = true;
3751 CodeCompleteExpression(S, Data);
Douglas Gregord328d572009-09-21 18:10:23 +00003752 return;
Douglas Gregor85b50632010-07-28 21:50:18 +00003753 }
Douglas Gregord328d572009-09-21 18:10:23 +00003754
3755 // Code-complete the cases of a switch statement over an enumeration type
3756 // by providing the list of
John McCall5939b162011-08-06 07:30:58 +00003757 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003758 if (EnumDecl *Def = Enum->getDefinition())
3759 Enum = Def;
Douglas Gregord328d572009-09-21 18:10:23 +00003760
3761 // Determine which enumerators we have already seen in the switch statement.
3762 // FIXME: Ideally, we would also be able to look *past* the code-completion
3763 // token, in case we are code-completing in the middle of the switch and not
3764 // at the end. However, we aren't able to do so at the moment.
3765 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Douglas Gregorf2510672009-09-21 19:57:38 +00003766 NestedNameSpecifier *Qualifier = 0;
Douglas Gregord328d572009-09-21 18:10:23 +00003767 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3768 SC = SC->getNextSwitchCase()) {
3769 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3770 if (!Case)
3771 continue;
3772
3773 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3774 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3775 if (EnumConstantDecl *Enumerator
3776 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3777 // We look into the AST of the case statement to determine which
3778 // enumerator was named. Alternatively, we could compute the value of
3779 // the integral constant expression, then compare it against the
3780 // values of each enumerator. However, value-based approach would not
3781 // work as well with C++ templates where enumerators declared within a
3782 // template are type- and value-dependent.
3783 EnumeratorsSeen.insert(Enumerator);
3784
Douglas Gregorf2510672009-09-21 19:57:38 +00003785 // If this is a qualified-id, keep track of the nested-name-specifier
3786 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregord328d572009-09-21 18:10:23 +00003787 //
3788 // switch (TagD.getKind()) {
3789 // case TagDecl::TK_enum:
3790 // break;
3791 // case XXX
3792 //
Douglas Gregorf2510672009-09-21 19:57:38 +00003793 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregord328d572009-09-21 18:10:23 +00003794 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3795 // TK_struct, and TK_class.
Douglas Gregor4bd90e52009-10-23 18:54:35 +00003796 Qualifier = DRE->getQualifier();
Douglas Gregord328d572009-09-21 18:10:23 +00003797 }
3798 }
3799
David Blaikiebbafb8a2012-03-11 07:00:24 +00003800 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorf2510672009-09-21 19:57:38 +00003801 // If there are no prior enumerators in C++, check whether we have to
3802 // qualify the names of the enumerators that we suggest, because they
3803 // may not be visible in this scope.
Douglas Gregord3cebdb2012-02-01 05:02:47 +00003804 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorf2510672009-09-21 19:57:38 +00003805 }
3806
Douglas Gregord328d572009-09-21 18:10:23 +00003807 // Add any enumerators that have not yet been mentioned.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003808 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003809 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003810 CodeCompletionContext::CCC_Expression);
Douglas Gregord328d572009-09-21 18:10:23 +00003811 Results.EnterNewScope();
3812 for (EnumDecl::enumerator_iterator E = Enum->enumerator_begin(),
3813 EEnd = Enum->enumerator_end();
3814 E != EEnd; ++E) {
David Blaikie40ed2972012-06-06 20:45:41 +00003815 if (EnumeratorsSeen.count(*E))
Douglas Gregord328d572009-09-21 18:10:23 +00003816 continue;
3817
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003818 CodeCompletionResult R(*E, CCP_EnumInCase, Qualifier);
Douglas Gregor3a69eaf2011-02-18 23:30:37 +00003819 Results.AddResult(R, CurContext, 0, false);
Douglas Gregord328d572009-09-21 18:10:23 +00003820 }
3821 Results.ExitScope();
Douglas Gregor285560922010-04-06 20:02:15 +00003822
Douglas Gregor21325842011-07-07 16:03:39 +00003823 //We need to make sure we're setting the right context,
3824 //so only say we include macros if the code completer says we do
3825 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3826 if (CodeCompleter->includeMacros()) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00003827 AddMacroResults(PP, Results, false);
Douglas Gregor21325842011-07-07 16:03:39 +00003828 kind = CodeCompletionContext::CCC_OtherWithMacros;
3829 }
3830
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003831 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00003832 kind,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003833 Results.data(),Results.size());
Douglas Gregord328d572009-09-21 18:10:23 +00003834}
3835
Douglas Gregorcabea402009-09-22 15:41:20 +00003836namespace {
3837 struct IsBetterOverloadCandidate {
3838 Sema &S;
John McCallbc077cf2010-02-08 23:07:23 +00003839 SourceLocation Loc;
Douglas Gregorcabea402009-09-22 15:41:20 +00003840
3841 public:
John McCallbc077cf2010-02-08 23:07:23 +00003842 explicit IsBetterOverloadCandidate(Sema &S, SourceLocation Loc)
3843 : S(S), Loc(Loc) { }
Douglas Gregorcabea402009-09-22 15:41:20 +00003844
3845 bool
3846 operator()(const OverloadCandidate &X, const OverloadCandidate &Y) const {
John McCall5c32be02010-08-24 20:38:10 +00003847 return isBetterOverloadCandidate(S, X, Y, Loc);
Douglas Gregorcabea402009-09-22 15:41:20 +00003848 }
3849 };
3850}
3851
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003852static bool anyNullArguments(ArrayRef<Expr *> Args) {
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003853 if (Args.size() && !Args.data())
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003854 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003855
3856 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003857 if (!Args[I])
3858 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003859
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003860 return false;
3861}
3862
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003863void Sema::CodeCompleteCall(Scope *S, Expr *FnIn, ArrayRef<Expr *> Args) {
Douglas Gregorcabea402009-09-22 15:41:20 +00003864 if (!CodeCompleter)
3865 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003866
3867 // When we're code-completing for a call, we fall back to ordinary
3868 // name code-completion whenever we can't produce specific
3869 // results. We may want to revisit this strategy in the future,
3870 // e.g., by merging the two kinds of results.
3871
Douglas Gregorcabea402009-09-22 15:41:20 +00003872 Expr *Fn = (Expr *)FnIn;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003873
Douglas Gregorcabea402009-09-22 15:41:20 +00003874 // Ignore type-dependent call expressions entirely.
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003875 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3876 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003877 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregorcabea402009-09-22 15:41:20 +00003878 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003879 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003880
John McCall57500772009-12-16 12:17:52 +00003881 // Build an overload candidate set based on the functions we find.
John McCallbc077cf2010-02-08 23:07:23 +00003882 SourceLocation Loc = Fn->getExprLoc();
3883 OverloadCandidateSet CandidateSet(Loc);
John McCall57500772009-12-16 12:17:52 +00003884
Douglas Gregorcabea402009-09-22 15:41:20 +00003885 // FIXME: What if we're calling something that isn't a function declaration?
3886 // FIXME: What if we're calling a pseudo-destructor?
3887 // FIXME: What if we're calling a member function?
3888
Douglas Gregorff59f672010-01-21 15:46:19 +00003889 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003890 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorff59f672010-01-21 15:46:19 +00003891
John McCall57500772009-12-16 12:17:52 +00003892 Expr *NakedFn = Fn->IgnoreParenCasts();
3893 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003894 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
John McCall57500772009-12-16 12:17:52 +00003895 /*PartialOverloading=*/ true);
3896 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3897 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorff59f672010-01-21 15:46:19 +00003898 if (FDecl) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00003899 if (!getLangOpts().CPlusPlus ||
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003900 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorff59f672010-01-21 15:46:19 +00003901 Results.push_back(ResultCandidate(FDecl));
3902 else
John McCallb89836b2010-01-26 01:37:31 +00003903 // FIXME: access?
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003904 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), Args,
3905 CandidateSet, false, /*PartialOverloading*/true);
Douglas Gregorff59f672010-01-21 15:46:19 +00003906 }
John McCall57500772009-12-16 12:17:52 +00003907 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003908
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003909 QualType ParamType;
3910
Douglas Gregorff59f672010-01-21 15:46:19 +00003911 if (!CandidateSet.empty()) {
3912 // Sort the overload candidate set by placing the best overloads first.
3913 std::stable_sort(CandidateSet.begin(), CandidateSet.end(),
John McCallbc077cf2010-02-08 23:07:23 +00003914 IsBetterOverloadCandidate(*this, Loc));
Douglas Gregorcabea402009-09-22 15:41:20 +00003915
Douglas Gregorff59f672010-01-21 15:46:19 +00003916 // Add the remaining viable overload candidates as code-completion reslults.
3917 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3918 CandEnd = CandidateSet.end();
3919 Cand != CandEnd; ++Cand) {
3920 if (Cand->Viable)
3921 Results.push_back(ResultCandidate(Cand->Function));
3922 }
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003923
3924 // From the viable candidates, try to determine the type of this parameter.
3925 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3926 if (const FunctionType *FType = Results[I].getFunctionType())
3927 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
Alp Toker9cacbab2014-01-20 20:26:09 +00003928 if (Args.size() < Proto->getNumParams()) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003929 if (ParamType.isNull())
Alp Toker9cacbab2014-01-20 20:26:09 +00003930 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003931 else if (!Context.hasSameUnqualifiedType(
Alp Toker9cacbab2014-01-20 20:26:09 +00003932 ParamType.getNonReferenceType(),
3933 Proto->getParamType(Args.size())
3934 .getNonReferenceType())) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003935 ParamType = QualType();
3936 break;
3937 }
3938 }
3939 }
3940 } else {
3941 // Try to determine the parameter type from the type of the expression
3942 // being called.
3943 QualType FunctionType = Fn->getType();
3944 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3945 FunctionType = Ptr->getPointeeType();
3946 else if (const BlockPointerType *BlockPtr
3947 = FunctionType->getAs<BlockPointerType>())
3948 FunctionType = BlockPtr->getPointeeType();
3949 else if (const MemberPointerType *MemPtr
3950 = FunctionType->getAs<MemberPointerType>())
3951 FunctionType = MemPtr->getPointeeType();
3952
3953 if (const FunctionProtoType *Proto
3954 = FunctionType->getAs<FunctionProtoType>()) {
Alp Toker9cacbab2014-01-20 20:26:09 +00003955 if (Args.size() < Proto->getNumParams())
3956 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003957 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003958 }
Douglas Gregor3ef59522009-12-11 19:06:04 +00003959
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003960 if (ParamType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003961 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003962 else
3963 CodeCompleteExpression(S, ParamType);
3964
Douglas Gregorc01890e2010-04-06 20:19:47 +00003965 if (!Results.empty())
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003966 CodeCompleter->ProcessOverloadCandidates(*this, Args.size(), Results.data(),
Douglas Gregor3ef59522009-12-11 19:06:04 +00003967 Results.size());
Douglas Gregorcabea402009-09-22 15:41:20 +00003968}
3969
John McCall48871652010-08-21 09:40:31 +00003970void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3971 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003972 if (!VD) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003973 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003974 return;
3975 }
3976
3977 CodeCompleteExpression(S, VD->getType());
3978}
3979
3980void Sema::CodeCompleteReturn(Scope *S) {
3981 QualType ResultType;
3982 if (isa<BlockDecl>(CurContext)) {
3983 if (BlockScopeInfo *BSI = getCurBlock())
3984 ResultType = BSI->ReturnType;
3985 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
3986 ResultType = Function->getResultType();
3987 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
3988 ResultType = Method->getResultType();
3989
3990 if (ResultType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003991 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003992 else
3993 CodeCompleteExpression(S, ResultType);
3994}
3995
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003996void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003997 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003998 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003999 mapCodeCompletionContext(*this, PCC_Statement));
4000 Results.setFilter(&ResultBuilder::IsOrdinaryName);
4001 Results.EnterNewScope();
4002
4003 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4004 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4005 CodeCompleter->includeGlobals());
4006
4007 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
4008
4009 // "else" block
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004010 CodeCompletionBuilder Builder(Results.getAllocator(),
4011 Results.getCodeCompletionTUInfo());
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004012 Builder.AddTypedTextChunk("else");
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00004013 if (Results.includeCodePatterns()) {
4014 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4015 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4016 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4017 Builder.AddPlaceholderChunk("statements");
4018 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4019 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4020 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004021 Results.AddResult(Builder.TakeString());
4022
4023 // "else if" block
4024 Builder.AddTypedTextChunk("else");
4025 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4026 Builder.AddTextChunk("if");
4027 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4028 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00004029 if (getLangOpts().CPlusPlus)
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004030 Builder.AddPlaceholderChunk("condition");
4031 else
4032 Builder.AddPlaceholderChunk("expression");
4033 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00004034 if (Results.includeCodePatterns()) {
4035 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4036 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4037 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4038 Builder.AddPlaceholderChunk("statements");
4039 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4040 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4041 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004042 Results.AddResult(Builder.TakeString());
4043
4044 Results.ExitScope();
4045
4046 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004047 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004048
4049 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00004050 AddMacroResults(PP, Results, false);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004051
4052 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4053 Results.data(),Results.size());
4054}
4055
Richard Trieu2bd04012011-09-09 02:00:50 +00004056void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004057 if (LHS)
4058 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4059 else
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004060 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004061}
4062
Jeffrey Yasskinc76498d2010-04-08 16:38:48 +00004063void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor2436e712009-09-17 21:32:03 +00004064 bool EnteringContext) {
4065 if (!SS.getScopeRep() || !CodeCompleter)
4066 return;
4067
Douglas Gregor3545ff42009-09-21 16:56:56 +00004068 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4069 if (!Ctx)
4070 return;
Douglas Gregor800f2f02009-12-11 18:28:39 +00004071
4072 // Try to instantiate any non-dependent declaration contexts before
4073 // we look in them.
John McCall0b66eb32010-05-01 00:40:08 +00004074 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregor800f2f02009-12-11 18:28:39 +00004075 return;
4076
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004077 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004078 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004079 CodeCompletionContext::CCC_Name);
Douglas Gregorac322ec2010-08-27 21:18:54 +00004080 Results.EnterNewScope();
Douglas Gregor0ac41382010-09-23 23:01:17 +00004081
Douglas Gregor3545ff42009-09-21 16:56:56 +00004082 // The "template" keyword can follow "::" in the grammar, but only
4083 // put it into the grammar if the nested-name-specifier is dependent.
Aaron Ballman4a979672014-01-03 13:56:08 +00004084 NestedNameSpecifier *NNS = SS.getScopeRep();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004085 if (!Results.empty() && NNS->isDependent())
Douglas Gregor78a21012010-01-14 16:01:26 +00004086 Results.AddResult("template");
Douglas Gregorac322ec2010-08-27 21:18:54 +00004087
4088 // Add calls to overridden virtual functions, if there are any.
4089 //
4090 // FIXME: This isn't wonderful, because we don't know whether we're actually
4091 // in a context that permits expressions. This is a general issue with
4092 // qualified-id completions.
4093 if (!EnteringContext)
4094 MaybeAddOverrideCalls(*this, Ctx, Results);
4095 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004096
Douglas Gregorac322ec2010-08-27 21:18:54 +00004097 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4098 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4099
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004100 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorc1679ec2011-07-25 17:48:11 +00004101 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004102 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00004103}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004104
4105void Sema::CodeCompleteUsing(Scope *S) {
4106 if (!CodeCompleter)
4107 return;
4108
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004109 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004110 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004111 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4112 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004113 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004114
4115 // If we aren't in class scope, we could see the "namespace" keyword.
4116 if (!S->isClassScope())
John McCall276321a2010-08-25 06:19:51 +00004117 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004118
4119 // After "using", we can see anything that would start a
4120 // nested-name-specifier.
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004121 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004122 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4123 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004124 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004125
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004126 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004127 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004128 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004129}
4130
4131void Sema::CodeCompleteUsingDirective(Scope *S) {
4132 if (!CodeCompleter)
4133 return;
4134
Douglas Gregor3545ff42009-09-21 16:56:56 +00004135 // After "using namespace", we expect to see a namespace name or namespace
4136 // alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004137 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004138 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004139 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004140 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004141 Results.EnterNewScope();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004142 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004143 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4144 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004145 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004146 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004147 CodeCompletionContext::CCC_Namespace,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004148 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004149}
4150
4151void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4152 if (!CodeCompleter)
4153 return;
4154
Ted Kremenekc37877d2013-10-08 17:08:03 +00004155 DeclContext *Ctx = S->getEntity();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004156 if (!S->getParent())
4157 Ctx = Context.getTranslationUnitDecl();
4158
Douglas Gregor0ac41382010-09-23 23:01:17 +00004159 bool SuppressedGlobalResults
4160 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4161
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004162 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004163 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004164 SuppressedGlobalResults
4165 ? CodeCompletionContext::CCC_Namespace
4166 : CodeCompletionContext::CCC_Other,
4167 &ResultBuilder::IsNamespace);
4168
4169 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00004170 // We only want to see those namespaces that have already been defined
4171 // within this scope, because its likely that the user is creating an
4172 // extended namespace declaration. Keep track of the most recent
4173 // definition of each namespace.
4174 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4175 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4176 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4177 NS != NSEnd; ++NS)
David Blaikie40ed2972012-06-06 20:45:41 +00004178 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004179
4180 // Add the most recent definition (or extended definition) of each
4181 // namespace to the list of results.
Douglas Gregor64b12b52009-09-22 23:31:26 +00004182 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004183 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregor78254c82012-03-27 23:34:16 +00004184 NS = OrigToLatest.begin(),
4185 NSEnd = OrigToLatest.end();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004186 NS != NSEnd; ++NS)
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004187 Results.AddResult(CodeCompletionResult(
4188 NS->second, Results.getBasePriority(NS->second), 0),
Douglas Gregorfc59ce12010-01-14 16:14:35 +00004189 CurContext, 0, false);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004190 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004191 }
4192
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004193 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004194 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004195 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004196}
4197
4198void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4199 if (!CodeCompleter)
4200 return;
4201
Douglas Gregor3545ff42009-09-21 16:56:56 +00004202 // After "namespace", we expect to see a namespace or alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004203 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004204 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004205 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004206 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004207 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004208 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4209 CodeCompleter->includeGlobals());
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004210 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004211 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004212 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004213}
4214
Douglas Gregorc811ede2009-09-18 20:05:18 +00004215void Sema::CodeCompleteOperatorName(Scope *S) {
4216 if (!CodeCompleter)
4217 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004218
John McCall276321a2010-08-25 06:19:51 +00004219 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004220 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004221 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004222 CodeCompletionContext::CCC_Type,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004223 &ResultBuilder::IsType);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004224 Results.EnterNewScope();
Douglas Gregorc811ede2009-09-18 20:05:18 +00004225
Douglas Gregor3545ff42009-09-21 16:56:56 +00004226 // Add the names of overloadable operators.
4227#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4228 if (std::strcmp(Spelling, "?")) \
Douglas Gregor78a21012010-01-14 16:01:26 +00004229 Results.AddResult(Result(Spelling));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004230#include "clang/Basic/OperatorKinds.def"
4231
4232 // Add any type names visible from the current scope
Douglas Gregor6ae4c522010-01-14 03:21:49 +00004233 Results.allowNestedNameSpecifiers();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004234 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004235 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4236 CodeCompleter->includeGlobals());
Douglas Gregor3545ff42009-09-21 16:56:56 +00004237
4238 // Add any type specifiers
David Blaikiebbafb8a2012-03-11 07:00:24 +00004239 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004240 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004241
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004242 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004243 CodeCompletionContext::CCC_Type,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004244 Results.data(),Results.size());
Douglas Gregorc811ede2009-09-18 20:05:18 +00004245}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004246
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004247void Sema::CodeCompleteConstructorInitializer(
4248 Decl *ConstructorD,
4249 ArrayRef <CXXCtorInitializer *> Initializers) {
Douglas Gregor75acd922011-09-27 23:30:47 +00004250 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004251 CXXConstructorDecl *Constructor
4252 = static_cast<CXXConstructorDecl *>(ConstructorD);
4253 if (!Constructor)
4254 return;
4255
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004256 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004257 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004258 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004259 Results.EnterNewScope();
4260
4261 // Fill in any already-initialized fields or base classes.
4262 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4263 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004264 for (unsigned I = 0, E = Initializers.size(); I != E; ++I) {
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004265 if (Initializers[I]->isBaseInitializer())
4266 InitializedBases.insert(
4267 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4268 else
Francois Pichetd583da02010-12-04 09:14:42 +00004269 InitializedFields.insert(cast<FieldDecl>(
4270 Initializers[I]->getAnyMember()));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004271 }
4272
4273 // Add completions for base classes.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004274 CodeCompletionBuilder Builder(Results.getAllocator(),
4275 Results.getCodeCompletionTUInfo());
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004276 bool SawLastInitializer = Initializers.empty();
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004277 CXXRecordDecl *ClassDecl = Constructor->getParent();
4278 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->bases_begin(),
4279 BaseEnd = ClassDecl->bases_end();
4280 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004281 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4282 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004283 = !Initializers.empty() &&
4284 Initializers.back()->isBaseInitializer() &&
Douglas Gregor99129ef2010-08-29 19:27:27 +00004285 Context.hasSameUnqualifiedType(Base->getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004286 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004287 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004288 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004289
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004290 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004291 Results.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004292 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004293 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4294 Builder.AddPlaceholderChunk("args");
4295 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4296 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004297 SawLastInitializer? CCP_NextInitializer
4298 : CCP_MemberDeclaration));
4299 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004300 }
4301
4302 // Add completions for virtual base classes.
4303 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->vbases_begin(),
4304 BaseEnd = ClassDecl->vbases_end();
4305 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004306 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4307 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004308 = !Initializers.empty() &&
4309 Initializers.back()->isBaseInitializer() &&
Douglas Gregor99129ef2010-08-29 19:27:27 +00004310 Context.hasSameUnqualifiedType(Base->getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004311 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004312 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004313 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004314
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004315 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004316 Builder.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004317 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004318 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4319 Builder.AddPlaceholderChunk("args");
4320 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4321 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004322 SawLastInitializer? CCP_NextInitializer
4323 : CCP_MemberDeclaration));
4324 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004325 }
4326
4327 // Add completions for members.
4328 for (CXXRecordDecl::field_iterator Field = ClassDecl->field_begin(),
4329 FieldEnd = ClassDecl->field_end();
4330 Field != FieldEnd; ++Field) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004331 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4332 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004333 = !Initializers.empty() &&
4334 Initializers.back()->isAnyMemberInitializer() &&
4335 Initializers.back()->getAnyMember() == *Field;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004336 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004337 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004338
4339 if (!Field->getDeclName())
4340 continue;
4341
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004342 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004343 Field->getIdentifier()->getName()));
4344 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4345 Builder.AddPlaceholderChunk("args");
4346 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4347 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004348 SawLastInitializer? CCP_NextInitializer
Douglas Gregorf3af3112010-09-09 21:42:20 +00004349 : CCP_MemberDeclaration,
Douglas Gregor78254c82012-03-27 23:34:16 +00004350 CXCursor_MemberRef,
4351 CXAvailability_Available,
David Blaikie40ed2972012-06-06 20:45:41 +00004352 *Field));
Douglas Gregor99129ef2010-08-29 19:27:27 +00004353 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004354 }
4355 Results.ExitScope();
4356
Douglas Gregor0ac41382010-09-23 23:01:17 +00004357 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004358 Results.data(), Results.size());
4359}
4360
Douglas Gregord8c61782012-02-15 15:34:24 +00004361/// \brief Determine whether this scope denotes a namespace.
4362static bool isNamespaceScope(Scope *S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00004363 DeclContext *DC = S->getEntity();
Douglas Gregord8c61782012-02-15 15:34:24 +00004364 if (!DC)
4365 return false;
4366
4367 return DC->isFileContext();
4368}
4369
4370void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4371 bool AfterAmpersand) {
4372 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004373 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord8c61782012-02-15 15:34:24 +00004374 CodeCompletionContext::CCC_Other);
4375 Results.EnterNewScope();
4376
4377 // Note what has already been captured.
4378 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4379 bool IncludedThis = false;
4380 for (SmallVectorImpl<LambdaCapture>::iterator C = Intro.Captures.begin(),
4381 CEnd = Intro.Captures.end();
4382 C != CEnd; ++C) {
4383 if (C->Kind == LCK_This) {
4384 IncludedThis = true;
4385 continue;
4386 }
4387
4388 Known.insert(C->Id);
4389 }
4390
4391 // Look for other capturable variables.
4392 for (; S && !isNamespaceScope(S); S = S->getParent()) {
4393 for (Scope::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
4394 D != DEnd; ++D) {
4395 VarDecl *Var = dyn_cast<VarDecl>(*D);
4396 if (!Var ||
4397 !Var->hasLocalStorage() ||
4398 Var->hasAttr<BlocksAttr>())
4399 continue;
4400
4401 if (Known.insert(Var->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004402 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
4403 CurContext, 0, false);
Douglas Gregord8c61782012-02-15 15:34:24 +00004404 }
4405 }
4406
4407 // Add 'this', if it would be valid.
4408 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4409 addThisCompletion(*this, Results);
4410
4411 Results.ExitScope();
4412
4413 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4414 Results.data(), Results.size());
4415}
4416
James Dennett596e4752012-06-14 03:11:41 +00004417/// Macro that optionally prepends an "@" to the string literal passed in via
4418/// Keyword, depending on whether NeedAt is true or false.
4419#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4420
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004421static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004422 ResultBuilder &Results,
4423 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004424 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004425 // Since we have an implementation, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004426 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004427
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004428 CodeCompletionBuilder Builder(Results.getAllocator(),
4429 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004430 if (LangOpts.ObjC2) {
4431 // @dynamic
James Dennett596e4752012-06-14 03:11:41 +00004432 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004433 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4434 Builder.AddPlaceholderChunk("property");
4435 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004436
4437 // @synthesize
James Dennett596e4752012-06-14 03:11:41 +00004438 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004439 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4440 Builder.AddPlaceholderChunk("property");
4441 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004442 }
4443}
4444
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004445static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004446 ResultBuilder &Results,
4447 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004448 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004449
4450 // Since we have an interface or protocol, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004451 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004452
4453 if (LangOpts.ObjC2) {
4454 // @property
James Dennett596e4752012-06-14 03:11:41 +00004455 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004456
4457 // @required
James Dennett596e4752012-06-14 03:11:41 +00004458 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004459
4460 // @optional
James Dennett596e4752012-06-14 03:11:41 +00004461 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004462 }
4463}
4464
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004465static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004466 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004467 CodeCompletionBuilder Builder(Results.getAllocator(),
4468 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004469
4470 // @class name ;
James Dennett596e4752012-06-14 03:11:41 +00004471 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004472 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4473 Builder.AddPlaceholderChunk("name");
4474 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004475
Douglas Gregorf4c33342010-05-28 00:22:41 +00004476 if (Results.includeCodePatterns()) {
4477 // @interface name
4478 // FIXME: Could introduce the whole pattern, including superclasses and
4479 // such.
James Dennett596e4752012-06-14 03:11:41 +00004480 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004481 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4482 Builder.AddPlaceholderChunk("class");
4483 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004484
Douglas Gregorf4c33342010-05-28 00:22:41 +00004485 // @protocol name
James Dennett596e4752012-06-14 03:11:41 +00004486 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004487 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4488 Builder.AddPlaceholderChunk("protocol");
4489 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004490
4491 // @implementation name
James Dennett596e4752012-06-14 03:11:41 +00004492 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004493 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4494 Builder.AddPlaceholderChunk("class");
4495 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004496 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004497
4498 // @compatibility_alias name
James Dennett596e4752012-06-14 03:11:41 +00004499 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004500 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4501 Builder.AddPlaceholderChunk("alias");
4502 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4503 Builder.AddPlaceholderChunk("class");
4504 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor61e36812013-03-07 23:26:24 +00004505
4506 if (Results.getSema().getLangOpts().Modules) {
4507 // @import name
4508 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "import"));
4509 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4510 Builder.AddPlaceholderChunk("module");
4511 Results.AddResult(Result(Builder.TakeString()));
4512 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004513}
4514
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004515void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004516 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004517 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004518 CodeCompletionContext::CCC_Other);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004519 Results.EnterNewScope();
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004520 if (isa<ObjCImplDecl>(CurContext))
David Blaikiebbafb8a2012-03-11 07:00:24 +00004521 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004522 else if (CurContext->isObjCContainer())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004523 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorf1934162010-01-13 21:24:21 +00004524 else
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004525 AddObjCTopLevelResults(Results, false);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004526 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004527 HandleCodeCompleteResults(this, CodeCompleter,
4528 CodeCompletionContext::CCC_Other,
4529 Results.data(),Results.size());
Douglas Gregorf48706c2009-12-07 09:27:33 +00004530}
4531
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004532static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004533 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004534 CodeCompletionBuilder Builder(Results.getAllocator(),
4535 Results.getCodeCompletionTUInfo());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004536
4537 // @encode ( type-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004538 const char *EncodeType = "char[]";
David Blaikiebbafb8a2012-03-11 07:00:24 +00004539 if (Results.getSema().getLangOpts().CPlusPlus ||
4540 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose9da05852012-06-15 18:19:56 +00004541 EncodeType = "const char[]";
Douglas Gregore5c79d52011-10-18 21:20:17 +00004542 Builder.AddResultTypeChunk(EncodeType);
James Dennett596e4752012-06-14 03:11:41 +00004543 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004544 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4545 Builder.AddPlaceholderChunk("type-name");
4546 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4547 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004548
4549 // @protocol ( protocol-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004550 Builder.AddResultTypeChunk("Protocol *");
James Dennett596e4752012-06-14 03:11:41 +00004551 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004552 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4553 Builder.AddPlaceholderChunk("protocol-name");
4554 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4555 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004556
4557 // @selector ( selector )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004558 Builder.AddResultTypeChunk("SEL");
James Dennett596e4752012-06-14 03:11:41 +00004559 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004560 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4561 Builder.AddPlaceholderChunk("selector");
4562 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4563 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004564
4565 // @"string"
4566 Builder.AddResultTypeChunk("NSString *");
4567 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4568 Builder.AddPlaceholderChunk("string");
4569 Builder.AddTextChunk("\"");
4570 Results.AddResult(Result(Builder.TakeString()));
4571
Douglas Gregor951de302012-07-17 23:24:47 +00004572 // @[objects, ...]
Jordan Rose9da05852012-06-15 18:19:56 +00004573 Builder.AddResultTypeChunk("NSArray *");
James Dennett596e4752012-06-14 03:11:41 +00004574 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004575 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004576 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4577 Results.AddResult(Result(Builder.TakeString()));
4578
Douglas Gregor951de302012-07-17 23:24:47 +00004579 // @{key : object, ...}
Jordan Rose9da05852012-06-15 18:19:56 +00004580 Builder.AddResultTypeChunk("NSDictionary *");
James Dennett596e4752012-06-14 03:11:41 +00004581 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"{"));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004582 Builder.AddPlaceholderChunk("key");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004583 Builder.AddChunk(CodeCompletionString::CK_Colon);
4584 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4585 Builder.AddPlaceholderChunk("object, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004586 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4587 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004588
Douglas Gregor951de302012-07-17 23:24:47 +00004589 // @(expression)
Jordan Rose9da05852012-06-15 18:19:56 +00004590 Builder.AddResultTypeChunk("id");
4591 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose9da05852012-06-15 18:19:56 +00004592 Builder.AddPlaceholderChunk("expression");
Jordan Rose9da05852012-06-15 18:19:56 +00004593 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4594 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004595}
4596
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004597static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004598 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004599 CodeCompletionBuilder Builder(Results.getAllocator(),
4600 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004601
Douglas Gregorf4c33342010-05-28 00:22:41 +00004602 if (Results.includeCodePatterns()) {
4603 // @try { statements } @catch ( declaration ) { statements } @finally
4604 // { statements }
James Dennett596e4752012-06-14 03:11:41 +00004605 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004606 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4607 Builder.AddPlaceholderChunk("statements");
4608 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4609 Builder.AddTextChunk("@catch");
4610 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4611 Builder.AddPlaceholderChunk("parameter");
4612 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4613 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4614 Builder.AddPlaceholderChunk("statements");
4615 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4616 Builder.AddTextChunk("@finally");
4617 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4618 Builder.AddPlaceholderChunk("statements");
4619 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4620 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004621 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004622
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004623 // @throw
James Dennett596e4752012-06-14 03:11:41 +00004624 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004625 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4626 Builder.AddPlaceholderChunk("expression");
4627 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004628
Douglas Gregorf4c33342010-05-28 00:22:41 +00004629 if (Results.includeCodePatterns()) {
4630 // @synchronized ( expression ) { statements }
James Dennett596e4752012-06-14 03:11:41 +00004631 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004632 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4633 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4634 Builder.AddPlaceholderChunk("expression");
4635 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4636 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4637 Builder.AddPlaceholderChunk("statements");
4638 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4639 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004640 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004641}
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004642
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004643static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00004644 ResultBuilder &Results,
4645 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004646 typedef CodeCompletionResult Result;
James Dennett596e4752012-06-14 03:11:41 +00004647 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4648 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4649 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004650 if (LangOpts.ObjC2)
James Dennett596e4752012-06-14 03:11:41 +00004651 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004652}
4653
4654void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004655 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004656 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004657 CodeCompletionContext::CCC_Other);
Douglas Gregor48d46252010-01-13 21:54:15 +00004658 Results.EnterNewScope();
David Blaikiebbafb8a2012-03-11 07:00:24 +00004659 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregor48d46252010-01-13 21:54:15 +00004660 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004661 HandleCodeCompleteResults(this, CodeCompleter,
4662 CodeCompletionContext::CCC_Other,
4663 Results.data(),Results.size());
Douglas Gregor48d46252010-01-13 21:54:15 +00004664}
4665
4666void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004667 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004668 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004669 CodeCompletionContext::CCC_Other);
Douglas Gregorf1934162010-01-13 21:24:21 +00004670 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004671 AddObjCStatementResults(Results, false);
4672 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004673 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004674 HandleCodeCompleteResults(this, CodeCompleter,
4675 CodeCompletionContext::CCC_Other,
4676 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004677}
4678
4679void Sema::CodeCompleteObjCAtExpression(Scope *S) {
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);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004683 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004684 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004685 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004686 HandleCodeCompleteResults(this, CodeCompleter,
4687 CodeCompletionContext::CCC_Other,
4688 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004689}
4690
Douglas Gregore6078da2009-11-19 00:14:45 +00004691/// \brief Determine whether the addition of the given flag to an Objective-C
4692/// property's attributes will cause a conflict.
Bill Wendling44426052012-12-20 19:22:21 +00004693static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregore6078da2009-11-19 00:14:45 +00004694 // Check if we've already added this flag.
Bill Wendling44426052012-12-20 19:22:21 +00004695 if (Attributes & NewFlag)
Douglas Gregore6078da2009-11-19 00:14:45 +00004696 return true;
4697
Bill Wendling44426052012-12-20 19:22:21 +00004698 Attributes |= NewFlag;
Douglas Gregore6078da2009-11-19 00:14:45 +00004699
4700 // Check for collisions with "readonly".
Bill Wendling44426052012-12-20 19:22:21 +00004701 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4702 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregore6078da2009-11-19 00:14:45 +00004703 return true;
4704
Jordan Rose53cb2f32012-08-20 20:01:13 +00004705 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendling44426052012-12-20 19:22:21 +00004706 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCall31168b02011-06-15 23:02:42 +00004707 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregore6078da2009-11-19 00:14:45 +00004708 ObjCDeclSpec::DQ_PR_copy |
Jordan Rose53cb2f32012-08-20 20:01:13 +00004709 ObjCDeclSpec::DQ_PR_retain |
4710 ObjCDeclSpec::DQ_PR_strong |
4711 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregore6078da2009-11-19 00:14:45 +00004712 if (AssignCopyRetMask &&
4713 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCall31168b02011-06-15 23:02:42 +00004714 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregore6078da2009-11-19 00:14:45 +00004715 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCall31168b02011-06-15 23:02:42 +00004716 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rose53cb2f32012-08-20 20:01:13 +00004717 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4718 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregore6078da2009-11-19 00:14:45 +00004719 return true;
4720
4721 return false;
4722}
4723
Douglas Gregor36029f42009-11-18 23:08:07 +00004724void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroff936354c2009-10-08 21:55:05 +00004725 if (!CodeCompleter)
4726 return;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004727
Bill Wendling44426052012-12-20 19:22:21 +00004728 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroff936354c2009-10-08 21:55:05 +00004729
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004730 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004731 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004732 CodeCompletionContext::CCC_Other);
Steve Naroff936354c2009-10-08 21:55:05 +00004733 Results.EnterNewScope();
Bill Wendling44426052012-12-20 19:22:21 +00004734 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall276321a2010-08-25 06:19:51 +00004735 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendling44426052012-12-20 19:22:21 +00004736 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall276321a2010-08-25 06:19:51 +00004737 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendling44426052012-12-20 19:22:21 +00004738 if (!ObjCPropertyFlagConflicts(Attributes,
John McCall31168b02011-06-15 23:02:42 +00004739 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4740 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendling44426052012-12-20 19:22:21 +00004741 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall276321a2010-08-25 06:19:51 +00004742 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendling44426052012-12-20 19:22:21 +00004743 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall276321a2010-08-25 06:19:51 +00004744 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendling44426052012-12-20 19:22:21 +00004745 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCall31168b02011-06-15 23:02:42 +00004746 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendling44426052012-12-20 19:22:21 +00004747 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall276321a2010-08-25 06:19:51 +00004748 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendling44426052012-12-20 19:22:21 +00004749 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall276321a2010-08-25 06:19:51 +00004750 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendling44426052012-12-20 19:22:21 +00004751 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian1c2d29e2011-06-11 17:14:27 +00004752 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rose53cb2f32012-08-20 20:01:13 +00004753
4754 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall3deb1ad2012-08-21 02:47:43 +00004755 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendling44426052012-12-20 19:22:21 +00004756 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rose53cb2f32012-08-20 20:01:13 +00004757 Results.AddResult(CodeCompletionResult("weak"));
4758
Bill Wendling44426052012-12-20 19:22:21 +00004759 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004760 CodeCompletionBuilder Setter(Results.getAllocator(),
4761 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004762 Setter.AddTypedTextChunk("setter");
4763 Setter.AddTextChunk(" = ");
4764 Setter.AddPlaceholderChunk("method");
4765 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004766 }
Bill Wendling44426052012-12-20 19:22:21 +00004767 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004768 CodeCompletionBuilder Getter(Results.getAllocator(),
4769 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004770 Getter.AddTypedTextChunk("getter");
4771 Getter.AddTextChunk(" = ");
4772 Getter.AddPlaceholderChunk("method");
4773 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004774 }
Steve Naroff936354c2009-10-08 21:55:05 +00004775 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004776 HandleCodeCompleteResults(this, CodeCompleter,
4777 CodeCompletionContext::CCC_Other,
4778 Results.data(),Results.size());
Steve Naroff936354c2009-10-08 21:55:05 +00004779}
Steve Naroffeae65032009-11-07 02:08:14 +00004780
James Dennettf1243872012-06-17 05:33:25 +00004781/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregorc8537c52009-11-19 07:41:15 +00004782/// via code completion.
4783enum ObjCMethodKind {
Dmitri Gribenko4280e5c2012-06-08 23:13:42 +00004784 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4785 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4786 MK_OneArgSelector ///< One-argument selector.
Douglas Gregorc8537c52009-11-19 07:41:15 +00004787};
4788
Douglas Gregor67c692c2010-08-26 15:07:07 +00004789static bool isAcceptableObjCSelector(Selector Sel,
4790 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004791 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004792 bool AllowSameLength = true) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004793 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor67c692c2010-08-26 15:07:07 +00004794 if (NumSelIdents > Sel.getNumArgs())
4795 return false;
4796
4797 switch (WantKind) {
4798 case MK_Any: break;
4799 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4800 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4801 }
4802
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004803 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4804 return false;
4805
Douglas Gregor67c692c2010-08-26 15:07:07 +00004806 for (unsigned I = 0; I != NumSelIdents; ++I)
4807 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4808 return false;
4809
4810 return true;
4811}
4812
Douglas Gregorc8537c52009-11-19 07:41:15 +00004813static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4814 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004815 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004816 bool AllowSameLength = true) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00004817 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004818 AllowSameLength);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004819}
Douglas Gregor1154e272010-09-16 16:06:31 +00004820
4821namespace {
4822 /// \brief A set of selectors, which is used to avoid introducing multiple
4823 /// completions with the same selector into the result set.
4824 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4825}
4826
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004827/// \brief Add all of the Objective-C methods in the given Objective-C
4828/// container to the set of results.
4829///
4830/// The container will be a class, protocol, category, or implementation of
4831/// any of the above. This mether will recurse to include methods from
4832/// the superclasses of classes along with their categories, protocols, and
4833/// implementations.
4834///
4835/// \param Container the container in which we'll look to find methods.
4836///
James Dennett596e4752012-06-14 03:11:41 +00004837/// \param WantInstanceMethods Whether to add instance methods (only); if
4838/// false, this routine will add factory methods (only).
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004839///
4840/// \param CurContext the context in which we're performing the lookup that
4841/// finds methods.
4842///
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004843/// \param AllowSameLength Whether we allow a method to be added to the list
4844/// when it has the same number of parameters as we have selector identifiers.
4845///
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004846/// \param Results the structure into which we'll add results.
4847static void AddObjCMethods(ObjCContainerDecl *Container,
4848 bool WantInstanceMethods,
Douglas Gregorc8537c52009-11-19 07:41:15 +00004849 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004850 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004851 DeclContext *CurContext,
Douglas Gregor1154e272010-09-16 16:06:31 +00004852 VisitedSelectorSet &Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004853 bool AllowSameLength,
Douglas Gregor416b5752010-08-25 01:08:01 +00004854 ResultBuilder &Results,
4855 bool InOriginalClass = true) {
John McCall276321a2010-08-25 06:19:51 +00004856 typedef CodeCompletionResult Result;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00004857 Container = getContainerDef(Container);
Douglas Gregor41778c32013-01-30 06:58:39 +00004858 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4859 bool isRootClass = IFace && !IFace->getSuperClass();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004860 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
4861 MEnd = Container->meth_end();
4862 M != MEnd; ++M) {
Douglas Gregor41778c32013-01-30 06:58:39 +00004863 // The instance methods on the root class can be messaged via the
4864 // metaclass.
4865 if (M->isInstanceMethod() == WantInstanceMethods ||
4866 (isRootClass && !WantInstanceMethods)) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00004867 // Check whether the selector identifiers we've been given are a
4868 // subset of the identifiers for this particular method.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004869 if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, AllowSameLength))
Douglas Gregor1b605f72009-11-19 01:08:35 +00004870 continue;
Douglas Gregorc8537c52009-11-19 07:41:15 +00004871
David Blaikie2d7c57e2012-04-30 02:36:29 +00004872 if (!Selectors.insert(M->getSelector()))
Douglas Gregor1154e272010-09-16 16:06:31 +00004873 continue;
4874
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004875 Result R = Result(*M, Results.getBasePriority(*M), 0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004876 R.StartParameter = SelIdents.size();
Douglas Gregorc8537c52009-11-19 07:41:15 +00004877 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor416b5752010-08-25 01:08:01 +00004878 if (!InOriginalClass)
4879 R.Priority += CCD_InBaseClass;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004880 Results.MaybeAddResult(R, CurContext);
4881 }
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004882 }
4883
Douglas Gregorf37c9492010-09-16 15:34:59 +00004884 // Visit the protocols of protocols.
4885 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregore6e48b12012-01-01 19:29:29 +00004886 if (Protocol->hasDefinition()) {
4887 const ObjCList<ObjCProtocolDecl> &Protocols
4888 = Protocol->getReferencedProtocols();
4889 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4890 E = Protocols.end();
4891 I != E; ++I)
4892 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004893 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregore6e48b12012-01-01 19:29:29 +00004894 }
Douglas Gregorf37c9492010-09-16 15:34:59 +00004895 }
4896
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00004897 if (!IFace || !IFace->hasDefinition())
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004898 return;
4899
4900 // Add methods in protocols.
Argyrios Kyrtzidise7f3ef32012-03-13 01:09:41 +00004901 for (ObjCInterfaceDecl::protocol_iterator I = IFace->protocol_begin(),
4902 E = IFace->protocol_end();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004903 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004904 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004905 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004906
4907 // Add methods in categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00004908 for (ObjCInterfaceDecl::known_categories_iterator
4909 Cat = IFace->known_categories_begin(),
4910 CatEnd = IFace->known_categories_end();
4911 Cat != CatEnd; ++Cat) {
4912 ObjCCategoryDecl *CatDecl = *Cat;
4913
4914 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004915 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004916 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004917
4918 // Add a categories protocol methods.
4919 const ObjCList<ObjCProtocolDecl> &Protocols
4920 = CatDecl->getReferencedProtocols();
4921 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4922 E = Protocols.end();
4923 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00004924 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004925 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004926 Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004927
4928 // Add methods in category implementations.
4929 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004930 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004931 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004932 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004933 }
4934
4935 // Add methods in superclass.
4936 if (IFace->getSuperClass())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004937 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004938 SelIdents, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004939 AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004940
4941 // Add methods in our implementation, if any.
4942 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004943 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004944 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004945 Results, InOriginalClass);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004946}
4947
4948
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004949void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004950 // Try to find the interface where getters might live.
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004951 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004952 if (!Class) {
4953 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004954 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004955 Class = Category->getClassInterface();
4956
4957 if (!Class)
4958 return;
4959 }
4960
4961 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004962 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004963 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004964 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004965 Results.EnterNewScope();
4966
Douglas Gregor1154e272010-09-16 16:06:31 +00004967 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004968 AddObjCMethods(Class, true, MK_ZeroArgSelector, None, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004969 /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004970 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004971 HandleCodeCompleteResults(this, CodeCompleter,
4972 CodeCompletionContext::CCC_Other,
4973 Results.data(),Results.size());
Douglas Gregorc8537c52009-11-19 07:41:15 +00004974}
4975
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004976void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004977 // Try to find the interface where setters might live.
4978 ObjCInterfaceDecl *Class
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004979 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004980 if (!Class) {
4981 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004982 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004983 Class = Category->getClassInterface();
4984
4985 if (!Class)
4986 return;
4987 }
4988
4989 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004990 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004991 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004992 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004993 Results.EnterNewScope();
4994
Douglas Gregor1154e272010-09-16 16:06:31 +00004995 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004996 AddObjCMethods(Class, true, MK_OneArgSelector, None, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004997 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004998
4999 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005000 HandleCodeCompleteResults(this, CodeCompleter,
5001 CodeCompletionContext::CCC_Other,
5002 Results.data(),Results.size());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005003}
5004
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005005void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
5006 bool IsParameter) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005007 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005008 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005009 CodeCompletionContext::CCC_Type);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005010 Results.EnterNewScope();
5011
5012 // Add context-sensitive, Objective-C parameter-passing keywords.
5013 bool AddedInOut = false;
5014 if ((DS.getObjCDeclQualifier() &
5015 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
5016 Results.AddResult("in");
5017 Results.AddResult("inout");
5018 AddedInOut = true;
5019 }
5020 if ((DS.getObjCDeclQualifier() &
5021 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
5022 Results.AddResult("out");
5023 if (!AddedInOut)
5024 Results.AddResult("inout");
5025 }
5026 if ((DS.getObjCDeclQualifier() &
5027 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
5028 ObjCDeclSpec::DQ_Oneway)) == 0) {
5029 Results.AddResult("bycopy");
5030 Results.AddResult("byref");
5031 Results.AddResult("oneway");
5032 }
5033
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005034 // If we're completing the return type of an Objective-C method and the
5035 // identifier IBAction refers to a macro, provide a completion item for
5036 // an action, e.g.,
5037 // IBAction)<#selector#>:(id)sender
5038 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
5039 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005040 CodeCompletionBuilder Builder(Results.getAllocator(),
5041 Results.getCodeCompletionTUInfo(),
5042 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005043 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005044 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005045 Builder.AddPlaceholderChunk("selector");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005046 Builder.AddChunk(CodeCompletionString::CK_Colon);
5047 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005048 Builder.AddTextChunk("id");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005049 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005050 Builder.AddTextChunk("sender");
5051 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
5052 }
Douglas Gregored1f5972013-01-30 07:11:43 +00005053
5054 // If we're completing the return type, provide 'instancetype'.
5055 if (!IsParameter) {
5056 Results.AddResult(CodeCompletionResult("instancetype"));
5057 }
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005058
Douglas Gregor99fa2642010-08-24 01:06:58 +00005059 // Add various builtin type names and specifiers.
5060 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5061 Results.ExitScope();
5062
5063 // Add the various type names
5064 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5065 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5066 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5067 CodeCompleter->includeGlobals());
5068
5069 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005070 AddMacroResults(PP, Results, false);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005071
5072 HandleCodeCompleteResults(this, CodeCompleter,
5073 CodeCompletionContext::CCC_Type,
5074 Results.data(), Results.size());
5075}
5076
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005077/// \brief When we have an expression with type "id", we may assume
5078/// that it has some more-specific class type based on knowledge of
5079/// common uses of Objective-C. This routine returns that class type,
5080/// or NULL if no better result could be determined.
5081static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregored0b69d2010-09-15 16:23:04 +00005082 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005083 if (!Msg)
5084 return 0;
5085
5086 Selector Sel = Msg->getSelector();
5087 if (Sel.isNull())
5088 return 0;
5089
5090 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5091 if (!Id)
5092 return 0;
5093
5094 ObjCMethodDecl *Method = Msg->getMethodDecl();
5095 if (!Method)
5096 return 0;
5097
5098 // Determine the class that we're sending the message to.
Douglas Gregor9a129192010-04-21 00:45:42 +00005099 ObjCInterfaceDecl *IFace = 0;
5100 switch (Msg->getReceiverKind()) {
5101 case ObjCMessageExpr::Class:
John McCall8b07ec22010-05-15 11:32:37 +00005102 if (const ObjCObjectType *ObjType
5103 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5104 IFace = ObjType->getInterface();
Douglas Gregor9a129192010-04-21 00:45:42 +00005105 break;
5106
5107 case ObjCMessageExpr::Instance: {
5108 QualType T = Msg->getInstanceReceiver()->getType();
5109 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5110 IFace = Ptr->getInterfaceDecl();
5111 break;
5112 }
5113
5114 case ObjCMessageExpr::SuperInstance:
5115 case ObjCMessageExpr::SuperClass:
5116 break;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005117 }
5118
5119 if (!IFace)
5120 return 0;
5121
5122 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5123 if (Method->isInstanceMethod())
5124 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5125 .Case("retain", IFace)
John McCall31168b02011-06-15 23:02:42 +00005126 .Case("strong", IFace)
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005127 .Case("autorelease", IFace)
5128 .Case("copy", IFace)
5129 .Case("copyWithZone", IFace)
5130 .Case("mutableCopy", IFace)
5131 .Case("mutableCopyWithZone", IFace)
5132 .Case("awakeFromCoder", IFace)
5133 .Case("replacementObjectFromCoder", IFace)
5134 .Case("class", IFace)
5135 .Case("classForCoder", IFace)
5136 .Case("superclass", Super)
5137 .Default(0);
5138
5139 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5140 .Case("new", IFace)
5141 .Case("alloc", IFace)
5142 .Case("allocWithZone", IFace)
5143 .Case("class", IFace)
5144 .Case("superclass", Super)
5145 .Default(0);
5146}
5147
Douglas Gregor6fc04132010-08-27 15:10:57 +00005148// Add a special completion for a message send to "super", which fills in the
5149// most likely case of forwarding all of our arguments to the superclass
5150// function.
5151///
5152/// \param S The semantic analysis object.
5153///
Dmitri Gribenkoadba9be2012-08-23 17:58:28 +00005154/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor6fc04132010-08-27 15:10:57 +00005155/// the "super" keyword. Otherwise, we just need to provide the arguments.
5156///
5157/// \param SelIdents The identifiers in the selector that have already been
5158/// provided as arguments for a send to "super".
5159///
Douglas Gregor6fc04132010-08-27 15:10:57 +00005160/// \param Results The set of results to augment.
5161///
5162/// \returns the Objective-C method declaration that would be invoked by
5163/// this "super" completion. If NULL, no completion was added.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005164static ObjCMethodDecl *AddSuperSendCompletion(
5165 Sema &S, bool NeedSuperKeyword,
5166 ArrayRef<IdentifierInfo *> SelIdents,
5167 ResultBuilder &Results) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005168 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5169 if (!CurMethod)
5170 return 0;
5171
5172 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5173 if (!Class)
5174 return 0;
5175
5176 // Try to find a superclass method with the same selector.
5177 ObjCMethodDecl *SuperMethod = 0;
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005178 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5179 // Check in the class
Douglas Gregor6fc04132010-08-27 15:10:57 +00005180 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5181 CurMethod->isInstanceMethod());
5182
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005183 // Check in categories or class extensions.
5184 if (!SuperMethod) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005185 for (ObjCInterfaceDecl::known_categories_iterator
5186 Cat = Class->known_categories_begin(),
5187 CatEnd = Class->known_categories_end();
5188 Cat != CatEnd; ++Cat) {
5189 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005190 CurMethod->isInstanceMethod())))
5191 break;
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005192 }
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005193 }
5194 }
5195
Douglas Gregor6fc04132010-08-27 15:10:57 +00005196 if (!SuperMethod)
5197 return 0;
5198
5199 // Check whether the superclass method has the same signature.
5200 if (CurMethod->param_size() != SuperMethod->param_size() ||
5201 CurMethod->isVariadic() != SuperMethod->isVariadic())
5202 return 0;
5203
5204 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5205 CurPEnd = CurMethod->param_end(),
5206 SuperP = SuperMethod->param_begin();
5207 CurP != CurPEnd; ++CurP, ++SuperP) {
5208 // Make sure the parameter types are compatible.
5209 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5210 (*SuperP)->getType()))
5211 return 0;
5212
5213 // Make sure we have a parameter name to forward!
5214 if (!(*CurP)->getIdentifier())
5215 return 0;
5216 }
5217
5218 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005219 CodeCompletionBuilder Builder(Results.getAllocator(),
5220 Results.getCodeCompletionTUInfo());
Douglas Gregor6fc04132010-08-27 15:10:57 +00005221
5222 // Give this completion a return type.
Douglas Gregor75acd922011-09-27 23:30:47 +00005223 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5224 Builder);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005225
5226 // If we need the "super" keyword, add it (plus some spacing).
5227 if (NeedSuperKeyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005228 Builder.AddTypedTextChunk("super");
5229 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005230 }
5231
5232 Selector Sel = CurMethod->getSelector();
5233 if (Sel.isUnarySelector()) {
5234 if (NeedSuperKeyword)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005235 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005236 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005237 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005238 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005239 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005240 } else {
5241 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5242 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005243 if (I > SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005244 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005245
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005246 if (I < SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005247 Builder.AddInformativeChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005248 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005249 Sel.getNameForSlot(I) + ":"));
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005250 else if (NeedSuperKeyword || I > SelIdents.size()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005251 Builder.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005252 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005253 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005254 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005255 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005256 } else {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005257 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005258 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005259 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005260 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005261 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005262 }
5263 }
5264 }
5265
Douglas Gregor78254c82012-03-27 23:34:16 +00005266 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5267 CCP_SuperCompletion));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005268 return SuperMethod;
5269}
5270
Douglas Gregora817a192010-05-27 23:06:34 +00005271void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall276321a2010-08-25 06:19:51 +00005272 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005273 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005274 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005275 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005276 getLangOpts().CPlusPlus11
Douglas Gregord8c61782012-02-15 15:34:24 +00005277 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5278 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregora817a192010-05-27 23:06:34 +00005279
Douglas Gregora817a192010-05-27 23:06:34 +00005280 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5281 Results.EnterNewScope();
Douglas Gregor39982192010-08-15 06:18:01 +00005282 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5283 CodeCompleter->includeGlobals());
Douglas Gregora817a192010-05-27 23:06:34 +00005284
5285 // If we are in an Objective-C method inside a class that has a superclass,
5286 // add "super" as an option.
5287 if (ObjCMethodDecl *Method = getCurMethodDecl())
5288 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor6fc04132010-08-27 15:10:57 +00005289 if (Iface->getSuperClass()) {
Douglas Gregora817a192010-05-27 23:06:34 +00005290 Results.AddResult(Result("super"));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005291
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005292 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, None, Results);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005293 }
Douglas Gregora817a192010-05-27 23:06:34 +00005294
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005295 if (getLangOpts().CPlusPlus11)
Douglas Gregord8c61782012-02-15 15:34:24 +00005296 addThisCompletion(*this, Results);
5297
Douglas Gregora817a192010-05-27 23:06:34 +00005298 Results.ExitScope();
5299
5300 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005301 AddMacroResults(PP, Results, false);
Douglas Gregor50832e02010-09-20 22:39:41 +00005302 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005303 Results.data(), Results.size());
Douglas Gregora817a192010-05-27 23:06:34 +00005304
5305}
5306
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005307void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005308 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005309 bool AtArgumentExpression) {
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005310 ObjCInterfaceDecl *CDecl = 0;
5311 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5312 // Figure out which interface we're in.
5313 CDecl = CurMethod->getClassInterface();
5314 if (!CDecl)
5315 return;
5316
5317 // Find the superclass of this class.
5318 CDecl = CDecl->getSuperClass();
5319 if (!CDecl)
5320 return;
5321
5322 if (CurMethod->isInstanceMethod()) {
5323 // We are inside an instance method, which means that the message
5324 // send [super ...] is actually calling an instance method on the
Douglas Gregor392a84b2010-10-13 21:24:53 +00005325 // current object.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005326 return CodeCompleteObjCInstanceMessage(S, 0, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005327 AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005328 CDecl);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005329 }
5330
5331 // Fall through to send to the superclass in CDecl.
5332 } else {
5333 // "super" may be the name of a type or variable. Figure out which
5334 // it is.
Argyrios Kyrtzidis3e56dd42013-03-14 22:56:43 +00005335 IdentifierInfo *Super = getSuperIdentifier();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005336 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5337 LookupOrdinaryName);
5338 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5339 // "super" names an interface. Use it.
5340 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCall8b07ec22010-05-15 11:32:37 +00005341 if (const ObjCObjectType *Iface
5342 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5343 CDecl = Iface->getInterface();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005344 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5345 // "super" names an unresolved type; we can't be more specific.
5346 } else {
5347 // Assume that "super" names some kind of value and parse that way.
5348 CXXScopeSpec SS;
Abramo Bagnara7945c982012-01-27 09:46:47 +00005349 SourceLocation TemplateKWLoc;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005350 UnqualifiedId id;
5351 id.setIdentifier(Super, SuperLoc);
Abramo Bagnara7945c982012-01-27 09:46:47 +00005352 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5353 false, false);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005354 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005355 SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005356 AtArgumentExpression);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005357 }
5358
5359 // Fall through
5360 }
5361
John McCallba7bf592010-08-24 05:47:05 +00005362 ParsedType Receiver;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005363 if (CDecl)
John McCallba7bf592010-08-24 05:47:05 +00005364 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005365 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005366 AtArgumentExpression,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005367 /*IsSuper=*/true);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005368}
5369
Douglas Gregor74661272010-09-21 00:03:25 +00005370/// \brief Given a set of code-completion results for the argument of a message
5371/// send, determine the preferred type (if any) for that argument expression.
5372static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5373 unsigned NumSelIdents) {
5374 typedef CodeCompletionResult Result;
5375 ASTContext &Context = Results.getSema().Context;
5376
5377 QualType PreferredType;
5378 unsigned BestPriority = CCP_Unlikely * 2;
5379 Result *ResultsData = Results.data();
5380 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5381 Result &R = ResultsData[I];
5382 if (R.Kind == Result::RK_Declaration &&
5383 isa<ObjCMethodDecl>(R.Declaration)) {
5384 if (R.Priority <= BestPriority) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00005385 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregor74661272010-09-21 00:03:25 +00005386 if (NumSelIdents <= Method->param_size()) {
5387 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
5388 ->getType();
5389 if (R.Priority < BestPriority || PreferredType.isNull()) {
5390 BestPriority = R.Priority;
5391 PreferredType = MyPreferredType;
5392 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5393 MyPreferredType)) {
5394 PreferredType = QualType();
5395 }
5396 }
5397 }
5398 }
5399 }
5400
5401 return PreferredType;
5402}
5403
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005404static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5405 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005406 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005407 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005408 bool IsSuper,
5409 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005410 typedef CodeCompletionResult Result;
Douglas Gregor8ce33212009-11-17 17:59:40 +00005411 ObjCInterfaceDecl *CDecl = 0;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005412
Douglas Gregor8ce33212009-11-17 17:59:40 +00005413 // If the given name refers to an interface type, retrieve the
5414 // corresponding declaration.
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005415 if (Receiver) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005416 QualType T = SemaRef.GetTypeFromParser(Receiver, 0);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005417 if (!T.isNull())
John McCall8b07ec22010-05-15 11:32:37 +00005418 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5419 CDecl = Interface->getInterface();
Douglas Gregor8ce33212009-11-17 17:59:40 +00005420 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005421
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005422 // Add all of the factory methods in this Objective-C class, its protocols,
5423 // superclasses, categories, implementation, etc.
Steve Naroffeae65032009-11-07 02:08:14 +00005424 Results.EnterNewScope();
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005425
Douglas Gregor6fc04132010-08-27 15:10:57 +00005426 // If this is a send-to-super, try to add the special "super" send
5427 // completion.
5428 if (IsSuper) {
5429 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005430 = AddSuperSendCompletion(SemaRef, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005431 Results.Ignore(SuperMethod);
5432 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005433
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005434 // If we're inside an Objective-C method definition, prefer its selector to
5435 // others.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005436 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005437 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005438
Douglas Gregor1154e272010-09-16 16:06:31 +00005439 VisitedSelectorSet Selectors;
Douglas Gregor6285f752010-04-06 16:40:00 +00005440 if (CDecl)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005441 AddObjCMethods(CDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005442 SemaRef.CurContext, Selectors, AtArgumentExpression,
5443 Results);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005444 else {
Douglas Gregor6285f752010-04-06 16:40:00 +00005445 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005446
Douglas Gregord720daf2010-04-06 17:30:22 +00005447 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005448 // pool from the AST file.
Axel Naumanndd433f02012-10-18 19:05:02 +00005449 if (SemaRef.getExternalSource()) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005450 for (uint32_t I = 0,
Axel Naumanndd433f02012-10-18 19:05:02 +00005451 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall75b960e2010-06-01 09:23:16 +00005452 I != N; ++I) {
Axel Naumanndd433f02012-10-18 19:05:02 +00005453 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005454 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005455 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005456
5457 SemaRef.ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005458 }
5459 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005460
5461 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5462 MEnd = SemaRef.MethodPool.end();
Sebastian Redl75d8a322010-08-02 23:18:59 +00005463 M != MEnd; ++M) {
5464 for (ObjCMethodList *MethList = &M->second.second;
5465 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005466 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005467 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005468 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005469
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005470 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005471 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005472 R.AllParametersAreInformative = false;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005473 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor6285f752010-04-06 16:40:00 +00005474 }
5475 }
5476 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005477
5478 Results.ExitScope();
5479}
Douglas Gregor6285f752010-04-06 16:40:00 +00005480
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005481void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005482 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005483 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005484 bool IsSuper) {
Douglas Gregor63745d52011-07-21 01:05:26 +00005485
5486 QualType T = this->GetTypeFromParser(Receiver);
5487
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005488 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005489 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005490 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005491 T, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005492
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005493 AddClassMessageCompletions(*this, S, Receiver, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005494 AtArgumentExpression, IsSuper, Results);
Douglas Gregor74661272010-09-21 00:03:25 +00005495
5496 // If we're actually at the argument expression (rather than prior to the
5497 // selector), we're actually performing code completion for an expression.
5498 // Determine whether we have a single, best method. If so, we can
5499 // code-complete the expression using the corresponding parameter type as
5500 // our preferred type, improving completion results.
5501 if (AtArgumentExpression) {
5502 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005503 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005504 if (PreferredType.isNull())
5505 CodeCompleteOrdinaryName(S, PCC_Expression);
5506 else
5507 CodeCompleteExpression(S, PreferredType);
5508 return;
5509 }
5510
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005511 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005512 Results.getCompletionContext(),
Douglas Gregor6fc04132010-08-27 15:10:57 +00005513 Results.data(), Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005514}
5515
Richard Trieu2bd04012011-09-09 02:00:50 +00005516void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005517 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005518 bool AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005519 ObjCInterfaceDecl *Super) {
John McCall276321a2010-08-25 06:19:51 +00005520 typedef CodeCompletionResult Result;
Steve Naroffeae65032009-11-07 02:08:14 +00005521
5522 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffeae65032009-11-07 02:08:14 +00005523
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005524 // If necessary, apply function/array conversion to the receiver.
5525 // C99 6.7.5.3p[7,8].
John Wiegley01296292011-04-08 18:41:53 +00005526 if (RecExpr) {
5527 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5528 if (Conv.isInvalid()) // conversion failed. bail.
5529 return;
5530 RecExpr = Conv.take();
5531 }
Douglas Gregor392a84b2010-10-13 21:24:53 +00005532 QualType ReceiverType = RecExpr? RecExpr->getType()
5533 : Super? Context.getObjCObjectPointerType(
5534 Context.getObjCInterfaceType(Super))
5535 : Context.getObjCIdType();
Steve Naroffeae65032009-11-07 02:08:14 +00005536
Douglas Gregordc520b02010-11-08 21:12:30 +00005537 // If we're messaging an expression with type "id" or "Class", check
5538 // whether we know something special about the receiver that allows
5539 // us to assume a more-specific receiver type.
5540 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType())
5541 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5542 if (ReceiverType->isObjCClassType())
5543 return CodeCompleteObjCClassMessage(S,
5544 ParsedType::make(Context.getObjCInterfaceType(IFace)),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005545 SelIdents,
Douglas Gregordc520b02010-11-08 21:12:30 +00005546 AtArgumentExpression, Super);
5547
5548 ReceiverType = Context.getObjCObjectPointerType(
5549 Context.getObjCInterfaceType(IFace));
5550 }
5551
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005552 // Build the set of methods we can see.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005553 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005554 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005555 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005556 ReceiverType, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005557
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005558 Results.EnterNewScope();
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005559
Douglas Gregor6fc04132010-08-27 15:10:57 +00005560 // If this is a send-to-super, try to add the special "super" send
5561 // completion.
Douglas Gregor392a84b2010-10-13 21:24:53 +00005562 if (Super) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005563 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005564 = AddSuperSendCompletion(*this, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005565 Results.Ignore(SuperMethod);
5566 }
5567
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005568 // If we're inside an Objective-C method definition, prefer its selector to
5569 // others.
5570 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5571 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005572
Douglas Gregor1154e272010-09-16 16:06:31 +00005573 // Keep track of the selectors we've already added.
5574 VisitedSelectorSet Selectors;
5575
Douglas Gregora3329fa2009-11-18 00:06:18 +00005576 // Handle messages to Class. This really isn't a message to an instance
5577 // method, so we treat it the same way we would treat a message send to a
5578 // class method.
5579 if (ReceiverType->isObjCClassType() ||
5580 ReceiverType->isObjCQualifiedClassType()) {
5581 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5582 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005583 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005584 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005585 }
5586 }
5587 // Handle messages to a qualified ID ("id<foo>").
5588 else if (const ObjCObjectPointerType *QualID
5589 = ReceiverType->getAsObjCQualifiedIdType()) {
5590 // Search protocols for instance methods.
5591 for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(),
5592 E = QualID->qual_end();
5593 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005594 AddObjCMethods(*I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005595 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005596 }
5597 // Handle messages to a pointer to interface type.
5598 else if (const ObjCObjectPointerType *IFacePtr
5599 = ReceiverType->getAsObjCInterfacePointerType()) {
5600 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregorc8537c52009-11-19 07:41:15 +00005601 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005602 CurContext, Selectors, AtArgumentExpression,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005603 Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005604
5605 // Search protocols for instance methods.
5606 for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(),
5607 E = IFacePtr->qual_end();
5608 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005609 AddObjCMethods(*I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005610 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005611 }
Douglas Gregor6285f752010-04-06 16:40:00 +00005612 // Handle messages to "id".
5613 else if (ReceiverType->isObjCIdType()) {
Douglas Gregord720daf2010-04-06 17:30:22 +00005614 // We're messaging "id", so provide all instance methods we know
5615 // about as code-completion results.
5616
5617 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005618 // pool from the AST file.
Douglas Gregord720daf2010-04-06 17:30:22 +00005619 if (ExternalSource) {
John McCall75b960e2010-06-01 09:23:16 +00005620 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5621 I != N; ++I) {
5622 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00005623 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005624 continue;
5625
Sebastian Redl75d8a322010-08-02 23:18:59 +00005626 ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005627 }
5628 }
5629
Sebastian Redl75d8a322010-08-02 23:18:59 +00005630 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5631 MEnd = MethodPool.end();
5632 M != MEnd; ++M) {
5633 for (ObjCMethodList *MethList = &M->second.first;
5634 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005635 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005636 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005637 continue;
Douglas Gregor1154e272010-09-16 16:06:31 +00005638
5639 if (!Selectors.insert(MethList->Method->getSelector()))
5640 continue;
5641
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005642 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005643 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005644 R.AllParametersAreInformative = false;
5645 Results.MaybeAddResult(R, CurContext);
5646 }
5647 }
5648 }
Steve Naroffeae65032009-11-07 02:08:14 +00005649 Results.ExitScope();
Douglas Gregor74661272010-09-21 00:03:25 +00005650
5651
5652 // If we're actually at the argument expression (rather than prior to the
5653 // selector), we're actually performing code completion for an expression.
5654 // Determine whether we have a single, best method. If so, we can
5655 // code-complete the expression using the corresponding parameter type as
5656 // our preferred type, improving completion results.
5657 if (AtArgumentExpression) {
5658 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005659 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005660 if (PreferredType.isNull())
5661 CodeCompleteOrdinaryName(S, PCC_Expression);
5662 else
5663 CodeCompleteExpression(S, PreferredType);
5664 return;
5665 }
5666
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005667 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005668 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005669 Results.data(),Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005670}
Douglas Gregorbaf69612009-11-18 04:19:12 +00005671
Douglas Gregor68762e72010-08-23 21:17:50 +00005672void Sema::CodeCompleteObjCForCollection(Scope *S,
5673 DeclGroupPtrTy IterationVar) {
5674 CodeCompleteExpressionData Data;
5675 Data.ObjCCollection = true;
5676
5677 if (IterationVar.getAsOpaquePtr()) {
Serge Pavlov9ddb76e2013-08-27 13:15:56 +00005678 DeclGroupRef DG = IterationVar.get();
Douglas Gregor68762e72010-08-23 21:17:50 +00005679 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5680 if (*I)
5681 Data.IgnoreDecls.push_back(*I);
5682 }
5683 }
5684
5685 CodeCompleteExpression(S, Data);
5686}
5687
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005688void Sema::CodeCompleteObjCSelector(Scope *S,
5689 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00005690 // If we have an external source, load the entire class method
5691 // pool from the AST file.
5692 if (ExternalSource) {
5693 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5694 I != N; ++I) {
5695 Selector Sel = ExternalSource->GetExternalSelector(I);
5696 if (Sel.isNull() || MethodPool.count(Sel))
5697 continue;
5698
5699 ReadMethodPool(Sel);
5700 }
5701 }
5702
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005703 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005704 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005705 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor67c692c2010-08-26 15:07:07 +00005706 Results.EnterNewScope();
5707 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5708 MEnd = MethodPool.end();
5709 M != MEnd; ++M) {
5710
5711 Selector Sel = M->first;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005712 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents))
Douglas Gregor67c692c2010-08-26 15:07:07 +00005713 continue;
5714
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005715 CodeCompletionBuilder Builder(Results.getAllocator(),
5716 Results.getCodeCompletionTUInfo());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005717 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005718 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005719 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005720 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005721 continue;
5722 }
5723
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005724 std::string Accumulator;
Douglas Gregor67c692c2010-08-26 15:07:07 +00005725 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005726 if (I == SelIdents.size()) {
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005727 if (!Accumulator.empty()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005728 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005729 Accumulator));
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005730 Accumulator.clear();
5731 }
5732 }
5733
Benjamin Kramer632500c2011-07-26 16:59:25 +00005734 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005735 Accumulator += ':';
Douglas Gregor67c692c2010-08-26 15:07:07 +00005736 }
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005737 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005738 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005739 }
5740 Results.ExitScope();
5741
5742 HandleCodeCompleteResults(this, CodeCompleter,
5743 CodeCompletionContext::CCC_SelectorName,
5744 Results.data(), Results.size());
5745}
5746
Douglas Gregorbaf69612009-11-18 04:19:12 +00005747/// \brief Add all of the protocol declarations that we find in the given
5748/// (translation unit) context.
5749static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005750 bool OnlyForwardDeclarations,
Douglas Gregorbaf69612009-11-18 04:19:12 +00005751 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005752 typedef CodeCompletionResult Result;
Douglas Gregorbaf69612009-11-18 04:19:12 +00005753
5754 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5755 DEnd = Ctx->decls_end();
5756 D != DEnd; ++D) {
5757 // Record any protocols we find.
5758 if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*D))
Douglas Gregore6e48b12012-01-01 19:29:29 +00005759 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005760 Results.AddResult(Result(Proto, Results.getBasePriority(Proto), 0),
5761 CurContext, 0, false);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005762 }
5763}
5764
5765void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5766 unsigned NumProtocols) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005767 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005768 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005769 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005770
Douglas Gregora3b23b02010-12-09 21:44:02 +00005771 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5772 Results.EnterNewScope();
5773
5774 // Tell the result set to ignore all of the protocols we have
5775 // already seen.
5776 // FIXME: This doesn't work when caching code-completion results.
5777 for (unsigned I = 0; I != NumProtocols; ++I)
5778 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5779 Protocols[I].second))
5780 Results.Ignore(Protocol);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005781
Douglas Gregora3b23b02010-12-09 21:44:02 +00005782 // Add all protocols.
5783 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5784 Results);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005785
Douglas Gregora3b23b02010-12-09 21:44:02 +00005786 Results.ExitScope();
5787 }
5788
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005789 HandleCodeCompleteResults(this, CodeCompleter,
5790 CodeCompletionContext::CCC_ObjCProtocolName,
5791 Results.data(),Results.size());
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005792}
5793
5794void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005795 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005796 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005797 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005798
Douglas Gregora3b23b02010-12-09 21:44:02 +00005799 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5800 Results.EnterNewScope();
5801
5802 // Add all protocols.
5803 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5804 Results);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005805
Douglas Gregora3b23b02010-12-09 21:44:02 +00005806 Results.ExitScope();
5807 }
5808
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005809 HandleCodeCompleteResults(this, CodeCompleter,
5810 CodeCompletionContext::CCC_ObjCProtocolName,
5811 Results.data(),Results.size());
Douglas Gregorbaf69612009-11-18 04:19:12 +00005812}
Douglas Gregor49c22a72009-11-18 16:26:39 +00005813
5814/// \brief Add all of the Objective-C interface declarations that we find in
5815/// the given (translation unit) context.
5816static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5817 bool OnlyForwardDeclarations,
5818 bool OnlyUnimplemented,
5819 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005820 typedef CodeCompletionResult Result;
Douglas Gregor49c22a72009-11-18 16:26:39 +00005821
5822 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5823 DEnd = Ctx->decls_end();
5824 D != DEnd; ++D) {
Douglas Gregor1c283312010-08-11 12:19:30 +00005825 // Record any interfaces we find.
5826 if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*D))
Douglas Gregordc9166c2011-12-15 20:29:51 +00005827 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregor1c283312010-08-11 12:19:30 +00005828 (!OnlyUnimplemented || !Class->getImplementation()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005829 Results.AddResult(Result(Class, Results.getBasePriority(Class), 0),
5830 CurContext, 0, false);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005831 }
5832}
5833
5834void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005835 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005836 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005837 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005838 Results.EnterNewScope();
5839
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005840 if (CodeCompleter->includeGlobals()) {
5841 // Add all classes.
5842 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5843 false, Results);
5844 }
5845
Douglas Gregor49c22a72009-11-18 16:26:39 +00005846 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005847
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005848 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005849 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005850 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005851}
5852
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005853void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5854 SourceLocation ClassNameLoc) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005855 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005856 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005857 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005858 Results.EnterNewScope();
5859
5860 // Make sure that we ignore the class we're currently defining.
5861 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005862 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005863 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor49c22a72009-11-18 16:26:39 +00005864 Results.Ignore(CurClass);
5865
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005866 if (CodeCompleter->includeGlobals()) {
5867 // Add all classes.
5868 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5869 false, Results);
5870 }
5871
Douglas Gregor49c22a72009-11-18 16:26:39 +00005872 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005873
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005874 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005875 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005876 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005877}
5878
5879void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005880 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005881 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005882 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005883 Results.EnterNewScope();
5884
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005885 if (CodeCompleter->includeGlobals()) {
5886 // Add all unimplemented classes.
5887 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5888 true, Results);
5889 }
5890
Douglas Gregor49c22a72009-11-18 16:26:39 +00005891 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005892
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005893 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005894 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005895 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005896}
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005897
5898void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005899 IdentifierInfo *ClassName,
5900 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005901 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005902
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005903 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005904 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005905 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005906
5907 // Ignore any categories we find that have already been implemented by this
5908 // interface.
5909 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5910 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005911 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005912 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
5913 for (ObjCInterfaceDecl::visible_categories_iterator
5914 Cat = Class->visible_categories_begin(),
5915 CatEnd = Class->visible_categories_end();
5916 Cat != CatEnd; ++Cat) {
5917 CategoryNames.insert(Cat->getIdentifier());
5918 }
5919 }
5920
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005921 // Add all of the categories we know about.
5922 Results.EnterNewScope();
5923 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
5924 for (DeclContext::decl_iterator D = TU->decls_begin(),
5925 DEnd = TU->decls_end();
5926 D != DEnd; ++D)
5927 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(*D))
5928 if (CategoryNames.insert(Category->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005929 Results.AddResult(Result(Category, Results.getBasePriority(Category),0),
5930 CurContext, 0, false);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005931 Results.ExitScope();
5932
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005933 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005934 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005935 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005936}
5937
5938void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005939 IdentifierInfo *ClassName,
5940 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005941 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005942
5943 // Find the corresponding interface. If we couldn't find the interface, the
5944 // program itself is ill-formed. However, we'll try to be helpful still by
5945 // providing the list of all of the categories we know about.
5946 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005947 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005948 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5949 if (!Class)
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005950 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005951
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005952 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005953 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005954 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005955
5956 // Add all of the categories that have have corresponding interface
5957 // declarations in this class and any of its superclasses, except for
5958 // already-implemented categories in the class itself.
5959 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5960 Results.EnterNewScope();
5961 bool IgnoreImplemented = true;
5962 while (Class) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005963 for (ObjCInterfaceDecl::visible_categories_iterator
5964 Cat = Class->visible_categories_begin(),
5965 CatEnd = Class->visible_categories_end();
5966 Cat != CatEnd; ++Cat) {
5967 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
5968 CategoryNames.insert(Cat->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005969 Results.AddResult(Result(*Cat, Results.getBasePriority(*Cat), 0),
5970 CurContext, 0, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005971 }
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005972
5973 Class = Class->getSuperClass();
5974 IgnoreImplemented = false;
5975 }
5976 Results.ExitScope();
5977
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005978 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005979 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005980 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005981}
Douglas Gregor5d649882009-11-18 22:32:06 +00005982
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005983void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005984 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005985 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005986 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00005987
5988 // Figure out where this @synthesize lives.
5989 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005990 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00005991 if (!Container ||
5992 (!isa<ObjCImplementationDecl>(Container) &&
5993 !isa<ObjCCategoryImplDecl>(Container)))
5994 return;
5995
5996 // Ignore any properties that have already been implemented.
Douglas Gregor9b4f3702012-06-12 13:44:08 +00005997 Container = getContainerDef(Container);
5998 for (DeclContext::decl_iterator D = Container->decls_begin(),
Douglas Gregor5d649882009-11-18 22:32:06 +00005999 DEnd = Container->decls_end();
6000 D != DEnd; ++D)
6001 if (ObjCPropertyImplDecl *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(*D))
6002 Results.Ignore(PropertyImpl->getPropertyDecl());
6003
6004 // Add any properties that we find.
Douglas Gregorb888acf2010-12-09 23:01:55 +00006005 AddedPropertiesSet AddedProperties;
Douglas Gregor5d649882009-11-18 22:32:06 +00006006 Results.EnterNewScope();
6007 if (ObjCImplementationDecl *ClassImpl
6008 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor95147142011-05-05 15:50:42 +00006009 AddObjCProperties(ClassImpl->getClassInterface(), false,
6010 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00006011 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00006012 else
6013 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor95147142011-05-05 15:50:42 +00006014 false, /*AllowNullaryMethods=*/false, CurContext,
6015 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00006016 Results.ExitScope();
6017
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006018 HandleCodeCompleteResults(this, CodeCompleter,
6019 CodeCompletionContext::CCC_Other,
6020 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006021}
6022
6023void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006024 IdentifierInfo *PropertyName) {
John McCall276321a2010-08-25 06:19:51 +00006025 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006026 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006027 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006028 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00006029
6030 // Figure out where this @synthesize lives.
6031 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006032 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00006033 if (!Container ||
6034 (!isa<ObjCImplementationDecl>(Container) &&
6035 !isa<ObjCCategoryImplDecl>(Container)))
6036 return;
6037
6038 // Figure out which interface we're looking into.
6039 ObjCInterfaceDecl *Class = 0;
6040 if (ObjCImplementationDecl *ClassImpl
6041 = dyn_cast<ObjCImplementationDecl>(Container))
6042 Class = ClassImpl->getClassInterface();
6043 else
6044 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
6045 ->getClassInterface();
6046
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006047 // Determine the type of the property we're synthesizing.
6048 QualType PropertyType = Context.getObjCIdType();
6049 if (Class) {
6050 if (ObjCPropertyDecl *Property
6051 = Class->FindPropertyDeclaration(PropertyName)) {
6052 PropertyType
6053 = Property->getType().getNonReferenceType().getUnqualifiedType();
6054
6055 // Give preference to ivars
6056 Results.setPreferredType(PropertyType);
6057 }
6058 }
6059
Douglas Gregor5d649882009-11-18 22:32:06 +00006060 // Add all of the instance variables in this class and its superclasses.
6061 Results.EnterNewScope();
Douglas Gregor331faa02011-04-18 14:13:53 +00006062 bool SawSimilarlyNamedIvar = false;
6063 std::string NameWithPrefix;
6064 NameWithPrefix += '_';
Benjamin Kramer632500c2011-07-26 16:59:25 +00006065 NameWithPrefix += PropertyName->getName();
Douglas Gregor331faa02011-04-18 14:13:53 +00006066 std::string NameWithSuffix = PropertyName->getName().str();
6067 NameWithSuffix += '_';
Douglas Gregor5d649882009-11-18 22:32:06 +00006068 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006069 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
6070 Ivar = Ivar->getNextIvar()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00006071 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), 0),
6072 CurContext, 0, false);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006073
Douglas Gregor331faa02011-04-18 14:13:53 +00006074 // Determine whether we've seen an ivar with a name similar to the
6075 // property.
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006076 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregor331faa02011-04-18 14:13:53 +00006077 NameWithPrefix == Ivar->getName() ||
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006078 NameWithSuffix == Ivar->getName())) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006079 SawSimilarlyNamedIvar = true;
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006080
6081 // Reduce the priority of this result by one, to give it a slight
6082 // advantage over other results whose names don't match so closely.
6083 if (Results.size() &&
6084 Results.data()[Results.size() - 1].Kind
6085 == CodeCompletionResult::RK_Declaration &&
6086 Results.data()[Results.size() - 1].Declaration == Ivar)
6087 Results.data()[Results.size() - 1].Priority--;
6088 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006089 }
Douglas Gregor5d649882009-11-18 22:32:06 +00006090 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006091
6092 if (!SawSimilarlyNamedIvar) {
6093 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006094 // an ivar of the appropriate type.
6095 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregor331faa02011-04-18 14:13:53 +00006096 typedef CodeCompletionResult Result;
6097 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006098 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6099 Priority,CXAvailability_Available);
Douglas Gregor331faa02011-04-18 14:13:53 +00006100
Douglas Gregor75acd922011-09-27 23:30:47 +00006101 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006102 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006103 Policy, Allocator));
Douglas Gregor331faa02011-04-18 14:13:53 +00006104 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6105 Results.AddResult(Result(Builder.TakeString(), Priority,
6106 CXCursor_ObjCIvarDecl));
6107 }
6108
Douglas Gregor5d649882009-11-18 22:32:06 +00006109 Results.ExitScope();
6110
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006111 HandleCodeCompleteResults(this, CodeCompleter,
6112 CodeCompletionContext::CCC_Other,
6113 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006114}
Douglas Gregor636a61e2010-04-07 00:21:17 +00006115
Douglas Gregor416b5752010-08-25 01:08:01 +00006116// Mapping from selectors to the methods that implement that selector, along
6117// with the "in original class" flag.
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006118typedef llvm::DenseMap<
6119 Selector, llvm::PointerIntPair<ObjCMethodDecl *, 1, bool> > KnownMethodsMap;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006120
6121/// \brief Find all of the methods that reside in the given container
6122/// (and its superclasses, protocols, etc.) that meet the given
6123/// criteria. Insert those methods into the map of known methods,
6124/// indexed by selector so they can be easily found.
6125static void FindImplementableMethods(ASTContext &Context,
6126 ObjCContainerDecl *Container,
6127 bool WantInstanceMethods,
6128 QualType ReturnType,
Douglas Gregor416b5752010-08-25 01:08:01 +00006129 KnownMethodsMap &KnownMethods,
6130 bool InOriginalClass = true) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006131 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006132 // Make sure we have a definition; that's what we'll walk.
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006133 if (!IFace->hasDefinition())
6134 return;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006135
6136 IFace = IFace->getDefinition();
6137 Container = IFace;
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006138
Douglas Gregor636a61e2010-04-07 00:21:17 +00006139 const ObjCList<ObjCProtocolDecl> &Protocols
6140 = IFace->getReferencedProtocols();
6141 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006142 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006143 I != E; ++I)
6144 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006145 KnownMethods, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006146
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006147 // Add methods from any class extensions and categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006148 for (ObjCInterfaceDecl::visible_categories_iterator
6149 Cat = IFace->visible_categories_begin(),
6150 CatEnd = IFace->visible_categories_end();
6151 Cat != CatEnd; ++Cat) {
6152 FindImplementableMethods(Context, *Cat, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006153 KnownMethods, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006154 }
6155
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006156 // Visit the superclass.
6157 if (IFace->getSuperClass())
6158 FindImplementableMethods(Context, IFace->getSuperClass(),
6159 WantInstanceMethods, ReturnType,
6160 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006161 }
6162
6163 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6164 // Recurse into protocols.
6165 const ObjCList<ObjCProtocolDecl> &Protocols
6166 = Category->getReferencedProtocols();
6167 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006168 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006169 I != E; ++I)
6170 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006171 KnownMethods, InOriginalClass);
6172
6173 // If this category is the original class, jump to the interface.
6174 if (InOriginalClass && Category->getClassInterface())
6175 FindImplementableMethods(Context, Category->getClassInterface(),
6176 WantInstanceMethods, ReturnType, KnownMethods,
6177 false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006178 }
6179
6180 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006181 // Make sure we have a definition; that's what we'll walk.
6182 if (!Protocol->hasDefinition())
6183 return;
6184 Protocol = Protocol->getDefinition();
6185 Container = Protocol;
6186
6187 // Recurse into protocols.
6188 const ObjCList<ObjCProtocolDecl> &Protocols
6189 = Protocol->getReferencedProtocols();
6190 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6191 E = Protocols.end();
6192 I != E; ++I)
6193 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6194 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006195 }
6196
6197 // Add methods in this container. This operation occurs last because
6198 // we want the methods from this container to override any methods
6199 // we've previously seen with the same selector.
6200 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
6201 MEnd = Container->meth_end();
6202 M != MEnd; ++M) {
David Blaikie2d7c57e2012-04-30 02:36:29 +00006203 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006204 if (!ReturnType.isNull() &&
David Blaikie2d7c57e2012-04-30 02:36:29 +00006205 !Context.hasSameUnqualifiedType(ReturnType, M->getResultType()))
Douglas Gregor636a61e2010-04-07 00:21:17 +00006206 continue;
6207
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006208 KnownMethods[M->getSelector()] =
6209 KnownMethodsMap::mapped_type(*M, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006210 }
6211 }
6212}
6213
Douglas Gregor669a25a2011-02-17 00:22:45 +00006214/// \brief Add the parenthesized return or parameter type chunk to a code
6215/// completion string.
6216static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor29979142012-04-10 18:35:07 +00006217 unsigned ObjCDeclQuals,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006218 ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006219 const PrintingPolicy &Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006220 CodeCompletionBuilder &Builder) {
6221 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor29979142012-04-10 18:35:07 +00006222 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6223 if (!Quals.empty())
6224 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor75acd922011-09-27 23:30:47 +00006225 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006226 Builder.getAllocator()));
6227 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6228}
6229
6230/// \brief Determine whether the given class is or inherits from a class by
6231/// the given name.
6232static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006233 StringRef Name) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006234 if (!Class)
6235 return false;
6236
6237 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6238 return true;
6239
6240 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6241}
6242
6243/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6244/// Key-Value Observing (KVO).
6245static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6246 bool IsInstanceMethod,
6247 QualType ReturnType,
6248 ASTContext &Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006249 VisitedSelectorSet &KnownSelectors,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006250 ResultBuilder &Results) {
6251 IdentifierInfo *PropName = Property->getIdentifier();
6252 if (!PropName || PropName->getLength() == 0)
6253 return;
6254
Douglas Gregor75acd922011-09-27 23:30:47 +00006255 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6256
Douglas Gregor669a25a2011-02-17 00:22:45 +00006257 // Builder that will create each code completion.
6258 typedef CodeCompletionResult Result;
6259 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006260 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor669a25a2011-02-17 00:22:45 +00006261
6262 // The selector table.
6263 SelectorTable &Selectors = Context.Selectors;
6264
6265 // The property name, copied into the code completion allocation region
6266 // on demand.
6267 struct KeyHolder {
6268 CodeCompletionAllocator &Allocator;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006269 StringRef Key;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006270 const char *CopiedKey;
6271
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006272 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006273 : Allocator(Allocator), Key(Key), CopiedKey(0) { }
6274
6275 operator const char *() {
6276 if (CopiedKey)
6277 return CopiedKey;
6278
6279 return CopiedKey = Allocator.CopyString(Key);
6280 }
6281 } Key(Allocator, PropName->getName());
6282
6283 // The uppercased name of the property name.
6284 std::string UpperKey = PropName->getName();
6285 if (!UpperKey.empty())
Jordan Rose4938f272013-02-09 10:09:43 +00006286 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006287
6288 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6289 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6290 Property->getType());
6291 bool ReturnTypeMatchesVoid
6292 = ReturnType.isNull() || ReturnType->isVoidType();
6293
6294 // Add the normal accessor -(type)key.
6295 if (IsInstanceMethod &&
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006296 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor669a25a2011-02-17 00:22:45 +00006297 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6298 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006299 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6300 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006301
6302 Builder.AddTypedTextChunk(Key);
6303 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6304 CXCursor_ObjCInstanceMethodDecl));
6305 }
6306
6307 // If we have an integral or boolean property (or the user has provided
6308 // an integral or boolean return type), add the accessor -(type)isKey.
6309 if (IsInstanceMethod &&
6310 ((!ReturnType.isNull() &&
6311 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6312 (ReturnType.isNull() &&
6313 (Property->getType()->isIntegerType() ||
6314 Property->getType()->isBooleanType())))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006315 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006316 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006317 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006318 if (ReturnType.isNull()) {
6319 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6320 Builder.AddTextChunk("BOOL");
6321 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6322 }
6323
6324 Builder.AddTypedTextChunk(
6325 Allocator.CopyString(SelectorId->getName()));
6326 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6327 CXCursor_ObjCInstanceMethodDecl));
6328 }
6329 }
6330
6331 // Add the normal mutator.
6332 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6333 !Property->getSetterMethodDecl()) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006334 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006335 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006336 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006337 if (ReturnType.isNull()) {
6338 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6339 Builder.AddTextChunk("void");
6340 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6341 }
6342
6343 Builder.AddTypedTextChunk(
6344 Allocator.CopyString(SelectorId->getName()));
6345 Builder.AddTypedTextChunk(":");
Douglas Gregor29979142012-04-10 18:35:07 +00006346 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6347 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006348 Builder.AddTextChunk(Key);
6349 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6350 CXCursor_ObjCInstanceMethodDecl));
6351 }
6352 }
6353
6354 // Indexed and unordered accessors
6355 unsigned IndexedGetterPriority = CCP_CodePattern;
6356 unsigned IndexedSetterPriority = CCP_CodePattern;
6357 unsigned UnorderedGetterPriority = CCP_CodePattern;
6358 unsigned UnorderedSetterPriority = CCP_CodePattern;
6359 if (const ObjCObjectPointerType *ObjCPointer
6360 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6361 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6362 // If this interface type is not provably derived from a known
6363 // collection, penalize the corresponding completions.
6364 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6365 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6366 if (!InheritsFromClassNamed(IFace, "NSArray"))
6367 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6368 }
6369
6370 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6371 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6372 if (!InheritsFromClassNamed(IFace, "NSSet"))
6373 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6374 }
6375 }
6376 } else {
6377 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6378 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6379 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6380 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6381 }
6382
6383 // Add -(NSUInteger)countOf<key>
6384 if (IsInstanceMethod &&
6385 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006386 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006387 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006388 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006389 if (ReturnType.isNull()) {
6390 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6391 Builder.AddTextChunk("NSUInteger");
6392 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6393 }
6394
6395 Builder.AddTypedTextChunk(
6396 Allocator.CopyString(SelectorId->getName()));
6397 Results.AddResult(Result(Builder.TakeString(),
6398 std::min(IndexedGetterPriority,
6399 UnorderedGetterPriority),
6400 CXCursor_ObjCInstanceMethodDecl));
6401 }
6402 }
6403
6404 // Indexed getters
6405 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6406 if (IsInstanceMethod &&
6407 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006408 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006409 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006410 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006411 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006412 if (ReturnType.isNull()) {
6413 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6414 Builder.AddTextChunk("id");
6415 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6416 }
6417
6418 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6419 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6420 Builder.AddTextChunk("NSUInteger");
6421 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6422 Builder.AddTextChunk("index");
6423 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6424 CXCursor_ObjCInstanceMethodDecl));
6425 }
6426 }
6427
6428 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6429 if (IsInstanceMethod &&
6430 (ReturnType.isNull() ||
6431 (ReturnType->isObjCObjectPointerType() &&
6432 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6433 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6434 ->getName() == "NSArray"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006435 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006436 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006437 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006438 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006439 if (ReturnType.isNull()) {
6440 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6441 Builder.AddTextChunk("NSArray *");
6442 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6443 }
6444
6445 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6446 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6447 Builder.AddTextChunk("NSIndexSet *");
6448 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6449 Builder.AddTextChunk("indexes");
6450 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6451 CXCursor_ObjCInstanceMethodDecl));
6452 }
6453 }
6454
6455 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6456 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006457 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006458 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006459 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006460 &Context.Idents.get("range")
6461 };
6462
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006463 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006464 if (ReturnType.isNull()) {
6465 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6466 Builder.AddTextChunk("void");
6467 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6468 }
6469
6470 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6471 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6472 Builder.AddPlaceholderChunk("object-type");
6473 Builder.AddTextChunk(" **");
6474 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6475 Builder.AddTextChunk("buffer");
6476 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6477 Builder.AddTypedTextChunk("range:");
6478 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6479 Builder.AddTextChunk("NSRange");
6480 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6481 Builder.AddTextChunk("inRange");
6482 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6483 CXCursor_ObjCInstanceMethodDecl));
6484 }
6485 }
6486
6487 // Mutable indexed accessors
6488
6489 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6490 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006491 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006492 IdentifierInfo *SelectorIds[2] = {
6493 &Context.Idents.get("insertObject"),
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006494 &Context.Idents.get(SelectorName)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006495 };
6496
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006497 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006498 if (ReturnType.isNull()) {
6499 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6500 Builder.AddTextChunk("void");
6501 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6502 }
6503
6504 Builder.AddTypedTextChunk("insertObject:");
6505 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6506 Builder.AddPlaceholderChunk("object-type");
6507 Builder.AddTextChunk(" *");
6508 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6509 Builder.AddTextChunk("object");
6510 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6511 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6512 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6513 Builder.AddPlaceholderChunk("NSUInteger");
6514 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6515 Builder.AddTextChunk("index");
6516 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6517 CXCursor_ObjCInstanceMethodDecl));
6518 }
6519 }
6520
6521 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6522 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006523 std::string SelectorName = (Twine("insert") + UpperKey).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("atIndexes")
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.AddTextChunk("NSArray *");
6539 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6540 Builder.AddTextChunk("array");
6541 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6542 Builder.AddTypedTextChunk("atIndexes:");
6543 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6544 Builder.AddPlaceholderChunk("NSIndexSet *");
6545 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6546 Builder.AddTextChunk("indexes");
6547 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6548 CXCursor_ObjCInstanceMethodDecl));
6549 }
6550 }
6551
6552 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6553 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006554 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006555 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006556 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006557 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006558 if (ReturnType.isNull()) {
6559 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6560 Builder.AddTextChunk("void");
6561 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6562 }
6563
6564 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6565 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6566 Builder.AddTextChunk("NSUInteger");
6567 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6568 Builder.AddTextChunk("index");
6569 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6570 CXCursor_ObjCInstanceMethodDecl));
6571 }
6572 }
6573
6574 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6575 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006576 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006577 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006578 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006579 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006580 if (ReturnType.isNull()) {
6581 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6582 Builder.AddTextChunk("void");
6583 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6584 }
6585
6586 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6587 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6588 Builder.AddTextChunk("NSIndexSet *");
6589 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6590 Builder.AddTextChunk("indexes");
6591 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6592 CXCursor_ObjCInstanceMethodDecl));
6593 }
6594 }
6595
6596 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6597 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006598 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006599 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006600 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006601 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006602 &Context.Idents.get("withObject")
6603 };
6604
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006605 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006606 if (ReturnType.isNull()) {
6607 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6608 Builder.AddTextChunk("void");
6609 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6610 }
6611
6612 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6613 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6614 Builder.AddPlaceholderChunk("NSUInteger");
6615 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6616 Builder.AddTextChunk("index");
6617 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6618 Builder.AddTypedTextChunk("withObject:");
6619 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6620 Builder.AddTextChunk("id");
6621 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6622 Builder.AddTextChunk("object");
6623 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6624 CXCursor_ObjCInstanceMethodDecl));
6625 }
6626 }
6627
6628 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6629 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006630 std::string SelectorName1
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006631 = (Twine("replace") + UpperKey + "AtIndexes").str();
6632 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006633 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006634 &Context.Idents.get(SelectorName1),
6635 &Context.Idents.get(SelectorName2)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006636 };
6637
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006638 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006639 if (ReturnType.isNull()) {
6640 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6641 Builder.AddTextChunk("void");
6642 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6643 }
6644
6645 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6646 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6647 Builder.AddPlaceholderChunk("NSIndexSet *");
6648 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6649 Builder.AddTextChunk("indexes");
6650 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6651 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6652 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6653 Builder.AddTextChunk("NSArray *");
6654 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6655 Builder.AddTextChunk("array");
6656 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6657 CXCursor_ObjCInstanceMethodDecl));
6658 }
6659 }
6660
6661 // Unordered getters
6662 // - (NSEnumerator *)enumeratorOfKey
6663 if (IsInstanceMethod &&
6664 (ReturnType.isNull() ||
6665 (ReturnType->isObjCObjectPointerType() &&
6666 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6667 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6668 ->getName() == "NSEnumerator"))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006669 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006670 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006671 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006672 if (ReturnType.isNull()) {
6673 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6674 Builder.AddTextChunk("NSEnumerator *");
6675 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6676 }
6677
6678 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6679 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6680 CXCursor_ObjCInstanceMethodDecl));
6681 }
6682 }
6683
6684 // - (type *)memberOfKey:(type *)object
6685 if (IsInstanceMethod &&
6686 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006687 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006688 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006689 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006690 if (ReturnType.isNull()) {
6691 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6692 Builder.AddPlaceholderChunk("object-type");
6693 Builder.AddTextChunk(" *");
6694 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6695 }
6696
6697 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6698 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6699 if (ReturnType.isNull()) {
6700 Builder.AddPlaceholderChunk("object-type");
6701 Builder.AddTextChunk(" *");
6702 } else {
6703 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006704 Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006705 Builder.getAllocator()));
6706 }
6707 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6708 Builder.AddTextChunk("object");
6709 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6710 CXCursor_ObjCInstanceMethodDecl));
6711 }
6712 }
6713
6714 // Mutable unordered accessors
6715 // - (void)addKeyObject:(type *)object
6716 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006717 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006718 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006719 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006720 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006721 if (ReturnType.isNull()) {
6722 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6723 Builder.AddTextChunk("void");
6724 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6725 }
6726
6727 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6728 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6729 Builder.AddPlaceholderChunk("object-type");
6730 Builder.AddTextChunk(" *");
6731 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6732 Builder.AddTextChunk("object");
6733 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6734 CXCursor_ObjCInstanceMethodDecl));
6735 }
6736 }
6737
6738 // - (void)addKey:(NSSet *)objects
6739 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006740 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006741 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006742 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006743 if (ReturnType.isNull()) {
6744 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6745 Builder.AddTextChunk("void");
6746 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6747 }
6748
6749 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6750 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6751 Builder.AddTextChunk("NSSet *");
6752 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6753 Builder.AddTextChunk("objects");
6754 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6755 CXCursor_ObjCInstanceMethodDecl));
6756 }
6757 }
6758
6759 // - (void)removeKeyObject:(type *)object
6760 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006761 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006762 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006763 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006764 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006765 if (ReturnType.isNull()) {
6766 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6767 Builder.AddTextChunk("void");
6768 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6769 }
6770
6771 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6772 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6773 Builder.AddPlaceholderChunk("object-type");
6774 Builder.AddTextChunk(" *");
6775 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6776 Builder.AddTextChunk("object");
6777 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6778 CXCursor_ObjCInstanceMethodDecl));
6779 }
6780 }
6781
6782 // - (void)removeKey:(NSSet *)objects
6783 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006784 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006785 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006786 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006787 if (ReturnType.isNull()) {
6788 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6789 Builder.AddTextChunk("void");
6790 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6791 }
6792
6793 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6794 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6795 Builder.AddTextChunk("NSSet *");
6796 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6797 Builder.AddTextChunk("objects");
6798 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6799 CXCursor_ObjCInstanceMethodDecl));
6800 }
6801 }
6802
6803 // - (void)intersectKey:(NSSet *)objects
6804 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006805 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006806 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006807 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006808 if (ReturnType.isNull()) {
6809 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6810 Builder.AddTextChunk("void");
6811 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6812 }
6813
6814 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6815 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6816 Builder.AddTextChunk("NSSet *");
6817 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6818 Builder.AddTextChunk("objects");
6819 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6820 CXCursor_ObjCInstanceMethodDecl));
6821 }
6822 }
6823
6824 // Key-Value Observing
6825 // + (NSSet *)keyPathsForValuesAffectingKey
6826 if (!IsInstanceMethod &&
6827 (ReturnType.isNull() ||
6828 (ReturnType->isObjCObjectPointerType() &&
6829 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6830 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6831 ->getName() == "NSSet"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006832 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006833 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006834 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006835 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006836 if (ReturnType.isNull()) {
6837 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6838 Builder.AddTextChunk("NSSet *");
6839 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6840 }
6841
6842 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6843 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor857bcda2011-06-02 04:02:27 +00006844 CXCursor_ObjCClassMethodDecl));
6845 }
6846 }
6847
6848 // + (BOOL)automaticallyNotifiesObserversForKey
6849 if (!IsInstanceMethod &&
6850 (ReturnType.isNull() ||
6851 ReturnType->isIntegerType() ||
6852 ReturnType->isBooleanType())) {
6853 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006854 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor857bcda2011-06-02 04:02:27 +00006855 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6856 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6857 if (ReturnType.isNull()) {
6858 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6859 Builder.AddTextChunk("BOOL");
6860 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6861 }
6862
6863 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6864 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6865 CXCursor_ObjCClassMethodDecl));
Douglas Gregor669a25a2011-02-17 00:22:45 +00006866 }
6867 }
6868}
6869
Douglas Gregor636a61e2010-04-07 00:21:17 +00006870void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6871 bool IsInstanceMethod,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006872 ParsedType ReturnTy) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006873 // Determine the return type of the method we're declaring, if
6874 // provided.
6875 QualType ReturnType = GetTypeFromParser(ReturnTy);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006876 Decl *IDecl = 0;
6877 if (CurContext->isObjCContainer()) {
6878 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6879 IDecl = cast<Decl>(OCD);
6880 }
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006881 // Determine where we should start searching for methods.
6882 ObjCContainerDecl *SearchDecl = 0;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006883 bool IsInImplementation = false;
John McCall48871652010-08-21 09:40:31 +00006884 if (Decl *D = IDecl) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006885 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6886 SearchDecl = Impl->getClassInterface();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006887 IsInImplementation = true;
6888 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006889 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006890 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006891 IsInImplementation = true;
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006892 } else
Douglas Gregor636a61e2010-04-07 00:21:17 +00006893 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006894 }
6895
6896 if (!SearchDecl && S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00006897 if (DeclContext *DC = S->getEntity())
Douglas Gregor636a61e2010-04-07 00:21:17 +00006898 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006899 }
6900
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006901 if (!SearchDecl) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006902 HandleCodeCompleteResults(this, CodeCompleter,
6903 CodeCompletionContext::CCC_Other,
6904 0, 0);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006905 return;
6906 }
6907
6908 // Find all of the methods that we could declare/implement here.
6909 KnownMethodsMap KnownMethods;
6910 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006911 ReturnType, KnownMethods);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006912
Douglas Gregor636a61e2010-04-07 00:21:17 +00006913 // Add declarations or definitions for each of the known methods.
John McCall276321a2010-08-25 06:19:51 +00006914 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006915 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006916 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006917 CodeCompletionContext::CCC_Other);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006918 Results.EnterNewScope();
Douglas Gregor75acd922011-09-27 23:30:47 +00006919 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006920 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6921 MEnd = KnownMethods.end();
6922 M != MEnd; ++M) {
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006923 ObjCMethodDecl *Method = M->second.getPointer();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006924 CodeCompletionBuilder Builder(Results.getAllocator(),
6925 Results.getCodeCompletionTUInfo());
Douglas Gregor636a61e2010-04-07 00:21:17 +00006926
6927 // If the result type was not already provided, add it to the
6928 // pattern as (type).
Douglas Gregor669a25a2011-02-17 00:22:45 +00006929 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006930 AddObjCPassingTypeChunk(Method->getResultType(),
6931 Method->getObjCDeclQualifier(),
6932 Context, Policy,
6933 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006934
6935 Selector Sel = Method->getSelector();
6936
6937 // Add the first part of the selector to the pattern.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006938 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006939 Sel.getNameForSlot(0)));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006940
6941 // Add parameters to the pattern.
6942 unsigned I = 0;
6943 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6944 PEnd = Method->param_end();
6945 P != PEnd; (void)++P, ++I) {
6946 // Add the part of the selector name.
6947 if (I == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006948 Builder.AddTypedTextChunk(":");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006949 else if (I < Sel.getNumArgs()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006950 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6951 Builder.AddTypedTextChunk(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006952 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006953 } else
6954 break;
6955
6956 // Add the parameter type.
Douglas Gregor29979142012-04-10 18:35:07 +00006957 AddObjCPassingTypeChunk((*P)->getOriginalType(),
6958 (*P)->getObjCDeclQualifier(),
6959 Context, Policy,
Douglas Gregor75acd922011-09-27 23:30:47 +00006960 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006961
6962 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006963 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006964 }
6965
6966 if (Method->isVariadic()) {
6967 if (Method->param_size() > 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006968 Builder.AddChunk(CodeCompletionString::CK_Comma);
6969 Builder.AddTextChunk("...");
Douglas Gregor400f5972010-08-31 05:13:43 +00006970 }
Douglas Gregor636a61e2010-04-07 00:21:17 +00006971
Douglas Gregord37c59d2010-05-28 00:57:46 +00006972 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006973 // We will be defining the method here, so add a compound statement.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006974 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6975 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6976 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006977 if (!Method->getResultType()->isVoidType()) {
6978 // If the result type is not void, add a return clause.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006979 Builder.AddTextChunk("return");
6980 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6981 Builder.AddPlaceholderChunk("expression");
6982 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006983 } else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006984 Builder.AddPlaceholderChunk("statements");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006985
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006986 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6987 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006988 }
6989
Douglas Gregor416b5752010-08-25 01:08:01 +00006990 unsigned Priority = CCP_CodePattern;
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006991 if (!M->second.getInt())
Douglas Gregor416b5752010-08-25 01:08:01 +00006992 Priority += CCD_InBaseClass;
6993
Douglas Gregor78254c82012-03-27 23:34:16 +00006994 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006995 }
6996
Douglas Gregor669a25a2011-02-17 00:22:45 +00006997 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
6998 // the properties in this class and its categories.
David Blaikiebbafb8a2012-03-11 07:00:24 +00006999 if (Context.getLangOpts().ObjC2) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00007000 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor669a25a2011-02-17 00:22:45 +00007001 Containers.push_back(SearchDecl);
7002
Douglas Gregord4a8ced2011-05-04 23:50:46 +00007003 VisitedSelectorSet KnownSelectors;
7004 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
7005 MEnd = KnownMethods.end();
7006 M != MEnd; ++M)
7007 KnownSelectors.insert(M->first);
7008
7009
Douglas Gregor669a25a2011-02-17 00:22:45 +00007010 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
7011 if (!IFace)
7012 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
7013 IFace = Category->getClassInterface();
7014
7015 if (IFace) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00007016 for (ObjCInterfaceDecl::visible_categories_iterator
7017 Cat = IFace->visible_categories_begin(),
7018 CatEnd = IFace->visible_categories_end();
7019 Cat != CatEnd; ++Cat) {
7020 Containers.push_back(*Cat);
7021 }
Douglas Gregor669a25a2011-02-17 00:22:45 +00007022 }
7023
7024 for (unsigned I = 0, N = Containers.size(); I != N; ++I) {
7025 for (ObjCContainerDecl::prop_iterator P = Containers[I]->prop_begin(),
7026 PEnd = Containers[I]->prop_end();
7027 P != PEnd; ++P) {
David Blaikie40ed2972012-06-06 20:45:41 +00007028 AddObjCKeyValueCompletions(*P, IsInstanceMethod, ReturnType, Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00007029 KnownSelectors, Results);
Douglas Gregor669a25a2011-02-17 00:22:45 +00007030 }
7031 }
7032 }
7033
Douglas Gregor636a61e2010-04-07 00:21:17 +00007034 Results.ExitScope();
7035
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007036 HandleCodeCompleteResults(this, CodeCompleter,
7037 CodeCompletionContext::CCC_Other,
7038 Results.data(),Results.size());
Douglas Gregor636a61e2010-04-07 00:21:17 +00007039}
Douglas Gregor95887f92010-07-08 23:20:03 +00007040
7041void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
7042 bool IsInstanceMethod,
Douglas Gregor45879692010-07-08 23:37:41 +00007043 bool AtParameterName,
John McCallba7bf592010-08-24 05:47:05 +00007044 ParsedType ReturnTy,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007045 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor95887f92010-07-08 23:20:03 +00007046 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00007047 // pool from the AST file.
Douglas Gregor95887f92010-07-08 23:20:03 +00007048 if (ExternalSource) {
7049 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
7050 I != N; ++I) {
7051 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00007052 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor95887f92010-07-08 23:20:03 +00007053 continue;
Sebastian Redl75d8a322010-08-02 23:18:59 +00007054
7055 ReadMethodPool(Sel);
Douglas Gregor95887f92010-07-08 23:20:03 +00007056 }
7057 }
7058
7059 // Build the set of methods we can see.
John McCall276321a2010-08-25 06:19:51 +00007060 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007061 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007062 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007063 CodeCompletionContext::CCC_Other);
Douglas Gregor95887f92010-07-08 23:20:03 +00007064
7065 if (ReturnTy)
7066 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redl75d8a322010-08-02 23:18:59 +00007067
Douglas Gregor95887f92010-07-08 23:20:03 +00007068 Results.EnterNewScope();
Sebastian Redl75d8a322010-08-02 23:18:59 +00007069 for (GlobalMethodPool::iterator M = MethodPool.begin(),
7070 MEnd = MethodPool.end();
7071 M != MEnd; ++M) {
7072 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
7073 &M->second.second;
7074 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00007075 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007076 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor95887f92010-07-08 23:20:03 +00007077 continue;
7078
Douglas Gregor45879692010-07-08 23:37:41 +00007079 if (AtParameterName) {
7080 // Suggest parameter names we've seen before.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007081 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor45879692010-07-08 23:37:41 +00007082 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
7083 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
7084 if (Param->getIdentifier()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007085 CodeCompletionBuilder Builder(Results.getAllocator(),
7086 Results.getCodeCompletionTUInfo());
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007087 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007088 Param->getIdentifier()->getName()));
7089 Results.AddResult(Builder.TakeString());
Douglas Gregor45879692010-07-08 23:37:41 +00007090 }
7091 }
7092
7093 continue;
7094 }
7095
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00007096 Result R(MethList->Method, Results.getBasePriority(MethList->Method), 0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007097 R.StartParameter = SelIdents.size();
Douglas Gregor95887f92010-07-08 23:20:03 +00007098 R.AllParametersAreInformative = false;
7099 R.DeclaringEntity = true;
7100 Results.MaybeAddResult(R, CurContext);
7101 }
7102 }
7103
7104 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007105 HandleCodeCompleteResults(this, CodeCompleter,
7106 CodeCompletionContext::CCC_Other,
7107 Results.data(),Results.size());
Douglas Gregor95887f92010-07-08 23:20:03 +00007108}
Douglas Gregorb14904c2010-08-13 22:48:40 +00007109
Douglas Gregorec00a262010-08-24 22:20:20 +00007110void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007111 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007112 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007113 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007114 Results.EnterNewScope();
7115
7116 // #if <condition>
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007117 CodeCompletionBuilder Builder(Results.getAllocator(),
7118 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007119 Builder.AddTypedTextChunk("if");
7120 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7121 Builder.AddPlaceholderChunk("condition");
7122 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007123
7124 // #ifdef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007125 Builder.AddTypedTextChunk("ifdef");
7126 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7127 Builder.AddPlaceholderChunk("macro");
7128 Results.AddResult(Builder.TakeString());
7129
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007130 // #ifndef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007131 Builder.AddTypedTextChunk("ifndef");
7132 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7133 Builder.AddPlaceholderChunk("macro");
7134 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007135
7136 if (InConditional) {
7137 // #elif <condition>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007138 Builder.AddTypedTextChunk("elif");
7139 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7140 Builder.AddPlaceholderChunk("condition");
7141 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007142
7143 // #else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007144 Builder.AddTypedTextChunk("else");
7145 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007146
7147 // #endif
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007148 Builder.AddTypedTextChunk("endif");
7149 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007150 }
7151
7152 // #include "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007153 Builder.AddTypedTextChunk("include");
7154 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7155 Builder.AddTextChunk("\"");
7156 Builder.AddPlaceholderChunk("header");
7157 Builder.AddTextChunk("\"");
7158 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007159
7160 // #include <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007161 Builder.AddTypedTextChunk("include");
7162 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7163 Builder.AddTextChunk("<");
7164 Builder.AddPlaceholderChunk("header");
7165 Builder.AddTextChunk(">");
7166 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007167
7168 // #define <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007169 Builder.AddTypedTextChunk("define");
7170 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7171 Builder.AddPlaceholderChunk("macro");
7172 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007173
7174 // #define <macro>(<args>)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007175 Builder.AddTypedTextChunk("define");
7176 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7177 Builder.AddPlaceholderChunk("macro");
7178 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7179 Builder.AddPlaceholderChunk("args");
7180 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7181 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007182
7183 // #undef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007184 Builder.AddTypedTextChunk("undef");
7185 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7186 Builder.AddPlaceholderChunk("macro");
7187 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007188
7189 // #line <number>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007190 Builder.AddTypedTextChunk("line");
7191 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7192 Builder.AddPlaceholderChunk("number");
7193 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007194
7195 // #line <number> "filename"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007196 Builder.AddTypedTextChunk("line");
7197 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7198 Builder.AddPlaceholderChunk("number");
7199 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7200 Builder.AddTextChunk("\"");
7201 Builder.AddPlaceholderChunk("filename");
7202 Builder.AddTextChunk("\"");
7203 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007204
7205 // #error <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007206 Builder.AddTypedTextChunk("error");
7207 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7208 Builder.AddPlaceholderChunk("message");
7209 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007210
7211 // #pragma <arguments>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007212 Builder.AddTypedTextChunk("pragma");
7213 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7214 Builder.AddPlaceholderChunk("arguments");
7215 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007216
David Blaikiebbafb8a2012-03-11 07:00:24 +00007217 if (getLangOpts().ObjC1) {
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007218 // #import "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007219 Builder.AddTypedTextChunk("import");
7220 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7221 Builder.AddTextChunk("\"");
7222 Builder.AddPlaceholderChunk("header");
7223 Builder.AddTextChunk("\"");
7224 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007225
7226 // #import <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007227 Builder.AddTypedTextChunk("import");
7228 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7229 Builder.AddTextChunk("<");
7230 Builder.AddPlaceholderChunk("header");
7231 Builder.AddTextChunk(">");
7232 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007233 }
7234
7235 // #include_next "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007236 Builder.AddTypedTextChunk("include_next");
7237 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7238 Builder.AddTextChunk("\"");
7239 Builder.AddPlaceholderChunk("header");
7240 Builder.AddTextChunk("\"");
7241 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007242
7243 // #include_next <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007244 Builder.AddTypedTextChunk("include_next");
7245 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7246 Builder.AddTextChunk("<");
7247 Builder.AddPlaceholderChunk("header");
7248 Builder.AddTextChunk(">");
7249 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007250
7251 // #warning <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007252 Builder.AddTypedTextChunk("warning");
7253 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7254 Builder.AddPlaceholderChunk("message");
7255 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007256
7257 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7258 // completions for them. And __include_macros is a Clang-internal extension
7259 // that we don't want to encourage anyone to use.
7260
7261 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7262 Results.ExitScope();
7263
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007264 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0de55ce2010-08-25 18:41:16 +00007265 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007266 Results.data(), Results.size());
7267}
7268
7269void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorec00a262010-08-24 22:20:20 +00007270 CodeCompleteOrdinaryName(S,
John McCallfaf5fb42010-08-26 23:41:50 +00007271 S->getFnParent()? Sema::PCC_RecoveryInFunction
7272 : Sema::PCC_Namespace);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007273}
7274
Douglas Gregorec00a262010-08-24 22:20:20 +00007275void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007276 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007277 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007278 IsDefinition? CodeCompletionContext::CCC_MacroName
7279 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor12785102010-08-24 20:21:13 +00007280 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7281 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007282 CodeCompletionBuilder Builder(Results.getAllocator(),
7283 Results.getCodeCompletionTUInfo());
Douglas Gregor12785102010-08-24 20:21:13 +00007284 Results.EnterNewScope();
7285 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7286 MEnd = PP.macro_end();
7287 M != MEnd; ++M) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007288 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007289 M->first->getName()));
Argyrios Kyrtzidis5c8b1cd2012-09-27 00:24:09 +00007290 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7291 CCP_CodePattern,
7292 CXCursor_MacroDefinition));
Douglas Gregor12785102010-08-24 20:21:13 +00007293 }
7294 Results.ExitScope();
7295 } else if (IsDefinition) {
7296 // FIXME: Can we detect when the user just wrote an include guard above?
7297 }
7298
Douglas Gregor0ac41382010-09-23 23:01:17 +00007299 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor12785102010-08-24 20:21:13 +00007300 Results.data(), Results.size());
7301}
7302
Douglas Gregorec00a262010-08-24 22:20:20 +00007303void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007304 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007305 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007306 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorec00a262010-08-24 22:20:20 +00007307
7308 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007309 AddMacroResults(PP, Results, true);
Douglas Gregorec00a262010-08-24 22:20:20 +00007310
7311 // defined (<macro>)
7312 Results.EnterNewScope();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007313 CodeCompletionBuilder Builder(Results.getAllocator(),
7314 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007315 Builder.AddTypedTextChunk("defined");
7316 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7317 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7318 Builder.AddPlaceholderChunk("macro");
7319 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7320 Results.AddResult(Builder.TakeString());
Douglas Gregorec00a262010-08-24 22:20:20 +00007321 Results.ExitScope();
7322
7323 HandleCodeCompleteResults(this, CodeCompleter,
7324 CodeCompletionContext::CCC_PreprocessorExpression,
7325 Results.data(), Results.size());
7326}
7327
7328void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7329 IdentifierInfo *Macro,
7330 MacroInfo *MacroInfo,
7331 unsigned Argument) {
7332 // FIXME: In the future, we could provide "overload" results, much like we
7333 // do for function calls.
7334
Argyrios Kyrtzidis75f6cd22011-08-18 19:41:28 +00007335 // Now just ignore this. There will be another code-completion callback
7336 // for the expanded tokens.
Douglas Gregorec00a262010-08-24 22:20:20 +00007337}
7338
Douglas Gregor11583702010-08-25 17:04:25 +00007339void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor11583702010-08-25 17:04:25 +00007340 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorea736372010-08-25 17:10:00 +00007341 CodeCompletionContext::CCC_NaturalLanguage,
Douglas Gregor11583702010-08-25 17:04:25 +00007342 0, 0);
7343}
7344
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007345void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007346 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00007347 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007348 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7349 CodeCompletionContext::CCC_Recovery);
Douglas Gregor39982192010-08-15 06:18:01 +00007350 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7351 CodeCompletionDeclConsumer Consumer(Builder,
7352 Context.getTranslationUnitDecl());
7353 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7354 Consumer);
7355 }
Douglas Gregorb14904c2010-08-13 22:48:40 +00007356
7357 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007358 AddMacroResults(PP, Builder, true);
Douglas Gregorb14904c2010-08-13 22:48:40 +00007359
7360 Results.clear();
7361 Results.insert(Results.end(),
7362 Builder.data(), Builder.data() + Builder.size());
7363}