blob: 00232f64b7715dbea3f9b2398ebdaeacad7c817a [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;
Alp Tokera2794f92014-01-22 07:29:52 +0000663 if (const FunctionDecl *Function = ND->getAsFunction())
Douglas Gregor603d81b2010-07-13 08:18:22 +0000664 T = Function->getCallResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000665 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregor603d81b2010-07-13 08:18:22 +0000666 T = Method->getSendResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000667 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000668 T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext()));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000669 else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000670 T = Property->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000671 else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000672 T = Value->getType();
673 else
674 return QualType();
Douglas Gregoraf670a82011-04-14 20:33:34 +0000675
676 // Dig through references, function pointers, and block pointers to
677 // get down to the likely type of an expression when the entity is
678 // used.
679 do {
680 if (const ReferenceType *Ref = T->getAs<ReferenceType>()) {
681 T = Ref->getPointeeType();
682 continue;
683 }
684
685 if (const PointerType *Pointer = T->getAs<PointerType>()) {
686 if (Pointer->getPointeeType()->isFunctionType()) {
687 T = Pointer->getPointeeType();
688 continue;
689 }
690
691 break;
692 }
693
694 if (const BlockPointerType *Block = T->getAs<BlockPointerType>()) {
695 T = Block->getPointeeType();
696 continue;
697 }
698
699 if (const FunctionType *Function = T->getAs<FunctionType>()) {
Alp Toker314cc812014-01-25 16:55:45 +0000700 T = Function->getReturnType();
Douglas Gregoraf670a82011-04-14 20:33:34 +0000701 continue;
702 }
703
704 break;
705 } while (true);
706
707 return T;
Douglas Gregor95887f92010-07-08 23:20:03 +0000708}
709
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000710unsigned ResultBuilder::getBasePriority(const NamedDecl *ND) {
711 if (!ND)
712 return CCP_Unlikely;
713
714 // Context-based decisions.
Richard Smith541b38b2013-09-20 01:15:31 +0000715 const DeclContext *LexicalDC = ND->getLexicalDeclContext();
716 if (LexicalDC->isFunctionOrMethod()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000717 // _cmd is relatively rare
718 if (const ImplicitParamDecl *ImplicitParam =
719 dyn_cast<ImplicitParamDecl>(ND))
720 if (ImplicitParam->getIdentifier() &&
721 ImplicitParam->getIdentifier()->isStr("_cmd"))
722 return CCP_ObjC_cmd;
723
724 return CCP_LocalDeclaration;
725 }
Richard Smith541b38b2013-09-20 01:15:31 +0000726
727 const DeclContext *DC = ND->getDeclContext()->getRedeclContext();
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000728 if (DC->isRecord() || isa<ObjCContainerDecl>(DC))
729 return CCP_MemberDeclaration;
730
731 // Content-based decisions.
732 if (isa<EnumConstantDecl>(ND))
733 return CCP_Constant;
734
Douglas Gregor52e0de42013-01-31 05:03:46 +0000735 // Use CCP_Type for type declarations unless we're in a statement, Objective-C
736 // message receiver, or parenthesized expression context. There, it's as
737 // likely that the user will want to write a type as other declarations.
738 if ((isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND)) &&
739 !(CompletionContext.getKind() == CodeCompletionContext::CCC_Statement ||
740 CompletionContext.getKind()
741 == CodeCompletionContext::CCC_ObjCMessageReceiver ||
742 CompletionContext.getKind()
743 == CodeCompletionContext::CCC_ParenthesizedExpression))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000744 return CCP_Type;
745
746 return CCP_Declaration;
747}
748
Douglas Gregor50832e02010-09-20 22:39:41 +0000749void ResultBuilder::AdjustResultPriorityForDecl(Result &R) {
750 // If this is an Objective-C method declaration whose selector matches our
751 // preferred selector, give it a priority boost.
752 if (!PreferredSelector.isNull())
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000753 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(R.Declaration))
Douglas Gregor50832e02010-09-20 22:39:41 +0000754 if (PreferredSelector == Method->getSelector())
755 R.Priority += CCD_SelectorMatch;
Douglas Gregor5fb901d2010-09-20 23:11:55 +0000756
Douglas Gregor50832e02010-09-20 22:39:41 +0000757 // If we have a preferred type, adjust the priority for results with exactly-
758 // matching or nearly-matching types.
759 if (!PreferredType.isNull()) {
760 QualType T = getDeclUsageType(SemaRef.Context, R.Declaration);
761 if (!T.isNull()) {
762 CanQualType TC = SemaRef.Context.getCanonicalType(T);
763 // Check for exactly-matching types (modulo qualifiers).
764 if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC))
765 R.Priority /= CCF_ExactTypeMatch;
766 // Check for nearly-matching types, based on classification of each.
767 else if ((getSimplifiedTypeClass(PreferredType)
Douglas Gregor95887f92010-07-08 23:20:03 +0000768 == getSimplifiedTypeClass(TC)) &&
Douglas Gregor50832e02010-09-20 22:39:41 +0000769 !(PreferredType->isEnumeralType() && TC->isEnumeralType()))
770 R.Priority /= CCF_SimilarTypeMatch;
771 }
772 }
Douglas Gregor95887f92010-07-08 23:20:03 +0000773}
774
Douglas Gregor0212fd72010-09-21 16:06:22 +0000775void ResultBuilder::MaybeAddConstructorResults(Result R) {
David Blaikiebbafb8a2012-03-11 07:00:24 +0000776 if (!SemaRef.getLangOpts().CPlusPlus || !R.Declaration ||
Douglas Gregor0212fd72010-09-21 16:06:22 +0000777 !CompletionContext.wantConstructorResults())
778 return;
779
780 ASTContext &Context = SemaRef.Context;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000781 const NamedDecl *D = R.Declaration;
782 const CXXRecordDecl *Record = 0;
783 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D))
Douglas Gregor0212fd72010-09-21 16:06:22 +0000784 Record = ClassTemplate->getTemplatedDecl();
785 else if ((Record = dyn_cast<CXXRecordDecl>(D))) {
786 // Skip specializations and partial specializations.
787 if (isa<ClassTemplateSpecializationDecl>(Record))
788 return;
789 } else {
790 // There are no constructors here.
791 return;
792 }
793
794 Record = Record->getDefinition();
795 if (!Record)
796 return;
797
798
799 QualType RecordTy = Context.getTypeDeclType(Record);
800 DeclarationName ConstructorName
801 = Context.DeclarationNames.getCXXConstructorName(
802 Context.getCanonicalType(RecordTy));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000803 DeclContext::lookup_const_result Ctors = Record->lookup(ConstructorName);
804 for (DeclContext::lookup_const_iterator I = Ctors.begin(),
805 E = Ctors.end();
806 I != E; ++I) {
David Blaikieff7d47a2012-12-19 00:45:41 +0000807 R.Declaration = *I;
Douglas Gregor0212fd72010-09-21 16:06:22 +0000808 R.CursorKind = getCursorKindForDecl(R.Declaration);
809 Results.push_back(R);
810 }
811}
812
Douglas Gregor7c208612010-01-14 00:20:49 +0000813void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) {
814 assert(!ShadowMaps.empty() && "Must enter into a results scope");
815
816 if (R.Kind != Result::RK_Declaration) {
817 // For non-declaration results, just add the result.
818 Results.push_back(R);
819 return;
820 }
821
822 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000823 if (const UsingShadowDecl *Using =
824 dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000825 MaybeAddResult(Result(Using->getTargetDecl(),
826 getBasePriority(Using->getTargetDecl()),
827 R.Qualifier),
828 CurContext);
Douglas Gregor7c208612010-01-14 00:20:49 +0000829 return;
830 }
831
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000832 const Decl *CanonDecl = R.Declaration->getCanonicalDecl();
Douglas Gregor7c208612010-01-14 00:20:49 +0000833 unsigned IDNS = CanonDecl->getIdentifierNamespace();
834
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000835 bool AsNestedNameSpecifier = false;
836 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregor7c208612010-01-14 00:20:49 +0000837 return;
838
Douglas Gregor0212fd72010-09-21 16:06:22 +0000839 // C++ constructors are never found by name lookup.
840 if (isa<CXXConstructorDecl>(R.Declaration))
841 return;
842
Douglas Gregor3545ff42009-09-21 16:56:56 +0000843 ShadowMap &SMap = ShadowMaps.back();
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000844 ShadowMapEntry::iterator I, IEnd;
845 ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName());
846 if (NamePos != SMap.end()) {
847 I = NamePos->second.begin();
848 IEnd = NamePos->second.end();
849 }
850
851 for (; I != IEnd; ++I) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000852 const NamedDecl *ND = I->first;
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000853 unsigned Index = I->second;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000854 if (ND->getCanonicalDecl() == CanonDecl) {
855 // This is a redeclaration. Always pick the newer declaration.
Douglas Gregor3545ff42009-09-21 16:56:56 +0000856 Results[Index].Declaration = R.Declaration;
857
Douglas Gregor3545ff42009-09-21 16:56:56 +0000858 // We're done.
859 return;
860 }
861 }
862
863 // This is a new declaration in this scope. However, check whether this
864 // declaration name is hidden by a similarly-named declaration in an outer
865 // scope.
866 std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end();
867 --SMEnd;
868 for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000869 ShadowMapEntry::iterator I, IEnd;
870 ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName());
871 if (NamePos != SM->end()) {
872 I = NamePos->second.begin();
873 IEnd = NamePos->second.end();
874 }
875 for (; I != IEnd; ++I) {
Douglas Gregor3545ff42009-09-21 16:56:56 +0000876 // A tag declaration does not hide a non-tag declaration.
John McCalle87beb22010-04-23 18:46:30 +0000877 if (I->first->hasTagIdentifierNamespace() &&
Richard Smith541b38b2013-09-20 01:15:31 +0000878 (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary |
879 Decl::IDNS_LocalExtern | Decl::IDNS_ObjCProtocol)))
Douglas Gregor3545ff42009-09-21 16:56:56 +0000880 continue;
881
882 // Protocols are in distinct namespaces from everything else.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000883 if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000884 || (IDNS & Decl::IDNS_ObjCProtocol)) &&
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000885 I->first->getIdentifierNamespace() != IDNS)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000886 continue;
887
888 // The newly-added result is hidden by an entry in the shadow map.
Douglas Gregore0717ab2010-01-14 00:41:07 +0000889 if (CheckHiddenResult(R, CurContext, I->first))
Douglas Gregor3545ff42009-09-21 16:56:56 +0000890 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000891
892 break;
893 }
894 }
895
896 // Make sure that any given declaration only shows up in the result set once.
897 if (!AllDeclsFound.insert(CanonDecl))
898 return;
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000899
Douglas Gregore412a5a2009-09-23 22:26:46 +0000900 // If the filter is for nested-name-specifiers, then this result starts a
901 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000902 if (AsNestedNameSpecifier) {
Douglas Gregore412a5a2009-09-23 22:26:46 +0000903 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000904 R.Priority = CCP_NestedNameSpecifier;
Douglas Gregor50832e02010-09-20 22:39:41 +0000905 } else
906 AdjustResultPriorityForDecl(R);
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000907
Douglas Gregor5bf52692009-09-22 23:15:58 +0000908 // If this result is supposed to have an informative qualifier, add one.
Douglas Gregore412a5a2009-09-23 22:26:46 +0000909 if (R.QualifierIsInformative && !R.Qualifier &&
910 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000911 const DeclContext *Ctx = R.Declaration->getDeclContext();
912 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregor5bf52692009-09-22 23:15:58 +0000913 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000914 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregor5bf52692009-09-22 23:15:58 +0000915 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
916 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
917 else
918 R.QualifierIsInformative = false;
919 }
Douglas Gregore412a5a2009-09-23 22:26:46 +0000920
Douglas Gregor3545ff42009-09-21 16:56:56 +0000921 // Insert this result into the set of results and into the current shadow
922 // map.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000923 SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size());
Douglas Gregor3545ff42009-09-21 16:56:56 +0000924 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +0000925
926 if (!AsNestedNameSpecifier)
927 MaybeAddConstructorResults(R);
Douglas Gregor3545ff42009-09-21 16:56:56 +0000928}
929
Douglas Gregorc580c522010-01-14 01:09:38 +0000930void ResultBuilder::AddResult(Result R, DeclContext *CurContext,
Douglas Gregor09bbc652010-01-14 15:47:35 +0000931 NamedDecl *Hiding, bool InBaseClass = false) {
Douglas Gregor78a21012010-01-14 16:01:26 +0000932 if (R.Kind != Result::RK_Declaration) {
933 // For non-declaration results, just add the result.
934 Results.push_back(R);
935 return;
936 }
937
Douglas Gregorc580c522010-01-14 01:09:38 +0000938 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000939 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000940 AddResult(Result(Using->getTargetDecl(),
941 getBasePriority(Using->getTargetDecl()),
942 R.Qualifier),
943 CurContext, Hiding);
Douglas Gregorc580c522010-01-14 01:09:38 +0000944 return;
945 }
946
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000947 bool AsNestedNameSpecifier = false;
948 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregorc580c522010-01-14 01:09:38 +0000949 return;
950
Douglas Gregor0212fd72010-09-21 16:06:22 +0000951 // C++ constructors are never found by name lookup.
952 if (isa<CXXConstructorDecl>(R.Declaration))
953 return;
954
Douglas Gregorc580c522010-01-14 01:09:38 +0000955 if (Hiding && CheckHiddenResult(R, CurContext, Hiding))
956 return;
Nick Lewyckyc3921482012-04-03 21:44:08 +0000957
Douglas Gregorc580c522010-01-14 01:09:38 +0000958 // Make sure that any given declaration only shows up in the result set once.
959 if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl()))
960 return;
961
962 // If the filter is for nested-name-specifiers, then this result starts a
963 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000964 if (AsNestedNameSpecifier) {
Douglas Gregorc580c522010-01-14 01:09:38 +0000965 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000966 R.Priority = CCP_NestedNameSpecifier;
967 }
Douglas Gregor09bbc652010-01-14 15:47:35 +0000968 else if (Filter == &ResultBuilder::IsMember && !R.Qualifier && InBaseClass &&
969 isa<CXXRecordDecl>(R.Declaration->getDeclContext()
Sebastian Redl50c68252010-08-31 00:36:30 +0000970 ->getRedeclContext()))
Douglas Gregor09bbc652010-01-14 15:47:35 +0000971 R.QualifierIsInformative = true;
972
Douglas Gregorc580c522010-01-14 01:09:38 +0000973 // If this result is supposed to have an informative qualifier, add one.
974 if (R.QualifierIsInformative && !R.Qualifier &&
975 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000976 const DeclContext *Ctx = R.Declaration->getDeclContext();
977 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregorc580c522010-01-14 01:09:38 +0000978 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000979 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregorc580c522010-01-14 01:09:38 +0000980 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000981 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregorc580c522010-01-14 01:09:38 +0000982 else
983 R.QualifierIsInformative = false;
984 }
985
Douglas Gregora2db7932010-05-26 22:00:08 +0000986 // Adjust the priority if this result comes from a base class.
987 if (InBaseClass)
988 R.Priority += CCD_InBaseClass;
989
Douglas Gregor50832e02010-09-20 22:39:41 +0000990 AdjustResultPriorityForDecl(R);
Douglas Gregor7aa6b222010-05-30 01:49:25 +0000991
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000992 if (HasObjectTypeQualifiers)
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000993 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(R.Declaration))
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000994 if (Method->isInstance()) {
995 Qualifiers MethodQuals
996 = Qualifiers::fromCVRMask(Method->getTypeQualifiers());
997 if (ObjectTypeQualifiers == MethodQuals)
998 R.Priority += CCD_ObjectQualifierMatch;
999 else if (ObjectTypeQualifiers - MethodQuals) {
1000 // The method cannot be invoked, because doing so would drop
1001 // qualifiers.
1002 return;
1003 }
1004 }
1005
Douglas Gregorc580c522010-01-14 01:09:38 +00001006 // Insert this result into the set of results.
1007 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +00001008
1009 if (!AsNestedNameSpecifier)
1010 MaybeAddConstructorResults(R);
Douglas Gregorc580c522010-01-14 01:09:38 +00001011}
1012
Douglas Gregor78a21012010-01-14 16:01:26 +00001013void ResultBuilder::AddResult(Result R) {
1014 assert(R.Kind != Result::RK_Declaration &&
1015 "Declaration results need more context");
1016 Results.push_back(R);
1017}
1018
Douglas Gregor3545ff42009-09-21 16:56:56 +00001019/// \brief Enter into a new scope.
1020void ResultBuilder::EnterNewScope() {
1021 ShadowMaps.push_back(ShadowMap());
1022}
1023
1024/// \brief Exit from the current scope.
1025void ResultBuilder::ExitScope() {
Douglas Gregor05e7ca32009-12-06 20:23:50 +00001026 for (ShadowMap::iterator E = ShadowMaps.back().begin(),
1027 EEnd = ShadowMaps.back().end();
1028 E != EEnd;
1029 ++E)
1030 E->second.Destroy();
1031
Douglas Gregor3545ff42009-09-21 16:56:56 +00001032 ShadowMaps.pop_back();
1033}
1034
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001035/// \brief Determines whether this given declaration will be found by
1036/// ordinary name lookup.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001037bool ResultBuilder::IsOrdinaryName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001038 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1039
Richard Smith541b38b2013-09-20 01:15:31 +00001040 // If name lookup finds a local extern declaration, then we are in a
1041 // context where it behaves like an ordinary name.
1042 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001043 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9858ed52010-06-15 20:26:51 +00001044 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001045 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor05fcf842010-11-02 20:36:02 +00001046 if (isa<ObjCIvarDecl>(ND))
1047 return true;
Douglas Gregor05fcf842010-11-02 20:36:02 +00001048 }
1049
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001050 return ND->getIdentifierNamespace() & IDNS;
1051}
1052
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001053/// \brief Determines whether this given declaration will be found by
Douglas Gregor70febae2010-05-28 00:49:12 +00001054/// ordinary name lookup but is not a type name.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001055bool ResultBuilder::IsOrdinaryNonTypeName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001056 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1057 if (isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND))
1058 return false;
1059
Richard Smith541b38b2013-09-20 01:15:31 +00001060 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001061 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9858ed52010-06-15 20:26:51 +00001062 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001063 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor05fcf842010-11-02 20:36:02 +00001064 if (isa<ObjCIvarDecl>(ND))
1065 return true;
Douglas Gregor05fcf842010-11-02 20:36:02 +00001066 }
1067
Douglas Gregor70febae2010-05-28 00:49:12 +00001068 return ND->getIdentifierNamespace() & IDNS;
1069}
1070
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001071bool ResultBuilder::IsIntegralConstantValue(const NamedDecl *ND) const {
Douglas Gregor85b50632010-07-28 21:50:18 +00001072 if (!IsOrdinaryNonTypeName(ND))
1073 return 0;
1074
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001075 if (const ValueDecl *VD = dyn_cast<ValueDecl>(ND->getUnderlyingDecl()))
Douglas Gregor85b50632010-07-28 21:50:18 +00001076 if (VD->getType()->isIntegralOrEnumerationType())
1077 return true;
1078
1079 return false;
1080}
1081
Douglas Gregor70febae2010-05-28 00:49:12 +00001082/// \brief Determines whether this given declaration will be found by
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001083/// ordinary name lookup.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001084bool ResultBuilder::IsOrdinaryNonValueName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001085 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1086
Richard Smith541b38b2013-09-20 01:15:31 +00001087 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001088 if (SemaRef.getLangOpts().CPlusPlus)
John McCalle87beb22010-04-23 18:46:30 +00001089 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001090
1091 return (ND->getIdentifierNamespace() & IDNS) &&
Douglas Gregor70febae2010-05-28 00:49:12 +00001092 !isa<ValueDecl>(ND) && !isa<FunctionTemplateDecl>(ND) &&
1093 !isa<ObjCPropertyDecl>(ND);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001094}
1095
Douglas Gregor3545ff42009-09-21 16:56:56 +00001096/// \brief Determines whether the given declaration is suitable as the
1097/// start of a C++ nested-name-specifier, e.g., a class or namespace.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001098bool ResultBuilder::IsNestedNameSpecifier(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001099 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001100 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001101 ND = ClassTemplate->getTemplatedDecl();
1102
1103 return SemaRef.isAcceptableNestedNameSpecifier(ND);
1104}
1105
1106/// \brief Determines whether the given declaration is an enumeration.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001107bool ResultBuilder::IsEnum(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001108 return isa<EnumDecl>(ND);
1109}
1110
1111/// \brief Determines whether the given declaration is a class or struct.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001112bool ResultBuilder::IsClassOrStruct(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001113 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001114 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001115 ND = ClassTemplate->getTemplatedDecl();
Joao Matosdc86f942012-08-31 18:45:21 +00001116
1117 // For purposes of this check, interfaces match too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001118 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara6150c882010-05-11 21:36:43 +00001119 return RD->getTagKind() == TTK_Class ||
Joao Matosdc86f942012-08-31 18:45:21 +00001120 RD->getTagKind() == TTK_Struct ||
1121 RD->getTagKind() == TTK_Interface;
Douglas Gregor3545ff42009-09-21 16:56:56 +00001122
1123 return false;
1124}
1125
1126/// \brief Determines whether the given declaration is a union.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001127bool ResultBuilder::IsUnion(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001128 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001129 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001130 ND = ClassTemplate->getTemplatedDecl();
1131
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001132 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara6150c882010-05-11 21:36:43 +00001133 return RD->getTagKind() == TTK_Union;
Douglas Gregor3545ff42009-09-21 16:56:56 +00001134
1135 return false;
1136}
1137
1138/// \brief Determines whether the given declaration is a namespace.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001139bool ResultBuilder::IsNamespace(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001140 return isa<NamespaceDecl>(ND);
1141}
1142
1143/// \brief Determines whether the given declaration is a namespace or
1144/// namespace alias.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001145bool ResultBuilder::IsNamespaceOrAlias(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001146 return isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND);
1147}
1148
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001149/// \brief Determines whether the given declaration is a type.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001150bool ResultBuilder::IsType(const NamedDecl *ND) const {
1151 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor99fa2642010-08-24 01:06:58 +00001152 ND = Using->getTargetDecl();
1153
1154 return isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND);
Douglas Gregor3545ff42009-09-21 16:56:56 +00001155}
1156
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001157/// \brief Determines which members of a class should be visible via
1158/// "." or "->". Only value declarations, nested name specifiers, and
1159/// using declarations thereof should show up.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001160bool ResultBuilder::IsMember(const NamedDecl *ND) const {
1161 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001162 ND = Using->getTargetDecl();
1163
Douglas Gregor70788392009-12-11 18:14:22 +00001164 return isa<ValueDecl>(ND) || isa<FunctionTemplateDecl>(ND) ||
1165 isa<ObjCPropertyDecl>(ND);
Douglas Gregore412a5a2009-09-23 22:26:46 +00001166}
1167
Douglas Gregora817a192010-05-27 23:06:34 +00001168static bool isObjCReceiverType(ASTContext &C, QualType T) {
1169 T = C.getCanonicalType(T);
1170 switch (T->getTypeClass()) {
1171 case Type::ObjCObject:
1172 case Type::ObjCInterface:
1173 case Type::ObjCObjectPointer:
1174 return true;
1175
1176 case Type::Builtin:
1177 switch (cast<BuiltinType>(T)->getKind()) {
1178 case BuiltinType::ObjCId:
1179 case BuiltinType::ObjCClass:
1180 case BuiltinType::ObjCSel:
1181 return true;
1182
1183 default:
1184 break;
1185 }
1186 return false;
1187
1188 default:
1189 break;
1190 }
1191
David Blaikiebbafb8a2012-03-11 07:00:24 +00001192 if (!C.getLangOpts().CPlusPlus)
Douglas Gregora817a192010-05-27 23:06:34 +00001193 return false;
1194
1195 // FIXME: We could perform more analysis here to determine whether a
1196 // particular class type has any conversions to Objective-C types. For now,
1197 // just accept all class types.
1198 return T->isDependentType() || T->isRecordType();
1199}
1200
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001201bool ResultBuilder::IsObjCMessageReceiver(const NamedDecl *ND) const {
Douglas Gregora817a192010-05-27 23:06:34 +00001202 QualType T = getDeclUsageType(SemaRef.Context, ND);
1203 if (T.isNull())
1204 return false;
1205
1206 T = SemaRef.Context.getBaseElementType(T);
1207 return isObjCReceiverType(SemaRef.Context, T);
1208}
1209
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001210bool ResultBuilder::IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const {
Douglas Gregord8c61782012-02-15 15:34:24 +00001211 if (IsObjCMessageReceiver(ND))
1212 return true;
1213
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001214 const VarDecl *Var = dyn_cast<VarDecl>(ND);
Douglas Gregord8c61782012-02-15 15:34:24 +00001215 if (!Var)
1216 return false;
1217
1218 return Var->hasLocalStorage() && !Var->hasAttr<BlocksAttr>();
1219}
1220
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001221bool ResultBuilder::IsObjCCollection(const NamedDecl *ND) const {
David Blaikiebbafb8a2012-03-11 07:00:24 +00001222 if ((SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryName(ND)) ||
1223 (!SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryNonTypeName(ND)))
Douglas Gregor68762e72010-08-23 21:17:50 +00001224 return false;
1225
1226 QualType T = getDeclUsageType(SemaRef.Context, ND);
1227 if (T.isNull())
1228 return false;
1229
1230 T = SemaRef.Context.getBaseElementType(T);
1231 return T->isObjCObjectType() || T->isObjCObjectPointerType() ||
1232 T->isObjCIdType() ||
David Blaikiebbafb8a2012-03-11 07:00:24 +00001233 (SemaRef.getLangOpts().CPlusPlus && T->isRecordType());
Douglas Gregor68762e72010-08-23 21:17:50 +00001234}
Douglas Gregora817a192010-05-27 23:06:34 +00001235
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001236bool ResultBuilder::IsImpossibleToSatisfy(const NamedDecl *ND) const {
Douglas Gregor0ac41382010-09-23 23:01:17 +00001237 return false;
1238}
1239
James Dennettf1243872012-06-17 05:33:25 +00001240/// \brief Determines whether the given declaration is an Objective-C
Douglas Gregor2b8162b2010-01-14 16:08:12 +00001241/// instance variable.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001242bool ResultBuilder::IsObjCIvar(const NamedDecl *ND) const {
Douglas Gregor2b8162b2010-01-14 16:08:12 +00001243 return isa<ObjCIvarDecl>(ND);
1244}
1245
Douglas Gregorc580c522010-01-14 01:09:38 +00001246namespace {
1247 /// \brief Visible declaration consumer that adds a code-completion result
1248 /// for each visible declaration.
1249 class CodeCompletionDeclConsumer : public VisibleDeclConsumer {
1250 ResultBuilder &Results;
1251 DeclContext *CurContext;
1252
1253 public:
1254 CodeCompletionDeclConsumer(ResultBuilder &Results, DeclContext *CurContext)
1255 : Results(Results), CurContext(CurContext) { }
1256
Erik Verbruggen2e657ff2011-10-06 07:27:49 +00001257 virtual void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
1258 bool InBaseClass) {
1259 bool Accessible = true;
Douglas Gregor03ba1882011-11-03 16:51:37 +00001260 if (Ctx)
1261 Accessible = Results.getSema().IsSimplyAccessible(ND, Ctx);
1262
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00001263 ResultBuilder::Result Result(ND, Results.getBasePriority(ND), 0, false,
1264 Accessible);
Erik Verbruggen2e657ff2011-10-06 07:27:49 +00001265 Results.AddResult(Result, CurContext, Hiding, InBaseClass);
Douglas Gregorc580c522010-01-14 01:09:38 +00001266 }
1267 };
1268}
1269
Douglas Gregor3545ff42009-09-21 16:56:56 +00001270/// \brief Add type specifiers for the current language as keyword results.
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001271static void AddTypeSpecifierResults(const LangOptions &LangOpts,
Douglas Gregor3545ff42009-09-21 16:56:56 +00001272 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001273 typedef CodeCompletionResult Result;
Douglas Gregora2db7932010-05-26 22:00:08 +00001274 Results.AddResult(Result("short", CCP_Type));
1275 Results.AddResult(Result("long", CCP_Type));
1276 Results.AddResult(Result("signed", CCP_Type));
1277 Results.AddResult(Result("unsigned", CCP_Type));
1278 Results.AddResult(Result("void", CCP_Type));
1279 Results.AddResult(Result("char", CCP_Type));
1280 Results.AddResult(Result("int", CCP_Type));
1281 Results.AddResult(Result("float", CCP_Type));
1282 Results.AddResult(Result("double", CCP_Type));
1283 Results.AddResult(Result("enum", CCP_Type));
1284 Results.AddResult(Result("struct", CCP_Type));
1285 Results.AddResult(Result("union", CCP_Type));
1286 Results.AddResult(Result("const", CCP_Type));
1287 Results.AddResult(Result("volatile", CCP_Type));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001288
Douglas Gregor3545ff42009-09-21 16:56:56 +00001289 if (LangOpts.C99) {
1290 // C99-specific
Douglas Gregora2db7932010-05-26 22:00:08 +00001291 Results.AddResult(Result("_Complex", CCP_Type));
1292 Results.AddResult(Result("_Imaginary", CCP_Type));
1293 Results.AddResult(Result("_Bool", CCP_Type));
1294 Results.AddResult(Result("restrict", CCP_Type));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001295 }
1296
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001297 CodeCompletionBuilder Builder(Results.getAllocator(),
1298 Results.getCodeCompletionTUInfo());
Douglas Gregor3545ff42009-09-21 16:56:56 +00001299 if (LangOpts.CPlusPlus) {
1300 // C++-specific
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00001301 Results.AddResult(Result("bool", CCP_Type +
1302 (LangOpts.ObjC1? CCD_bool_in_ObjC : 0)));
Douglas Gregora2db7932010-05-26 22:00:08 +00001303 Results.AddResult(Result("class", CCP_Type));
1304 Results.AddResult(Result("wchar_t", CCP_Type));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001305
Douglas Gregorf4c33342010-05-28 00:22:41 +00001306 // typename qualified-id
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001307 Builder.AddTypedTextChunk("typename");
1308 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1309 Builder.AddPlaceholderChunk("qualifier");
1310 Builder.AddTextChunk("::");
1311 Builder.AddPlaceholderChunk("name");
1312 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001313
Richard Smith2bf7fdb2013-01-02 11:42:31 +00001314 if (LangOpts.CPlusPlus11) {
Douglas Gregora2db7932010-05-26 22:00:08 +00001315 Results.AddResult(Result("auto", CCP_Type));
1316 Results.AddResult(Result("char16_t", CCP_Type));
1317 Results.AddResult(Result("char32_t", CCP_Type));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001318
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001319 Builder.AddTypedTextChunk("decltype");
1320 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1321 Builder.AddPlaceholderChunk("expression");
1322 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1323 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001324 }
1325 }
1326
1327 // GNU extensions
1328 if (LangOpts.GNUMode) {
1329 // FIXME: Enable when we actually support decimal floating point.
Douglas Gregor78a21012010-01-14 16:01:26 +00001330 // Results.AddResult(Result("_Decimal32"));
1331 // Results.AddResult(Result("_Decimal64"));
1332 // Results.AddResult(Result("_Decimal128"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001333
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001334 Builder.AddTypedTextChunk("typeof");
1335 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1336 Builder.AddPlaceholderChunk("expression");
1337 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001338
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001339 Builder.AddTypedTextChunk("typeof");
1340 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1341 Builder.AddPlaceholderChunk("type");
1342 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1343 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001344 }
1345}
1346
John McCallfaf5fb42010-08-26 23:41:50 +00001347static void AddStorageSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001348 const LangOptions &LangOpts,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001349 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001350 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001351 // Note: we don't suggest either "auto" or "register", because both
1352 // are pointless as storage specifiers. Elsewhere, we suggest "auto"
1353 // in C++0x as a type specifier.
Douglas Gregor78a21012010-01-14 16:01:26 +00001354 Results.AddResult(Result("extern"));
1355 Results.AddResult(Result("static"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001356}
1357
John McCallfaf5fb42010-08-26 23:41:50 +00001358static void AddFunctionSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001359 const LangOptions &LangOpts,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001360 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001361 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001362 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001363 case Sema::PCC_Class:
1364 case Sema::PCC_MemberTemplate:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001365 if (LangOpts.CPlusPlus) {
Douglas Gregor78a21012010-01-14 16:01:26 +00001366 Results.AddResult(Result("explicit"));
1367 Results.AddResult(Result("friend"));
1368 Results.AddResult(Result("mutable"));
1369 Results.AddResult(Result("virtual"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001370 }
1371 // Fall through
1372
John McCallfaf5fb42010-08-26 23:41:50 +00001373 case Sema::PCC_ObjCInterface:
1374 case Sema::PCC_ObjCImplementation:
1375 case Sema::PCC_Namespace:
1376 case Sema::PCC_Template:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001377 if (LangOpts.CPlusPlus || LangOpts.C99)
Douglas Gregor78a21012010-01-14 16:01:26 +00001378 Results.AddResult(Result("inline"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001379 break;
1380
John McCallfaf5fb42010-08-26 23:41:50 +00001381 case Sema::PCC_ObjCInstanceVariableList:
1382 case Sema::PCC_Expression:
1383 case Sema::PCC_Statement:
1384 case Sema::PCC_ForInit:
1385 case Sema::PCC_Condition:
1386 case Sema::PCC_RecoveryInFunction:
1387 case Sema::PCC_Type:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001388 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor80039242011-02-15 20:33:25 +00001389 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001390 break;
1391 }
1392}
1393
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001394static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt);
1395static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt);
1396static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00001397 ResultBuilder &Results,
1398 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001399static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00001400 ResultBuilder &Results,
1401 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001402static void AddObjCInterfaceResults(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 AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt);
Douglas Gregorf1934162010-01-13 21:24:21 +00001406
Douglas Gregorf4c33342010-05-28 00:22:41 +00001407static void AddTypedefResult(ResultBuilder &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001408 CodeCompletionBuilder Builder(Results.getAllocator(),
1409 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001410 Builder.AddTypedTextChunk("typedef");
1411 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1412 Builder.AddPlaceholderChunk("type");
1413 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1414 Builder.AddPlaceholderChunk("name");
1415 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001416}
1417
John McCallfaf5fb42010-08-26 23:41:50 +00001418static bool WantTypesInContext(Sema::ParserCompletionContext CCC,
Douglas Gregor70febae2010-05-28 00:49:12 +00001419 const LangOptions &LangOpts) {
Douglas Gregor70febae2010-05-28 00:49:12 +00001420 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001421 case Sema::PCC_Namespace:
1422 case Sema::PCC_Class:
1423 case Sema::PCC_ObjCInstanceVariableList:
1424 case Sema::PCC_Template:
1425 case Sema::PCC_MemberTemplate:
1426 case Sema::PCC_Statement:
1427 case Sema::PCC_RecoveryInFunction:
1428 case Sema::PCC_Type:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001429 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor80039242011-02-15 20:33:25 +00001430 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor70febae2010-05-28 00:49:12 +00001431 return true;
1432
John McCallfaf5fb42010-08-26 23:41:50 +00001433 case Sema::PCC_Expression:
1434 case Sema::PCC_Condition:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001435 return LangOpts.CPlusPlus;
1436
1437 case Sema::PCC_ObjCInterface:
1438 case Sema::PCC_ObjCImplementation:
Douglas Gregor70febae2010-05-28 00:49:12 +00001439 return false;
1440
John McCallfaf5fb42010-08-26 23:41:50 +00001441 case Sema::PCC_ForInit:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001442 return LangOpts.CPlusPlus || LangOpts.ObjC1 || LangOpts.C99;
Douglas Gregor70febae2010-05-28 00:49:12 +00001443 }
David Blaikie8a40f702012-01-17 06:56:22 +00001444
1445 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor70febae2010-05-28 00:49:12 +00001446}
1447
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00001448static PrintingPolicy getCompletionPrintingPolicy(const ASTContext &Context,
1449 const Preprocessor &PP) {
1450 PrintingPolicy Policy = Sema::getPrintingPolicy(Context, PP);
Douglas Gregore5c79d52011-10-18 21:20:17 +00001451 Policy.AnonymousTagLocations = false;
1452 Policy.SuppressStrongLifetime = true;
Douglas Gregor2e10cf92011-11-03 00:16:13 +00001453 Policy.SuppressUnwrittenScope = true;
Douglas Gregore5c79d52011-10-18 21:20:17 +00001454 return Policy;
1455}
1456
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00001457/// \brief Retrieve a printing policy suitable for code completion.
1458static PrintingPolicy getCompletionPrintingPolicy(Sema &S) {
1459 return getCompletionPrintingPolicy(S.Context, S.PP);
1460}
1461
Douglas Gregore5c79d52011-10-18 21:20:17 +00001462/// \brief Retrieve the string representation of the given type as a string
1463/// that has the appropriate lifetime for code completion.
1464///
1465/// This routine provides a fast path where we provide constant strings for
1466/// common type names.
1467static const char *GetCompletionTypeString(QualType T,
1468 ASTContext &Context,
1469 const PrintingPolicy &Policy,
1470 CodeCompletionAllocator &Allocator) {
1471 if (!T.getLocalQualifiers()) {
1472 // Built-in type names are constant strings.
1473 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T))
Argyrios Kyrtzidisbbff3da2012-05-05 04:20:28 +00001474 return BT->getNameAsCString(Policy);
Douglas Gregore5c79d52011-10-18 21:20:17 +00001475
1476 // Anonymous tag types are constant strings.
1477 if (const TagType *TagT = dyn_cast<TagType>(T))
1478 if (TagDecl *Tag = TagT->getDecl())
John McCall5ea95772013-03-09 00:54:27 +00001479 if (!Tag->hasNameForLinkage()) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001480 switch (Tag->getTagKind()) {
1481 case TTK_Struct: return "struct <anonymous>";
Joao Matosdc86f942012-08-31 18:45:21 +00001482 case TTK_Interface: return "__interface <anonymous>";
1483 case TTK_Class: return "class <anonymous>";
Douglas Gregore5c79d52011-10-18 21:20:17 +00001484 case TTK_Union: return "union <anonymous>";
1485 case TTK_Enum: return "enum <anonymous>";
1486 }
1487 }
1488 }
1489
1490 // Slow path: format the type as a string.
1491 std::string Result;
1492 T.getAsStringInternal(Result, Policy);
1493 return Allocator.CopyString(Result);
1494}
1495
Douglas Gregord8c61782012-02-15 15:34:24 +00001496/// \brief Add a completion for "this", if we're in a member function.
1497static void addThisCompletion(Sema &S, ResultBuilder &Results) {
1498 QualType ThisTy = S.getCurrentThisType();
1499 if (ThisTy.isNull())
1500 return;
1501
1502 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001503 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregord8c61782012-02-15 15:34:24 +00001504 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
1505 Builder.AddResultTypeChunk(GetCompletionTypeString(ThisTy,
1506 S.Context,
1507 Policy,
1508 Allocator));
1509 Builder.AddTypedTextChunk("this");
Joao Matosdc86f942012-08-31 18:45:21 +00001510 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregord8c61782012-02-15 15:34:24 +00001511}
1512
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001513/// \brief Add language constructs that show up for "ordinary" names.
John McCallfaf5fb42010-08-26 23:41:50 +00001514static void AddOrdinaryNameResults(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001515 Scope *S,
1516 Sema &SemaRef,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001517 ResultBuilder &Results) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001518 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001519 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregore5c79d52011-10-18 21:20:17 +00001520 PrintingPolicy Policy = getCompletionPrintingPolicy(SemaRef);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001521
John McCall276321a2010-08-25 06:19:51 +00001522 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001523 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001524 case Sema::PCC_Namespace:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001525 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001526 if (Results.includeCodePatterns()) {
1527 // namespace <identifier> { declarations }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001528 Builder.AddTypedTextChunk("namespace");
1529 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1530 Builder.AddPlaceholderChunk("identifier");
1531 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1532 Builder.AddPlaceholderChunk("declarations");
1533 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1534 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1535 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001536 }
1537
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001538 // namespace identifier = identifier ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001539 Builder.AddTypedTextChunk("namespace");
1540 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1541 Builder.AddPlaceholderChunk("name");
1542 Builder.AddChunk(CodeCompletionString::CK_Equal);
1543 Builder.AddPlaceholderChunk("namespace");
1544 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001545
1546 // Using directives
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001547 Builder.AddTypedTextChunk("using");
1548 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1549 Builder.AddTextChunk("namespace");
1550 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1551 Builder.AddPlaceholderChunk("identifier");
1552 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001553
1554 // asm(string-literal)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001555 Builder.AddTypedTextChunk("asm");
1556 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1557 Builder.AddPlaceholderChunk("string-literal");
1558 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1559 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001560
Douglas Gregorf4c33342010-05-28 00:22:41 +00001561 if (Results.includeCodePatterns()) {
1562 // Explicit template instantiation
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001563 Builder.AddTypedTextChunk("template");
1564 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1565 Builder.AddPlaceholderChunk("declaration");
1566 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001567 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001568 }
Douglas Gregorf1934162010-01-13 21:24:21 +00001569
David Blaikiebbafb8a2012-03-11 07:00:24 +00001570 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001571 AddObjCTopLevelResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001572
Douglas Gregorf4c33342010-05-28 00:22:41 +00001573 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001574 // Fall through
1575
John McCallfaf5fb42010-08-26 23:41:50 +00001576 case Sema::PCC_Class:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001577 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001578 // Using declaration
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001579 Builder.AddTypedTextChunk("using");
1580 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1581 Builder.AddPlaceholderChunk("qualifier");
1582 Builder.AddTextChunk("::");
1583 Builder.AddPlaceholderChunk("name");
1584 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001585
Douglas Gregorf4c33342010-05-28 00:22:41 +00001586 // using typename qualifier::name (only in a dependent context)
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001587 if (SemaRef.CurContext->isDependentContext()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001588 Builder.AddTypedTextChunk("using");
1589 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1590 Builder.AddTextChunk("typename");
1591 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1592 Builder.AddPlaceholderChunk("qualifier");
1593 Builder.AddTextChunk("::");
1594 Builder.AddPlaceholderChunk("name");
1595 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001596 }
1597
John McCallfaf5fb42010-08-26 23:41:50 +00001598 if (CCC == Sema::PCC_Class) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001599 AddTypedefResult(Results);
1600
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001601 // public:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001602 Builder.AddTypedTextChunk("public");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001603 if (Results.includeCodePatterns())
1604 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001605 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001606
1607 // protected:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001608 Builder.AddTypedTextChunk("protected");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001609 if (Results.includeCodePatterns())
1610 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001611 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001612
1613 // private:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001614 Builder.AddTypedTextChunk("private");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001615 if (Results.includeCodePatterns())
1616 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001617 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001618 }
1619 }
1620 // Fall through
1621
John McCallfaf5fb42010-08-26 23:41:50 +00001622 case Sema::PCC_Template:
1623 case Sema::PCC_MemberTemplate:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001624 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001625 // template < parameters >
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001626 Builder.AddTypedTextChunk("template");
1627 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1628 Builder.AddPlaceholderChunk("parameters");
1629 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1630 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001631 }
1632
David Blaikiebbafb8a2012-03-11 07:00:24 +00001633 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1634 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001635 break;
1636
John McCallfaf5fb42010-08-26 23:41:50 +00001637 case Sema::PCC_ObjCInterface:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001638 AddObjCInterfaceResults(SemaRef.getLangOpts(), Results, true);
1639 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1640 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001641 break;
1642
John McCallfaf5fb42010-08-26 23:41:50 +00001643 case Sema::PCC_ObjCImplementation:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001644 AddObjCImplementationResults(SemaRef.getLangOpts(), Results, true);
1645 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1646 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001647 break;
1648
John McCallfaf5fb42010-08-26 23:41:50 +00001649 case Sema::PCC_ObjCInstanceVariableList:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001650 AddObjCVisibilityResults(SemaRef.getLangOpts(), Results, true);
Douglas Gregor48d46252010-01-13 21:54:15 +00001651 break;
1652
John McCallfaf5fb42010-08-26 23:41:50 +00001653 case Sema::PCC_RecoveryInFunction:
1654 case Sema::PCC_Statement: {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001655 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001656
David Blaikiebbafb8a2012-03-11 07:00:24 +00001657 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns() &&
1658 SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001659 Builder.AddTypedTextChunk("try");
1660 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1661 Builder.AddPlaceholderChunk("statements");
1662 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1663 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1664 Builder.AddTextChunk("catch");
1665 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1666 Builder.AddPlaceholderChunk("declaration");
1667 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1668 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1669 Builder.AddPlaceholderChunk("statements");
1670 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1671 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1672 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001673 }
David Blaikiebbafb8a2012-03-11 07:00:24 +00001674 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001675 AddObjCStatementResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001676
Douglas Gregorf64acca2010-05-25 21:41:55 +00001677 if (Results.includeCodePatterns()) {
1678 // if (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001679 Builder.AddTypedTextChunk("if");
1680 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001681 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001682 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001683 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001684 Builder.AddPlaceholderChunk("expression");
1685 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1686 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1687 Builder.AddPlaceholderChunk("statements");
1688 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1689 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1690 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001691
Douglas Gregorf64acca2010-05-25 21:41:55 +00001692 // switch (condition) { }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001693 Builder.AddTypedTextChunk("switch");
1694 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001695 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001696 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001697 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001698 Builder.AddPlaceholderChunk("expression");
1699 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1700 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1701 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1702 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1703 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001704 }
1705
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001706 // Switch-specific statements.
John McCallaab3e412010-08-25 08:40:02 +00001707 if (!SemaRef.getCurFunction()->SwitchStack.empty()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001708 // case expression:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001709 Builder.AddTypedTextChunk("case");
1710 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1711 Builder.AddPlaceholderChunk("expression");
1712 Builder.AddChunk(CodeCompletionString::CK_Colon);
1713 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001714
1715 // default:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001716 Builder.AddTypedTextChunk("default");
1717 Builder.AddChunk(CodeCompletionString::CK_Colon);
1718 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001719 }
1720
Douglas Gregorf64acca2010-05-25 21:41:55 +00001721 if (Results.includeCodePatterns()) {
1722 /// while (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001723 Builder.AddTypedTextChunk("while");
1724 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001725 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001726 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001727 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001728 Builder.AddPlaceholderChunk("expression");
1729 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1730 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1731 Builder.AddPlaceholderChunk("statements");
1732 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1733 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1734 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001735
1736 // do { statements } while ( expression );
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001737 Builder.AddTypedTextChunk("do");
1738 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1739 Builder.AddPlaceholderChunk("statements");
1740 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1741 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1742 Builder.AddTextChunk("while");
1743 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1744 Builder.AddPlaceholderChunk("expression");
1745 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1746 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001747
Douglas Gregorf64acca2010-05-25 21:41:55 +00001748 // for ( for-init-statement ; condition ; expression ) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001749 Builder.AddTypedTextChunk("for");
1750 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001751 if (SemaRef.getLangOpts().CPlusPlus || SemaRef.getLangOpts().C99)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001752 Builder.AddPlaceholderChunk("init-statement");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001753 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001754 Builder.AddPlaceholderChunk("init-expression");
1755 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1756 Builder.AddPlaceholderChunk("condition");
1757 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1758 Builder.AddPlaceholderChunk("inc-expression");
1759 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1760 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1761 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1762 Builder.AddPlaceholderChunk("statements");
1763 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1764 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1765 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001766 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001767
1768 if (S->getContinueParent()) {
1769 // continue ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001770 Builder.AddTypedTextChunk("continue");
1771 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001772 }
1773
1774 if (S->getBreakParent()) {
1775 // break ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001776 Builder.AddTypedTextChunk("break");
1777 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001778 }
1779
1780 // "return expression ;" or "return ;", depending on whether we
1781 // know the function is void or not.
1782 bool isVoid = false;
1783 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00001784 isVoid = Function->getReturnType()->isVoidType();
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001785 else if (ObjCMethodDecl *Method
1786 = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00001787 isVoid = Method->getReturnType()->isVoidType();
Douglas Gregor9a28e842010-03-01 23:15:13 +00001788 else if (SemaRef.getCurBlock() &&
1789 !SemaRef.getCurBlock()->ReturnType.isNull())
1790 isVoid = SemaRef.getCurBlock()->ReturnType->isVoidType();
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001791 Builder.AddTypedTextChunk("return");
Douglas Gregor44272ca2010-02-18 04:06:48 +00001792 if (!isVoid) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001793 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1794 Builder.AddPlaceholderChunk("expression");
Douglas Gregor44272ca2010-02-18 04:06:48 +00001795 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001796 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001797
Douglas Gregorf4c33342010-05-28 00:22:41 +00001798 // goto identifier ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001799 Builder.AddTypedTextChunk("goto");
1800 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1801 Builder.AddPlaceholderChunk("label");
1802 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001803
Douglas Gregorf4c33342010-05-28 00:22:41 +00001804 // Using directives
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001805 Builder.AddTypedTextChunk("using");
1806 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1807 Builder.AddTextChunk("namespace");
1808 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1809 Builder.AddPlaceholderChunk("identifier");
1810 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001811 }
1812
1813 // Fall through (for statement expressions).
John McCallfaf5fb42010-08-26 23:41:50 +00001814 case Sema::PCC_ForInit:
1815 case Sema::PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001816 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001817 // Fall through: conditions and statements can have expressions.
1818
Douglas Gregor5e35d592010-09-14 23:59:36 +00001819 case Sema::PCC_ParenthesizedExpression:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001820 if (SemaRef.getLangOpts().ObjCAutoRefCount &&
John McCall31168b02011-06-15 23:02:42 +00001821 CCC == Sema::PCC_ParenthesizedExpression) {
1822 // (__bridge <type>)<expression>
1823 Builder.AddTypedTextChunk("__bridge");
1824 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1825 Builder.AddPlaceholderChunk("type");
1826 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1827 Builder.AddPlaceholderChunk("expression");
1828 Results.AddResult(Result(Builder.TakeString()));
1829
1830 // (__bridge_transfer <Objective-C type>)<expression>
1831 Builder.AddTypedTextChunk("__bridge_transfer");
1832 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1833 Builder.AddPlaceholderChunk("Objective-C type");
1834 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1835 Builder.AddPlaceholderChunk("expression");
1836 Results.AddResult(Result(Builder.TakeString()));
1837
1838 // (__bridge_retained <CF type>)<expression>
1839 Builder.AddTypedTextChunk("__bridge_retained");
1840 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1841 Builder.AddPlaceholderChunk("CF type");
1842 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1843 Builder.AddPlaceholderChunk("expression");
1844 Results.AddResult(Result(Builder.TakeString()));
1845 }
1846 // Fall through
1847
John McCallfaf5fb42010-08-26 23:41:50 +00001848 case Sema::PCC_Expression: {
David Blaikiebbafb8a2012-03-11 07:00:24 +00001849 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001850 // 'this', if we're in a non-static member function.
Douglas Gregord8c61782012-02-15 15:34:24 +00001851 addThisCompletion(SemaRef, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001852
Douglas Gregore5c79d52011-10-18 21:20:17 +00001853 // true
1854 Builder.AddResultTypeChunk("bool");
1855 Builder.AddTypedTextChunk("true");
1856 Results.AddResult(Result(Builder.TakeString()));
1857
1858 // false
1859 Builder.AddResultTypeChunk("bool");
1860 Builder.AddTypedTextChunk("false");
1861 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001862
David Blaikiebbafb8a2012-03-11 07:00:24 +00001863 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001864 // dynamic_cast < type-id > ( expression )
1865 Builder.AddTypedTextChunk("dynamic_cast");
1866 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1867 Builder.AddPlaceholderChunk("type");
1868 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1869 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1870 Builder.AddPlaceholderChunk("expression");
1871 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1872 Results.AddResult(Result(Builder.TakeString()));
1873 }
Douglas Gregorf4c33342010-05-28 00:22:41 +00001874
1875 // static_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001876 Builder.AddTypedTextChunk("static_cast");
1877 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1878 Builder.AddPlaceholderChunk("type");
1879 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1880 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1881 Builder.AddPlaceholderChunk("expression");
1882 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1883 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001884
Douglas Gregorf4c33342010-05-28 00:22:41 +00001885 // reinterpret_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001886 Builder.AddTypedTextChunk("reinterpret_cast");
1887 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1888 Builder.AddPlaceholderChunk("type");
1889 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1890 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1891 Builder.AddPlaceholderChunk("expression");
1892 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1893 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001894
Douglas Gregorf4c33342010-05-28 00:22:41 +00001895 // const_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001896 Builder.AddTypedTextChunk("const_cast");
1897 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1898 Builder.AddPlaceholderChunk("type");
1899 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1900 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1901 Builder.AddPlaceholderChunk("expression");
1902 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1903 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001904
David Blaikiebbafb8a2012-03-11 07:00:24 +00001905 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001906 // typeid ( expression-or-type )
Douglas Gregore5c79d52011-10-18 21:20:17 +00001907 Builder.AddResultTypeChunk("std::type_info");
Douglas Gregorc05f6572011-04-12 02:47:21 +00001908 Builder.AddTypedTextChunk("typeid");
1909 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1910 Builder.AddPlaceholderChunk("expression-or-type");
1911 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1912 Results.AddResult(Result(Builder.TakeString()));
1913 }
1914
Douglas Gregorf4c33342010-05-28 00:22:41 +00001915 // new T ( ... )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001916 Builder.AddTypedTextChunk("new");
1917 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1918 Builder.AddPlaceholderChunk("type");
1919 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1920 Builder.AddPlaceholderChunk("expressions");
1921 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1922 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001923
Douglas Gregorf4c33342010-05-28 00:22:41 +00001924 // new T [ ] ( ... )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001925 Builder.AddTypedTextChunk("new");
1926 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1927 Builder.AddPlaceholderChunk("type");
1928 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1929 Builder.AddPlaceholderChunk("size");
1930 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1931 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1932 Builder.AddPlaceholderChunk("expressions");
1933 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1934 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001935
Douglas Gregorf4c33342010-05-28 00:22:41 +00001936 // delete expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001937 Builder.AddResultTypeChunk("void");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001938 Builder.AddTypedTextChunk("delete");
1939 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1940 Builder.AddPlaceholderChunk("expression");
1941 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001942
Douglas Gregorf4c33342010-05-28 00:22:41 +00001943 // delete [] expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001944 Builder.AddResultTypeChunk("void");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001945 Builder.AddTypedTextChunk("delete");
1946 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1947 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1948 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1949 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1950 Builder.AddPlaceholderChunk("expression");
1951 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001952
David Blaikiebbafb8a2012-03-11 07:00:24 +00001953 if (SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001954 // throw expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001955 Builder.AddResultTypeChunk("void");
Douglas Gregorc05f6572011-04-12 02:47:21 +00001956 Builder.AddTypedTextChunk("throw");
1957 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1958 Builder.AddPlaceholderChunk("expression");
1959 Results.AddResult(Result(Builder.TakeString()));
1960 }
Douglas Gregor4205fef2011-10-18 16:29:03 +00001961
Douglas Gregora2db7932010-05-26 22:00:08 +00001962 // FIXME: Rethrow?
Douglas Gregor4205fef2011-10-18 16:29:03 +00001963
Richard Smith2bf7fdb2013-01-02 11:42:31 +00001964 if (SemaRef.getLangOpts().CPlusPlus11) {
Douglas Gregor4205fef2011-10-18 16:29:03 +00001965 // nullptr
Douglas Gregore5c79d52011-10-18 21:20:17 +00001966 Builder.AddResultTypeChunk("std::nullptr_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001967 Builder.AddTypedTextChunk("nullptr");
1968 Results.AddResult(Result(Builder.TakeString()));
1969
1970 // alignof
Douglas Gregore5c79d52011-10-18 21:20:17 +00001971 Builder.AddResultTypeChunk("size_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001972 Builder.AddTypedTextChunk("alignof");
1973 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1974 Builder.AddPlaceholderChunk("type");
1975 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1976 Results.AddResult(Result(Builder.TakeString()));
1977
1978 // noexcept
Douglas Gregore5c79d52011-10-18 21:20:17 +00001979 Builder.AddResultTypeChunk("bool");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001980 Builder.AddTypedTextChunk("noexcept");
1981 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1982 Builder.AddPlaceholderChunk("expression");
1983 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1984 Results.AddResult(Result(Builder.TakeString()));
1985
1986 // sizeof... expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001987 Builder.AddResultTypeChunk("size_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001988 Builder.AddTypedTextChunk("sizeof...");
1989 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1990 Builder.AddPlaceholderChunk("parameter-pack");
1991 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1992 Results.AddResult(Result(Builder.TakeString()));
1993 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001994 }
1995
David Blaikiebbafb8a2012-03-11 07:00:24 +00001996 if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001997 // Add "super", if we're in an Objective-C class with a superclass.
Ted Kremenek305a0a72010-05-31 21:43:10 +00001998 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) {
1999 // The interface can be NULL.
2000 if (ObjCInterfaceDecl *ID = Method->getClassInterface())
Douglas Gregore5c79d52011-10-18 21:20:17 +00002001 if (ID->getSuperClass()) {
2002 std::string SuperType;
2003 SuperType = ID->getSuperClass()->getNameAsString();
2004 if (Method->isInstanceMethod())
2005 SuperType += " *";
2006
2007 Builder.AddResultTypeChunk(Allocator.CopyString(SuperType));
2008 Builder.AddTypedTextChunk("super");
2009 Results.AddResult(Result(Builder.TakeString()));
2010 }
Ted Kremenek305a0a72010-05-31 21:43:10 +00002011 }
2012
Douglas Gregorf98e6a22010-01-13 23:51:12 +00002013 AddObjCExpressionResults(Results, true);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002014 }
2015
Jordan Rose58d54722012-06-30 21:33:57 +00002016 if (SemaRef.getLangOpts().C11) {
2017 // _Alignof
2018 Builder.AddResultTypeChunk("size_t");
2019 if (SemaRef.getASTContext().Idents.get("alignof").hasMacroDefinition())
2020 Builder.AddTypedTextChunk("alignof");
2021 else
2022 Builder.AddTypedTextChunk("_Alignof");
2023 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2024 Builder.AddPlaceholderChunk("type");
2025 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2026 Results.AddResult(Result(Builder.TakeString()));
2027 }
2028
Douglas Gregorf4c33342010-05-28 00:22:41 +00002029 // sizeof expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00002030 Builder.AddResultTypeChunk("size_t");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002031 Builder.AddTypedTextChunk("sizeof");
2032 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2033 Builder.AddPlaceholderChunk("expression-or-type");
2034 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2035 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002036 break;
2037 }
Douglas Gregor99fa2642010-08-24 01:06:58 +00002038
John McCallfaf5fb42010-08-26 23:41:50 +00002039 case Sema::PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00002040 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor99fa2642010-08-24 01:06:58 +00002041 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002042 }
2043
David Blaikiebbafb8a2012-03-11 07:00:24 +00002044 if (WantTypesInContext(CCC, SemaRef.getLangOpts()))
2045 AddTypeSpecifierResults(SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002046
David Blaikiebbafb8a2012-03-11 07:00:24 +00002047 if (SemaRef.getLangOpts().CPlusPlus && CCC != Sema::PCC_Type)
Douglas Gregor78a21012010-01-14 16:01:26 +00002048 Results.AddResult(Result("operator"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002049}
2050
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002051/// \brief If the given declaration has an associated type, add it as a result
2052/// type chunk.
2053static void AddResultTypeChunk(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002054 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002055 const NamedDecl *ND,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002056 CodeCompletionBuilder &Result) {
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002057 if (!ND)
2058 return;
Douglas Gregor0212fd72010-09-21 16:06:22 +00002059
2060 // Skip constructors and conversion functions, which have their return types
2061 // built into their names.
2062 if (isa<CXXConstructorDecl>(ND) || isa<CXXConversionDecl>(ND))
2063 return;
2064
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002065 // Determine the type of the declaration (if it has a type).
Alp Tokera2794f92014-01-22 07:29:52 +00002066 QualType T;
2067 if (const FunctionDecl *Function = ND->getAsFunction())
Alp Toker314cc812014-01-25 16:55:45 +00002068 T = Function->getReturnType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002069 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Alp Toker314cc812014-01-25 16:55:45 +00002070 T = Method->getReturnType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002071 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002072 T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext()));
2073 else if (isa<UnresolvedUsingValueDecl>(ND)) {
2074 /* Do nothing: ignore unresolved using declarations*/
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002075 } else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND)) {
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002076 T = Value->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002077 } else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002078 T = Property->getType();
2079
2080 if (T.isNull() || Context.hasSameType(T, Context.DependentTy))
2081 return;
2082
Douglas Gregor75acd922011-09-27 23:30:47 +00002083 Result.AddResultTypeChunk(GetCompletionTypeString(T, Context, Policy,
Douglas Gregor304f9b02011-02-01 21:15:40 +00002084 Result.getAllocator()));
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002085}
2086
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002087static void MaybeAddSentinel(ASTContext &Context,
2088 const NamedDecl *FunctionOrMethod,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002089 CodeCompletionBuilder &Result) {
Douglas Gregordbb71db2010-08-23 23:51:41 +00002090 if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>())
2091 if (Sentinel->getSentinel() == 0) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00002092 if (Context.getLangOpts().ObjC1 &&
Douglas Gregordbb71db2010-08-23 23:51:41 +00002093 Context.Idents.get("nil").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002094 Result.AddTextChunk(", nil");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002095 else if (Context.Idents.get("NULL").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002096 Result.AddTextChunk(", NULL");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002097 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002098 Result.AddTextChunk(", (void*)0");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002099 }
2100}
2101
Douglas Gregor8f08d742011-07-30 07:55:26 +00002102static std::string formatObjCParamQualifiers(unsigned ObjCQuals) {
2103 std::string Result;
2104 if (ObjCQuals & Decl::OBJC_TQ_In)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002105 Result += "in ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002106 else if (ObjCQuals & Decl::OBJC_TQ_Inout)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002107 Result += "inout ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002108 else if (ObjCQuals & Decl::OBJC_TQ_Out)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002109 Result += "out ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002110 if (ObjCQuals & Decl::OBJC_TQ_Bycopy)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002111 Result += "bycopy ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002112 else if (ObjCQuals & Decl::OBJC_TQ_Byref)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002113 Result += "byref ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002114 if (ObjCQuals & Decl::OBJC_TQ_Oneway)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002115 Result += "oneway ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002116 return Result;
2117}
2118
Douglas Gregore90dd002010-08-24 16:15:59 +00002119static std::string FormatFunctionParameter(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002120 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002121 const ParmVarDecl *Param,
Douglas Gregord793e7c2011-10-18 04:23:19 +00002122 bool SuppressName = false,
2123 bool SuppressBlock = false) {
Douglas Gregore90dd002010-08-24 16:15:59 +00002124 bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext());
2125 if (Param->getType()->isDependentType() ||
2126 !Param->getType()->isBlockPointerType()) {
2127 // The argument for a dependent or non-block parameter is a placeholder
2128 // containing that parameter's type.
2129 std::string Result;
2130
Douglas Gregor981a0c42010-08-29 19:47:46 +00002131 if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002132 Result = Param->getIdentifier()->getName();
2133
John McCall31168b02011-06-15 23:02:42 +00002134 Param->getType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002135
2136 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002137 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2138 + Result + ")";
Douglas Gregor981a0c42010-08-29 19:47:46 +00002139 if (Param->getIdentifier() && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002140 Result += Param->getIdentifier()->getName();
2141 }
2142 return Result;
2143 }
2144
2145 // The argument for a block pointer parameter is a block literal with
2146 // the appropriate type.
David Blaikie6adc78e2013-02-18 22:06:02 +00002147 FunctionTypeLoc Block;
2148 FunctionProtoTypeLoc BlockProto;
Douglas Gregore90dd002010-08-24 16:15:59 +00002149 TypeLoc TL;
2150 if (TypeSourceInfo *TSInfo = Param->getTypeSourceInfo()) {
2151 TL = TSInfo->getTypeLoc().getUnqualifiedLoc();
2152 while (true) {
2153 // Look through typedefs.
Douglas Gregord793e7c2011-10-18 04:23:19 +00002154 if (!SuppressBlock) {
David Blaikie6adc78e2013-02-18 22:06:02 +00002155 if (TypedefTypeLoc TypedefTL = TL.getAs<TypedefTypeLoc>()) {
2156 if (TypeSourceInfo *InnerTSInfo =
2157 TypedefTL.getTypedefNameDecl()->getTypeSourceInfo()) {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002158 TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc();
2159 continue;
2160 }
2161 }
2162
2163 // Look through qualified types
David Blaikie6adc78e2013-02-18 22:06:02 +00002164 if (QualifiedTypeLoc QualifiedTL = TL.getAs<QualifiedTypeLoc>()) {
2165 TL = QualifiedTL.getUnqualifiedLoc();
Douglas Gregore90dd002010-08-24 16:15:59 +00002166 continue;
2167 }
2168 }
2169
Douglas Gregore90dd002010-08-24 16:15:59 +00002170 // Try to get the function prototype behind the block pointer type,
2171 // then we're done.
David Blaikie6adc78e2013-02-18 22:06:02 +00002172 if (BlockPointerTypeLoc BlockPtr = TL.getAs<BlockPointerTypeLoc>()) {
2173 TL = BlockPtr.getPointeeLoc().IgnoreParens();
2174 Block = TL.getAs<FunctionTypeLoc>();
2175 BlockProto = TL.getAs<FunctionProtoTypeLoc>();
Douglas Gregore90dd002010-08-24 16:15:59 +00002176 }
2177 break;
2178 }
2179 }
2180
2181 if (!Block) {
2182 // We were unable to find a FunctionProtoTypeLoc with parameter names
2183 // for the block; just use the parameter type as a placeholder.
2184 std::string Result;
Douglas Gregord793e7c2011-10-18 04:23:19 +00002185 if (!ObjCMethodParam && Param->getIdentifier())
2186 Result = Param->getIdentifier()->getName();
2187
John McCall31168b02011-06-15 23:02:42 +00002188 Param->getType().getUnqualifiedType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002189
2190 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002191 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2192 + Result + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002193 if (Param->getIdentifier())
2194 Result += Param->getIdentifier()->getName();
2195 }
2196
2197 return Result;
2198 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002199
Douglas Gregore90dd002010-08-24 16:15:59 +00002200 // We have the function prototype behind the block pointer type, as it was
2201 // written in the source.
Douglas Gregor67da50e2010-09-08 22:47:51 +00002202 std::string Result;
Alp Toker314cc812014-01-25 16:55:45 +00002203 QualType ResultType = Block.getTypePtr()->getReturnType();
Douglas Gregord793e7c2011-10-18 04:23:19 +00002204 if (!ResultType->isVoidType() || SuppressBlock)
John McCall31168b02011-06-15 23:02:42 +00002205 ResultType.getAsStringInternal(Result, Policy);
Douglas Gregord793e7c2011-10-18 04:23:19 +00002206
2207 // Format the parameter list.
2208 std::string Params;
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002209 if (!BlockProto || Block.getNumParams() == 0) {
David Blaikie6adc78e2013-02-18 22:06:02 +00002210 if (BlockProto && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002211 Params = "(...)";
Douglas Gregoraf25cfa2010-10-02 23:49:58 +00002212 else
Douglas Gregord793e7c2011-10-18 04:23:19 +00002213 Params = "(void)";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002214 } else {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002215 Params += "(";
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002216 for (unsigned I = 0, N = Block.getNumParams(); I != N; ++I) {
Douglas Gregor67da50e2010-09-08 22:47:51 +00002217 if (I)
Douglas Gregord793e7c2011-10-18 04:23:19 +00002218 Params += ", ";
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002219 Params += FormatFunctionParameter(Context, Policy, Block.getParam(I),
2220 /*SuppressName=*/false,
Douglas Gregord793e7c2011-10-18 04:23:19 +00002221 /*SuppressBlock=*/true);
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002222
David Blaikie6adc78e2013-02-18 22:06:02 +00002223 if (I == N - 1 && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002224 Params += ", ...";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002225 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002226 Params += ")";
Douglas Gregor400f5972010-08-31 05:13:43 +00002227 }
Douglas Gregor67da50e2010-09-08 22:47:51 +00002228
Douglas Gregord793e7c2011-10-18 04:23:19 +00002229 if (SuppressBlock) {
2230 // Format as a parameter.
2231 Result = Result + " (^";
2232 if (Param->getIdentifier())
2233 Result += Param->getIdentifier()->getName();
2234 Result += ")";
2235 Result += Params;
2236 } else {
2237 // Format as a block literal argument.
2238 Result = '^' + Result;
2239 Result += Params;
2240
2241 if (Param->getIdentifier())
2242 Result += Param->getIdentifier()->getName();
2243 }
2244
Douglas Gregore90dd002010-08-24 16:15:59 +00002245 return Result;
2246}
2247
Douglas Gregor3545ff42009-09-21 16:56:56 +00002248/// \brief Add function parameter chunks to the given code completion string.
2249static void AddFunctionParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002250 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002251 const FunctionDecl *Function,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002252 CodeCompletionBuilder &Result,
2253 unsigned Start = 0,
2254 bool InOptional = false) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002255 bool FirstParameter = true;
Douglas Gregor9eb77012009-11-07 00:00:49 +00002256
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002257 for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002258 const ParmVarDecl *Param = Function->getParamDecl(P);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002259
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002260 if (Param->hasDefaultArg() && !InOptional) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002261 // When we see an optional default argument, put that argument and
2262 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002263 CodeCompletionBuilder Opt(Result.getAllocator(),
2264 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002265 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002266 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002267 AddFunctionParameterChunks(Context, Policy, Function, Opt, P, true);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002268 Result.AddOptionalChunk(Opt.TakeString());
2269 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002270 }
2271
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002272 if (FirstParameter)
2273 FirstParameter = false;
2274 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002275 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002276
2277 InOptional = false;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002278
2279 // Format the placeholder string.
Douglas Gregor75acd922011-09-27 23:30:47 +00002280 std::string PlaceholderStr = FormatFunctionParameter(Context, Policy,
2281 Param);
Douglas Gregore90dd002010-08-24 16:15:59 +00002282
Douglas Gregor400f5972010-08-31 05:13:43 +00002283 if (Function->isVariadic() && P == N - 1)
2284 PlaceholderStr += ", ...";
2285
Douglas Gregor3545ff42009-09-21 16:56:56 +00002286 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002287 Result.AddPlaceholderChunk(
2288 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002289 }
Douglas Gregorba449032009-09-22 21:42:17 +00002290
2291 if (const FunctionProtoType *Proto
2292 = Function->getType()->getAs<FunctionProtoType>())
Douglas Gregordbb71db2010-08-23 23:51:41 +00002293 if (Proto->isVariadic()) {
Alp Toker9cacbab2014-01-20 20:26:09 +00002294 if (Proto->getNumParams() == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002295 Result.AddPlaceholderChunk("...");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002296
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002297 MaybeAddSentinel(Context, Function, Result);
Douglas Gregordbb71db2010-08-23 23:51:41 +00002298 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002299}
2300
2301/// \brief Add template parameter chunks to the given code completion string.
2302static void AddTemplateParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002303 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002304 const TemplateDecl *Template,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002305 CodeCompletionBuilder &Result,
2306 unsigned MaxParameters = 0,
2307 unsigned Start = 0,
2308 bool InDefaultArg = false) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002309 bool FirstParameter = true;
2310
2311 TemplateParameterList *Params = Template->getTemplateParameters();
2312 TemplateParameterList::iterator PEnd = Params->end();
2313 if (MaxParameters)
2314 PEnd = Params->begin() + MaxParameters;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002315 for (TemplateParameterList::iterator P = Params->begin() + Start;
2316 P != PEnd; ++P) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002317 bool HasDefaultArg = false;
2318 std::string PlaceholderStr;
2319 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) {
2320 if (TTP->wasDeclaredWithTypename())
2321 PlaceholderStr = "typename";
2322 else
2323 PlaceholderStr = "class";
2324
2325 if (TTP->getIdentifier()) {
2326 PlaceholderStr += ' ';
2327 PlaceholderStr += TTP->getIdentifier()->getName();
2328 }
2329
2330 HasDefaultArg = TTP->hasDefaultArgument();
2331 } else if (NonTypeTemplateParmDecl *NTTP
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002332 = dyn_cast<NonTypeTemplateParmDecl>(*P)) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002333 if (NTTP->getIdentifier())
2334 PlaceholderStr = NTTP->getIdentifier()->getName();
John McCall31168b02011-06-15 23:02:42 +00002335 NTTP->getType().getAsStringInternal(PlaceholderStr, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002336 HasDefaultArg = NTTP->hasDefaultArgument();
2337 } else {
2338 assert(isa<TemplateTemplateParmDecl>(*P));
2339 TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P);
2340
2341 // Since putting the template argument list into the placeholder would
2342 // be very, very long, we just use an abbreviation.
2343 PlaceholderStr = "template<...> class";
2344 if (TTP->getIdentifier()) {
2345 PlaceholderStr += ' ';
2346 PlaceholderStr += TTP->getIdentifier()->getName();
2347 }
2348
2349 HasDefaultArg = TTP->hasDefaultArgument();
2350 }
2351
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002352 if (HasDefaultArg && !InDefaultArg) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002353 // When we see an optional default argument, put that argument and
2354 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002355 CodeCompletionBuilder Opt(Result.getAllocator(),
2356 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002357 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002358 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002359 AddTemplateParameterChunks(Context, Policy, Template, Opt, MaxParameters,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002360 P - Params->begin(), true);
2361 Result.AddOptionalChunk(Opt.TakeString());
2362 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002363 }
2364
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002365 InDefaultArg = false;
2366
Douglas Gregor3545ff42009-09-21 16:56:56 +00002367 if (FirstParameter)
2368 FirstParameter = false;
2369 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002370 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002371
2372 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002373 Result.AddPlaceholderChunk(
2374 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002375 }
2376}
2377
Douglas Gregorf2510672009-09-21 19:57:38 +00002378/// \brief Add a qualifier to the given code-completion string, if the
2379/// provided nested-name-specifier is non-NULL.
Douglas Gregor0f622362009-12-11 18:44:16 +00002380static void
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002381AddQualifierToCompletionString(CodeCompletionBuilder &Result,
Douglas Gregor0f622362009-12-11 18:44:16 +00002382 NestedNameSpecifier *Qualifier,
2383 bool QualifierIsInformative,
Douglas Gregor75acd922011-09-27 23:30:47 +00002384 ASTContext &Context,
2385 const PrintingPolicy &Policy) {
Douglas Gregorf2510672009-09-21 19:57:38 +00002386 if (!Qualifier)
2387 return;
2388
2389 std::string PrintedNNS;
2390 {
2391 llvm::raw_string_ostream OS(PrintedNNS);
Douglas Gregor75acd922011-09-27 23:30:47 +00002392 Qualifier->print(OS, Policy);
Douglas Gregorf2510672009-09-21 19:57:38 +00002393 }
Douglas Gregor5bf52692009-09-22 23:15:58 +00002394 if (QualifierIsInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002395 Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregor5bf52692009-09-22 23:15:58 +00002396 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002397 Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregorf2510672009-09-21 19:57:38 +00002398}
2399
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002400static void
2401AddFunctionTypeQualsToCompletionString(CodeCompletionBuilder &Result,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002402 const FunctionDecl *Function) {
Douglas Gregor0f622362009-12-11 18:44:16 +00002403 const FunctionProtoType *Proto
2404 = Function->getType()->getAs<FunctionProtoType>();
2405 if (!Proto || !Proto->getTypeQuals())
2406 return;
2407
Douglas Gregor304f9b02011-02-01 21:15:40 +00002408 // FIXME: Add ref-qualifier!
2409
2410 // Handle single qualifiers without copying
2411 if (Proto->getTypeQuals() == Qualifiers::Const) {
2412 Result.AddInformativeChunk(" const");
2413 return;
2414 }
2415
2416 if (Proto->getTypeQuals() == Qualifiers::Volatile) {
2417 Result.AddInformativeChunk(" volatile");
2418 return;
2419 }
2420
2421 if (Proto->getTypeQuals() == Qualifiers::Restrict) {
2422 Result.AddInformativeChunk(" restrict");
2423 return;
2424 }
2425
2426 // Handle multiple qualifiers.
Douglas Gregor0f622362009-12-11 18:44:16 +00002427 std::string QualsStr;
David Blaikief5697e52012-08-10 00:55:35 +00002428 if (Proto->isConst())
Douglas Gregor0f622362009-12-11 18:44:16 +00002429 QualsStr += " const";
David Blaikief5697e52012-08-10 00:55:35 +00002430 if (Proto->isVolatile())
Douglas Gregor0f622362009-12-11 18:44:16 +00002431 QualsStr += " volatile";
David Blaikief5697e52012-08-10 00:55:35 +00002432 if (Proto->isRestrict())
Douglas Gregor0f622362009-12-11 18:44:16 +00002433 QualsStr += " restrict";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002434 Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr));
Douglas Gregor0f622362009-12-11 18:44:16 +00002435}
2436
Douglas Gregor0212fd72010-09-21 16:06:22 +00002437/// \brief Add the name of the given declaration
Douglas Gregor75acd922011-09-27 23:30:47 +00002438static void AddTypedNameChunk(ASTContext &Context, const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002439 const NamedDecl *ND,
2440 CodeCompletionBuilder &Result) {
Douglas Gregor0212fd72010-09-21 16:06:22 +00002441 DeclarationName Name = ND->getDeclName();
2442 if (!Name)
2443 return;
2444
2445 switch (Name.getNameKind()) {
Douglas Gregor304f9b02011-02-01 21:15:40 +00002446 case DeclarationName::CXXOperatorName: {
2447 const char *OperatorName = 0;
2448 switch (Name.getCXXOverloadedOperator()) {
2449 case OO_None:
2450 case OO_Conditional:
2451 case NUM_OVERLOADED_OPERATORS:
2452 OperatorName = "operator";
2453 break;
2454
2455#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
2456 case OO_##Name: OperatorName = "operator" Spelling; break;
2457#define OVERLOADED_OPERATOR_MULTI(Name,Spelling,Unary,Binary,MemberOnly)
2458#include "clang/Basic/OperatorKinds.def"
2459
2460 case OO_New: OperatorName = "operator new"; break;
2461 case OO_Delete: OperatorName = "operator delete"; break;
2462 case OO_Array_New: OperatorName = "operator new[]"; break;
2463 case OO_Array_Delete: OperatorName = "operator delete[]"; break;
2464 case OO_Call: OperatorName = "operator()"; break;
2465 case OO_Subscript: OperatorName = "operator[]"; break;
2466 }
2467 Result.AddTypedTextChunk(OperatorName);
2468 break;
2469 }
2470
Douglas Gregor0212fd72010-09-21 16:06:22 +00002471 case DeclarationName::Identifier:
2472 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor0212fd72010-09-21 16:06:22 +00002473 case DeclarationName::CXXDestructorName:
2474 case DeclarationName::CXXLiteralOperatorName:
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002475 Result.AddTypedTextChunk(
2476 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002477 break;
2478
2479 case DeclarationName::CXXUsingDirective:
2480 case DeclarationName::ObjCZeroArgSelector:
2481 case DeclarationName::ObjCOneArgSelector:
2482 case DeclarationName::ObjCMultiArgSelector:
2483 break;
2484
2485 case DeclarationName::CXXConstructorName: {
2486 CXXRecordDecl *Record = 0;
2487 QualType Ty = Name.getCXXNameType();
2488 if (const RecordType *RecordTy = Ty->getAs<RecordType>())
2489 Record = cast<CXXRecordDecl>(RecordTy->getDecl());
2490 else if (const InjectedClassNameType *InjectedTy
2491 = Ty->getAs<InjectedClassNameType>())
2492 Record = InjectedTy->getDecl();
2493 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002494 Result.AddTypedTextChunk(
2495 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002496 break;
2497 }
2498
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002499 Result.AddTypedTextChunk(
2500 Result.getAllocator().CopyString(Record->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002501 if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002502 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Douglas Gregor75acd922011-09-27 23:30:47 +00002503 AddTemplateParameterChunks(Context, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002504 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002505 }
2506 break;
2507 }
2508 }
2509}
2510
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002511CodeCompletionString *CodeCompletionResult::CreateCodeCompletionString(Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002512 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002513 CodeCompletionTUInfo &CCTUInfo,
2514 bool IncludeBriefComments) {
2515 return CreateCodeCompletionString(S.Context, S.PP, Allocator, CCTUInfo,
2516 IncludeBriefComments);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002517}
2518
Douglas Gregor3545ff42009-09-21 16:56:56 +00002519/// \brief If possible, create a new code completion string for the given
2520/// result.
2521///
2522/// \returns Either a new, heap-allocated code completion string describing
2523/// how to use this result, or NULL to indicate that the string or name of the
2524/// result is all that is needed.
2525CodeCompletionString *
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002526CodeCompletionResult::CreateCodeCompletionString(ASTContext &Ctx,
2527 Preprocessor &PP,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002528 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002529 CodeCompletionTUInfo &CCTUInfo,
2530 bool IncludeBriefComments) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002531 CodeCompletionBuilder Result(Allocator, CCTUInfo, Priority, Availability);
Douglas Gregor9eb77012009-11-07 00:00:49 +00002532
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002533 PrintingPolicy Policy = getCompletionPrintingPolicy(Ctx, PP);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002534 if (Kind == RK_Pattern) {
2535 Pattern->Priority = Priority;
2536 Pattern->Availability = Availability;
Douglas Gregor78254c82012-03-27 23:34:16 +00002537
2538 if (Declaration) {
2539 Result.addParentContext(Declaration->getDeclContext());
Douglas Gregor78254c82012-03-27 23:34:16 +00002540 Pattern->ParentName = Result.getParentName();
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002541 // Provide code completion comment for self.GetterName where
2542 // GetterName is the getter method for a property with name
2543 // different from the property name (declared via a property
2544 // getter attribute.
2545 const NamedDecl *ND = Declaration;
2546 if (const ObjCMethodDecl *M = dyn_cast<ObjCMethodDecl>(ND))
2547 if (M->isPropertyAccessor())
2548 if (const ObjCPropertyDecl *PDecl = M->findPropertyDecl())
2549 if (PDecl->getGetterName() == M->getSelector() &&
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002550 PDecl->getIdentifier() != M->getIdentifier()) {
2551 if (const RawComment *RC =
2552 Ctx.getRawCommentForAnyRedecl(M)) {
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002553 Result.addBriefComment(RC->getBriefText(Ctx));
2554 Pattern->BriefComment = Result.getBriefComment();
2555 }
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002556 else if (const RawComment *RC =
2557 Ctx.getRawCommentForAnyRedecl(PDecl)) {
2558 Result.addBriefComment(RC->getBriefText(Ctx));
2559 Pattern->BriefComment = Result.getBriefComment();
2560 }
2561 }
Douglas Gregor78254c82012-03-27 23:34:16 +00002562 }
2563
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002564 return Pattern;
2565 }
Douglas Gregorf09935f2009-12-01 05:55:20 +00002566
Douglas Gregorf09935f2009-12-01 05:55:20 +00002567 if (Kind == RK_Keyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002568 Result.AddTypedTextChunk(Keyword);
2569 return Result.TakeString();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002570 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002571
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002572 if (Kind == RK_Macro) {
Argyrios Kyrtzidis09c9e812013-02-20 00:54:57 +00002573 const MacroDirective *MD = PP.getMacroDirectiveHistory(Macro);
2574 assert(MD && "Not a macro?");
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00002575 const MacroInfo *MI = MD->getMacroInfo();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002576
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002577 Result.AddTypedTextChunk(
2578 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregorf09935f2009-12-01 05:55:20 +00002579
2580 if (!MI->isFunctionLike())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002581 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002582
2583 // Format a function-like macro with placeholders for the arguments.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002584 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor0c505312011-07-30 08:17:44 +00002585 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002586
2587 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
2588 if (MI->isC99Varargs()) {
2589 --AEnd;
2590
2591 if (A == AEnd) {
2592 Result.AddPlaceholderChunk("...");
2593 }
Douglas Gregor0c505312011-07-30 08:17:44 +00002594 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002595
Douglas Gregor0c505312011-07-30 08:17:44 +00002596 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002597 if (A != MI->arg_begin())
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002598 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3aa55262012-01-21 00:43:38 +00002599
2600 if (MI->isVariadic() && (A+1) == AEnd) {
Dylan Noblesmith2c1dd272012-02-05 02:13:05 +00002601 SmallString<32> Arg = (*A)->getName();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002602 if (MI->isC99Varargs())
2603 Arg += ", ...";
2604 else
2605 Arg += "...";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002606 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor3aa55262012-01-21 00:43:38 +00002607 break;
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002608 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002609
2610 // Non-variadic macros are simple.
2611 Result.AddPlaceholderChunk(
2612 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregor0c505312011-07-30 08:17:44 +00002613 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002614 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002615 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002616 }
2617
Douglas Gregorf64acca2010-05-25 21:41:55 +00002618 assert(Kind == RK_Declaration && "Missed a result kind?");
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002619 const NamedDecl *ND = Declaration;
Douglas Gregor78254c82012-03-27 23:34:16 +00002620 Result.addParentContext(ND->getDeclContext());
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002621
2622 if (IncludeBriefComments) {
2623 // Add documentation comment, if it exists.
Dmitri Gribenkoa43ec182012-08-11 00:51:43 +00002624 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(ND)) {
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002625 Result.addBriefComment(RC->getBriefText(Ctx));
Fariborz Jahanian15a0b552013-02-28 17:47:14 +00002626 }
2627 else if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2628 if (OMD->isPropertyAccessor())
2629 if (const ObjCPropertyDecl *PDecl = OMD->findPropertyDecl())
2630 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(PDecl))
2631 Result.addBriefComment(RC->getBriefText(Ctx));
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002632 }
2633
Douglas Gregor9eb77012009-11-07 00:00:49 +00002634 if (StartsNestedNameSpecifier) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002635 Result.AddTypedTextChunk(
2636 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002637 Result.AddTextChunk("::");
2638 return Result.TakeString();
Douglas Gregor9eb77012009-11-07 00:00:49 +00002639 }
Erik Verbruggen98ea7f62011-10-14 15:31:08 +00002640
2641 for (Decl::attr_iterator i = ND->attr_begin(); i != ND->attr_end(); ++i) {
2642 if (AnnotateAttr *Attr = dyn_cast_or_null<AnnotateAttr>(*i)) {
2643 Result.AddAnnotation(Result.getAllocator().CopyString(Attr->getAnnotation()));
2644 }
2645 }
Douglas Gregor9eb77012009-11-07 00:00:49 +00002646
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002647 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002648
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002649 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002650 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002651 Ctx, Policy);
2652 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002653 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002654 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002655 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002656 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002657 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002658 }
2659
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002660 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002661 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002662 Ctx, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002663 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002664 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002665
Douglas Gregor3545ff42009-09-21 16:56:56 +00002666 // Figure out which template parameters are deduced (or have default
2667 // arguments).
Benjamin Kramere0513cb2012-01-30 16:17:39 +00002668 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002669 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002670 unsigned LastDeducibleArgument;
2671 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2672 --LastDeducibleArgument) {
2673 if (!Deduced[LastDeducibleArgument - 1]) {
2674 // C++0x: Figure out if the template argument has a default. If so,
2675 // the user doesn't need to type this argument.
2676 // FIXME: We need to abstract template parameters better!
2677 bool HasDefaultArg = false;
2678 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002679 LastDeducibleArgument - 1);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002680 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2681 HasDefaultArg = TTP->hasDefaultArgument();
2682 else if (NonTypeTemplateParmDecl *NTTP
2683 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2684 HasDefaultArg = NTTP->hasDefaultArgument();
2685 else {
2686 assert(isa<TemplateTemplateParmDecl>(Param));
2687 HasDefaultArg
Douglas Gregor9eb77012009-11-07 00:00:49 +00002688 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002689 }
2690
2691 if (!HasDefaultArg)
2692 break;
2693 }
2694 }
2695
2696 if (LastDeducibleArgument) {
2697 // Some of the function template arguments cannot be deduced from a
2698 // function call, so we introduce an explicit template argument list
2699 // containing all of the arguments up to the first deducible argument.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002700 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002701 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor3545ff42009-09-21 16:56:56 +00002702 LastDeducibleArgument);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002703 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002704 }
2705
2706 // Add the function parameters
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002707 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002708 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002709 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002710 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002711 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002712 }
2713
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002714 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002715 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002716 Ctx, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002717 Result.AddTypedTextChunk(
2718 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002719 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002720 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002721 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002722 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002723 }
2724
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002725 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregord3c5d792009-11-17 16:44:22 +00002726 Selector Sel = Method->getSelector();
2727 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002728 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002729 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002730 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002731 }
2732
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002733 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregor1b605f72009-11-19 01:08:35 +00002734 SelName += ':';
2735 if (StartParameter == 0)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002736 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002737 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002738 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002739
2740 // If there is only one parameter, and we're past it, add an empty
2741 // typed-text chunk since there is nothing to type.
2742 if (Method->param_size() == 1)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002743 Result.AddTypedTextChunk("");
Douglas Gregor1b605f72009-11-19 01:08:35 +00002744 }
Douglas Gregord3c5d792009-11-17 16:44:22 +00002745 unsigned Idx = 0;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002746 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2747 PEnd = Method->param_end();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002748 P != PEnd; (void)++P, ++Idx) {
2749 if (Idx > 0) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00002750 std::string Keyword;
2751 if (Idx > StartParameter)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002752 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregord3c5d792009-11-17 16:44:22 +00002753 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramer632500c2011-07-26 16:59:25 +00002754 Keyword += II->getName();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002755 Keyword += ":";
Douglas Gregor95887f92010-07-08 23:20:03 +00002756 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002757 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002758 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002759 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002760 }
Douglas Gregor1b605f72009-11-19 01:08:35 +00002761
2762 // If we're before the starting parameter, skip the placeholder.
2763 if (Idx < StartParameter)
2764 continue;
Douglas Gregord3c5d792009-11-17 16:44:22 +00002765
2766 std::string Arg;
Douglas Gregore90dd002010-08-24 16:15:59 +00002767
2768 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002769 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregore90dd002010-08-24 16:15:59 +00002770 else {
John McCall31168b02011-06-15 23:02:42 +00002771 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor8f08d742011-07-30 07:55:26 +00002772 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2773 + Arg + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002774 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregor981a0c42010-08-29 19:47:46 +00002775 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramer632500c2011-07-26 16:59:25 +00002776 Arg += II->getName();
Douglas Gregore90dd002010-08-24 16:15:59 +00002777 }
2778
Douglas Gregor400f5972010-08-31 05:13:43 +00002779 if (Method->isVariadic() && (P + 1) == PEnd)
2780 Arg += ", ...";
2781
Douglas Gregor95887f92010-07-08 23:20:03 +00002782 if (DeclaringEntity)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002783 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor95887f92010-07-08 23:20:03 +00002784 else if (AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002785 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8537c52009-11-19 07:41:15 +00002786 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002787 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002788 }
2789
Douglas Gregor04c5f972009-12-23 00:21:46 +00002790 if (Method->isVariadic()) {
Douglas Gregor400f5972010-08-31 05:13:43 +00002791 if (Method->param_size() == 0) {
2792 if (DeclaringEntity)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002793 Result.AddTextChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002794 else if (AllParametersAreInformative)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002795 Result.AddInformativeChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002796 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002797 Result.AddPlaceholderChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002798 }
Douglas Gregordbb71db2010-08-23 23:51:41 +00002799
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002800 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor04c5f972009-12-23 00:21:46 +00002801 }
2802
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002803 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002804 }
2805
Douglas Gregorf09935f2009-12-01 05:55:20 +00002806 if (Qualifier)
Douglas Gregor5bf52692009-09-22 23:15:58 +00002807 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002808 Ctx, Policy);
Douglas Gregorf09935f2009-12-01 05:55:20 +00002809
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002810 Result.AddTypedTextChunk(
2811 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002812 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002813}
2814
Douglas Gregorf0f51982009-09-23 00:34:09 +00002815CodeCompletionString *
2816CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2817 unsigned CurrentArg,
Douglas Gregor36e3b5c2010-10-11 21:37:58 +00002818 Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002819 CodeCompletionAllocator &Allocator,
2820 CodeCompletionTUInfo &CCTUInfo) const {
Douglas Gregor75acd922011-09-27 23:30:47 +00002821 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCall31168b02011-06-15 23:02:42 +00002822
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002823 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002824 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002825 FunctionDecl *FDecl = getFunction();
Douglas Gregor75acd922011-09-27 23:30:47 +00002826 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002827 const FunctionProtoType *Proto
2828 = dyn_cast<FunctionProtoType>(getFunctionType());
2829 if (!FDecl && !Proto) {
2830 // Function without a prototype. Just give the return type and a
2831 // highlighted ellipsis.
2832 const FunctionType *FT = getFunctionType();
Alp Toker314cc812014-01-25 16:55:45 +00002833 Result.AddTextChunk(GetCompletionTypeString(FT->getReturnType(), S.Context,
2834 Policy, Result.getAllocator()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002835 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2836 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2837 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002838 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002839 }
2840
2841 if (FDecl)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002842 Result.AddTextChunk(
2843 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002844 else
Alp Toker314cc812014-01-25 16:55:45 +00002845 Result.AddTextChunk(Result.getAllocator().CopyString(
2846 Proto->getReturnType().getAsString(Policy)));
2847
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002848 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Alp Toker9cacbab2014-01-20 20:26:09 +00002849 unsigned NumParams = FDecl ? FDecl->getNumParams() : Proto->getNumParams();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002850 for (unsigned I = 0; I != NumParams; ++I) {
2851 if (I)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002852 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002853
2854 std::string ArgString;
2855 QualType ArgType;
2856
2857 if (FDecl) {
2858 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2859 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2860 } else {
Alp Toker9cacbab2014-01-20 20:26:09 +00002861 ArgType = Proto->getParamType(I);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002862 }
2863
John McCall31168b02011-06-15 23:02:42 +00002864 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002865
2866 if (I == CurrentArg)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002867 Result.AddChunk(CodeCompletionString::CK_CurrentParameter,
2868 Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002869 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002870 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002871 }
2872
2873 if (Proto && Proto->isVariadic()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002874 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002875 if (CurrentArg < NumParams)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002876 Result.AddTextChunk("...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002877 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002878 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002879 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002880 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002881
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002882 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002883}
2884
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002885unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002886 const LangOptions &LangOpts,
Douglas Gregor6e240332010-08-16 16:18:59 +00002887 bool PreferredTypeIsPointer) {
2888 unsigned Priority = CCP_Macro;
2889
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002890 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2891 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2892 MacroName.equals("Nil")) {
Douglas Gregor6e240332010-08-16 16:18:59 +00002893 Priority = CCP_Constant;
2894 if (PreferredTypeIsPointer)
2895 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002896 }
2897 // Treat "YES", "NO", "true", and "false" as constants.
2898 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2899 MacroName.equals("true") || MacroName.equals("false"))
2900 Priority = CCP_Constant;
2901 // Treat "bool" as a type.
2902 else if (MacroName.equals("bool"))
2903 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2904
Douglas Gregor6e240332010-08-16 16:18:59 +00002905
2906 return Priority;
2907}
2908
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002909CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002910 if (!D)
2911 return CXCursor_UnexposedDecl;
2912
2913 switch (D->getKind()) {
2914 case Decl::Enum: return CXCursor_EnumDecl;
2915 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2916 case Decl::Field: return CXCursor_FieldDecl;
2917 case Decl::Function:
2918 return CXCursor_FunctionDecl;
2919 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2920 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002921 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregordeafd0b2011-12-27 22:43:10 +00002922
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002923 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002924 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2925 case Decl::ObjCMethod:
2926 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2927 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2928 case Decl::CXXMethod: return CXCursor_CXXMethod;
2929 case Decl::CXXConstructor: return CXCursor_Constructor;
2930 case Decl::CXXDestructor: return CXCursor_Destructor;
2931 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2932 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002933 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002934 case Decl::ParmVar: return CXCursor_ParmDecl;
2935 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smithdda56e42011-04-15 14:24:37 +00002936 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002937 case Decl::Var: return CXCursor_VarDecl;
2938 case Decl::Namespace: return CXCursor_Namespace;
2939 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2940 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2941 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2942 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2943 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2944 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis12afd702011-09-30 17:58:23 +00002945 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002946 case Decl::ClassTemplatePartialSpecialization:
2947 return CXCursor_ClassTemplatePartialSpecialization;
2948 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor3e653b32012-04-30 23:41:16 +00002949 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002950
2951 case Decl::Using:
2952 case Decl::UnresolvedUsingValue:
2953 case Decl::UnresolvedUsingTypename:
2954 return CXCursor_UsingDeclaration;
2955
Douglas Gregor4cd65962011-06-03 23:08:58 +00002956 case Decl::ObjCPropertyImpl:
2957 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2958 case ObjCPropertyImplDecl::Dynamic:
2959 return CXCursor_ObjCDynamicDecl;
2960
2961 case ObjCPropertyImplDecl::Synthesize:
2962 return CXCursor_ObjCSynthesizeDecl;
2963 }
Argyrios Kyrtzidis50e5b1d2012-10-05 00:22:24 +00002964
2965 case Decl::Import:
2966 return CXCursor_ModuleImportDecl;
Douglas Gregor4cd65962011-06-03 23:08:58 +00002967
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002968 default:
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002969 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002970 switch (TD->getTagKind()) {
Joao Matosdc86f942012-08-31 18:45:21 +00002971 case TTK_Interface: // fall through
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002972 case TTK_Struct: return CXCursor_StructDecl;
2973 case TTK_Class: return CXCursor_ClassDecl;
2974 case TTK_Union: return CXCursor_UnionDecl;
2975 case TTK_Enum: return CXCursor_EnumDecl;
2976 }
2977 }
2978 }
2979
2980 return CXCursor_UnexposedDecl;
2981}
2982
Douglas Gregor55b037b2010-07-08 20:55:51 +00002983static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor8cb17462012-10-09 16:01:50 +00002984 bool IncludeUndefined,
Douglas Gregor55b037b2010-07-08 20:55:51 +00002985 bool TargetTypeIsPointer = false) {
John McCall276321a2010-08-25 06:19:51 +00002986 typedef CodeCompletionResult Result;
Douglas Gregor55b037b2010-07-08 20:55:51 +00002987
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002988 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002989
Douglas Gregor9eb77012009-11-07 00:00:49 +00002990 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2991 MEnd = PP.macro_end();
Douglas Gregor55b037b2010-07-08 20:55:51 +00002992 M != MEnd; ++M) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00002993 if (IncludeUndefined || M->first->hasMacroDefinition())
2994 Results.AddResult(Result(M->first,
Douglas Gregor6e240332010-08-16 16:18:59 +00002995 getMacroUsagePriority(M->first->getName(),
David Blaikiebbafb8a2012-03-11 07:00:24 +00002996 PP.getLangOpts(),
Douglas Gregor6e240332010-08-16 16:18:59 +00002997 TargetTypeIsPointer)));
Douglas Gregor55b037b2010-07-08 20:55:51 +00002998 }
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002999
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003000 Results.ExitScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003001
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003002}
3003
Douglas Gregorce0e8562010-08-23 21:54:33 +00003004static void AddPrettyFunctionResults(const LangOptions &LangOpts,
3005 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003006 typedef CodeCompletionResult Result;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003007
3008 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003009
Douglas Gregorce0e8562010-08-23 21:54:33 +00003010 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
3011 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith2bf7fdb2013-01-02 11:42:31 +00003012 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregorce0e8562010-08-23 21:54:33 +00003013 Results.AddResult(Result("__func__", CCP_Constant));
3014 Results.ExitScope();
3015}
3016
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003017static void HandleCodeCompleteResults(Sema *S,
3018 CodeCompleteConsumer *CodeCompleter,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003019 CodeCompletionContext Context,
John McCall276321a2010-08-25 06:19:51 +00003020 CodeCompletionResult *Results,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003021 unsigned NumResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00003022 if (CodeCompleter)
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003023 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor3545ff42009-09-21 16:56:56 +00003024}
3025
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003026static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3027 Sema::ParserCompletionContext PCC) {
3028 switch (PCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00003029 case Sema::PCC_Namespace:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003030 return CodeCompletionContext::CCC_TopLevel;
3031
John McCallfaf5fb42010-08-26 23:41:50 +00003032 case Sema::PCC_Class:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003033 return CodeCompletionContext::CCC_ClassStructUnion;
3034
John McCallfaf5fb42010-08-26 23:41:50 +00003035 case Sema::PCC_ObjCInterface:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003036 return CodeCompletionContext::CCC_ObjCInterface;
3037
John McCallfaf5fb42010-08-26 23:41:50 +00003038 case Sema::PCC_ObjCImplementation:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003039 return CodeCompletionContext::CCC_ObjCImplementation;
3040
John McCallfaf5fb42010-08-26 23:41:50 +00003041 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003042 return CodeCompletionContext::CCC_ObjCIvarList;
3043
John McCallfaf5fb42010-08-26 23:41:50 +00003044 case Sema::PCC_Template:
3045 case Sema::PCC_MemberTemplate:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003046 if (S.CurContext->isFileContext())
3047 return CodeCompletionContext::CCC_TopLevel;
David Blaikie8a40f702012-01-17 06:56:22 +00003048 if (S.CurContext->isRecord())
Douglas Gregor0ac41382010-09-23 23:01:17 +00003049 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie8a40f702012-01-17 06:56:22 +00003050 return CodeCompletionContext::CCC_Other;
Douglas Gregor0ac41382010-09-23 23:01:17 +00003051
John McCallfaf5fb42010-08-26 23:41:50 +00003052 case Sema::PCC_RecoveryInFunction:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003053 return CodeCompletionContext::CCC_Recovery;
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003054
John McCallfaf5fb42010-08-26 23:41:50 +00003055 case Sema::PCC_ForInit:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003056 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3057 S.getLangOpts().ObjC1)
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003058 return CodeCompletionContext::CCC_ParenthesizedExpression;
3059 else
3060 return CodeCompletionContext::CCC_Expression;
3061
3062 case Sema::PCC_Expression:
John McCallfaf5fb42010-08-26 23:41:50 +00003063 case Sema::PCC_Condition:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003064 return CodeCompletionContext::CCC_Expression;
3065
John McCallfaf5fb42010-08-26 23:41:50 +00003066 case Sema::PCC_Statement:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003067 return CodeCompletionContext::CCC_Statement;
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003068
John McCallfaf5fb42010-08-26 23:41:50 +00003069 case Sema::PCC_Type:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003070 return CodeCompletionContext::CCC_Type;
Douglas Gregor5e35d592010-09-14 23:59:36 +00003071
3072 case Sema::PCC_ParenthesizedExpression:
3073 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor80039242011-02-15 20:33:25 +00003074
3075 case Sema::PCC_LocalDeclarationSpecifiers:
3076 return CodeCompletionContext::CCC_Type;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003077 }
David Blaikie8a40f702012-01-17 06:56:22 +00003078
3079 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003080}
3081
Douglas Gregorac322ec2010-08-27 21:18:54 +00003082/// \brief If we're in a C++ virtual member function, add completion results
3083/// that invoke the functions we override, since it's common to invoke the
3084/// overridden function as well as adding new functionality.
3085///
3086/// \param S The semantic analysis object for which we are generating results.
3087///
3088/// \param InContext This context in which the nested-name-specifier preceding
3089/// the code-completion point
3090static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3091 ResultBuilder &Results) {
3092 // Look through blocks.
3093 DeclContext *CurContext = S.CurContext;
3094 while (isa<BlockDecl>(CurContext))
3095 CurContext = CurContext->getParent();
3096
3097
3098 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3099 if (!Method || !Method->isVirtual())
3100 return;
3101
3102 // We need to have names for all of the parameters, if we're going to
3103 // generate a forwarding call.
3104 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3105 PEnd = Method->param_end();
3106 P != PEnd;
3107 ++P) {
3108 if (!(*P)->getDeclName())
3109 return;
3110 }
3111
Douglas Gregor75acd922011-09-27 23:30:47 +00003112 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003113 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3114 MEnd = Method->end_overridden_methods();
3115 M != MEnd; ++M) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003116 CodeCompletionBuilder Builder(Results.getAllocator(),
3117 Results.getCodeCompletionTUInfo());
Dmitri Gribenko6cfb1532013-02-14 13:53:30 +00003118 const CXXMethodDecl *Overridden = *M;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003119 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3120 continue;
3121
3122 // If we need a nested-name-specifier, add one now.
3123 if (!InContext) {
3124 NestedNameSpecifier *NNS
3125 = getRequiredQualification(S.Context, CurContext,
3126 Overridden->getDeclContext());
3127 if (NNS) {
3128 std::string Str;
3129 llvm::raw_string_ostream OS(Str);
Douglas Gregor75acd922011-09-27 23:30:47 +00003130 NNS->print(OS, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003131 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003132 }
3133 } else if (!InContext->Equals(Overridden->getDeclContext()))
3134 continue;
3135
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003136 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003137 Overridden->getNameAsString()));
3138 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003139 bool FirstParam = true;
3140 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3141 PEnd = Method->param_end();
3142 P != PEnd; ++P) {
3143 if (FirstParam)
3144 FirstParam = false;
3145 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003146 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003147
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003148 Builder.AddPlaceholderChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003149 (*P)->getIdentifier()->getName()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003150 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003151 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3152 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorac322ec2010-08-27 21:18:54 +00003153 CCP_SuperCompletion,
Douglas Gregor78254c82012-03-27 23:34:16 +00003154 CXCursor_CXXMethod,
3155 CXAvailability_Available,
3156 Overridden));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003157 Results.Ignore(Overridden);
3158 }
3159}
3160
Douglas Gregor07f43572012-01-29 18:15:03 +00003161void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3162 ModuleIdPath Path) {
3163 typedef CodeCompletionResult Result;
3164 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003165 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor07f43572012-01-29 18:15:03 +00003166 CodeCompletionContext::CCC_Other);
3167 Results.EnterNewScope();
3168
3169 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003170 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor07f43572012-01-29 18:15:03 +00003171 typedef CodeCompletionResult Result;
3172 if (Path.empty()) {
3173 // Enumerate all top-level modules.
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003174 SmallVector<Module *, 8> Modules;
Douglas Gregor07f43572012-01-29 18:15:03 +00003175 PP.getHeaderSearchInfo().collectAllModules(Modules);
3176 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3177 Builder.AddTypedTextChunk(
3178 Builder.getAllocator().CopyString(Modules[I]->Name));
3179 Results.AddResult(Result(Builder.TakeString(),
3180 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003181 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003182 Modules[I]->isAvailable()
3183 ? CXAvailability_Available
3184 : CXAvailability_NotAvailable));
3185 }
Daniel Jasper07e6c402013-08-05 20:26:17 +00003186 } else if (getLangOpts().Modules) {
Douglas Gregor07f43572012-01-29 18:15:03 +00003187 // Load the named module.
3188 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3189 Module::AllVisible,
3190 /*IsInclusionDirective=*/false);
3191 // Enumerate submodules.
3192 if (Mod) {
3193 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3194 SubEnd = Mod->submodule_end();
3195 Sub != SubEnd; ++Sub) {
3196
3197 Builder.AddTypedTextChunk(
3198 Builder.getAllocator().CopyString((*Sub)->Name));
3199 Results.AddResult(Result(Builder.TakeString(),
3200 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003201 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003202 (*Sub)->isAvailable()
3203 ? CXAvailability_Available
3204 : CXAvailability_NotAvailable));
3205 }
3206 }
3207 }
3208 Results.ExitScope();
3209 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3210 Results.data(),Results.size());
3211}
3212
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003213void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003214 ParserCompletionContext CompletionContext) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003215 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003216 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003217 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003218 Results.EnterNewScope();
Douglas Gregor50832e02010-09-20 22:39:41 +00003219
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003220 // Determine how to filter results, e.g., so that the names of
3221 // values (functions, enumerators, function templates, etc.) are
3222 // only allowed where we can have an expression.
3223 switch (CompletionContext) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003224 case PCC_Namespace:
3225 case PCC_Class:
3226 case PCC_ObjCInterface:
3227 case PCC_ObjCImplementation:
3228 case PCC_ObjCInstanceVariableList:
3229 case PCC_Template:
3230 case PCC_MemberTemplate:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003231 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003232 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003233 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3234 break;
3235
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003236 case PCC_Statement:
Douglas Gregor5e35d592010-09-14 23:59:36 +00003237 case PCC_ParenthesizedExpression:
Douglas Gregor4d755e82010-08-24 23:58:17 +00003238 case PCC_Expression:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003239 case PCC_ForInit:
3240 case PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003241 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor70febae2010-05-28 00:49:12 +00003242 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3243 else
3244 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003245
David Blaikiebbafb8a2012-03-11 07:00:24 +00003246 if (getLangOpts().CPlusPlus)
Douglas Gregorac322ec2010-08-27 21:18:54 +00003247 MaybeAddOverrideCalls(*this, /*InContext=*/0, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003248 break;
Douglas Gregor6da3db42010-05-25 05:58:43 +00003249
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003250 case PCC_RecoveryInFunction:
Douglas Gregor6da3db42010-05-25 05:58:43 +00003251 // Unfiltered
3252 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003253 }
3254
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003255 // If we are in a C++ non-static member function, check the qualifiers on
3256 // the member function to filter/prioritize the results list.
3257 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3258 if (CurMethod->isInstance())
3259 Results.setObjectTypeQualifiers(
3260 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3261
Douglas Gregorc580c522010-01-14 01:09:38 +00003262 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003263 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3264 CodeCompleter->includeGlobals());
Douglas Gregor92253692009-12-07 09:54:55 +00003265
Douglas Gregorf98e6a22010-01-13 23:51:12 +00003266 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor92253692009-12-07 09:54:55 +00003267 Results.ExitScope();
3268
Douglas Gregorce0e8562010-08-23 21:54:33 +00003269 switch (CompletionContext) {
Douglas Gregor5e35d592010-09-14 23:59:36 +00003270 case PCC_ParenthesizedExpression:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003271 case PCC_Expression:
3272 case PCC_Statement:
3273 case PCC_RecoveryInFunction:
3274 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00003275 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003276 break;
3277
3278 case PCC_Namespace:
3279 case PCC_Class:
3280 case PCC_ObjCInterface:
3281 case PCC_ObjCImplementation:
3282 case PCC_ObjCInstanceVariableList:
3283 case PCC_Template:
3284 case PCC_MemberTemplate:
3285 case PCC_ForInit:
3286 case PCC_Condition:
3287 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003288 case PCC_LocalDeclarationSpecifiers:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003289 break;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003290 }
3291
Douglas Gregor9eb77012009-11-07 00:00:49 +00003292 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003293 AddMacroResults(PP, Results, false);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003294
Douglas Gregor50832e02010-09-20 22:39:41 +00003295 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003296 Results.data(),Results.size());
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00003297}
3298
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003299static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3300 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003301 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00003302 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003303 bool IsSuper,
3304 ResultBuilder &Results);
3305
3306void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3307 bool AllowNonIdentifiers,
3308 bool AllowNestedNameSpecifiers) {
John McCall276321a2010-08-25 06:19:51 +00003309 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003310 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003311 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003312 AllowNestedNameSpecifiers
3313 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3314 : CodeCompletionContext::CCC_Name);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003315 Results.EnterNewScope();
3316
3317 // Type qualifiers can come after names.
3318 Results.AddResult(Result("const"));
3319 Results.AddResult(Result("volatile"));
David Blaikiebbafb8a2012-03-11 07:00:24 +00003320 if (getLangOpts().C99)
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003321 Results.AddResult(Result("restrict"));
3322
David Blaikiebbafb8a2012-03-11 07:00:24 +00003323 if (getLangOpts().CPlusPlus) {
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003324 if (AllowNonIdentifiers) {
3325 Results.AddResult(Result("operator"));
3326 }
3327
3328 // Add nested-name-specifiers.
3329 if (AllowNestedNameSpecifiers) {
3330 Results.allowNestedNameSpecifiers();
Douglas Gregor0ac41382010-09-23 23:01:17 +00003331 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003332 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3333 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3334 CodeCompleter->includeGlobals());
Douglas Gregor0ac41382010-09-23 23:01:17 +00003335 Results.setFilter(0);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003336 }
3337 }
3338 Results.ExitScope();
3339
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003340 // If we're in a context where we might have an expression (rather than a
3341 // declaration), and what we've seen so far is an Objective-C type that could
3342 // be a receiver of a class message, this may be a class message send with
3343 // the initial opening bracket '[' missing. Add appropriate completions.
3344 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003345 DS.getParsedSpecifiers() == DeclSpec::PQ_TypeSpecifier &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003346 DS.getTypeSpecType() == DeclSpec::TST_typename &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003347 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3348 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003349 !DS.isTypeAltiVecVector() &&
3350 S &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003351 (S->getFlags() & Scope::DeclScope) != 0 &&
3352 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3353 Scope::FunctionPrototypeScope |
3354 Scope::AtCatchScope)) == 0) {
3355 ParsedType T = DS.getRepAsType();
3356 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003357 AddClassMessageCompletions(*this, S, T, None, false, false, Results);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003358 }
3359
Douglas Gregor56ccce02010-08-24 04:59:56 +00003360 // Note that we intentionally suppress macro results here, since we do not
3361 // encourage using macros to produce the names of entities.
3362
Douglas Gregor0ac41382010-09-23 23:01:17 +00003363 HandleCodeCompleteResults(this, CodeCompleter,
3364 Results.getCompletionContext(),
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003365 Results.data(), Results.size());
3366}
3367
Douglas Gregor68762e72010-08-23 21:17:50 +00003368struct Sema::CodeCompleteExpressionData {
3369 CodeCompleteExpressionData(QualType PreferredType = QualType())
3370 : PreferredType(PreferredType), IntegralConstantExpression(false),
3371 ObjCCollection(false) { }
3372
3373 QualType PreferredType;
3374 bool IntegralConstantExpression;
3375 bool ObjCCollection;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003376 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregor68762e72010-08-23 21:17:50 +00003377};
3378
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003379/// \brief Perform code-completion in an expression context when we know what
3380/// type we're looking for.
Douglas Gregor68762e72010-08-23 21:17:50 +00003381void Sema::CodeCompleteExpression(Scope *S,
3382 const CodeCompleteExpressionData &Data) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003383 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003384 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003385 CodeCompletionContext::CCC_Expression);
Douglas Gregor68762e72010-08-23 21:17:50 +00003386 if (Data.ObjCCollection)
3387 Results.setFilter(&ResultBuilder::IsObjCCollection);
3388 else if (Data.IntegralConstantExpression)
Douglas Gregor85b50632010-07-28 21:50:18 +00003389 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003390 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003391 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3392 else
3393 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregor68762e72010-08-23 21:17:50 +00003394
3395 if (!Data.PreferredType.isNull())
3396 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3397
3398 // Ignore any declarations that we were told that we don't care about.
3399 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3400 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003401
3402 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003403 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3404 CodeCompleter->includeGlobals());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003405
3406 Results.EnterNewScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003407 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003408 Results.ExitScope();
3409
Douglas Gregor55b037b2010-07-08 20:55:51 +00003410 bool PreferredTypeIsPointer = false;
Douglas Gregor68762e72010-08-23 21:17:50 +00003411 if (!Data.PreferredType.isNull())
3412 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3413 || Data.PreferredType->isMemberPointerType()
3414 || Data.PreferredType->isBlockPointerType();
Douglas Gregor55b037b2010-07-08 20:55:51 +00003415
Douglas Gregorce0e8562010-08-23 21:54:33 +00003416 if (S->getFnParent() &&
3417 !Data.ObjCCollection &&
3418 !Data.IntegralConstantExpression)
David Blaikiebbafb8a2012-03-11 07:00:24 +00003419 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003420
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003421 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003422 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003423 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor68762e72010-08-23 21:17:50 +00003424 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3425 Data.PreferredType),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003426 Results.data(),Results.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003427}
3428
Douglas Gregoreda7e542010-09-18 01:28:11 +00003429void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3430 if (E.isInvalid())
3431 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003432 else if (getLangOpts().ObjC1)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003433 CodeCompleteObjCInstanceMessage(S, E.take(), None, false);
Douglas Gregored0b69d2010-09-15 16:23:04 +00003434}
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003435
Douglas Gregorb888acf2010-12-09 23:01:55 +00003436/// \brief The set of properties that have already been added, referenced by
3437/// property name.
3438typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3439
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003440/// \brief Retrieve the container definition, if any?
3441static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3442 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3443 if (Interface->hasDefinition())
3444 return Interface->getDefinition();
3445
3446 return Interface;
3447 }
3448
3449 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3450 if (Protocol->hasDefinition())
3451 return Protocol->getDefinition();
3452
3453 return Protocol;
3454 }
3455 return Container;
3456}
3457
3458static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor5d649882009-11-18 22:32:06 +00003459 bool AllowCategories,
Douglas Gregor95147142011-05-05 15:50:42 +00003460 bool AllowNullaryMethods,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003461 DeclContext *CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003462 AddedPropertiesSet &AddedProperties,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003463 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003464 typedef CodeCompletionResult Result;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003465
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003466 // Retrieve the definition.
3467 Container = getContainerDef(Container);
3468
Douglas Gregor9291bad2009-11-18 01:29:26 +00003469 // Add properties in this container.
3470 for (ObjCContainerDecl::prop_iterator P = Container->prop_begin(),
3471 PEnd = Container->prop_end();
3472 P != PEnd;
Douglas Gregorb888acf2010-12-09 23:01:55 +00003473 ++P) {
3474 if (AddedProperties.insert(P->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003475 Results.MaybeAddResult(Result(*P, Results.getBasePriority(*P), 0),
3476 CurContext);
Douglas Gregorb888acf2010-12-09 23:01:55 +00003477 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003478
Douglas Gregor95147142011-05-05 15:50:42 +00003479 // Add nullary methods
3480 if (AllowNullaryMethods) {
3481 ASTContext &Context = Container->getASTContext();
Douglas Gregor75acd922011-09-27 23:30:47 +00003482 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Douglas Gregor95147142011-05-05 15:50:42 +00003483 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
3484 MEnd = Container->meth_end();
3485 M != MEnd; ++M) {
3486 if (M->getSelector().isUnarySelector())
3487 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3488 if (AddedProperties.insert(Name)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003489 CodeCompletionBuilder Builder(Results.getAllocator(),
3490 Results.getCodeCompletionTUInfo());
David Blaikie40ed2972012-06-06 20:45:41 +00003491 AddResultTypeChunk(Context, Policy, *M, Builder);
Douglas Gregor95147142011-05-05 15:50:42 +00003492 Builder.AddTypedTextChunk(
3493 Results.getAllocator().CopyString(Name->getName()));
3494
David Blaikie40ed2972012-06-06 20:45:41 +00003495 Results.MaybeAddResult(Result(Builder.TakeString(), *M,
Douglas Gregor78254c82012-03-27 23:34:16 +00003496 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor95147142011-05-05 15:50:42 +00003497 CurContext);
3498 }
3499 }
3500 }
3501
3502
Douglas Gregor9291bad2009-11-18 01:29:26 +00003503 // Add properties in referenced protocols.
3504 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3505 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
3506 PEnd = Protocol->protocol_end();
3507 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003508 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3509 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003510 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor5d649882009-11-18 22:32:06 +00003511 if (AllowCategories) {
3512 // Look through categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00003513 for (ObjCInterfaceDecl::known_categories_iterator
3514 Cat = IFace->known_categories_begin(),
3515 CatEnd = IFace->known_categories_end();
3516 Cat != CatEnd; ++Cat)
3517 AddObjCProperties(*Cat, AllowCategories, AllowNullaryMethods,
Douglas Gregor95147142011-05-05 15:50:42 +00003518 CurContext, AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00003519 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003520
3521 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003522 for (ObjCInterfaceDecl::all_protocol_iterator
3523 I = IFace->all_referenced_protocol_begin(),
3524 E = IFace->all_referenced_protocol_end(); I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003525 AddObjCProperties(*I, AllowCategories, AllowNullaryMethods, CurContext,
3526 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003527
3528 // Look in the superclass.
3529 if (IFace->getSuperClass())
Douglas Gregor95147142011-05-05 15:50:42 +00003530 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3531 AllowNullaryMethods, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003532 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003533 } else if (const ObjCCategoryDecl *Category
3534 = dyn_cast<ObjCCategoryDecl>(Container)) {
3535 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003536 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
3537 PEnd = Category->protocol_end();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003538 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003539 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3540 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003541 }
3542}
3543
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003544void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor2436e712009-09-17 21:32:03 +00003545 SourceLocation OpLoc,
3546 bool IsArrow) {
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003547 if (!Base || !CodeCompleter)
Douglas Gregor2436e712009-09-17 21:32:03 +00003548 return;
3549
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003550 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3551 if (ConvertedBase.isInvalid())
3552 return;
3553 Base = ConvertedBase.get();
3554
John McCall276321a2010-08-25 06:19:51 +00003555 typedef CodeCompletionResult Result;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003556
Douglas Gregor2436e712009-09-17 21:32:03 +00003557 QualType BaseType = Base->getType();
Douglas Gregor3545ff42009-09-21 16:56:56 +00003558
3559 if (IsArrow) {
3560 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3561 BaseType = Ptr->getPointeeType();
3562 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003563 /*Do nothing*/ ;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003564 else
3565 return;
3566 }
3567
Douglas Gregor21325842011-07-07 16:03:39 +00003568 enum CodeCompletionContext::Kind contextKind;
3569
3570 if (IsArrow) {
3571 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3572 }
3573 else {
3574 if (BaseType->isObjCObjectPointerType() ||
3575 BaseType->isObjCObjectOrInterfaceType()) {
3576 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3577 }
3578 else {
3579 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3580 }
3581 }
3582
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003583 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003584 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00003585 CodeCompletionContext(contextKind,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003586 BaseType),
3587 &ResultBuilder::IsMember);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003588 Results.EnterNewScope();
3589 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003590 // Indicate that we are performing a member access, and the cv-qualifiers
3591 // for the base object type.
3592 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3593
Douglas Gregor9291bad2009-11-18 01:29:26 +00003594 // Access to a C/C++ class, struct, or union.
Douglas Gregor6ae4c522010-01-14 03:21:49 +00003595 Results.allowNestedNameSpecifiers();
Douglas Gregor09bbc652010-01-14 15:47:35 +00003596 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003597 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3598 CodeCompleter->includeGlobals());
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003599
David Blaikiebbafb8a2012-03-11 07:00:24 +00003600 if (getLangOpts().CPlusPlus) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003601 if (!Results.empty()) {
3602 // The "template" keyword can follow "->" or "." in the grammar.
3603 // However, we only want to suggest the template keyword if something
3604 // is dependent.
3605 bool IsDependent = BaseType->isDependentType();
3606 if (!IsDependent) {
3607 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
Ted Kremenekc37877d2013-10-08 17:08:03 +00003608 if (DeclContext *Ctx = DepScope->getEntity()) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003609 IsDependent = Ctx->isDependentContext();
3610 break;
3611 }
3612 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003613
Douglas Gregor9291bad2009-11-18 01:29:26 +00003614 if (IsDependent)
Douglas Gregor78a21012010-01-14 16:01:26 +00003615 Results.AddResult(Result("template"));
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003616 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003617 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003618 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3619 // Objective-C property reference.
Douglas Gregorb888acf2010-12-09 23:01:55 +00003620 AddedPropertiesSet AddedProperties;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003621
3622 // Add property results based on our interface.
3623 const ObjCObjectPointerType *ObjCPtr
3624 = BaseType->getAsObjCInterfacePointerType();
3625 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor95147142011-05-05 15:50:42 +00003626 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3627 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003628 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003629
3630 // Add properties from the protocols in a qualified interface.
3631 for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(),
3632 E = ObjCPtr->qual_end();
3633 I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003634 AddObjCProperties(*I, true, /*AllowNullaryMethods=*/true, CurContext,
3635 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003636 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCall8b07ec22010-05-15 11:32:37 +00003637 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003638 // Objective-C instance variable access.
3639 ObjCInterfaceDecl *Class = 0;
3640 if (const ObjCObjectPointerType *ObjCPtr
3641 = BaseType->getAs<ObjCObjectPointerType>())
3642 Class = ObjCPtr->getInterfaceDecl();
3643 else
John McCall8b07ec22010-05-15 11:32:37 +00003644 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003645
3646 // Add all ivars from this class and its superclasses.
Douglas Gregor2b8162b2010-01-14 16:08:12 +00003647 if (Class) {
3648 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3649 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor39982192010-08-15 06:18:01 +00003650 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3651 CodeCompleter->includeGlobals());
Douglas Gregor9291bad2009-11-18 01:29:26 +00003652 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003653 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003654
3655 // FIXME: How do we cope with isa?
3656
3657 Results.ExitScope();
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003658
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003659 // Hand off the results found for code completion.
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003660 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003661 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003662 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00003663}
3664
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003665void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3666 if (!CodeCompleter)
3667 return;
3668
Douglas Gregor3545ff42009-09-21 16:56:56 +00003669 ResultBuilder::LookupFilter Filter = 0;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003670 enum CodeCompletionContext::Kind ContextKind
3671 = CodeCompletionContext::CCC_Other;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003672 switch ((DeclSpec::TST)TagSpec) {
3673 case DeclSpec::TST_enum:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003674 Filter = &ResultBuilder::IsEnum;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003675 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003676 break;
3677
3678 case DeclSpec::TST_union:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003679 Filter = &ResultBuilder::IsUnion;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003680 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003681 break;
3682
3683 case DeclSpec::TST_struct:
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003684 case DeclSpec::TST_class:
Joao Matosdc86f942012-08-31 18:45:21 +00003685 case DeclSpec::TST_interface:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003686 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003687 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003688 break;
3689
3690 default:
David Blaikie83d382b2011-09-23 05:06:16 +00003691 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003692 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003693
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003694 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3695 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00003696 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCalle87beb22010-04-23 18:46:30 +00003697
3698 // First pass: look for tags.
3699 Results.setFilter(Filter);
Douglas Gregor39982192010-08-15 06:18:01 +00003700 LookupVisibleDecls(S, LookupTagName, Consumer,
3701 CodeCompleter->includeGlobals());
John McCalle87beb22010-04-23 18:46:30 +00003702
Douglas Gregor39982192010-08-15 06:18:01 +00003703 if (CodeCompleter->includeGlobals()) {
3704 // Second pass: look for nested name specifiers.
3705 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3706 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3707 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003708
Douglas Gregor0ac41382010-09-23 23:01:17 +00003709 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003710 Results.data(),Results.size());
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003711}
3712
Douglas Gregor28c78432010-08-27 17:35:51 +00003713void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003714 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003715 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003716 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor28c78432010-08-27 17:35:51 +00003717 Results.EnterNewScope();
3718 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3719 Results.AddResult("const");
3720 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3721 Results.AddResult("volatile");
David Blaikiebbafb8a2012-03-11 07:00:24 +00003722 if (getLangOpts().C99 &&
Douglas Gregor28c78432010-08-27 17:35:51 +00003723 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3724 Results.AddResult("restrict");
Richard Smith8e1ac332013-03-28 01:55:44 +00003725 if (getLangOpts().C11 &&
3726 !(DS.getTypeQualifiers() & DeclSpec::TQ_atomic))
3727 Results.AddResult("_Atomic");
Douglas Gregor28c78432010-08-27 17:35:51 +00003728 Results.ExitScope();
3729 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003730 Results.getCompletionContext(),
Douglas Gregor28c78432010-08-27 17:35:51 +00003731 Results.data(), Results.size());
3732}
3733
Douglas Gregord328d572009-09-21 18:10:23 +00003734void Sema::CodeCompleteCase(Scope *S) {
John McCallaab3e412010-08-25 08:40:02 +00003735 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregord328d572009-09-21 18:10:23 +00003736 return;
John McCall5939b162011-08-06 07:30:58 +00003737
John McCallaab3e412010-08-25 08:40:02 +00003738 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCall5939b162011-08-06 07:30:58 +00003739 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3740 if (!type->isEnumeralType()) {
3741 CodeCompleteExpressionData Data(type);
Douglas Gregor68762e72010-08-23 21:17:50 +00003742 Data.IntegralConstantExpression = true;
3743 CodeCompleteExpression(S, Data);
Douglas Gregord328d572009-09-21 18:10:23 +00003744 return;
Douglas Gregor85b50632010-07-28 21:50:18 +00003745 }
Douglas Gregord328d572009-09-21 18:10:23 +00003746
3747 // Code-complete the cases of a switch statement over an enumeration type
3748 // by providing the list of
John McCall5939b162011-08-06 07:30:58 +00003749 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003750 if (EnumDecl *Def = Enum->getDefinition())
3751 Enum = Def;
Douglas Gregord328d572009-09-21 18:10:23 +00003752
3753 // Determine which enumerators we have already seen in the switch statement.
3754 // FIXME: Ideally, we would also be able to look *past* the code-completion
3755 // token, in case we are code-completing in the middle of the switch and not
3756 // at the end. However, we aren't able to do so at the moment.
3757 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Douglas Gregorf2510672009-09-21 19:57:38 +00003758 NestedNameSpecifier *Qualifier = 0;
Douglas Gregord328d572009-09-21 18:10:23 +00003759 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3760 SC = SC->getNextSwitchCase()) {
3761 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3762 if (!Case)
3763 continue;
3764
3765 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3766 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3767 if (EnumConstantDecl *Enumerator
3768 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3769 // We look into the AST of the case statement to determine which
3770 // enumerator was named. Alternatively, we could compute the value of
3771 // the integral constant expression, then compare it against the
3772 // values of each enumerator. However, value-based approach would not
3773 // work as well with C++ templates where enumerators declared within a
3774 // template are type- and value-dependent.
3775 EnumeratorsSeen.insert(Enumerator);
3776
Douglas Gregorf2510672009-09-21 19:57:38 +00003777 // If this is a qualified-id, keep track of the nested-name-specifier
3778 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregord328d572009-09-21 18:10:23 +00003779 //
3780 // switch (TagD.getKind()) {
3781 // case TagDecl::TK_enum:
3782 // break;
3783 // case XXX
3784 //
Douglas Gregorf2510672009-09-21 19:57:38 +00003785 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregord328d572009-09-21 18:10:23 +00003786 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3787 // TK_struct, and TK_class.
Douglas Gregor4bd90e52009-10-23 18:54:35 +00003788 Qualifier = DRE->getQualifier();
Douglas Gregord328d572009-09-21 18:10:23 +00003789 }
3790 }
3791
David Blaikiebbafb8a2012-03-11 07:00:24 +00003792 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorf2510672009-09-21 19:57:38 +00003793 // If there are no prior enumerators in C++, check whether we have to
3794 // qualify the names of the enumerators that we suggest, because they
3795 // may not be visible in this scope.
Douglas Gregord3cebdb2012-02-01 05:02:47 +00003796 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorf2510672009-09-21 19:57:38 +00003797 }
3798
Douglas Gregord328d572009-09-21 18:10:23 +00003799 // Add any enumerators that have not yet been mentioned.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003800 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003801 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003802 CodeCompletionContext::CCC_Expression);
Douglas Gregord328d572009-09-21 18:10:23 +00003803 Results.EnterNewScope();
3804 for (EnumDecl::enumerator_iterator E = Enum->enumerator_begin(),
3805 EEnd = Enum->enumerator_end();
3806 E != EEnd; ++E) {
David Blaikie40ed2972012-06-06 20:45:41 +00003807 if (EnumeratorsSeen.count(*E))
Douglas Gregord328d572009-09-21 18:10:23 +00003808 continue;
3809
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003810 CodeCompletionResult R(*E, CCP_EnumInCase, Qualifier);
Douglas Gregor3a69eaf2011-02-18 23:30:37 +00003811 Results.AddResult(R, CurContext, 0, false);
Douglas Gregord328d572009-09-21 18:10:23 +00003812 }
3813 Results.ExitScope();
Douglas Gregor285560922010-04-06 20:02:15 +00003814
Douglas Gregor21325842011-07-07 16:03:39 +00003815 //We need to make sure we're setting the right context,
3816 //so only say we include macros if the code completer says we do
3817 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3818 if (CodeCompleter->includeMacros()) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00003819 AddMacroResults(PP, Results, false);
Douglas Gregor21325842011-07-07 16:03:39 +00003820 kind = CodeCompletionContext::CCC_OtherWithMacros;
3821 }
3822
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003823 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00003824 kind,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003825 Results.data(),Results.size());
Douglas Gregord328d572009-09-21 18:10:23 +00003826}
3827
Douglas Gregorcabea402009-09-22 15:41:20 +00003828namespace {
3829 struct IsBetterOverloadCandidate {
3830 Sema &S;
John McCallbc077cf2010-02-08 23:07:23 +00003831 SourceLocation Loc;
Douglas Gregorcabea402009-09-22 15:41:20 +00003832
3833 public:
John McCallbc077cf2010-02-08 23:07:23 +00003834 explicit IsBetterOverloadCandidate(Sema &S, SourceLocation Loc)
3835 : S(S), Loc(Loc) { }
Douglas Gregorcabea402009-09-22 15:41:20 +00003836
3837 bool
3838 operator()(const OverloadCandidate &X, const OverloadCandidate &Y) const {
John McCall5c32be02010-08-24 20:38:10 +00003839 return isBetterOverloadCandidate(S, X, Y, Loc);
Douglas Gregorcabea402009-09-22 15:41:20 +00003840 }
3841 };
3842}
3843
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003844static bool anyNullArguments(ArrayRef<Expr *> Args) {
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003845 if (Args.size() && !Args.data())
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003846 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003847
3848 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003849 if (!Args[I])
3850 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003851
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003852 return false;
3853}
3854
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003855void Sema::CodeCompleteCall(Scope *S, Expr *FnIn, ArrayRef<Expr *> Args) {
Douglas Gregorcabea402009-09-22 15:41:20 +00003856 if (!CodeCompleter)
3857 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003858
3859 // When we're code-completing for a call, we fall back to ordinary
3860 // name code-completion whenever we can't produce specific
3861 // results. We may want to revisit this strategy in the future,
3862 // e.g., by merging the two kinds of results.
3863
Douglas Gregorcabea402009-09-22 15:41:20 +00003864 Expr *Fn = (Expr *)FnIn;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003865
Douglas Gregorcabea402009-09-22 15:41:20 +00003866 // Ignore type-dependent call expressions entirely.
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003867 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3868 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003869 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregorcabea402009-09-22 15:41:20 +00003870 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003871 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003872
John McCall57500772009-12-16 12:17:52 +00003873 // Build an overload candidate set based on the functions we find.
John McCallbc077cf2010-02-08 23:07:23 +00003874 SourceLocation Loc = Fn->getExprLoc();
3875 OverloadCandidateSet CandidateSet(Loc);
John McCall57500772009-12-16 12:17:52 +00003876
Douglas Gregorcabea402009-09-22 15:41:20 +00003877 // FIXME: What if we're calling something that isn't a function declaration?
3878 // FIXME: What if we're calling a pseudo-destructor?
3879 // FIXME: What if we're calling a member function?
3880
Douglas Gregorff59f672010-01-21 15:46:19 +00003881 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003882 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorff59f672010-01-21 15:46:19 +00003883
John McCall57500772009-12-16 12:17:52 +00003884 Expr *NakedFn = Fn->IgnoreParenCasts();
3885 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003886 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
John McCall57500772009-12-16 12:17:52 +00003887 /*PartialOverloading=*/ true);
3888 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3889 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorff59f672010-01-21 15:46:19 +00003890 if (FDecl) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00003891 if (!getLangOpts().CPlusPlus ||
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003892 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorff59f672010-01-21 15:46:19 +00003893 Results.push_back(ResultCandidate(FDecl));
3894 else
John McCallb89836b2010-01-26 01:37:31 +00003895 // FIXME: access?
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003896 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), Args,
3897 CandidateSet, false, /*PartialOverloading*/true);
Douglas Gregorff59f672010-01-21 15:46:19 +00003898 }
John McCall57500772009-12-16 12:17:52 +00003899 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003900
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003901 QualType ParamType;
3902
Douglas Gregorff59f672010-01-21 15:46:19 +00003903 if (!CandidateSet.empty()) {
3904 // Sort the overload candidate set by placing the best overloads first.
3905 std::stable_sort(CandidateSet.begin(), CandidateSet.end(),
John McCallbc077cf2010-02-08 23:07:23 +00003906 IsBetterOverloadCandidate(*this, Loc));
Douglas Gregorcabea402009-09-22 15:41:20 +00003907
Douglas Gregorff59f672010-01-21 15:46:19 +00003908 // Add the remaining viable overload candidates as code-completion reslults.
3909 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3910 CandEnd = CandidateSet.end();
3911 Cand != CandEnd; ++Cand) {
3912 if (Cand->Viable)
3913 Results.push_back(ResultCandidate(Cand->Function));
3914 }
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003915
3916 // From the viable candidates, try to determine the type of this parameter.
3917 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3918 if (const FunctionType *FType = Results[I].getFunctionType())
3919 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
Alp Toker9cacbab2014-01-20 20:26:09 +00003920 if (Args.size() < Proto->getNumParams()) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003921 if (ParamType.isNull())
Alp Toker9cacbab2014-01-20 20:26:09 +00003922 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003923 else if (!Context.hasSameUnqualifiedType(
Alp Toker9cacbab2014-01-20 20:26:09 +00003924 ParamType.getNonReferenceType(),
3925 Proto->getParamType(Args.size())
3926 .getNonReferenceType())) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003927 ParamType = QualType();
3928 break;
3929 }
3930 }
3931 }
3932 } else {
3933 // Try to determine the parameter type from the type of the expression
3934 // being called.
3935 QualType FunctionType = Fn->getType();
3936 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3937 FunctionType = Ptr->getPointeeType();
3938 else if (const BlockPointerType *BlockPtr
3939 = FunctionType->getAs<BlockPointerType>())
3940 FunctionType = BlockPtr->getPointeeType();
3941 else if (const MemberPointerType *MemPtr
3942 = FunctionType->getAs<MemberPointerType>())
3943 FunctionType = MemPtr->getPointeeType();
3944
3945 if (const FunctionProtoType *Proto
3946 = FunctionType->getAs<FunctionProtoType>()) {
Alp Toker9cacbab2014-01-20 20:26:09 +00003947 if (Args.size() < Proto->getNumParams())
3948 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003949 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003950 }
Douglas Gregor3ef59522009-12-11 19:06:04 +00003951
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003952 if (ParamType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003953 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003954 else
3955 CodeCompleteExpression(S, ParamType);
3956
Douglas Gregorc01890e2010-04-06 20:19:47 +00003957 if (!Results.empty())
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003958 CodeCompleter->ProcessOverloadCandidates(*this, Args.size(), Results.data(),
Douglas Gregor3ef59522009-12-11 19:06:04 +00003959 Results.size());
Douglas Gregorcabea402009-09-22 15:41:20 +00003960}
3961
John McCall48871652010-08-21 09:40:31 +00003962void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3963 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003964 if (!VD) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003965 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003966 return;
3967 }
3968
3969 CodeCompleteExpression(S, VD->getType());
3970}
3971
3972void Sema::CodeCompleteReturn(Scope *S) {
3973 QualType ResultType;
3974 if (isa<BlockDecl>(CurContext)) {
3975 if (BlockScopeInfo *BSI = getCurBlock())
3976 ResultType = BSI->ReturnType;
3977 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00003978 ResultType = Function->getReturnType();
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003979 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00003980 ResultType = Method->getReturnType();
3981
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003982 if (ResultType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003983 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003984 else
3985 CodeCompleteExpression(S, ResultType);
3986}
3987
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003988void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003989 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003990 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003991 mapCodeCompletionContext(*this, PCC_Statement));
3992 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3993 Results.EnterNewScope();
3994
3995 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3996 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3997 CodeCompleter->includeGlobals());
3998
3999 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
4000
4001 // "else" block
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004002 CodeCompletionBuilder Builder(Results.getAllocator(),
4003 Results.getCodeCompletionTUInfo());
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004004 Builder.AddTypedTextChunk("else");
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00004005 if (Results.includeCodePatterns()) {
4006 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4007 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4008 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4009 Builder.AddPlaceholderChunk("statements");
4010 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4011 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4012 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004013 Results.AddResult(Builder.TakeString());
4014
4015 // "else if" block
4016 Builder.AddTypedTextChunk("else");
4017 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4018 Builder.AddTextChunk("if");
4019 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4020 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00004021 if (getLangOpts().CPlusPlus)
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004022 Builder.AddPlaceholderChunk("condition");
4023 else
4024 Builder.AddPlaceholderChunk("expression");
4025 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00004026 if (Results.includeCodePatterns()) {
4027 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4028 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4029 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4030 Builder.AddPlaceholderChunk("statements");
4031 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4032 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4033 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004034 Results.AddResult(Builder.TakeString());
4035
4036 Results.ExitScope();
4037
4038 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004039 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004040
4041 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00004042 AddMacroResults(PP, Results, false);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004043
4044 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4045 Results.data(),Results.size());
4046}
4047
Richard Trieu2bd04012011-09-09 02:00:50 +00004048void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004049 if (LHS)
4050 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4051 else
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004052 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004053}
4054
Jeffrey Yasskinc76498d2010-04-08 16:38:48 +00004055void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor2436e712009-09-17 21:32:03 +00004056 bool EnteringContext) {
4057 if (!SS.getScopeRep() || !CodeCompleter)
4058 return;
4059
Douglas Gregor3545ff42009-09-21 16:56:56 +00004060 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4061 if (!Ctx)
4062 return;
Douglas Gregor800f2f02009-12-11 18:28:39 +00004063
4064 // Try to instantiate any non-dependent declaration contexts before
4065 // we look in them.
John McCall0b66eb32010-05-01 00:40:08 +00004066 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregor800f2f02009-12-11 18:28:39 +00004067 return;
4068
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004069 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004070 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004071 CodeCompletionContext::CCC_Name);
Douglas Gregorac322ec2010-08-27 21:18:54 +00004072 Results.EnterNewScope();
Douglas Gregor0ac41382010-09-23 23:01:17 +00004073
Douglas Gregor3545ff42009-09-21 16:56:56 +00004074 // The "template" keyword can follow "::" in the grammar, but only
4075 // put it into the grammar if the nested-name-specifier is dependent.
Aaron Ballman4a979672014-01-03 13:56:08 +00004076 NestedNameSpecifier *NNS = SS.getScopeRep();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004077 if (!Results.empty() && NNS->isDependent())
Douglas Gregor78a21012010-01-14 16:01:26 +00004078 Results.AddResult("template");
Douglas Gregorac322ec2010-08-27 21:18:54 +00004079
4080 // Add calls to overridden virtual functions, if there are any.
4081 //
4082 // FIXME: This isn't wonderful, because we don't know whether we're actually
4083 // in a context that permits expressions. This is a general issue with
4084 // qualified-id completions.
4085 if (!EnteringContext)
4086 MaybeAddOverrideCalls(*this, Ctx, Results);
4087 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004088
Douglas Gregorac322ec2010-08-27 21:18:54 +00004089 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4090 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4091
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004092 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorc1679ec2011-07-25 17:48:11 +00004093 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004094 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00004095}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004096
4097void Sema::CodeCompleteUsing(Scope *S) {
4098 if (!CodeCompleter)
4099 return;
4100
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004101 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004102 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004103 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4104 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004105 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004106
4107 // If we aren't in class scope, we could see the "namespace" keyword.
4108 if (!S->isClassScope())
John McCall276321a2010-08-25 06:19:51 +00004109 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004110
4111 // After "using", we can see anything that would start a
4112 // nested-name-specifier.
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004113 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004114 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4115 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004116 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004117
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004118 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004119 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004120 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004121}
4122
4123void Sema::CodeCompleteUsingDirective(Scope *S) {
4124 if (!CodeCompleter)
4125 return;
4126
Douglas Gregor3545ff42009-09-21 16:56:56 +00004127 // After "using namespace", we expect to see a namespace name or namespace
4128 // alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004129 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004130 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004131 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004132 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004133 Results.EnterNewScope();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004134 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004135 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4136 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004137 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004138 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004139 CodeCompletionContext::CCC_Namespace,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004140 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004141}
4142
4143void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4144 if (!CodeCompleter)
4145 return;
4146
Ted Kremenekc37877d2013-10-08 17:08:03 +00004147 DeclContext *Ctx = S->getEntity();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004148 if (!S->getParent())
4149 Ctx = Context.getTranslationUnitDecl();
4150
Douglas Gregor0ac41382010-09-23 23:01:17 +00004151 bool SuppressedGlobalResults
4152 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4153
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004154 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004155 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004156 SuppressedGlobalResults
4157 ? CodeCompletionContext::CCC_Namespace
4158 : CodeCompletionContext::CCC_Other,
4159 &ResultBuilder::IsNamespace);
4160
4161 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00004162 // We only want to see those namespaces that have already been defined
4163 // within this scope, because its likely that the user is creating an
4164 // extended namespace declaration. Keep track of the most recent
4165 // definition of each namespace.
4166 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4167 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4168 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4169 NS != NSEnd; ++NS)
David Blaikie40ed2972012-06-06 20:45:41 +00004170 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004171
4172 // Add the most recent definition (or extended definition) of each
4173 // namespace to the list of results.
Douglas Gregor64b12b52009-09-22 23:31:26 +00004174 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004175 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregor78254c82012-03-27 23:34:16 +00004176 NS = OrigToLatest.begin(),
4177 NSEnd = OrigToLatest.end();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004178 NS != NSEnd; ++NS)
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004179 Results.AddResult(CodeCompletionResult(
4180 NS->second, Results.getBasePriority(NS->second), 0),
Douglas Gregorfc59ce12010-01-14 16:14:35 +00004181 CurContext, 0, false);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004182 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004183 }
4184
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004185 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004186 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004187 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004188}
4189
4190void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4191 if (!CodeCompleter)
4192 return;
4193
Douglas Gregor3545ff42009-09-21 16:56:56 +00004194 // After "namespace", we expect to see a namespace or alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004195 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004196 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004197 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004198 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004199 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004200 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4201 CodeCompleter->includeGlobals());
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004202 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004203 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004204 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004205}
4206
Douglas Gregorc811ede2009-09-18 20:05:18 +00004207void Sema::CodeCompleteOperatorName(Scope *S) {
4208 if (!CodeCompleter)
4209 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004210
John McCall276321a2010-08-25 06:19:51 +00004211 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004212 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004213 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004214 CodeCompletionContext::CCC_Type,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004215 &ResultBuilder::IsType);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004216 Results.EnterNewScope();
Douglas Gregorc811ede2009-09-18 20:05:18 +00004217
Douglas Gregor3545ff42009-09-21 16:56:56 +00004218 // Add the names of overloadable operators.
4219#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4220 if (std::strcmp(Spelling, "?")) \
Douglas Gregor78a21012010-01-14 16:01:26 +00004221 Results.AddResult(Result(Spelling));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004222#include "clang/Basic/OperatorKinds.def"
4223
4224 // Add any type names visible from the current scope
Douglas Gregor6ae4c522010-01-14 03:21:49 +00004225 Results.allowNestedNameSpecifiers();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004226 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004227 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4228 CodeCompleter->includeGlobals());
Douglas Gregor3545ff42009-09-21 16:56:56 +00004229
4230 // Add any type specifiers
David Blaikiebbafb8a2012-03-11 07:00:24 +00004231 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004232 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004233
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004234 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004235 CodeCompletionContext::CCC_Type,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004236 Results.data(),Results.size());
Douglas Gregorc811ede2009-09-18 20:05:18 +00004237}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004238
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004239void Sema::CodeCompleteConstructorInitializer(
4240 Decl *ConstructorD,
4241 ArrayRef <CXXCtorInitializer *> Initializers) {
Douglas Gregor75acd922011-09-27 23:30:47 +00004242 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004243 CXXConstructorDecl *Constructor
4244 = static_cast<CXXConstructorDecl *>(ConstructorD);
4245 if (!Constructor)
4246 return;
4247
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004248 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004249 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004250 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004251 Results.EnterNewScope();
4252
4253 // Fill in any already-initialized fields or base classes.
4254 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4255 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004256 for (unsigned I = 0, E = Initializers.size(); I != E; ++I) {
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004257 if (Initializers[I]->isBaseInitializer())
4258 InitializedBases.insert(
4259 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4260 else
Francois Pichetd583da02010-12-04 09:14:42 +00004261 InitializedFields.insert(cast<FieldDecl>(
4262 Initializers[I]->getAnyMember()));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004263 }
4264
4265 // Add completions for base classes.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004266 CodeCompletionBuilder Builder(Results.getAllocator(),
4267 Results.getCodeCompletionTUInfo());
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004268 bool SawLastInitializer = Initializers.empty();
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004269 CXXRecordDecl *ClassDecl = Constructor->getParent();
4270 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->bases_begin(),
4271 BaseEnd = ClassDecl->bases_end();
4272 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004273 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4274 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004275 = !Initializers.empty() &&
4276 Initializers.back()->isBaseInitializer() &&
Douglas Gregor99129ef2010-08-29 19:27:27 +00004277 Context.hasSameUnqualifiedType(Base->getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004278 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004279 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004280 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004281
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004282 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004283 Results.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004284 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004285 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4286 Builder.AddPlaceholderChunk("args");
4287 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4288 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004289 SawLastInitializer? CCP_NextInitializer
4290 : CCP_MemberDeclaration));
4291 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004292 }
4293
4294 // Add completions for virtual base classes.
4295 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->vbases_begin(),
4296 BaseEnd = ClassDecl->vbases_end();
4297 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004298 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4299 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004300 = !Initializers.empty() &&
4301 Initializers.back()->isBaseInitializer() &&
Douglas Gregor99129ef2010-08-29 19:27:27 +00004302 Context.hasSameUnqualifiedType(Base->getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004303 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004304 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004305 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004306
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004307 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004308 Builder.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004309 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004310 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4311 Builder.AddPlaceholderChunk("args");
4312 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4313 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004314 SawLastInitializer? CCP_NextInitializer
4315 : CCP_MemberDeclaration));
4316 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004317 }
4318
4319 // Add completions for members.
4320 for (CXXRecordDecl::field_iterator Field = ClassDecl->field_begin(),
4321 FieldEnd = ClassDecl->field_end();
4322 Field != FieldEnd; ++Field) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004323 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4324 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004325 = !Initializers.empty() &&
4326 Initializers.back()->isAnyMemberInitializer() &&
4327 Initializers.back()->getAnyMember() == *Field;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004328 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004329 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004330
4331 if (!Field->getDeclName())
4332 continue;
4333
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004334 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004335 Field->getIdentifier()->getName()));
4336 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4337 Builder.AddPlaceholderChunk("args");
4338 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4339 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004340 SawLastInitializer? CCP_NextInitializer
Douglas Gregorf3af3112010-09-09 21:42:20 +00004341 : CCP_MemberDeclaration,
Douglas Gregor78254c82012-03-27 23:34:16 +00004342 CXCursor_MemberRef,
4343 CXAvailability_Available,
David Blaikie40ed2972012-06-06 20:45:41 +00004344 *Field));
Douglas Gregor99129ef2010-08-29 19:27:27 +00004345 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004346 }
4347 Results.ExitScope();
4348
Douglas Gregor0ac41382010-09-23 23:01:17 +00004349 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004350 Results.data(), Results.size());
4351}
4352
Douglas Gregord8c61782012-02-15 15:34:24 +00004353/// \brief Determine whether this scope denotes a namespace.
4354static bool isNamespaceScope(Scope *S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00004355 DeclContext *DC = S->getEntity();
Douglas Gregord8c61782012-02-15 15:34:24 +00004356 if (!DC)
4357 return false;
4358
4359 return DC->isFileContext();
4360}
4361
4362void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4363 bool AfterAmpersand) {
4364 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004365 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord8c61782012-02-15 15:34:24 +00004366 CodeCompletionContext::CCC_Other);
4367 Results.EnterNewScope();
4368
4369 // Note what has already been captured.
4370 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4371 bool IncludedThis = false;
4372 for (SmallVectorImpl<LambdaCapture>::iterator C = Intro.Captures.begin(),
4373 CEnd = Intro.Captures.end();
4374 C != CEnd; ++C) {
4375 if (C->Kind == LCK_This) {
4376 IncludedThis = true;
4377 continue;
4378 }
4379
4380 Known.insert(C->Id);
4381 }
4382
4383 // Look for other capturable variables.
4384 for (; S && !isNamespaceScope(S); S = S->getParent()) {
4385 for (Scope::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
4386 D != DEnd; ++D) {
4387 VarDecl *Var = dyn_cast<VarDecl>(*D);
4388 if (!Var ||
4389 !Var->hasLocalStorage() ||
4390 Var->hasAttr<BlocksAttr>())
4391 continue;
4392
4393 if (Known.insert(Var->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004394 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
4395 CurContext, 0, false);
Douglas Gregord8c61782012-02-15 15:34:24 +00004396 }
4397 }
4398
4399 // Add 'this', if it would be valid.
4400 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4401 addThisCompletion(*this, Results);
4402
4403 Results.ExitScope();
4404
4405 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4406 Results.data(), Results.size());
4407}
4408
James Dennett596e4752012-06-14 03:11:41 +00004409/// Macro that optionally prepends an "@" to the string literal passed in via
4410/// Keyword, depending on whether NeedAt is true or false.
4411#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4412
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004413static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004414 ResultBuilder &Results,
4415 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004416 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004417 // Since we have an implementation, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004418 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004419
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004420 CodeCompletionBuilder Builder(Results.getAllocator(),
4421 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004422 if (LangOpts.ObjC2) {
4423 // @dynamic
James Dennett596e4752012-06-14 03:11:41 +00004424 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004425 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4426 Builder.AddPlaceholderChunk("property");
4427 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004428
4429 // @synthesize
James Dennett596e4752012-06-14 03:11:41 +00004430 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004431 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4432 Builder.AddPlaceholderChunk("property");
4433 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004434 }
4435}
4436
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004437static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004438 ResultBuilder &Results,
4439 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004440 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004441
4442 // Since we have an interface or protocol, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004443 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004444
4445 if (LangOpts.ObjC2) {
4446 // @property
James Dennett596e4752012-06-14 03:11:41 +00004447 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004448
4449 // @required
James Dennett596e4752012-06-14 03:11:41 +00004450 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004451
4452 // @optional
James Dennett596e4752012-06-14 03:11:41 +00004453 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004454 }
4455}
4456
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004457static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004458 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004459 CodeCompletionBuilder Builder(Results.getAllocator(),
4460 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004461
4462 // @class name ;
James Dennett596e4752012-06-14 03:11:41 +00004463 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004464 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4465 Builder.AddPlaceholderChunk("name");
4466 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004467
Douglas Gregorf4c33342010-05-28 00:22:41 +00004468 if (Results.includeCodePatterns()) {
4469 // @interface name
4470 // FIXME: Could introduce the whole pattern, including superclasses and
4471 // such.
James Dennett596e4752012-06-14 03:11:41 +00004472 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004473 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4474 Builder.AddPlaceholderChunk("class");
4475 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004476
Douglas Gregorf4c33342010-05-28 00:22:41 +00004477 // @protocol name
James Dennett596e4752012-06-14 03:11:41 +00004478 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004479 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4480 Builder.AddPlaceholderChunk("protocol");
4481 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004482
4483 // @implementation name
James Dennett596e4752012-06-14 03:11:41 +00004484 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004485 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4486 Builder.AddPlaceholderChunk("class");
4487 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004488 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004489
4490 // @compatibility_alias name
James Dennett596e4752012-06-14 03:11:41 +00004491 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004492 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4493 Builder.AddPlaceholderChunk("alias");
4494 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4495 Builder.AddPlaceholderChunk("class");
4496 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor61e36812013-03-07 23:26:24 +00004497
4498 if (Results.getSema().getLangOpts().Modules) {
4499 // @import name
4500 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "import"));
4501 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4502 Builder.AddPlaceholderChunk("module");
4503 Results.AddResult(Result(Builder.TakeString()));
4504 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004505}
4506
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004507void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004508 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004509 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004510 CodeCompletionContext::CCC_Other);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004511 Results.EnterNewScope();
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004512 if (isa<ObjCImplDecl>(CurContext))
David Blaikiebbafb8a2012-03-11 07:00:24 +00004513 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004514 else if (CurContext->isObjCContainer())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004515 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorf1934162010-01-13 21:24:21 +00004516 else
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004517 AddObjCTopLevelResults(Results, false);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004518 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004519 HandleCodeCompleteResults(this, CodeCompleter,
4520 CodeCompletionContext::CCC_Other,
4521 Results.data(),Results.size());
Douglas Gregorf48706c2009-12-07 09:27:33 +00004522}
4523
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004524static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004525 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004526 CodeCompletionBuilder Builder(Results.getAllocator(),
4527 Results.getCodeCompletionTUInfo());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004528
4529 // @encode ( type-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004530 const char *EncodeType = "char[]";
David Blaikiebbafb8a2012-03-11 07:00:24 +00004531 if (Results.getSema().getLangOpts().CPlusPlus ||
4532 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose9da05852012-06-15 18:19:56 +00004533 EncodeType = "const char[]";
Douglas Gregore5c79d52011-10-18 21:20:17 +00004534 Builder.AddResultTypeChunk(EncodeType);
James Dennett596e4752012-06-14 03:11:41 +00004535 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004536 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4537 Builder.AddPlaceholderChunk("type-name");
4538 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4539 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004540
4541 // @protocol ( protocol-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004542 Builder.AddResultTypeChunk("Protocol *");
James Dennett596e4752012-06-14 03:11:41 +00004543 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004544 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4545 Builder.AddPlaceholderChunk("protocol-name");
4546 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4547 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004548
4549 // @selector ( selector )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004550 Builder.AddResultTypeChunk("SEL");
James Dennett596e4752012-06-14 03:11:41 +00004551 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004552 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4553 Builder.AddPlaceholderChunk("selector");
4554 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4555 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004556
4557 // @"string"
4558 Builder.AddResultTypeChunk("NSString *");
4559 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4560 Builder.AddPlaceholderChunk("string");
4561 Builder.AddTextChunk("\"");
4562 Results.AddResult(Result(Builder.TakeString()));
4563
Douglas Gregor951de302012-07-17 23:24:47 +00004564 // @[objects, ...]
Jordan Rose9da05852012-06-15 18:19:56 +00004565 Builder.AddResultTypeChunk("NSArray *");
James Dennett596e4752012-06-14 03:11:41 +00004566 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004567 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004568 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4569 Results.AddResult(Result(Builder.TakeString()));
4570
Douglas Gregor951de302012-07-17 23:24:47 +00004571 // @{key : object, ...}
Jordan Rose9da05852012-06-15 18:19:56 +00004572 Builder.AddResultTypeChunk("NSDictionary *");
James Dennett596e4752012-06-14 03:11:41 +00004573 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"{"));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004574 Builder.AddPlaceholderChunk("key");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004575 Builder.AddChunk(CodeCompletionString::CK_Colon);
4576 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4577 Builder.AddPlaceholderChunk("object, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004578 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4579 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004580
Douglas Gregor951de302012-07-17 23:24:47 +00004581 // @(expression)
Jordan Rose9da05852012-06-15 18:19:56 +00004582 Builder.AddResultTypeChunk("id");
4583 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose9da05852012-06-15 18:19:56 +00004584 Builder.AddPlaceholderChunk("expression");
Jordan Rose9da05852012-06-15 18:19:56 +00004585 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4586 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004587}
4588
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004589static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004590 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004591 CodeCompletionBuilder Builder(Results.getAllocator(),
4592 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004593
Douglas Gregorf4c33342010-05-28 00:22:41 +00004594 if (Results.includeCodePatterns()) {
4595 // @try { statements } @catch ( declaration ) { statements } @finally
4596 // { statements }
James Dennett596e4752012-06-14 03:11:41 +00004597 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004598 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4599 Builder.AddPlaceholderChunk("statements");
4600 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4601 Builder.AddTextChunk("@catch");
4602 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4603 Builder.AddPlaceholderChunk("parameter");
4604 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4605 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4606 Builder.AddPlaceholderChunk("statements");
4607 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4608 Builder.AddTextChunk("@finally");
4609 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4610 Builder.AddPlaceholderChunk("statements");
4611 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4612 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004613 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004614
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004615 // @throw
James Dennett596e4752012-06-14 03:11:41 +00004616 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004617 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4618 Builder.AddPlaceholderChunk("expression");
4619 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004620
Douglas Gregorf4c33342010-05-28 00:22:41 +00004621 if (Results.includeCodePatterns()) {
4622 // @synchronized ( expression ) { statements }
James Dennett596e4752012-06-14 03:11:41 +00004623 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004624 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4625 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4626 Builder.AddPlaceholderChunk("expression");
4627 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4628 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4629 Builder.AddPlaceholderChunk("statements");
4630 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4631 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004632 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004633}
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004634
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004635static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00004636 ResultBuilder &Results,
4637 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004638 typedef CodeCompletionResult Result;
James Dennett596e4752012-06-14 03:11:41 +00004639 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4640 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4641 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004642 if (LangOpts.ObjC2)
James Dennett596e4752012-06-14 03:11:41 +00004643 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004644}
4645
4646void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004647 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004648 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004649 CodeCompletionContext::CCC_Other);
Douglas Gregor48d46252010-01-13 21:54:15 +00004650 Results.EnterNewScope();
David Blaikiebbafb8a2012-03-11 07:00:24 +00004651 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregor48d46252010-01-13 21:54:15 +00004652 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004653 HandleCodeCompleteResults(this, CodeCompleter,
4654 CodeCompletionContext::CCC_Other,
4655 Results.data(),Results.size());
Douglas Gregor48d46252010-01-13 21:54:15 +00004656}
4657
4658void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004659 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004660 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004661 CodeCompletionContext::CCC_Other);
Douglas Gregorf1934162010-01-13 21:24:21 +00004662 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004663 AddObjCStatementResults(Results, false);
4664 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004665 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004666 HandleCodeCompleteResults(this, CodeCompleter,
4667 CodeCompletionContext::CCC_Other,
4668 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004669}
4670
4671void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004672 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004673 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004674 CodeCompletionContext::CCC_Other);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004675 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004676 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004677 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004678 HandleCodeCompleteResults(this, CodeCompleter,
4679 CodeCompletionContext::CCC_Other,
4680 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004681}
4682
Douglas Gregore6078da2009-11-19 00:14:45 +00004683/// \brief Determine whether the addition of the given flag to an Objective-C
4684/// property's attributes will cause a conflict.
Bill Wendling44426052012-12-20 19:22:21 +00004685static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregore6078da2009-11-19 00:14:45 +00004686 // Check if we've already added this flag.
Bill Wendling44426052012-12-20 19:22:21 +00004687 if (Attributes & NewFlag)
Douglas Gregore6078da2009-11-19 00:14:45 +00004688 return true;
4689
Bill Wendling44426052012-12-20 19:22:21 +00004690 Attributes |= NewFlag;
Douglas Gregore6078da2009-11-19 00:14:45 +00004691
4692 // Check for collisions with "readonly".
Bill Wendling44426052012-12-20 19:22:21 +00004693 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4694 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregore6078da2009-11-19 00:14:45 +00004695 return true;
4696
Jordan Rose53cb2f32012-08-20 20:01:13 +00004697 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendling44426052012-12-20 19:22:21 +00004698 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCall31168b02011-06-15 23:02:42 +00004699 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregore6078da2009-11-19 00:14:45 +00004700 ObjCDeclSpec::DQ_PR_copy |
Jordan Rose53cb2f32012-08-20 20:01:13 +00004701 ObjCDeclSpec::DQ_PR_retain |
4702 ObjCDeclSpec::DQ_PR_strong |
4703 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregore6078da2009-11-19 00:14:45 +00004704 if (AssignCopyRetMask &&
4705 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCall31168b02011-06-15 23:02:42 +00004706 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregore6078da2009-11-19 00:14:45 +00004707 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCall31168b02011-06-15 23:02:42 +00004708 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rose53cb2f32012-08-20 20:01:13 +00004709 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4710 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregore6078da2009-11-19 00:14:45 +00004711 return true;
4712
4713 return false;
4714}
4715
Douglas Gregor36029f42009-11-18 23:08:07 +00004716void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroff936354c2009-10-08 21:55:05 +00004717 if (!CodeCompleter)
4718 return;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004719
Bill Wendling44426052012-12-20 19:22:21 +00004720 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroff936354c2009-10-08 21:55:05 +00004721
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004722 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004723 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004724 CodeCompletionContext::CCC_Other);
Steve Naroff936354c2009-10-08 21:55:05 +00004725 Results.EnterNewScope();
Bill Wendling44426052012-12-20 19:22:21 +00004726 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall276321a2010-08-25 06:19:51 +00004727 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendling44426052012-12-20 19:22:21 +00004728 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall276321a2010-08-25 06:19:51 +00004729 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendling44426052012-12-20 19:22:21 +00004730 if (!ObjCPropertyFlagConflicts(Attributes,
John McCall31168b02011-06-15 23:02:42 +00004731 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4732 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendling44426052012-12-20 19:22:21 +00004733 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall276321a2010-08-25 06:19:51 +00004734 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendling44426052012-12-20 19:22:21 +00004735 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall276321a2010-08-25 06:19:51 +00004736 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendling44426052012-12-20 19:22:21 +00004737 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCall31168b02011-06-15 23:02:42 +00004738 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendling44426052012-12-20 19:22:21 +00004739 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall276321a2010-08-25 06:19:51 +00004740 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendling44426052012-12-20 19:22:21 +00004741 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall276321a2010-08-25 06:19:51 +00004742 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendling44426052012-12-20 19:22:21 +00004743 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian1c2d29e2011-06-11 17:14:27 +00004744 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rose53cb2f32012-08-20 20:01:13 +00004745
4746 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall3deb1ad2012-08-21 02:47:43 +00004747 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendling44426052012-12-20 19:22:21 +00004748 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rose53cb2f32012-08-20 20:01:13 +00004749 Results.AddResult(CodeCompletionResult("weak"));
4750
Bill Wendling44426052012-12-20 19:22:21 +00004751 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004752 CodeCompletionBuilder Setter(Results.getAllocator(),
4753 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004754 Setter.AddTypedTextChunk("setter");
Argyrios Kyrtzidis7bbb8812014-02-20 07:55:15 +00004755 Setter.AddTextChunk("=");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004756 Setter.AddPlaceholderChunk("method");
4757 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004758 }
Bill Wendling44426052012-12-20 19:22:21 +00004759 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004760 CodeCompletionBuilder Getter(Results.getAllocator(),
4761 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004762 Getter.AddTypedTextChunk("getter");
Argyrios Kyrtzidis7bbb8812014-02-20 07:55:15 +00004763 Getter.AddTextChunk("=");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004764 Getter.AddPlaceholderChunk("method");
4765 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004766 }
Steve Naroff936354c2009-10-08 21:55:05 +00004767 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004768 HandleCodeCompleteResults(this, CodeCompleter,
4769 CodeCompletionContext::CCC_Other,
4770 Results.data(),Results.size());
Steve Naroff936354c2009-10-08 21:55:05 +00004771}
Steve Naroffeae65032009-11-07 02:08:14 +00004772
James Dennettf1243872012-06-17 05:33:25 +00004773/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregorc8537c52009-11-19 07:41:15 +00004774/// via code completion.
4775enum ObjCMethodKind {
Dmitri Gribenko4280e5c2012-06-08 23:13:42 +00004776 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4777 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4778 MK_OneArgSelector ///< One-argument selector.
Douglas Gregorc8537c52009-11-19 07:41:15 +00004779};
4780
Douglas Gregor67c692c2010-08-26 15:07:07 +00004781static bool isAcceptableObjCSelector(Selector Sel,
4782 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004783 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004784 bool AllowSameLength = true) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004785 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor67c692c2010-08-26 15:07:07 +00004786 if (NumSelIdents > Sel.getNumArgs())
4787 return false;
4788
4789 switch (WantKind) {
4790 case MK_Any: break;
4791 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4792 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4793 }
4794
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004795 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4796 return false;
4797
Douglas Gregor67c692c2010-08-26 15:07:07 +00004798 for (unsigned I = 0; I != NumSelIdents; ++I)
4799 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4800 return false;
4801
4802 return true;
4803}
4804
Douglas Gregorc8537c52009-11-19 07:41:15 +00004805static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4806 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004807 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004808 bool AllowSameLength = true) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00004809 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004810 AllowSameLength);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004811}
Douglas Gregor1154e272010-09-16 16:06:31 +00004812
4813namespace {
4814 /// \brief A set of selectors, which is used to avoid introducing multiple
4815 /// completions with the same selector into the result set.
4816 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4817}
4818
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004819/// \brief Add all of the Objective-C methods in the given Objective-C
4820/// container to the set of results.
4821///
4822/// The container will be a class, protocol, category, or implementation of
4823/// any of the above. This mether will recurse to include methods from
4824/// the superclasses of classes along with their categories, protocols, and
4825/// implementations.
4826///
4827/// \param Container the container in which we'll look to find methods.
4828///
James Dennett596e4752012-06-14 03:11:41 +00004829/// \param WantInstanceMethods Whether to add instance methods (only); if
4830/// false, this routine will add factory methods (only).
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004831///
4832/// \param CurContext the context in which we're performing the lookup that
4833/// finds methods.
4834///
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004835/// \param AllowSameLength Whether we allow a method to be added to the list
4836/// when it has the same number of parameters as we have selector identifiers.
4837///
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004838/// \param Results the structure into which we'll add results.
4839static void AddObjCMethods(ObjCContainerDecl *Container,
4840 bool WantInstanceMethods,
Douglas Gregorc8537c52009-11-19 07:41:15 +00004841 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004842 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004843 DeclContext *CurContext,
Douglas Gregor1154e272010-09-16 16:06:31 +00004844 VisitedSelectorSet &Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004845 bool AllowSameLength,
Douglas Gregor416b5752010-08-25 01:08:01 +00004846 ResultBuilder &Results,
4847 bool InOriginalClass = true) {
John McCall276321a2010-08-25 06:19:51 +00004848 typedef CodeCompletionResult Result;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00004849 Container = getContainerDef(Container);
Douglas Gregor41778c32013-01-30 06:58:39 +00004850 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4851 bool isRootClass = IFace && !IFace->getSuperClass();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004852 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
4853 MEnd = Container->meth_end();
4854 M != MEnd; ++M) {
Douglas Gregor41778c32013-01-30 06:58:39 +00004855 // The instance methods on the root class can be messaged via the
4856 // metaclass.
4857 if (M->isInstanceMethod() == WantInstanceMethods ||
4858 (isRootClass && !WantInstanceMethods)) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00004859 // Check whether the selector identifiers we've been given are a
4860 // subset of the identifiers for this particular method.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004861 if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, AllowSameLength))
Douglas Gregor1b605f72009-11-19 01:08:35 +00004862 continue;
Douglas Gregorc8537c52009-11-19 07:41:15 +00004863
David Blaikie2d7c57e2012-04-30 02:36:29 +00004864 if (!Selectors.insert(M->getSelector()))
Douglas Gregor1154e272010-09-16 16:06:31 +00004865 continue;
4866
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004867 Result R = Result(*M, Results.getBasePriority(*M), 0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004868 R.StartParameter = SelIdents.size();
Douglas Gregorc8537c52009-11-19 07:41:15 +00004869 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor416b5752010-08-25 01:08:01 +00004870 if (!InOriginalClass)
4871 R.Priority += CCD_InBaseClass;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004872 Results.MaybeAddResult(R, CurContext);
4873 }
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004874 }
4875
Douglas Gregorf37c9492010-09-16 15:34:59 +00004876 // Visit the protocols of protocols.
4877 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregore6e48b12012-01-01 19:29:29 +00004878 if (Protocol->hasDefinition()) {
4879 const ObjCList<ObjCProtocolDecl> &Protocols
4880 = Protocol->getReferencedProtocols();
4881 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4882 E = Protocols.end();
4883 I != E; ++I)
4884 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004885 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregore6e48b12012-01-01 19:29:29 +00004886 }
Douglas Gregorf37c9492010-09-16 15:34:59 +00004887 }
4888
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00004889 if (!IFace || !IFace->hasDefinition())
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004890 return;
4891
4892 // Add methods in protocols.
Argyrios Kyrtzidise7f3ef32012-03-13 01:09:41 +00004893 for (ObjCInterfaceDecl::protocol_iterator I = IFace->protocol_begin(),
4894 E = IFace->protocol_end();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004895 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004896 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004897 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004898
4899 // Add methods in categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00004900 for (ObjCInterfaceDecl::known_categories_iterator
4901 Cat = IFace->known_categories_begin(),
4902 CatEnd = IFace->known_categories_end();
4903 Cat != CatEnd; ++Cat) {
4904 ObjCCategoryDecl *CatDecl = *Cat;
4905
4906 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004907 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004908 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004909
4910 // Add a categories protocol methods.
4911 const ObjCList<ObjCProtocolDecl> &Protocols
4912 = CatDecl->getReferencedProtocols();
4913 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4914 E = Protocols.end();
4915 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00004916 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004917 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004918 Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004919
4920 // Add methods in category implementations.
4921 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004922 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004923 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004924 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004925 }
4926
4927 // Add methods in superclass.
4928 if (IFace->getSuperClass())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004929 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004930 SelIdents, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004931 AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004932
4933 // Add methods in our implementation, if any.
4934 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004935 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004936 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004937 Results, InOriginalClass);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004938}
4939
4940
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004941void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004942 // Try to find the interface where getters might live.
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004943 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004944 if (!Class) {
4945 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004946 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004947 Class = Category->getClassInterface();
4948
4949 if (!Class)
4950 return;
4951 }
4952
4953 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004954 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004955 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004956 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004957 Results.EnterNewScope();
4958
Douglas Gregor1154e272010-09-16 16:06:31 +00004959 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004960 AddObjCMethods(Class, true, MK_ZeroArgSelector, None, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004961 /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004962 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004963 HandleCodeCompleteResults(this, CodeCompleter,
4964 CodeCompletionContext::CCC_Other,
4965 Results.data(),Results.size());
Douglas Gregorc8537c52009-11-19 07:41:15 +00004966}
4967
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004968void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004969 // Try to find the interface where setters might live.
4970 ObjCInterfaceDecl *Class
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004971 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004972 if (!Class) {
4973 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004974 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004975 Class = Category->getClassInterface();
4976
4977 if (!Class)
4978 return;
4979 }
4980
4981 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004982 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004983 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004984 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004985 Results.EnterNewScope();
4986
Douglas Gregor1154e272010-09-16 16:06:31 +00004987 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004988 AddObjCMethods(Class, true, MK_OneArgSelector, None, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004989 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004990
4991 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004992 HandleCodeCompleteResults(this, CodeCompleter,
4993 CodeCompletionContext::CCC_Other,
4994 Results.data(),Results.size());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004995}
4996
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004997void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
4998 bool IsParameter) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004999 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005000 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005001 CodeCompletionContext::CCC_Type);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005002 Results.EnterNewScope();
5003
5004 // Add context-sensitive, Objective-C parameter-passing keywords.
5005 bool AddedInOut = false;
5006 if ((DS.getObjCDeclQualifier() &
5007 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
5008 Results.AddResult("in");
5009 Results.AddResult("inout");
5010 AddedInOut = true;
5011 }
5012 if ((DS.getObjCDeclQualifier() &
5013 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
5014 Results.AddResult("out");
5015 if (!AddedInOut)
5016 Results.AddResult("inout");
5017 }
5018 if ((DS.getObjCDeclQualifier() &
5019 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
5020 ObjCDeclSpec::DQ_Oneway)) == 0) {
5021 Results.AddResult("bycopy");
5022 Results.AddResult("byref");
5023 Results.AddResult("oneway");
5024 }
5025
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005026 // If we're completing the return type of an Objective-C method and the
5027 // identifier IBAction refers to a macro, provide a completion item for
5028 // an action, e.g.,
5029 // IBAction)<#selector#>:(id)sender
5030 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
5031 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005032 CodeCompletionBuilder Builder(Results.getAllocator(),
5033 Results.getCodeCompletionTUInfo(),
5034 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005035 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005036 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005037 Builder.AddPlaceholderChunk("selector");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005038 Builder.AddChunk(CodeCompletionString::CK_Colon);
5039 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005040 Builder.AddTextChunk("id");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005041 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005042 Builder.AddTextChunk("sender");
5043 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
5044 }
Douglas Gregored1f5972013-01-30 07:11:43 +00005045
5046 // If we're completing the return type, provide 'instancetype'.
5047 if (!IsParameter) {
5048 Results.AddResult(CodeCompletionResult("instancetype"));
5049 }
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005050
Douglas Gregor99fa2642010-08-24 01:06:58 +00005051 // Add various builtin type names and specifiers.
5052 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5053 Results.ExitScope();
5054
5055 // Add the various type names
5056 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5057 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5058 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5059 CodeCompleter->includeGlobals());
5060
5061 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005062 AddMacroResults(PP, Results, false);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005063
5064 HandleCodeCompleteResults(this, CodeCompleter,
5065 CodeCompletionContext::CCC_Type,
5066 Results.data(), Results.size());
5067}
5068
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005069/// \brief When we have an expression with type "id", we may assume
5070/// that it has some more-specific class type based on knowledge of
5071/// common uses of Objective-C. This routine returns that class type,
5072/// or NULL if no better result could be determined.
5073static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregored0b69d2010-09-15 16:23:04 +00005074 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005075 if (!Msg)
5076 return 0;
5077
5078 Selector Sel = Msg->getSelector();
5079 if (Sel.isNull())
5080 return 0;
5081
5082 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5083 if (!Id)
5084 return 0;
5085
5086 ObjCMethodDecl *Method = Msg->getMethodDecl();
5087 if (!Method)
5088 return 0;
5089
5090 // Determine the class that we're sending the message to.
Douglas Gregor9a129192010-04-21 00:45:42 +00005091 ObjCInterfaceDecl *IFace = 0;
5092 switch (Msg->getReceiverKind()) {
5093 case ObjCMessageExpr::Class:
John McCall8b07ec22010-05-15 11:32:37 +00005094 if (const ObjCObjectType *ObjType
5095 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5096 IFace = ObjType->getInterface();
Douglas Gregor9a129192010-04-21 00:45:42 +00005097 break;
5098
5099 case ObjCMessageExpr::Instance: {
5100 QualType T = Msg->getInstanceReceiver()->getType();
5101 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5102 IFace = Ptr->getInterfaceDecl();
5103 break;
5104 }
5105
5106 case ObjCMessageExpr::SuperInstance:
5107 case ObjCMessageExpr::SuperClass:
5108 break;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005109 }
5110
5111 if (!IFace)
5112 return 0;
5113
5114 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5115 if (Method->isInstanceMethod())
5116 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5117 .Case("retain", IFace)
John McCall31168b02011-06-15 23:02:42 +00005118 .Case("strong", IFace)
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005119 .Case("autorelease", IFace)
5120 .Case("copy", IFace)
5121 .Case("copyWithZone", IFace)
5122 .Case("mutableCopy", IFace)
5123 .Case("mutableCopyWithZone", IFace)
5124 .Case("awakeFromCoder", IFace)
5125 .Case("replacementObjectFromCoder", IFace)
5126 .Case("class", IFace)
5127 .Case("classForCoder", IFace)
5128 .Case("superclass", Super)
5129 .Default(0);
5130
5131 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5132 .Case("new", IFace)
5133 .Case("alloc", IFace)
5134 .Case("allocWithZone", IFace)
5135 .Case("class", IFace)
5136 .Case("superclass", Super)
5137 .Default(0);
5138}
5139
Douglas Gregor6fc04132010-08-27 15:10:57 +00005140// Add a special completion for a message send to "super", which fills in the
5141// most likely case of forwarding all of our arguments to the superclass
5142// function.
5143///
5144/// \param S The semantic analysis object.
5145///
Dmitri Gribenkoadba9be2012-08-23 17:58:28 +00005146/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor6fc04132010-08-27 15:10:57 +00005147/// the "super" keyword. Otherwise, we just need to provide the arguments.
5148///
5149/// \param SelIdents The identifiers in the selector that have already been
5150/// provided as arguments for a send to "super".
5151///
Douglas Gregor6fc04132010-08-27 15:10:57 +00005152/// \param Results The set of results to augment.
5153///
5154/// \returns the Objective-C method declaration that would be invoked by
5155/// this "super" completion. If NULL, no completion was added.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005156static ObjCMethodDecl *AddSuperSendCompletion(
5157 Sema &S, bool NeedSuperKeyword,
5158 ArrayRef<IdentifierInfo *> SelIdents,
5159 ResultBuilder &Results) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005160 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5161 if (!CurMethod)
5162 return 0;
5163
5164 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5165 if (!Class)
5166 return 0;
5167
5168 // Try to find a superclass method with the same selector.
5169 ObjCMethodDecl *SuperMethod = 0;
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005170 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5171 // Check in the class
Douglas Gregor6fc04132010-08-27 15:10:57 +00005172 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5173 CurMethod->isInstanceMethod());
5174
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005175 // Check in categories or class extensions.
5176 if (!SuperMethod) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005177 for (ObjCInterfaceDecl::known_categories_iterator
5178 Cat = Class->known_categories_begin(),
5179 CatEnd = Class->known_categories_end();
5180 Cat != CatEnd; ++Cat) {
5181 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005182 CurMethod->isInstanceMethod())))
5183 break;
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005184 }
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005185 }
5186 }
5187
Douglas Gregor6fc04132010-08-27 15:10:57 +00005188 if (!SuperMethod)
5189 return 0;
5190
5191 // Check whether the superclass method has the same signature.
5192 if (CurMethod->param_size() != SuperMethod->param_size() ||
5193 CurMethod->isVariadic() != SuperMethod->isVariadic())
5194 return 0;
5195
5196 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5197 CurPEnd = CurMethod->param_end(),
5198 SuperP = SuperMethod->param_begin();
5199 CurP != CurPEnd; ++CurP, ++SuperP) {
5200 // Make sure the parameter types are compatible.
5201 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5202 (*SuperP)->getType()))
5203 return 0;
5204
5205 // Make sure we have a parameter name to forward!
5206 if (!(*CurP)->getIdentifier())
5207 return 0;
5208 }
5209
5210 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005211 CodeCompletionBuilder Builder(Results.getAllocator(),
5212 Results.getCodeCompletionTUInfo());
Douglas Gregor6fc04132010-08-27 15:10:57 +00005213
5214 // Give this completion a return type.
Douglas Gregor75acd922011-09-27 23:30:47 +00005215 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5216 Builder);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005217
5218 // If we need the "super" keyword, add it (plus some spacing).
5219 if (NeedSuperKeyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005220 Builder.AddTypedTextChunk("super");
5221 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005222 }
5223
5224 Selector Sel = CurMethod->getSelector();
5225 if (Sel.isUnarySelector()) {
5226 if (NeedSuperKeyword)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005227 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005228 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005229 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005230 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005231 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005232 } else {
5233 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5234 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005235 if (I > SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005236 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005237
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005238 if (I < SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005239 Builder.AddInformativeChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005240 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005241 Sel.getNameForSlot(I) + ":"));
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005242 else if (NeedSuperKeyword || I > SelIdents.size()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005243 Builder.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005244 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005245 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005246 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005247 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005248 } else {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005249 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005250 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005251 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005252 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005253 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005254 }
5255 }
5256 }
5257
Douglas Gregor78254c82012-03-27 23:34:16 +00005258 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5259 CCP_SuperCompletion));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005260 return SuperMethod;
5261}
5262
Douglas Gregora817a192010-05-27 23:06:34 +00005263void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall276321a2010-08-25 06:19:51 +00005264 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005265 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005266 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005267 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005268 getLangOpts().CPlusPlus11
Douglas Gregord8c61782012-02-15 15:34:24 +00005269 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5270 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregora817a192010-05-27 23:06:34 +00005271
Douglas Gregora817a192010-05-27 23:06:34 +00005272 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5273 Results.EnterNewScope();
Douglas Gregor39982192010-08-15 06:18:01 +00005274 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5275 CodeCompleter->includeGlobals());
Douglas Gregora817a192010-05-27 23:06:34 +00005276
5277 // If we are in an Objective-C method inside a class that has a superclass,
5278 // add "super" as an option.
5279 if (ObjCMethodDecl *Method = getCurMethodDecl())
5280 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor6fc04132010-08-27 15:10:57 +00005281 if (Iface->getSuperClass()) {
Douglas Gregora817a192010-05-27 23:06:34 +00005282 Results.AddResult(Result("super"));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005283
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005284 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, None, Results);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005285 }
Douglas Gregora817a192010-05-27 23:06:34 +00005286
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005287 if (getLangOpts().CPlusPlus11)
Douglas Gregord8c61782012-02-15 15:34:24 +00005288 addThisCompletion(*this, Results);
5289
Douglas Gregora817a192010-05-27 23:06:34 +00005290 Results.ExitScope();
5291
5292 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005293 AddMacroResults(PP, Results, false);
Douglas Gregor50832e02010-09-20 22:39:41 +00005294 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005295 Results.data(), Results.size());
Douglas Gregora817a192010-05-27 23:06:34 +00005296
5297}
5298
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005299void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005300 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005301 bool AtArgumentExpression) {
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005302 ObjCInterfaceDecl *CDecl = 0;
5303 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5304 // Figure out which interface we're in.
5305 CDecl = CurMethod->getClassInterface();
5306 if (!CDecl)
5307 return;
5308
5309 // Find the superclass of this class.
5310 CDecl = CDecl->getSuperClass();
5311 if (!CDecl)
5312 return;
5313
5314 if (CurMethod->isInstanceMethod()) {
5315 // We are inside an instance method, which means that the message
5316 // send [super ...] is actually calling an instance method on the
Douglas Gregor392a84b2010-10-13 21:24:53 +00005317 // current object.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005318 return CodeCompleteObjCInstanceMessage(S, 0, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005319 AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005320 CDecl);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005321 }
5322
5323 // Fall through to send to the superclass in CDecl.
5324 } else {
5325 // "super" may be the name of a type or variable. Figure out which
5326 // it is.
Argyrios Kyrtzidis3e56dd42013-03-14 22:56:43 +00005327 IdentifierInfo *Super = getSuperIdentifier();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005328 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5329 LookupOrdinaryName);
5330 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5331 // "super" names an interface. Use it.
5332 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCall8b07ec22010-05-15 11:32:37 +00005333 if (const ObjCObjectType *Iface
5334 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5335 CDecl = Iface->getInterface();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005336 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5337 // "super" names an unresolved type; we can't be more specific.
5338 } else {
5339 // Assume that "super" names some kind of value and parse that way.
5340 CXXScopeSpec SS;
Abramo Bagnara7945c982012-01-27 09:46:47 +00005341 SourceLocation TemplateKWLoc;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005342 UnqualifiedId id;
5343 id.setIdentifier(Super, SuperLoc);
Abramo Bagnara7945c982012-01-27 09:46:47 +00005344 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5345 false, false);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005346 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005347 SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005348 AtArgumentExpression);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005349 }
5350
5351 // Fall through
5352 }
5353
John McCallba7bf592010-08-24 05:47:05 +00005354 ParsedType Receiver;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005355 if (CDecl)
John McCallba7bf592010-08-24 05:47:05 +00005356 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005357 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005358 AtArgumentExpression,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005359 /*IsSuper=*/true);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005360}
5361
Douglas Gregor74661272010-09-21 00:03:25 +00005362/// \brief Given a set of code-completion results for the argument of a message
5363/// send, determine the preferred type (if any) for that argument expression.
5364static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5365 unsigned NumSelIdents) {
5366 typedef CodeCompletionResult Result;
5367 ASTContext &Context = Results.getSema().Context;
5368
5369 QualType PreferredType;
5370 unsigned BestPriority = CCP_Unlikely * 2;
5371 Result *ResultsData = Results.data();
5372 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5373 Result &R = ResultsData[I];
5374 if (R.Kind == Result::RK_Declaration &&
5375 isa<ObjCMethodDecl>(R.Declaration)) {
5376 if (R.Priority <= BestPriority) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00005377 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregor74661272010-09-21 00:03:25 +00005378 if (NumSelIdents <= Method->param_size()) {
5379 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
5380 ->getType();
5381 if (R.Priority < BestPriority || PreferredType.isNull()) {
5382 BestPriority = R.Priority;
5383 PreferredType = MyPreferredType;
5384 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5385 MyPreferredType)) {
5386 PreferredType = QualType();
5387 }
5388 }
5389 }
5390 }
5391 }
5392
5393 return PreferredType;
5394}
5395
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005396static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5397 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005398 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005399 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005400 bool IsSuper,
5401 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005402 typedef CodeCompletionResult Result;
Douglas Gregor8ce33212009-11-17 17:59:40 +00005403 ObjCInterfaceDecl *CDecl = 0;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005404
Douglas Gregor8ce33212009-11-17 17:59:40 +00005405 // If the given name refers to an interface type, retrieve the
5406 // corresponding declaration.
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005407 if (Receiver) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005408 QualType T = SemaRef.GetTypeFromParser(Receiver, 0);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005409 if (!T.isNull())
John McCall8b07ec22010-05-15 11:32:37 +00005410 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5411 CDecl = Interface->getInterface();
Douglas Gregor8ce33212009-11-17 17:59:40 +00005412 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005413
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005414 // Add all of the factory methods in this Objective-C class, its protocols,
5415 // superclasses, categories, implementation, etc.
Steve Naroffeae65032009-11-07 02:08:14 +00005416 Results.EnterNewScope();
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005417
Douglas Gregor6fc04132010-08-27 15:10:57 +00005418 // If this is a send-to-super, try to add the special "super" send
5419 // completion.
5420 if (IsSuper) {
5421 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005422 = AddSuperSendCompletion(SemaRef, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005423 Results.Ignore(SuperMethod);
5424 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005425
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005426 // If we're inside an Objective-C method definition, prefer its selector to
5427 // others.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005428 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005429 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005430
Douglas Gregor1154e272010-09-16 16:06:31 +00005431 VisitedSelectorSet Selectors;
Douglas Gregor6285f752010-04-06 16:40:00 +00005432 if (CDecl)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005433 AddObjCMethods(CDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005434 SemaRef.CurContext, Selectors, AtArgumentExpression,
5435 Results);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005436 else {
Douglas Gregor6285f752010-04-06 16:40:00 +00005437 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005438
Douglas Gregord720daf2010-04-06 17:30:22 +00005439 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005440 // pool from the AST file.
Axel Naumanndd433f02012-10-18 19:05:02 +00005441 if (SemaRef.getExternalSource()) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005442 for (uint32_t I = 0,
Axel Naumanndd433f02012-10-18 19:05:02 +00005443 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall75b960e2010-06-01 09:23:16 +00005444 I != N; ++I) {
Axel Naumanndd433f02012-10-18 19:05:02 +00005445 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005446 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005447 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005448
5449 SemaRef.ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005450 }
5451 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005452
5453 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5454 MEnd = SemaRef.MethodPool.end();
Sebastian Redl75d8a322010-08-02 23:18:59 +00005455 M != MEnd; ++M) {
5456 for (ObjCMethodList *MethList = &M->second.second;
5457 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005458 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005459 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005460 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005461
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005462 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005463 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005464 R.AllParametersAreInformative = false;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005465 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor6285f752010-04-06 16:40:00 +00005466 }
5467 }
5468 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005469
5470 Results.ExitScope();
5471}
Douglas Gregor6285f752010-04-06 16:40:00 +00005472
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005473void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005474 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005475 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005476 bool IsSuper) {
Douglas Gregor63745d52011-07-21 01:05:26 +00005477
5478 QualType T = this->GetTypeFromParser(Receiver);
5479
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005480 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005481 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005482 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005483 T, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005484
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005485 AddClassMessageCompletions(*this, S, Receiver, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005486 AtArgumentExpression, IsSuper, Results);
Douglas Gregor74661272010-09-21 00:03:25 +00005487
5488 // If we're actually at the argument expression (rather than prior to the
5489 // selector), we're actually performing code completion for an expression.
5490 // Determine whether we have a single, best method. If so, we can
5491 // code-complete the expression using the corresponding parameter type as
5492 // our preferred type, improving completion results.
5493 if (AtArgumentExpression) {
5494 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005495 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005496 if (PreferredType.isNull())
5497 CodeCompleteOrdinaryName(S, PCC_Expression);
5498 else
5499 CodeCompleteExpression(S, PreferredType);
5500 return;
5501 }
5502
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005503 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005504 Results.getCompletionContext(),
Douglas Gregor6fc04132010-08-27 15:10:57 +00005505 Results.data(), Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005506}
5507
Richard Trieu2bd04012011-09-09 02:00:50 +00005508void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005509 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005510 bool AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005511 ObjCInterfaceDecl *Super) {
John McCall276321a2010-08-25 06:19:51 +00005512 typedef CodeCompletionResult Result;
Steve Naroffeae65032009-11-07 02:08:14 +00005513
5514 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffeae65032009-11-07 02:08:14 +00005515
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005516 // If necessary, apply function/array conversion to the receiver.
5517 // C99 6.7.5.3p[7,8].
John Wiegley01296292011-04-08 18:41:53 +00005518 if (RecExpr) {
5519 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5520 if (Conv.isInvalid()) // conversion failed. bail.
5521 return;
5522 RecExpr = Conv.take();
5523 }
Douglas Gregor392a84b2010-10-13 21:24:53 +00005524 QualType ReceiverType = RecExpr? RecExpr->getType()
5525 : Super? Context.getObjCObjectPointerType(
5526 Context.getObjCInterfaceType(Super))
5527 : Context.getObjCIdType();
Steve Naroffeae65032009-11-07 02:08:14 +00005528
Douglas Gregordc520b02010-11-08 21:12:30 +00005529 // If we're messaging an expression with type "id" or "Class", check
5530 // whether we know something special about the receiver that allows
5531 // us to assume a more-specific receiver type.
5532 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType())
5533 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5534 if (ReceiverType->isObjCClassType())
5535 return CodeCompleteObjCClassMessage(S,
5536 ParsedType::make(Context.getObjCInterfaceType(IFace)),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005537 SelIdents,
Douglas Gregordc520b02010-11-08 21:12:30 +00005538 AtArgumentExpression, Super);
5539
5540 ReceiverType = Context.getObjCObjectPointerType(
5541 Context.getObjCInterfaceType(IFace));
5542 }
5543
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005544 // Build the set of methods we can see.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005545 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005546 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005547 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005548 ReceiverType, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005549
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005550 Results.EnterNewScope();
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005551
Douglas Gregor6fc04132010-08-27 15:10:57 +00005552 // If this is a send-to-super, try to add the special "super" send
5553 // completion.
Douglas Gregor392a84b2010-10-13 21:24:53 +00005554 if (Super) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005555 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005556 = AddSuperSendCompletion(*this, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005557 Results.Ignore(SuperMethod);
5558 }
5559
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005560 // If we're inside an Objective-C method definition, prefer its selector to
5561 // others.
5562 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5563 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005564
Douglas Gregor1154e272010-09-16 16:06:31 +00005565 // Keep track of the selectors we've already added.
5566 VisitedSelectorSet Selectors;
5567
Douglas Gregora3329fa2009-11-18 00:06:18 +00005568 // Handle messages to Class. This really isn't a message to an instance
5569 // method, so we treat it the same way we would treat a message send to a
5570 // class method.
5571 if (ReceiverType->isObjCClassType() ||
5572 ReceiverType->isObjCQualifiedClassType()) {
5573 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5574 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005575 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005576 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005577 }
5578 }
5579 // Handle messages to a qualified ID ("id<foo>").
5580 else if (const ObjCObjectPointerType *QualID
5581 = ReceiverType->getAsObjCQualifiedIdType()) {
5582 // Search protocols for instance methods.
5583 for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(),
5584 E = QualID->qual_end();
5585 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005586 AddObjCMethods(*I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005587 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005588 }
5589 // Handle messages to a pointer to interface type.
5590 else if (const ObjCObjectPointerType *IFacePtr
5591 = ReceiverType->getAsObjCInterfacePointerType()) {
5592 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregorc8537c52009-11-19 07:41:15 +00005593 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005594 CurContext, Selectors, AtArgumentExpression,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005595 Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005596
5597 // Search protocols for instance methods.
5598 for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(),
5599 E = IFacePtr->qual_end();
5600 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005601 AddObjCMethods(*I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005602 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005603 }
Douglas Gregor6285f752010-04-06 16:40:00 +00005604 // Handle messages to "id".
5605 else if (ReceiverType->isObjCIdType()) {
Douglas Gregord720daf2010-04-06 17:30:22 +00005606 // We're messaging "id", so provide all instance methods we know
5607 // about as code-completion results.
5608
5609 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005610 // pool from the AST file.
Douglas Gregord720daf2010-04-06 17:30:22 +00005611 if (ExternalSource) {
John McCall75b960e2010-06-01 09:23:16 +00005612 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5613 I != N; ++I) {
5614 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00005615 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005616 continue;
5617
Sebastian Redl75d8a322010-08-02 23:18:59 +00005618 ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005619 }
5620 }
5621
Sebastian Redl75d8a322010-08-02 23:18:59 +00005622 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5623 MEnd = MethodPool.end();
5624 M != MEnd; ++M) {
5625 for (ObjCMethodList *MethList = &M->second.first;
5626 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005627 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005628 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005629 continue;
Douglas Gregor1154e272010-09-16 16:06:31 +00005630
5631 if (!Selectors.insert(MethList->Method->getSelector()))
5632 continue;
5633
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005634 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005635 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005636 R.AllParametersAreInformative = false;
5637 Results.MaybeAddResult(R, CurContext);
5638 }
5639 }
5640 }
Steve Naroffeae65032009-11-07 02:08:14 +00005641 Results.ExitScope();
Douglas Gregor74661272010-09-21 00:03:25 +00005642
5643
5644 // If we're actually at the argument expression (rather than prior to the
5645 // selector), we're actually performing code completion for an expression.
5646 // Determine whether we have a single, best method. If so, we can
5647 // code-complete the expression using the corresponding parameter type as
5648 // our preferred type, improving completion results.
5649 if (AtArgumentExpression) {
5650 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005651 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005652 if (PreferredType.isNull())
5653 CodeCompleteOrdinaryName(S, PCC_Expression);
5654 else
5655 CodeCompleteExpression(S, PreferredType);
5656 return;
5657 }
5658
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005659 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005660 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005661 Results.data(),Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005662}
Douglas Gregorbaf69612009-11-18 04:19:12 +00005663
Douglas Gregor68762e72010-08-23 21:17:50 +00005664void Sema::CodeCompleteObjCForCollection(Scope *S,
5665 DeclGroupPtrTy IterationVar) {
5666 CodeCompleteExpressionData Data;
5667 Data.ObjCCollection = true;
5668
5669 if (IterationVar.getAsOpaquePtr()) {
Serge Pavlov9ddb76e2013-08-27 13:15:56 +00005670 DeclGroupRef DG = IterationVar.get();
Douglas Gregor68762e72010-08-23 21:17:50 +00005671 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5672 if (*I)
5673 Data.IgnoreDecls.push_back(*I);
5674 }
5675 }
5676
5677 CodeCompleteExpression(S, Data);
5678}
5679
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005680void Sema::CodeCompleteObjCSelector(Scope *S,
5681 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00005682 // If we have an external source, load the entire class method
5683 // pool from the AST file.
5684 if (ExternalSource) {
5685 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5686 I != N; ++I) {
5687 Selector Sel = ExternalSource->GetExternalSelector(I);
5688 if (Sel.isNull() || MethodPool.count(Sel))
5689 continue;
5690
5691 ReadMethodPool(Sel);
5692 }
5693 }
5694
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005695 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005696 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005697 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor67c692c2010-08-26 15:07:07 +00005698 Results.EnterNewScope();
5699 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5700 MEnd = MethodPool.end();
5701 M != MEnd; ++M) {
5702
5703 Selector Sel = M->first;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005704 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents))
Douglas Gregor67c692c2010-08-26 15:07:07 +00005705 continue;
5706
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005707 CodeCompletionBuilder Builder(Results.getAllocator(),
5708 Results.getCodeCompletionTUInfo());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005709 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005710 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005711 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005712 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005713 continue;
5714 }
5715
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005716 std::string Accumulator;
Douglas Gregor67c692c2010-08-26 15:07:07 +00005717 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005718 if (I == SelIdents.size()) {
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005719 if (!Accumulator.empty()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005720 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005721 Accumulator));
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005722 Accumulator.clear();
5723 }
5724 }
5725
Benjamin Kramer632500c2011-07-26 16:59:25 +00005726 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005727 Accumulator += ':';
Douglas Gregor67c692c2010-08-26 15:07:07 +00005728 }
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005729 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005730 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005731 }
5732 Results.ExitScope();
5733
5734 HandleCodeCompleteResults(this, CodeCompleter,
5735 CodeCompletionContext::CCC_SelectorName,
5736 Results.data(), Results.size());
5737}
5738
Douglas Gregorbaf69612009-11-18 04:19:12 +00005739/// \brief Add all of the protocol declarations that we find in the given
5740/// (translation unit) context.
5741static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005742 bool OnlyForwardDeclarations,
Douglas Gregorbaf69612009-11-18 04:19:12 +00005743 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005744 typedef CodeCompletionResult Result;
Douglas Gregorbaf69612009-11-18 04:19:12 +00005745
5746 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5747 DEnd = Ctx->decls_end();
5748 D != DEnd; ++D) {
5749 // Record any protocols we find.
5750 if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*D))
Douglas Gregore6e48b12012-01-01 19:29:29 +00005751 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005752 Results.AddResult(Result(Proto, Results.getBasePriority(Proto), 0),
5753 CurContext, 0, false);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005754 }
5755}
5756
5757void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5758 unsigned NumProtocols) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005759 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005760 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005761 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005762
Douglas Gregora3b23b02010-12-09 21:44:02 +00005763 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5764 Results.EnterNewScope();
5765
5766 // Tell the result set to ignore all of the protocols we have
5767 // already seen.
5768 // FIXME: This doesn't work when caching code-completion results.
5769 for (unsigned I = 0; I != NumProtocols; ++I)
5770 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5771 Protocols[I].second))
5772 Results.Ignore(Protocol);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005773
Douglas Gregora3b23b02010-12-09 21:44:02 +00005774 // Add all protocols.
5775 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5776 Results);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005777
Douglas Gregora3b23b02010-12-09 21:44:02 +00005778 Results.ExitScope();
5779 }
5780
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005781 HandleCodeCompleteResults(this, CodeCompleter,
5782 CodeCompletionContext::CCC_ObjCProtocolName,
5783 Results.data(),Results.size());
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005784}
5785
5786void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005787 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005788 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005789 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005790
Douglas Gregora3b23b02010-12-09 21:44:02 +00005791 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5792 Results.EnterNewScope();
5793
5794 // Add all protocols.
5795 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5796 Results);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005797
Douglas Gregora3b23b02010-12-09 21:44:02 +00005798 Results.ExitScope();
5799 }
5800
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005801 HandleCodeCompleteResults(this, CodeCompleter,
5802 CodeCompletionContext::CCC_ObjCProtocolName,
5803 Results.data(),Results.size());
Douglas Gregorbaf69612009-11-18 04:19:12 +00005804}
Douglas Gregor49c22a72009-11-18 16:26:39 +00005805
5806/// \brief Add all of the Objective-C interface declarations that we find in
5807/// the given (translation unit) context.
5808static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5809 bool OnlyForwardDeclarations,
5810 bool OnlyUnimplemented,
5811 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005812 typedef CodeCompletionResult Result;
Douglas Gregor49c22a72009-11-18 16:26:39 +00005813
5814 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5815 DEnd = Ctx->decls_end();
5816 D != DEnd; ++D) {
Douglas Gregor1c283312010-08-11 12:19:30 +00005817 // Record any interfaces we find.
5818 if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*D))
Douglas Gregordc9166c2011-12-15 20:29:51 +00005819 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregor1c283312010-08-11 12:19:30 +00005820 (!OnlyUnimplemented || !Class->getImplementation()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005821 Results.AddResult(Result(Class, Results.getBasePriority(Class), 0),
5822 CurContext, 0, false);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005823 }
5824}
5825
5826void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005827 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005828 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005829 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005830 Results.EnterNewScope();
5831
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005832 if (CodeCompleter->includeGlobals()) {
5833 // Add all classes.
5834 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5835 false, Results);
5836 }
5837
Douglas Gregor49c22a72009-11-18 16:26:39 +00005838 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005839
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005840 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005841 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005842 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005843}
5844
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005845void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5846 SourceLocation ClassNameLoc) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005847 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005848 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005849 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005850 Results.EnterNewScope();
5851
5852 // Make sure that we ignore the class we're currently defining.
5853 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005854 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005855 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor49c22a72009-11-18 16:26:39 +00005856 Results.Ignore(CurClass);
5857
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005858 if (CodeCompleter->includeGlobals()) {
5859 // Add all classes.
5860 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5861 false, Results);
5862 }
5863
Douglas Gregor49c22a72009-11-18 16:26:39 +00005864 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005865
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005866 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005867 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005868 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005869}
5870
5871void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005872 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005873 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005874 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005875 Results.EnterNewScope();
5876
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005877 if (CodeCompleter->includeGlobals()) {
5878 // Add all unimplemented classes.
5879 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5880 true, Results);
5881 }
5882
Douglas Gregor49c22a72009-11-18 16:26:39 +00005883 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005884
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005885 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005886 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005887 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005888}
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005889
5890void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005891 IdentifierInfo *ClassName,
5892 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005893 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005894
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005895 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005896 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005897 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005898
5899 // Ignore any categories we find that have already been implemented by this
5900 // interface.
5901 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5902 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005903 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005904 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
5905 for (ObjCInterfaceDecl::visible_categories_iterator
5906 Cat = Class->visible_categories_begin(),
5907 CatEnd = Class->visible_categories_end();
5908 Cat != CatEnd; ++Cat) {
5909 CategoryNames.insert(Cat->getIdentifier());
5910 }
5911 }
5912
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005913 // Add all of the categories we know about.
5914 Results.EnterNewScope();
5915 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
5916 for (DeclContext::decl_iterator D = TU->decls_begin(),
5917 DEnd = TU->decls_end();
5918 D != DEnd; ++D)
5919 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(*D))
5920 if (CategoryNames.insert(Category->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005921 Results.AddResult(Result(Category, Results.getBasePriority(Category),0),
5922 CurContext, 0, false);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005923 Results.ExitScope();
5924
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005925 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005926 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005927 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005928}
5929
5930void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005931 IdentifierInfo *ClassName,
5932 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005933 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005934
5935 // Find the corresponding interface. If we couldn't find the interface, the
5936 // program itself is ill-formed. However, we'll try to be helpful still by
5937 // providing the list of all of the categories we know about.
5938 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005939 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005940 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5941 if (!Class)
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005942 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005943
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005944 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005945 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005946 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005947
5948 // Add all of the categories that have have corresponding interface
5949 // declarations in this class and any of its superclasses, except for
5950 // already-implemented categories in the class itself.
5951 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5952 Results.EnterNewScope();
5953 bool IgnoreImplemented = true;
5954 while (Class) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005955 for (ObjCInterfaceDecl::visible_categories_iterator
5956 Cat = Class->visible_categories_begin(),
5957 CatEnd = Class->visible_categories_end();
5958 Cat != CatEnd; ++Cat) {
5959 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
5960 CategoryNames.insert(Cat->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005961 Results.AddResult(Result(*Cat, Results.getBasePriority(*Cat), 0),
5962 CurContext, 0, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005963 }
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005964
5965 Class = Class->getSuperClass();
5966 IgnoreImplemented = false;
5967 }
5968 Results.ExitScope();
5969
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005970 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005971 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005972 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005973}
Douglas Gregor5d649882009-11-18 22:32:06 +00005974
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005975void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005976 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005977 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005978 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00005979
5980 // Figure out where this @synthesize lives.
5981 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005982 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00005983 if (!Container ||
5984 (!isa<ObjCImplementationDecl>(Container) &&
5985 !isa<ObjCCategoryImplDecl>(Container)))
5986 return;
5987
5988 // Ignore any properties that have already been implemented.
Douglas Gregor9b4f3702012-06-12 13:44:08 +00005989 Container = getContainerDef(Container);
5990 for (DeclContext::decl_iterator D = Container->decls_begin(),
Douglas Gregor5d649882009-11-18 22:32:06 +00005991 DEnd = Container->decls_end();
5992 D != DEnd; ++D)
5993 if (ObjCPropertyImplDecl *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(*D))
5994 Results.Ignore(PropertyImpl->getPropertyDecl());
5995
5996 // Add any properties that we find.
Douglas Gregorb888acf2010-12-09 23:01:55 +00005997 AddedPropertiesSet AddedProperties;
Douglas Gregor5d649882009-11-18 22:32:06 +00005998 Results.EnterNewScope();
5999 if (ObjCImplementationDecl *ClassImpl
6000 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor95147142011-05-05 15:50:42 +00006001 AddObjCProperties(ClassImpl->getClassInterface(), false,
6002 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00006003 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00006004 else
6005 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor95147142011-05-05 15:50:42 +00006006 false, /*AllowNullaryMethods=*/false, CurContext,
6007 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00006008 Results.ExitScope();
6009
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006010 HandleCodeCompleteResults(this, CodeCompleter,
6011 CodeCompletionContext::CCC_Other,
6012 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006013}
6014
6015void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006016 IdentifierInfo *PropertyName) {
John McCall276321a2010-08-25 06:19:51 +00006017 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006018 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006019 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006020 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00006021
6022 // Figure out where this @synthesize lives.
6023 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006024 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00006025 if (!Container ||
6026 (!isa<ObjCImplementationDecl>(Container) &&
6027 !isa<ObjCCategoryImplDecl>(Container)))
6028 return;
6029
6030 // Figure out which interface we're looking into.
6031 ObjCInterfaceDecl *Class = 0;
6032 if (ObjCImplementationDecl *ClassImpl
6033 = dyn_cast<ObjCImplementationDecl>(Container))
6034 Class = ClassImpl->getClassInterface();
6035 else
6036 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
6037 ->getClassInterface();
6038
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006039 // Determine the type of the property we're synthesizing.
6040 QualType PropertyType = Context.getObjCIdType();
6041 if (Class) {
6042 if (ObjCPropertyDecl *Property
6043 = Class->FindPropertyDeclaration(PropertyName)) {
6044 PropertyType
6045 = Property->getType().getNonReferenceType().getUnqualifiedType();
6046
6047 // Give preference to ivars
6048 Results.setPreferredType(PropertyType);
6049 }
6050 }
6051
Douglas Gregor5d649882009-11-18 22:32:06 +00006052 // Add all of the instance variables in this class and its superclasses.
6053 Results.EnterNewScope();
Douglas Gregor331faa02011-04-18 14:13:53 +00006054 bool SawSimilarlyNamedIvar = false;
6055 std::string NameWithPrefix;
6056 NameWithPrefix += '_';
Benjamin Kramer632500c2011-07-26 16:59:25 +00006057 NameWithPrefix += PropertyName->getName();
Douglas Gregor331faa02011-04-18 14:13:53 +00006058 std::string NameWithSuffix = PropertyName->getName().str();
6059 NameWithSuffix += '_';
Douglas Gregor5d649882009-11-18 22:32:06 +00006060 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006061 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
6062 Ivar = Ivar->getNextIvar()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00006063 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), 0),
6064 CurContext, 0, false);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006065
Douglas Gregor331faa02011-04-18 14:13:53 +00006066 // Determine whether we've seen an ivar with a name similar to the
6067 // property.
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006068 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregor331faa02011-04-18 14:13:53 +00006069 NameWithPrefix == Ivar->getName() ||
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006070 NameWithSuffix == Ivar->getName())) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006071 SawSimilarlyNamedIvar = true;
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006072
6073 // Reduce the priority of this result by one, to give it a slight
6074 // advantage over other results whose names don't match so closely.
6075 if (Results.size() &&
6076 Results.data()[Results.size() - 1].Kind
6077 == CodeCompletionResult::RK_Declaration &&
6078 Results.data()[Results.size() - 1].Declaration == Ivar)
6079 Results.data()[Results.size() - 1].Priority--;
6080 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006081 }
Douglas Gregor5d649882009-11-18 22:32:06 +00006082 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006083
6084 if (!SawSimilarlyNamedIvar) {
6085 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006086 // an ivar of the appropriate type.
6087 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregor331faa02011-04-18 14:13:53 +00006088 typedef CodeCompletionResult Result;
6089 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006090 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6091 Priority,CXAvailability_Available);
Douglas Gregor331faa02011-04-18 14:13:53 +00006092
Douglas Gregor75acd922011-09-27 23:30:47 +00006093 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006094 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006095 Policy, Allocator));
Douglas Gregor331faa02011-04-18 14:13:53 +00006096 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6097 Results.AddResult(Result(Builder.TakeString(), Priority,
6098 CXCursor_ObjCIvarDecl));
6099 }
6100
Douglas Gregor5d649882009-11-18 22:32:06 +00006101 Results.ExitScope();
6102
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006103 HandleCodeCompleteResults(this, CodeCompleter,
6104 CodeCompletionContext::CCC_Other,
6105 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006106}
Douglas Gregor636a61e2010-04-07 00:21:17 +00006107
Douglas Gregor416b5752010-08-25 01:08:01 +00006108// Mapping from selectors to the methods that implement that selector, along
6109// with the "in original class" flag.
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006110typedef llvm::DenseMap<
6111 Selector, llvm::PointerIntPair<ObjCMethodDecl *, 1, bool> > KnownMethodsMap;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006112
6113/// \brief Find all of the methods that reside in the given container
6114/// (and its superclasses, protocols, etc.) that meet the given
6115/// criteria. Insert those methods into the map of known methods,
6116/// indexed by selector so they can be easily found.
6117static void FindImplementableMethods(ASTContext &Context,
6118 ObjCContainerDecl *Container,
6119 bool WantInstanceMethods,
6120 QualType ReturnType,
Douglas Gregor416b5752010-08-25 01:08:01 +00006121 KnownMethodsMap &KnownMethods,
6122 bool InOriginalClass = true) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006123 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006124 // Make sure we have a definition; that's what we'll walk.
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006125 if (!IFace->hasDefinition())
6126 return;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006127
6128 IFace = IFace->getDefinition();
6129 Container = IFace;
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006130
Douglas Gregor636a61e2010-04-07 00:21:17 +00006131 const ObjCList<ObjCProtocolDecl> &Protocols
6132 = IFace->getReferencedProtocols();
6133 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006134 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006135 I != E; ++I)
6136 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006137 KnownMethods, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006138
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006139 // Add methods from any class extensions and categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006140 for (ObjCInterfaceDecl::visible_categories_iterator
6141 Cat = IFace->visible_categories_begin(),
6142 CatEnd = IFace->visible_categories_end();
6143 Cat != CatEnd; ++Cat) {
6144 FindImplementableMethods(Context, *Cat, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006145 KnownMethods, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006146 }
6147
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006148 // Visit the superclass.
6149 if (IFace->getSuperClass())
6150 FindImplementableMethods(Context, IFace->getSuperClass(),
6151 WantInstanceMethods, ReturnType,
6152 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006153 }
6154
6155 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6156 // Recurse into protocols.
6157 const ObjCList<ObjCProtocolDecl> &Protocols
6158 = Category->getReferencedProtocols();
6159 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006160 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006161 I != E; ++I)
6162 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006163 KnownMethods, InOriginalClass);
6164
6165 // If this category is the original class, jump to the interface.
6166 if (InOriginalClass && Category->getClassInterface())
6167 FindImplementableMethods(Context, Category->getClassInterface(),
6168 WantInstanceMethods, ReturnType, KnownMethods,
6169 false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006170 }
6171
6172 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006173 // Make sure we have a definition; that's what we'll walk.
6174 if (!Protocol->hasDefinition())
6175 return;
6176 Protocol = Protocol->getDefinition();
6177 Container = Protocol;
6178
6179 // Recurse into protocols.
6180 const ObjCList<ObjCProtocolDecl> &Protocols
6181 = Protocol->getReferencedProtocols();
6182 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6183 E = Protocols.end();
6184 I != E; ++I)
6185 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6186 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006187 }
6188
6189 // Add methods in this container. This operation occurs last because
6190 // we want the methods from this container to override any methods
6191 // we've previously seen with the same selector.
6192 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
6193 MEnd = Container->meth_end();
6194 M != MEnd; ++M) {
David Blaikie2d7c57e2012-04-30 02:36:29 +00006195 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006196 if (!ReturnType.isNull() &&
Alp Toker314cc812014-01-25 16:55:45 +00006197 !Context.hasSameUnqualifiedType(ReturnType, M->getReturnType()))
Douglas Gregor636a61e2010-04-07 00:21:17 +00006198 continue;
6199
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006200 KnownMethods[M->getSelector()] =
6201 KnownMethodsMap::mapped_type(*M, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006202 }
6203 }
6204}
6205
Douglas Gregor669a25a2011-02-17 00:22:45 +00006206/// \brief Add the parenthesized return or parameter type chunk to a code
6207/// completion string.
6208static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor29979142012-04-10 18:35:07 +00006209 unsigned ObjCDeclQuals,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006210 ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006211 const PrintingPolicy &Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006212 CodeCompletionBuilder &Builder) {
6213 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor29979142012-04-10 18:35:07 +00006214 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6215 if (!Quals.empty())
6216 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor75acd922011-09-27 23:30:47 +00006217 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006218 Builder.getAllocator()));
6219 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6220}
6221
6222/// \brief Determine whether the given class is or inherits from a class by
6223/// the given name.
6224static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006225 StringRef Name) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006226 if (!Class)
6227 return false;
6228
6229 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6230 return true;
6231
6232 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6233}
6234
6235/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6236/// Key-Value Observing (KVO).
6237static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6238 bool IsInstanceMethod,
6239 QualType ReturnType,
6240 ASTContext &Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006241 VisitedSelectorSet &KnownSelectors,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006242 ResultBuilder &Results) {
6243 IdentifierInfo *PropName = Property->getIdentifier();
6244 if (!PropName || PropName->getLength() == 0)
6245 return;
6246
Douglas Gregor75acd922011-09-27 23:30:47 +00006247 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6248
Douglas Gregor669a25a2011-02-17 00:22:45 +00006249 // Builder that will create each code completion.
6250 typedef CodeCompletionResult Result;
6251 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006252 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor669a25a2011-02-17 00:22:45 +00006253
6254 // The selector table.
6255 SelectorTable &Selectors = Context.Selectors;
6256
6257 // The property name, copied into the code completion allocation region
6258 // on demand.
6259 struct KeyHolder {
6260 CodeCompletionAllocator &Allocator;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006261 StringRef Key;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006262 const char *CopiedKey;
6263
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006264 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006265 : Allocator(Allocator), Key(Key), CopiedKey(0) { }
6266
6267 operator const char *() {
6268 if (CopiedKey)
6269 return CopiedKey;
6270
6271 return CopiedKey = Allocator.CopyString(Key);
6272 }
6273 } Key(Allocator, PropName->getName());
6274
6275 // The uppercased name of the property name.
6276 std::string UpperKey = PropName->getName();
6277 if (!UpperKey.empty())
Jordan Rose4938f272013-02-09 10:09:43 +00006278 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006279
6280 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6281 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6282 Property->getType());
6283 bool ReturnTypeMatchesVoid
6284 = ReturnType.isNull() || ReturnType->isVoidType();
6285
6286 // Add the normal accessor -(type)key.
6287 if (IsInstanceMethod &&
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006288 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor669a25a2011-02-17 00:22:45 +00006289 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6290 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006291 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6292 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006293
6294 Builder.AddTypedTextChunk(Key);
6295 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6296 CXCursor_ObjCInstanceMethodDecl));
6297 }
6298
6299 // If we have an integral or boolean property (or the user has provided
6300 // an integral or boolean return type), add the accessor -(type)isKey.
6301 if (IsInstanceMethod &&
6302 ((!ReturnType.isNull() &&
6303 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6304 (ReturnType.isNull() &&
6305 (Property->getType()->isIntegerType() ||
6306 Property->getType()->isBooleanType())))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006307 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006308 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006309 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006310 if (ReturnType.isNull()) {
6311 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6312 Builder.AddTextChunk("BOOL");
6313 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6314 }
6315
6316 Builder.AddTypedTextChunk(
6317 Allocator.CopyString(SelectorId->getName()));
6318 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6319 CXCursor_ObjCInstanceMethodDecl));
6320 }
6321 }
6322
6323 // Add the normal mutator.
6324 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6325 !Property->getSetterMethodDecl()) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006326 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006327 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006328 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006329 if (ReturnType.isNull()) {
6330 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6331 Builder.AddTextChunk("void");
6332 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6333 }
6334
6335 Builder.AddTypedTextChunk(
6336 Allocator.CopyString(SelectorId->getName()));
6337 Builder.AddTypedTextChunk(":");
Douglas Gregor29979142012-04-10 18:35:07 +00006338 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6339 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006340 Builder.AddTextChunk(Key);
6341 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6342 CXCursor_ObjCInstanceMethodDecl));
6343 }
6344 }
6345
6346 // Indexed and unordered accessors
6347 unsigned IndexedGetterPriority = CCP_CodePattern;
6348 unsigned IndexedSetterPriority = CCP_CodePattern;
6349 unsigned UnorderedGetterPriority = CCP_CodePattern;
6350 unsigned UnorderedSetterPriority = CCP_CodePattern;
6351 if (const ObjCObjectPointerType *ObjCPointer
6352 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6353 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6354 // If this interface type is not provably derived from a known
6355 // collection, penalize the corresponding completions.
6356 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6357 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6358 if (!InheritsFromClassNamed(IFace, "NSArray"))
6359 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6360 }
6361
6362 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6363 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6364 if (!InheritsFromClassNamed(IFace, "NSSet"))
6365 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6366 }
6367 }
6368 } else {
6369 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6370 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6371 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6372 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6373 }
6374
6375 // Add -(NSUInteger)countOf<key>
6376 if (IsInstanceMethod &&
6377 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006378 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006379 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006380 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006381 if (ReturnType.isNull()) {
6382 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6383 Builder.AddTextChunk("NSUInteger");
6384 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6385 }
6386
6387 Builder.AddTypedTextChunk(
6388 Allocator.CopyString(SelectorId->getName()));
6389 Results.AddResult(Result(Builder.TakeString(),
6390 std::min(IndexedGetterPriority,
6391 UnorderedGetterPriority),
6392 CXCursor_ObjCInstanceMethodDecl));
6393 }
6394 }
6395
6396 // Indexed getters
6397 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6398 if (IsInstanceMethod &&
6399 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006400 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006401 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006402 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006403 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006404 if (ReturnType.isNull()) {
6405 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6406 Builder.AddTextChunk("id");
6407 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6408 }
6409
6410 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6411 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6412 Builder.AddTextChunk("NSUInteger");
6413 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6414 Builder.AddTextChunk("index");
6415 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6416 CXCursor_ObjCInstanceMethodDecl));
6417 }
6418 }
6419
6420 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6421 if (IsInstanceMethod &&
6422 (ReturnType.isNull() ||
6423 (ReturnType->isObjCObjectPointerType() &&
6424 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6425 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6426 ->getName() == "NSArray"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006427 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006428 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006429 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006430 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006431 if (ReturnType.isNull()) {
6432 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6433 Builder.AddTextChunk("NSArray *");
6434 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6435 }
6436
6437 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6438 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6439 Builder.AddTextChunk("NSIndexSet *");
6440 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6441 Builder.AddTextChunk("indexes");
6442 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6443 CXCursor_ObjCInstanceMethodDecl));
6444 }
6445 }
6446
6447 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6448 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006449 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006450 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006451 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006452 &Context.Idents.get("range")
6453 };
6454
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006455 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006456 if (ReturnType.isNull()) {
6457 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6458 Builder.AddTextChunk("void");
6459 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6460 }
6461
6462 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6463 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6464 Builder.AddPlaceholderChunk("object-type");
6465 Builder.AddTextChunk(" **");
6466 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6467 Builder.AddTextChunk("buffer");
6468 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6469 Builder.AddTypedTextChunk("range:");
6470 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6471 Builder.AddTextChunk("NSRange");
6472 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6473 Builder.AddTextChunk("inRange");
6474 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6475 CXCursor_ObjCInstanceMethodDecl));
6476 }
6477 }
6478
6479 // Mutable indexed accessors
6480
6481 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6482 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006483 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006484 IdentifierInfo *SelectorIds[2] = {
6485 &Context.Idents.get("insertObject"),
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006486 &Context.Idents.get(SelectorName)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006487 };
6488
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006489 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006490 if (ReturnType.isNull()) {
6491 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6492 Builder.AddTextChunk("void");
6493 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6494 }
6495
6496 Builder.AddTypedTextChunk("insertObject:");
6497 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6498 Builder.AddPlaceholderChunk("object-type");
6499 Builder.AddTextChunk(" *");
6500 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6501 Builder.AddTextChunk("object");
6502 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6503 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6504 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6505 Builder.AddPlaceholderChunk("NSUInteger");
6506 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6507 Builder.AddTextChunk("index");
6508 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6509 CXCursor_ObjCInstanceMethodDecl));
6510 }
6511 }
6512
6513 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6514 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006515 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006516 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006517 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006518 &Context.Idents.get("atIndexes")
6519 };
6520
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006521 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006522 if (ReturnType.isNull()) {
6523 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6524 Builder.AddTextChunk("void");
6525 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6526 }
6527
6528 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6529 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6530 Builder.AddTextChunk("NSArray *");
6531 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6532 Builder.AddTextChunk("array");
6533 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6534 Builder.AddTypedTextChunk("atIndexes:");
6535 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6536 Builder.AddPlaceholderChunk("NSIndexSet *");
6537 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6538 Builder.AddTextChunk("indexes");
6539 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6540 CXCursor_ObjCInstanceMethodDecl));
6541 }
6542 }
6543
6544 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6545 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006546 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006547 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006548 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006549 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006550 if (ReturnType.isNull()) {
6551 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6552 Builder.AddTextChunk("void");
6553 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6554 }
6555
6556 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6557 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6558 Builder.AddTextChunk("NSUInteger");
6559 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6560 Builder.AddTextChunk("index");
6561 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6562 CXCursor_ObjCInstanceMethodDecl));
6563 }
6564 }
6565
6566 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6567 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006568 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006569 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006570 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006571 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006572 if (ReturnType.isNull()) {
6573 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6574 Builder.AddTextChunk("void");
6575 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6576 }
6577
6578 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6579 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6580 Builder.AddTextChunk("NSIndexSet *");
6581 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6582 Builder.AddTextChunk("indexes");
6583 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6584 CXCursor_ObjCInstanceMethodDecl));
6585 }
6586 }
6587
6588 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6589 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006590 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006591 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006592 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006593 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006594 &Context.Idents.get("withObject")
6595 };
6596
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006597 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006598 if (ReturnType.isNull()) {
6599 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6600 Builder.AddTextChunk("void");
6601 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6602 }
6603
6604 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6605 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6606 Builder.AddPlaceholderChunk("NSUInteger");
6607 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6608 Builder.AddTextChunk("index");
6609 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6610 Builder.AddTypedTextChunk("withObject:");
6611 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6612 Builder.AddTextChunk("id");
6613 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6614 Builder.AddTextChunk("object");
6615 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6616 CXCursor_ObjCInstanceMethodDecl));
6617 }
6618 }
6619
6620 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6621 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006622 std::string SelectorName1
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006623 = (Twine("replace") + UpperKey + "AtIndexes").str();
6624 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006625 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006626 &Context.Idents.get(SelectorName1),
6627 &Context.Idents.get(SelectorName2)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006628 };
6629
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006630 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006631 if (ReturnType.isNull()) {
6632 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6633 Builder.AddTextChunk("void");
6634 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6635 }
6636
6637 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6638 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6639 Builder.AddPlaceholderChunk("NSIndexSet *");
6640 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6641 Builder.AddTextChunk("indexes");
6642 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6643 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6644 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6645 Builder.AddTextChunk("NSArray *");
6646 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6647 Builder.AddTextChunk("array");
6648 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6649 CXCursor_ObjCInstanceMethodDecl));
6650 }
6651 }
6652
6653 // Unordered getters
6654 // - (NSEnumerator *)enumeratorOfKey
6655 if (IsInstanceMethod &&
6656 (ReturnType.isNull() ||
6657 (ReturnType->isObjCObjectPointerType() &&
6658 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6659 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6660 ->getName() == "NSEnumerator"))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006661 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006662 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006663 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006664 if (ReturnType.isNull()) {
6665 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6666 Builder.AddTextChunk("NSEnumerator *");
6667 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6668 }
6669
6670 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6671 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6672 CXCursor_ObjCInstanceMethodDecl));
6673 }
6674 }
6675
6676 // - (type *)memberOfKey:(type *)object
6677 if (IsInstanceMethod &&
6678 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006679 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006680 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006681 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006682 if (ReturnType.isNull()) {
6683 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6684 Builder.AddPlaceholderChunk("object-type");
6685 Builder.AddTextChunk(" *");
6686 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6687 }
6688
6689 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6690 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6691 if (ReturnType.isNull()) {
6692 Builder.AddPlaceholderChunk("object-type");
6693 Builder.AddTextChunk(" *");
6694 } else {
6695 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006696 Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006697 Builder.getAllocator()));
6698 }
6699 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6700 Builder.AddTextChunk("object");
6701 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6702 CXCursor_ObjCInstanceMethodDecl));
6703 }
6704 }
6705
6706 // Mutable unordered accessors
6707 // - (void)addKeyObject:(type *)object
6708 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006709 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006710 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006711 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006712 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006713 if (ReturnType.isNull()) {
6714 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6715 Builder.AddTextChunk("void");
6716 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6717 }
6718
6719 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6720 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6721 Builder.AddPlaceholderChunk("object-type");
6722 Builder.AddTextChunk(" *");
6723 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6724 Builder.AddTextChunk("object");
6725 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6726 CXCursor_ObjCInstanceMethodDecl));
6727 }
6728 }
6729
6730 // - (void)addKey:(NSSet *)objects
6731 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006732 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006733 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006734 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006735 if (ReturnType.isNull()) {
6736 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6737 Builder.AddTextChunk("void");
6738 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6739 }
6740
6741 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6742 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6743 Builder.AddTextChunk("NSSet *");
6744 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6745 Builder.AddTextChunk("objects");
6746 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6747 CXCursor_ObjCInstanceMethodDecl));
6748 }
6749 }
6750
6751 // - (void)removeKeyObject:(type *)object
6752 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006753 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006754 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006755 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006756 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006757 if (ReturnType.isNull()) {
6758 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6759 Builder.AddTextChunk("void");
6760 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6761 }
6762
6763 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6764 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6765 Builder.AddPlaceholderChunk("object-type");
6766 Builder.AddTextChunk(" *");
6767 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6768 Builder.AddTextChunk("object");
6769 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6770 CXCursor_ObjCInstanceMethodDecl));
6771 }
6772 }
6773
6774 // - (void)removeKey:(NSSet *)objects
6775 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006776 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006777 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006778 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006779 if (ReturnType.isNull()) {
6780 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6781 Builder.AddTextChunk("void");
6782 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6783 }
6784
6785 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6786 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6787 Builder.AddTextChunk("NSSet *");
6788 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6789 Builder.AddTextChunk("objects");
6790 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6791 CXCursor_ObjCInstanceMethodDecl));
6792 }
6793 }
6794
6795 // - (void)intersectKey:(NSSet *)objects
6796 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006797 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006798 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006799 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006800 if (ReturnType.isNull()) {
6801 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6802 Builder.AddTextChunk("void");
6803 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6804 }
6805
6806 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6807 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6808 Builder.AddTextChunk("NSSet *");
6809 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6810 Builder.AddTextChunk("objects");
6811 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6812 CXCursor_ObjCInstanceMethodDecl));
6813 }
6814 }
6815
6816 // Key-Value Observing
6817 // + (NSSet *)keyPathsForValuesAffectingKey
6818 if (!IsInstanceMethod &&
6819 (ReturnType.isNull() ||
6820 (ReturnType->isObjCObjectPointerType() &&
6821 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6822 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6823 ->getName() == "NSSet"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006824 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006825 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006826 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006827 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006828 if (ReturnType.isNull()) {
6829 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6830 Builder.AddTextChunk("NSSet *");
6831 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6832 }
6833
6834 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6835 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor857bcda2011-06-02 04:02:27 +00006836 CXCursor_ObjCClassMethodDecl));
6837 }
6838 }
6839
6840 // + (BOOL)automaticallyNotifiesObserversForKey
6841 if (!IsInstanceMethod &&
6842 (ReturnType.isNull() ||
6843 ReturnType->isIntegerType() ||
6844 ReturnType->isBooleanType())) {
6845 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006846 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor857bcda2011-06-02 04:02:27 +00006847 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6848 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6849 if (ReturnType.isNull()) {
6850 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6851 Builder.AddTextChunk("BOOL");
6852 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6853 }
6854
6855 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6856 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6857 CXCursor_ObjCClassMethodDecl));
Douglas Gregor669a25a2011-02-17 00:22:45 +00006858 }
6859 }
6860}
6861
Douglas Gregor636a61e2010-04-07 00:21:17 +00006862void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6863 bool IsInstanceMethod,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006864 ParsedType ReturnTy) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006865 // Determine the return type of the method we're declaring, if
6866 // provided.
6867 QualType ReturnType = GetTypeFromParser(ReturnTy);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006868 Decl *IDecl = 0;
6869 if (CurContext->isObjCContainer()) {
6870 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6871 IDecl = cast<Decl>(OCD);
6872 }
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006873 // Determine where we should start searching for methods.
6874 ObjCContainerDecl *SearchDecl = 0;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006875 bool IsInImplementation = false;
John McCall48871652010-08-21 09:40:31 +00006876 if (Decl *D = IDecl) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006877 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6878 SearchDecl = Impl->getClassInterface();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006879 IsInImplementation = true;
6880 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006881 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006882 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006883 IsInImplementation = true;
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006884 } else
Douglas Gregor636a61e2010-04-07 00:21:17 +00006885 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006886 }
6887
6888 if (!SearchDecl && S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00006889 if (DeclContext *DC = S->getEntity())
Douglas Gregor636a61e2010-04-07 00:21:17 +00006890 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006891 }
6892
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006893 if (!SearchDecl) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006894 HandleCodeCompleteResults(this, CodeCompleter,
6895 CodeCompletionContext::CCC_Other,
6896 0, 0);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006897 return;
6898 }
6899
6900 // Find all of the methods that we could declare/implement here.
6901 KnownMethodsMap KnownMethods;
6902 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006903 ReturnType, KnownMethods);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006904
Douglas Gregor636a61e2010-04-07 00:21:17 +00006905 // Add declarations or definitions for each of the known methods.
John McCall276321a2010-08-25 06:19:51 +00006906 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006907 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006908 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006909 CodeCompletionContext::CCC_Other);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006910 Results.EnterNewScope();
Douglas Gregor75acd922011-09-27 23:30:47 +00006911 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006912 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6913 MEnd = KnownMethods.end();
6914 M != MEnd; ++M) {
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006915 ObjCMethodDecl *Method = M->second.getPointer();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006916 CodeCompletionBuilder Builder(Results.getAllocator(),
6917 Results.getCodeCompletionTUInfo());
Douglas Gregor636a61e2010-04-07 00:21:17 +00006918
6919 // If the result type was not already provided, add it to the
6920 // pattern as (type).
Douglas Gregor669a25a2011-02-17 00:22:45 +00006921 if (ReturnType.isNull())
Alp Toker314cc812014-01-25 16:55:45 +00006922 AddObjCPassingTypeChunk(Method->getReturnType(),
6923 Method->getObjCDeclQualifier(), Context, Policy,
6924 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006925
6926 Selector Sel = Method->getSelector();
6927
6928 // Add the first part of the selector to the pattern.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006929 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006930 Sel.getNameForSlot(0)));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006931
6932 // Add parameters to the pattern.
6933 unsigned I = 0;
6934 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6935 PEnd = Method->param_end();
6936 P != PEnd; (void)++P, ++I) {
6937 // Add the part of the selector name.
6938 if (I == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006939 Builder.AddTypedTextChunk(":");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006940 else if (I < Sel.getNumArgs()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006941 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6942 Builder.AddTypedTextChunk(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006943 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006944 } else
6945 break;
6946
6947 // Add the parameter type.
Douglas Gregor29979142012-04-10 18:35:07 +00006948 AddObjCPassingTypeChunk((*P)->getOriginalType(),
6949 (*P)->getObjCDeclQualifier(),
6950 Context, Policy,
Douglas Gregor75acd922011-09-27 23:30:47 +00006951 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006952
6953 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006954 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006955 }
6956
6957 if (Method->isVariadic()) {
6958 if (Method->param_size() > 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006959 Builder.AddChunk(CodeCompletionString::CK_Comma);
6960 Builder.AddTextChunk("...");
Douglas Gregor400f5972010-08-31 05:13:43 +00006961 }
Douglas Gregor636a61e2010-04-07 00:21:17 +00006962
Douglas Gregord37c59d2010-05-28 00:57:46 +00006963 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006964 // We will be defining the method here, so add a compound statement.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006965 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6966 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6967 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Alp Toker314cc812014-01-25 16:55:45 +00006968 if (!Method->getReturnType()->isVoidType()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006969 // If the result type is not void, add a return clause.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006970 Builder.AddTextChunk("return");
6971 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6972 Builder.AddPlaceholderChunk("expression");
6973 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006974 } else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006975 Builder.AddPlaceholderChunk("statements");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006976
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006977 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6978 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006979 }
6980
Douglas Gregor416b5752010-08-25 01:08:01 +00006981 unsigned Priority = CCP_CodePattern;
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006982 if (!M->second.getInt())
Douglas Gregor416b5752010-08-25 01:08:01 +00006983 Priority += CCD_InBaseClass;
6984
Douglas Gregor78254c82012-03-27 23:34:16 +00006985 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006986 }
6987
Douglas Gregor669a25a2011-02-17 00:22:45 +00006988 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
6989 // the properties in this class and its categories.
David Blaikiebbafb8a2012-03-11 07:00:24 +00006990 if (Context.getLangOpts().ObjC2) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006991 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006992 Containers.push_back(SearchDecl);
6993
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006994 VisitedSelectorSet KnownSelectors;
6995 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6996 MEnd = KnownMethods.end();
6997 M != MEnd; ++M)
6998 KnownSelectors.insert(M->first);
6999
7000
Douglas Gregor669a25a2011-02-17 00:22:45 +00007001 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
7002 if (!IFace)
7003 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
7004 IFace = Category->getClassInterface();
7005
7006 if (IFace) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00007007 for (ObjCInterfaceDecl::visible_categories_iterator
7008 Cat = IFace->visible_categories_begin(),
7009 CatEnd = IFace->visible_categories_end();
7010 Cat != CatEnd; ++Cat) {
7011 Containers.push_back(*Cat);
7012 }
Douglas Gregor669a25a2011-02-17 00:22:45 +00007013 }
7014
7015 for (unsigned I = 0, N = Containers.size(); I != N; ++I) {
7016 for (ObjCContainerDecl::prop_iterator P = Containers[I]->prop_begin(),
7017 PEnd = Containers[I]->prop_end();
7018 P != PEnd; ++P) {
David Blaikie40ed2972012-06-06 20:45:41 +00007019 AddObjCKeyValueCompletions(*P, IsInstanceMethod, ReturnType, Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00007020 KnownSelectors, Results);
Douglas Gregor669a25a2011-02-17 00:22:45 +00007021 }
7022 }
7023 }
7024
Douglas Gregor636a61e2010-04-07 00:21:17 +00007025 Results.ExitScope();
7026
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007027 HandleCodeCompleteResults(this, CodeCompleter,
7028 CodeCompletionContext::CCC_Other,
7029 Results.data(),Results.size());
Douglas Gregor636a61e2010-04-07 00:21:17 +00007030}
Douglas Gregor95887f92010-07-08 23:20:03 +00007031
7032void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
7033 bool IsInstanceMethod,
Douglas Gregor45879692010-07-08 23:37:41 +00007034 bool AtParameterName,
John McCallba7bf592010-08-24 05:47:05 +00007035 ParsedType ReturnTy,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007036 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor95887f92010-07-08 23:20:03 +00007037 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00007038 // pool from the AST file.
Douglas Gregor95887f92010-07-08 23:20:03 +00007039 if (ExternalSource) {
7040 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
7041 I != N; ++I) {
7042 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00007043 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor95887f92010-07-08 23:20:03 +00007044 continue;
Sebastian Redl75d8a322010-08-02 23:18:59 +00007045
7046 ReadMethodPool(Sel);
Douglas Gregor95887f92010-07-08 23:20:03 +00007047 }
7048 }
7049
7050 // Build the set of methods we can see.
John McCall276321a2010-08-25 06:19:51 +00007051 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007052 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007053 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007054 CodeCompletionContext::CCC_Other);
Douglas Gregor95887f92010-07-08 23:20:03 +00007055
7056 if (ReturnTy)
7057 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redl75d8a322010-08-02 23:18:59 +00007058
Douglas Gregor95887f92010-07-08 23:20:03 +00007059 Results.EnterNewScope();
Sebastian Redl75d8a322010-08-02 23:18:59 +00007060 for (GlobalMethodPool::iterator M = MethodPool.begin(),
7061 MEnd = MethodPool.end();
7062 M != MEnd; ++M) {
7063 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
7064 &M->second.second;
7065 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00007066 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007067 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor95887f92010-07-08 23:20:03 +00007068 continue;
7069
Douglas Gregor45879692010-07-08 23:37:41 +00007070 if (AtParameterName) {
7071 // Suggest parameter names we've seen before.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007072 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor45879692010-07-08 23:37:41 +00007073 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
7074 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
7075 if (Param->getIdentifier()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007076 CodeCompletionBuilder Builder(Results.getAllocator(),
7077 Results.getCodeCompletionTUInfo());
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007078 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007079 Param->getIdentifier()->getName()));
7080 Results.AddResult(Builder.TakeString());
Douglas Gregor45879692010-07-08 23:37:41 +00007081 }
7082 }
7083
7084 continue;
7085 }
7086
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00007087 Result R(MethList->Method, Results.getBasePriority(MethList->Method), 0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007088 R.StartParameter = SelIdents.size();
Douglas Gregor95887f92010-07-08 23:20:03 +00007089 R.AllParametersAreInformative = false;
7090 R.DeclaringEntity = true;
7091 Results.MaybeAddResult(R, CurContext);
7092 }
7093 }
7094
7095 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007096 HandleCodeCompleteResults(this, CodeCompleter,
7097 CodeCompletionContext::CCC_Other,
7098 Results.data(),Results.size());
Douglas Gregor95887f92010-07-08 23:20:03 +00007099}
Douglas Gregorb14904c2010-08-13 22:48:40 +00007100
Douglas Gregorec00a262010-08-24 22:20:20 +00007101void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007102 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007103 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007104 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007105 Results.EnterNewScope();
7106
7107 // #if <condition>
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007108 CodeCompletionBuilder Builder(Results.getAllocator(),
7109 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007110 Builder.AddTypedTextChunk("if");
7111 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7112 Builder.AddPlaceholderChunk("condition");
7113 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007114
7115 // #ifdef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007116 Builder.AddTypedTextChunk("ifdef");
7117 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7118 Builder.AddPlaceholderChunk("macro");
7119 Results.AddResult(Builder.TakeString());
7120
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007121 // #ifndef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007122 Builder.AddTypedTextChunk("ifndef");
7123 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7124 Builder.AddPlaceholderChunk("macro");
7125 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007126
7127 if (InConditional) {
7128 // #elif <condition>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007129 Builder.AddTypedTextChunk("elif");
7130 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7131 Builder.AddPlaceholderChunk("condition");
7132 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007133
7134 // #else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007135 Builder.AddTypedTextChunk("else");
7136 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007137
7138 // #endif
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007139 Builder.AddTypedTextChunk("endif");
7140 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007141 }
7142
7143 // #include "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007144 Builder.AddTypedTextChunk("include");
7145 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7146 Builder.AddTextChunk("\"");
7147 Builder.AddPlaceholderChunk("header");
7148 Builder.AddTextChunk("\"");
7149 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007150
7151 // #include <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007152 Builder.AddTypedTextChunk("include");
7153 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7154 Builder.AddTextChunk("<");
7155 Builder.AddPlaceholderChunk("header");
7156 Builder.AddTextChunk(">");
7157 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007158
7159 // #define <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007160 Builder.AddTypedTextChunk("define");
7161 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7162 Builder.AddPlaceholderChunk("macro");
7163 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007164
7165 // #define <macro>(<args>)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007166 Builder.AddTypedTextChunk("define");
7167 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7168 Builder.AddPlaceholderChunk("macro");
7169 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7170 Builder.AddPlaceholderChunk("args");
7171 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7172 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007173
7174 // #undef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007175 Builder.AddTypedTextChunk("undef");
7176 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7177 Builder.AddPlaceholderChunk("macro");
7178 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007179
7180 // #line <number>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007181 Builder.AddTypedTextChunk("line");
7182 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7183 Builder.AddPlaceholderChunk("number");
7184 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007185
7186 // #line <number> "filename"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007187 Builder.AddTypedTextChunk("line");
7188 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7189 Builder.AddPlaceholderChunk("number");
7190 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7191 Builder.AddTextChunk("\"");
7192 Builder.AddPlaceholderChunk("filename");
7193 Builder.AddTextChunk("\"");
7194 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007195
7196 // #error <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007197 Builder.AddTypedTextChunk("error");
7198 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7199 Builder.AddPlaceholderChunk("message");
7200 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007201
7202 // #pragma <arguments>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007203 Builder.AddTypedTextChunk("pragma");
7204 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7205 Builder.AddPlaceholderChunk("arguments");
7206 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007207
David Blaikiebbafb8a2012-03-11 07:00:24 +00007208 if (getLangOpts().ObjC1) {
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007209 // #import "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007210 Builder.AddTypedTextChunk("import");
7211 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7212 Builder.AddTextChunk("\"");
7213 Builder.AddPlaceholderChunk("header");
7214 Builder.AddTextChunk("\"");
7215 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007216
7217 // #import <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007218 Builder.AddTypedTextChunk("import");
7219 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7220 Builder.AddTextChunk("<");
7221 Builder.AddPlaceholderChunk("header");
7222 Builder.AddTextChunk(">");
7223 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007224 }
7225
7226 // #include_next "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007227 Builder.AddTypedTextChunk("include_next");
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 // #include_next <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007235 Builder.AddTypedTextChunk("include_next");
7236 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7237 Builder.AddTextChunk("<");
7238 Builder.AddPlaceholderChunk("header");
7239 Builder.AddTextChunk(">");
7240 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007241
7242 // #warning <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007243 Builder.AddTypedTextChunk("warning");
7244 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7245 Builder.AddPlaceholderChunk("message");
7246 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007247
7248 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7249 // completions for them. And __include_macros is a Clang-internal extension
7250 // that we don't want to encourage anyone to use.
7251
7252 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7253 Results.ExitScope();
7254
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007255 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0de55ce2010-08-25 18:41:16 +00007256 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007257 Results.data(), Results.size());
7258}
7259
7260void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorec00a262010-08-24 22:20:20 +00007261 CodeCompleteOrdinaryName(S,
John McCallfaf5fb42010-08-26 23:41:50 +00007262 S->getFnParent()? Sema::PCC_RecoveryInFunction
7263 : Sema::PCC_Namespace);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007264}
7265
Douglas Gregorec00a262010-08-24 22:20:20 +00007266void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007267 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007268 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007269 IsDefinition? CodeCompletionContext::CCC_MacroName
7270 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor12785102010-08-24 20:21:13 +00007271 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7272 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007273 CodeCompletionBuilder Builder(Results.getAllocator(),
7274 Results.getCodeCompletionTUInfo());
Douglas Gregor12785102010-08-24 20:21:13 +00007275 Results.EnterNewScope();
7276 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7277 MEnd = PP.macro_end();
7278 M != MEnd; ++M) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007279 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007280 M->first->getName()));
Argyrios Kyrtzidis5c8b1cd2012-09-27 00:24:09 +00007281 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7282 CCP_CodePattern,
7283 CXCursor_MacroDefinition));
Douglas Gregor12785102010-08-24 20:21:13 +00007284 }
7285 Results.ExitScope();
7286 } else if (IsDefinition) {
7287 // FIXME: Can we detect when the user just wrote an include guard above?
7288 }
7289
Douglas Gregor0ac41382010-09-23 23:01:17 +00007290 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor12785102010-08-24 20:21:13 +00007291 Results.data(), Results.size());
7292}
7293
Douglas Gregorec00a262010-08-24 22:20:20 +00007294void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007295 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007296 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007297 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorec00a262010-08-24 22:20:20 +00007298
7299 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007300 AddMacroResults(PP, Results, true);
Douglas Gregorec00a262010-08-24 22:20:20 +00007301
7302 // defined (<macro>)
7303 Results.EnterNewScope();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007304 CodeCompletionBuilder Builder(Results.getAllocator(),
7305 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007306 Builder.AddTypedTextChunk("defined");
7307 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7308 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7309 Builder.AddPlaceholderChunk("macro");
7310 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7311 Results.AddResult(Builder.TakeString());
Douglas Gregorec00a262010-08-24 22:20:20 +00007312 Results.ExitScope();
7313
7314 HandleCodeCompleteResults(this, CodeCompleter,
7315 CodeCompletionContext::CCC_PreprocessorExpression,
7316 Results.data(), Results.size());
7317}
7318
7319void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7320 IdentifierInfo *Macro,
7321 MacroInfo *MacroInfo,
7322 unsigned Argument) {
7323 // FIXME: In the future, we could provide "overload" results, much like we
7324 // do for function calls.
7325
Argyrios Kyrtzidis75f6cd22011-08-18 19:41:28 +00007326 // Now just ignore this. There will be another code-completion callback
7327 // for the expanded tokens.
Douglas Gregorec00a262010-08-24 22:20:20 +00007328}
7329
Douglas Gregor11583702010-08-25 17:04:25 +00007330void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor11583702010-08-25 17:04:25 +00007331 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorea736372010-08-25 17:10:00 +00007332 CodeCompletionContext::CCC_NaturalLanguage,
Douglas Gregor11583702010-08-25 17:04:25 +00007333 0, 0);
7334}
7335
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007336void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007337 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00007338 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007339 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7340 CodeCompletionContext::CCC_Recovery);
Douglas Gregor39982192010-08-15 06:18:01 +00007341 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7342 CodeCompletionDeclConsumer Consumer(Builder,
7343 Context.getTranslationUnitDecl());
7344 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7345 Consumer);
7346 }
Douglas Gregorb14904c2010-08-13 22:48:40 +00007347
7348 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007349 AddMacroResults(PP, Builder, true);
Douglas Gregorb14904c2010-08-13 22:48:40 +00007350
7351 Results.clear();
7352 Results.insert(Results.end(),
7353 Builder.data(), Builder.data() + Builder.size());
7354}