blob: 2cc7b85a7f58733789af0e94d4f3b992b1e10bfd [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()) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000467 const DeclContext *Parent = TargetParents.back();
Douglas Gregor2af2f672009-09-21 20:12:40 +0000468 TargetParents.pop_back();
469
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000470 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Parent)) {
Douglas Gregor68762e72010-08-23 21:17:50 +0000471 if (!Namespace->getIdentifier())
472 continue;
473
Douglas Gregor2af2f672009-09-21 20:12:40 +0000474 Result = NestedNameSpecifier::Create(Context, Result, Namespace);
Douglas Gregor68762e72010-08-23 21:17:50 +0000475 }
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000476 else if (const TagDecl *TD = dyn_cast<TagDecl>(Parent))
Douglas Gregor2af2f672009-09-21 20:12:40 +0000477 Result = NestedNameSpecifier::Create(Context, Result,
478 false,
479 Context.getTypeDeclType(TD).getTypePtr());
Douglas Gregor9eb77012009-11-07 00:00:49 +0000480 }
Douglas Gregor2af2f672009-09-21 20:12:40 +0000481 return Result;
482}
483
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000484bool ResultBuilder::isInterestingDecl(const NamedDecl *ND,
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000485 bool &AsNestedNameSpecifier) const {
486 AsNestedNameSpecifier = false;
487
Douglas Gregor7c208612010-01-14 00:20:49 +0000488 ND = ND->getUnderlyingDecl();
489 unsigned IDNS = ND->getIdentifierNamespace();
Douglas Gregor58acf322009-10-09 22:16:47 +0000490
491 // Skip unnamed entities.
Douglas Gregor7c208612010-01-14 00:20:49 +0000492 if (!ND->getDeclName())
493 return false;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000494
495 // Friend declarations and declarations introduced due to friends are never
496 // added as results.
John McCallbbbbe4e2010-03-11 07:50:04 +0000497 if (IDNS & (Decl::IDNS_OrdinaryFriend | Decl::IDNS_TagFriend))
Douglas Gregor7c208612010-01-14 00:20:49 +0000498 return false;
499
Douglas Gregor99fe2ad2009-12-11 17:31:05 +0000500 // Class template (partial) specializations are never added as results.
Douglas Gregor7c208612010-01-14 00:20:49 +0000501 if (isa<ClassTemplateSpecializationDecl>(ND) ||
502 isa<ClassTemplatePartialSpecializationDecl>(ND))
503 return false;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000504
Douglas Gregor99fe2ad2009-12-11 17:31:05 +0000505 // Using declarations themselves are never added as results.
Douglas Gregor7c208612010-01-14 00:20:49 +0000506 if (isa<UsingDecl>(ND))
507 return false;
508
509 // Some declarations have reserved names that we don't want to ever show.
510 if (const IdentifierInfo *Id = ND->getIdentifier()) {
Douglas Gregor3545ff42009-09-21 16:56:56 +0000511 // __va_list_tag is a freak of nature. Find it and skip it.
512 if (Id->isStr("__va_list_tag") || Id->isStr("__builtin_va_list"))
Douglas Gregor7c208612010-01-14 00:20:49 +0000513 return false;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000514
Douglas Gregor58acf322009-10-09 22:16:47 +0000515 // Filter out names reserved for the implementation (C99 7.1.3,
Douglas Gregor9f1570d2010-07-14 17:44:04 +0000516 // C++ [lib.global.names]) if they come from a system header.
Daniel Dunbar2c422dc92009-10-18 20:26:12 +0000517 //
518 // FIXME: Add predicate for this.
Douglas Gregor58acf322009-10-09 22:16:47 +0000519 if (Id->getLength() >= 2) {
Daniel Dunbar2c422dc92009-10-18 20:26:12 +0000520 const char *Name = Id->getNameStart();
Douglas Gregor58acf322009-10-09 22:16:47 +0000521 if (Name[0] == '_' &&
Douglas Gregor9f1570d2010-07-14 17:44:04 +0000522 (Name[1] == '_' || (Name[1] >= 'A' && Name[1] <= 'Z')) &&
523 (ND->getLocation().isInvalid() ||
524 SemaRef.SourceMgr.isInSystemHeader(
525 SemaRef.SourceMgr.getSpellingLoc(ND->getLocation()))))
Douglas Gregor7c208612010-01-14 00:20:49 +0000526 return false;
Douglas Gregor58acf322009-10-09 22:16:47 +0000527 }
Douglas Gregor3545ff42009-09-21 16:56:56 +0000528 }
Douglas Gregor2927c0c2010-11-09 03:59:40 +0000529
Douglas Gregor59cab552010-08-16 23:05:20 +0000530 if (Filter == &ResultBuilder::IsNestedNameSpecifier ||
531 ((isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND)) &&
532 Filter != &ResultBuilder::IsNamespace &&
Douglas Gregor0ac41382010-09-23 23:01:17 +0000533 Filter != &ResultBuilder::IsNamespaceOrAlias &&
534 Filter != 0))
Douglas Gregor59cab552010-08-16 23:05:20 +0000535 AsNestedNameSpecifier = true;
536
Douglas Gregor3545ff42009-09-21 16:56:56 +0000537 // Filter out any unwanted results.
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000538 if (Filter && !(this->*Filter)(ND)) {
539 // Check whether it is interesting as a nested-name-specifier.
David Blaikiebbafb8a2012-03-11 07:00:24 +0000540 if (AllowNestedNameSpecifiers && SemaRef.getLangOpts().CPlusPlus &&
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000541 IsNestedNameSpecifier(ND) &&
542 (Filter != &ResultBuilder::IsMember ||
543 (isa<CXXRecordDecl>(ND) &&
544 cast<CXXRecordDecl>(ND)->isInjectedClassName()))) {
545 AsNestedNameSpecifier = true;
546 return true;
547 }
548
Douglas Gregor7c208612010-01-14 00:20:49 +0000549 return false;
Douglas Gregor59cab552010-08-16 23:05:20 +0000550 }
Douglas Gregor7c208612010-01-14 00:20:49 +0000551 // ... then it must be interesting!
552 return true;
553}
554
Douglas Gregore0717ab2010-01-14 00:41:07 +0000555bool ResultBuilder::CheckHiddenResult(Result &R, DeclContext *CurContext,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000556 const NamedDecl *Hiding) {
Douglas Gregore0717ab2010-01-14 00:41:07 +0000557 // In C, there is no way to refer to a hidden name.
558 // FIXME: This isn't true; we can find a tag name hidden by an ordinary
559 // name if we introduce the tag type.
David Blaikiebbafb8a2012-03-11 07:00:24 +0000560 if (!SemaRef.getLangOpts().CPlusPlus)
Douglas Gregore0717ab2010-01-14 00:41:07 +0000561 return true;
562
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000563 const DeclContext *HiddenCtx =
564 R.Declaration->getDeclContext()->getRedeclContext();
Douglas Gregore0717ab2010-01-14 00:41:07 +0000565
566 // There is no way to qualify a name declared in a function or method.
567 if (HiddenCtx->isFunctionOrMethod())
568 return true;
569
Sebastian Redl50c68252010-08-31 00:36:30 +0000570 if (HiddenCtx == Hiding->getDeclContext()->getRedeclContext())
Douglas Gregore0717ab2010-01-14 00:41:07 +0000571 return true;
572
573 // We can refer to the result with the appropriate qualification. Do it.
574 R.Hidden = true;
575 R.QualifierIsInformative = false;
576
577 if (!R.Qualifier)
578 R.Qualifier = getRequiredQualification(SemaRef.Context,
579 CurContext,
580 R.Declaration->getDeclContext());
581 return false;
582}
583
Douglas Gregor95887f92010-07-08 23:20:03 +0000584/// \brief A simplified classification of types used to determine whether two
585/// types are "similar enough" when adjusting priorities.
Douglas Gregor6e240332010-08-16 16:18:59 +0000586SimplifiedTypeClass clang::getSimplifiedTypeClass(CanQualType T) {
Douglas Gregor95887f92010-07-08 23:20:03 +0000587 switch (T->getTypeClass()) {
588 case Type::Builtin:
589 switch (cast<BuiltinType>(T)->getKind()) {
590 case BuiltinType::Void:
591 return STC_Void;
592
593 case BuiltinType::NullPtr:
594 return STC_Pointer;
595
596 case BuiltinType::Overload:
597 case BuiltinType::Dependent:
Douglas Gregor95887f92010-07-08 23:20:03 +0000598 return STC_Other;
599
600 case BuiltinType::ObjCId:
601 case BuiltinType::ObjCClass:
602 case BuiltinType::ObjCSel:
603 return STC_ObjectiveC;
604
605 default:
606 return STC_Arithmetic;
607 }
David Blaikie8a40f702012-01-17 06:56:22 +0000608
Douglas Gregor95887f92010-07-08 23:20:03 +0000609 case Type::Complex:
610 return STC_Arithmetic;
611
612 case Type::Pointer:
613 return STC_Pointer;
614
615 case Type::BlockPointer:
616 return STC_Block;
617
618 case Type::LValueReference:
619 case Type::RValueReference:
620 return getSimplifiedTypeClass(T->getAs<ReferenceType>()->getPointeeType());
621
622 case Type::ConstantArray:
623 case Type::IncompleteArray:
624 case Type::VariableArray:
625 case Type::DependentSizedArray:
626 return STC_Array;
627
628 case Type::DependentSizedExtVector:
629 case Type::Vector:
630 case Type::ExtVector:
631 return STC_Arithmetic;
632
633 case Type::FunctionProto:
634 case Type::FunctionNoProto:
635 return STC_Function;
636
637 case Type::Record:
638 return STC_Record;
639
640 case Type::Enum:
641 return STC_Arithmetic;
642
643 case Type::ObjCObject:
644 case Type::ObjCInterface:
645 case Type::ObjCObjectPointer:
646 return STC_ObjectiveC;
647
648 default:
649 return STC_Other;
650 }
651}
652
653/// \brief Get the type that a given expression will have if this declaration
654/// is used as an expression in its "typical" code-completion form.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000655QualType clang::getDeclUsageType(ASTContext &C, const NamedDecl *ND) {
Douglas Gregor95887f92010-07-08 23:20:03 +0000656 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
657
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000658 if (const TypeDecl *Type = dyn_cast<TypeDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000659 return C.getTypeDeclType(Type);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000660 if (const ObjCInterfaceDecl *Iface = dyn_cast<ObjCInterfaceDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000661 return C.getObjCInterfaceType(Iface);
662
663 QualType T;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000664 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND))
Douglas Gregor603d81b2010-07-13 08:18:22 +0000665 T = Function->getCallResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000666 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregor603d81b2010-07-13 08:18:22 +0000667 T = Method->getSendResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000668 else if (const FunctionTemplateDecl *FunTmpl =
669 dyn_cast<FunctionTemplateDecl>(ND))
Douglas Gregor603d81b2010-07-13 08:18:22 +0000670 T = FunTmpl->getTemplatedDecl()->getCallResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000671 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000672 T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext()));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000673 else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000674 T = Property->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000675 else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000676 T = Value->getType();
677 else
678 return QualType();
Douglas Gregoraf670a82011-04-14 20:33:34 +0000679
680 // Dig through references, function pointers, and block pointers to
681 // get down to the likely type of an expression when the entity is
682 // used.
683 do {
684 if (const ReferenceType *Ref = T->getAs<ReferenceType>()) {
685 T = Ref->getPointeeType();
686 continue;
687 }
688
689 if (const PointerType *Pointer = T->getAs<PointerType>()) {
690 if (Pointer->getPointeeType()->isFunctionType()) {
691 T = Pointer->getPointeeType();
692 continue;
693 }
694
695 break;
696 }
697
698 if (const BlockPointerType *Block = T->getAs<BlockPointerType>()) {
699 T = Block->getPointeeType();
700 continue;
701 }
702
703 if (const FunctionType *Function = T->getAs<FunctionType>()) {
704 T = Function->getResultType();
705 continue;
706 }
707
708 break;
709 } while (true);
710
711 return T;
Douglas Gregor95887f92010-07-08 23:20:03 +0000712}
713
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000714unsigned ResultBuilder::getBasePriority(const NamedDecl *ND) {
715 if (!ND)
716 return CCP_Unlikely;
717
718 // Context-based decisions.
719 const DeclContext *DC = ND->getDeclContext()->getRedeclContext();
720 if (DC->isFunctionOrMethod() || isa<BlockDecl>(DC)) {
721 // _cmd is relatively rare
722 if (const ImplicitParamDecl *ImplicitParam =
723 dyn_cast<ImplicitParamDecl>(ND))
724 if (ImplicitParam->getIdentifier() &&
725 ImplicitParam->getIdentifier()->isStr("_cmd"))
726 return CCP_ObjC_cmd;
727
728 return CCP_LocalDeclaration;
729 }
730 if (DC->isRecord() || isa<ObjCContainerDecl>(DC))
731 return CCP_MemberDeclaration;
732
733 // Content-based decisions.
734 if (isa<EnumConstantDecl>(ND))
735 return CCP_Constant;
736
Douglas Gregor52e0de42013-01-31 05:03:46 +0000737 // Use CCP_Type for type declarations unless we're in a statement, Objective-C
738 // message receiver, or parenthesized expression context. There, it's as
739 // likely that the user will want to write a type as other declarations.
740 if ((isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND)) &&
741 !(CompletionContext.getKind() == CodeCompletionContext::CCC_Statement ||
742 CompletionContext.getKind()
743 == CodeCompletionContext::CCC_ObjCMessageReceiver ||
744 CompletionContext.getKind()
745 == CodeCompletionContext::CCC_ParenthesizedExpression))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000746 return CCP_Type;
747
748 return CCP_Declaration;
749}
750
Douglas Gregor50832e02010-09-20 22:39:41 +0000751void ResultBuilder::AdjustResultPriorityForDecl(Result &R) {
752 // If this is an Objective-C method declaration whose selector matches our
753 // preferred selector, give it a priority boost.
754 if (!PreferredSelector.isNull())
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000755 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(R.Declaration))
Douglas Gregor50832e02010-09-20 22:39:41 +0000756 if (PreferredSelector == Method->getSelector())
757 R.Priority += CCD_SelectorMatch;
Douglas Gregor5fb901d2010-09-20 23:11:55 +0000758
Douglas Gregor50832e02010-09-20 22:39:41 +0000759 // If we have a preferred type, adjust the priority for results with exactly-
760 // matching or nearly-matching types.
761 if (!PreferredType.isNull()) {
762 QualType T = getDeclUsageType(SemaRef.Context, R.Declaration);
763 if (!T.isNull()) {
764 CanQualType TC = SemaRef.Context.getCanonicalType(T);
765 // Check for exactly-matching types (modulo qualifiers).
766 if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC))
767 R.Priority /= CCF_ExactTypeMatch;
768 // Check for nearly-matching types, based on classification of each.
769 else if ((getSimplifiedTypeClass(PreferredType)
Douglas Gregor95887f92010-07-08 23:20:03 +0000770 == getSimplifiedTypeClass(TC)) &&
Douglas Gregor50832e02010-09-20 22:39:41 +0000771 !(PreferredType->isEnumeralType() && TC->isEnumeralType()))
772 R.Priority /= CCF_SimilarTypeMatch;
773 }
774 }
Douglas Gregor95887f92010-07-08 23:20:03 +0000775}
776
Douglas Gregor0212fd72010-09-21 16:06:22 +0000777void ResultBuilder::MaybeAddConstructorResults(Result R) {
David Blaikiebbafb8a2012-03-11 07:00:24 +0000778 if (!SemaRef.getLangOpts().CPlusPlus || !R.Declaration ||
Douglas Gregor0212fd72010-09-21 16:06:22 +0000779 !CompletionContext.wantConstructorResults())
780 return;
781
782 ASTContext &Context = SemaRef.Context;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000783 const NamedDecl *D = R.Declaration;
784 const CXXRecordDecl *Record = 0;
785 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D))
Douglas Gregor0212fd72010-09-21 16:06:22 +0000786 Record = ClassTemplate->getTemplatedDecl();
787 else if ((Record = dyn_cast<CXXRecordDecl>(D))) {
788 // Skip specializations and partial specializations.
789 if (isa<ClassTemplateSpecializationDecl>(Record))
790 return;
791 } else {
792 // There are no constructors here.
793 return;
794 }
795
796 Record = Record->getDefinition();
797 if (!Record)
798 return;
799
800
801 QualType RecordTy = Context.getTypeDeclType(Record);
802 DeclarationName ConstructorName
803 = Context.DeclarationNames.getCXXConstructorName(
804 Context.getCanonicalType(RecordTy));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000805 DeclContext::lookup_const_result Ctors = Record->lookup(ConstructorName);
806 for (DeclContext::lookup_const_iterator I = Ctors.begin(),
807 E = Ctors.end();
808 I != E; ++I) {
David Blaikieff7d47a2012-12-19 00:45:41 +0000809 R.Declaration = *I;
Douglas Gregor0212fd72010-09-21 16:06:22 +0000810 R.CursorKind = getCursorKindForDecl(R.Declaration);
811 Results.push_back(R);
812 }
813}
814
Douglas Gregor7c208612010-01-14 00:20:49 +0000815void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) {
816 assert(!ShadowMaps.empty() && "Must enter into a results scope");
817
818 if (R.Kind != Result::RK_Declaration) {
819 // For non-declaration results, just add the result.
820 Results.push_back(R);
821 return;
822 }
823
824 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000825 if (const UsingShadowDecl *Using =
826 dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000827 MaybeAddResult(Result(Using->getTargetDecl(),
828 getBasePriority(Using->getTargetDecl()),
829 R.Qualifier),
830 CurContext);
Douglas Gregor7c208612010-01-14 00:20:49 +0000831 return;
832 }
833
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000834 const Decl *CanonDecl = R.Declaration->getCanonicalDecl();
Douglas Gregor7c208612010-01-14 00:20:49 +0000835 unsigned IDNS = CanonDecl->getIdentifierNamespace();
836
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000837 bool AsNestedNameSpecifier = false;
838 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregor7c208612010-01-14 00:20:49 +0000839 return;
840
Douglas Gregor0212fd72010-09-21 16:06:22 +0000841 // C++ constructors are never found by name lookup.
842 if (isa<CXXConstructorDecl>(R.Declaration))
843 return;
844
Douglas Gregor3545ff42009-09-21 16:56:56 +0000845 ShadowMap &SMap = ShadowMaps.back();
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000846 ShadowMapEntry::iterator I, IEnd;
847 ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName());
848 if (NamePos != SMap.end()) {
849 I = NamePos->second.begin();
850 IEnd = NamePos->second.end();
851 }
852
853 for (; I != IEnd; ++I) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000854 const NamedDecl *ND = I->first;
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000855 unsigned Index = I->second;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000856 if (ND->getCanonicalDecl() == CanonDecl) {
857 // This is a redeclaration. Always pick the newer declaration.
Douglas Gregor3545ff42009-09-21 16:56:56 +0000858 Results[Index].Declaration = R.Declaration;
859
Douglas Gregor3545ff42009-09-21 16:56:56 +0000860 // We're done.
861 return;
862 }
863 }
864
865 // This is a new declaration in this scope. However, check whether this
866 // declaration name is hidden by a similarly-named declaration in an outer
867 // scope.
868 std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end();
869 --SMEnd;
870 for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000871 ShadowMapEntry::iterator I, IEnd;
872 ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName());
873 if (NamePos != SM->end()) {
874 I = NamePos->second.begin();
875 IEnd = NamePos->second.end();
876 }
877 for (; I != IEnd; ++I) {
Douglas Gregor3545ff42009-09-21 16:56:56 +0000878 // A tag declaration does not hide a non-tag declaration.
John McCalle87beb22010-04-23 18:46:30 +0000879 if (I->first->hasTagIdentifierNamespace() &&
Douglas Gregor3545ff42009-09-21 16:56:56 +0000880 (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary |
881 Decl::IDNS_ObjCProtocol)))
882 continue;
883
884 // Protocols are in distinct namespaces from everything else.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000885 if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000886 || (IDNS & Decl::IDNS_ObjCProtocol)) &&
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000887 I->first->getIdentifierNamespace() != IDNS)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000888 continue;
889
890 // The newly-added result is hidden by an entry in the shadow map.
Douglas Gregore0717ab2010-01-14 00:41:07 +0000891 if (CheckHiddenResult(R, CurContext, I->first))
Douglas Gregor3545ff42009-09-21 16:56:56 +0000892 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000893
894 break;
895 }
896 }
897
898 // Make sure that any given declaration only shows up in the result set once.
899 if (!AllDeclsFound.insert(CanonDecl))
900 return;
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000901
Douglas Gregore412a5a2009-09-23 22:26:46 +0000902 // If the filter is for nested-name-specifiers, then this result starts a
903 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000904 if (AsNestedNameSpecifier) {
Douglas Gregore412a5a2009-09-23 22:26:46 +0000905 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000906 R.Priority = CCP_NestedNameSpecifier;
Douglas Gregor50832e02010-09-20 22:39:41 +0000907 } else
908 AdjustResultPriorityForDecl(R);
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000909
Douglas Gregor5bf52692009-09-22 23:15:58 +0000910 // If this result is supposed to have an informative qualifier, add one.
Douglas Gregore412a5a2009-09-23 22:26:46 +0000911 if (R.QualifierIsInformative && !R.Qualifier &&
912 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000913 const DeclContext *Ctx = R.Declaration->getDeclContext();
914 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregor5bf52692009-09-22 23:15:58 +0000915 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000916 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregor5bf52692009-09-22 23:15:58 +0000917 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
918 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
919 else
920 R.QualifierIsInformative = false;
921 }
Douglas Gregore412a5a2009-09-23 22:26:46 +0000922
Douglas Gregor3545ff42009-09-21 16:56:56 +0000923 // Insert this result into the set of results and into the current shadow
924 // map.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000925 SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size());
Douglas Gregor3545ff42009-09-21 16:56:56 +0000926 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +0000927
928 if (!AsNestedNameSpecifier)
929 MaybeAddConstructorResults(R);
Douglas Gregor3545ff42009-09-21 16:56:56 +0000930}
931
Douglas Gregorc580c522010-01-14 01:09:38 +0000932void ResultBuilder::AddResult(Result R, DeclContext *CurContext,
Douglas Gregor09bbc652010-01-14 15:47:35 +0000933 NamedDecl *Hiding, bool InBaseClass = false) {
Douglas Gregor78a21012010-01-14 16:01:26 +0000934 if (R.Kind != Result::RK_Declaration) {
935 // For non-declaration results, just add the result.
936 Results.push_back(R);
937 return;
938 }
939
Douglas Gregorc580c522010-01-14 01:09:38 +0000940 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000941 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000942 AddResult(Result(Using->getTargetDecl(),
943 getBasePriority(Using->getTargetDecl()),
944 R.Qualifier),
945 CurContext, Hiding);
Douglas Gregorc580c522010-01-14 01:09:38 +0000946 return;
947 }
948
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000949 bool AsNestedNameSpecifier = false;
950 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregorc580c522010-01-14 01:09:38 +0000951 return;
952
Douglas Gregor0212fd72010-09-21 16:06:22 +0000953 // C++ constructors are never found by name lookup.
954 if (isa<CXXConstructorDecl>(R.Declaration))
955 return;
956
Douglas Gregorc580c522010-01-14 01:09:38 +0000957 if (Hiding && CheckHiddenResult(R, CurContext, Hiding))
958 return;
Nick Lewyckyc3921482012-04-03 21:44:08 +0000959
Douglas Gregorc580c522010-01-14 01:09:38 +0000960 // Make sure that any given declaration only shows up in the result set once.
961 if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl()))
962 return;
963
964 // If the filter is for nested-name-specifiers, then this result starts a
965 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000966 if (AsNestedNameSpecifier) {
Douglas Gregorc580c522010-01-14 01:09:38 +0000967 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000968 R.Priority = CCP_NestedNameSpecifier;
969 }
Douglas Gregor09bbc652010-01-14 15:47:35 +0000970 else if (Filter == &ResultBuilder::IsMember && !R.Qualifier && InBaseClass &&
971 isa<CXXRecordDecl>(R.Declaration->getDeclContext()
Sebastian Redl50c68252010-08-31 00:36:30 +0000972 ->getRedeclContext()))
Douglas Gregor09bbc652010-01-14 15:47:35 +0000973 R.QualifierIsInformative = true;
974
Douglas Gregorc580c522010-01-14 01:09:38 +0000975 // If this result is supposed to have an informative qualifier, add one.
976 if (R.QualifierIsInformative && !R.Qualifier &&
977 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000978 const DeclContext *Ctx = R.Declaration->getDeclContext();
979 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregorc580c522010-01-14 01:09:38 +0000980 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000981 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregorc580c522010-01-14 01:09:38 +0000982 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000983 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregorc580c522010-01-14 01:09:38 +0000984 else
985 R.QualifierIsInformative = false;
986 }
987
Douglas Gregora2db7932010-05-26 22:00:08 +0000988 // Adjust the priority if this result comes from a base class.
989 if (InBaseClass)
990 R.Priority += CCD_InBaseClass;
991
Douglas Gregor50832e02010-09-20 22:39:41 +0000992 AdjustResultPriorityForDecl(R);
Douglas Gregor7aa6b222010-05-30 01:49:25 +0000993
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000994 if (HasObjectTypeQualifiers)
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000995 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(R.Declaration))
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000996 if (Method->isInstance()) {
997 Qualifiers MethodQuals
998 = Qualifiers::fromCVRMask(Method->getTypeQualifiers());
999 if (ObjectTypeQualifiers == MethodQuals)
1000 R.Priority += CCD_ObjectQualifierMatch;
1001 else if (ObjectTypeQualifiers - MethodQuals) {
1002 // The method cannot be invoked, because doing so would drop
1003 // qualifiers.
1004 return;
1005 }
1006 }
1007
Douglas Gregorc580c522010-01-14 01:09:38 +00001008 // Insert this result into the set of results.
1009 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +00001010
1011 if (!AsNestedNameSpecifier)
1012 MaybeAddConstructorResults(R);
Douglas Gregorc580c522010-01-14 01:09:38 +00001013}
1014
Douglas Gregor78a21012010-01-14 16:01:26 +00001015void ResultBuilder::AddResult(Result R) {
1016 assert(R.Kind != Result::RK_Declaration &&
1017 "Declaration results need more context");
1018 Results.push_back(R);
1019}
1020
Douglas Gregor3545ff42009-09-21 16:56:56 +00001021/// \brief Enter into a new scope.
1022void ResultBuilder::EnterNewScope() {
1023 ShadowMaps.push_back(ShadowMap());
1024}
1025
1026/// \brief Exit from the current scope.
1027void ResultBuilder::ExitScope() {
Douglas Gregor05e7ca32009-12-06 20:23:50 +00001028 for (ShadowMap::iterator E = ShadowMaps.back().begin(),
1029 EEnd = ShadowMaps.back().end();
1030 E != EEnd;
1031 ++E)
1032 E->second.Destroy();
1033
Douglas Gregor3545ff42009-09-21 16:56:56 +00001034 ShadowMaps.pop_back();
1035}
1036
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001037/// \brief Determines whether this given declaration will be found by
1038/// ordinary name lookup.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001039bool ResultBuilder::IsOrdinaryName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001040 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1041
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001042 unsigned IDNS = Decl::IDNS_Ordinary;
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
1060 unsigned IDNS = Decl::IDNS_Ordinary;
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
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001087 unsigned IDNS = Decl::IDNS_Ordinary;
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())
1479 if (!Tag->getIdentifier() && !Tag->getTypedefNameForAnonDecl()) {
1480 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))
1784 isVoid = Function->getResultType()->isVoidType();
1785 else if (ObjCMethodDecl *Method
1786 = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext))
1787 isVoid = Method->getResultType()->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).
Douglas Gregor0212fd72010-09-21 16:06:22 +00002066 QualType T;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002067 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002068 T = Function->getResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002069 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002070 T = Method->getResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002071 else if (const FunctionTemplateDecl *FunTmpl =
2072 dyn_cast<FunctionTemplateDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002073 T = FunTmpl->getTemplatedDecl()->getResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002074 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002075 T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext()));
2076 else if (isa<UnresolvedUsingValueDecl>(ND)) {
2077 /* Do nothing: ignore unresolved using declarations*/
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002078 } else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND)) {
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002079 T = Value->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002080 } else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002081 T = Property->getType();
2082
2083 if (T.isNull() || Context.hasSameType(T, Context.DependentTy))
2084 return;
2085
Douglas Gregor75acd922011-09-27 23:30:47 +00002086 Result.AddResultTypeChunk(GetCompletionTypeString(T, Context, Policy,
Douglas Gregor304f9b02011-02-01 21:15:40 +00002087 Result.getAllocator()));
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002088}
2089
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002090static void MaybeAddSentinel(ASTContext &Context,
2091 const NamedDecl *FunctionOrMethod,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002092 CodeCompletionBuilder &Result) {
Douglas Gregordbb71db2010-08-23 23:51:41 +00002093 if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>())
2094 if (Sentinel->getSentinel() == 0) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00002095 if (Context.getLangOpts().ObjC1 &&
Douglas Gregordbb71db2010-08-23 23:51:41 +00002096 Context.Idents.get("nil").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002097 Result.AddTextChunk(", nil");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002098 else if (Context.Idents.get("NULL").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002099 Result.AddTextChunk(", NULL");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002100 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002101 Result.AddTextChunk(", (void*)0");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002102 }
2103}
2104
Douglas Gregor8f08d742011-07-30 07:55:26 +00002105static std::string formatObjCParamQualifiers(unsigned ObjCQuals) {
2106 std::string Result;
2107 if (ObjCQuals & Decl::OBJC_TQ_In)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002108 Result += "in ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002109 else if (ObjCQuals & Decl::OBJC_TQ_Inout)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002110 Result += "inout ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002111 else if (ObjCQuals & Decl::OBJC_TQ_Out)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002112 Result += "out ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002113 if (ObjCQuals & Decl::OBJC_TQ_Bycopy)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002114 Result += "bycopy ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002115 else if (ObjCQuals & Decl::OBJC_TQ_Byref)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002116 Result += "byref ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002117 if (ObjCQuals & Decl::OBJC_TQ_Oneway)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002118 Result += "oneway ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002119 return Result;
2120}
2121
Douglas Gregore90dd002010-08-24 16:15:59 +00002122static std::string FormatFunctionParameter(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002123 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002124 const ParmVarDecl *Param,
Douglas Gregord793e7c2011-10-18 04:23:19 +00002125 bool SuppressName = false,
2126 bool SuppressBlock = false) {
Douglas Gregore90dd002010-08-24 16:15:59 +00002127 bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext());
2128 if (Param->getType()->isDependentType() ||
2129 !Param->getType()->isBlockPointerType()) {
2130 // The argument for a dependent or non-block parameter is a placeholder
2131 // containing that parameter's type.
2132 std::string Result;
2133
Douglas Gregor981a0c42010-08-29 19:47:46 +00002134 if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002135 Result = Param->getIdentifier()->getName();
2136
John McCall31168b02011-06-15 23:02:42 +00002137 Param->getType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002138
2139 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002140 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2141 + Result + ")";
Douglas Gregor981a0c42010-08-29 19:47:46 +00002142 if (Param->getIdentifier() && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002143 Result += Param->getIdentifier()->getName();
2144 }
2145 return Result;
2146 }
2147
2148 // The argument for a block pointer parameter is a block literal with
2149 // the appropriate type.
David Blaikie6adc78e2013-02-18 22:06:02 +00002150 FunctionTypeLoc Block;
2151 FunctionProtoTypeLoc BlockProto;
Douglas Gregore90dd002010-08-24 16:15:59 +00002152 TypeLoc TL;
2153 if (TypeSourceInfo *TSInfo = Param->getTypeSourceInfo()) {
2154 TL = TSInfo->getTypeLoc().getUnqualifiedLoc();
2155 while (true) {
2156 // Look through typedefs.
Douglas Gregord793e7c2011-10-18 04:23:19 +00002157 if (!SuppressBlock) {
David Blaikie6adc78e2013-02-18 22:06:02 +00002158 if (TypedefTypeLoc TypedefTL = TL.getAs<TypedefTypeLoc>()) {
2159 if (TypeSourceInfo *InnerTSInfo =
2160 TypedefTL.getTypedefNameDecl()->getTypeSourceInfo()) {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002161 TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc();
2162 continue;
2163 }
2164 }
2165
2166 // Look through qualified types
David Blaikie6adc78e2013-02-18 22:06:02 +00002167 if (QualifiedTypeLoc QualifiedTL = TL.getAs<QualifiedTypeLoc>()) {
2168 TL = QualifiedTL.getUnqualifiedLoc();
Douglas Gregore90dd002010-08-24 16:15:59 +00002169 continue;
2170 }
2171 }
2172
Douglas Gregore90dd002010-08-24 16:15:59 +00002173 // Try to get the function prototype behind the block pointer type,
2174 // then we're done.
David Blaikie6adc78e2013-02-18 22:06:02 +00002175 if (BlockPointerTypeLoc BlockPtr = TL.getAs<BlockPointerTypeLoc>()) {
2176 TL = BlockPtr.getPointeeLoc().IgnoreParens();
2177 Block = TL.getAs<FunctionTypeLoc>();
2178 BlockProto = TL.getAs<FunctionProtoTypeLoc>();
Douglas Gregore90dd002010-08-24 16:15:59 +00002179 }
2180 break;
2181 }
2182 }
2183
2184 if (!Block) {
2185 // We were unable to find a FunctionProtoTypeLoc with parameter names
2186 // for the block; just use the parameter type as a placeholder.
2187 std::string Result;
Douglas Gregord793e7c2011-10-18 04:23:19 +00002188 if (!ObjCMethodParam && Param->getIdentifier())
2189 Result = Param->getIdentifier()->getName();
2190
John McCall31168b02011-06-15 23:02:42 +00002191 Param->getType().getUnqualifiedType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002192
2193 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002194 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2195 + Result + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002196 if (Param->getIdentifier())
2197 Result += Param->getIdentifier()->getName();
2198 }
2199
2200 return Result;
2201 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002202
Douglas Gregore90dd002010-08-24 16:15:59 +00002203 // We have the function prototype behind the block pointer type, as it was
2204 // written in the source.
Douglas Gregor67da50e2010-09-08 22:47:51 +00002205 std::string Result;
David Blaikie6adc78e2013-02-18 22:06:02 +00002206 QualType ResultType = Block.getTypePtr()->getResultType();
Douglas Gregord793e7c2011-10-18 04:23:19 +00002207 if (!ResultType->isVoidType() || SuppressBlock)
John McCall31168b02011-06-15 23:02:42 +00002208 ResultType.getAsStringInternal(Result, Policy);
Douglas Gregord793e7c2011-10-18 04:23:19 +00002209
2210 // Format the parameter list.
2211 std::string Params;
David Blaikie6adc78e2013-02-18 22:06:02 +00002212 if (!BlockProto || Block.getNumArgs() == 0) {
2213 if (BlockProto && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002214 Params = "(...)";
Douglas Gregoraf25cfa2010-10-02 23:49:58 +00002215 else
Douglas Gregord793e7c2011-10-18 04:23:19 +00002216 Params = "(void)";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002217 } else {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002218 Params += "(";
David Blaikie6adc78e2013-02-18 22:06:02 +00002219 for (unsigned I = 0, N = Block.getNumArgs(); I != N; ++I) {
Douglas Gregor67da50e2010-09-08 22:47:51 +00002220 if (I)
Douglas Gregord793e7c2011-10-18 04:23:19 +00002221 Params += ", ";
David Blaikie6adc78e2013-02-18 22:06:02 +00002222 Params += FormatFunctionParameter(Context, Policy, Block.getArg(I),
Douglas Gregord793e7c2011-10-18 04:23:19 +00002223 /*SuppressName=*/false,
2224 /*SuppressBlock=*/true);
Douglas Gregor67da50e2010-09-08 22:47:51 +00002225
David Blaikie6adc78e2013-02-18 22:06:02 +00002226 if (I == N - 1 && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002227 Params += ", ...";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002228 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002229 Params += ")";
Douglas Gregor400f5972010-08-31 05:13:43 +00002230 }
Douglas Gregor67da50e2010-09-08 22:47:51 +00002231
Douglas Gregord793e7c2011-10-18 04:23:19 +00002232 if (SuppressBlock) {
2233 // Format as a parameter.
2234 Result = Result + " (^";
2235 if (Param->getIdentifier())
2236 Result += Param->getIdentifier()->getName();
2237 Result += ")";
2238 Result += Params;
2239 } else {
2240 // Format as a block literal argument.
2241 Result = '^' + Result;
2242 Result += Params;
2243
2244 if (Param->getIdentifier())
2245 Result += Param->getIdentifier()->getName();
2246 }
2247
Douglas Gregore90dd002010-08-24 16:15:59 +00002248 return Result;
2249}
2250
Douglas Gregor3545ff42009-09-21 16:56:56 +00002251/// \brief Add function parameter chunks to the given code completion string.
2252static void AddFunctionParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002253 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002254 const FunctionDecl *Function,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002255 CodeCompletionBuilder &Result,
2256 unsigned Start = 0,
2257 bool InOptional = false) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002258 bool FirstParameter = true;
Douglas Gregor9eb77012009-11-07 00:00:49 +00002259
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002260 for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002261 const ParmVarDecl *Param = Function->getParamDecl(P);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002262
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002263 if (Param->hasDefaultArg() && !InOptional) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002264 // When we see an optional default argument, put that argument and
2265 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002266 CodeCompletionBuilder Opt(Result.getAllocator(),
2267 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002268 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002269 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002270 AddFunctionParameterChunks(Context, Policy, Function, Opt, P, true);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002271 Result.AddOptionalChunk(Opt.TakeString());
2272 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002273 }
2274
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002275 if (FirstParameter)
2276 FirstParameter = false;
2277 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002278 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002279
2280 InOptional = false;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002281
2282 // Format the placeholder string.
Douglas Gregor75acd922011-09-27 23:30:47 +00002283 std::string PlaceholderStr = FormatFunctionParameter(Context, Policy,
2284 Param);
Douglas Gregore90dd002010-08-24 16:15:59 +00002285
Douglas Gregor400f5972010-08-31 05:13:43 +00002286 if (Function->isVariadic() && P == N - 1)
2287 PlaceholderStr += ", ...";
2288
Douglas Gregor3545ff42009-09-21 16:56:56 +00002289 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002290 Result.AddPlaceholderChunk(
2291 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002292 }
Douglas Gregorba449032009-09-22 21:42:17 +00002293
2294 if (const FunctionProtoType *Proto
2295 = Function->getType()->getAs<FunctionProtoType>())
Douglas Gregordbb71db2010-08-23 23:51:41 +00002296 if (Proto->isVariadic()) {
Douglas Gregor400f5972010-08-31 05:13:43 +00002297 if (Proto->getNumArgs() == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002298 Result.AddPlaceholderChunk("...");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002299
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002300 MaybeAddSentinel(Context, Function, Result);
Douglas Gregordbb71db2010-08-23 23:51:41 +00002301 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002302}
2303
2304/// \brief Add template parameter chunks to the given code completion string.
2305static void AddTemplateParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002306 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002307 const TemplateDecl *Template,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002308 CodeCompletionBuilder &Result,
2309 unsigned MaxParameters = 0,
2310 unsigned Start = 0,
2311 bool InDefaultArg = false) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002312 bool FirstParameter = true;
2313
2314 TemplateParameterList *Params = Template->getTemplateParameters();
2315 TemplateParameterList::iterator PEnd = Params->end();
2316 if (MaxParameters)
2317 PEnd = Params->begin() + MaxParameters;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002318 for (TemplateParameterList::iterator P = Params->begin() + Start;
2319 P != PEnd; ++P) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002320 bool HasDefaultArg = false;
2321 std::string PlaceholderStr;
2322 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) {
2323 if (TTP->wasDeclaredWithTypename())
2324 PlaceholderStr = "typename";
2325 else
2326 PlaceholderStr = "class";
2327
2328 if (TTP->getIdentifier()) {
2329 PlaceholderStr += ' ';
2330 PlaceholderStr += TTP->getIdentifier()->getName();
2331 }
2332
2333 HasDefaultArg = TTP->hasDefaultArgument();
2334 } else if (NonTypeTemplateParmDecl *NTTP
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002335 = dyn_cast<NonTypeTemplateParmDecl>(*P)) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002336 if (NTTP->getIdentifier())
2337 PlaceholderStr = NTTP->getIdentifier()->getName();
John McCall31168b02011-06-15 23:02:42 +00002338 NTTP->getType().getAsStringInternal(PlaceholderStr, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002339 HasDefaultArg = NTTP->hasDefaultArgument();
2340 } else {
2341 assert(isa<TemplateTemplateParmDecl>(*P));
2342 TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P);
2343
2344 // Since putting the template argument list into the placeholder would
2345 // be very, very long, we just use an abbreviation.
2346 PlaceholderStr = "template<...> class";
2347 if (TTP->getIdentifier()) {
2348 PlaceholderStr += ' ';
2349 PlaceholderStr += TTP->getIdentifier()->getName();
2350 }
2351
2352 HasDefaultArg = TTP->hasDefaultArgument();
2353 }
2354
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002355 if (HasDefaultArg && !InDefaultArg) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002356 // When we see an optional default argument, put that argument and
2357 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002358 CodeCompletionBuilder Opt(Result.getAllocator(),
2359 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002360 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002361 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002362 AddTemplateParameterChunks(Context, Policy, Template, Opt, MaxParameters,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002363 P - Params->begin(), true);
2364 Result.AddOptionalChunk(Opt.TakeString());
2365 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002366 }
2367
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002368 InDefaultArg = false;
2369
Douglas Gregor3545ff42009-09-21 16:56:56 +00002370 if (FirstParameter)
2371 FirstParameter = false;
2372 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002373 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002374
2375 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002376 Result.AddPlaceholderChunk(
2377 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002378 }
2379}
2380
Douglas Gregorf2510672009-09-21 19:57:38 +00002381/// \brief Add a qualifier to the given code-completion string, if the
2382/// provided nested-name-specifier is non-NULL.
Douglas Gregor0f622362009-12-11 18:44:16 +00002383static void
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002384AddQualifierToCompletionString(CodeCompletionBuilder &Result,
Douglas Gregor0f622362009-12-11 18:44:16 +00002385 NestedNameSpecifier *Qualifier,
2386 bool QualifierIsInformative,
Douglas Gregor75acd922011-09-27 23:30:47 +00002387 ASTContext &Context,
2388 const PrintingPolicy &Policy) {
Douglas Gregorf2510672009-09-21 19:57:38 +00002389 if (!Qualifier)
2390 return;
2391
2392 std::string PrintedNNS;
2393 {
2394 llvm::raw_string_ostream OS(PrintedNNS);
Douglas Gregor75acd922011-09-27 23:30:47 +00002395 Qualifier->print(OS, Policy);
Douglas Gregorf2510672009-09-21 19:57:38 +00002396 }
Douglas Gregor5bf52692009-09-22 23:15:58 +00002397 if (QualifierIsInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002398 Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregor5bf52692009-09-22 23:15:58 +00002399 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002400 Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregorf2510672009-09-21 19:57:38 +00002401}
2402
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002403static void
2404AddFunctionTypeQualsToCompletionString(CodeCompletionBuilder &Result,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002405 const FunctionDecl *Function) {
Douglas Gregor0f622362009-12-11 18:44:16 +00002406 const FunctionProtoType *Proto
2407 = Function->getType()->getAs<FunctionProtoType>();
2408 if (!Proto || !Proto->getTypeQuals())
2409 return;
2410
Douglas Gregor304f9b02011-02-01 21:15:40 +00002411 // FIXME: Add ref-qualifier!
2412
2413 // Handle single qualifiers without copying
2414 if (Proto->getTypeQuals() == Qualifiers::Const) {
2415 Result.AddInformativeChunk(" const");
2416 return;
2417 }
2418
2419 if (Proto->getTypeQuals() == Qualifiers::Volatile) {
2420 Result.AddInformativeChunk(" volatile");
2421 return;
2422 }
2423
2424 if (Proto->getTypeQuals() == Qualifiers::Restrict) {
2425 Result.AddInformativeChunk(" restrict");
2426 return;
2427 }
2428
2429 // Handle multiple qualifiers.
Douglas Gregor0f622362009-12-11 18:44:16 +00002430 std::string QualsStr;
David Blaikief5697e52012-08-10 00:55:35 +00002431 if (Proto->isConst())
Douglas Gregor0f622362009-12-11 18:44:16 +00002432 QualsStr += " const";
David Blaikief5697e52012-08-10 00:55:35 +00002433 if (Proto->isVolatile())
Douglas Gregor0f622362009-12-11 18:44:16 +00002434 QualsStr += " volatile";
David Blaikief5697e52012-08-10 00:55:35 +00002435 if (Proto->isRestrict())
Douglas Gregor0f622362009-12-11 18:44:16 +00002436 QualsStr += " restrict";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002437 Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr));
Douglas Gregor0f622362009-12-11 18:44:16 +00002438}
2439
Douglas Gregor0212fd72010-09-21 16:06:22 +00002440/// \brief Add the name of the given declaration
Douglas Gregor75acd922011-09-27 23:30:47 +00002441static void AddTypedNameChunk(ASTContext &Context, const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002442 const NamedDecl *ND,
2443 CodeCompletionBuilder &Result) {
Douglas Gregor0212fd72010-09-21 16:06:22 +00002444 DeclarationName Name = ND->getDeclName();
2445 if (!Name)
2446 return;
2447
2448 switch (Name.getNameKind()) {
Douglas Gregor304f9b02011-02-01 21:15:40 +00002449 case DeclarationName::CXXOperatorName: {
2450 const char *OperatorName = 0;
2451 switch (Name.getCXXOverloadedOperator()) {
2452 case OO_None:
2453 case OO_Conditional:
2454 case NUM_OVERLOADED_OPERATORS:
2455 OperatorName = "operator";
2456 break;
2457
2458#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
2459 case OO_##Name: OperatorName = "operator" Spelling; break;
2460#define OVERLOADED_OPERATOR_MULTI(Name,Spelling,Unary,Binary,MemberOnly)
2461#include "clang/Basic/OperatorKinds.def"
2462
2463 case OO_New: OperatorName = "operator new"; break;
2464 case OO_Delete: OperatorName = "operator delete"; break;
2465 case OO_Array_New: OperatorName = "operator new[]"; break;
2466 case OO_Array_Delete: OperatorName = "operator delete[]"; break;
2467 case OO_Call: OperatorName = "operator()"; break;
2468 case OO_Subscript: OperatorName = "operator[]"; break;
2469 }
2470 Result.AddTypedTextChunk(OperatorName);
2471 break;
2472 }
2473
Douglas Gregor0212fd72010-09-21 16:06:22 +00002474 case DeclarationName::Identifier:
2475 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor0212fd72010-09-21 16:06:22 +00002476 case DeclarationName::CXXDestructorName:
2477 case DeclarationName::CXXLiteralOperatorName:
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002478 Result.AddTypedTextChunk(
2479 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002480 break;
2481
2482 case DeclarationName::CXXUsingDirective:
2483 case DeclarationName::ObjCZeroArgSelector:
2484 case DeclarationName::ObjCOneArgSelector:
2485 case DeclarationName::ObjCMultiArgSelector:
2486 break;
2487
2488 case DeclarationName::CXXConstructorName: {
2489 CXXRecordDecl *Record = 0;
2490 QualType Ty = Name.getCXXNameType();
2491 if (const RecordType *RecordTy = Ty->getAs<RecordType>())
2492 Record = cast<CXXRecordDecl>(RecordTy->getDecl());
2493 else if (const InjectedClassNameType *InjectedTy
2494 = Ty->getAs<InjectedClassNameType>())
2495 Record = InjectedTy->getDecl();
2496 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002497 Result.AddTypedTextChunk(
2498 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002499 break;
2500 }
2501
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002502 Result.AddTypedTextChunk(
2503 Result.getAllocator().CopyString(Record->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002504 if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002505 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Douglas Gregor75acd922011-09-27 23:30:47 +00002506 AddTemplateParameterChunks(Context, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002507 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002508 }
2509 break;
2510 }
2511 }
2512}
2513
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002514CodeCompletionString *CodeCompletionResult::CreateCodeCompletionString(Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002515 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002516 CodeCompletionTUInfo &CCTUInfo,
2517 bool IncludeBriefComments) {
2518 return CreateCodeCompletionString(S.Context, S.PP, Allocator, CCTUInfo,
2519 IncludeBriefComments);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002520}
2521
Douglas Gregor3545ff42009-09-21 16:56:56 +00002522/// \brief If possible, create a new code completion string for the given
2523/// result.
2524///
2525/// \returns Either a new, heap-allocated code completion string describing
2526/// how to use this result, or NULL to indicate that the string or name of the
2527/// result is all that is needed.
2528CodeCompletionString *
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002529CodeCompletionResult::CreateCodeCompletionString(ASTContext &Ctx,
2530 Preprocessor &PP,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002531 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002532 CodeCompletionTUInfo &CCTUInfo,
2533 bool IncludeBriefComments) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002534 CodeCompletionBuilder Result(Allocator, CCTUInfo, Priority, Availability);
Douglas Gregor9eb77012009-11-07 00:00:49 +00002535
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002536 PrintingPolicy Policy = getCompletionPrintingPolicy(Ctx, PP);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002537 if (Kind == RK_Pattern) {
2538 Pattern->Priority = Priority;
2539 Pattern->Availability = Availability;
Douglas Gregor78254c82012-03-27 23:34:16 +00002540
2541 if (Declaration) {
2542 Result.addParentContext(Declaration->getDeclContext());
Douglas Gregor78254c82012-03-27 23:34:16 +00002543 Pattern->ParentName = Result.getParentName();
2544 }
2545
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002546 return Pattern;
2547 }
Douglas Gregorf09935f2009-12-01 05:55:20 +00002548
Douglas Gregorf09935f2009-12-01 05:55:20 +00002549 if (Kind == RK_Keyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002550 Result.AddTypedTextChunk(Keyword);
2551 return Result.TakeString();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002552 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002553
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002554 if (Kind == RK_Macro) {
Argyrios Kyrtzidis09c9e812013-02-20 00:54:57 +00002555 const MacroDirective *MD = PP.getMacroDirectiveHistory(Macro);
2556 assert(MD && "Not a macro?");
2557 const MacroInfo *MI = MD->getInfo();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002558
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002559 Result.AddTypedTextChunk(
2560 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregorf09935f2009-12-01 05:55:20 +00002561
2562 if (!MI->isFunctionLike())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002563 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002564
2565 // Format a function-like macro with placeholders for the arguments.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002566 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor0c505312011-07-30 08:17:44 +00002567 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002568
2569 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
2570 if (MI->isC99Varargs()) {
2571 --AEnd;
2572
2573 if (A == AEnd) {
2574 Result.AddPlaceholderChunk("...");
2575 }
Douglas Gregor0c505312011-07-30 08:17:44 +00002576 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002577
Douglas Gregor0c505312011-07-30 08:17:44 +00002578 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002579 if (A != MI->arg_begin())
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002580 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3aa55262012-01-21 00:43:38 +00002581
2582 if (MI->isVariadic() && (A+1) == AEnd) {
Dylan Noblesmith2c1dd272012-02-05 02:13:05 +00002583 SmallString<32> Arg = (*A)->getName();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002584 if (MI->isC99Varargs())
2585 Arg += ", ...";
2586 else
2587 Arg += "...";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002588 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor3aa55262012-01-21 00:43:38 +00002589 break;
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002590 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002591
2592 // Non-variadic macros are simple.
2593 Result.AddPlaceholderChunk(
2594 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregor0c505312011-07-30 08:17:44 +00002595 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002596 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002597 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002598 }
2599
Douglas Gregorf64acca2010-05-25 21:41:55 +00002600 assert(Kind == RK_Declaration && "Missed a result kind?");
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002601 const NamedDecl *ND = Declaration;
Douglas Gregor78254c82012-03-27 23:34:16 +00002602 Result.addParentContext(ND->getDeclContext());
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002603
2604 if (IncludeBriefComments) {
2605 // Add documentation comment, if it exists.
Dmitri Gribenkoa43ec182012-08-11 00:51:43 +00002606 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(ND)) {
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002607 Result.addBriefComment(RC->getBriefText(Ctx));
2608 }
2609 }
2610
Douglas Gregor9eb77012009-11-07 00:00:49 +00002611 if (StartsNestedNameSpecifier) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002612 Result.AddTypedTextChunk(
2613 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002614 Result.AddTextChunk("::");
2615 return Result.TakeString();
Douglas Gregor9eb77012009-11-07 00:00:49 +00002616 }
Erik Verbruggen98ea7f62011-10-14 15:31:08 +00002617
2618 for (Decl::attr_iterator i = ND->attr_begin(); i != ND->attr_end(); ++i) {
2619 if (AnnotateAttr *Attr = dyn_cast_or_null<AnnotateAttr>(*i)) {
2620 Result.AddAnnotation(Result.getAllocator().CopyString(Attr->getAnnotation()));
2621 }
2622 }
Douglas Gregor9eb77012009-11-07 00:00:49 +00002623
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002624 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002625
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002626 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002627 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002628 Ctx, Policy);
2629 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002630 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002631 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002632 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002633 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002634 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002635 }
2636
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002637 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002638 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002639 Ctx, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002640 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002641 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002642
Douglas Gregor3545ff42009-09-21 16:56:56 +00002643 // Figure out which template parameters are deduced (or have default
2644 // arguments).
Benjamin Kramere0513cb2012-01-30 16:17:39 +00002645 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002646 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002647 unsigned LastDeducibleArgument;
2648 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2649 --LastDeducibleArgument) {
2650 if (!Deduced[LastDeducibleArgument - 1]) {
2651 // C++0x: Figure out if the template argument has a default. If so,
2652 // the user doesn't need to type this argument.
2653 // FIXME: We need to abstract template parameters better!
2654 bool HasDefaultArg = false;
2655 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002656 LastDeducibleArgument - 1);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002657 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2658 HasDefaultArg = TTP->hasDefaultArgument();
2659 else if (NonTypeTemplateParmDecl *NTTP
2660 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2661 HasDefaultArg = NTTP->hasDefaultArgument();
2662 else {
2663 assert(isa<TemplateTemplateParmDecl>(Param));
2664 HasDefaultArg
Douglas Gregor9eb77012009-11-07 00:00:49 +00002665 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002666 }
2667
2668 if (!HasDefaultArg)
2669 break;
2670 }
2671 }
2672
2673 if (LastDeducibleArgument) {
2674 // Some of the function template arguments cannot be deduced from a
2675 // function call, so we introduce an explicit template argument list
2676 // containing all of the arguments up to the first deducible argument.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002677 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002678 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor3545ff42009-09-21 16:56:56 +00002679 LastDeducibleArgument);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002680 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002681 }
2682
2683 // Add the function parameters
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002684 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002685 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002686 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002687 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002688 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002689 }
2690
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002691 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002692 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002693 Ctx, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002694 Result.AddTypedTextChunk(
2695 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002696 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002697 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002698 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002699 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002700 }
2701
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002702 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregord3c5d792009-11-17 16:44:22 +00002703 Selector Sel = Method->getSelector();
2704 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002705 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002706 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002707 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002708 }
2709
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002710 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregor1b605f72009-11-19 01:08:35 +00002711 SelName += ':';
2712 if (StartParameter == 0)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002713 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002714 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002715 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002716
2717 // If there is only one parameter, and we're past it, add an empty
2718 // typed-text chunk since there is nothing to type.
2719 if (Method->param_size() == 1)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002720 Result.AddTypedTextChunk("");
Douglas Gregor1b605f72009-11-19 01:08:35 +00002721 }
Douglas Gregord3c5d792009-11-17 16:44:22 +00002722 unsigned Idx = 0;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002723 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2724 PEnd = Method->param_end();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002725 P != PEnd; (void)++P, ++Idx) {
2726 if (Idx > 0) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00002727 std::string Keyword;
2728 if (Idx > StartParameter)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002729 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregord3c5d792009-11-17 16:44:22 +00002730 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramer632500c2011-07-26 16:59:25 +00002731 Keyword += II->getName();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002732 Keyword += ":";
Douglas Gregor95887f92010-07-08 23:20:03 +00002733 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002734 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002735 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002736 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002737 }
Douglas Gregor1b605f72009-11-19 01:08:35 +00002738
2739 // If we're before the starting parameter, skip the placeholder.
2740 if (Idx < StartParameter)
2741 continue;
Douglas Gregord3c5d792009-11-17 16:44:22 +00002742
2743 std::string Arg;
Douglas Gregore90dd002010-08-24 16:15:59 +00002744
2745 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002746 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregore90dd002010-08-24 16:15:59 +00002747 else {
John McCall31168b02011-06-15 23:02:42 +00002748 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor8f08d742011-07-30 07:55:26 +00002749 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2750 + Arg + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002751 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregor981a0c42010-08-29 19:47:46 +00002752 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramer632500c2011-07-26 16:59:25 +00002753 Arg += II->getName();
Douglas Gregore90dd002010-08-24 16:15:59 +00002754 }
2755
Douglas Gregor400f5972010-08-31 05:13:43 +00002756 if (Method->isVariadic() && (P + 1) == PEnd)
2757 Arg += ", ...";
2758
Douglas Gregor95887f92010-07-08 23:20:03 +00002759 if (DeclaringEntity)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002760 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor95887f92010-07-08 23:20:03 +00002761 else if (AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002762 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8537c52009-11-19 07:41:15 +00002763 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002764 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002765 }
2766
Douglas Gregor04c5f972009-12-23 00:21:46 +00002767 if (Method->isVariadic()) {
Douglas Gregor400f5972010-08-31 05:13:43 +00002768 if (Method->param_size() == 0) {
2769 if (DeclaringEntity)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002770 Result.AddTextChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002771 else if (AllParametersAreInformative)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002772 Result.AddInformativeChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002773 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002774 Result.AddPlaceholderChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002775 }
Douglas Gregordbb71db2010-08-23 23:51:41 +00002776
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002777 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor04c5f972009-12-23 00:21:46 +00002778 }
2779
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002780 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002781 }
2782
Douglas Gregorf09935f2009-12-01 05:55:20 +00002783 if (Qualifier)
Douglas Gregor5bf52692009-09-22 23:15:58 +00002784 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002785 Ctx, Policy);
Douglas Gregorf09935f2009-12-01 05:55:20 +00002786
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002787 Result.AddTypedTextChunk(
2788 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002789 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002790}
2791
Douglas Gregorf0f51982009-09-23 00:34:09 +00002792CodeCompletionString *
2793CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2794 unsigned CurrentArg,
Douglas Gregor36e3b5c2010-10-11 21:37:58 +00002795 Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002796 CodeCompletionAllocator &Allocator,
2797 CodeCompletionTUInfo &CCTUInfo) const {
Douglas Gregor75acd922011-09-27 23:30:47 +00002798 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCall31168b02011-06-15 23:02:42 +00002799
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002800 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002801 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002802 FunctionDecl *FDecl = getFunction();
Douglas Gregor75acd922011-09-27 23:30:47 +00002803 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002804 const FunctionProtoType *Proto
2805 = dyn_cast<FunctionProtoType>(getFunctionType());
2806 if (!FDecl && !Proto) {
2807 // Function without a prototype. Just give the return type and a
2808 // highlighted ellipsis.
2809 const FunctionType *FT = getFunctionType();
Douglas Gregor304f9b02011-02-01 21:15:40 +00002810 Result.AddTextChunk(GetCompletionTypeString(FT->getResultType(),
Douglas Gregor75acd922011-09-27 23:30:47 +00002811 S.Context, Policy,
Douglas Gregor304f9b02011-02-01 21:15:40 +00002812 Result.getAllocator()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002813 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2814 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2815 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002816 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002817 }
2818
2819 if (FDecl)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002820 Result.AddTextChunk(
2821 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002822 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002823 Result.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002824 Result.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00002825 Proto->getResultType().getAsString(Policy)));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002826
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002827 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002828 unsigned NumParams = FDecl? FDecl->getNumParams() : Proto->getNumArgs();
2829 for (unsigned I = 0; I != NumParams; ++I) {
2830 if (I)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002831 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002832
2833 std::string ArgString;
2834 QualType ArgType;
2835
2836 if (FDecl) {
2837 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2838 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2839 } else {
2840 ArgType = Proto->getArgType(I);
2841 }
2842
John McCall31168b02011-06-15 23:02:42 +00002843 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002844
2845 if (I == CurrentArg)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002846 Result.AddChunk(CodeCompletionString::CK_CurrentParameter,
2847 Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002848 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002849 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002850 }
2851
2852 if (Proto && Proto->isVariadic()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002853 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002854 if (CurrentArg < NumParams)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002855 Result.AddTextChunk("...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002856 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002857 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002858 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002859 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002860
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002861 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002862}
2863
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002864unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002865 const LangOptions &LangOpts,
Douglas Gregor6e240332010-08-16 16:18:59 +00002866 bool PreferredTypeIsPointer) {
2867 unsigned Priority = CCP_Macro;
2868
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002869 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2870 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2871 MacroName.equals("Nil")) {
Douglas Gregor6e240332010-08-16 16:18:59 +00002872 Priority = CCP_Constant;
2873 if (PreferredTypeIsPointer)
2874 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002875 }
2876 // Treat "YES", "NO", "true", and "false" as constants.
2877 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2878 MacroName.equals("true") || MacroName.equals("false"))
2879 Priority = CCP_Constant;
2880 // Treat "bool" as a type.
2881 else if (MacroName.equals("bool"))
2882 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2883
Douglas Gregor6e240332010-08-16 16:18:59 +00002884
2885 return Priority;
2886}
2887
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002888CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002889 if (!D)
2890 return CXCursor_UnexposedDecl;
2891
2892 switch (D->getKind()) {
2893 case Decl::Enum: return CXCursor_EnumDecl;
2894 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2895 case Decl::Field: return CXCursor_FieldDecl;
2896 case Decl::Function:
2897 return CXCursor_FunctionDecl;
2898 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2899 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002900 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregordeafd0b2011-12-27 22:43:10 +00002901
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002902 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002903 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2904 case Decl::ObjCMethod:
2905 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2906 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2907 case Decl::CXXMethod: return CXCursor_CXXMethod;
2908 case Decl::CXXConstructor: return CXCursor_Constructor;
2909 case Decl::CXXDestructor: return CXCursor_Destructor;
2910 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2911 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002912 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002913 case Decl::ParmVar: return CXCursor_ParmDecl;
2914 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smithdda56e42011-04-15 14:24:37 +00002915 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002916 case Decl::Var: return CXCursor_VarDecl;
2917 case Decl::Namespace: return CXCursor_Namespace;
2918 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2919 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2920 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2921 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2922 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2923 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis12afd702011-09-30 17:58:23 +00002924 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002925 case Decl::ClassTemplatePartialSpecialization:
2926 return CXCursor_ClassTemplatePartialSpecialization;
2927 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor3e653b32012-04-30 23:41:16 +00002928 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002929
2930 case Decl::Using:
2931 case Decl::UnresolvedUsingValue:
2932 case Decl::UnresolvedUsingTypename:
2933 return CXCursor_UsingDeclaration;
2934
Douglas Gregor4cd65962011-06-03 23:08:58 +00002935 case Decl::ObjCPropertyImpl:
2936 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2937 case ObjCPropertyImplDecl::Dynamic:
2938 return CXCursor_ObjCDynamicDecl;
2939
2940 case ObjCPropertyImplDecl::Synthesize:
2941 return CXCursor_ObjCSynthesizeDecl;
2942 }
Argyrios Kyrtzidis50e5b1d2012-10-05 00:22:24 +00002943
2944 case Decl::Import:
2945 return CXCursor_ModuleImportDecl;
Douglas Gregor4cd65962011-06-03 23:08:58 +00002946
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002947 default:
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002948 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002949 switch (TD->getTagKind()) {
Joao Matosdc86f942012-08-31 18:45:21 +00002950 case TTK_Interface: // fall through
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002951 case TTK_Struct: return CXCursor_StructDecl;
2952 case TTK_Class: return CXCursor_ClassDecl;
2953 case TTK_Union: return CXCursor_UnionDecl;
2954 case TTK_Enum: return CXCursor_EnumDecl;
2955 }
2956 }
2957 }
2958
2959 return CXCursor_UnexposedDecl;
2960}
2961
Douglas Gregor55b037b2010-07-08 20:55:51 +00002962static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor8cb17462012-10-09 16:01:50 +00002963 bool IncludeUndefined,
Douglas Gregor55b037b2010-07-08 20:55:51 +00002964 bool TargetTypeIsPointer = false) {
John McCall276321a2010-08-25 06:19:51 +00002965 typedef CodeCompletionResult Result;
Douglas Gregor55b037b2010-07-08 20:55:51 +00002966
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002967 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002968
Douglas Gregor9eb77012009-11-07 00:00:49 +00002969 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2970 MEnd = PP.macro_end();
Douglas Gregor55b037b2010-07-08 20:55:51 +00002971 M != MEnd; ++M) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00002972 if (IncludeUndefined || M->first->hasMacroDefinition())
2973 Results.AddResult(Result(M->first,
Douglas Gregor6e240332010-08-16 16:18:59 +00002974 getMacroUsagePriority(M->first->getName(),
David Blaikiebbafb8a2012-03-11 07:00:24 +00002975 PP.getLangOpts(),
Douglas Gregor6e240332010-08-16 16:18:59 +00002976 TargetTypeIsPointer)));
Douglas Gregor55b037b2010-07-08 20:55:51 +00002977 }
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002978
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002979 Results.ExitScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002980
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002981}
2982
Douglas Gregorce0e8562010-08-23 21:54:33 +00002983static void AddPrettyFunctionResults(const LangOptions &LangOpts,
2984 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00002985 typedef CodeCompletionResult Result;
Douglas Gregorce0e8562010-08-23 21:54:33 +00002986
2987 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002988
Douglas Gregorce0e8562010-08-23 21:54:33 +00002989 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
2990 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith2bf7fdb2013-01-02 11:42:31 +00002991 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregorce0e8562010-08-23 21:54:33 +00002992 Results.AddResult(Result("__func__", CCP_Constant));
2993 Results.ExitScope();
2994}
2995
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00002996static void HandleCodeCompleteResults(Sema *S,
2997 CodeCompleteConsumer *CodeCompleter,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00002998 CodeCompletionContext Context,
John McCall276321a2010-08-25 06:19:51 +00002999 CodeCompletionResult *Results,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003000 unsigned NumResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00003001 if (CodeCompleter)
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003002 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor3545ff42009-09-21 16:56:56 +00003003}
3004
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003005static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3006 Sema::ParserCompletionContext PCC) {
3007 switch (PCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00003008 case Sema::PCC_Namespace:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003009 return CodeCompletionContext::CCC_TopLevel;
3010
John McCallfaf5fb42010-08-26 23:41:50 +00003011 case Sema::PCC_Class:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003012 return CodeCompletionContext::CCC_ClassStructUnion;
3013
John McCallfaf5fb42010-08-26 23:41:50 +00003014 case Sema::PCC_ObjCInterface:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003015 return CodeCompletionContext::CCC_ObjCInterface;
3016
John McCallfaf5fb42010-08-26 23:41:50 +00003017 case Sema::PCC_ObjCImplementation:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003018 return CodeCompletionContext::CCC_ObjCImplementation;
3019
John McCallfaf5fb42010-08-26 23:41:50 +00003020 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003021 return CodeCompletionContext::CCC_ObjCIvarList;
3022
John McCallfaf5fb42010-08-26 23:41:50 +00003023 case Sema::PCC_Template:
3024 case Sema::PCC_MemberTemplate:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003025 if (S.CurContext->isFileContext())
3026 return CodeCompletionContext::CCC_TopLevel;
David Blaikie8a40f702012-01-17 06:56:22 +00003027 if (S.CurContext->isRecord())
Douglas Gregor0ac41382010-09-23 23:01:17 +00003028 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie8a40f702012-01-17 06:56:22 +00003029 return CodeCompletionContext::CCC_Other;
Douglas Gregor0ac41382010-09-23 23:01:17 +00003030
John McCallfaf5fb42010-08-26 23:41:50 +00003031 case Sema::PCC_RecoveryInFunction:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003032 return CodeCompletionContext::CCC_Recovery;
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003033
John McCallfaf5fb42010-08-26 23:41:50 +00003034 case Sema::PCC_ForInit:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003035 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3036 S.getLangOpts().ObjC1)
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003037 return CodeCompletionContext::CCC_ParenthesizedExpression;
3038 else
3039 return CodeCompletionContext::CCC_Expression;
3040
3041 case Sema::PCC_Expression:
John McCallfaf5fb42010-08-26 23:41:50 +00003042 case Sema::PCC_Condition:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003043 return CodeCompletionContext::CCC_Expression;
3044
John McCallfaf5fb42010-08-26 23:41:50 +00003045 case Sema::PCC_Statement:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003046 return CodeCompletionContext::CCC_Statement;
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003047
John McCallfaf5fb42010-08-26 23:41:50 +00003048 case Sema::PCC_Type:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003049 return CodeCompletionContext::CCC_Type;
Douglas Gregor5e35d592010-09-14 23:59:36 +00003050
3051 case Sema::PCC_ParenthesizedExpression:
3052 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor80039242011-02-15 20:33:25 +00003053
3054 case Sema::PCC_LocalDeclarationSpecifiers:
3055 return CodeCompletionContext::CCC_Type;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003056 }
David Blaikie8a40f702012-01-17 06:56:22 +00003057
3058 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003059}
3060
Douglas Gregorac322ec2010-08-27 21:18:54 +00003061/// \brief If we're in a C++ virtual member function, add completion results
3062/// that invoke the functions we override, since it's common to invoke the
3063/// overridden function as well as adding new functionality.
3064///
3065/// \param S The semantic analysis object for which we are generating results.
3066///
3067/// \param InContext This context in which the nested-name-specifier preceding
3068/// the code-completion point
3069static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3070 ResultBuilder &Results) {
3071 // Look through blocks.
3072 DeclContext *CurContext = S.CurContext;
3073 while (isa<BlockDecl>(CurContext))
3074 CurContext = CurContext->getParent();
3075
3076
3077 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3078 if (!Method || !Method->isVirtual())
3079 return;
3080
3081 // We need to have names for all of the parameters, if we're going to
3082 // generate a forwarding call.
3083 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3084 PEnd = Method->param_end();
3085 P != PEnd;
3086 ++P) {
3087 if (!(*P)->getDeclName())
3088 return;
3089 }
3090
Douglas Gregor75acd922011-09-27 23:30:47 +00003091 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003092 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3093 MEnd = Method->end_overridden_methods();
3094 M != MEnd; ++M) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003095 CodeCompletionBuilder Builder(Results.getAllocator(),
3096 Results.getCodeCompletionTUInfo());
Dmitri Gribenko6cfb1532013-02-14 13:53:30 +00003097 const CXXMethodDecl *Overridden = *M;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003098 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3099 continue;
3100
3101 // If we need a nested-name-specifier, add one now.
3102 if (!InContext) {
3103 NestedNameSpecifier *NNS
3104 = getRequiredQualification(S.Context, CurContext,
3105 Overridden->getDeclContext());
3106 if (NNS) {
3107 std::string Str;
3108 llvm::raw_string_ostream OS(Str);
Douglas Gregor75acd922011-09-27 23:30:47 +00003109 NNS->print(OS, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003110 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003111 }
3112 } else if (!InContext->Equals(Overridden->getDeclContext()))
3113 continue;
3114
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003115 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003116 Overridden->getNameAsString()));
3117 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003118 bool FirstParam = true;
3119 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3120 PEnd = Method->param_end();
3121 P != PEnd; ++P) {
3122 if (FirstParam)
3123 FirstParam = false;
3124 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003125 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003126
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003127 Builder.AddPlaceholderChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003128 (*P)->getIdentifier()->getName()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003129 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003130 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3131 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorac322ec2010-08-27 21:18:54 +00003132 CCP_SuperCompletion,
Douglas Gregor78254c82012-03-27 23:34:16 +00003133 CXCursor_CXXMethod,
3134 CXAvailability_Available,
3135 Overridden));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003136 Results.Ignore(Overridden);
3137 }
3138}
3139
Douglas Gregor07f43572012-01-29 18:15:03 +00003140void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3141 ModuleIdPath Path) {
3142 typedef CodeCompletionResult Result;
3143 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003144 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor07f43572012-01-29 18:15:03 +00003145 CodeCompletionContext::CCC_Other);
3146 Results.EnterNewScope();
3147
3148 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003149 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor07f43572012-01-29 18:15:03 +00003150 typedef CodeCompletionResult Result;
3151 if (Path.empty()) {
3152 // Enumerate all top-level modules.
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003153 SmallVector<Module *, 8> Modules;
Douglas Gregor07f43572012-01-29 18:15:03 +00003154 PP.getHeaderSearchInfo().collectAllModules(Modules);
3155 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3156 Builder.AddTypedTextChunk(
3157 Builder.getAllocator().CopyString(Modules[I]->Name));
3158 Results.AddResult(Result(Builder.TakeString(),
3159 CCP_Declaration,
3160 CXCursor_NotImplemented,
3161 Modules[I]->isAvailable()
3162 ? CXAvailability_Available
3163 : CXAvailability_NotAvailable));
3164 }
3165 } else {
3166 // Load the named module.
3167 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3168 Module::AllVisible,
3169 /*IsInclusionDirective=*/false);
3170 // Enumerate submodules.
3171 if (Mod) {
3172 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3173 SubEnd = Mod->submodule_end();
3174 Sub != SubEnd; ++Sub) {
3175
3176 Builder.AddTypedTextChunk(
3177 Builder.getAllocator().CopyString((*Sub)->Name));
3178 Results.AddResult(Result(Builder.TakeString(),
3179 CCP_Declaration,
3180 CXCursor_NotImplemented,
3181 (*Sub)->isAvailable()
3182 ? CXAvailability_Available
3183 : CXAvailability_NotAvailable));
3184 }
3185 }
3186 }
3187 Results.ExitScope();
3188 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3189 Results.data(),Results.size());
3190}
3191
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003192void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003193 ParserCompletionContext CompletionContext) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003194 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003195 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003196 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003197 Results.EnterNewScope();
Douglas Gregor50832e02010-09-20 22:39:41 +00003198
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003199 // Determine how to filter results, e.g., so that the names of
3200 // values (functions, enumerators, function templates, etc.) are
3201 // only allowed where we can have an expression.
3202 switch (CompletionContext) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003203 case PCC_Namespace:
3204 case PCC_Class:
3205 case PCC_ObjCInterface:
3206 case PCC_ObjCImplementation:
3207 case PCC_ObjCInstanceVariableList:
3208 case PCC_Template:
3209 case PCC_MemberTemplate:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003210 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003211 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003212 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3213 break;
3214
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003215 case PCC_Statement:
Douglas Gregor5e35d592010-09-14 23:59:36 +00003216 case PCC_ParenthesizedExpression:
Douglas Gregor4d755e82010-08-24 23:58:17 +00003217 case PCC_Expression:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003218 case PCC_ForInit:
3219 case PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003220 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor70febae2010-05-28 00:49:12 +00003221 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3222 else
3223 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003224
David Blaikiebbafb8a2012-03-11 07:00:24 +00003225 if (getLangOpts().CPlusPlus)
Douglas Gregorac322ec2010-08-27 21:18:54 +00003226 MaybeAddOverrideCalls(*this, /*InContext=*/0, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003227 break;
Douglas Gregor6da3db42010-05-25 05:58:43 +00003228
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003229 case PCC_RecoveryInFunction:
Douglas Gregor6da3db42010-05-25 05:58:43 +00003230 // Unfiltered
3231 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003232 }
3233
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003234 // If we are in a C++ non-static member function, check the qualifiers on
3235 // the member function to filter/prioritize the results list.
3236 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3237 if (CurMethod->isInstance())
3238 Results.setObjectTypeQualifiers(
3239 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3240
Douglas Gregorc580c522010-01-14 01:09:38 +00003241 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003242 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3243 CodeCompleter->includeGlobals());
Douglas Gregor92253692009-12-07 09:54:55 +00003244
Douglas Gregorf98e6a22010-01-13 23:51:12 +00003245 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor92253692009-12-07 09:54:55 +00003246 Results.ExitScope();
3247
Douglas Gregorce0e8562010-08-23 21:54:33 +00003248 switch (CompletionContext) {
Douglas Gregor5e35d592010-09-14 23:59:36 +00003249 case PCC_ParenthesizedExpression:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003250 case PCC_Expression:
3251 case PCC_Statement:
3252 case PCC_RecoveryInFunction:
3253 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00003254 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003255 break;
3256
3257 case PCC_Namespace:
3258 case PCC_Class:
3259 case PCC_ObjCInterface:
3260 case PCC_ObjCImplementation:
3261 case PCC_ObjCInstanceVariableList:
3262 case PCC_Template:
3263 case PCC_MemberTemplate:
3264 case PCC_ForInit:
3265 case PCC_Condition:
3266 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003267 case PCC_LocalDeclarationSpecifiers:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003268 break;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003269 }
3270
Douglas Gregor9eb77012009-11-07 00:00:49 +00003271 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003272 AddMacroResults(PP, Results, false);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003273
Douglas Gregor50832e02010-09-20 22:39:41 +00003274 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003275 Results.data(),Results.size());
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00003276}
3277
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003278static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3279 ParsedType Receiver,
3280 IdentifierInfo **SelIdents,
3281 unsigned NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00003282 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003283 bool IsSuper,
3284 ResultBuilder &Results);
3285
3286void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3287 bool AllowNonIdentifiers,
3288 bool AllowNestedNameSpecifiers) {
John McCall276321a2010-08-25 06:19:51 +00003289 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003290 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003291 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003292 AllowNestedNameSpecifiers
3293 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3294 : CodeCompletionContext::CCC_Name);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003295 Results.EnterNewScope();
3296
3297 // Type qualifiers can come after names.
3298 Results.AddResult(Result("const"));
3299 Results.AddResult(Result("volatile"));
David Blaikiebbafb8a2012-03-11 07:00:24 +00003300 if (getLangOpts().C99)
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003301 Results.AddResult(Result("restrict"));
3302
David Blaikiebbafb8a2012-03-11 07:00:24 +00003303 if (getLangOpts().CPlusPlus) {
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003304 if (AllowNonIdentifiers) {
3305 Results.AddResult(Result("operator"));
3306 }
3307
3308 // Add nested-name-specifiers.
3309 if (AllowNestedNameSpecifiers) {
3310 Results.allowNestedNameSpecifiers();
Douglas Gregor0ac41382010-09-23 23:01:17 +00003311 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003312 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3313 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3314 CodeCompleter->includeGlobals());
Douglas Gregor0ac41382010-09-23 23:01:17 +00003315 Results.setFilter(0);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003316 }
3317 }
3318 Results.ExitScope();
3319
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003320 // If we're in a context where we might have an expression (rather than a
3321 // declaration), and what we've seen so far is an Objective-C type that could
3322 // be a receiver of a class message, this may be a class message send with
3323 // the initial opening bracket '[' missing. Add appropriate completions.
3324 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
3325 DS.getTypeSpecType() == DeclSpec::TST_typename &&
3326 DS.getStorageClassSpecAsWritten() == DeclSpec::SCS_unspecified &&
3327 !DS.isThreadSpecified() && !DS.isExternInLinkageSpec() &&
3328 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3329 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
3330 DS.getTypeQualifiers() == 0 &&
3331 S &&
3332 (S->getFlags() & Scope::DeclScope) != 0 &&
3333 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3334 Scope::FunctionPrototypeScope |
3335 Scope::AtCatchScope)) == 0) {
3336 ParsedType T = DS.getRepAsType();
3337 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Douglas Gregorf86e4da2010-09-20 23:34:21 +00003338 AddClassMessageCompletions(*this, S, T, 0, 0, false, false, Results);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003339 }
3340
Douglas Gregor56ccce02010-08-24 04:59:56 +00003341 // Note that we intentionally suppress macro results here, since we do not
3342 // encourage using macros to produce the names of entities.
3343
Douglas Gregor0ac41382010-09-23 23:01:17 +00003344 HandleCodeCompleteResults(this, CodeCompleter,
3345 Results.getCompletionContext(),
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003346 Results.data(), Results.size());
3347}
3348
Douglas Gregor68762e72010-08-23 21:17:50 +00003349struct Sema::CodeCompleteExpressionData {
3350 CodeCompleteExpressionData(QualType PreferredType = QualType())
3351 : PreferredType(PreferredType), IntegralConstantExpression(false),
3352 ObjCCollection(false) { }
3353
3354 QualType PreferredType;
3355 bool IntegralConstantExpression;
3356 bool ObjCCollection;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003357 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregor68762e72010-08-23 21:17:50 +00003358};
3359
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003360/// \brief Perform code-completion in an expression context when we know what
3361/// type we're looking for.
Douglas Gregor68762e72010-08-23 21:17:50 +00003362void Sema::CodeCompleteExpression(Scope *S,
3363 const CodeCompleteExpressionData &Data) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003364 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003365 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003366 CodeCompletionContext::CCC_Expression);
Douglas Gregor68762e72010-08-23 21:17:50 +00003367 if (Data.ObjCCollection)
3368 Results.setFilter(&ResultBuilder::IsObjCCollection);
3369 else if (Data.IntegralConstantExpression)
Douglas Gregor85b50632010-07-28 21:50:18 +00003370 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003371 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003372 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3373 else
3374 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregor68762e72010-08-23 21:17:50 +00003375
3376 if (!Data.PreferredType.isNull())
3377 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3378
3379 // Ignore any declarations that we were told that we don't care about.
3380 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3381 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003382
3383 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003384 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3385 CodeCompleter->includeGlobals());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003386
3387 Results.EnterNewScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003388 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003389 Results.ExitScope();
3390
Douglas Gregor55b037b2010-07-08 20:55:51 +00003391 bool PreferredTypeIsPointer = false;
Douglas Gregor68762e72010-08-23 21:17:50 +00003392 if (!Data.PreferredType.isNull())
3393 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3394 || Data.PreferredType->isMemberPointerType()
3395 || Data.PreferredType->isBlockPointerType();
Douglas Gregor55b037b2010-07-08 20:55:51 +00003396
Douglas Gregorce0e8562010-08-23 21:54:33 +00003397 if (S->getFnParent() &&
3398 !Data.ObjCCollection &&
3399 !Data.IntegralConstantExpression)
David Blaikiebbafb8a2012-03-11 07:00:24 +00003400 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003401
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003402 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003403 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003404 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor68762e72010-08-23 21:17:50 +00003405 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3406 Data.PreferredType),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003407 Results.data(),Results.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003408}
3409
Douglas Gregoreda7e542010-09-18 01:28:11 +00003410void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3411 if (E.isInvalid())
3412 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003413 else if (getLangOpts().ObjC1)
Douglas Gregoreda7e542010-09-18 01:28:11 +00003414 CodeCompleteObjCInstanceMessage(S, E.take(), 0, 0, false);
Douglas Gregored0b69d2010-09-15 16:23:04 +00003415}
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003416
Douglas Gregorb888acf2010-12-09 23:01:55 +00003417/// \brief The set of properties that have already been added, referenced by
3418/// property name.
3419typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3420
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003421/// \brief Retrieve the container definition, if any?
3422static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3423 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3424 if (Interface->hasDefinition())
3425 return Interface->getDefinition();
3426
3427 return Interface;
3428 }
3429
3430 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3431 if (Protocol->hasDefinition())
3432 return Protocol->getDefinition();
3433
3434 return Protocol;
3435 }
3436 return Container;
3437}
3438
3439static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor5d649882009-11-18 22:32:06 +00003440 bool AllowCategories,
Douglas Gregor95147142011-05-05 15:50:42 +00003441 bool AllowNullaryMethods,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003442 DeclContext *CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003443 AddedPropertiesSet &AddedProperties,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003444 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003445 typedef CodeCompletionResult Result;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003446
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003447 // Retrieve the definition.
3448 Container = getContainerDef(Container);
3449
Douglas Gregor9291bad2009-11-18 01:29:26 +00003450 // Add properties in this container.
3451 for (ObjCContainerDecl::prop_iterator P = Container->prop_begin(),
3452 PEnd = Container->prop_end();
3453 P != PEnd;
Douglas Gregorb888acf2010-12-09 23:01:55 +00003454 ++P) {
3455 if (AddedProperties.insert(P->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003456 Results.MaybeAddResult(Result(*P, Results.getBasePriority(*P), 0),
3457 CurContext);
Douglas Gregorb888acf2010-12-09 23:01:55 +00003458 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003459
Douglas Gregor95147142011-05-05 15:50:42 +00003460 // Add nullary methods
3461 if (AllowNullaryMethods) {
3462 ASTContext &Context = Container->getASTContext();
Douglas Gregor75acd922011-09-27 23:30:47 +00003463 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Douglas Gregor95147142011-05-05 15:50:42 +00003464 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
3465 MEnd = Container->meth_end();
3466 M != MEnd; ++M) {
3467 if (M->getSelector().isUnarySelector())
3468 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3469 if (AddedProperties.insert(Name)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003470 CodeCompletionBuilder Builder(Results.getAllocator(),
3471 Results.getCodeCompletionTUInfo());
David Blaikie40ed2972012-06-06 20:45:41 +00003472 AddResultTypeChunk(Context, Policy, *M, Builder);
Douglas Gregor95147142011-05-05 15:50:42 +00003473 Builder.AddTypedTextChunk(
3474 Results.getAllocator().CopyString(Name->getName()));
3475
David Blaikie40ed2972012-06-06 20:45:41 +00003476 Results.MaybeAddResult(Result(Builder.TakeString(), *M,
Douglas Gregor78254c82012-03-27 23:34:16 +00003477 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor95147142011-05-05 15:50:42 +00003478 CurContext);
3479 }
3480 }
3481 }
3482
3483
Douglas Gregor9291bad2009-11-18 01:29:26 +00003484 // Add properties in referenced protocols.
3485 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3486 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
3487 PEnd = Protocol->protocol_end();
3488 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003489 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3490 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003491 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor5d649882009-11-18 22:32:06 +00003492 if (AllowCategories) {
3493 // Look through categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00003494 for (ObjCInterfaceDecl::known_categories_iterator
3495 Cat = IFace->known_categories_begin(),
3496 CatEnd = IFace->known_categories_end();
3497 Cat != CatEnd; ++Cat)
3498 AddObjCProperties(*Cat, AllowCategories, AllowNullaryMethods,
Douglas Gregor95147142011-05-05 15:50:42 +00003499 CurContext, AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00003500 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003501
3502 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003503 for (ObjCInterfaceDecl::all_protocol_iterator
3504 I = IFace->all_referenced_protocol_begin(),
3505 E = IFace->all_referenced_protocol_end(); I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003506 AddObjCProperties(*I, AllowCategories, AllowNullaryMethods, CurContext,
3507 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003508
3509 // Look in the superclass.
3510 if (IFace->getSuperClass())
Douglas Gregor95147142011-05-05 15:50:42 +00003511 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3512 AllowNullaryMethods, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003513 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003514 } else if (const ObjCCategoryDecl *Category
3515 = dyn_cast<ObjCCategoryDecl>(Container)) {
3516 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003517 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
3518 PEnd = Category->protocol_end();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003519 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003520 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3521 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003522 }
3523}
3524
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003525void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor2436e712009-09-17 21:32:03 +00003526 SourceLocation OpLoc,
3527 bool IsArrow) {
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003528 if (!Base || !CodeCompleter)
Douglas Gregor2436e712009-09-17 21:32:03 +00003529 return;
3530
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003531 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3532 if (ConvertedBase.isInvalid())
3533 return;
3534 Base = ConvertedBase.get();
3535
John McCall276321a2010-08-25 06:19:51 +00003536 typedef CodeCompletionResult Result;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003537
Douglas Gregor2436e712009-09-17 21:32:03 +00003538 QualType BaseType = Base->getType();
Douglas Gregor3545ff42009-09-21 16:56:56 +00003539
3540 if (IsArrow) {
3541 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3542 BaseType = Ptr->getPointeeType();
3543 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003544 /*Do nothing*/ ;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003545 else
3546 return;
3547 }
3548
Douglas Gregor21325842011-07-07 16:03:39 +00003549 enum CodeCompletionContext::Kind contextKind;
3550
3551 if (IsArrow) {
3552 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3553 }
3554 else {
3555 if (BaseType->isObjCObjectPointerType() ||
3556 BaseType->isObjCObjectOrInterfaceType()) {
3557 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3558 }
3559 else {
3560 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3561 }
3562 }
3563
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003564 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003565 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00003566 CodeCompletionContext(contextKind,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003567 BaseType),
3568 &ResultBuilder::IsMember);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003569 Results.EnterNewScope();
3570 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003571 // Indicate that we are performing a member access, and the cv-qualifiers
3572 // for the base object type.
3573 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3574
Douglas Gregor9291bad2009-11-18 01:29:26 +00003575 // Access to a C/C++ class, struct, or union.
Douglas Gregor6ae4c522010-01-14 03:21:49 +00003576 Results.allowNestedNameSpecifiers();
Douglas Gregor09bbc652010-01-14 15:47:35 +00003577 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003578 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3579 CodeCompleter->includeGlobals());
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003580
David Blaikiebbafb8a2012-03-11 07:00:24 +00003581 if (getLangOpts().CPlusPlus) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003582 if (!Results.empty()) {
3583 // The "template" keyword can follow "->" or "." in the grammar.
3584 // However, we only want to suggest the template keyword if something
3585 // is dependent.
3586 bool IsDependent = BaseType->isDependentType();
3587 if (!IsDependent) {
3588 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
3589 if (DeclContext *Ctx = (DeclContext *)DepScope->getEntity()) {
3590 IsDependent = Ctx->isDependentContext();
3591 break;
3592 }
3593 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003594
Douglas Gregor9291bad2009-11-18 01:29:26 +00003595 if (IsDependent)
Douglas Gregor78a21012010-01-14 16:01:26 +00003596 Results.AddResult(Result("template"));
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003597 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003598 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003599 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3600 // Objective-C property reference.
Douglas Gregorb888acf2010-12-09 23:01:55 +00003601 AddedPropertiesSet AddedProperties;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003602
3603 // Add property results based on our interface.
3604 const ObjCObjectPointerType *ObjCPtr
3605 = BaseType->getAsObjCInterfacePointerType();
3606 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor95147142011-05-05 15:50:42 +00003607 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3608 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003609 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003610
3611 // Add properties from the protocols in a qualified interface.
3612 for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(),
3613 E = ObjCPtr->qual_end();
3614 I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003615 AddObjCProperties(*I, true, /*AllowNullaryMethods=*/true, CurContext,
3616 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003617 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCall8b07ec22010-05-15 11:32:37 +00003618 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003619 // Objective-C instance variable access.
3620 ObjCInterfaceDecl *Class = 0;
3621 if (const ObjCObjectPointerType *ObjCPtr
3622 = BaseType->getAs<ObjCObjectPointerType>())
3623 Class = ObjCPtr->getInterfaceDecl();
3624 else
John McCall8b07ec22010-05-15 11:32:37 +00003625 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003626
3627 // Add all ivars from this class and its superclasses.
Douglas Gregor2b8162b2010-01-14 16:08:12 +00003628 if (Class) {
3629 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3630 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor39982192010-08-15 06:18:01 +00003631 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3632 CodeCompleter->includeGlobals());
Douglas Gregor9291bad2009-11-18 01:29:26 +00003633 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003634 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003635
3636 // FIXME: How do we cope with isa?
3637
3638 Results.ExitScope();
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003639
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003640 // Hand off the results found for code completion.
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003641 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003642 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003643 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00003644}
3645
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003646void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3647 if (!CodeCompleter)
3648 return;
3649
Douglas Gregor3545ff42009-09-21 16:56:56 +00003650 ResultBuilder::LookupFilter Filter = 0;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003651 enum CodeCompletionContext::Kind ContextKind
3652 = CodeCompletionContext::CCC_Other;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003653 switch ((DeclSpec::TST)TagSpec) {
3654 case DeclSpec::TST_enum:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003655 Filter = &ResultBuilder::IsEnum;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003656 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003657 break;
3658
3659 case DeclSpec::TST_union:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003660 Filter = &ResultBuilder::IsUnion;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003661 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003662 break;
3663
3664 case DeclSpec::TST_struct:
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003665 case DeclSpec::TST_class:
Joao Matosdc86f942012-08-31 18:45:21 +00003666 case DeclSpec::TST_interface:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003667 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003668 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003669 break;
3670
3671 default:
David Blaikie83d382b2011-09-23 05:06:16 +00003672 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003673 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003674
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003675 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3676 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00003677 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCalle87beb22010-04-23 18:46:30 +00003678
3679 // First pass: look for tags.
3680 Results.setFilter(Filter);
Douglas Gregor39982192010-08-15 06:18:01 +00003681 LookupVisibleDecls(S, LookupTagName, Consumer,
3682 CodeCompleter->includeGlobals());
John McCalle87beb22010-04-23 18:46:30 +00003683
Douglas Gregor39982192010-08-15 06:18:01 +00003684 if (CodeCompleter->includeGlobals()) {
3685 // Second pass: look for nested name specifiers.
3686 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3687 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3688 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003689
Douglas Gregor0ac41382010-09-23 23:01:17 +00003690 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003691 Results.data(),Results.size());
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003692}
3693
Douglas Gregor28c78432010-08-27 17:35:51 +00003694void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003695 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003696 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003697 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor28c78432010-08-27 17:35:51 +00003698 Results.EnterNewScope();
3699 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3700 Results.AddResult("const");
3701 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3702 Results.AddResult("volatile");
David Blaikiebbafb8a2012-03-11 07:00:24 +00003703 if (getLangOpts().C99 &&
Douglas Gregor28c78432010-08-27 17:35:51 +00003704 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3705 Results.AddResult("restrict");
3706 Results.ExitScope();
3707 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003708 Results.getCompletionContext(),
Douglas Gregor28c78432010-08-27 17:35:51 +00003709 Results.data(), Results.size());
3710}
3711
Douglas Gregord328d572009-09-21 18:10:23 +00003712void Sema::CodeCompleteCase(Scope *S) {
John McCallaab3e412010-08-25 08:40:02 +00003713 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregord328d572009-09-21 18:10:23 +00003714 return;
John McCall5939b162011-08-06 07:30:58 +00003715
John McCallaab3e412010-08-25 08:40:02 +00003716 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCall5939b162011-08-06 07:30:58 +00003717 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3718 if (!type->isEnumeralType()) {
3719 CodeCompleteExpressionData Data(type);
Douglas Gregor68762e72010-08-23 21:17:50 +00003720 Data.IntegralConstantExpression = true;
3721 CodeCompleteExpression(S, Data);
Douglas Gregord328d572009-09-21 18:10:23 +00003722 return;
Douglas Gregor85b50632010-07-28 21:50:18 +00003723 }
Douglas Gregord328d572009-09-21 18:10:23 +00003724
3725 // Code-complete the cases of a switch statement over an enumeration type
3726 // by providing the list of
John McCall5939b162011-08-06 07:30:58 +00003727 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003728 if (EnumDecl *Def = Enum->getDefinition())
3729 Enum = Def;
Douglas Gregord328d572009-09-21 18:10:23 +00003730
3731 // Determine which enumerators we have already seen in the switch statement.
3732 // FIXME: Ideally, we would also be able to look *past* the code-completion
3733 // token, in case we are code-completing in the middle of the switch and not
3734 // at the end. However, we aren't able to do so at the moment.
3735 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Douglas Gregorf2510672009-09-21 19:57:38 +00003736 NestedNameSpecifier *Qualifier = 0;
Douglas Gregord328d572009-09-21 18:10:23 +00003737 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3738 SC = SC->getNextSwitchCase()) {
3739 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3740 if (!Case)
3741 continue;
3742
3743 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3744 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3745 if (EnumConstantDecl *Enumerator
3746 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3747 // We look into the AST of the case statement to determine which
3748 // enumerator was named. Alternatively, we could compute the value of
3749 // the integral constant expression, then compare it against the
3750 // values of each enumerator. However, value-based approach would not
3751 // work as well with C++ templates where enumerators declared within a
3752 // template are type- and value-dependent.
3753 EnumeratorsSeen.insert(Enumerator);
3754
Douglas Gregorf2510672009-09-21 19:57:38 +00003755 // If this is a qualified-id, keep track of the nested-name-specifier
3756 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregord328d572009-09-21 18:10:23 +00003757 //
3758 // switch (TagD.getKind()) {
3759 // case TagDecl::TK_enum:
3760 // break;
3761 // case XXX
3762 //
Douglas Gregorf2510672009-09-21 19:57:38 +00003763 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregord328d572009-09-21 18:10:23 +00003764 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3765 // TK_struct, and TK_class.
Douglas Gregor4bd90e52009-10-23 18:54:35 +00003766 Qualifier = DRE->getQualifier();
Douglas Gregord328d572009-09-21 18:10:23 +00003767 }
3768 }
3769
David Blaikiebbafb8a2012-03-11 07:00:24 +00003770 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorf2510672009-09-21 19:57:38 +00003771 // If there are no prior enumerators in C++, check whether we have to
3772 // qualify the names of the enumerators that we suggest, because they
3773 // may not be visible in this scope.
Douglas Gregord3cebdb2012-02-01 05:02:47 +00003774 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorf2510672009-09-21 19:57:38 +00003775 }
3776
Douglas Gregord328d572009-09-21 18:10:23 +00003777 // Add any enumerators that have not yet been mentioned.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003778 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003779 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003780 CodeCompletionContext::CCC_Expression);
Douglas Gregord328d572009-09-21 18:10:23 +00003781 Results.EnterNewScope();
3782 for (EnumDecl::enumerator_iterator E = Enum->enumerator_begin(),
3783 EEnd = Enum->enumerator_end();
3784 E != EEnd; ++E) {
David Blaikie40ed2972012-06-06 20:45:41 +00003785 if (EnumeratorsSeen.count(*E))
Douglas Gregord328d572009-09-21 18:10:23 +00003786 continue;
3787
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003788 CodeCompletionResult R(*E, CCP_EnumInCase, Qualifier);
Douglas Gregor3a69eaf2011-02-18 23:30:37 +00003789 Results.AddResult(R, CurContext, 0, false);
Douglas Gregord328d572009-09-21 18:10:23 +00003790 }
3791 Results.ExitScope();
Douglas Gregor285560922010-04-06 20:02:15 +00003792
Douglas Gregor21325842011-07-07 16:03:39 +00003793 //We need to make sure we're setting the right context,
3794 //so only say we include macros if the code completer says we do
3795 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3796 if (CodeCompleter->includeMacros()) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00003797 AddMacroResults(PP, Results, false);
Douglas Gregor21325842011-07-07 16:03:39 +00003798 kind = CodeCompletionContext::CCC_OtherWithMacros;
3799 }
3800
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003801 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00003802 kind,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003803 Results.data(),Results.size());
Douglas Gregord328d572009-09-21 18:10:23 +00003804}
3805
Douglas Gregorcabea402009-09-22 15:41:20 +00003806namespace {
3807 struct IsBetterOverloadCandidate {
3808 Sema &S;
John McCallbc077cf2010-02-08 23:07:23 +00003809 SourceLocation Loc;
Douglas Gregorcabea402009-09-22 15:41:20 +00003810
3811 public:
John McCallbc077cf2010-02-08 23:07:23 +00003812 explicit IsBetterOverloadCandidate(Sema &S, SourceLocation Loc)
3813 : S(S), Loc(Loc) { }
Douglas Gregorcabea402009-09-22 15:41:20 +00003814
3815 bool
3816 operator()(const OverloadCandidate &X, const OverloadCandidate &Y) const {
John McCall5c32be02010-08-24 20:38:10 +00003817 return isBetterOverloadCandidate(S, X, Y, Loc);
Douglas Gregorcabea402009-09-22 15:41:20 +00003818 }
3819 };
3820}
3821
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003822static bool anyNullArguments(llvm::ArrayRef<Expr*> Args) {
3823 if (Args.size() && !Args.data())
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003824 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003825
3826 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003827 if (!Args[I])
3828 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003829
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003830 return false;
3831}
3832
Richard Trieu2bd04012011-09-09 02:00:50 +00003833void Sema::CodeCompleteCall(Scope *S, Expr *FnIn,
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003834 llvm::ArrayRef<Expr *> Args) {
Douglas Gregorcabea402009-09-22 15:41:20 +00003835 if (!CodeCompleter)
3836 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003837
3838 // When we're code-completing for a call, we fall back to ordinary
3839 // name code-completion whenever we can't produce specific
3840 // results. We may want to revisit this strategy in the future,
3841 // e.g., by merging the two kinds of results.
3842
Douglas Gregorcabea402009-09-22 15:41:20 +00003843 Expr *Fn = (Expr *)FnIn;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003844
Douglas Gregorcabea402009-09-22 15:41:20 +00003845 // Ignore type-dependent call expressions entirely.
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003846 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3847 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003848 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregorcabea402009-09-22 15:41:20 +00003849 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003850 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003851
John McCall57500772009-12-16 12:17:52 +00003852 // Build an overload candidate set based on the functions we find.
John McCallbc077cf2010-02-08 23:07:23 +00003853 SourceLocation Loc = Fn->getExprLoc();
3854 OverloadCandidateSet CandidateSet(Loc);
John McCall57500772009-12-16 12:17:52 +00003855
Douglas Gregorcabea402009-09-22 15:41:20 +00003856 // FIXME: What if we're calling something that isn't a function declaration?
3857 // FIXME: What if we're calling a pseudo-destructor?
3858 // FIXME: What if we're calling a member function?
3859
Douglas Gregorff59f672010-01-21 15:46:19 +00003860 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003861 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorff59f672010-01-21 15:46:19 +00003862
John McCall57500772009-12-16 12:17:52 +00003863 Expr *NakedFn = Fn->IgnoreParenCasts();
3864 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003865 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
John McCall57500772009-12-16 12:17:52 +00003866 /*PartialOverloading=*/ true);
3867 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3868 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorff59f672010-01-21 15:46:19 +00003869 if (FDecl) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00003870 if (!getLangOpts().CPlusPlus ||
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003871 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorff59f672010-01-21 15:46:19 +00003872 Results.push_back(ResultCandidate(FDecl));
3873 else
John McCallb89836b2010-01-26 01:37:31 +00003874 // FIXME: access?
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003875 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), Args,
3876 CandidateSet, false, /*PartialOverloading*/true);
Douglas Gregorff59f672010-01-21 15:46:19 +00003877 }
John McCall57500772009-12-16 12:17:52 +00003878 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003879
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003880 QualType ParamType;
3881
Douglas Gregorff59f672010-01-21 15:46:19 +00003882 if (!CandidateSet.empty()) {
3883 // Sort the overload candidate set by placing the best overloads first.
3884 std::stable_sort(CandidateSet.begin(), CandidateSet.end(),
John McCallbc077cf2010-02-08 23:07:23 +00003885 IsBetterOverloadCandidate(*this, Loc));
Douglas Gregorcabea402009-09-22 15:41:20 +00003886
Douglas Gregorff59f672010-01-21 15:46:19 +00003887 // Add the remaining viable overload candidates as code-completion reslults.
3888 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3889 CandEnd = CandidateSet.end();
3890 Cand != CandEnd; ++Cand) {
3891 if (Cand->Viable)
3892 Results.push_back(ResultCandidate(Cand->Function));
3893 }
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003894
3895 // From the viable candidates, try to determine the type of this parameter.
3896 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3897 if (const FunctionType *FType = Results[I].getFunctionType())
3898 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003899 if (Args.size() < Proto->getNumArgs()) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003900 if (ParamType.isNull())
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003901 ParamType = Proto->getArgType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003902 else if (!Context.hasSameUnqualifiedType(
3903 ParamType.getNonReferenceType(),
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003904 Proto->getArgType(Args.size()).getNonReferenceType())) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003905 ParamType = QualType();
3906 break;
3907 }
3908 }
3909 }
3910 } else {
3911 // Try to determine the parameter type from the type of the expression
3912 // being called.
3913 QualType FunctionType = Fn->getType();
3914 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3915 FunctionType = Ptr->getPointeeType();
3916 else if (const BlockPointerType *BlockPtr
3917 = FunctionType->getAs<BlockPointerType>())
3918 FunctionType = BlockPtr->getPointeeType();
3919 else if (const MemberPointerType *MemPtr
3920 = FunctionType->getAs<MemberPointerType>())
3921 FunctionType = MemPtr->getPointeeType();
3922
3923 if (const FunctionProtoType *Proto
3924 = FunctionType->getAs<FunctionProtoType>()) {
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003925 if (Args.size() < Proto->getNumArgs())
3926 ParamType = Proto->getArgType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003927 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003928 }
Douglas Gregor3ef59522009-12-11 19:06:04 +00003929
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003930 if (ParamType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003931 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003932 else
3933 CodeCompleteExpression(S, ParamType);
3934
Douglas Gregorc01890e2010-04-06 20:19:47 +00003935 if (!Results.empty())
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003936 CodeCompleter->ProcessOverloadCandidates(*this, Args.size(), Results.data(),
Douglas Gregor3ef59522009-12-11 19:06:04 +00003937 Results.size());
Douglas Gregorcabea402009-09-22 15:41:20 +00003938}
3939
John McCall48871652010-08-21 09:40:31 +00003940void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3941 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003942 if (!VD) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003943 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003944 return;
3945 }
3946
3947 CodeCompleteExpression(S, VD->getType());
3948}
3949
3950void Sema::CodeCompleteReturn(Scope *S) {
3951 QualType ResultType;
3952 if (isa<BlockDecl>(CurContext)) {
3953 if (BlockScopeInfo *BSI = getCurBlock())
3954 ResultType = BSI->ReturnType;
3955 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
3956 ResultType = Function->getResultType();
3957 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
3958 ResultType = Method->getResultType();
3959
3960 if (ResultType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003961 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003962 else
3963 CodeCompleteExpression(S, ResultType);
3964}
3965
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003966void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003967 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003968 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003969 mapCodeCompletionContext(*this, PCC_Statement));
3970 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3971 Results.EnterNewScope();
3972
3973 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3974 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3975 CodeCompleter->includeGlobals());
3976
3977 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
3978
3979 // "else" block
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003980 CodeCompletionBuilder Builder(Results.getAllocator(),
3981 Results.getCodeCompletionTUInfo());
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003982 Builder.AddTypedTextChunk("else");
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00003983 if (Results.includeCodePatterns()) {
3984 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3985 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
3986 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3987 Builder.AddPlaceholderChunk("statements");
3988 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3989 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
3990 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003991 Results.AddResult(Builder.TakeString());
3992
3993 // "else if" block
3994 Builder.AddTypedTextChunk("else");
3995 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3996 Builder.AddTextChunk("if");
3997 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3998 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003999 if (getLangOpts().CPlusPlus)
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004000 Builder.AddPlaceholderChunk("condition");
4001 else
4002 Builder.AddPlaceholderChunk("expression");
4003 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00004004 if (Results.includeCodePatterns()) {
4005 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4006 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4007 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4008 Builder.AddPlaceholderChunk("statements");
4009 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4010 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4011 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004012 Results.AddResult(Builder.TakeString());
4013
4014 Results.ExitScope();
4015
4016 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004017 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004018
4019 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00004020 AddMacroResults(PP, Results, false);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004021
4022 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4023 Results.data(),Results.size());
4024}
4025
Richard Trieu2bd04012011-09-09 02:00:50 +00004026void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004027 if (LHS)
4028 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4029 else
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004030 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004031}
4032
Jeffrey Yasskinc76498d2010-04-08 16:38:48 +00004033void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor2436e712009-09-17 21:32:03 +00004034 bool EnteringContext) {
4035 if (!SS.getScopeRep() || !CodeCompleter)
4036 return;
4037
Douglas Gregor3545ff42009-09-21 16:56:56 +00004038 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4039 if (!Ctx)
4040 return;
Douglas Gregor800f2f02009-12-11 18:28:39 +00004041
4042 // Try to instantiate any non-dependent declaration contexts before
4043 // we look in them.
John McCall0b66eb32010-05-01 00:40:08 +00004044 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregor800f2f02009-12-11 18:28:39 +00004045 return;
4046
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004047 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004048 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004049 CodeCompletionContext::CCC_Name);
Douglas Gregorac322ec2010-08-27 21:18:54 +00004050 Results.EnterNewScope();
Douglas Gregor0ac41382010-09-23 23:01:17 +00004051
Douglas Gregor3545ff42009-09-21 16:56:56 +00004052 // The "template" keyword can follow "::" in the grammar, but only
4053 // put it into the grammar if the nested-name-specifier is dependent.
4054 NestedNameSpecifier *NNS = (NestedNameSpecifier *)SS.getScopeRep();
4055 if (!Results.empty() && NNS->isDependent())
Douglas Gregor78a21012010-01-14 16:01:26 +00004056 Results.AddResult("template");
Douglas Gregorac322ec2010-08-27 21:18:54 +00004057
4058 // Add calls to overridden virtual functions, if there are any.
4059 //
4060 // FIXME: This isn't wonderful, because we don't know whether we're actually
4061 // in a context that permits expressions. This is a general issue with
4062 // qualified-id completions.
4063 if (!EnteringContext)
4064 MaybeAddOverrideCalls(*this, Ctx, Results);
4065 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004066
Douglas Gregorac322ec2010-08-27 21:18:54 +00004067 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4068 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4069
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004070 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorc1679ec2011-07-25 17:48:11 +00004071 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004072 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00004073}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004074
4075void Sema::CodeCompleteUsing(Scope *S) {
4076 if (!CodeCompleter)
4077 return;
4078
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004079 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004080 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004081 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4082 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004083 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004084
4085 // If we aren't in class scope, we could see the "namespace" keyword.
4086 if (!S->isClassScope())
John McCall276321a2010-08-25 06:19:51 +00004087 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004088
4089 // After "using", we can see anything that would start a
4090 // nested-name-specifier.
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004091 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004092 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4093 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004094 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004095
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004096 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004097 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004098 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004099}
4100
4101void Sema::CodeCompleteUsingDirective(Scope *S) {
4102 if (!CodeCompleter)
4103 return;
4104
Douglas Gregor3545ff42009-09-21 16:56:56 +00004105 // After "using namespace", we expect to see a namespace name or namespace
4106 // alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004107 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004108 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004109 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004110 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004111 Results.EnterNewScope();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004112 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004113 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4114 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004115 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004116 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004117 CodeCompletionContext::CCC_Namespace,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004118 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004119}
4120
4121void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4122 if (!CodeCompleter)
4123 return;
4124
Douglas Gregor3545ff42009-09-21 16:56:56 +00004125 DeclContext *Ctx = (DeclContext *)S->getEntity();
4126 if (!S->getParent())
4127 Ctx = Context.getTranslationUnitDecl();
4128
Douglas Gregor0ac41382010-09-23 23:01:17 +00004129 bool SuppressedGlobalResults
4130 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4131
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004132 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004133 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004134 SuppressedGlobalResults
4135 ? CodeCompletionContext::CCC_Namespace
4136 : CodeCompletionContext::CCC_Other,
4137 &ResultBuilder::IsNamespace);
4138
4139 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00004140 // We only want to see those namespaces that have already been defined
4141 // within this scope, because its likely that the user is creating an
4142 // extended namespace declaration. Keep track of the most recent
4143 // definition of each namespace.
4144 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4145 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4146 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4147 NS != NSEnd; ++NS)
David Blaikie40ed2972012-06-06 20:45:41 +00004148 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004149
4150 // Add the most recent definition (or extended definition) of each
4151 // namespace to the list of results.
Douglas Gregor64b12b52009-09-22 23:31:26 +00004152 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004153 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregor78254c82012-03-27 23:34:16 +00004154 NS = OrigToLatest.begin(),
4155 NSEnd = OrigToLatest.end();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004156 NS != NSEnd; ++NS)
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004157 Results.AddResult(CodeCompletionResult(
4158 NS->second, Results.getBasePriority(NS->second), 0),
Douglas Gregorfc59ce12010-01-14 16:14:35 +00004159 CurContext, 0, false);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004160 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004161 }
4162
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004163 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004164 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004165 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004166}
4167
4168void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4169 if (!CodeCompleter)
4170 return;
4171
Douglas Gregor3545ff42009-09-21 16:56:56 +00004172 // After "namespace", we expect to see a namespace or alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004173 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004174 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004175 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004176 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004177 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004178 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4179 CodeCompleter->includeGlobals());
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004180 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004181 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004182 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004183}
4184
Douglas Gregorc811ede2009-09-18 20:05:18 +00004185void Sema::CodeCompleteOperatorName(Scope *S) {
4186 if (!CodeCompleter)
4187 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004188
John McCall276321a2010-08-25 06:19:51 +00004189 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004190 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004191 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004192 CodeCompletionContext::CCC_Type,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004193 &ResultBuilder::IsType);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004194 Results.EnterNewScope();
Douglas Gregorc811ede2009-09-18 20:05:18 +00004195
Douglas Gregor3545ff42009-09-21 16:56:56 +00004196 // Add the names of overloadable operators.
4197#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4198 if (std::strcmp(Spelling, "?")) \
Douglas Gregor78a21012010-01-14 16:01:26 +00004199 Results.AddResult(Result(Spelling));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004200#include "clang/Basic/OperatorKinds.def"
4201
4202 // Add any type names visible from the current scope
Douglas Gregor6ae4c522010-01-14 03:21:49 +00004203 Results.allowNestedNameSpecifiers();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004204 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004205 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4206 CodeCompleter->includeGlobals());
Douglas Gregor3545ff42009-09-21 16:56:56 +00004207
4208 // Add any type specifiers
David Blaikiebbafb8a2012-03-11 07:00:24 +00004209 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004210 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004211
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004212 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004213 CodeCompletionContext::CCC_Type,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004214 Results.data(),Results.size());
Douglas Gregorc811ede2009-09-18 20:05:18 +00004215}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004216
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004217void Sema::CodeCompleteConstructorInitializer(Decl *ConstructorD,
Alexis Hunt1d792652011-01-08 20:30:50 +00004218 CXXCtorInitializer** Initializers,
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004219 unsigned NumInitializers) {
Douglas Gregor75acd922011-09-27 23:30:47 +00004220 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004221 CXXConstructorDecl *Constructor
4222 = static_cast<CXXConstructorDecl *>(ConstructorD);
4223 if (!Constructor)
4224 return;
4225
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004226 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004227 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004228 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004229 Results.EnterNewScope();
4230
4231 // Fill in any already-initialized fields or base classes.
4232 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4233 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
4234 for (unsigned I = 0; I != NumInitializers; ++I) {
4235 if (Initializers[I]->isBaseInitializer())
4236 InitializedBases.insert(
4237 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4238 else
Francois Pichetd583da02010-12-04 09:14:42 +00004239 InitializedFields.insert(cast<FieldDecl>(
4240 Initializers[I]->getAnyMember()));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004241 }
4242
4243 // Add completions for base classes.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004244 CodeCompletionBuilder Builder(Results.getAllocator(),
4245 Results.getCodeCompletionTUInfo());
Douglas Gregor99129ef2010-08-29 19:27:27 +00004246 bool SawLastInitializer = (NumInitializers == 0);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004247 CXXRecordDecl *ClassDecl = Constructor->getParent();
4248 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->bases_begin(),
4249 BaseEnd = ClassDecl->bases_end();
4250 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004251 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4252 SawLastInitializer
4253 = NumInitializers > 0 &&
4254 Initializers[NumInitializers - 1]->isBaseInitializer() &&
4255 Context.hasSameUnqualifiedType(Base->getType(),
4256 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004257 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004258 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004259
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004260 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004261 Results.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004262 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004263 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4264 Builder.AddPlaceholderChunk("args");
4265 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4266 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004267 SawLastInitializer? CCP_NextInitializer
4268 : CCP_MemberDeclaration));
4269 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004270 }
4271
4272 // Add completions for virtual base classes.
4273 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->vbases_begin(),
4274 BaseEnd = ClassDecl->vbases_end();
4275 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004276 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4277 SawLastInitializer
4278 = NumInitializers > 0 &&
4279 Initializers[NumInitializers - 1]->isBaseInitializer() &&
4280 Context.hasSameUnqualifiedType(Base->getType(),
4281 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004282 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004283 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004284
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004285 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004286 Builder.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004287 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004288 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4289 Builder.AddPlaceholderChunk("args");
4290 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4291 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004292 SawLastInitializer? CCP_NextInitializer
4293 : CCP_MemberDeclaration));
4294 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004295 }
4296
4297 // Add completions for members.
4298 for (CXXRecordDecl::field_iterator Field = ClassDecl->field_begin(),
4299 FieldEnd = ClassDecl->field_end();
4300 Field != FieldEnd; ++Field) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004301 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4302 SawLastInitializer
4303 = NumInitializers > 0 &&
Francois Pichetd583da02010-12-04 09:14:42 +00004304 Initializers[NumInitializers - 1]->isAnyMemberInitializer() &&
David Blaikie40ed2972012-06-06 20:45:41 +00004305 Initializers[NumInitializers - 1]->getAnyMember() == *Field;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004306 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004307 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004308
4309 if (!Field->getDeclName())
4310 continue;
4311
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004312 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004313 Field->getIdentifier()->getName()));
4314 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4315 Builder.AddPlaceholderChunk("args");
4316 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4317 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004318 SawLastInitializer? CCP_NextInitializer
Douglas Gregorf3af3112010-09-09 21:42:20 +00004319 : CCP_MemberDeclaration,
Douglas Gregor78254c82012-03-27 23:34:16 +00004320 CXCursor_MemberRef,
4321 CXAvailability_Available,
David Blaikie40ed2972012-06-06 20:45:41 +00004322 *Field));
Douglas Gregor99129ef2010-08-29 19:27:27 +00004323 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004324 }
4325 Results.ExitScope();
4326
Douglas Gregor0ac41382010-09-23 23:01:17 +00004327 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004328 Results.data(), Results.size());
4329}
4330
Douglas Gregord8c61782012-02-15 15:34:24 +00004331/// \brief Determine whether this scope denotes a namespace.
4332static bool isNamespaceScope(Scope *S) {
4333 DeclContext *DC = static_cast<DeclContext *>(S->getEntity());
4334 if (!DC)
4335 return false;
4336
4337 return DC->isFileContext();
4338}
4339
4340void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4341 bool AfterAmpersand) {
4342 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004343 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord8c61782012-02-15 15:34:24 +00004344 CodeCompletionContext::CCC_Other);
4345 Results.EnterNewScope();
4346
4347 // Note what has already been captured.
4348 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4349 bool IncludedThis = false;
4350 for (SmallVectorImpl<LambdaCapture>::iterator C = Intro.Captures.begin(),
4351 CEnd = Intro.Captures.end();
4352 C != CEnd; ++C) {
4353 if (C->Kind == LCK_This) {
4354 IncludedThis = true;
4355 continue;
4356 }
4357
4358 Known.insert(C->Id);
4359 }
4360
4361 // Look for other capturable variables.
4362 for (; S && !isNamespaceScope(S); S = S->getParent()) {
4363 for (Scope::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
4364 D != DEnd; ++D) {
4365 VarDecl *Var = dyn_cast<VarDecl>(*D);
4366 if (!Var ||
4367 !Var->hasLocalStorage() ||
4368 Var->hasAttr<BlocksAttr>())
4369 continue;
4370
4371 if (Known.insert(Var->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004372 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
4373 CurContext, 0, false);
Douglas Gregord8c61782012-02-15 15:34:24 +00004374 }
4375 }
4376
4377 // Add 'this', if it would be valid.
4378 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4379 addThisCompletion(*this, Results);
4380
4381 Results.ExitScope();
4382
4383 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4384 Results.data(), Results.size());
4385}
4386
James Dennett596e4752012-06-14 03:11:41 +00004387/// Macro that optionally prepends an "@" to the string literal passed in via
4388/// Keyword, depending on whether NeedAt is true or false.
4389#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4390
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004391static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004392 ResultBuilder &Results,
4393 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004394 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004395 // Since we have an implementation, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004396 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004397
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004398 CodeCompletionBuilder Builder(Results.getAllocator(),
4399 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004400 if (LangOpts.ObjC2) {
4401 // @dynamic
James Dennett596e4752012-06-14 03:11:41 +00004402 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004403 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4404 Builder.AddPlaceholderChunk("property");
4405 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004406
4407 // @synthesize
James Dennett596e4752012-06-14 03:11:41 +00004408 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004409 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4410 Builder.AddPlaceholderChunk("property");
4411 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004412 }
4413}
4414
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004415static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004416 ResultBuilder &Results,
4417 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004418 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004419
4420 // Since we have an interface or protocol, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004421 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004422
4423 if (LangOpts.ObjC2) {
4424 // @property
James Dennett596e4752012-06-14 03:11:41 +00004425 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004426
4427 // @required
James Dennett596e4752012-06-14 03:11:41 +00004428 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004429
4430 // @optional
James Dennett596e4752012-06-14 03:11:41 +00004431 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004432 }
4433}
4434
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004435static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004436 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004437 CodeCompletionBuilder Builder(Results.getAllocator(),
4438 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004439
4440 // @class name ;
James Dennett596e4752012-06-14 03:11:41 +00004441 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004442 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4443 Builder.AddPlaceholderChunk("name");
4444 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004445
Douglas Gregorf4c33342010-05-28 00:22:41 +00004446 if (Results.includeCodePatterns()) {
4447 // @interface name
4448 // FIXME: Could introduce the whole pattern, including superclasses and
4449 // such.
James Dennett596e4752012-06-14 03:11:41 +00004450 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004451 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4452 Builder.AddPlaceholderChunk("class");
4453 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004454
Douglas Gregorf4c33342010-05-28 00:22:41 +00004455 // @protocol name
James Dennett596e4752012-06-14 03:11:41 +00004456 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004457 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4458 Builder.AddPlaceholderChunk("protocol");
4459 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004460
4461 // @implementation name
James Dennett596e4752012-06-14 03:11:41 +00004462 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004463 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4464 Builder.AddPlaceholderChunk("class");
4465 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004466 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004467
4468 // @compatibility_alias name
James Dennett596e4752012-06-14 03:11:41 +00004469 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004470 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4471 Builder.AddPlaceholderChunk("alias");
4472 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4473 Builder.AddPlaceholderChunk("class");
4474 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004475}
4476
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004477void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004478 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004479 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004480 CodeCompletionContext::CCC_Other);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004481 Results.EnterNewScope();
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004482 if (isa<ObjCImplDecl>(CurContext))
David Blaikiebbafb8a2012-03-11 07:00:24 +00004483 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004484 else if (CurContext->isObjCContainer())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004485 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorf1934162010-01-13 21:24:21 +00004486 else
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004487 AddObjCTopLevelResults(Results, false);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004488 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004489 HandleCodeCompleteResults(this, CodeCompleter,
4490 CodeCompletionContext::CCC_Other,
4491 Results.data(),Results.size());
Douglas Gregorf48706c2009-12-07 09:27:33 +00004492}
4493
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004494static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004495 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004496 CodeCompletionBuilder Builder(Results.getAllocator(),
4497 Results.getCodeCompletionTUInfo());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004498
4499 // @encode ( type-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004500 const char *EncodeType = "char[]";
David Blaikiebbafb8a2012-03-11 07:00:24 +00004501 if (Results.getSema().getLangOpts().CPlusPlus ||
4502 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose9da05852012-06-15 18:19:56 +00004503 EncodeType = "const char[]";
Douglas Gregore5c79d52011-10-18 21:20:17 +00004504 Builder.AddResultTypeChunk(EncodeType);
James Dennett596e4752012-06-14 03:11:41 +00004505 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004506 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4507 Builder.AddPlaceholderChunk("type-name");
4508 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4509 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004510
4511 // @protocol ( protocol-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004512 Builder.AddResultTypeChunk("Protocol *");
James Dennett596e4752012-06-14 03:11:41 +00004513 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004514 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4515 Builder.AddPlaceholderChunk("protocol-name");
4516 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4517 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004518
4519 // @selector ( selector )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004520 Builder.AddResultTypeChunk("SEL");
James Dennett596e4752012-06-14 03:11:41 +00004521 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004522 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4523 Builder.AddPlaceholderChunk("selector");
4524 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4525 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004526
4527 // @"string"
4528 Builder.AddResultTypeChunk("NSString *");
4529 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4530 Builder.AddPlaceholderChunk("string");
4531 Builder.AddTextChunk("\"");
4532 Results.AddResult(Result(Builder.TakeString()));
4533
Douglas Gregor951de302012-07-17 23:24:47 +00004534 // @[objects, ...]
Jordan Rose9da05852012-06-15 18:19:56 +00004535 Builder.AddResultTypeChunk("NSArray *");
James Dennett596e4752012-06-14 03:11:41 +00004536 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004537 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004538 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4539 Results.AddResult(Result(Builder.TakeString()));
4540
Douglas Gregor951de302012-07-17 23:24:47 +00004541 // @{key : object, ...}
Jordan Rose9da05852012-06-15 18:19:56 +00004542 Builder.AddResultTypeChunk("NSDictionary *");
James Dennett596e4752012-06-14 03:11:41 +00004543 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"{"));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004544 Builder.AddPlaceholderChunk("key");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004545 Builder.AddChunk(CodeCompletionString::CK_Colon);
4546 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4547 Builder.AddPlaceholderChunk("object, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004548 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4549 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004550
Douglas Gregor951de302012-07-17 23:24:47 +00004551 // @(expression)
Jordan Rose9da05852012-06-15 18:19:56 +00004552 Builder.AddResultTypeChunk("id");
4553 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose9da05852012-06-15 18:19:56 +00004554 Builder.AddPlaceholderChunk("expression");
Jordan Rose9da05852012-06-15 18:19:56 +00004555 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4556 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004557}
4558
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004559static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004560 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004561 CodeCompletionBuilder Builder(Results.getAllocator(),
4562 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004563
Douglas Gregorf4c33342010-05-28 00:22:41 +00004564 if (Results.includeCodePatterns()) {
4565 // @try { statements } @catch ( declaration ) { statements } @finally
4566 // { statements }
James Dennett596e4752012-06-14 03:11:41 +00004567 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004568 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4569 Builder.AddPlaceholderChunk("statements");
4570 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4571 Builder.AddTextChunk("@catch");
4572 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4573 Builder.AddPlaceholderChunk("parameter");
4574 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4575 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4576 Builder.AddPlaceholderChunk("statements");
4577 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4578 Builder.AddTextChunk("@finally");
4579 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4580 Builder.AddPlaceholderChunk("statements");
4581 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4582 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004583 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004584
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004585 // @throw
James Dennett596e4752012-06-14 03:11:41 +00004586 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004587 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4588 Builder.AddPlaceholderChunk("expression");
4589 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004590
Douglas Gregorf4c33342010-05-28 00:22:41 +00004591 if (Results.includeCodePatterns()) {
4592 // @synchronized ( expression ) { statements }
James Dennett596e4752012-06-14 03:11:41 +00004593 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004594 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4595 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4596 Builder.AddPlaceholderChunk("expression");
4597 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4598 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4599 Builder.AddPlaceholderChunk("statements");
4600 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4601 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004602 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004603}
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004604
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004605static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00004606 ResultBuilder &Results,
4607 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004608 typedef CodeCompletionResult Result;
James Dennett596e4752012-06-14 03:11:41 +00004609 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4610 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4611 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004612 if (LangOpts.ObjC2)
James Dennett596e4752012-06-14 03:11:41 +00004613 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004614}
4615
4616void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004617 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004618 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004619 CodeCompletionContext::CCC_Other);
Douglas Gregor48d46252010-01-13 21:54:15 +00004620 Results.EnterNewScope();
David Blaikiebbafb8a2012-03-11 07:00:24 +00004621 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregor48d46252010-01-13 21:54:15 +00004622 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004623 HandleCodeCompleteResults(this, CodeCompleter,
4624 CodeCompletionContext::CCC_Other,
4625 Results.data(),Results.size());
Douglas Gregor48d46252010-01-13 21:54:15 +00004626}
4627
4628void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004629 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004630 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004631 CodeCompletionContext::CCC_Other);
Douglas Gregorf1934162010-01-13 21:24:21 +00004632 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004633 AddObjCStatementResults(Results, false);
4634 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004635 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004636 HandleCodeCompleteResults(this, CodeCompleter,
4637 CodeCompletionContext::CCC_Other,
4638 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004639}
4640
4641void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004642 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004643 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004644 CodeCompletionContext::CCC_Other);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004645 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004646 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004647 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004648 HandleCodeCompleteResults(this, CodeCompleter,
4649 CodeCompletionContext::CCC_Other,
4650 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004651}
4652
Douglas Gregore6078da2009-11-19 00:14:45 +00004653/// \brief Determine whether the addition of the given flag to an Objective-C
4654/// property's attributes will cause a conflict.
Bill Wendling44426052012-12-20 19:22:21 +00004655static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregore6078da2009-11-19 00:14:45 +00004656 // Check if we've already added this flag.
Bill Wendling44426052012-12-20 19:22:21 +00004657 if (Attributes & NewFlag)
Douglas Gregore6078da2009-11-19 00:14:45 +00004658 return true;
4659
Bill Wendling44426052012-12-20 19:22:21 +00004660 Attributes |= NewFlag;
Douglas Gregore6078da2009-11-19 00:14:45 +00004661
4662 // Check for collisions with "readonly".
Bill Wendling44426052012-12-20 19:22:21 +00004663 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4664 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregore6078da2009-11-19 00:14:45 +00004665 return true;
4666
Jordan Rose53cb2f32012-08-20 20:01:13 +00004667 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendling44426052012-12-20 19:22:21 +00004668 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCall31168b02011-06-15 23:02:42 +00004669 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregore6078da2009-11-19 00:14:45 +00004670 ObjCDeclSpec::DQ_PR_copy |
Jordan Rose53cb2f32012-08-20 20:01:13 +00004671 ObjCDeclSpec::DQ_PR_retain |
4672 ObjCDeclSpec::DQ_PR_strong |
4673 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregore6078da2009-11-19 00:14:45 +00004674 if (AssignCopyRetMask &&
4675 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCall31168b02011-06-15 23:02:42 +00004676 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregore6078da2009-11-19 00:14:45 +00004677 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCall31168b02011-06-15 23:02:42 +00004678 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rose53cb2f32012-08-20 20:01:13 +00004679 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4680 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregore6078da2009-11-19 00:14:45 +00004681 return true;
4682
4683 return false;
4684}
4685
Douglas Gregor36029f42009-11-18 23:08:07 +00004686void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroff936354c2009-10-08 21:55:05 +00004687 if (!CodeCompleter)
4688 return;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004689
Bill Wendling44426052012-12-20 19:22:21 +00004690 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroff936354c2009-10-08 21:55:05 +00004691
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004692 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004693 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004694 CodeCompletionContext::CCC_Other);
Steve Naroff936354c2009-10-08 21:55:05 +00004695 Results.EnterNewScope();
Bill Wendling44426052012-12-20 19:22:21 +00004696 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall276321a2010-08-25 06:19:51 +00004697 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendling44426052012-12-20 19:22:21 +00004698 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall276321a2010-08-25 06:19:51 +00004699 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendling44426052012-12-20 19:22:21 +00004700 if (!ObjCPropertyFlagConflicts(Attributes,
John McCall31168b02011-06-15 23:02:42 +00004701 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4702 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendling44426052012-12-20 19:22:21 +00004703 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall276321a2010-08-25 06:19:51 +00004704 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendling44426052012-12-20 19:22:21 +00004705 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall276321a2010-08-25 06:19:51 +00004706 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendling44426052012-12-20 19:22:21 +00004707 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCall31168b02011-06-15 23:02:42 +00004708 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendling44426052012-12-20 19:22:21 +00004709 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall276321a2010-08-25 06:19:51 +00004710 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendling44426052012-12-20 19:22:21 +00004711 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall276321a2010-08-25 06:19:51 +00004712 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendling44426052012-12-20 19:22:21 +00004713 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian1c2d29e2011-06-11 17:14:27 +00004714 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rose53cb2f32012-08-20 20:01:13 +00004715
4716 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall3deb1ad2012-08-21 02:47:43 +00004717 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendling44426052012-12-20 19:22:21 +00004718 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rose53cb2f32012-08-20 20:01:13 +00004719 Results.AddResult(CodeCompletionResult("weak"));
4720
Bill Wendling44426052012-12-20 19:22:21 +00004721 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004722 CodeCompletionBuilder Setter(Results.getAllocator(),
4723 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004724 Setter.AddTypedTextChunk("setter");
4725 Setter.AddTextChunk(" = ");
4726 Setter.AddPlaceholderChunk("method");
4727 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004728 }
Bill Wendling44426052012-12-20 19:22:21 +00004729 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004730 CodeCompletionBuilder Getter(Results.getAllocator(),
4731 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004732 Getter.AddTypedTextChunk("getter");
4733 Getter.AddTextChunk(" = ");
4734 Getter.AddPlaceholderChunk("method");
4735 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004736 }
Steve Naroff936354c2009-10-08 21:55:05 +00004737 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004738 HandleCodeCompleteResults(this, CodeCompleter,
4739 CodeCompletionContext::CCC_Other,
4740 Results.data(),Results.size());
Steve Naroff936354c2009-10-08 21:55:05 +00004741}
Steve Naroffeae65032009-11-07 02:08:14 +00004742
James Dennettf1243872012-06-17 05:33:25 +00004743/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregorc8537c52009-11-19 07:41:15 +00004744/// via code completion.
4745enum ObjCMethodKind {
Dmitri Gribenko4280e5c2012-06-08 23:13:42 +00004746 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4747 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4748 MK_OneArgSelector ///< One-argument selector.
Douglas Gregorc8537c52009-11-19 07:41:15 +00004749};
4750
Douglas Gregor67c692c2010-08-26 15:07:07 +00004751static bool isAcceptableObjCSelector(Selector Sel,
4752 ObjCMethodKind WantKind,
4753 IdentifierInfo **SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004754 unsigned NumSelIdents,
4755 bool AllowSameLength = true) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00004756 if (NumSelIdents > Sel.getNumArgs())
4757 return false;
4758
4759 switch (WantKind) {
4760 case MK_Any: break;
4761 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4762 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4763 }
4764
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004765 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4766 return false;
4767
Douglas Gregor67c692c2010-08-26 15:07:07 +00004768 for (unsigned I = 0; I != NumSelIdents; ++I)
4769 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4770 return false;
4771
4772 return true;
4773}
4774
Douglas Gregorc8537c52009-11-19 07:41:15 +00004775static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4776 ObjCMethodKind WantKind,
4777 IdentifierInfo **SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004778 unsigned NumSelIdents,
4779 bool AllowSameLength = true) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00004780 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004781 NumSelIdents, AllowSameLength);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004782}
Douglas Gregor1154e272010-09-16 16:06:31 +00004783
4784namespace {
4785 /// \brief A set of selectors, which is used to avoid introducing multiple
4786 /// completions with the same selector into the result set.
4787 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4788}
4789
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004790/// \brief Add all of the Objective-C methods in the given Objective-C
4791/// container to the set of results.
4792///
4793/// The container will be a class, protocol, category, or implementation of
4794/// any of the above. This mether will recurse to include methods from
4795/// the superclasses of classes along with their categories, protocols, and
4796/// implementations.
4797///
4798/// \param Container the container in which we'll look to find methods.
4799///
James Dennett596e4752012-06-14 03:11:41 +00004800/// \param WantInstanceMethods Whether to add instance methods (only); if
4801/// false, this routine will add factory methods (only).
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004802///
4803/// \param CurContext the context in which we're performing the lookup that
4804/// finds methods.
4805///
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004806/// \param AllowSameLength Whether we allow a method to be added to the list
4807/// when it has the same number of parameters as we have selector identifiers.
4808///
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004809/// \param Results the structure into which we'll add results.
4810static void AddObjCMethods(ObjCContainerDecl *Container,
4811 bool WantInstanceMethods,
Douglas Gregorc8537c52009-11-19 07:41:15 +00004812 ObjCMethodKind WantKind,
Douglas Gregor1b605f72009-11-19 01:08:35 +00004813 IdentifierInfo **SelIdents,
4814 unsigned NumSelIdents,
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004815 DeclContext *CurContext,
Douglas Gregor1154e272010-09-16 16:06:31 +00004816 VisitedSelectorSet &Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004817 bool AllowSameLength,
Douglas Gregor416b5752010-08-25 01:08:01 +00004818 ResultBuilder &Results,
4819 bool InOriginalClass = true) {
John McCall276321a2010-08-25 06:19:51 +00004820 typedef CodeCompletionResult Result;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00004821 Container = getContainerDef(Container);
Douglas Gregor41778c32013-01-30 06:58:39 +00004822 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4823 bool isRootClass = IFace && !IFace->getSuperClass();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004824 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
4825 MEnd = Container->meth_end();
4826 M != MEnd; ++M) {
Douglas Gregor41778c32013-01-30 06:58:39 +00004827 // The instance methods on the root class can be messaged via the
4828 // metaclass.
4829 if (M->isInstanceMethod() == WantInstanceMethods ||
4830 (isRootClass && !WantInstanceMethods)) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00004831 // Check whether the selector identifiers we've been given are a
4832 // subset of the identifiers for this particular method.
David Blaikie40ed2972012-06-06 20:45:41 +00004833 if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, NumSelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004834 AllowSameLength))
Douglas Gregor1b605f72009-11-19 01:08:35 +00004835 continue;
Douglas Gregorc8537c52009-11-19 07:41:15 +00004836
David Blaikie2d7c57e2012-04-30 02:36:29 +00004837 if (!Selectors.insert(M->getSelector()))
Douglas Gregor1154e272010-09-16 16:06:31 +00004838 continue;
4839
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004840 Result R = Result(*M, Results.getBasePriority(*M), 0);
Douglas Gregor1b605f72009-11-19 01:08:35 +00004841 R.StartParameter = NumSelIdents;
Douglas Gregorc8537c52009-11-19 07:41:15 +00004842 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor416b5752010-08-25 01:08:01 +00004843 if (!InOriginalClass)
4844 R.Priority += CCD_InBaseClass;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004845 Results.MaybeAddResult(R, CurContext);
4846 }
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004847 }
4848
Douglas Gregorf37c9492010-09-16 15:34:59 +00004849 // Visit the protocols of protocols.
4850 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregore6e48b12012-01-01 19:29:29 +00004851 if (Protocol->hasDefinition()) {
4852 const ObjCList<ObjCProtocolDecl> &Protocols
4853 = Protocol->getReferencedProtocols();
4854 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4855 E = Protocols.end();
4856 I != E; ++I)
4857 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
4858 NumSelIdents, CurContext, Selectors, AllowSameLength,
4859 Results, false);
4860 }
Douglas Gregorf37c9492010-09-16 15:34:59 +00004861 }
4862
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00004863 if (!IFace || !IFace->hasDefinition())
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004864 return;
4865
4866 // Add methods in protocols.
Argyrios Kyrtzidise7f3ef32012-03-13 01:09:41 +00004867 for (ObjCInterfaceDecl::protocol_iterator I = IFace->protocol_begin(),
4868 E = IFace->protocol_end();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004869 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00004870 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents, NumSelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004871 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004872
4873 // Add methods in categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00004874 for (ObjCInterfaceDecl::known_categories_iterator
4875 Cat = IFace->known_categories_begin(),
4876 CatEnd = IFace->known_categories_end();
4877 Cat != CatEnd; ++Cat) {
4878 ObjCCategoryDecl *CatDecl = *Cat;
4879
4880 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004881 NumSelIdents, CurContext, Selectors, AllowSameLength,
4882 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004883
4884 // Add a categories protocol methods.
4885 const ObjCList<ObjCProtocolDecl> &Protocols
4886 = CatDecl->getReferencedProtocols();
4887 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4888 E = Protocols.end();
4889 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00004890 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004891 NumSelIdents, CurContext, Selectors, AllowSameLength,
4892 Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004893
4894 // Add methods in category implementations.
4895 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004896 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004897 NumSelIdents, CurContext, Selectors, AllowSameLength,
4898 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004899 }
4900
4901 // Add methods in superclass.
4902 if (IFace->getSuperClass())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004903 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004904 SelIdents, NumSelIdents, CurContext, Selectors,
4905 AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004906
4907 // Add methods in our implementation, if any.
4908 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004909 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004910 NumSelIdents, CurContext, Selectors, AllowSameLength,
4911 Results, InOriginalClass);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004912}
4913
4914
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004915void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004916 // Try to find the interface where getters might live.
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004917 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004918 if (!Class) {
4919 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004920 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004921 Class = Category->getClassInterface();
4922
4923 if (!Class)
4924 return;
4925 }
4926
4927 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004928 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004929 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004930 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004931 Results.EnterNewScope();
4932
Douglas Gregor1154e272010-09-16 16:06:31 +00004933 VisitedSelectorSet Selectors;
4934 AddObjCMethods(Class, true, MK_ZeroArgSelector, 0, 0, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004935 /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004936 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004937 HandleCodeCompleteResults(this, CodeCompleter,
4938 CodeCompletionContext::CCC_Other,
4939 Results.data(),Results.size());
Douglas Gregorc8537c52009-11-19 07:41:15 +00004940}
4941
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004942void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004943 // Try to find the interface where setters might live.
4944 ObjCInterfaceDecl *Class
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004945 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004946 if (!Class) {
4947 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004948 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004949 Class = Category->getClassInterface();
4950
4951 if (!Class)
4952 return;
4953 }
4954
4955 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004956 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004957 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004958 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004959 Results.EnterNewScope();
4960
Douglas Gregor1154e272010-09-16 16:06:31 +00004961 VisitedSelectorSet Selectors;
4962 AddObjCMethods(Class, true, MK_OneArgSelector, 0, 0, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004963 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004964
4965 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004966 HandleCodeCompleteResults(this, CodeCompleter,
4967 CodeCompletionContext::CCC_Other,
4968 Results.data(),Results.size());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004969}
4970
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004971void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
4972 bool IsParameter) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004973 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004974 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004975 CodeCompletionContext::CCC_Type);
Douglas Gregor99fa2642010-08-24 01:06:58 +00004976 Results.EnterNewScope();
4977
4978 // Add context-sensitive, Objective-C parameter-passing keywords.
4979 bool AddedInOut = false;
4980 if ((DS.getObjCDeclQualifier() &
4981 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
4982 Results.AddResult("in");
4983 Results.AddResult("inout");
4984 AddedInOut = true;
4985 }
4986 if ((DS.getObjCDeclQualifier() &
4987 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
4988 Results.AddResult("out");
4989 if (!AddedInOut)
4990 Results.AddResult("inout");
4991 }
4992 if ((DS.getObjCDeclQualifier() &
4993 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
4994 ObjCDeclSpec::DQ_Oneway)) == 0) {
4995 Results.AddResult("bycopy");
4996 Results.AddResult("byref");
4997 Results.AddResult("oneway");
4998 }
4999
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005000 // If we're completing the return type of an Objective-C method and the
5001 // identifier IBAction refers to a macro, provide a completion item for
5002 // an action, e.g.,
5003 // IBAction)<#selector#>:(id)sender
5004 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
5005 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005006 CodeCompletionBuilder Builder(Results.getAllocator(),
5007 Results.getCodeCompletionTUInfo(),
5008 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005009 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005010 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005011 Builder.AddPlaceholderChunk("selector");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005012 Builder.AddChunk(CodeCompletionString::CK_Colon);
5013 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005014 Builder.AddTextChunk("id");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005015 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005016 Builder.AddTextChunk("sender");
5017 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
5018 }
Douglas Gregored1f5972013-01-30 07:11:43 +00005019
5020 // If we're completing the return type, provide 'instancetype'.
5021 if (!IsParameter) {
5022 Results.AddResult(CodeCompletionResult("instancetype"));
5023 }
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005024
Douglas Gregor99fa2642010-08-24 01:06:58 +00005025 // Add various builtin type names and specifiers.
5026 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5027 Results.ExitScope();
5028
5029 // Add the various type names
5030 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5031 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5032 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5033 CodeCompleter->includeGlobals());
5034
5035 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005036 AddMacroResults(PP, Results, false);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005037
5038 HandleCodeCompleteResults(this, CodeCompleter,
5039 CodeCompletionContext::CCC_Type,
5040 Results.data(), Results.size());
5041}
5042
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005043/// \brief When we have an expression with type "id", we may assume
5044/// that it has some more-specific class type based on knowledge of
5045/// common uses of Objective-C. This routine returns that class type,
5046/// or NULL if no better result could be determined.
5047static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregored0b69d2010-09-15 16:23:04 +00005048 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005049 if (!Msg)
5050 return 0;
5051
5052 Selector Sel = Msg->getSelector();
5053 if (Sel.isNull())
5054 return 0;
5055
5056 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5057 if (!Id)
5058 return 0;
5059
5060 ObjCMethodDecl *Method = Msg->getMethodDecl();
5061 if (!Method)
5062 return 0;
5063
5064 // Determine the class that we're sending the message to.
Douglas Gregor9a129192010-04-21 00:45:42 +00005065 ObjCInterfaceDecl *IFace = 0;
5066 switch (Msg->getReceiverKind()) {
5067 case ObjCMessageExpr::Class:
John McCall8b07ec22010-05-15 11:32:37 +00005068 if (const ObjCObjectType *ObjType
5069 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5070 IFace = ObjType->getInterface();
Douglas Gregor9a129192010-04-21 00:45:42 +00005071 break;
5072
5073 case ObjCMessageExpr::Instance: {
5074 QualType T = Msg->getInstanceReceiver()->getType();
5075 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5076 IFace = Ptr->getInterfaceDecl();
5077 break;
5078 }
5079
5080 case ObjCMessageExpr::SuperInstance:
5081 case ObjCMessageExpr::SuperClass:
5082 break;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005083 }
5084
5085 if (!IFace)
5086 return 0;
5087
5088 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5089 if (Method->isInstanceMethod())
5090 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5091 .Case("retain", IFace)
John McCall31168b02011-06-15 23:02:42 +00005092 .Case("strong", IFace)
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005093 .Case("autorelease", IFace)
5094 .Case("copy", IFace)
5095 .Case("copyWithZone", IFace)
5096 .Case("mutableCopy", IFace)
5097 .Case("mutableCopyWithZone", IFace)
5098 .Case("awakeFromCoder", IFace)
5099 .Case("replacementObjectFromCoder", IFace)
5100 .Case("class", IFace)
5101 .Case("classForCoder", IFace)
5102 .Case("superclass", Super)
5103 .Default(0);
5104
5105 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5106 .Case("new", IFace)
5107 .Case("alloc", IFace)
5108 .Case("allocWithZone", IFace)
5109 .Case("class", IFace)
5110 .Case("superclass", Super)
5111 .Default(0);
5112}
5113
Douglas Gregor6fc04132010-08-27 15:10:57 +00005114// Add a special completion for a message send to "super", which fills in the
5115// most likely case of forwarding all of our arguments to the superclass
5116// function.
5117///
5118/// \param S The semantic analysis object.
5119///
Dmitri Gribenkoadba9be2012-08-23 17:58:28 +00005120/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor6fc04132010-08-27 15:10:57 +00005121/// the "super" keyword. Otherwise, we just need to provide the arguments.
5122///
5123/// \param SelIdents The identifiers in the selector that have already been
5124/// provided as arguments for a send to "super".
5125///
5126/// \param NumSelIdents The number of identifiers in \p SelIdents.
5127///
5128/// \param Results The set of results to augment.
5129///
5130/// \returns the Objective-C method declaration that would be invoked by
5131/// this "super" completion. If NULL, no completion was added.
5132static ObjCMethodDecl *AddSuperSendCompletion(Sema &S, bool NeedSuperKeyword,
5133 IdentifierInfo **SelIdents,
5134 unsigned NumSelIdents,
5135 ResultBuilder &Results) {
5136 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5137 if (!CurMethod)
5138 return 0;
5139
5140 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5141 if (!Class)
5142 return 0;
5143
5144 // Try to find a superclass method with the same selector.
5145 ObjCMethodDecl *SuperMethod = 0;
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005146 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5147 // Check in the class
Douglas Gregor6fc04132010-08-27 15:10:57 +00005148 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5149 CurMethod->isInstanceMethod());
5150
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005151 // Check in categories or class extensions.
5152 if (!SuperMethod) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005153 for (ObjCInterfaceDecl::known_categories_iterator
5154 Cat = Class->known_categories_begin(),
5155 CatEnd = Class->known_categories_end();
5156 Cat != CatEnd; ++Cat) {
5157 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005158 CurMethod->isInstanceMethod())))
5159 break;
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005160 }
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005161 }
5162 }
5163
Douglas Gregor6fc04132010-08-27 15:10:57 +00005164 if (!SuperMethod)
5165 return 0;
5166
5167 // Check whether the superclass method has the same signature.
5168 if (CurMethod->param_size() != SuperMethod->param_size() ||
5169 CurMethod->isVariadic() != SuperMethod->isVariadic())
5170 return 0;
5171
5172 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5173 CurPEnd = CurMethod->param_end(),
5174 SuperP = SuperMethod->param_begin();
5175 CurP != CurPEnd; ++CurP, ++SuperP) {
5176 // Make sure the parameter types are compatible.
5177 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5178 (*SuperP)->getType()))
5179 return 0;
5180
5181 // Make sure we have a parameter name to forward!
5182 if (!(*CurP)->getIdentifier())
5183 return 0;
5184 }
5185
5186 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005187 CodeCompletionBuilder Builder(Results.getAllocator(),
5188 Results.getCodeCompletionTUInfo());
Douglas Gregor6fc04132010-08-27 15:10:57 +00005189
5190 // Give this completion a return type.
Douglas Gregor75acd922011-09-27 23:30:47 +00005191 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5192 Builder);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005193
5194 // If we need the "super" keyword, add it (plus some spacing).
5195 if (NeedSuperKeyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005196 Builder.AddTypedTextChunk("super");
5197 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005198 }
5199
5200 Selector Sel = CurMethod->getSelector();
5201 if (Sel.isUnarySelector()) {
5202 if (NeedSuperKeyword)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005203 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005204 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005205 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005206 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005207 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005208 } else {
5209 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5210 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
5211 if (I > NumSelIdents)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005212 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005213
5214 if (I < NumSelIdents)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005215 Builder.AddInformativeChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005216 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005217 Sel.getNameForSlot(I) + ":"));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005218 else if (NeedSuperKeyword || I > NumSelIdents) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005219 Builder.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005220 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005221 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005222 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005223 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005224 } else {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005225 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005226 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005227 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005228 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005229 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005230 }
5231 }
5232 }
5233
Douglas Gregor78254c82012-03-27 23:34:16 +00005234 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5235 CCP_SuperCompletion));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005236 return SuperMethod;
5237}
5238
Douglas Gregora817a192010-05-27 23:06:34 +00005239void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall276321a2010-08-25 06:19:51 +00005240 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005241 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005242 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005243 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005244 getLangOpts().CPlusPlus11
Douglas Gregord8c61782012-02-15 15:34:24 +00005245 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5246 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregora817a192010-05-27 23:06:34 +00005247
Douglas Gregora817a192010-05-27 23:06:34 +00005248 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5249 Results.EnterNewScope();
Douglas Gregor39982192010-08-15 06:18:01 +00005250 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5251 CodeCompleter->includeGlobals());
Douglas Gregora817a192010-05-27 23:06:34 +00005252
5253 // If we are in an Objective-C method inside a class that has a superclass,
5254 // add "super" as an option.
5255 if (ObjCMethodDecl *Method = getCurMethodDecl())
5256 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor6fc04132010-08-27 15:10:57 +00005257 if (Iface->getSuperClass()) {
Douglas Gregora817a192010-05-27 23:06:34 +00005258 Results.AddResult(Result("super"));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005259
5260 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, 0, 0, Results);
5261 }
Douglas Gregora817a192010-05-27 23:06:34 +00005262
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005263 if (getLangOpts().CPlusPlus11)
Douglas Gregord8c61782012-02-15 15:34:24 +00005264 addThisCompletion(*this, Results);
5265
Douglas Gregora817a192010-05-27 23:06:34 +00005266 Results.ExitScope();
5267
5268 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005269 AddMacroResults(PP, Results, false);
Douglas Gregor50832e02010-09-20 22:39:41 +00005270 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005271 Results.data(), Results.size());
Douglas Gregora817a192010-05-27 23:06:34 +00005272
5273}
5274
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005275void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
5276 IdentifierInfo **SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005277 unsigned NumSelIdents,
5278 bool AtArgumentExpression) {
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005279 ObjCInterfaceDecl *CDecl = 0;
5280 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5281 // Figure out which interface we're in.
5282 CDecl = CurMethod->getClassInterface();
5283 if (!CDecl)
5284 return;
5285
5286 // Find the superclass of this class.
5287 CDecl = CDecl->getSuperClass();
5288 if (!CDecl)
5289 return;
5290
5291 if (CurMethod->isInstanceMethod()) {
5292 // We are inside an instance method, which means that the message
5293 // send [super ...] is actually calling an instance method on the
Douglas Gregor392a84b2010-10-13 21:24:53 +00005294 // current object.
5295 return CodeCompleteObjCInstanceMessage(S, 0,
Douglas Gregor6fc04132010-08-27 15:10:57 +00005296 SelIdents, NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005297 AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005298 CDecl);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005299 }
5300
5301 // Fall through to send to the superclass in CDecl.
5302 } else {
5303 // "super" may be the name of a type or variable. Figure out which
5304 // it is.
5305 IdentifierInfo *Super = &Context.Idents.get("super");
5306 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5307 LookupOrdinaryName);
5308 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5309 // "super" names an interface. Use it.
5310 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCall8b07ec22010-05-15 11:32:37 +00005311 if (const ObjCObjectType *Iface
5312 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5313 CDecl = Iface->getInterface();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005314 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5315 // "super" names an unresolved type; we can't be more specific.
5316 } else {
5317 // Assume that "super" names some kind of value and parse that way.
5318 CXXScopeSpec SS;
Abramo Bagnara7945c982012-01-27 09:46:47 +00005319 SourceLocation TemplateKWLoc;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005320 UnqualifiedId id;
5321 id.setIdentifier(Super, SuperLoc);
Abramo Bagnara7945c982012-01-27 09:46:47 +00005322 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5323 false, false);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005324 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005325 SelIdents, NumSelIdents,
5326 AtArgumentExpression);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005327 }
5328
5329 // Fall through
5330 }
5331
John McCallba7bf592010-08-24 05:47:05 +00005332 ParsedType Receiver;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005333 if (CDecl)
John McCallba7bf592010-08-24 05:47:05 +00005334 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005335 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005336 NumSelIdents, AtArgumentExpression,
5337 /*IsSuper=*/true);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005338}
5339
Douglas Gregor74661272010-09-21 00:03:25 +00005340/// \brief Given a set of code-completion results for the argument of a message
5341/// send, determine the preferred type (if any) for that argument expression.
5342static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5343 unsigned NumSelIdents) {
5344 typedef CodeCompletionResult Result;
5345 ASTContext &Context = Results.getSema().Context;
5346
5347 QualType PreferredType;
5348 unsigned BestPriority = CCP_Unlikely * 2;
5349 Result *ResultsData = Results.data();
5350 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5351 Result &R = ResultsData[I];
5352 if (R.Kind == Result::RK_Declaration &&
5353 isa<ObjCMethodDecl>(R.Declaration)) {
5354 if (R.Priority <= BestPriority) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00005355 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregor74661272010-09-21 00:03:25 +00005356 if (NumSelIdents <= Method->param_size()) {
5357 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
5358 ->getType();
5359 if (R.Priority < BestPriority || PreferredType.isNull()) {
5360 BestPriority = R.Priority;
5361 PreferredType = MyPreferredType;
5362 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5363 MyPreferredType)) {
5364 PreferredType = QualType();
5365 }
5366 }
5367 }
5368 }
5369 }
5370
5371 return PreferredType;
5372}
5373
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005374static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5375 ParsedType Receiver,
5376 IdentifierInfo **SelIdents,
5377 unsigned NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005378 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005379 bool IsSuper,
5380 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005381 typedef CodeCompletionResult Result;
Douglas Gregor8ce33212009-11-17 17:59:40 +00005382 ObjCInterfaceDecl *CDecl = 0;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005383
Douglas Gregor8ce33212009-11-17 17:59:40 +00005384 // If the given name refers to an interface type, retrieve the
5385 // corresponding declaration.
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005386 if (Receiver) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005387 QualType T = SemaRef.GetTypeFromParser(Receiver, 0);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005388 if (!T.isNull())
John McCall8b07ec22010-05-15 11:32:37 +00005389 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5390 CDecl = Interface->getInterface();
Douglas Gregor8ce33212009-11-17 17:59:40 +00005391 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005392
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005393 // Add all of the factory methods in this Objective-C class, its protocols,
5394 // superclasses, categories, implementation, etc.
Steve Naroffeae65032009-11-07 02:08:14 +00005395 Results.EnterNewScope();
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005396
Douglas Gregor6fc04132010-08-27 15:10:57 +00005397 // If this is a send-to-super, try to add the special "super" send
5398 // completion.
5399 if (IsSuper) {
5400 if (ObjCMethodDecl *SuperMethod
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005401 = AddSuperSendCompletion(SemaRef, false, SelIdents, NumSelIdents,
5402 Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005403 Results.Ignore(SuperMethod);
5404 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005405
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005406 // If we're inside an Objective-C method definition, prefer its selector to
5407 // others.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005408 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005409 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005410
Douglas Gregor1154e272010-09-16 16:06:31 +00005411 VisitedSelectorSet Selectors;
Douglas Gregor6285f752010-04-06 16:40:00 +00005412 if (CDecl)
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005413 AddObjCMethods(CDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005414 SemaRef.CurContext, Selectors, AtArgumentExpression,
5415 Results);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005416 else {
Douglas Gregor6285f752010-04-06 16:40:00 +00005417 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005418
Douglas Gregord720daf2010-04-06 17:30:22 +00005419 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005420 // pool from the AST file.
Axel Naumanndd433f02012-10-18 19:05:02 +00005421 if (SemaRef.getExternalSource()) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005422 for (uint32_t I = 0,
Axel Naumanndd433f02012-10-18 19:05:02 +00005423 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall75b960e2010-06-01 09:23:16 +00005424 I != N; ++I) {
Axel Naumanndd433f02012-10-18 19:05:02 +00005425 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005426 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005427 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005428
5429 SemaRef.ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005430 }
5431 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005432
5433 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5434 MEnd = SemaRef.MethodPool.end();
Sebastian Redl75d8a322010-08-02 23:18:59 +00005435 M != MEnd; ++M) {
5436 for (ObjCMethodList *MethList = &M->second.second;
5437 MethList && MethList->Method;
Douglas Gregor6285f752010-04-06 16:40:00 +00005438 MethList = MethList->Next) {
5439 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5440 NumSelIdents))
5441 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005442
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005443 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Douglas Gregor6285f752010-04-06 16:40:00 +00005444 R.StartParameter = NumSelIdents;
5445 R.AllParametersAreInformative = false;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005446 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor6285f752010-04-06 16:40:00 +00005447 }
5448 }
5449 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005450
5451 Results.ExitScope();
5452}
Douglas Gregor6285f752010-04-06 16:40:00 +00005453
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005454void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
5455 IdentifierInfo **SelIdents,
5456 unsigned NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005457 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005458 bool IsSuper) {
Douglas Gregor63745d52011-07-21 01:05:26 +00005459
5460 QualType T = this->GetTypeFromParser(Receiver);
5461
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005462 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005463 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005464 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Douglas Gregorea777402011-07-26 15:24:30 +00005465 T, SelIdents, NumSelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005466
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005467 AddClassMessageCompletions(*this, S, Receiver, SelIdents, NumSelIdents,
5468 AtArgumentExpression, IsSuper, Results);
Douglas Gregor74661272010-09-21 00:03:25 +00005469
5470 // If we're actually at the argument expression (rather than prior to the
5471 // selector), we're actually performing code completion for an expression.
5472 // Determine whether we have a single, best method. If so, we can
5473 // code-complete the expression using the corresponding parameter type as
5474 // our preferred type, improving completion results.
5475 if (AtArgumentExpression) {
5476 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Douglas Gregor63745d52011-07-21 01:05:26 +00005477 NumSelIdents);
Douglas Gregor74661272010-09-21 00:03:25 +00005478 if (PreferredType.isNull())
5479 CodeCompleteOrdinaryName(S, PCC_Expression);
5480 else
5481 CodeCompleteExpression(S, PreferredType);
5482 return;
5483 }
5484
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005485 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005486 Results.getCompletionContext(),
Douglas Gregor6fc04132010-08-27 15:10:57 +00005487 Results.data(), Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005488}
5489
Richard Trieu2bd04012011-09-09 02:00:50 +00005490void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Douglas Gregor1b605f72009-11-19 01:08:35 +00005491 IdentifierInfo **SelIdents,
Douglas Gregor6fc04132010-08-27 15:10:57 +00005492 unsigned NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005493 bool AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005494 ObjCInterfaceDecl *Super) {
John McCall276321a2010-08-25 06:19:51 +00005495 typedef CodeCompletionResult Result;
Steve Naroffeae65032009-11-07 02:08:14 +00005496
5497 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffeae65032009-11-07 02:08:14 +00005498
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005499 // If necessary, apply function/array conversion to the receiver.
5500 // C99 6.7.5.3p[7,8].
John Wiegley01296292011-04-08 18:41:53 +00005501 if (RecExpr) {
5502 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5503 if (Conv.isInvalid()) // conversion failed. bail.
5504 return;
5505 RecExpr = Conv.take();
5506 }
Douglas Gregor392a84b2010-10-13 21:24:53 +00005507 QualType ReceiverType = RecExpr? RecExpr->getType()
5508 : Super? Context.getObjCObjectPointerType(
5509 Context.getObjCInterfaceType(Super))
5510 : Context.getObjCIdType();
Steve Naroffeae65032009-11-07 02:08:14 +00005511
Douglas Gregordc520b02010-11-08 21:12:30 +00005512 // If we're messaging an expression with type "id" or "Class", check
5513 // whether we know something special about the receiver that allows
5514 // us to assume a more-specific receiver type.
5515 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType())
5516 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5517 if (ReceiverType->isObjCClassType())
5518 return CodeCompleteObjCClassMessage(S,
5519 ParsedType::make(Context.getObjCInterfaceType(IFace)),
5520 SelIdents, NumSelIdents,
5521 AtArgumentExpression, Super);
5522
5523 ReceiverType = Context.getObjCObjectPointerType(
5524 Context.getObjCInterfaceType(IFace));
5525 }
5526
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005527 // Build the set of methods we can see.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005528 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005529 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005530 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Douglas Gregorea777402011-07-26 15:24:30 +00005531 ReceiverType, SelIdents, NumSelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005532
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005533 Results.EnterNewScope();
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005534
Douglas Gregor6fc04132010-08-27 15:10:57 +00005535 // If this is a send-to-super, try to add the special "super" send
5536 // completion.
Douglas Gregor392a84b2010-10-13 21:24:53 +00005537 if (Super) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005538 if (ObjCMethodDecl *SuperMethod
5539 = AddSuperSendCompletion(*this, false, SelIdents, NumSelIdents,
5540 Results))
5541 Results.Ignore(SuperMethod);
5542 }
5543
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005544 // If we're inside an Objective-C method definition, prefer its selector to
5545 // others.
5546 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5547 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005548
Douglas Gregor1154e272010-09-16 16:06:31 +00005549 // Keep track of the selectors we've already added.
5550 VisitedSelectorSet Selectors;
5551
Douglas Gregora3329fa2009-11-18 00:06:18 +00005552 // Handle messages to Class. This really isn't a message to an instance
5553 // method, so we treat it the same way we would treat a message send to a
5554 // class method.
5555 if (ReceiverType->isObjCClassType() ||
5556 ReceiverType->isObjCQualifiedClassType()) {
5557 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5558 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Douglas Gregorc8537c52009-11-19 07:41:15 +00005559 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005560 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005561 }
5562 }
5563 // Handle messages to a qualified ID ("id<foo>").
5564 else if (const ObjCObjectPointerType *QualID
5565 = ReceiverType->getAsObjCQualifiedIdType()) {
5566 // Search protocols for instance methods.
5567 for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(),
5568 E = QualID->qual_end();
5569 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00005570 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005571 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005572 }
5573 // Handle messages to a pointer to interface type.
5574 else if (const ObjCObjectPointerType *IFacePtr
5575 = ReceiverType->getAsObjCInterfacePointerType()) {
5576 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregorc8537c52009-11-19 07:41:15 +00005577 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005578 NumSelIdents, CurContext, Selectors, AtArgumentExpression,
5579 Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005580
5581 // Search protocols for instance methods.
5582 for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(),
5583 E = IFacePtr->qual_end();
5584 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00005585 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005586 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005587 }
Douglas Gregor6285f752010-04-06 16:40:00 +00005588 // Handle messages to "id".
5589 else if (ReceiverType->isObjCIdType()) {
Douglas Gregord720daf2010-04-06 17:30:22 +00005590 // We're messaging "id", so provide all instance methods we know
5591 // about as code-completion results.
5592
5593 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005594 // pool from the AST file.
Douglas Gregord720daf2010-04-06 17:30:22 +00005595 if (ExternalSource) {
John McCall75b960e2010-06-01 09:23:16 +00005596 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5597 I != N; ++I) {
5598 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00005599 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005600 continue;
5601
Sebastian Redl75d8a322010-08-02 23:18:59 +00005602 ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005603 }
5604 }
5605
Sebastian Redl75d8a322010-08-02 23:18:59 +00005606 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5607 MEnd = MethodPool.end();
5608 M != MEnd; ++M) {
5609 for (ObjCMethodList *MethList = &M->second.first;
5610 MethList && MethList->Method;
Douglas Gregor6285f752010-04-06 16:40:00 +00005611 MethList = MethList->Next) {
5612 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5613 NumSelIdents))
5614 continue;
Douglas Gregor1154e272010-09-16 16:06:31 +00005615
5616 if (!Selectors.insert(MethList->Method->getSelector()))
5617 continue;
5618
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005619 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Douglas Gregor6285f752010-04-06 16:40:00 +00005620 R.StartParameter = NumSelIdents;
5621 R.AllParametersAreInformative = false;
5622 Results.MaybeAddResult(R, CurContext);
5623 }
5624 }
5625 }
Steve Naroffeae65032009-11-07 02:08:14 +00005626 Results.ExitScope();
Douglas Gregor74661272010-09-21 00:03:25 +00005627
5628
5629 // If we're actually at the argument expression (rather than prior to the
5630 // selector), we're actually performing code completion for an expression.
5631 // Determine whether we have a single, best method. If so, we can
5632 // code-complete the expression using the corresponding parameter type as
5633 // our preferred type, improving completion results.
5634 if (AtArgumentExpression) {
5635 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
5636 NumSelIdents);
5637 if (PreferredType.isNull())
5638 CodeCompleteOrdinaryName(S, PCC_Expression);
5639 else
5640 CodeCompleteExpression(S, PreferredType);
5641 return;
5642 }
5643
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005644 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005645 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005646 Results.data(),Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005647}
Douglas Gregorbaf69612009-11-18 04:19:12 +00005648
Douglas Gregor68762e72010-08-23 21:17:50 +00005649void Sema::CodeCompleteObjCForCollection(Scope *S,
5650 DeclGroupPtrTy IterationVar) {
5651 CodeCompleteExpressionData Data;
5652 Data.ObjCCollection = true;
5653
5654 if (IterationVar.getAsOpaquePtr()) {
5655 DeclGroupRef DG = IterationVar.getAsVal<DeclGroupRef>();
5656 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5657 if (*I)
5658 Data.IgnoreDecls.push_back(*I);
5659 }
5660 }
5661
5662 CodeCompleteExpression(S, Data);
5663}
5664
Douglas Gregor67c692c2010-08-26 15:07:07 +00005665void Sema::CodeCompleteObjCSelector(Scope *S, IdentifierInfo **SelIdents,
5666 unsigned NumSelIdents) {
5667 // If we have an external source, load the entire class method
5668 // pool from the AST file.
5669 if (ExternalSource) {
5670 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5671 I != N; ++I) {
5672 Selector Sel = ExternalSource->GetExternalSelector(I);
5673 if (Sel.isNull() || MethodPool.count(Sel))
5674 continue;
5675
5676 ReadMethodPool(Sel);
5677 }
5678 }
5679
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005680 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005681 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005682 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor67c692c2010-08-26 15:07:07 +00005683 Results.EnterNewScope();
5684 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5685 MEnd = MethodPool.end();
5686 M != MEnd; ++M) {
5687
5688 Selector Sel = M->first;
5689 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents, NumSelIdents))
5690 continue;
5691
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005692 CodeCompletionBuilder Builder(Results.getAllocator(),
5693 Results.getCodeCompletionTUInfo());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005694 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005695 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005696 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005697 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005698 continue;
5699 }
5700
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005701 std::string Accumulator;
Douglas Gregor67c692c2010-08-26 15:07:07 +00005702 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005703 if (I == NumSelIdents) {
5704 if (!Accumulator.empty()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005705 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005706 Accumulator));
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005707 Accumulator.clear();
5708 }
5709 }
5710
Benjamin Kramer632500c2011-07-26 16:59:25 +00005711 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005712 Accumulator += ':';
Douglas Gregor67c692c2010-08-26 15:07:07 +00005713 }
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005714 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005715 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005716 }
5717 Results.ExitScope();
5718
5719 HandleCodeCompleteResults(this, CodeCompleter,
5720 CodeCompletionContext::CCC_SelectorName,
5721 Results.data(), Results.size());
5722}
5723
Douglas Gregorbaf69612009-11-18 04:19:12 +00005724/// \brief Add all of the protocol declarations that we find in the given
5725/// (translation unit) context.
5726static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005727 bool OnlyForwardDeclarations,
Douglas Gregorbaf69612009-11-18 04:19:12 +00005728 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005729 typedef CodeCompletionResult Result;
Douglas Gregorbaf69612009-11-18 04:19:12 +00005730
5731 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5732 DEnd = Ctx->decls_end();
5733 D != DEnd; ++D) {
5734 // Record any protocols we find.
5735 if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*D))
Douglas Gregore6e48b12012-01-01 19:29:29 +00005736 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005737 Results.AddResult(Result(Proto, Results.getBasePriority(Proto), 0),
5738 CurContext, 0, false);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005739 }
5740}
5741
5742void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5743 unsigned NumProtocols) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005744 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005745 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005746 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005747
Douglas Gregora3b23b02010-12-09 21:44:02 +00005748 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5749 Results.EnterNewScope();
5750
5751 // Tell the result set to ignore all of the protocols we have
5752 // already seen.
5753 // FIXME: This doesn't work when caching code-completion results.
5754 for (unsigned I = 0; I != NumProtocols; ++I)
5755 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5756 Protocols[I].second))
5757 Results.Ignore(Protocol);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005758
Douglas Gregora3b23b02010-12-09 21:44:02 +00005759 // Add all protocols.
5760 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5761 Results);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005762
Douglas Gregora3b23b02010-12-09 21:44:02 +00005763 Results.ExitScope();
5764 }
5765
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005766 HandleCodeCompleteResults(this, CodeCompleter,
5767 CodeCompletionContext::CCC_ObjCProtocolName,
5768 Results.data(),Results.size());
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005769}
5770
5771void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005772 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005773 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005774 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005775
Douglas Gregora3b23b02010-12-09 21:44:02 +00005776 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5777 Results.EnterNewScope();
5778
5779 // Add all protocols.
5780 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5781 Results);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005782
Douglas Gregora3b23b02010-12-09 21:44:02 +00005783 Results.ExitScope();
5784 }
5785
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005786 HandleCodeCompleteResults(this, CodeCompleter,
5787 CodeCompletionContext::CCC_ObjCProtocolName,
5788 Results.data(),Results.size());
Douglas Gregorbaf69612009-11-18 04:19:12 +00005789}
Douglas Gregor49c22a72009-11-18 16:26:39 +00005790
5791/// \brief Add all of the Objective-C interface declarations that we find in
5792/// the given (translation unit) context.
5793static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5794 bool OnlyForwardDeclarations,
5795 bool OnlyUnimplemented,
5796 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005797 typedef CodeCompletionResult Result;
Douglas Gregor49c22a72009-11-18 16:26:39 +00005798
5799 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5800 DEnd = Ctx->decls_end();
5801 D != DEnd; ++D) {
Douglas Gregor1c283312010-08-11 12:19:30 +00005802 // Record any interfaces we find.
5803 if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*D))
Douglas Gregordc9166c2011-12-15 20:29:51 +00005804 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregor1c283312010-08-11 12:19:30 +00005805 (!OnlyUnimplemented || !Class->getImplementation()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005806 Results.AddResult(Result(Class, Results.getBasePriority(Class), 0),
5807 CurContext, 0, false);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005808 }
5809}
5810
5811void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005812 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005813 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005814 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005815 Results.EnterNewScope();
5816
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005817 if (CodeCompleter->includeGlobals()) {
5818 // Add all classes.
5819 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5820 false, Results);
5821 }
5822
Douglas Gregor49c22a72009-11-18 16:26:39 +00005823 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005824
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005825 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005826 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005827 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005828}
5829
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005830void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5831 SourceLocation ClassNameLoc) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005832 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005833 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005834 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005835 Results.EnterNewScope();
5836
5837 // Make sure that we ignore the class we're currently defining.
5838 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005839 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005840 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor49c22a72009-11-18 16:26:39 +00005841 Results.Ignore(CurClass);
5842
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005843 if (CodeCompleter->includeGlobals()) {
5844 // Add all classes.
5845 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5846 false, Results);
5847 }
5848
Douglas Gregor49c22a72009-11-18 16:26:39 +00005849 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005850
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005851 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005852 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005853 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005854}
5855
5856void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005857 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005858 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005859 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005860 Results.EnterNewScope();
5861
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005862 if (CodeCompleter->includeGlobals()) {
5863 // Add all unimplemented classes.
5864 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5865 true, Results);
5866 }
5867
Douglas Gregor49c22a72009-11-18 16:26:39 +00005868 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005869
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005870 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005871 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005872 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005873}
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005874
5875void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005876 IdentifierInfo *ClassName,
5877 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005878 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005879
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005880 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005881 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005882 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005883
5884 // Ignore any categories we find that have already been implemented by this
5885 // interface.
5886 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5887 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005888 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005889 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
5890 for (ObjCInterfaceDecl::visible_categories_iterator
5891 Cat = Class->visible_categories_begin(),
5892 CatEnd = Class->visible_categories_end();
5893 Cat != CatEnd; ++Cat) {
5894 CategoryNames.insert(Cat->getIdentifier());
5895 }
5896 }
5897
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005898 // Add all of the categories we know about.
5899 Results.EnterNewScope();
5900 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
5901 for (DeclContext::decl_iterator D = TU->decls_begin(),
5902 DEnd = TU->decls_end();
5903 D != DEnd; ++D)
5904 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(*D))
5905 if (CategoryNames.insert(Category->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005906 Results.AddResult(Result(Category, Results.getBasePriority(Category),0),
5907 CurContext, 0, false);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005908 Results.ExitScope();
5909
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005910 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005911 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005912 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005913}
5914
5915void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005916 IdentifierInfo *ClassName,
5917 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005918 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005919
5920 // Find the corresponding interface. If we couldn't find the interface, the
5921 // program itself is ill-formed. However, we'll try to be helpful still by
5922 // providing the list of all of the categories we know about.
5923 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005924 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005925 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5926 if (!Class)
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005927 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005928
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005929 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005930 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005931 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005932
5933 // Add all of the categories that have have corresponding interface
5934 // declarations in this class and any of its superclasses, except for
5935 // already-implemented categories in the class itself.
5936 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5937 Results.EnterNewScope();
5938 bool IgnoreImplemented = true;
5939 while (Class) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005940 for (ObjCInterfaceDecl::visible_categories_iterator
5941 Cat = Class->visible_categories_begin(),
5942 CatEnd = Class->visible_categories_end();
5943 Cat != CatEnd; ++Cat) {
5944 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
5945 CategoryNames.insert(Cat->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005946 Results.AddResult(Result(*Cat, Results.getBasePriority(*Cat), 0),
5947 CurContext, 0, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005948 }
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005949
5950 Class = Class->getSuperClass();
5951 IgnoreImplemented = false;
5952 }
5953 Results.ExitScope();
5954
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005955 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005956 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005957 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005958}
Douglas Gregor5d649882009-11-18 22:32:06 +00005959
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005960void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005961 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005962 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005963 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00005964
5965 // Figure out where this @synthesize lives.
5966 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005967 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00005968 if (!Container ||
5969 (!isa<ObjCImplementationDecl>(Container) &&
5970 !isa<ObjCCategoryImplDecl>(Container)))
5971 return;
5972
5973 // Ignore any properties that have already been implemented.
Douglas Gregor9b4f3702012-06-12 13:44:08 +00005974 Container = getContainerDef(Container);
5975 for (DeclContext::decl_iterator D = Container->decls_begin(),
Douglas Gregor5d649882009-11-18 22:32:06 +00005976 DEnd = Container->decls_end();
5977 D != DEnd; ++D)
5978 if (ObjCPropertyImplDecl *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(*D))
5979 Results.Ignore(PropertyImpl->getPropertyDecl());
5980
5981 // Add any properties that we find.
Douglas Gregorb888acf2010-12-09 23:01:55 +00005982 AddedPropertiesSet AddedProperties;
Douglas Gregor5d649882009-11-18 22:32:06 +00005983 Results.EnterNewScope();
5984 if (ObjCImplementationDecl *ClassImpl
5985 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor95147142011-05-05 15:50:42 +00005986 AddObjCProperties(ClassImpl->getClassInterface(), false,
5987 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00005988 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00005989 else
5990 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor95147142011-05-05 15:50:42 +00005991 false, /*AllowNullaryMethods=*/false, CurContext,
5992 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00005993 Results.ExitScope();
5994
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005995 HandleCodeCompleteResults(this, CodeCompleter,
5996 CodeCompletionContext::CCC_Other,
5997 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00005998}
5999
6000void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006001 IdentifierInfo *PropertyName) {
John McCall276321a2010-08-25 06:19:51 +00006002 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006003 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006004 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006005 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00006006
6007 // Figure out where this @synthesize lives.
6008 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006009 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00006010 if (!Container ||
6011 (!isa<ObjCImplementationDecl>(Container) &&
6012 !isa<ObjCCategoryImplDecl>(Container)))
6013 return;
6014
6015 // Figure out which interface we're looking into.
6016 ObjCInterfaceDecl *Class = 0;
6017 if (ObjCImplementationDecl *ClassImpl
6018 = dyn_cast<ObjCImplementationDecl>(Container))
6019 Class = ClassImpl->getClassInterface();
6020 else
6021 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
6022 ->getClassInterface();
6023
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006024 // Determine the type of the property we're synthesizing.
6025 QualType PropertyType = Context.getObjCIdType();
6026 if (Class) {
6027 if (ObjCPropertyDecl *Property
6028 = Class->FindPropertyDeclaration(PropertyName)) {
6029 PropertyType
6030 = Property->getType().getNonReferenceType().getUnqualifiedType();
6031
6032 // Give preference to ivars
6033 Results.setPreferredType(PropertyType);
6034 }
6035 }
6036
Douglas Gregor5d649882009-11-18 22:32:06 +00006037 // Add all of the instance variables in this class and its superclasses.
6038 Results.EnterNewScope();
Douglas Gregor331faa02011-04-18 14:13:53 +00006039 bool SawSimilarlyNamedIvar = false;
6040 std::string NameWithPrefix;
6041 NameWithPrefix += '_';
Benjamin Kramer632500c2011-07-26 16:59:25 +00006042 NameWithPrefix += PropertyName->getName();
Douglas Gregor331faa02011-04-18 14:13:53 +00006043 std::string NameWithSuffix = PropertyName->getName().str();
6044 NameWithSuffix += '_';
Douglas Gregor5d649882009-11-18 22:32:06 +00006045 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006046 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
6047 Ivar = Ivar->getNextIvar()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00006048 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), 0),
6049 CurContext, 0, false);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006050
Douglas Gregor331faa02011-04-18 14:13:53 +00006051 // Determine whether we've seen an ivar with a name similar to the
6052 // property.
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006053 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregor331faa02011-04-18 14:13:53 +00006054 NameWithPrefix == Ivar->getName() ||
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006055 NameWithSuffix == Ivar->getName())) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006056 SawSimilarlyNamedIvar = true;
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006057
6058 // Reduce the priority of this result by one, to give it a slight
6059 // advantage over other results whose names don't match so closely.
6060 if (Results.size() &&
6061 Results.data()[Results.size() - 1].Kind
6062 == CodeCompletionResult::RK_Declaration &&
6063 Results.data()[Results.size() - 1].Declaration == Ivar)
6064 Results.data()[Results.size() - 1].Priority--;
6065 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006066 }
Douglas Gregor5d649882009-11-18 22:32:06 +00006067 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006068
6069 if (!SawSimilarlyNamedIvar) {
6070 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006071 // an ivar of the appropriate type.
6072 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregor331faa02011-04-18 14:13:53 +00006073 typedef CodeCompletionResult Result;
6074 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006075 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6076 Priority,CXAvailability_Available);
Douglas Gregor331faa02011-04-18 14:13:53 +00006077
Douglas Gregor75acd922011-09-27 23:30:47 +00006078 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006079 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006080 Policy, Allocator));
Douglas Gregor331faa02011-04-18 14:13:53 +00006081 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6082 Results.AddResult(Result(Builder.TakeString(), Priority,
6083 CXCursor_ObjCIvarDecl));
6084 }
6085
Douglas Gregor5d649882009-11-18 22:32:06 +00006086 Results.ExitScope();
6087
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006088 HandleCodeCompleteResults(this, CodeCompleter,
6089 CodeCompletionContext::CCC_Other,
6090 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006091}
Douglas Gregor636a61e2010-04-07 00:21:17 +00006092
Douglas Gregor416b5752010-08-25 01:08:01 +00006093// Mapping from selectors to the methods that implement that selector, along
6094// with the "in original class" flag.
6095typedef llvm::DenseMap<Selector, std::pair<ObjCMethodDecl *, bool> >
6096 KnownMethodsMap;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006097
6098/// \brief Find all of the methods that reside in the given container
6099/// (and its superclasses, protocols, etc.) that meet the given
6100/// criteria. Insert those methods into the map of known methods,
6101/// indexed by selector so they can be easily found.
6102static void FindImplementableMethods(ASTContext &Context,
6103 ObjCContainerDecl *Container,
6104 bool WantInstanceMethods,
6105 QualType ReturnType,
Douglas Gregor416b5752010-08-25 01:08:01 +00006106 KnownMethodsMap &KnownMethods,
6107 bool InOriginalClass = true) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006108 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006109 // Make sure we have a definition; that's what we'll walk.
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006110 if (!IFace->hasDefinition())
6111 return;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006112
6113 IFace = IFace->getDefinition();
6114 Container = IFace;
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006115
Douglas Gregor636a61e2010-04-07 00:21:17 +00006116 const ObjCList<ObjCProtocolDecl> &Protocols
6117 = IFace->getReferencedProtocols();
6118 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006119 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006120 I != E; ++I)
6121 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006122 KnownMethods, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006123
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006124 // Add methods from any class extensions and categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006125 for (ObjCInterfaceDecl::visible_categories_iterator
6126 Cat = IFace->visible_categories_begin(),
6127 CatEnd = IFace->visible_categories_end();
6128 Cat != CatEnd; ++Cat) {
6129 FindImplementableMethods(Context, *Cat, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006130 KnownMethods, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006131 }
6132
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006133 // Visit the superclass.
6134 if (IFace->getSuperClass())
6135 FindImplementableMethods(Context, IFace->getSuperClass(),
6136 WantInstanceMethods, ReturnType,
6137 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006138 }
6139
6140 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6141 // Recurse into protocols.
6142 const ObjCList<ObjCProtocolDecl> &Protocols
6143 = Category->getReferencedProtocols();
6144 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006145 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006146 I != E; ++I)
6147 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006148 KnownMethods, InOriginalClass);
6149
6150 // If this category is the original class, jump to the interface.
6151 if (InOriginalClass && Category->getClassInterface())
6152 FindImplementableMethods(Context, Category->getClassInterface(),
6153 WantInstanceMethods, ReturnType, KnownMethods,
6154 false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006155 }
6156
6157 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006158 // Make sure we have a definition; that's what we'll walk.
6159 if (!Protocol->hasDefinition())
6160 return;
6161 Protocol = Protocol->getDefinition();
6162 Container = Protocol;
6163
6164 // Recurse into protocols.
6165 const ObjCList<ObjCProtocolDecl> &Protocols
6166 = Protocol->getReferencedProtocols();
6167 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6168 E = Protocols.end();
6169 I != E; ++I)
6170 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6171 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006172 }
6173
6174 // Add methods in this container. This operation occurs last because
6175 // we want the methods from this container to override any methods
6176 // we've previously seen with the same selector.
6177 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
6178 MEnd = Container->meth_end();
6179 M != MEnd; ++M) {
David Blaikie2d7c57e2012-04-30 02:36:29 +00006180 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006181 if (!ReturnType.isNull() &&
David Blaikie2d7c57e2012-04-30 02:36:29 +00006182 !Context.hasSameUnqualifiedType(ReturnType, M->getResultType()))
Douglas Gregor636a61e2010-04-07 00:21:17 +00006183 continue;
6184
David Blaikie40ed2972012-06-06 20:45:41 +00006185 KnownMethods[M->getSelector()] = std::make_pair(*M, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006186 }
6187 }
6188}
6189
Douglas Gregor669a25a2011-02-17 00:22:45 +00006190/// \brief Add the parenthesized return or parameter type chunk to a code
6191/// completion string.
6192static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor29979142012-04-10 18:35:07 +00006193 unsigned ObjCDeclQuals,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006194 ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006195 const PrintingPolicy &Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006196 CodeCompletionBuilder &Builder) {
6197 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor29979142012-04-10 18:35:07 +00006198 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6199 if (!Quals.empty())
6200 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor75acd922011-09-27 23:30:47 +00006201 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006202 Builder.getAllocator()));
6203 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6204}
6205
6206/// \brief Determine whether the given class is or inherits from a class by
6207/// the given name.
6208static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006209 StringRef Name) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006210 if (!Class)
6211 return false;
6212
6213 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6214 return true;
6215
6216 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6217}
6218
6219/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6220/// Key-Value Observing (KVO).
6221static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6222 bool IsInstanceMethod,
6223 QualType ReturnType,
6224 ASTContext &Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006225 VisitedSelectorSet &KnownSelectors,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006226 ResultBuilder &Results) {
6227 IdentifierInfo *PropName = Property->getIdentifier();
6228 if (!PropName || PropName->getLength() == 0)
6229 return;
6230
Douglas Gregor75acd922011-09-27 23:30:47 +00006231 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6232
Douglas Gregor669a25a2011-02-17 00:22:45 +00006233 // Builder that will create each code completion.
6234 typedef CodeCompletionResult Result;
6235 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006236 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor669a25a2011-02-17 00:22:45 +00006237
6238 // The selector table.
6239 SelectorTable &Selectors = Context.Selectors;
6240
6241 // The property name, copied into the code completion allocation region
6242 // on demand.
6243 struct KeyHolder {
6244 CodeCompletionAllocator &Allocator;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006245 StringRef Key;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006246 const char *CopiedKey;
6247
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006248 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006249 : Allocator(Allocator), Key(Key), CopiedKey(0) { }
6250
6251 operator const char *() {
6252 if (CopiedKey)
6253 return CopiedKey;
6254
6255 return CopiedKey = Allocator.CopyString(Key);
6256 }
6257 } Key(Allocator, PropName->getName());
6258
6259 // The uppercased name of the property name.
6260 std::string UpperKey = PropName->getName();
6261 if (!UpperKey.empty())
Jordan Rose4938f272013-02-09 10:09:43 +00006262 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006263
6264 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6265 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6266 Property->getType());
6267 bool ReturnTypeMatchesVoid
6268 = ReturnType.isNull() || ReturnType->isVoidType();
6269
6270 // Add the normal accessor -(type)key.
6271 if (IsInstanceMethod &&
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006272 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor669a25a2011-02-17 00:22:45 +00006273 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6274 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006275 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6276 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006277
6278 Builder.AddTypedTextChunk(Key);
6279 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6280 CXCursor_ObjCInstanceMethodDecl));
6281 }
6282
6283 // If we have an integral or boolean property (or the user has provided
6284 // an integral or boolean return type), add the accessor -(type)isKey.
6285 if (IsInstanceMethod &&
6286 ((!ReturnType.isNull() &&
6287 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6288 (ReturnType.isNull() &&
6289 (Property->getType()->isIntegerType() ||
6290 Property->getType()->isBooleanType())))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006291 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006292 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006293 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006294 if (ReturnType.isNull()) {
6295 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6296 Builder.AddTextChunk("BOOL");
6297 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6298 }
6299
6300 Builder.AddTypedTextChunk(
6301 Allocator.CopyString(SelectorId->getName()));
6302 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6303 CXCursor_ObjCInstanceMethodDecl));
6304 }
6305 }
6306
6307 // Add the normal mutator.
6308 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6309 !Property->getSetterMethodDecl()) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006310 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006311 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006312 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006313 if (ReturnType.isNull()) {
6314 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6315 Builder.AddTextChunk("void");
6316 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6317 }
6318
6319 Builder.AddTypedTextChunk(
6320 Allocator.CopyString(SelectorId->getName()));
6321 Builder.AddTypedTextChunk(":");
Douglas Gregor29979142012-04-10 18:35:07 +00006322 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6323 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006324 Builder.AddTextChunk(Key);
6325 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6326 CXCursor_ObjCInstanceMethodDecl));
6327 }
6328 }
6329
6330 // Indexed and unordered accessors
6331 unsigned IndexedGetterPriority = CCP_CodePattern;
6332 unsigned IndexedSetterPriority = CCP_CodePattern;
6333 unsigned UnorderedGetterPriority = CCP_CodePattern;
6334 unsigned UnorderedSetterPriority = CCP_CodePattern;
6335 if (const ObjCObjectPointerType *ObjCPointer
6336 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6337 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6338 // If this interface type is not provably derived from a known
6339 // collection, penalize the corresponding completions.
6340 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6341 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6342 if (!InheritsFromClassNamed(IFace, "NSArray"))
6343 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6344 }
6345
6346 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6347 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6348 if (!InheritsFromClassNamed(IFace, "NSSet"))
6349 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6350 }
6351 }
6352 } else {
6353 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6354 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6355 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6356 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6357 }
6358
6359 // Add -(NSUInteger)countOf<key>
6360 if (IsInstanceMethod &&
6361 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006362 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006363 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006364 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006365 if (ReturnType.isNull()) {
6366 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6367 Builder.AddTextChunk("NSUInteger");
6368 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6369 }
6370
6371 Builder.AddTypedTextChunk(
6372 Allocator.CopyString(SelectorId->getName()));
6373 Results.AddResult(Result(Builder.TakeString(),
6374 std::min(IndexedGetterPriority,
6375 UnorderedGetterPriority),
6376 CXCursor_ObjCInstanceMethodDecl));
6377 }
6378 }
6379
6380 // Indexed getters
6381 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6382 if (IsInstanceMethod &&
6383 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006384 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006385 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006386 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006387 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006388 if (ReturnType.isNull()) {
6389 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6390 Builder.AddTextChunk("id");
6391 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6392 }
6393
6394 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6395 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6396 Builder.AddTextChunk("NSUInteger");
6397 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6398 Builder.AddTextChunk("index");
6399 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6400 CXCursor_ObjCInstanceMethodDecl));
6401 }
6402 }
6403
6404 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6405 if (IsInstanceMethod &&
6406 (ReturnType.isNull() ||
6407 (ReturnType->isObjCObjectPointerType() &&
6408 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6409 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6410 ->getName() == "NSArray"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006411 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006412 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006413 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006414 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006415 if (ReturnType.isNull()) {
6416 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6417 Builder.AddTextChunk("NSArray *");
6418 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6419 }
6420
6421 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6422 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6423 Builder.AddTextChunk("NSIndexSet *");
6424 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6425 Builder.AddTextChunk("indexes");
6426 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6427 CXCursor_ObjCInstanceMethodDecl));
6428 }
6429 }
6430
6431 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6432 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006433 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006434 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006435 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006436 &Context.Idents.get("range")
6437 };
6438
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006439 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006440 if (ReturnType.isNull()) {
6441 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6442 Builder.AddTextChunk("void");
6443 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6444 }
6445
6446 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6447 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6448 Builder.AddPlaceholderChunk("object-type");
6449 Builder.AddTextChunk(" **");
6450 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6451 Builder.AddTextChunk("buffer");
6452 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6453 Builder.AddTypedTextChunk("range:");
6454 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6455 Builder.AddTextChunk("NSRange");
6456 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6457 Builder.AddTextChunk("inRange");
6458 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6459 CXCursor_ObjCInstanceMethodDecl));
6460 }
6461 }
6462
6463 // Mutable indexed accessors
6464
6465 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6466 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006467 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006468 IdentifierInfo *SelectorIds[2] = {
6469 &Context.Idents.get("insertObject"),
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006470 &Context.Idents.get(SelectorName)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006471 };
6472
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006473 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006474 if (ReturnType.isNull()) {
6475 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6476 Builder.AddTextChunk("void");
6477 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6478 }
6479
6480 Builder.AddTypedTextChunk("insertObject:");
6481 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6482 Builder.AddPlaceholderChunk("object-type");
6483 Builder.AddTextChunk(" *");
6484 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6485 Builder.AddTextChunk("object");
6486 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6487 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6488 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6489 Builder.AddPlaceholderChunk("NSUInteger");
6490 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6491 Builder.AddTextChunk("index");
6492 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6493 CXCursor_ObjCInstanceMethodDecl));
6494 }
6495 }
6496
6497 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6498 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006499 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006500 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006501 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006502 &Context.Idents.get("atIndexes")
6503 };
6504
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006505 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006506 if (ReturnType.isNull()) {
6507 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6508 Builder.AddTextChunk("void");
6509 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6510 }
6511
6512 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6513 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6514 Builder.AddTextChunk("NSArray *");
6515 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6516 Builder.AddTextChunk("array");
6517 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6518 Builder.AddTypedTextChunk("atIndexes:");
6519 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6520 Builder.AddPlaceholderChunk("NSIndexSet *");
6521 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6522 Builder.AddTextChunk("indexes");
6523 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6524 CXCursor_ObjCInstanceMethodDecl));
6525 }
6526 }
6527
6528 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6529 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006530 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006531 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006532 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006533 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006534 if (ReturnType.isNull()) {
6535 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6536 Builder.AddTextChunk("void");
6537 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6538 }
6539
6540 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6541 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6542 Builder.AddTextChunk("NSUInteger");
6543 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6544 Builder.AddTextChunk("index");
6545 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6546 CXCursor_ObjCInstanceMethodDecl));
6547 }
6548 }
6549
6550 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6551 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006552 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006553 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006554 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006555 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006556 if (ReturnType.isNull()) {
6557 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6558 Builder.AddTextChunk("void");
6559 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6560 }
6561
6562 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6563 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6564 Builder.AddTextChunk("NSIndexSet *");
6565 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6566 Builder.AddTextChunk("indexes");
6567 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6568 CXCursor_ObjCInstanceMethodDecl));
6569 }
6570 }
6571
6572 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6573 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006574 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006575 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006576 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006577 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006578 &Context.Idents.get("withObject")
6579 };
6580
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006581 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006582 if (ReturnType.isNull()) {
6583 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6584 Builder.AddTextChunk("void");
6585 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6586 }
6587
6588 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6589 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6590 Builder.AddPlaceholderChunk("NSUInteger");
6591 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6592 Builder.AddTextChunk("index");
6593 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6594 Builder.AddTypedTextChunk("withObject:");
6595 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6596 Builder.AddTextChunk("id");
6597 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6598 Builder.AddTextChunk("object");
6599 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6600 CXCursor_ObjCInstanceMethodDecl));
6601 }
6602 }
6603
6604 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6605 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006606 std::string SelectorName1
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006607 = (Twine("replace") + UpperKey + "AtIndexes").str();
6608 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006609 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006610 &Context.Idents.get(SelectorName1),
6611 &Context.Idents.get(SelectorName2)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006612 };
6613
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006614 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006615 if (ReturnType.isNull()) {
6616 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6617 Builder.AddTextChunk("void");
6618 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6619 }
6620
6621 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6622 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6623 Builder.AddPlaceholderChunk("NSIndexSet *");
6624 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6625 Builder.AddTextChunk("indexes");
6626 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6627 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6628 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6629 Builder.AddTextChunk("NSArray *");
6630 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6631 Builder.AddTextChunk("array");
6632 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6633 CXCursor_ObjCInstanceMethodDecl));
6634 }
6635 }
6636
6637 // Unordered getters
6638 // - (NSEnumerator *)enumeratorOfKey
6639 if (IsInstanceMethod &&
6640 (ReturnType.isNull() ||
6641 (ReturnType->isObjCObjectPointerType() &&
6642 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6643 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6644 ->getName() == "NSEnumerator"))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006645 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006646 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006647 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006648 if (ReturnType.isNull()) {
6649 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6650 Builder.AddTextChunk("NSEnumerator *");
6651 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6652 }
6653
6654 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6655 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6656 CXCursor_ObjCInstanceMethodDecl));
6657 }
6658 }
6659
6660 // - (type *)memberOfKey:(type *)object
6661 if (IsInstanceMethod &&
6662 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006663 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006664 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006665 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006666 if (ReturnType.isNull()) {
6667 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6668 Builder.AddPlaceholderChunk("object-type");
6669 Builder.AddTextChunk(" *");
6670 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6671 }
6672
6673 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6674 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6675 if (ReturnType.isNull()) {
6676 Builder.AddPlaceholderChunk("object-type");
6677 Builder.AddTextChunk(" *");
6678 } else {
6679 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006680 Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006681 Builder.getAllocator()));
6682 }
6683 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6684 Builder.AddTextChunk("object");
6685 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6686 CXCursor_ObjCInstanceMethodDecl));
6687 }
6688 }
6689
6690 // Mutable unordered accessors
6691 // - (void)addKeyObject:(type *)object
6692 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006693 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006694 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006695 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006696 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006697 if (ReturnType.isNull()) {
6698 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6699 Builder.AddTextChunk("void");
6700 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6701 }
6702
6703 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6704 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6705 Builder.AddPlaceholderChunk("object-type");
6706 Builder.AddTextChunk(" *");
6707 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6708 Builder.AddTextChunk("object");
6709 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6710 CXCursor_ObjCInstanceMethodDecl));
6711 }
6712 }
6713
6714 // - (void)addKey:(NSSet *)objects
6715 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006716 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006717 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006718 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006719 if (ReturnType.isNull()) {
6720 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6721 Builder.AddTextChunk("void");
6722 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6723 }
6724
6725 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6726 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6727 Builder.AddTextChunk("NSSet *");
6728 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6729 Builder.AddTextChunk("objects");
6730 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6731 CXCursor_ObjCInstanceMethodDecl));
6732 }
6733 }
6734
6735 // - (void)removeKeyObject:(type *)object
6736 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006737 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006738 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006739 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006740 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006741 if (ReturnType.isNull()) {
6742 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6743 Builder.AddTextChunk("void");
6744 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6745 }
6746
6747 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6748 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6749 Builder.AddPlaceholderChunk("object-type");
6750 Builder.AddTextChunk(" *");
6751 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6752 Builder.AddTextChunk("object");
6753 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6754 CXCursor_ObjCInstanceMethodDecl));
6755 }
6756 }
6757
6758 // - (void)removeKey:(NSSet *)objects
6759 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006760 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006761 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006762 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006763 if (ReturnType.isNull()) {
6764 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6765 Builder.AddTextChunk("void");
6766 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6767 }
6768
6769 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6770 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6771 Builder.AddTextChunk("NSSet *");
6772 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6773 Builder.AddTextChunk("objects");
6774 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6775 CXCursor_ObjCInstanceMethodDecl));
6776 }
6777 }
6778
6779 // - (void)intersectKey:(NSSet *)objects
6780 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006781 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006782 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006783 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006784 if (ReturnType.isNull()) {
6785 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6786 Builder.AddTextChunk("void");
6787 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6788 }
6789
6790 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6791 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6792 Builder.AddTextChunk("NSSet *");
6793 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6794 Builder.AddTextChunk("objects");
6795 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6796 CXCursor_ObjCInstanceMethodDecl));
6797 }
6798 }
6799
6800 // Key-Value Observing
6801 // + (NSSet *)keyPathsForValuesAffectingKey
6802 if (!IsInstanceMethod &&
6803 (ReturnType.isNull() ||
6804 (ReturnType->isObjCObjectPointerType() &&
6805 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6806 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6807 ->getName() == "NSSet"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006808 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006809 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006810 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006811 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006812 if (ReturnType.isNull()) {
6813 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6814 Builder.AddTextChunk("NSSet *");
6815 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6816 }
6817
6818 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6819 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor857bcda2011-06-02 04:02:27 +00006820 CXCursor_ObjCClassMethodDecl));
6821 }
6822 }
6823
6824 // + (BOOL)automaticallyNotifiesObserversForKey
6825 if (!IsInstanceMethod &&
6826 (ReturnType.isNull() ||
6827 ReturnType->isIntegerType() ||
6828 ReturnType->isBooleanType())) {
6829 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006830 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor857bcda2011-06-02 04:02:27 +00006831 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6832 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6833 if (ReturnType.isNull()) {
6834 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6835 Builder.AddTextChunk("BOOL");
6836 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6837 }
6838
6839 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6840 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6841 CXCursor_ObjCClassMethodDecl));
Douglas Gregor669a25a2011-02-17 00:22:45 +00006842 }
6843 }
6844}
6845
Douglas Gregor636a61e2010-04-07 00:21:17 +00006846void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6847 bool IsInstanceMethod,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006848 ParsedType ReturnTy) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006849 // Determine the return type of the method we're declaring, if
6850 // provided.
6851 QualType ReturnType = GetTypeFromParser(ReturnTy);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006852 Decl *IDecl = 0;
6853 if (CurContext->isObjCContainer()) {
6854 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6855 IDecl = cast<Decl>(OCD);
6856 }
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006857 // Determine where we should start searching for methods.
6858 ObjCContainerDecl *SearchDecl = 0;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006859 bool IsInImplementation = false;
John McCall48871652010-08-21 09:40:31 +00006860 if (Decl *D = IDecl) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006861 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6862 SearchDecl = Impl->getClassInterface();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006863 IsInImplementation = true;
6864 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006865 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006866 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006867 IsInImplementation = true;
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006868 } else
Douglas Gregor636a61e2010-04-07 00:21:17 +00006869 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006870 }
6871
6872 if (!SearchDecl && S) {
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006873 if (DeclContext *DC = static_cast<DeclContext *>(S->getEntity()))
Douglas Gregor636a61e2010-04-07 00:21:17 +00006874 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006875 }
6876
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006877 if (!SearchDecl) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006878 HandleCodeCompleteResults(this, CodeCompleter,
6879 CodeCompletionContext::CCC_Other,
6880 0, 0);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006881 return;
6882 }
6883
6884 // Find all of the methods that we could declare/implement here.
6885 KnownMethodsMap KnownMethods;
6886 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006887 ReturnType, KnownMethods);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006888
Douglas Gregor636a61e2010-04-07 00:21:17 +00006889 // Add declarations or definitions for each of the known methods.
John McCall276321a2010-08-25 06:19:51 +00006890 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006891 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006892 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006893 CodeCompletionContext::CCC_Other);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006894 Results.EnterNewScope();
Douglas Gregor75acd922011-09-27 23:30:47 +00006895 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006896 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6897 MEnd = KnownMethods.end();
6898 M != MEnd; ++M) {
Douglas Gregor416b5752010-08-25 01:08:01 +00006899 ObjCMethodDecl *Method = M->second.first;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006900 CodeCompletionBuilder Builder(Results.getAllocator(),
6901 Results.getCodeCompletionTUInfo());
Douglas Gregor636a61e2010-04-07 00:21:17 +00006902
6903 // If the result type was not already provided, add it to the
6904 // pattern as (type).
Douglas Gregor669a25a2011-02-17 00:22:45 +00006905 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006906 AddObjCPassingTypeChunk(Method->getResultType(),
6907 Method->getObjCDeclQualifier(),
6908 Context, Policy,
6909 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006910
6911 Selector Sel = Method->getSelector();
6912
6913 // Add the first part of the selector to the pattern.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006914 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006915 Sel.getNameForSlot(0)));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006916
6917 // Add parameters to the pattern.
6918 unsigned I = 0;
6919 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6920 PEnd = Method->param_end();
6921 P != PEnd; (void)++P, ++I) {
6922 // Add the part of the selector name.
6923 if (I == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006924 Builder.AddTypedTextChunk(":");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006925 else if (I < Sel.getNumArgs()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006926 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6927 Builder.AddTypedTextChunk(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006928 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006929 } else
6930 break;
6931
6932 // Add the parameter type.
Douglas Gregor29979142012-04-10 18:35:07 +00006933 AddObjCPassingTypeChunk((*P)->getOriginalType(),
6934 (*P)->getObjCDeclQualifier(),
6935 Context, Policy,
Douglas Gregor75acd922011-09-27 23:30:47 +00006936 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006937
6938 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006939 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006940 }
6941
6942 if (Method->isVariadic()) {
6943 if (Method->param_size() > 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006944 Builder.AddChunk(CodeCompletionString::CK_Comma);
6945 Builder.AddTextChunk("...");
Douglas Gregor400f5972010-08-31 05:13:43 +00006946 }
Douglas Gregor636a61e2010-04-07 00:21:17 +00006947
Douglas Gregord37c59d2010-05-28 00:57:46 +00006948 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006949 // We will be defining the method here, so add a compound statement.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006950 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6951 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6952 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006953 if (!Method->getResultType()->isVoidType()) {
6954 // If the result type is not void, add a return clause.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006955 Builder.AddTextChunk("return");
6956 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6957 Builder.AddPlaceholderChunk("expression");
6958 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006959 } else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006960 Builder.AddPlaceholderChunk("statements");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006961
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006962 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6963 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006964 }
6965
Douglas Gregor416b5752010-08-25 01:08:01 +00006966 unsigned Priority = CCP_CodePattern;
6967 if (!M->second.second)
6968 Priority += CCD_InBaseClass;
6969
Douglas Gregor78254c82012-03-27 23:34:16 +00006970 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006971 }
6972
Douglas Gregor669a25a2011-02-17 00:22:45 +00006973 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
6974 // the properties in this class and its categories.
David Blaikiebbafb8a2012-03-11 07:00:24 +00006975 if (Context.getLangOpts().ObjC2) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006976 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006977 Containers.push_back(SearchDecl);
6978
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006979 VisitedSelectorSet KnownSelectors;
6980 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6981 MEnd = KnownMethods.end();
6982 M != MEnd; ++M)
6983 KnownSelectors.insert(M->first);
6984
6985
Douglas Gregor669a25a2011-02-17 00:22:45 +00006986 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
6987 if (!IFace)
6988 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
6989 IFace = Category->getClassInterface();
6990
6991 if (IFace) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006992 for (ObjCInterfaceDecl::visible_categories_iterator
6993 Cat = IFace->visible_categories_begin(),
6994 CatEnd = IFace->visible_categories_end();
6995 Cat != CatEnd; ++Cat) {
6996 Containers.push_back(*Cat);
6997 }
Douglas Gregor669a25a2011-02-17 00:22:45 +00006998 }
6999
7000 for (unsigned I = 0, N = Containers.size(); I != N; ++I) {
7001 for (ObjCContainerDecl::prop_iterator P = Containers[I]->prop_begin(),
7002 PEnd = Containers[I]->prop_end();
7003 P != PEnd; ++P) {
David Blaikie40ed2972012-06-06 20:45:41 +00007004 AddObjCKeyValueCompletions(*P, IsInstanceMethod, ReturnType, Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00007005 KnownSelectors, Results);
Douglas Gregor669a25a2011-02-17 00:22:45 +00007006 }
7007 }
7008 }
7009
Douglas Gregor636a61e2010-04-07 00:21:17 +00007010 Results.ExitScope();
7011
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007012 HandleCodeCompleteResults(this, CodeCompleter,
7013 CodeCompletionContext::CCC_Other,
7014 Results.data(),Results.size());
Douglas Gregor636a61e2010-04-07 00:21:17 +00007015}
Douglas Gregor95887f92010-07-08 23:20:03 +00007016
7017void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
7018 bool IsInstanceMethod,
Douglas Gregor45879692010-07-08 23:37:41 +00007019 bool AtParameterName,
John McCallba7bf592010-08-24 05:47:05 +00007020 ParsedType ReturnTy,
Douglas Gregor95887f92010-07-08 23:20:03 +00007021 IdentifierInfo **SelIdents,
7022 unsigned NumSelIdents) {
Douglas Gregor95887f92010-07-08 23:20:03 +00007023 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00007024 // pool from the AST file.
Douglas Gregor95887f92010-07-08 23:20:03 +00007025 if (ExternalSource) {
7026 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
7027 I != N; ++I) {
7028 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00007029 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor95887f92010-07-08 23:20:03 +00007030 continue;
Sebastian Redl75d8a322010-08-02 23:18:59 +00007031
7032 ReadMethodPool(Sel);
Douglas Gregor95887f92010-07-08 23:20:03 +00007033 }
7034 }
7035
7036 // Build the set of methods we can see.
John McCall276321a2010-08-25 06:19:51 +00007037 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007038 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007039 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007040 CodeCompletionContext::CCC_Other);
Douglas Gregor95887f92010-07-08 23:20:03 +00007041
7042 if (ReturnTy)
7043 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redl75d8a322010-08-02 23:18:59 +00007044
Douglas Gregor95887f92010-07-08 23:20:03 +00007045 Results.EnterNewScope();
Sebastian Redl75d8a322010-08-02 23:18:59 +00007046 for (GlobalMethodPool::iterator M = MethodPool.begin(),
7047 MEnd = MethodPool.end();
7048 M != MEnd; ++M) {
7049 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
7050 &M->second.second;
7051 MethList && MethList->Method;
Douglas Gregor95887f92010-07-08 23:20:03 +00007052 MethList = MethList->Next) {
7053 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
7054 NumSelIdents))
7055 continue;
7056
Douglas Gregor45879692010-07-08 23:37:41 +00007057 if (AtParameterName) {
7058 // Suggest parameter names we've seen before.
7059 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
7060 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
7061 if (Param->getIdentifier()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007062 CodeCompletionBuilder Builder(Results.getAllocator(),
7063 Results.getCodeCompletionTUInfo());
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007064 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007065 Param->getIdentifier()->getName()));
7066 Results.AddResult(Builder.TakeString());
Douglas Gregor45879692010-07-08 23:37:41 +00007067 }
7068 }
7069
7070 continue;
7071 }
7072
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00007073 Result R(MethList->Method, Results.getBasePriority(MethList->Method), 0);
Douglas Gregor95887f92010-07-08 23:20:03 +00007074 R.StartParameter = NumSelIdents;
7075 R.AllParametersAreInformative = false;
7076 R.DeclaringEntity = true;
7077 Results.MaybeAddResult(R, CurContext);
7078 }
7079 }
7080
7081 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007082 HandleCodeCompleteResults(this, CodeCompleter,
7083 CodeCompletionContext::CCC_Other,
7084 Results.data(),Results.size());
Douglas Gregor95887f92010-07-08 23:20:03 +00007085}
Douglas Gregorb14904c2010-08-13 22:48:40 +00007086
Douglas Gregorec00a262010-08-24 22:20:20 +00007087void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007088 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007089 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007090 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007091 Results.EnterNewScope();
7092
7093 // #if <condition>
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007094 CodeCompletionBuilder Builder(Results.getAllocator(),
7095 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007096 Builder.AddTypedTextChunk("if");
7097 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7098 Builder.AddPlaceholderChunk("condition");
7099 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007100
7101 // #ifdef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007102 Builder.AddTypedTextChunk("ifdef");
7103 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7104 Builder.AddPlaceholderChunk("macro");
7105 Results.AddResult(Builder.TakeString());
7106
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007107 // #ifndef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007108 Builder.AddTypedTextChunk("ifndef");
7109 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7110 Builder.AddPlaceholderChunk("macro");
7111 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007112
7113 if (InConditional) {
7114 // #elif <condition>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007115 Builder.AddTypedTextChunk("elif");
7116 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7117 Builder.AddPlaceholderChunk("condition");
7118 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007119
7120 // #else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007121 Builder.AddTypedTextChunk("else");
7122 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007123
7124 // #endif
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007125 Builder.AddTypedTextChunk("endif");
7126 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007127 }
7128
7129 // #include "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007130 Builder.AddTypedTextChunk("include");
7131 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7132 Builder.AddTextChunk("\"");
7133 Builder.AddPlaceholderChunk("header");
7134 Builder.AddTextChunk("\"");
7135 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007136
7137 // #include <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007138 Builder.AddTypedTextChunk("include");
7139 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7140 Builder.AddTextChunk("<");
7141 Builder.AddPlaceholderChunk("header");
7142 Builder.AddTextChunk(">");
7143 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007144
7145 // #define <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007146 Builder.AddTypedTextChunk("define");
7147 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7148 Builder.AddPlaceholderChunk("macro");
7149 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007150
7151 // #define <macro>(<args>)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007152 Builder.AddTypedTextChunk("define");
7153 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7154 Builder.AddPlaceholderChunk("macro");
7155 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7156 Builder.AddPlaceholderChunk("args");
7157 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7158 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007159
7160 // #undef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007161 Builder.AddTypedTextChunk("undef");
7162 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7163 Builder.AddPlaceholderChunk("macro");
7164 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007165
7166 // #line <number>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007167 Builder.AddTypedTextChunk("line");
7168 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7169 Builder.AddPlaceholderChunk("number");
7170 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007171
7172 // #line <number> "filename"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007173 Builder.AddTypedTextChunk("line");
7174 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7175 Builder.AddPlaceholderChunk("number");
7176 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7177 Builder.AddTextChunk("\"");
7178 Builder.AddPlaceholderChunk("filename");
7179 Builder.AddTextChunk("\"");
7180 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007181
7182 // #error <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007183 Builder.AddTypedTextChunk("error");
7184 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7185 Builder.AddPlaceholderChunk("message");
7186 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007187
7188 // #pragma <arguments>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007189 Builder.AddTypedTextChunk("pragma");
7190 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7191 Builder.AddPlaceholderChunk("arguments");
7192 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007193
David Blaikiebbafb8a2012-03-11 07:00:24 +00007194 if (getLangOpts().ObjC1) {
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007195 // #import "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007196 Builder.AddTypedTextChunk("import");
7197 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7198 Builder.AddTextChunk("\"");
7199 Builder.AddPlaceholderChunk("header");
7200 Builder.AddTextChunk("\"");
7201 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007202
7203 // #import <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007204 Builder.AddTypedTextChunk("import");
7205 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7206 Builder.AddTextChunk("<");
7207 Builder.AddPlaceholderChunk("header");
7208 Builder.AddTextChunk(">");
7209 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007210 }
7211
7212 // #include_next "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007213 Builder.AddTypedTextChunk("include_next");
7214 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7215 Builder.AddTextChunk("\"");
7216 Builder.AddPlaceholderChunk("header");
7217 Builder.AddTextChunk("\"");
7218 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007219
7220 // #include_next <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007221 Builder.AddTypedTextChunk("include_next");
7222 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7223 Builder.AddTextChunk("<");
7224 Builder.AddPlaceholderChunk("header");
7225 Builder.AddTextChunk(">");
7226 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007227
7228 // #warning <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007229 Builder.AddTypedTextChunk("warning");
7230 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7231 Builder.AddPlaceholderChunk("message");
7232 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007233
7234 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7235 // completions for them. And __include_macros is a Clang-internal extension
7236 // that we don't want to encourage anyone to use.
7237
7238 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7239 Results.ExitScope();
7240
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007241 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0de55ce2010-08-25 18:41:16 +00007242 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007243 Results.data(), Results.size());
7244}
7245
7246void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorec00a262010-08-24 22:20:20 +00007247 CodeCompleteOrdinaryName(S,
John McCallfaf5fb42010-08-26 23:41:50 +00007248 S->getFnParent()? Sema::PCC_RecoveryInFunction
7249 : Sema::PCC_Namespace);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007250}
7251
Douglas Gregorec00a262010-08-24 22:20:20 +00007252void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007253 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007254 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007255 IsDefinition? CodeCompletionContext::CCC_MacroName
7256 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor12785102010-08-24 20:21:13 +00007257 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7258 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007259 CodeCompletionBuilder Builder(Results.getAllocator(),
7260 Results.getCodeCompletionTUInfo());
Douglas Gregor12785102010-08-24 20:21:13 +00007261 Results.EnterNewScope();
7262 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7263 MEnd = PP.macro_end();
7264 M != MEnd; ++M) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007265 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007266 M->first->getName()));
Argyrios Kyrtzidis5c8b1cd2012-09-27 00:24:09 +00007267 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7268 CCP_CodePattern,
7269 CXCursor_MacroDefinition));
Douglas Gregor12785102010-08-24 20:21:13 +00007270 }
7271 Results.ExitScope();
7272 } else if (IsDefinition) {
7273 // FIXME: Can we detect when the user just wrote an include guard above?
7274 }
7275
Douglas Gregor0ac41382010-09-23 23:01:17 +00007276 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor12785102010-08-24 20:21:13 +00007277 Results.data(), Results.size());
7278}
7279
Douglas Gregorec00a262010-08-24 22:20:20 +00007280void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007281 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007282 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007283 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorec00a262010-08-24 22:20:20 +00007284
7285 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007286 AddMacroResults(PP, Results, true);
Douglas Gregorec00a262010-08-24 22:20:20 +00007287
7288 // defined (<macro>)
7289 Results.EnterNewScope();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007290 CodeCompletionBuilder Builder(Results.getAllocator(),
7291 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007292 Builder.AddTypedTextChunk("defined");
7293 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7294 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7295 Builder.AddPlaceholderChunk("macro");
7296 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7297 Results.AddResult(Builder.TakeString());
Douglas Gregorec00a262010-08-24 22:20:20 +00007298 Results.ExitScope();
7299
7300 HandleCodeCompleteResults(this, CodeCompleter,
7301 CodeCompletionContext::CCC_PreprocessorExpression,
7302 Results.data(), Results.size());
7303}
7304
7305void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7306 IdentifierInfo *Macro,
7307 MacroInfo *MacroInfo,
7308 unsigned Argument) {
7309 // FIXME: In the future, we could provide "overload" results, much like we
7310 // do for function calls.
7311
Argyrios Kyrtzidis75f6cd22011-08-18 19:41:28 +00007312 // Now just ignore this. There will be another code-completion callback
7313 // for the expanded tokens.
Douglas Gregorec00a262010-08-24 22:20:20 +00007314}
7315
Douglas Gregor11583702010-08-25 17:04:25 +00007316void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor11583702010-08-25 17:04:25 +00007317 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorea736372010-08-25 17:10:00 +00007318 CodeCompletionContext::CCC_NaturalLanguage,
Douglas Gregor11583702010-08-25 17:04:25 +00007319 0, 0);
7320}
7321
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007322void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007323 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00007324 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007325 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7326 CodeCompletionContext::CCC_Recovery);
Douglas Gregor39982192010-08-15 06:18:01 +00007327 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7328 CodeCompletionDeclConsumer Consumer(Builder,
7329 Context.getTranslationUnitDecl());
7330 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7331 Consumer);
7332 }
Douglas Gregorb14904c2010-08-13 22:48:40 +00007333
7334 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007335 AddMacroResults(PP, Builder, true);
Douglas Gregorb14904c2010-08-13 22:48:40 +00007336
7337 Results.clear();
7338 Results.insert(Results.end(),
7339 Builder.data(), Builder.data() + Builder.size());
7340}