blob: b48836495f4c47f570d09debb9e81460cb38f7aa [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())
John McCall5ea95772013-03-09 00:54:27 +00001479 if (!Tag->hasNameForLinkage()) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001480 switch (Tag->getTagKind()) {
1481 case TTK_Struct: return "struct <anonymous>";
Joao Matosdc86f942012-08-31 18:45:21 +00001482 case TTK_Interface: return "__interface <anonymous>";
1483 case TTK_Class: return "class <anonymous>";
Douglas Gregore5c79d52011-10-18 21:20:17 +00001484 case TTK_Union: return "union <anonymous>";
1485 case TTK_Enum: return "enum <anonymous>";
1486 }
1487 }
1488 }
1489
1490 // Slow path: format the type as a string.
1491 std::string Result;
1492 T.getAsStringInternal(Result, Policy);
1493 return Allocator.CopyString(Result);
1494}
1495
Douglas Gregord8c61782012-02-15 15:34:24 +00001496/// \brief Add a completion for "this", if we're in a member function.
1497static void addThisCompletion(Sema &S, ResultBuilder &Results) {
1498 QualType ThisTy = S.getCurrentThisType();
1499 if (ThisTy.isNull())
1500 return;
1501
1502 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001503 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregord8c61782012-02-15 15:34:24 +00001504 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
1505 Builder.AddResultTypeChunk(GetCompletionTypeString(ThisTy,
1506 S.Context,
1507 Policy,
1508 Allocator));
1509 Builder.AddTypedTextChunk("this");
Joao Matosdc86f942012-08-31 18:45:21 +00001510 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregord8c61782012-02-15 15:34:24 +00001511}
1512
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001513/// \brief Add language constructs that show up for "ordinary" names.
John McCallfaf5fb42010-08-26 23:41:50 +00001514static void AddOrdinaryNameResults(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001515 Scope *S,
1516 Sema &SemaRef,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001517 ResultBuilder &Results) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001518 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001519 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregore5c79d52011-10-18 21:20:17 +00001520 PrintingPolicy Policy = getCompletionPrintingPolicy(SemaRef);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001521
John McCall276321a2010-08-25 06:19:51 +00001522 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001523 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001524 case Sema::PCC_Namespace:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001525 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001526 if (Results.includeCodePatterns()) {
1527 // namespace <identifier> { declarations }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001528 Builder.AddTypedTextChunk("namespace");
1529 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1530 Builder.AddPlaceholderChunk("identifier");
1531 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1532 Builder.AddPlaceholderChunk("declarations");
1533 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1534 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1535 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001536 }
1537
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001538 // namespace identifier = identifier ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001539 Builder.AddTypedTextChunk("namespace");
1540 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1541 Builder.AddPlaceholderChunk("name");
1542 Builder.AddChunk(CodeCompletionString::CK_Equal);
1543 Builder.AddPlaceholderChunk("namespace");
1544 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001545
1546 // Using directives
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001547 Builder.AddTypedTextChunk("using");
1548 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1549 Builder.AddTextChunk("namespace");
1550 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1551 Builder.AddPlaceholderChunk("identifier");
1552 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001553
1554 // asm(string-literal)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001555 Builder.AddTypedTextChunk("asm");
1556 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1557 Builder.AddPlaceholderChunk("string-literal");
1558 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1559 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001560
Douglas Gregorf4c33342010-05-28 00:22:41 +00001561 if (Results.includeCodePatterns()) {
1562 // Explicit template instantiation
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001563 Builder.AddTypedTextChunk("template");
1564 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1565 Builder.AddPlaceholderChunk("declaration");
1566 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001567 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001568 }
Douglas Gregorf1934162010-01-13 21:24:21 +00001569
David Blaikiebbafb8a2012-03-11 07:00:24 +00001570 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001571 AddObjCTopLevelResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001572
Douglas Gregorf4c33342010-05-28 00:22:41 +00001573 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001574 // Fall through
1575
John McCallfaf5fb42010-08-26 23:41:50 +00001576 case Sema::PCC_Class:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001577 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001578 // Using declaration
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001579 Builder.AddTypedTextChunk("using");
1580 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1581 Builder.AddPlaceholderChunk("qualifier");
1582 Builder.AddTextChunk("::");
1583 Builder.AddPlaceholderChunk("name");
1584 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001585
Douglas Gregorf4c33342010-05-28 00:22:41 +00001586 // using typename qualifier::name (only in a dependent context)
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001587 if (SemaRef.CurContext->isDependentContext()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001588 Builder.AddTypedTextChunk("using");
1589 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1590 Builder.AddTextChunk("typename");
1591 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1592 Builder.AddPlaceholderChunk("qualifier");
1593 Builder.AddTextChunk("::");
1594 Builder.AddPlaceholderChunk("name");
1595 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001596 }
1597
John McCallfaf5fb42010-08-26 23:41:50 +00001598 if (CCC == Sema::PCC_Class) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001599 AddTypedefResult(Results);
1600
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001601 // public:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001602 Builder.AddTypedTextChunk("public");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001603 if (Results.includeCodePatterns())
1604 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001605 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001606
1607 // protected:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001608 Builder.AddTypedTextChunk("protected");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001609 if (Results.includeCodePatterns())
1610 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001611 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001612
1613 // private:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001614 Builder.AddTypedTextChunk("private");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001615 if (Results.includeCodePatterns())
1616 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001617 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001618 }
1619 }
1620 // Fall through
1621
John McCallfaf5fb42010-08-26 23:41:50 +00001622 case Sema::PCC_Template:
1623 case Sema::PCC_MemberTemplate:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001624 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001625 // template < parameters >
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001626 Builder.AddTypedTextChunk("template");
1627 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1628 Builder.AddPlaceholderChunk("parameters");
1629 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1630 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001631 }
1632
David Blaikiebbafb8a2012-03-11 07:00:24 +00001633 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1634 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001635 break;
1636
John McCallfaf5fb42010-08-26 23:41:50 +00001637 case Sema::PCC_ObjCInterface:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001638 AddObjCInterfaceResults(SemaRef.getLangOpts(), Results, true);
1639 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1640 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001641 break;
1642
John McCallfaf5fb42010-08-26 23:41:50 +00001643 case Sema::PCC_ObjCImplementation:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001644 AddObjCImplementationResults(SemaRef.getLangOpts(), Results, true);
1645 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1646 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001647 break;
1648
John McCallfaf5fb42010-08-26 23:41:50 +00001649 case Sema::PCC_ObjCInstanceVariableList:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001650 AddObjCVisibilityResults(SemaRef.getLangOpts(), Results, true);
Douglas Gregor48d46252010-01-13 21:54:15 +00001651 break;
1652
John McCallfaf5fb42010-08-26 23:41:50 +00001653 case Sema::PCC_RecoveryInFunction:
1654 case Sema::PCC_Statement: {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001655 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001656
David Blaikiebbafb8a2012-03-11 07:00:24 +00001657 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns() &&
1658 SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001659 Builder.AddTypedTextChunk("try");
1660 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1661 Builder.AddPlaceholderChunk("statements");
1662 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1663 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1664 Builder.AddTextChunk("catch");
1665 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1666 Builder.AddPlaceholderChunk("declaration");
1667 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1668 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1669 Builder.AddPlaceholderChunk("statements");
1670 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1671 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1672 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001673 }
David Blaikiebbafb8a2012-03-11 07:00:24 +00001674 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001675 AddObjCStatementResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001676
Douglas Gregorf64acca2010-05-25 21:41:55 +00001677 if (Results.includeCodePatterns()) {
1678 // if (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001679 Builder.AddTypedTextChunk("if");
1680 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001681 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001682 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001683 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001684 Builder.AddPlaceholderChunk("expression");
1685 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1686 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1687 Builder.AddPlaceholderChunk("statements");
1688 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1689 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1690 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001691
Douglas Gregorf64acca2010-05-25 21:41:55 +00001692 // switch (condition) { }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001693 Builder.AddTypedTextChunk("switch");
1694 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001695 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001696 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001697 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001698 Builder.AddPlaceholderChunk("expression");
1699 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1700 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1701 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1702 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1703 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001704 }
1705
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001706 // Switch-specific statements.
John McCallaab3e412010-08-25 08:40:02 +00001707 if (!SemaRef.getCurFunction()->SwitchStack.empty()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001708 // case expression:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001709 Builder.AddTypedTextChunk("case");
1710 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1711 Builder.AddPlaceholderChunk("expression");
1712 Builder.AddChunk(CodeCompletionString::CK_Colon);
1713 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001714
1715 // default:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001716 Builder.AddTypedTextChunk("default");
1717 Builder.AddChunk(CodeCompletionString::CK_Colon);
1718 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001719 }
1720
Douglas Gregorf64acca2010-05-25 21:41:55 +00001721 if (Results.includeCodePatterns()) {
1722 /// while (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001723 Builder.AddTypedTextChunk("while");
1724 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001725 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001726 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001727 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001728 Builder.AddPlaceholderChunk("expression");
1729 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1730 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1731 Builder.AddPlaceholderChunk("statements");
1732 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1733 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1734 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001735
1736 // do { statements } while ( expression );
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001737 Builder.AddTypedTextChunk("do");
1738 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1739 Builder.AddPlaceholderChunk("statements");
1740 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1741 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1742 Builder.AddTextChunk("while");
1743 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1744 Builder.AddPlaceholderChunk("expression");
1745 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1746 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001747
Douglas Gregorf64acca2010-05-25 21:41:55 +00001748 // for ( for-init-statement ; condition ; expression ) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001749 Builder.AddTypedTextChunk("for");
1750 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001751 if (SemaRef.getLangOpts().CPlusPlus || SemaRef.getLangOpts().C99)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001752 Builder.AddPlaceholderChunk("init-statement");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001753 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001754 Builder.AddPlaceholderChunk("init-expression");
1755 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1756 Builder.AddPlaceholderChunk("condition");
1757 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1758 Builder.AddPlaceholderChunk("inc-expression");
1759 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1760 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1761 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1762 Builder.AddPlaceholderChunk("statements");
1763 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1764 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1765 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001766 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001767
1768 if (S->getContinueParent()) {
1769 // continue ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001770 Builder.AddTypedTextChunk("continue");
1771 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001772 }
1773
1774 if (S->getBreakParent()) {
1775 // break ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001776 Builder.AddTypedTextChunk("break");
1777 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001778 }
1779
1780 // "return expression ;" or "return ;", depending on whether we
1781 // know the function is void or not.
1782 bool isVoid = false;
1783 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext))
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();
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002544 // Provide code completion comment for self.GetterName where
2545 // GetterName is the getter method for a property with name
2546 // different from the property name (declared via a property
2547 // getter attribute.
2548 const NamedDecl *ND = Declaration;
2549 if (const ObjCMethodDecl *M = dyn_cast<ObjCMethodDecl>(ND))
2550 if (M->isPropertyAccessor())
2551 if (const ObjCPropertyDecl *PDecl = M->findPropertyDecl())
2552 if (PDecl->getGetterName() == M->getSelector() &&
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002553 PDecl->getIdentifier() != M->getIdentifier()) {
2554 if (const RawComment *RC =
2555 Ctx.getRawCommentForAnyRedecl(M)) {
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002556 Result.addBriefComment(RC->getBriefText(Ctx));
2557 Pattern->BriefComment = Result.getBriefComment();
2558 }
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002559 else if (const RawComment *RC =
2560 Ctx.getRawCommentForAnyRedecl(PDecl)) {
2561 Result.addBriefComment(RC->getBriefText(Ctx));
2562 Pattern->BriefComment = Result.getBriefComment();
2563 }
2564 }
Douglas Gregor78254c82012-03-27 23:34:16 +00002565 }
2566
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002567 return Pattern;
2568 }
Douglas Gregorf09935f2009-12-01 05:55:20 +00002569
Douglas Gregorf09935f2009-12-01 05:55:20 +00002570 if (Kind == RK_Keyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002571 Result.AddTypedTextChunk(Keyword);
2572 return Result.TakeString();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002573 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002574
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002575 if (Kind == RK_Macro) {
Argyrios Kyrtzidis09c9e812013-02-20 00:54:57 +00002576 const MacroDirective *MD = PP.getMacroDirectiveHistory(Macro);
2577 assert(MD && "Not a macro?");
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00002578 const MacroInfo *MI = MD->getMacroInfo();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002579
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002580 Result.AddTypedTextChunk(
2581 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregorf09935f2009-12-01 05:55:20 +00002582
2583 if (!MI->isFunctionLike())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002584 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002585
2586 // Format a function-like macro with placeholders for the arguments.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002587 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor0c505312011-07-30 08:17:44 +00002588 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002589
2590 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
2591 if (MI->isC99Varargs()) {
2592 --AEnd;
2593
2594 if (A == AEnd) {
2595 Result.AddPlaceholderChunk("...");
2596 }
Douglas Gregor0c505312011-07-30 08:17:44 +00002597 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002598
Douglas Gregor0c505312011-07-30 08:17:44 +00002599 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002600 if (A != MI->arg_begin())
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002601 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3aa55262012-01-21 00:43:38 +00002602
2603 if (MI->isVariadic() && (A+1) == AEnd) {
Dylan Noblesmith2c1dd272012-02-05 02:13:05 +00002604 SmallString<32> Arg = (*A)->getName();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002605 if (MI->isC99Varargs())
2606 Arg += ", ...";
2607 else
2608 Arg += "...";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002609 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor3aa55262012-01-21 00:43:38 +00002610 break;
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002611 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002612
2613 // Non-variadic macros are simple.
2614 Result.AddPlaceholderChunk(
2615 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregor0c505312011-07-30 08:17:44 +00002616 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002617 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002618 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002619 }
2620
Douglas Gregorf64acca2010-05-25 21:41:55 +00002621 assert(Kind == RK_Declaration && "Missed a result kind?");
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002622 const NamedDecl *ND = Declaration;
Douglas Gregor78254c82012-03-27 23:34:16 +00002623 Result.addParentContext(ND->getDeclContext());
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002624
2625 if (IncludeBriefComments) {
2626 // Add documentation comment, if it exists.
Dmitri Gribenkoa43ec182012-08-11 00:51:43 +00002627 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(ND)) {
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002628 Result.addBriefComment(RC->getBriefText(Ctx));
Fariborz Jahanian15a0b552013-02-28 17:47:14 +00002629 }
2630 else if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2631 if (OMD->isPropertyAccessor())
2632 if (const ObjCPropertyDecl *PDecl = OMD->findPropertyDecl())
2633 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(PDecl))
2634 Result.addBriefComment(RC->getBriefText(Ctx));
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002635 }
2636
Douglas Gregor9eb77012009-11-07 00:00:49 +00002637 if (StartsNestedNameSpecifier) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002638 Result.AddTypedTextChunk(
2639 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002640 Result.AddTextChunk("::");
2641 return Result.TakeString();
Douglas Gregor9eb77012009-11-07 00:00:49 +00002642 }
Erik Verbruggen98ea7f62011-10-14 15:31:08 +00002643
2644 for (Decl::attr_iterator i = ND->attr_begin(); i != ND->attr_end(); ++i) {
2645 if (AnnotateAttr *Attr = dyn_cast_or_null<AnnotateAttr>(*i)) {
2646 Result.AddAnnotation(Result.getAllocator().CopyString(Attr->getAnnotation()));
2647 }
2648 }
Douglas Gregor9eb77012009-11-07 00:00:49 +00002649
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002650 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002651
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002652 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002653 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002654 Ctx, Policy);
2655 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002656 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002657 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002658 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002659 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002660 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002661 }
2662
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002663 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002664 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002665 Ctx, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002666 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002667 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002668
Douglas Gregor3545ff42009-09-21 16:56:56 +00002669 // Figure out which template parameters are deduced (or have default
2670 // arguments).
Benjamin Kramere0513cb2012-01-30 16:17:39 +00002671 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002672 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002673 unsigned LastDeducibleArgument;
2674 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2675 --LastDeducibleArgument) {
2676 if (!Deduced[LastDeducibleArgument - 1]) {
2677 // C++0x: Figure out if the template argument has a default. If so,
2678 // the user doesn't need to type this argument.
2679 // FIXME: We need to abstract template parameters better!
2680 bool HasDefaultArg = false;
2681 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002682 LastDeducibleArgument - 1);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002683 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2684 HasDefaultArg = TTP->hasDefaultArgument();
2685 else if (NonTypeTemplateParmDecl *NTTP
2686 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2687 HasDefaultArg = NTTP->hasDefaultArgument();
2688 else {
2689 assert(isa<TemplateTemplateParmDecl>(Param));
2690 HasDefaultArg
Douglas Gregor9eb77012009-11-07 00:00:49 +00002691 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002692 }
2693
2694 if (!HasDefaultArg)
2695 break;
2696 }
2697 }
2698
2699 if (LastDeducibleArgument) {
2700 // Some of the function template arguments cannot be deduced from a
2701 // function call, so we introduce an explicit template argument list
2702 // containing all of the arguments up to the first deducible argument.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002703 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002704 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor3545ff42009-09-21 16:56:56 +00002705 LastDeducibleArgument);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002706 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002707 }
2708
2709 // Add the function parameters
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002710 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002711 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002712 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002713 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002714 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002715 }
2716
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002717 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002718 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002719 Ctx, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002720 Result.AddTypedTextChunk(
2721 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002722 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002723 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002724 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002725 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002726 }
2727
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002728 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregord3c5d792009-11-17 16:44:22 +00002729 Selector Sel = Method->getSelector();
2730 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002731 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002732 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002733 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002734 }
2735
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002736 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregor1b605f72009-11-19 01:08:35 +00002737 SelName += ':';
2738 if (StartParameter == 0)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002739 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002740 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002741 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002742
2743 // If there is only one parameter, and we're past it, add an empty
2744 // typed-text chunk since there is nothing to type.
2745 if (Method->param_size() == 1)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002746 Result.AddTypedTextChunk("");
Douglas Gregor1b605f72009-11-19 01:08:35 +00002747 }
Douglas Gregord3c5d792009-11-17 16:44:22 +00002748 unsigned Idx = 0;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002749 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2750 PEnd = Method->param_end();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002751 P != PEnd; (void)++P, ++Idx) {
2752 if (Idx > 0) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00002753 std::string Keyword;
2754 if (Idx > StartParameter)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002755 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregord3c5d792009-11-17 16:44:22 +00002756 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramer632500c2011-07-26 16:59:25 +00002757 Keyword += II->getName();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002758 Keyword += ":";
Douglas Gregor95887f92010-07-08 23:20:03 +00002759 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002760 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002761 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002762 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002763 }
Douglas Gregor1b605f72009-11-19 01:08:35 +00002764
2765 // If we're before the starting parameter, skip the placeholder.
2766 if (Idx < StartParameter)
2767 continue;
Douglas Gregord3c5d792009-11-17 16:44:22 +00002768
2769 std::string Arg;
Douglas Gregore90dd002010-08-24 16:15:59 +00002770
2771 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002772 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregore90dd002010-08-24 16:15:59 +00002773 else {
John McCall31168b02011-06-15 23:02:42 +00002774 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor8f08d742011-07-30 07:55:26 +00002775 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2776 + Arg + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002777 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregor981a0c42010-08-29 19:47:46 +00002778 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramer632500c2011-07-26 16:59:25 +00002779 Arg += II->getName();
Douglas Gregore90dd002010-08-24 16:15:59 +00002780 }
2781
Douglas Gregor400f5972010-08-31 05:13:43 +00002782 if (Method->isVariadic() && (P + 1) == PEnd)
2783 Arg += ", ...";
2784
Douglas Gregor95887f92010-07-08 23:20:03 +00002785 if (DeclaringEntity)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002786 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor95887f92010-07-08 23:20:03 +00002787 else if (AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002788 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8537c52009-11-19 07:41:15 +00002789 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002790 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002791 }
2792
Douglas Gregor04c5f972009-12-23 00:21:46 +00002793 if (Method->isVariadic()) {
Douglas Gregor400f5972010-08-31 05:13:43 +00002794 if (Method->param_size() == 0) {
2795 if (DeclaringEntity)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002796 Result.AddTextChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002797 else if (AllParametersAreInformative)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002798 Result.AddInformativeChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002799 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002800 Result.AddPlaceholderChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002801 }
Douglas Gregordbb71db2010-08-23 23:51:41 +00002802
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002803 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor04c5f972009-12-23 00:21:46 +00002804 }
2805
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002806 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002807 }
2808
Douglas Gregorf09935f2009-12-01 05:55:20 +00002809 if (Qualifier)
Douglas Gregor5bf52692009-09-22 23:15:58 +00002810 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002811 Ctx, Policy);
Douglas Gregorf09935f2009-12-01 05:55:20 +00002812
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002813 Result.AddTypedTextChunk(
2814 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002815 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002816}
2817
Douglas Gregorf0f51982009-09-23 00:34:09 +00002818CodeCompletionString *
2819CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2820 unsigned CurrentArg,
Douglas Gregor36e3b5c2010-10-11 21:37:58 +00002821 Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002822 CodeCompletionAllocator &Allocator,
2823 CodeCompletionTUInfo &CCTUInfo) const {
Douglas Gregor75acd922011-09-27 23:30:47 +00002824 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCall31168b02011-06-15 23:02:42 +00002825
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002826 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002827 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002828 FunctionDecl *FDecl = getFunction();
Douglas Gregor75acd922011-09-27 23:30:47 +00002829 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002830 const FunctionProtoType *Proto
2831 = dyn_cast<FunctionProtoType>(getFunctionType());
2832 if (!FDecl && !Proto) {
2833 // Function without a prototype. Just give the return type and a
2834 // highlighted ellipsis.
2835 const FunctionType *FT = getFunctionType();
Douglas Gregor304f9b02011-02-01 21:15:40 +00002836 Result.AddTextChunk(GetCompletionTypeString(FT->getResultType(),
Douglas Gregor75acd922011-09-27 23:30:47 +00002837 S.Context, Policy,
Douglas Gregor304f9b02011-02-01 21:15:40 +00002838 Result.getAllocator()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002839 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2840 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2841 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002842 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002843 }
2844
2845 if (FDecl)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002846 Result.AddTextChunk(
2847 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002848 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002849 Result.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002850 Result.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00002851 Proto->getResultType().getAsString(Policy)));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002852
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002853 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002854 unsigned NumParams = FDecl? FDecl->getNumParams() : Proto->getNumArgs();
2855 for (unsigned I = 0; I != NumParams; ++I) {
2856 if (I)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002857 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002858
2859 std::string ArgString;
2860 QualType ArgType;
2861
2862 if (FDecl) {
2863 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2864 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2865 } else {
2866 ArgType = Proto->getArgType(I);
2867 }
2868
John McCall31168b02011-06-15 23:02:42 +00002869 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002870
2871 if (I == CurrentArg)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002872 Result.AddChunk(CodeCompletionString::CK_CurrentParameter,
2873 Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002874 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002875 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002876 }
2877
2878 if (Proto && Proto->isVariadic()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002879 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002880 if (CurrentArg < NumParams)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002881 Result.AddTextChunk("...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002882 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002883 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002884 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002885 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002886
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002887 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002888}
2889
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002890unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002891 const LangOptions &LangOpts,
Douglas Gregor6e240332010-08-16 16:18:59 +00002892 bool PreferredTypeIsPointer) {
2893 unsigned Priority = CCP_Macro;
2894
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002895 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2896 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2897 MacroName.equals("Nil")) {
Douglas Gregor6e240332010-08-16 16:18:59 +00002898 Priority = CCP_Constant;
2899 if (PreferredTypeIsPointer)
2900 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002901 }
2902 // Treat "YES", "NO", "true", and "false" as constants.
2903 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2904 MacroName.equals("true") || MacroName.equals("false"))
2905 Priority = CCP_Constant;
2906 // Treat "bool" as a type.
2907 else if (MacroName.equals("bool"))
2908 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2909
Douglas Gregor6e240332010-08-16 16:18:59 +00002910
2911 return Priority;
2912}
2913
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002914CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002915 if (!D)
2916 return CXCursor_UnexposedDecl;
2917
2918 switch (D->getKind()) {
2919 case Decl::Enum: return CXCursor_EnumDecl;
2920 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2921 case Decl::Field: return CXCursor_FieldDecl;
2922 case Decl::Function:
2923 return CXCursor_FunctionDecl;
2924 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2925 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002926 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregordeafd0b2011-12-27 22:43:10 +00002927
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002928 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002929 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2930 case Decl::ObjCMethod:
2931 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2932 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2933 case Decl::CXXMethod: return CXCursor_CXXMethod;
2934 case Decl::CXXConstructor: return CXCursor_Constructor;
2935 case Decl::CXXDestructor: return CXCursor_Destructor;
2936 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2937 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002938 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002939 case Decl::ParmVar: return CXCursor_ParmDecl;
2940 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smithdda56e42011-04-15 14:24:37 +00002941 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002942 case Decl::Var: return CXCursor_VarDecl;
2943 case Decl::Namespace: return CXCursor_Namespace;
2944 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2945 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2946 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2947 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2948 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2949 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis12afd702011-09-30 17:58:23 +00002950 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002951 case Decl::ClassTemplatePartialSpecialization:
2952 return CXCursor_ClassTemplatePartialSpecialization;
2953 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor3e653b32012-04-30 23:41:16 +00002954 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002955
2956 case Decl::Using:
2957 case Decl::UnresolvedUsingValue:
2958 case Decl::UnresolvedUsingTypename:
2959 return CXCursor_UsingDeclaration;
2960
Douglas Gregor4cd65962011-06-03 23:08:58 +00002961 case Decl::ObjCPropertyImpl:
2962 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2963 case ObjCPropertyImplDecl::Dynamic:
2964 return CXCursor_ObjCDynamicDecl;
2965
2966 case ObjCPropertyImplDecl::Synthesize:
2967 return CXCursor_ObjCSynthesizeDecl;
2968 }
Argyrios Kyrtzidis50e5b1d2012-10-05 00:22:24 +00002969
2970 case Decl::Import:
2971 return CXCursor_ModuleImportDecl;
Douglas Gregor4cd65962011-06-03 23:08:58 +00002972
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002973 default:
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002974 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002975 switch (TD->getTagKind()) {
Joao Matosdc86f942012-08-31 18:45:21 +00002976 case TTK_Interface: // fall through
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002977 case TTK_Struct: return CXCursor_StructDecl;
2978 case TTK_Class: return CXCursor_ClassDecl;
2979 case TTK_Union: return CXCursor_UnionDecl;
2980 case TTK_Enum: return CXCursor_EnumDecl;
2981 }
2982 }
2983 }
2984
2985 return CXCursor_UnexposedDecl;
2986}
2987
Douglas Gregor55b037b2010-07-08 20:55:51 +00002988static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor8cb17462012-10-09 16:01:50 +00002989 bool IncludeUndefined,
Douglas Gregor55b037b2010-07-08 20:55:51 +00002990 bool TargetTypeIsPointer = false) {
John McCall276321a2010-08-25 06:19:51 +00002991 typedef CodeCompletionResult Result;
Douglas Gregor55b037b2010-07-08 20:55:51 +00002992
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002993 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002994
Douglas Gregor9eb77012009-11-07 00:00:49 +00002995 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2996 MEnd = PP.macro_end();
Douglas Gregor55b037b2010-07-08 20:55:51 +00002997 M != MEnd; ++M) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00002998 if (IncludeUndefined || M->first->hasMacroDefinition())
2999 Results.AddResult(Result(M->first,
Douglas Gregor6e240332010-08-16 16:18:59 +00003000 getMacroUsagePriority(M->first->getName(),
David Blaikiebbafb8a2012-03-11 07:00:24 +00003001 PP.getLangOpts(),
Douglas Gregor6e240332010-08-16 16:18:59 +00003002 TargetTypeIsPointer)));
Douglas Gregor55b037b2010-07-08 20:55:51 +00003003 }
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003004
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003005 Results.ExitScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003006
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003007}
3008
Douglas Gregorce0e8562010-08-23 21:54:33 +00003009static void AddPrettyFunctionResults(const LangOptions &LangOpts,
3010 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003011 typedef CodeCompletionResult Result;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003012
3013 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003014
Douglas Gregorce0e8562010-08-23 21:54:33 +00003015 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
3016 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith2bf7fdb2013-01-02 11:42:31 +00003017 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregorce0e8562010-08-23 21:54:33 +00003018 Results.AddResult(Result("__func__", CCP_Constant));
3019 Results.ExitScope();
3020}
3021
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003022static void HandleCodeCompleteResults(Sema *S,
3023 CodeCompleteConsumer *CodeCompleter,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003024 CodeCompletionContext Context,
John McCall276321a2010-08-25 06:19:51 +00003025 CodeCompletionResult *Results,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003026 unsigned NumResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00003027 if (CodeCompleter)
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003028 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor3545ff42009-09-21 16:56:56 +00003029}
3030
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003031static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3032 Sema::ParserCompletionContext PCC) {
3033 switch (PCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00003034 case Sema::PCC_Namespace:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003035 return CodeCompletionContext::CCC_TopLevel;
3036
John McCallfaf5fb42010-08-26 23:41:50 +00003037 case Sema::PCC_Class:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003038 return CodeCompletionContext::CCC_ClassStructUnion;
3039
John McCallfaf5fb42010-08-26 23:41:50 +00003040 case Sema::PCC_ObjCInterface:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003041 return CodeCompletionContext::CCC_ObjCInterface;
3042
John McCallfaf5fb42010-08-26 23:41:50 +00003043 case Sema::PCC_ObjCImplementation:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003044 return CodeCompletionContext::CCC_ObjCImplementation;
3045
John McCallfaf5fb42010-08-26 23:41:50 +00003046 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003047 return CodeCompletionContext::CCC_ObjCIvarList;
3048
John McCallfaf5fb42010-08-26 23:41:50 +00003049 case Sema::PCC_Template:
3050 case Sema::PCC_MemberTemplate:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003051 if (S.CurContext->isFileContext())
3052 return CodeCompletionContext::CCC_TopLevel;
David Blaikie8a40f702012-01-17 06:56:22 +00003053 if (S.CurContext->isRecord())
Douglas Gregor0ac41382010-09-23 23:01:17 +00003054 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie8a40f702012-01-17 06:56:22 +00003055 return CodeCompletionContext::CCC_Other;
Douglas Gregor0ac41382010-09-23 23:01:17 +00003056
John McCallfaf5fb42010-08-26 23:41:50 +00003057 case Sema::PCC_RecoveryInFunction:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003058 return CodeCompletionContext::CCC_Recovery;
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003059
John McCallfaf5fb42010-08-26 23:41:50 +00003060 case Sema::PCC_ForInit:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003061 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3062 S.getLangOpts().ObjC1)
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003063 return CodeCompletionContext::CCC_ParenthesizedExpression;
3064 else
3065 return CodeCompletionContext::CCC_Expression;
3066
3067 case Sema::PCC_Expression:
John McCallfaf5fb42010-08-26 23:41:50 +00003068 case Sema::PCC_Condition:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003069 return CodeCompletionContext::CCC_Expression;
3070
John McCallfaf5fb42010-08-26 23:41:50 +00003071 case Sema::PCC_Statement:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003072 return CodeCompletionContext::CCC_Statement;
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003073
John McCallfaf5fb42010-08-26 23:41:50 +00003074 case Sema::PCC_Type:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003075 return CodeCompletionContext::CCC_Type;
Douglas Gregor5e35d592010-09-14 23:59:36 +00003076
3077 case Sema::PCC_ParenthesizedExpression:
3078 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor80039242011-02-15 20:33:25 +00003079
3080 case Sema::PCC_LocalDeclarationSpecifiers:
3081 return CodeCompletionContext::CCC_Type;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003082 }
David Blaikie8a40f702012-01-17 06:56:22 +00003083
3084 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003085}
3086
Douglas Gregorac322ec2010-08-27 21:18:54 +00003087/// \brief If we're in a C++ virtual member function, add completion results
3088/// that invoke the functions we override, since it's common to invoke the
3089/// overridden function as well as adding new functionality.
3090///
3091/// \param S The semantic analysis object for which we are generating results.
3092///
3093/// \param InContext This context in which the nested-name-specifier preceding
3094/// the code-completion point
3095static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3096 ResultBuilder &Results) {
3097 // Look through blocks.
3098 DeclContext *CurContext = S.CurContext;
3099 while (isa<BlockDecl>(CurContext))
3100 CurContext = CurContext->getParent();
3101
3102
3103 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3104 if (!Method || !Method->isVirtual())
3105 return;
3106
3107 // We need to have names for all of the parameters, if we're going to
3108 // generate a forwarding call.
3109 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3110 PEnd = Method->param_end();
3111 P != PEnd;
3112 ++P) {
3113 if (!(*P)->getDeclName())
3114 return;
3115 }
3116
Douglas Gregor75acd922011-09-27 23:30:47 +00003117 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003118 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3119 MEnd = Method->end_overridden_methods();
3120 M != MEnd; ++M) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003121 CodeCompletionBuilder Builder(Results.getAllocator(),
3122 Results.getCodeCompletionTUInfo());
Dmitri Gribenko6cfb1532013-02-14 13:53:30 +00003123 const CXXMethodDecl *Overridden = *M;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003124 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3125 continue;
3126
3127 // If we need a nested-name-specifier, add one now.
3128 if (!InContext) {
3129 NestedNameSpecifier *NNS
3130 = getRequiredQualification(S.Context, CurContext,
3131 Overridden->getDeclContext());
3132 if (NNS) {
3133 std::string Str;
3134 llvm::raw_string_ostream OS(Str);
Douglas Gregor75acd922011-09-27 23:30:47 +00003135 NNS->print(OS, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003136 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003137 }
3138 } else if (!InContext->Equals(Overridden->getDeclContext()))
3139 continue;
3140
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003141 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003142 Overridden->getNameAsString()));
3143 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003144 bool FirstParam = true;
3145 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3146 PEnd = Method->param_end();
3147 P != PEnd; ++P) {
3148 if (FirstParam)
3149 FirstParam = false;
3150 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003151 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003152
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003153 Builder.AddPlaceholderChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003154 (*P)->getIdentifier()->getName()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003155 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003156 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3157 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorac322ec2010-08-27 21:18:54 +00003158 CCP_SuperCompletion,
Douglas Gregor78254c82012-03-27 23:34:16 +00003159 CXCursor_CXXMethod,
3160 CXAvailability_Available,
3161 Overridden));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003162 Results.Ignore(Overridden);
3163 }
3164}
3165
Douglas Gregor07f43572012-01-29 18:15:03 +00003166void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3167 ModuleIdPath Path) {
3168 typedef CodeCompletionResult Result;
3169 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003170 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor07f43572012-01-29 18:15:03 +00003171 CodeCompletionContext::CCC_Other);
3172 Results.EnterNewScope();
3173
3174 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003175 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor07f43572012-01-29 18:15:03 +00003176 typedef CodeCompletionResult Result;
3177 if (Path.empty()) {
3178 // Enumerate all top-level modules.
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003179 SmallVector<Module *, 8> Modules;
Douglas Gregor07f43572012-01-29 18:15:03 +00003180 PP.getHeaderSearchInfo().collectAllModules(Modules);
3181 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3182 Builder.AddTypedTextChunk(
3183 Builder.getAllocator().CopyString(Modules[I]->Name));
3184 Results.AddResult(Result(Builder.TakeString(),
3185 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003186 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003187 Modules[I]->isAvailable()
3188 ? CXAvailability_Available
3189 : CXAvailability_NotAvailable));
3190 }
3191 } else {
3192 // Load the named module.
3193 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3194 Module::AllVisible,
3195 /*IsInclusionDirective=*/false);
3196 // Enumerate submodules.
3197 if (Mod) {
3198 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3199 SubEnd = Mod->submodule_end();
3200 Sub != SubEnd; ++Sub) {
3201
3202 Builder.AddTypedTextChunk(
3203 Builder.getAllocator().CopyString((*Sub)->Name));
3204 Results.AddResult(Result(Builder.TakeString(),
3205 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003206 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003207 (*Sub)->isAvailable()
3208 ? CXAvailability_Available
3209 : CXAvailability_NotAvailable));
3210 }
3211 }
3212 }
3213 Results.ExitScope();
3214 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3215 Results.data(),Results.size());
3216}
3217
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003218void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003219 ParserCompletionContext CompletionContext) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003220 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003221 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003222 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003223 Results.EnterNewScope();
Douglas Gregor50832e02010-09-20 22:39:41 +00003224
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003225 // Determine how to filter results, e.g., so that the names of
3226 // values (functions, enumerators, function templates, etc.) are
3227 // only allowed where we can have an expression.
3228 switch (CompletionContext) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003229 case PCC_Namespace:
3230 case PCC_Class:
3231 case PCC_ObjCInterface:
3232 case PCC_ObjCImplementation:
3233 case PCC_ObjCInstanceVariableList:
3234 case PCC_Template:
3235 case PCC_MemberTemplate:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003236 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003237 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003238 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3239 break;
3240
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003241 case PCC_Statement:
Douglas Gregor5e35d592010-09-14 23:59:36 +00003242 case PCC_ParenthesizedExpression:
Douglas Gregor4d755e82010-08-24 23:58:17 +00003243 case PCC_Expression:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003244 case PCC_ForInit:
3245 case PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003246 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor70febae2010-05-28 00:49:12 +00003247 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3248 else
3249 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003250
David Blaikiebbafb8a2012-03-11 07:00:24 +00003251 if (getLangOpts().CPlusPlus)
Douglas Gregorac322ec2010-08-27 21:18:54 +00003252 MaybeAddOverrideCalls(*this, /*InContext=*/0, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003253 break;
Douglas Gregor6da3db42010-05-25 05:58:43 +00003254
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003255 case PCC_RecoveryInFunction:
Douglas Gregor6da3db42010-05-25 05:58:43 +00003256 // Unfiltered
3257 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003258 }
3259
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003260 // If we are in a C++ non-static member function, check the qualifiers on
3261 // the member function to filter/prioritize the results list.
3262 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3263 if (CurMethod->isInstance())
3264 Results.setObjectTypeQualifiers(
3265 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3266
Douglas Gregorc580c522010-01-14 01:09:38 +00003267 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003268 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3269 CodeCompleter->includeGlobals());
Douglas Gregor92253692009-12-07 09:54:55 +00003270
Douglas Gregorf98e6a22010-01-13 23:51:12 +00003271 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor92253692009-12-07 09:54:55 +00003272 Results.ExitScope();
3273
Douglas Gregorce0e8562010-08-23 21:54:33 +00003274 switch (CompletionContext) {
Douglas Gregor5e35d592010-09-14 23:59:36 +00003275 case PCC_ParenthesizedExpression:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003276 case PCC_Expression:
3277 case PCC_Statement:
3278 case PCC_RecoveryInFunction:
3279 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00003280 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003281 break;
3282
3283 case PCC_Namespace:
3284 case PCC_Class:
3285 case PCC_ObjCInterface:
3286 case PCC_ObjCImplementation:
3287 case PCC_ObjCInstanceVariableList:
3288 case PCC_Template:
3289 case PCC_MemberTemplate:
3290 case PCC_ForInit:
3291 case PCC_Condition:
3292 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003293 case PCC_LocalDeclarationSpecifiers:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003294 break;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003295 }
3296
Douglas Gregor9eb77012009-11-07 00:00:49 +00003297 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003298 AddMacroResults(PP, Results, false);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003299
Douglas Gregor50832e02010-09-20 22:39:41 +00003300 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003301 Results.data(),Results.size());
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00003302}
3303
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003304static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3305 ParsedType Receiver,
3306 IdentifierInfo **SelIdents,
3307 unsigned NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00003308 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003309 bool IsSuper,
3310 ResultBuilder &Results);
3311
3312void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3313 bool AllowNonIdentifiers,
3314 bool AllowNestedNameSpecifiers) {
John McCall276321a2010-08-25 06:19:51 +00003315 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003316 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003317 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003318 AllowNestedNameSpecifiers
3319 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3320 : CodeCompletionContext::CCC_Name);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003321 Results.EnterNewScope();
3322
3323 // Type qualifiers can come after names.
3324 Results.AddResult(Result("const"));
3325 Results.AddResult(Result("volatile"));
David Blaikiebbafb8a2012-03-11 07:00:24 +00003326 if (getLangOpts().C99)
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003327 Results.AddResult(Result("restrict"));
3328
David Blaikiebbafb8a2012-03-11 07:00:24 +00003329 if (getLangOpts().CPlusPlus) {
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003330 if (AllowNonIdentifiers) {
3331 Results.AddResult(Result("operator"));
3332 }
3333
3334 // Add nested-name-specifiers.
3335 if (AllowNestedNameSpecifiers) {
3336 Results.allowNestedNameSpecifiers();
Douglas Gregor0ac41382010-09-23 23:01:17 +00003337 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003338 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3339 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3340 CodeCompleter->includeGlobals());
Douglas Gregor0ac41382010-09-23 23:01:17 +00003341 Results.setFilter(0);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003342 }
3343 }
3344 Results.ExitScope();
3345
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003346 // If we're in a context where we might have an expression (rather than a
3347 // declaration), and what we've seen so far is an Objective-C type that could
3348 // be a receiver of a class message, this may be a class message send with
3349 // the initial opening bracket '[' missing. Add appropriate completions.
3350 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003351 DS.getParsedSpecifiers() == DeclSpec::PQ_TypeSpecifier &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003352 DS.getTypeSpecType() == DeclSpec::TST_typename &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003353 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3354 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003355 !DS.isTypeAltiVecVector() &&
3356 S &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003357 (S->getFlags() & Scope::DeclScope) != 0 &&
3358 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3359 Scope::FunctionPrototypeScope |
3360 Scope::AtCatchScope)) == 0) {
3361 ParsedType T = DS.getRepAsType();
3362 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Douglas Gregorf86e4da2010-09-20 23:34:21 +00003363 AddClassMessageCompletions(*this, S, T, 0, 0, false, false, Results);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003364 }
3365
Douglas Gregor56ccce02010-08-24 04:59:56 +00003366 // Note that we intentionally suppress macro results here, since we do not
3367 // encourage using macros to produce the names of entities.
3368
Douglas Gregor0ac41382010-09-23 23:01:17 +00003369 HandleCodeCompleteResults(this, CodeCompleter,
3370 Results.getCompletionContext(),
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003371 Results.data(), Results.size());
3372}
3373
Douglas Gregor68762e72010-08-23 21:17:50 +00003374struct Sema::CodeCompleteExpressionData {
3375 CodeCompleteExpressionData(QualType PreferredType = QualType())
3376 : PreferredType(PreferredType), IntegralConstantExpression(false),
3377 ObjCCollection(false) { }
3378
3379 QualType PreferredType;
3380 bool IntegralConstantExpression;
3381 bool ObjCCollection;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003382 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregor68762e72010-08-23 21:17:50 +00003383};
3384
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003385/// \brief Perform code-completion in an expression context when we know what
3386/// type we're looking for.
Douglas Gregor68762e72010-08-23 21:17:50 +00003387void Sema::CodeCompleteExpression(Scope *S,
3388 const CodeCompleteExpressionData &Data) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003389 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003390 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003391 CodeCompletionContext::CCC_Expression);
Douglas Gregor68762e72010-08-23 21:17:50 +00003392 if (Data.ObjCCollection)
3393 Results.setFilter(&ResultBuilder::IsObjCCollection);
3394 else if (Data.IntegralConstantExpression)
Douglas Gregor85b50632010-07-28 21:50:18 +00003395 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003396 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003397 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3398 else
3399 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregor68762e72010-08-23 21:17:50 +00003400
3401 if (!Data.PreferredType.isNull())
3402 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3403
3404 // Ignore any declarations that we were told that we don't care about.
3405 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3406 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003407
3408 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003409 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3410 CodeCompleter->includeGlobals());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003411
3412 Results.EnterNewScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003413 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003414 Results.ExitScope();
3415
Douglas Gregor55b037b2010-07-08 20:55:51 +00003416 bool PreferredTypeIsPointer = false;
Douglas Gregor68762e72010-08-23 21:17:50 +00003417 if (!Data.PreferredType.isNull())
3418 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3419 || Data.PreferredType->isMemberPointerType()
3420 || Data.PreferredType->isBlockPointerType();
Douglas Gregor55b037b2010-07-08 20:55:51 +00003421
Douglas Gregorce0e8562010-08-23 21:54:33 +00003422 if (S->getFnParent() &&
3423 !Data.ObjCCollection &&
3424 !Data.IntegralConstantExpression)
David Blaikiebbafb8a2012-03-11 07:00:24 +00003425 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003426
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003427 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003428 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003429 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor68762e72010-08-23 21:17:50 +00003430 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3431 Data.PreferredType),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003432 Results.data(),Results.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003433}
3434
Douglas Gregoreda7e542010-09-18 01:28:11 +00003435void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3436 if (E.isInvalid())
3437 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003438 else if (getLangOpts().ObjC1)
Douglas Gregoreda7e542010-09-18 01:28:11 +00003439 CodeCompleteObjCInstanceMessage(S, E.take(), 0, 0, false);
Douglas Gregored0b69d2010-09-15 16:23:04 +00003440}
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003441
Douglas Gregorb888acf2010-12-09 23:01:55 +00003442/// \brief The set of properties that have already been added, referenced by
3443/// property name.
3444typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3445
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003446/// \brief Retrieve the container definition, if any?
3447static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3448 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3449 if (Interface->hasDefinition())
3450 return Interface->getDefinition();
3451
3452 return Interface;
3453 }
3454
3455 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3456 if (Protocol->hasDefinition())
3457 return Protocol->getDefinition();
3458
3459 return Protocol;
3460 }
3461 return Container;
3462}
3463
3464static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor5d649882009-11-18 22:32:06 +00003465 bool AllowCategories,
Douglas Gregor95147142011-05-05 15:50:42 +00003466 bool AllowNullaryMethods,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003467 DeclContext *CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003468 AddedPropertiesSet &AddedProperties,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003469 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003470 typedef CodeCompletionResult Result;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003471
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003472 // Retrieve the definition.
3473 Container = getContainerDef(Container);
3474
Douglas Gregor9291bad2009-11-18 01:29:26 +00003475 // Add properties in this container.
3476 for (ObjCContainerDecl::prop_iterator P = Container->prop_begin(),
3477 PEnd = Container->prop_end();
3478 P != PEnd;
Douglas Gregorb888acf2010-12-09 23:01:55 +00003479 ++P) {
3480 if (AddedProperties.insert(P->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003481 Results.MaybeAddResult(Result(*P, Results.getBasePriority(*P), 0),
3482 CurContext);
Douglas Gregorb888acf2010-12-09 23:01:55 +00003483 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003484
Douglas Gregor95147142011-05-05 15:50:42 +00003485 // Add nullary methods
3486 if (AllowNullaryMethods) {
3487 ASTContext &Context = Container->getASTContext();
Douglas Gregor75acd922011-09-27 23:30:47 +00003488 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Douglas Gregor95147142011-05-05 15:50:42 +00003489 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
3490 MEnd = Container->meth_end();
3491 M != MEnd; ++M) {
3492 if (M->getSelector().isUnarySelector())
3493 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3494 if (AddedProperties.insert(Name)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003495 CodeCompletionBuilder Builder(Results.getAllocator(),
3496 Results.getCodeCompletionTUInfo());
David Blaikie40ed2972012-06-06 20:45:41 +00003497 AddResultTypeChunk(Context, Policy, *M, Builder);
Douglas Gregor95147142011-05-05 15:50:42 +00003498 Builder.AddTypedTextChunk(
3499 Results.getAllocator().CopyString(Name->getName()));
3500
David Blaikie40ed2972012-06-06 20:45:41 +00003501 Results.MaybeAddResult(Result(Builder.TakeString(), *M,
Douglas Gregor78254c82012-03-27 23:34:16 +00003502 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor95147142011-05-05 15:50:42 +00003503 CurContext);
3504 }
3505 }
3506 }
3507
3508
Douglas Gregor9291bad2009-11-18 01:29:26 +00003509 // Add properties in referenced protocols.
3510 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3511 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
3512 PEnd = Protocol->protocol_end();
3513 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003514 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3515 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003516 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor5d649882009-11-18 22:32:06 +00003517 if (AllowCategories) {
3518 // Look through categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00003519 for (ObjCInterfaceDecl::known_categories_iterator
3520 Cat = IFace->known_categories_begin(),
3521 CatEnd = IFace->known_categories_end();
3522 Cat != CatEnd; ++Cat)
3523 AddObjCProperties(*Cat, AllowCategories, AllowNullaryMethods,
Douglas Gregor95147142011-05-05 15:50:42 +00003524 CurContext, AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00003525 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003526
3527 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003528 for (ObjCInterfaceDecl::all_protocol_iterator
3529 I = IFace->all_referenced_protocol_begin(),
3530 E = IFace->all_referenced_protocol_end(); I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003531 AddObjCProperties(*I, AllowCategories, AllowNullaryMethods, CurContext,
3532 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003533
3534 // Look in the superclass.
3535 if (IFace->getSuperClass())
Douglas Gregor95147142011-05-05 15:50:42 +00003536 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3537 AllowNullaryMethods, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003538 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003539 } else if (const ObjCCategoryDecl *Category
3540 = dyn_cast<ObjCCategoryDecl>(Container)) {
3541 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003542 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
3543 PEnd = Category->protocol_end();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003544 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003545 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3546 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003547 }
3548}
3549
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003550void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor2436e712009-09-17 21:32:03 +00003551 SourceLocation OpLoc,
3552 bool IsArrow) {
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003553 if (!Base || !CodeCompleter)
Douglas Gregor2436e712009-09-17 21:32:03 +00003554 return;
3555
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003556 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3557 if (ConvertedBase.isInvalid())
3558 return;
3559 Base = ConvertedBase.get();
3560
John McCall276321a2010-08-25 06:19:51 +00003561 typedef CodeCompletionResult Result;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003562
Douglas Gregor2436e712009-09-17 21:32:03 +00003563 QualType BaseType = Base->getType();
Douglas Gregor3545ff42009-09-21 16:56:56 +00003564
3565 if (IsArrow) {
3566 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3567 BaseType = Ptr->getPointeeType();
3568 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003569 /*Do nothing*/ ;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003570 else
3571 return;
3572 }
3573
Douglas Gregor21325842011-07-07 16:03:39 +00003574 enum CodeCompletionContext::Kind contextKind;
3575
3576 if (IsArrow) {
3577 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3578 }
3579 else {
3580 if (BaseType->isObjCObjectPointerType() ||
3581 BaseType->isObjCObjectOrInterfaceType()) {
3582 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3583 }
3584 else {
3585 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3586 }
3587 }
3588
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003589 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003590 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00003591 CodeCompletionContext(contextKind,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003592 BaseType),
3593 &ResultBuilder::IsMember);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003594 Results.EnterNewScope();
3595 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003596 // Indicate that we are performing a member access, and the cv-qualifiers
3597 // for the base object type.
3598 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3599
Douglas Gregor9291bad2009-11-18 01:29:26 +00003600 // Access to a C/C++ class, struct, or union.
Douglas Gregor6ae4c522010-01-14 03:21:49 +00003601 Results.allowNestedNameSpecifiers();
Douglas Gregor09bbc652010-01-14 15:47:35 +00003602 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003603 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3604 CodeCompleter->includeGlobals());
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003605
David Blaikiebbafb8a2012-03-11 07:00:24 +00003606 if (getLangOpts().CPlusPlus) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003607 if (!Results.empty()) {
3608 // The "template" keyword can follow "->" or "." in the grammar.
3609 // However, we only want to suggest the template keyword if something
3610 // is dependent.
3611 bool IsDependent = BaseType->isDependentType();
3612 if (!IsDependent) {
3613 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
3614 if (DeclContext *Ctx = (DeclContext *)DepScope->getEntity()) {
3615 IsDependent = Ctx->isDependentContext();
3616 break;
3617 }
3618 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003619
Douglas Gregor9291bad2009-11-18 01:29:26 +00003620 if (IsDependent)
Douglas Gregor78a21012010-01-14 16:01:26 +00003621 Results.AddResult(Result("template"));
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003622 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003623 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003624 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3625 // Objective-C property reference.
Douglas Gregorb888acf2010-12-09 23:01:55 +00003626 AddedPropertiesSet AddedProperties;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003627
3628 // Add property results based on our interface.
3629 const ObjCObjectPointerType *ObjCPtr
3630 = BaseType->getAsObjCInterfacePointerType();
3631 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor95147142011-05-05 15:50:42 +00003632 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3633 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003634 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003635
3636 // Add properties from the protocols in a qualified interface.
3637 for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(),
3638 E = ObjCPtr->qual_end();
3639 I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003640 AddObjCProperties(*I, true, /*AllowNullaryMethods=*/true, CurContext,
3641 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003642 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCall8b07ec22010-05-15 11:32:37 +00003643 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003644 // Objective-C instance variable access.
3645 ObjCInterfaceDecl *Class = 0;
3646 if (const ObjCObjectPointerType *ObjCPtr
3647 = BaseType->getAs<ObjCObjectPointerType>())
3648 Class = ObjCPtr->getInterfaceDecl();
3649 else
John McCall8b07ec22010-05-15 11:32:37 +00003650 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003651
3652 // Add all ivars from this class and its superclasses.
Douglas Gregor2b8162b2010-01-14 16:08:12 +00003653 if (Class) {
3654 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3655 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor39982192010-08-15 06:18:01 +00003656 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3657 CodeCompleter->includeGlobals());
Douglas Gregor9291bad2009-11-18 01:29:26 +00003658 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003659 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003660
3661 // FIXME: How do we cope with isa?
3662
3663 Results.ExitScope();
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003664
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003665 // Hand off the results found for code completion.
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003666 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003667 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003668 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00003669}
3670
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003671void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3672 if (!CodeCompleter)
3673 return;
3674
Douglas Gregor3545ff42009-09-21 16:56:56 +00003675 ResultBuilder::LookupFilter Filter = 0;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003676 enum CodeCompletionContext::Kind ContextKind
3677 = CodeCompletionContext::CCC_Other;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003678 switch ((DeclSpec::TST)TagSpec) {
3679 case DeclSpec::TST_enum:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003680 Filter = &ResultBuilder::IsEnum;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003681 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003682 break;
3683
3684 case DeclSpec::TST_union:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003685 Filter = &ResultBuilder::IsUnion;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003686 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003687 break;
3688
3689 case DeclSpec::TST_struct:
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003690 case DeclSpec::TST_class:
Joao Matosdc86f942012-08-31 18:45:21 +00003691 case DeclSpec::TST_interface:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003692 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003693 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003694 break;
3695
3696 default:
David Blaikie83d382b2011-09-23 05:06:16 +00003697 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003698 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003699
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003700 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3701 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00003702 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCalle87beb22010-04-23 18:46:30 +00003703
3704 // First pass: look for tags.
3705 Results.setFilter(Filter);
Douglas Gregor39982192010-08-15 06:18:01 +00003706 LookupVisibleDecls(S, LookupTagName, Consumer,
3707 CodeCompleter->includeGlobals());
John McCalle87beb22010-04-23 18:46:30 +00003708
Douglas Gregor39982192010-08-15 06:18:01 +00003709 if (CodeCompleter->includeGlobals()) {
3710 // Second pass: look for nested name specifiers.
3711 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3712 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3713 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003714
Douglas Gregor0ac41382010-09-23 23:01:17 +00003715 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003716 Results.data(),Results.size());
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003717}
3718
Douglas Gregor28c78432010-08-27 17:35:51 +00003719void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003720 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003721 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003722 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor28c78432010-08-27 17:35:51 +00003723 Results.EnterNewScope();
3724 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3725 Results.AddResult("const");
3726 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3727 Results.AddResult("volatile");
David Blaikiebbafb8a2012-03-11 07:00:24 +00003728 if (getLangOpts().C99 &&
Douglas Gregor28c78432010-08-27 17:35:51 +00003729 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3730 Results.AddResult("restrict");
Richard Smith8e1ac332013-03-28 01:55:44 +00003731 if (getLangOpts().C11 &&
3732 !(DS.getTypeQualifiers() & DeclSpec::TQ_atomic))
3733 Results.AddResult("_Atomic");
Douglas Gregor28c78432010-08-27 17:35:51 +00003734 Results.ExitScope();
3735 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003736 Results.getCompletionContext(),
Douglas Gregor28c78432010-08-27 17:35:51 +00003737 Results.data(), Results.size());
3738}
3739
Douglas Gregord328d572009-09-21 18:10:23 +00003740void Sema::CodeCompleteCase(Scope *S) {
John McCallaab3e412010-08-25 08:40:02 +00003741 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregord328d572009-09-21 18:10:23 +00003742 return;
John McCall5939b162011-08-06 07:30:58 +00003743
John McCallaab3e412010-08-25 08:40:02 +00003744 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCall5939b162011-08-06 07:30:58 +00003745 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3746 if (!type->isEnumeralType()) {
3747 CodeCompleteExpressionData Data(type);
Douglas Gregor68762e72010-08-23 21:17:50 +00003748 Data.IntegralConstantExpression = true;
3749 CodeCompleteExpression(S, Data);
Douglas Gregord328d572009-09-21 18:10:23 +00003750 return;
Douglas Gregor85b50632010-07-28 21:50:18 +00003751 }
Douglas Gregord328d572009-09-21 18:10:23 +00003752
3753 // Code-complete the cases of a switch statement over an enumeration type
3754 // by providing the list of
John McCall5939b162011-08-06 07:30:58 +00003755 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003756 if (EnumDecl *Def = Enum->getDefinition())
3757 Enum = Def;
Douglas Gregord328d572009-09-21 18:10:23 +00003758
3759 // Determine which enumerators we have already seen in the switch statement.
3760 // FIXME: Ideally, we would also be able to look *past* the code-completion
3761 // token, in case we are code-completing in the middle of the switch and not
3762 // at the end. However, we aren't able to do so at the moment.
3763 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Douglas Gregorf2510672009-09-21 19:57:38 +00003764 NestedNameSpecifier *Qualifier = 0;
Douglas Gregord328d572009-09-21 18:10:23 +00003765 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3766 SC = SC->getNextSwitchCase()) {
3767 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3768 if (!Case)
3769 continue;
3770
3771 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3772 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3773 if (EnumConstantDecl *Enumerator
3774 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3775 // We look into the AST of the case statement to determine which
3776 // enumerator was named. Alternatively, we could compute the value of
3777 // the integral constant expression, then compare it against the
3778 // values of each enumerator. However, value-based approach would not
3779 // work as well with C++ templates where enumerators declared within a
3780 // template are type- and value-dependent.
3781 EnumeratorsSeen.insert(Enumerator);
3782
Douglas Gregorf2510672009-09-21 19:57:38 +00003783 // If this is a qualified-id, keep track of the nested-name-specifier
3784 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregord328d572009-09-21 18:10:23 +00003785 //
3786 // switch (TagD.getKind()) {
3787 // case TagDecl::TK_enum:
3788 // break;
3789 // case XXX
3790 //
Douglas Gregorf2510672009-09-21 19:57:38 +00003791 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregord328d572009-09-21 18:10:23 +00003792 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3793 // TK_struct, and TK_class.
Douglas Gregor4bd90e52009-10-23 18:54:35 +00003794 Qualifier = DRE->getQualifier();
Douglas Gregord328d572009-09-21 18:10:23 +00003795 }
3796 }
3797
David Blaikiebbafb8a2012-03-11 07:00:24 +00003798 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorf2510672009-09-21 19:57:38 +00003799 // If there are no prior enumerators in C++, check whether we have to
3800 // qualify the names of the enumerators that we suggest, because they
3801 // may not be visible in this scope.
Douglas Gregord3cebdb2012-02-01 05:02:47 +00003802 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorf2510672009-09-21 19:57:38 +00003803 }
3804
Douglas Gregord328d572009-09-21 18:10:23 +00003805 // Add any enumerators that have not yet been mentioned.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003806 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003807 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003808 CodeCompletionContext::CCC_Expression);
Douglas Gregord328d572009-09-21 18:10:23 +00003809 Results.EnterNewScope();
3810 for (EnumDecl::enumerator_iterator E = Enum->enumerator_begin(),
3811 EEnd = Enum->enumerator_end();
3812 E != EEnd; ++E) {
David Blaikie40ed2972012-06-06 20:45:41 +00003813 if (EnumeratorsSeen.count(*E))
Douglas Gregord328d572009-09-21 18:10:23 +00003814 continue;
3815
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003816 CodeCompletionResult R(*E, CCP_EnumInCase, Qualifier);
Douglas Gregor3a69eaf2011-02-18 23:30:37 +00003817 Results.AddResult(R, CurContext, 0, false);
Douglas Gregord328d572009-09-21 18:10:23 +00003818 }
3819 Results.ExitScope();
Douglas Gregor285560922010-04-06 20:02:15 +00003820
Douglas Gregor21325842011-07-07 16:03:39 +00003821 //We need to make sure we're setting the right context,
3822 //so only say we include macros if the code completer says we do
3823 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3824 if (CodeCompleter->includeMacros()) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00003825 AddMacroResults(PP, Results, false);
Douglas Gregor21325842011-07-07 16:03:39 +00003826 kind = CodeCompletionContext::CCC_OtherWithMacros;
3827 }
3828
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003829 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00003830 kind,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003831 Results.data(),Results.size());
Douglas Gregord328d572009-09-21 18:10:23 +00003832}
3833
Douglas Gregorcabea402009-09-22 15:41:20 +00003834namespace {
3835 struct IsBetterOverloadCandidate {
3836 Sema &S;
John McCallbc077cf2010-02-08 23:07:23 +00003837 SourceLocation Loc;
Douglas Gregorcabea402009-09-22 15:41:20 +00003838
3839 public:
John McCallbc077cf2010-02-08 23:07:23 +00003840 explicit IsBetterOverloadCandidate(Sema &S, SourceLocation Loc)
3841 : S(S), Loc(Loc) { }
Douglas Gregorcabea402009-09-22 15:41:20 +00003842
3843 bool
3844 operator()(const OverloadCandidate &X, const OverloadCandidate &Y) const {
John McCall5c32be02010-08-24 20:38:10 +00003845 return isBetterOverloadCandidate(S, X, Y, Loc);
Douglas Gregorcabea402009-09-22 15:41:20 +00003846 }
3847 };
3848}
3849
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003850static bool anyNullArguments(llvm::ArrayRef<Expr*> Args) {
3851 if (Args.size() && !Args.data())
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003852 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003853
3854 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003855 if (!Args[I])
3856 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003857
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003858 return false;
3859}
3860
Richard Trieu2bd04012011-09-09 02:00:50 +00003861void Sema::CodeCompleteCall(Scope *S, Expr *FnIn,
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003862 llvm::ArrayRef<Expr *> Args) {
Douglas Gregorcabea402009-09-22 15:41:20 +00003863 if (!CodeCompleter)
3864 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003865
3866 // When we're code-completing for a call, we fall back to ordinary
3867 // name code-completion whenever we can't produce specific
3868 // results. We may want to revisit this strategy in the future,
3869 // e.g., by merging the two kinds of results.
3870
Douglas Gregorcabea402009-09-22 15:41:20 +00003871 Expr *Fn = (Expr *)FnIn;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003872
Douglas Gregorcabea402009-09-22 15:41:20 +00003873 // Ignore type-dependent call expressions entirely.
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003874 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3875 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003876 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregorcabea402009-09-22 15:41:20 +00003877 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003878 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003879
John McCall57500772009-12-16 12:17:52 +00003880 // Build an overload candidate set based on the functions we find.
John McCallbc077cf2010-02-08 23:07:23 +00003881 SourceLocation Loc = Fn->getExprLoc();
3882 OverloadCandidateSet CandidateSet(Loc);
John McCall57500772009-12-16 12:17:52 +00003883
Douglas Gregorcabea402009-09-22 15:41:20 +00003884 // FIXME: What if we're calling something that isn't a function declaration?
3885 // FIXME: What if we're calling a pseudo-destructor?
3886 // FIXME: What if we're calling a member function?
3887
Douglas Gregorff59f672010-01-21 15:46:19 +00003888 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003889 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorff59f672010-01-21 15:46:19 +00003890
John McCall57500772009-12-16 12:17:52 +00003891 Expr *NakedFn = Fn->IgnoreParenCasts();
3892 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003893 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
John McCall57500772009-12-16 12:17:52 +00003894 /*PartialOverloading=*/ true);
3895 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3896 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorff59f672010-01-21 15:46:19 +00003897 if (FDecl) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00003898 if (!getLangOpts().CPlusPlus ||
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003899 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorff59f672010-01-21 15:46:19 +00003900 Results.push_back(ResultCandidate(FDecl));
3901 else
John McCallb89836b2010-01-26 01:37:31 +00003902 // FIXME: access?
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003903 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), Args,
3904 CandidateSet, false, /*PartialOverloading*/true);
Douglas Gregorff59f672010-01-21 15:46:19 +00003905 }
John McCall57500772009-12-16 12:17:52 +00003906 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003907
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003908 QualType ParamType;
3909
Douglas Gregorff59f672010-01-21 15:46:19 +00003910 if (!CandidateSet.empty()) {
3911 // Sort the overload candidate set by placing the best overloads first.
3912 std::stable_sort(CandidateSet.begin(), CandidateSet.end(),
John McCallbc077cf2010-02-08 23:07:23 +00003913 IsBetterOverloadCandidate(*this, Loc));
Douglas Gregorcabea402009-09-22 15:41:20 +00003914
Douglas Gregorff59f672010-01-21 15:46:19 +00003915 // Add the remaining viable overload candidates as code-completion reslults.
3916 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3917 CandEnd = CandidateSet.end();
3918 Cand != CandEnd; ++Cand) {
3919 if (Cand->Viable)
3920 Results.push_back(ResultCandidate(Cand->Function));
3921 }
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003922
3923 // From the viable candidates, try to determine the type of this parameter.
3924 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3925 if (const FunctionType *FType = Results[I].getFunctionType())
3926 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003927 if (Args.size() < Proto->getNumArgs()) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003928 if (ParamType.isNull())
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003929 ParamType = Proto->getArgType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003930 else if (!Context.hasSameUnqualifiedType(
3931 ParamType.getNonReferenceType(),
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003932 Proto->getArgType(Args.size()).getNonReferenceType())) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003933 ParamType = QualType();
3934 break;
3935 }
3936 }
3937 }
3938 } else {
3939 // Try to determine the parameter type from the type of the expression
3940 // being called.
3941 QualType FunctionType = Fn->getType();
3942 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3943 FunctionType = Ptr->getPointeeType();
3944 else if (const BlockPointerType *BlockPtr
3945 = FunctionType->getAs<BlockPointerType>())
3946 FunctionType = BlockPtr->getPointeeType();
3947 else if (const MemberPointerType *MemPtr
3948 = FunctionType->getAs<MemberPointerType>())
3949 FunctionType = MemPtr->getPointeeType();
3950
3951 if (const FunctionProtoType *Proto
3952 = FunctionType->getAs<FunctionProtoType>()) {
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003953 if (Args.size() < Proto->getNumArgs())
3954 ParamType = Proto->getArgType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003955 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003956 }
Douglas Gregor3ef59522009-12-11 19:06:04 +00003957
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003958 if (ParamType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003959 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003960 else
3961 CodeCompleteExpression(S, ParamType);
3962
Douglas Gregorc01890e2010-04-06 20:19:47 +00003963 if (!Results.empty())
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003964 CodeCompleter->ProcessOverloadCandidates(*this, Args.size(), Results.data(),
Douglas Gregor3ef59522009-12-11 19:06:04 +00003965 Results.size());
Douglas Gregorcabea402009-09-22 15:41:20 +00003966}
3967
John McCall48871652010-08-21 09:40:31 +00003968void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3969 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003970 if (!VD) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003971 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003972 return;
3973 }
3974
3975 CodeCompleteExpression(S, VD->getType());
3976}
3977
3978void Sema::CodeCompleteReturn(Scope *S) {
3979 QualType ResultType;
3980 if (isa<BlockDecl>(CurContext)) {
3981 if (BlockScopeInfo *BSI = getCurBlock())
3982 ResultType = BSI->ReturnType;
3983 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
3984 ResultType = Function->getResultType();
3985 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
3986 ResultType = Method->getResultType();
3987
3988 if (ResultType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003989 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003990 else
3991 CodeCompleteExpression(S, ResultType);
3992}
3993
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003994void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003995 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003996 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003997 mapCodeCompletionContext(*this, PCC_Statement));
3998 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3999 Results.EnterNewScope();
4000
4001 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4002 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4003 CodeCompleter->includeGlobals());
4004
4005 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
4006
4007 // "else" block
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004008 CodeCompletionBuilder Builder(Results.getAllocator(),
4009 Results.getCodeCompletionTUInfo());
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004010 Builder.AddTypedTextChunk("else");
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00004011 if (Results.includeCodePatterns()) {
4012 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4013 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4014 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4015 Builder.AddPlaceholderChunk("statements");
4016 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4017 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4018 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004019 Results.AddResult(Builder.TakeString());
4020
4021 // "else if" block
4022 Builder.AddTypedTextChunk("else");
4023 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4024 Builder.AddTextChunk("if");
4025 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4026 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00004027 if (getLangOpts().CPlusPlus)
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004028 Builder.AddPlaceholderChunk("condition");
4029 else
4030 Builder.AddPlaceholderChunk("expression");
4031 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00004032 if (Results.includeCodePatterns()) {
4033 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4034 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4035 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4036 Builder.AddPlaceholderChunk("statements");
4037 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4038 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4039 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004040 Results.AddResult(Builder.TakeString());
4041
4042 Results.ExitScope();
4043
4044 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004045 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004046
4047 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00004048 AddMacroResults(PP, Results, false);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004049
4050 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4051 Results.data(),Results.size());
4052}
4053
Richard Trieu2bd04012011-09-09 02:00:50 +00004054void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004055 if (LHS)
4056 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4057 else
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004058 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004059}
4060
Jeffrey Yasskinc76498d2010-04-08 16:38:48 +00004061void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor2436e712009-09-17 21:32:03 +00004062 bool EnteringContext) {
4063 if (!SS.getScopeRep() || !CodeCompleter)
4064 return;
4065
Douglas Gregor3545ff42009-09-21 16:56:56 +00004066 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4067 if (!Ctx)
4068 return;
Douglas Gregor800f2f02009-12-11 18:28:39 +00004069
4070 // Try to instantiate any non-dependent declaration contexts before
4071 // we look in them.
John McCall0b66eb32010-05-01 00:40:08 +00004072 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregor800f2f02009-12-11 18:28:39 +00004073 return;
4074
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004075 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004076 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004077 CodeCompletionContext::CCC_Name);
Douglas Gregorac322ec2010-08-27 21:18:54 +00004078 Results.EnterNewScope();
Douglas Gregor0ac41382010-09-23 23:01:17 +00004079
Douglas Gregor3545ff42009-09-21 16:56:56 +00004080 // The "template" keyword can follow "::" in the grammar, but only
4081 // put it into the grammar if the nested-name-specifier is dependent.
4082 NestedNameSpecifier *NNS = (NestedNameSpecifier *)SS.getScopeRep();
4083 if (!Results.empty() && NNS->isDependent())
Douglas Gregor78a21012010-01-14 16:01:26 +00004084 Results.AddResult("template");
Douglas Gregorac322ec2010-08-27 21:18:54 +00004085
4086 // Add calls to overridden virtual functions, if there are any.
4087 //
4088 // FIXME: This isn't wonderful, because we don't know whether we're actually
4089 // in a context that permits expressions. This is a general issue with
4090 // qualified-id completions.
4091 if (!EnteringContext)
4092 MaybeAddOverrideCalls(*this, Ctx, Results);
4093 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004094
Douglas Gregorac322ec2010-08-27 21:18:54 +00004095 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4096 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4097
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004098 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorc1679ec2011-07-25 17:48:11 +00004099 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004100 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00004101}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004102
4103void Sema::CodeCompleteUsing(Scope *S) {
4104 if (!CodeCompleter)
4105 return;
4106
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004107 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004108 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004109 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4110 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004111 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004112
4113 // If we aren't in class scope, we could see the "namespace" keyword.
4114 if (!S->isClassScope())
John McCall276321a2010-08-25 06:19:51 +00004115 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004116
4117 // After "using", we can see anything that would start a
4118 // nested-name-specifier.
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004119 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004120 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4121 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004122 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004123
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004124 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004125 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004126 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004127}
4128
4129void Sema::CodeCompleteUsingDirective(Scope *S) {
4130 if (!CodeCompleter)
4131 return;
4132
Douglas Gregor3545ff42009-09-21 16:56:56 +00004133 // After "using namespace", we expect to see a namespace name or namespace
4134 // alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004135 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004136 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004137 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004138 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004139 Results.EnterNewScope();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004140 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004141 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4142 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004143 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004144 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004145 CodeCompletionContext::CCC_Namespace,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004146 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004147}
4148
4149void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4150 if (!CodeCompleter)
4151 return;
4152
Douglas Gregor3545ff42009-09-21 16:56:56 +00004153 DeclContext *Ctx = (DeclContext *)S->getEntity();
4154 if (!S->getParent())
4155 Ctx = Context.getTranslationUnitDecl();
4156
Douglas Gregor0ac41382010-09-23 23:01:17 +00004157 bool SuppressedGlobalResults
4158 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4159
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004160 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004161 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004162 SuppressedGlobalResults
4163 ? CodeCompletionContext::CCC_Namespace
4164 : CodeCompletionContext::CCC_Other,
4165 &ResultBuilder::IsNamespace);
4166
4167 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00004168 // We only want to see those namespaces that have already been defined
4169 // within this scope, because its likely that the user is creating an
4170 // extended namespace declaration. Keep track of the most recent
4171 // definition of each namespace.
4172 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4173 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4174 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4175 NS != NSEnd; ++NS)
David Blaikie40ed2972012-06-06 20:45:41 +00004176 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004177
4178 // Add the most recent definition (or extended definition) of each
4179 // namespace to the list of results.
Douglas Gregor64b12b52009-09-22 23:31:26 +00004180 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004181 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregor78254c82012-03-27 23:34:16 +00004182 NS = OrigToLatest.begin(),
4183 NSEnd = OrigToLatest.end();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004184 NS != NSEnd; ++NS)
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004185 Results.AddResult(CodeCompletionResult(
4186 NS->second, Results.getBasePriority(NS->second), 0),
Douglas Gregorfc59ce12010-01-14 16:14:35 +00004187 CurContext, 0, false);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004188 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004189 }
4190
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004191 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004192 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004193 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004194}
4195
4196void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4197 if (!CodeCompleter)
4198 return;
4199
Douglas Gregor3545ff42009-09-21 16:56:56 +00004200 // After "namespace", we expect to see a namespace or alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004201 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004202 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004203 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004204 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004205 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004206 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4207 CodeCompleter->includeGlobals());
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004208 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004209 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004210 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004211}
4212
Douglas Gregorc811ede2009-09-18 20:05:18 +00004213void Sema::CodeCompleteOperatorName(Scope *S) {
4214 if (!CodeCompleter)
4215 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004216
John McCall276321a2010-08-25 06:19:51 +00004217 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004218 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004219 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004220 CodeCompletionContext::CCC_Type,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004221 &ResultBuilder::IsType);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004222 Results.EnterNewScope();
Douglas Gregorc811ede2009-09-18 20:05:18 +00004223
Douglas Gregor3545ff42009-09-21 16:56:56 +00004224 // Add the names of overloadable operators.
4225#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4226 if (std::strcmp(Spelling, "?")) \
Douglas Gregor78a21012010-01-14 16:01:26 +00004227 Results.AddResult(Result(Spelling));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004228#include "clang/Basic/OperatorKinds.def"
4229
4230 // Add any type names visible from the current scope
Douglas Gregor6ae4c522010-01-14 03:21:49 +00004231 Results.allowNestedNameSpecifiers();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004232 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004233 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4234 CodeCompleter->includeGlobals());
Douglas Gregor3545ff42009-09-21 16:56:56 +00004235
4236 // Add any type specifiers
David Blaikiebbafb8a2012-03-11 07:00:24 +00004237 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004238 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004239
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004240 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004241 CodeCompletionContext::CCC_Type,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004242 Results.data(),Results.size());
Douglas Gregorc811ede2009-09-18 20:05:18 +00004243}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004244
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004245void Sema::CodeCompleteConstructorInitializer(Decl *ConstructorD,
Alexis Hunt1d792652011-01-08 20:30:50 +00004246 CXXCtorInitializer** Initializers,
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004247 unsigned NumInitializers) {
Douglas Gregor75acd922011-09-27 23:30:47 +00004248 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004249 CXXConstructorDecl *Constructor
4250 = static_cast<CXXConstructorDecl *>(ConstructorD);
4251 if (!Constructor)
4252 return;
4253
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004254 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004255 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004256 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004257 Results.EnterNewScope();
4258
4259 // Fill in any already-initialized fields or base classes.
4260 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4261 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
4262 for (unsigned I = 0; I != NumInitializers; ++I) {
4263 if (Initializers[I]->isBaseInitializer())
4264 InitializedBases.insert(
4265 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4266 else
Francois Pichetd583da02010-12-04 09:14:42 +00004267 InitializedFields.insert(cast<FieldDecl>(
4268 Initializers[I]->getAnyMember()));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004269 }
4270
4271 // Add completions for base classes.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004272 CodeCompletionBuilder Builder(Results.getAllocator(),
4273 Results.getCodeCompletionTUInfo());
Douglas Gregor99129ef2010-08-29 19:27:27 +00004274 bool SawLastInitializer = (NumInitializers == 0);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004275 CXXRecordDecl *ClassDecl = Constructor->getParent();
4276 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->bases_begin(),
4277 BaseEnd = ClassDecl->bases_end();
4278 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004279 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4280 SawLastInitializer
4281 = NumInitializers > 0 &&
4282 Initializers[NumInitializers - 1]->isBaseInitializer() &&
4283 Context.hasSameUnqualifiedType(Base->getType(),
4284 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004285 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004286 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004287
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004288 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004289 Results.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004290 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004291 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4292 Builder.AddPlaceholderChunk("args");
4293 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4294 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004295 SawLastInitializer? CCP_NextInitializer
4296 : CCP_MemberDeclaration));
4297 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004298 }
4299
4300 // Add completions for virtual base classes.
4301 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->vbases_begin(),
4302 BaseEnd = ClassDecl->vbases_end();
4303 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004304 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4305 SawLastInitializer
4306 = NumInitializers > 0 &&
4307 Initializers[NumInitializers - 1]->isBaseInitializer() &&
4308 Context.hasSameUnqualifiedType(Base->getType(),
4309 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004310 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004311 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004312
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004313 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004314 Builder.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004315 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004316 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4317 Builder.AddPlaceholderChunk("args");
4318 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4319 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004320 SawLastInitializer? CCP_NextInitializer
4321 : CCP_MemberDeclaration));
4322 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004323 }
4324
4325 // Add completions for members.
4326 for (CXXRecordDecl::field_iterator Field = ClassDecl->field_begin(),
4327 FieldEnd = ClassDecl->field_end();
4328 Field != FieldEnd; ++Field) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004329 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4330 SawLastInitializer
4331 = NumInitializers > 0 &&
Francois Pichetd583da02010-12-04 09:14:42 +00004332 Initializers[NumInitializers - 1]->isAnyMemberInitializer() &&
David Blaikie40ed2972012-06-06 20:45:41 +00004333 Initializers[NumInitializers - 1]->getAnyMember() == *Field;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004334 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004335 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004336
4337 if (!Field->getDeclName())
4338 continue;
4339
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004340 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004341 Field->getIdentifier()->getName()));
4342 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4343 Builder.AddPlaceholderChunk("args");
4344 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4345 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004346 SawLastInitializer? CCP_NextInitializer
Douglas Gregorf3af3112010-09-09 21:42:20 +00004347 : CCP_MemberDeclaration,
Douglas Gregor78254c82012-03-27 23:34:16 +00004348 CXCursor_MemberRef,
4349 CXAvailability_Available,
David Blaikie40ed2972012-06-06 20:45:41 +00004350 *Field));
Douglas Gregor99129ef2010-08-29 19:27:27 +00004351 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004352 }
4353 Results.ExitScope();
4354
Douglas Gregor0ac41382010-09-23 23:01:17 +00004355 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004356 Results.data(), Results.size());
4357}
4358
Douglas Gregord8c61782012-02-15 15:34:24 +00004359/// \brief Determine whether this scope denotes a namespace.
4360static bool isNamespaceScope(Scope *S) {
4361 DeclContext *DC = static_cast<DeclContext *>(S->getEntity());
4362 if (!DC)
4363 return false;
4364
4365 return DC->isFileContext();
4366}
4367
4368void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4369 bool AfterAmpersand) {
4370 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004371 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord8c61782012-02-15 15:34:24 +00004372 CodeCompletionContext::CCC_Other);
4373 Results.EnterNewScope();
4374
4375 // Note what has already been captured.
4376 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4377 bool IncludedThis = false;
4378 for (SmallVectorImpl<LambdaCapture>::iterator C = Intro.Captures.begin(),
4379 CEnd = Intro.Captures.end();
4380 C != CEnd; ++C) {
4381 if (C->Kind == LCK_This) {
4382 IncludedThis = true;
4383 continue;
4384 }
4385
4386 Known.insert(C->Id);
4387 }
4388
4389 // Look for other capturable variables.
4390 for (; S && !isNamespaceScope(S); S = S->getParent()) {
4391 for (Scope::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
4392 D != DEnd; ++D) {
4393 VarDecl *Var = dyn_cast<VarDecl>(*D);
4394 if (!Var ||
4395 !Var->hasLocalStorage() ||
4396 Var->hasAttr<BlocksAttr>())
4397 continue;
4398
4399 if (Known.insert(Var->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004400 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
4401 CurContext, 0, false);
Douglas Gregord8c61782012-02-15 15:34:24 +00004402 }
4403 }
4404
4405 // Add 'this', if it would be valid.
4406 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4407 addThisCompletion(*this, Results);
4408
4409 Results.ExitScope();
4410
4411 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4412 Results.data(), Results.size());
4413}
4414
James Dennett596e4752012-06-14 03:11:41 +00004415/// Macro that optionally prepends an "@" to the string literal passed in via
4416/// Keyword, depending on whether NeedAt is true or false.
4417#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4418
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004419static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004420 ResultBuilder &Results,
4421 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004422 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004423 // Since we have an implementation, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004424 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004425
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004426 CodeCompletionBuilder Builder(Results.getAllocator(),
4427 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004428 if (LangOpts.ObjC2) {
4429 // @dynamic
James Dennett596e4752012-06-14 03:11:41 +00004430 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004431 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4432 Builder.AddPlaceholderChunk("property");
4433 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004434
4435 // @synthesize
James Dennett596e4752012-06-14 03:11:41 +00004436 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004437 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4438 Builder.AddPlaceholderChunk("property");
4439 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004440 }
4441}
4442
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004443static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004444 ResultBuilder &Results,
4445 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004446 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004447
4448 // Since we have an interface or protocol, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004449 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004450
4451 if (LangOpts.ObjC2) {
4452 // @property
James Dennett596e4752012-06-14 03:11:41 +00004453 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004454
4455 // @required
James Dennett596e4752012-06-14 03:11:41 +00004456 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004457
4458 // @optional
James Dennett596e4752012-06-14 03:11:41 +00004459 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004460 }
4461}
4462
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004463static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004464 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004465 CodeCompletionBuilder Builder(Results.getAllocator(),
4466 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004467
4468 // @class name ;
James Dennett596e4752012-06-14 03:11:41 +00004469 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004470 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4471 Builder.AddPlaceholderChunk("name");
4472 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004473
Douglas Gregorf4c33342010-05-28 00:22:41 +00004474 if (Results.includeCodePatterns()) {
4475 // @interface name
4476 // FIXME: Could introduce the whole pattern, including superclasses and
4477 // such.
James Dennett596e4752012-06-14 03:11:41 +00004478 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004479 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4480 Builder.AddPlaceholderChunk("class");
4481 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004482
Douglas Gregorf4c33342010-05-28 00:22:41 +00004483 // @protocol name
James Dennett596e4752012-06-14 03:11:41 +00004484 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004485 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4486 Builder.AddPlaceholderChunk("protocol");
4487 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004488
4489 // @implementation name
James Dennett596e4752012-06-14 03:11:41 +00004490 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004491 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4492 Builder.AddPlaceholderChunk("class");
4493 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004494 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004495
4496 // @compatibility_alias name
James Dennett596e4752012-06-14 03:11:41 +00004497 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004498 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4499 Builder.AddPlaceholderChunk("alias");
4500 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4501 Builder.AddPlaceholderChunk("class");
4502 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor61e36812013-03-07 23:26:24 +00004503
4504 if (Results.getSema().getLangOpts().Modules) {
4505 // @import name
4506 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "import"));
4507 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4508 Builder.AddPlaceholderChunk("module");
4509 Results.AddResult(Result(Builder.TakeString()));
4510 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004511}
4512
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004513void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004514 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004515 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004516 CodeCompletionContext::CCC_Other);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004517 Results.EnterNewScope();
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004518 if (isa<ObjCImplDecl>(CurContext))
David Blaikiebbafb8a2012-03-11 07:00:24 +00004519 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004520 else if (CurContext->isObjCContainer())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004521 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorf1934162010-01-13 21:24:21 +00004522 else
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004523 AddObjCTopLevelResults(Results, false);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004524 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004525 HandleCodeCompleteResults(this, CodeCompleter,
4526 CodeCompletionContext::CCC_Other,
4527 Results.data(),Results.size());
Douglas Gregorf48706c2009-12-07 09:27:33 +00004528}
4529
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004530static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004531 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004532 CodeCompletionBuilder Builder(Results.getAllocator(),
4533 Results.getCodeCompletionTUInfo());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004534
4535 // @encode ( type-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004536 const char *EncodeType = "char[]";
David Blaikiebbafb8a2012-03-11 07:00:24 +00004537 if (Results.getSema().getLangOpts().CPlusPlus ||
4538 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose9da05852012-06-15 18:19:56 +00004539 EncodeType = "const char[]";
Douglas Gregore5c79d52011-10-18 21:20:17 +00004540 Builder.AddResultTypeChunk(EncodeType);
James Dennett596e4752012-06-14 03:11:41 +00004541 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004542 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4543 Builder.AddPlaceholderChunk("type-name");
4544 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4545 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004546
4547 // @protocol ( protocol-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004548 Builder.AddResultTypeChunk("Protocol *");
James Dennett596e4752012-06-14 03:11:41 +00004549 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004550 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4551 Builder.AddPlaceholderChunk("protocol-name");
4552 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4553 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004554
4555 // @selector ( selector )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004556 Builder.AddResultTypeChunk("SEL");
James Dennett596e4752012-06-14 03:11:41 +00004557 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004558 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4559 Builder.AddPlaceholderChunk("selector");
4560 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4561 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004562
4563 // @"string"
4564 Builder.AddResultTypeChunk("NSString *");
4565 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4566 Builder.AddPlaceholderChunk("string");
4567 Builder.AddTextChunk("\"");
4568 Results.AddResult(Result(Builder.TakeString()));
4569
Douglas Gregor951de302012-07-17 23:24:47 +00004570 // @[objects, ...]
Jordan Rose9da05852012-06-15 18:19:56 +00004571 Builder.AddResultTypeChunk("NSArray *");
James Dennett596e4752012-06-14 03:11:41 +00004572 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004573 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004574 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4575 Results.AddResult(Result(Builder.TakeString()));
4576
Douglas Gregor951de302012-07-17 23:24:47 +00004577 // @{key : object, ...}
Jordan Rose9da05852012-06-15 18:19:56 +00004578 Builder.AddResultTypeChunk("NSDictionary *");
James Dennett596e4752012-06-14 03:11:41 +00004579 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"{"));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004580 Builder.AddPlaceholderChunk("key");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004581 Builder.AddChunk(CodeCompletionString::CK_Colon);
4582 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4583 Builder.AddPlaceholderChunk("object, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004584 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4585 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004586
Douglas Gregor951de302012-07-17 23:24:47 +00004587 // @(expression)
Jordan Rose9da05852012-06-15 18:19:56 +00004588 Builder.AddResultTypeChunk("id");
4589 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose9da05852012-06-15 18:19:56 +00004590 Builder.AddPlaceholderChunk("expression");
Jordan Rose9da05852012-06-15 18:19:56 +00004591 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4592 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004593}
4594
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004595static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004596 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004597 CodeCompletionBuilder Builder(Results.getAllocator(),
4598 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004599
Douglas Gregorf4c33342010-05-28 00:22:41 +00004600 if (Results.includeCodePatterns()) {
4601 // @try { statements } @catch ( declaration ) { statements } @finally
4602 // { statements }
James Dennett596e4752012-06-14 03:11:41 +00004603 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004604 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4605 Builder.AddPlaceholderChunk("statements");
4606 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4607 Builder.AddTextChunk("@catch");
4608 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4609 Builder.AddPlaceholderChunk("parameter");
4610 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4611 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4612 Builder.AddPlaceholderChunk("statements");
4613 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4614 Builder.AddTextChunk("@finally");
4615 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4616 Builder.AddPlaceholderChunk("statements");
4617 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4618 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004619 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004620
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004621 // @throw
James Dennett596e4752012-06-14 03:11:41 +00004622 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004623 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4624 Builder.AddPlaceholderChunk("expression");
4625 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004626
Douglas Gregorf4c33342010-05-28 00:22:41 +00004627 if (Results.includeCodePatterns()) {
4628 // @synchronized ( expression ) { statements }
James Dennett596e4752012-06-14 03:11:41 +00004629 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004630 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4631 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4632 Builder.AddPlaceholderChunk("expression");
4633 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4634 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4635 Builder.AddPlaceholderChunk("statements");
4636 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4637 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004638 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004639}
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004640
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004641static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00004642 ResultBuilder &Results,
4643 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004644 typedef CodeCompletionResult Result;
James Dennett596e4752012-06-14 03:11:41 +00004645 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4646 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4647 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004648 if (LangOpts.ObjC2)
James Dennett596e4752012-06-14 03:11:41 +00004649 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004650}
4651
4652void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004653 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004654 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004655 CodeCompletionContext::CCC_Other);
Douglas Gregor48d46252010-01-13 21:54:15 +00004656 Results.EnterNewScope();
David Blaikiebbafb8a2012-03-11 07:00:24 +00004657 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregor48d46252010-01-13 21:54:15 +00004658 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004659 HandleCodeCompleteResults(this, CodeCompleter,
4660 CodeCompletionContext::CCC_Other,
4661 Results.data(),Results.size());
Douglas Gregor48d46252010-01-13 21:54:15 +00004662}
4663
4664void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004665 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004666 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004667 CodeCompletionContext::CCC_Other);
Douglas Gregorf1934162010-01-13 21:24:21 +00004668 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004669 AddObjCStatementResults(Results, false);
4670 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004671 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004672 HandleCodeCompleteResults(this, CodeCompleter,
4673 CodeCompletionContext::CCC_Other,
4674 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004675}
4676
4677void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004678 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004679 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004680 CodeCompletionContext::CCC_Other);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004681 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004682 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004683 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004684 HandleCodeCompleteResults(this, CodeCompleter,
4685 CodeCompletionContext::CCC_Other,
4686 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004687}
4688
Douglas Gregore6078da2009-11-19 00:14:45 +00004689/// \brief Determine whether the addition of the given flag to an Objective-C
4690/// property's attributes will cause a conflict.
Bill Wendling44426052012-12-20 19:22:21 +00004691static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregore6078da2009-11-19 00:14:45 +00004692 // Check if we've already added this flag.
Bill Wendling44426052012-12-20 19:22:21 +00004693 if (Attributes & NewFlag)
Douglas Gregore6078da2009-11-19 00:14:45 +00004694 return true;
4695
Bill Wendling44426052012-12-20 19:22:21 +00004696 Attributes |= NewFlag;
Douglas Gregore6078da2009-11-19 00:14:45 +00004697
4698 // Check for collisions with "readonly".
Bill Wendling44426052012-12-20 19:22:21 +00004699 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4700 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregore6078da2009-11-19 00:14:45 +00004701 return true;
4702
Jordan Rose53cb2f32012-08-20 20:01:13 +00004703 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendling44426052012-12-20 19:22:21 +00004704 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCall31168b02011-06-15 23:02:42 +00004705 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregore6078da2009-11-19 00:14:45 +00004706 ObjCDeclSpec::DQ_PR_copy |
Jordan Rose53cb2f32012-08-20 20:01:13 +00004707 ObjCDeclSpec::DQ_PR_retain |
4708 ObjCDeclSpec::DQ_PR_strong |
4709 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregore6078da2009-11-19 00:14:45 +00004710 if (AssignCopyRetMask &&
4711 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCall31168b02011-06-15 23:02:42 +00004712 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregore6078da2009-11-19 00:14:45 +00004713 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCall31168b02011-06-15 23:02:42 +00004714 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rose53cb2f32012-08-20 20:01:13 +00004715 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4716 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregore6078da2009-11-19 00:14:45 +00004717 return true;
4718
4719 return false;
4720}
4721
Douglas Gregor36029f42009-11-18 23:08:07 +00004722void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroff936354c2009-10-08 21:55:05 +00004723 if (!CodeCompleter)
4724 return;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004725
Bill Wendling44426052012-12-20 19:22:21 +00004726 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroff936354c2009-10-08 21:55:05 +00004727
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004728 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004729 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004730 CodeCompletionContext::CCC_Other);
Steve Naroff936354c2009-10-08 21:55:05 +00004731 Results.EnterNewScope();
Bill Wendling44426052012-12-20 19:22:21 +00004732 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall276321a2010-08-25 06:19:51 +00004733 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendling44426052012-12-20 19:22:21 +00004734 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall276321a2010-08-25 06:19:51 +00004735 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendling44426052012-12-20 19:22:21 +00004736 if (!ObjCPropertyFlagConflicts(Attributes,
John McCall31168b02011-06-15 23:02:42 +00004737 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4738 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendling44426052012-12-20 19:22:21 +00004739 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall276321a2010-08-25 06:19:51 +00004740 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendling44426052012-12-20 19:22:21 +00004741 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall276321a2010-08-25 06:19:51 +00004742 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendling44426052012-12-20 19:22:21 +00004743 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCall31168b02011-06-15 23:02:42 +00004744 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendling44426052012-12-20 19:22:21 +00004745 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall276321a2010-08-25 06:19:51 +00004746 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendling44426052012-12-20 19:22:21 +00004747 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall276321a2010-08-25 06:19:51 +00004748 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendling44426052012-12-20 19:22:21 +00004749 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian1c2d29e2011-06-11 17:14:27 +00004750 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rose53cb2f32012-08-20 20:01:13 +00004751
4752 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall3deb1ad2012-08-21 02:47:43 +00004753 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendling44426052012-12-20 19:22:21 +00004754 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rose53cb2f32012-08-20 20:01:13 +00004755 Results.AddResult(CodeCompletionResult("weak"));
4756
Bill Wendling44426052012-12-20 19:22:21 +00004757 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004758 CodeCompletionBuilder Setter(Results.getAllocator(),
4759 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004760 Setter.AddTypedTextChunk("setter");
4761 Setter.AddTextChunk(" = ");
4762 Setter.AddPlaceholderChunk("method");
4763 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004764 }
Bill Wendling44426052012-12-20 19:22:21 +00004765 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004766 CodeCompletionBuilder Getter(Results.getAllocator(),
4767 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004768 Getter.AddTypedTextChunk("getter");
4769 Getter.AddTextChunk(" = ");
4770 Getter.AddPlaceholderChunk("method");
4771 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004772 }
Steve Naroff936354c2009-10-08 21:55:05 +00004773 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004774 HandleCodeCompleteResults(this, CodeCompleter,
4775 CodeCompletionContext::CCC_Other,
4776 Results.data(),Results.size());
Steve Naroff936354c2009-10-08 21:55:05 +00004777}
Steve Naroffeae65032009-11-07 02:08:14 +00004778
James Dennettf1243872012-06-17 05:33:25 +00004779/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregorc8537c52009-11-19 07:41:15 +00004780/// via code completion.
4781enum ObjCMethodKind {
Dmitri Gribenko4280e5c2012-06-08 23:13:42 +00004782 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4783 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4784 MK_OneArgSelector ///< One-argument selector.
Douglas Gregorc8537c52009-11-19 07:41:15 +00004785};
4786
Douglas Gregor67c692c2010-08-26 15:07:07 +00004787static bool isAcceptableObjCSelector(Selector Sel,
4788 ObjCMethodKind WantKind,
4789 IdentifierInfo **SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004790 unsigned NumSelIdents,
4791 bool AllowSameLength = true) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00004792 if (NumSelIdents > Sel.getNumArgs())
4793 return false;
4794
4795 switch (WantKind) {
4796 case MK_Any: break;
4797 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4798 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4799 }
4800
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004801 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4802 return false;
4803
Douglas Gregor67c692c2010-08-26 15:07:07 +00004804 for (unsigned I = 0; I != NumSelIdents; ++I)
4805 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4806 return false;
4807
4808 return true;
4809}
4810
Douglas Gregorc8537c52009-11-19 07:41:15 +00004811static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4812 ObjCMethodKind WantKind,
4813 IdentifierInfo **SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004814 unsigned NumSelIdents,
4815 bool AllowSameLength = true) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00004816 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004817 NumSelIdents, AllowSameLength);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004818}
Douglas Gregor1154e272010-09-16 16:06:31 +00004819
4820namespace {
4821 /// \brief A set of selectors, which is used to avoid introducing multiple
4822 /// completions with the same selector into the result set.
4823 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4824}
4825
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004826/// \brief Add all of the Objective-C methods in the given Objective-C
4827/// container to the set of results.
4828///
4829/// The container will be a class, protocol, category, or implementation of
4830/// any of the above. This mether will recurse to include methods from
4831/// the superclasses of classes along with their categories, protocols, and
4832/// implementations.
4833///
4834/// \param Container the container in which we'll look to find methods.
4835///
James Dennett596e4752012-06-14 03:11:41 +00004836/// \param WantInstanceMethods Whether to add instance methods (only); if
4837/// false, this routine will add factory methods (only).
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004838///
4839/// \param CurContext the context in which we're performing the lookup that
4840/// finds methods.
4841///
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004842/// \param AllowSameLength Whether we allow a method to be added to the list
4843/// when it has the same number of parameters as we have selector identifiers.
4844///
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004845/// \param Results the structure into which we'll add results.
4846static void AddObjCMethods(ObjCContainerDecl *Container,
4847 bool WantInstanceMethods,
Douglas Gregorc8537c52009-11-19 07:41:15 +00004848 ObjCMethodKind WantKind,
Douglas Gregor1b605f72009-11-19 01:08:35 +00004849 IdentifierInfo **SelIdents,
4850 unsigned NumSelIdents,
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004851 DeclContext *CurContext,
Douglas Gregor1154e272010-09-16 16:06:31 +00004852 VisitedSelectorSet &Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004853 bool AllowSameLength,
Douglas Gregor416b5752010-08-25 01:08:01 +00004854 ResultBuilder &Results,
4855 bool InOriginalClass = true) {
John McCall276321a2010-08-25 06:19:51 +00004856 typedef CodeCompletionResult Result;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00004857 Container = getContainerDef(Container);
Douglas Gregor41778c32013-01-30 06:58:39 +00004858 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4859 bool isRootClass = IFace && !IFace->getSuperClass();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004860 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
4861 MEnd = Container->meth_end();
4862 M != MEnd; ++M) {
Douglas Gregor41778c32013-01-30 06:58:39 +00004863 // The instance methods on the root class can be messaged via the
4864 // metaclass.
4865 if (M->isInstanceMethod() == WantInstanceMethods ||
4866 (isRootClass && !WantInstanceMethods)) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00004867 // Check whether the selector identifiers we've been given are a
4868 // subset of the identifiers for this particular method.
David Blaikie40ed2972012-06-06 20:45:41 +00004869 if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, NumSelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004870 AllowSameLength))
Douglas Gregor1b605f72009-11-19 01:08:35 +00004871 continue;
Douglas Gregorc8537c52009-11-19 07:41:15 +00004872
David Blaikie2d7c57e2012-04-30 02:36:29 +00004873 if (!Selectors.insert(M->getSelector()))
Douglas Gregor1154e272010-09-16 16:06:31 +00004874 continue;
4875
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004876 Result R = Result(*M, Results.getBasePriority(*M), 0);
Douglas Gregor1b605f72009-11-19 01:08:35 +00004877 R.StartParameter = NumSelIdents;
Douglas Gregorc8537c52009-11-19 07:41:15 +00004878 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor416b5752010-08-25 01:08:01 +00004879 if (!InOriginalClass)
4880 R.Priority += CCD_InBaseClass;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004881 Results.MaybeAddResult(R, CurContext);
4882 }
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004883 }
4884
Douglas Gregorf37c9492010-09-16 15:34:59 +00004885 // Visit the protocols of protocols.
4886 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregore6e48b12012-01-01 19:29:29 +00004887 if (Protocol->hasDefinition()) {
4888 const ObjCList<ObjCProtocolDecl> &Protocols
4889 = Protocol->getReferencedProtocols();
4890 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4891 E = Protocols.end();
4892 I != E; ++I)
4893 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
4894 NumSelIdents, CurContext, Selectors, AllowSameLength,
4895 Results, false);
4896 }
Douglas Gregorf37c9492010-09-16 15:34:59 +00004897 }
4898
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00004899 if (!IFace || !IFace->hasDefinition())
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004900 return;
4901
4902 // Add methods in protocols.
Argyrios Kyrtzidise7f3ef32012-03-13 01:09:41 +00004903 for (ObjCInterfaceDecl::protocol_iterator I = IFace->protocol_begin(),
4904 E = IFace->protocol_end();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004905 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00004906 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents, NumSelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004907 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004908
4909 // Add methods in categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00004910 for (ObjCInterfaceDecl::known_categories_iterator
4911 Cat = IFace->known_categories_begin(),
4912 CatEnd = IFace->known_categories_end();
4913 Cat != CatEnd; ++Cat) {
4914 ObjCCategoryDecl *CatDecl = *Cat;
4915
4916 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004917 NumSelIdents, CurContext, Selectors, AllowSameLength,
4918 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004919
4920 // Add a categories protocol methods.
4921 const ObjCList<ObjCProtocolDecl> &Protocols
4922 = CatDecl->getReferencedProtocols();
4923 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4924 E = Protocols.end();
4925 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00004926 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004927 NumSelIdents, CurContext, Selectors, AllowSameLength,
4928 Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004929
4930 // Add methods in category implementations.
4931 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004932 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004933 NumSelIdents, CurContext, Selectors, AllowSameLength,
4934 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004935 }
4936
4937 // Add methods in superclass.
4938 if (IFace->getSuperClass())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004939 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004940 SelIdents, NumSelIdents, CurContext, Selectors,
4941 AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004942
4943 // Add methods in our implementation, if any.
4944 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004945 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004946 NumSelIdents, CurContext, Selectors, AllowSameLength,
4947 Results, InOriginalClass);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004948}
4949
4950
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004951void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004952 // Try to find the interface where getters might live.
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004953 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004954 if (!Class) {
4955 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004956 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004957 Class = Category->getClassInterface();
4958
4959 if (!Class)
4960 return;
4961 }
4962
4963 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004964 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004965 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004966 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004967 Results.EnterNewScope();
4968
Douglas Gregor1154e272010-09-16 16:06:31 +00004969 VisitedSelectorSet Selectors;
4970 AddObjCMethods(Class, true, MK_ZeroArgSelector, 0, 0, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004971 /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004972 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004973 HandleCodeCompleteResults(this, CodeCompleter,
4974 CodeCompletionContext::CCC_Other,
4975 Results.data(),Results.size());
Douglas Gregorc8537c52009-11-19 07:41:15 +00004976}
4977
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004978void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004979 // Try to find the interface where setters might live.
4980 ObjCInterfaceDecl *Class
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004981 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004982 if (!Class) {
4983 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004984 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004985 Class = Category->getClassInterface();
4986
4987 if (!Class)
4988 return;
4989 }
4990
4991 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004992 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004993 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004994 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004995 Results.EnterNewScope();
4996
Douglas Gregor1154e272010-09-16 16:06:31 +00004997 VisitedSelectorSet Selectors;
4998 AddObjCMethods(Class, true, MK_OneArgSelector, 0, 0, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004999 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00005000
5001 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005002 HandleCodeCompleteResults(this, CodeCompleter,
5003 CodeCompletionContext::CCC_Other,
5004 Results.data(),Results.size());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005005}
5006
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005007void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
5008 bool IsParameter) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005009 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005010 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005011 CodeCompletionContext::CCC_Type);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005012 Results.EnterNewScope();
5013
5014 // Add context-sensitive, Objective-C parameter-passing keywords.
5015 bool AddedInOut = false;
5016 if ((DS.getObjCDeclQualifier() &
5017 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
5018 Results.AddResult("in");
5019 Results.AddResult("inout");
5020 AddedInOut = true;
5021 }
5022 if ((DS.getObjCDeclQualifier() &
5023 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
5024 Results.AddResult("out");
5025 if (!AddedInOut)
5026 Results.AddResult("inout");
5027 }
5028 if ((DS.getObjCDeclQualifier() &
5029 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
5030 ObjCDeclSpec::DQ_Oneway)) == 0) {
5031 Results.AddResult("bycopy");
5032 Results.AddResult("byref");
5033 Results.AddResult("oneway");
5034 }
5035
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005036 // If we're completing the return type of an Objective-C method and the
5037 // identifier IBAction refers to a macro, provide a completion item for
5038 // an action, e.g.,
5039 // IBAction)<#selector#>:(id)sender
5040 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
5041 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005042 CodeCompletionBuilder Builder(Results.getAllocator(),
5043 Results.getCodeCompletionTUInfo(),
5044 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005045 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005046 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005047 Builder.AddPlaceholderChunk("selector");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005048 Builder.AddChunk(CodeCompletionString::CK_Colon);
5049 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005050 Builder.AddTextChunk("id");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005051 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005052 Builder.AddTextChunk("sender");
5053 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
5054 }
Douglas Gregored1f5972013-01-30 07:11:43 +00005055
5056 // If we're completing the return type, provide 'instancetype'.
5057 if (!IsParameter) {
5058 Results.AddResult(CodeCompletionResult("instancetype"));
5059 }
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005060
Douglas Gregor99fa2642010-08-24 01:06:58 +00005061 // Add various builtin type names and specifiers.
5062 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5063 Results.ExitScope();
5064
5065 // Add the various type names
5066 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5067 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5068 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5069 CodeCompleter->includeGlobals());
5070
5071 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005072 AddMacroResults(PP, Results, false);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005073
5074 HandleCodeCompleteResults(this, CodeCompleter,
5075 CodeCompletionContext::CCC_Type,
5076 Results.data(), Results.size());
5077}
5078
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005079/// \brief When we have an expression with type "id", we may assume
5080/// that it has some more-specific class type based on knowledge of
5081/// common uses of Objective-C. This routine returns that class type,
5082/// or NULL if no better result could be determined.
5083static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregored0b69d2010-09-15 16:23:04 +00005084 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005085 if (!Msg)
5086 return 0;
5087
5088 Selector Sel = Msg->getSelector();
5089 if (Sel.isNull())
5090 return 0;
5091
5092 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5093 if (!Id)
5094 return 0;
5095
5096 ObjCMethodDecl *Method = Msg->getMethodDecl();
5097 if (!Method)
5098 return 0;
5099
5100 // Determine the class that we're sending the message to.
Douglas Gregor9a129192010-04-21 00:45:42 +00005101 ObjCInterfaceDecl *IFace = 0;
5102 switch (Msg->getReceiverKind()) {
5103 case ObjCMessageExpr::Class:
John McCall8b07ec22010-05-15 11:32:37 +00005104 if (const ObjCObjectType *ObjType
5105 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5106 IFace = ObjType->getInterface();
Douglas Gregor9a129192010-04-21 00:45:42 +00005107 break;
5108
5109 case ObjCMessageExpr::Instance: {
5110 QualType T = Msg->getInstanceReceiver()->getType();
5111 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5112 IFace = Ptr->getInterfaceDecl();
5113 break;
5114 }
5115
5116 case ObjCMessageExpr::SuperInstance:
5117 case ObjCMessageExpr::SuperClass:
5118 break;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005119 }
5120
5121 if (!IFace)
5122 return 0;
5123
5124 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5125 if (Method->isInstanceMethod())
5126 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5127 .Case("retain", IFace)
John McCall31168b02011-06-15 23:02:42 +00005128 .Case("strong", IFace)
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005129 .Case("autorelease", IFace)
5130 .Case("copy", IFace)
5131 .Case("copyWithZone", IFace)
5132 .Case("mutableCopy", IFace)
5133 .Case("mutableCopyWithZone", IFace)
5134 .Case("awakeFromCoder", IFace)
5135 .Case("replacementObjectFromCoder", IFace)
5136 .Case("class", IFace)
5137 .Case("classForCoder", IFace)
5138 .Case("superclass", Super)
5139 .Default(0);
5140
5141 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5142 .Case("new", IFace)
5143 .Case("alloc", IFace)
5144 .Case("allocWithZone", IFace)
5145 .Case("class", IFace)
5146 .Case("superclass", Super)
5147 .Default(0);
5148}
5149
Douglas Gregor6fc04132010-08-27 15:10:57 +00005150// Add a special completion for a message send to "super", which fills in the
5151// most likely case of forwarding all of our arguments to the superclass
5152// function.
5153///
5154/// \param S The semantic analysis object.
5155///
Dmitri Gribenkoadba9be2012-08-23 17:58:28 +00005156/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor6fc04132010-08-27 15:10:57 +00005157/// the "super" keyword. Otherwise, we just need to provide the arguments.
5158///
5159/// \param SelIdents The identifiers in the selector that have already been
5160/// provided as arguments for a send to "super".
5161///
5162/// \param NumSelIdents The number of identifiers in \p SelIdents.
5163///
5164/// \param Results The set of results to augment.
5165///
5166/// \returns the Objective-C method declaration that would be invoked by
5167/// this "super" completion. If NULL, no completion was added.
5168static ObjCMethodDecl *AddSuperSendCompletion(Sema &S, bool NeedSuperKeyword,
5169 IdentifierInfo **SelIdents,
5170 unsigned NumSelIdents,
5171 ResultBuilder &Results) {
5172 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5173 if (!CurMethod)
5174 return 0;
5175
5176 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5177 if (!Class)
5178 return 0;
5179
5180 // Try to find a superclass method with the same selector.
5181 ObjCMethodDecl *SuperMethod = 0;
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005182 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5183 // Check in the class
Douglas Gregor6fc04132010-08-27 15:10:57 +00005184 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5185 CurMethod->isInstanceMethod());
5186
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005187 // Check in categories or class extensions.
5188 if (!SuperMethod) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005189 for (ObjCInterfaceDecl::known_categories_iterator
5190 Cat = Class->known_categories_begin(),
5191 CatEnd = Class->known_categories_end();
5192 Cat != CatEnd; ++Cat) {
5193 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005194 CurMethod->isInstanceMethod())))
5195 break;
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005196 }
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005197 }
5198 }
5199
Douglas Gregor6fc04132010-08-27 15:10:57 +00005200 if (!SuperMethod)
5201 return 0;
5202
5203 // Check whether the superclass method has the same signature.
5204 if (CurMethod->param_size() != SuperMethod->param_size() ||
5205 CurMethod->isVariadic() != SuperMethod->isVariadic())
5206 return 0;
5207
5208 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5209 CurPEnd = CurMethod->param_end(),
5210 SuperP = SuperMethod->param_begin();
5211 CurP != CurPEnd; ++CurP, ++SuperP) {
5212 // Make sure the parameter types are compatible.
5213 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5214 (*SuperP)->getType()))
5215 return 0;
5216
5217 // Make sure we have a parameter name to forward!
5218 if (!(*CurP)->getIdentifier())
5219 return 0;
5220 }
5221
5222 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005223 CodeCompletionBuilder Builder(Results.getAllocator(),
5224 Results.getCodeCompletionTUInfo());
Douglas Gregor6fc04132010-08-27 15:10:57 +00005225
5226 // Give this completion a return type.
Douglas Gregor75acd922011-09-27 23:30:47 +00005227 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5228 Builder);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005229
5230 // If we need the "super" keyword, add it (plus some spacing).
5231 if (NeedSuperKeyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005232 Builder.AddTypedTextChunk("super");
5233 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005234 }
5235
5236 Selector Sel = CurMethod->getSelector();
5237 if (Sel.isUnarySelector()) {
5238 if (NeedSuperKeyword)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005239 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005240 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005241 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005242 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005243 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005244 } else {
5245 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5246 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
5247 if (I > NumSelIdents)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005248 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005249
5250 if (I < NumSelIdents)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005251 Builder.AddInformativeChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005252 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005253 Sel.getNameForSlot(I) + ":"));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005254 else if (NeedSuperKeyword || I > NumSelIdents) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005255 Builder.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005256 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005257 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005258 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005259 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005260 } else {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005261 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005262 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005263 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005264 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005265 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005266 }
5267 }
5268 }
5269
Douglas Gregor78254c82012-03-27 23:34:16 +00005270 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5271 CCP_SuperCompletion));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005272 return SuperMethod;
5273}
5274
Douglas Gregora817a192010-05-27 23:06:34 +00005275void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall276321a2010-08-25 06:19:51 +00005276 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005277 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005278 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005279 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005280 getLangOpts().CPlusPlus11
Douglas Gregord8c61782012-02-15 15:34:24 +00005281 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5282 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregora817a192010-05-27 23:06:34 +00005283
Douglas Gregora817a192010-05-27 23:06:34 +00005284 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5285 Results.EnterNewScope();
Douglas Gregor39982192010-08-15 06:18:01 +00005286 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5287 CodeCompleter->includeGlobals());
Douglas Gregora817a192010-05-27 23:06:34 +00005288
5289 // If we are in an Objective-C method inside a class that has a superclass,
5290 // add "super" as an option.
5291 if (ObjCMethodDecl *Method = getCurMethodDecl())
5292 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor6fc04132010-08-27 15:10:57 +00005293 if (Iface->getSuperClass()) {
Douglas Gregora817a192010-05-27 23:06:34 +00005294 Results.AddResult(Result("super"));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005295
5296 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, 0, 0, Results);
5297 }
Douglas Gregora817a192010-05-27 23:06:34 +00005298
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005299 if (getLangOpts().CPlusPlus11)
Douglas Gregord8c61782012-02-15 15:34:24 +00005300 addThisCompletion(*this, Results);
5301
Douglas Gregora817a192010-05-27 23:06:34 +00005302 Results.ExitScope();
5303
5304 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005305 AddMacroResults(PP, Results, false);
Douglas Gregor50832e02010-09-20 22:39:41 +00005306 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005307 Results.data(), Results.size());
Douglas Gregora817a192010-05-27 23:06:34 +00005308
5309}
5310
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005311void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
5312 IdentifierInfo **SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005313 unsigned NumSelIdents,
5314 bool AtArgumentExpression) {
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005315 ObjCInterfaceDecl *CDecl = 0;
5316 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5317 // Figure out which interface we're in.
5318 CDecl = CurMethod->getClassInterface();
5319 if (!CDecl)
5320 return;
5321
5322 // Find the superclass of this class.
5323 CDecl = CDecl->getSuperClass();
5324 if (!CDecl)
5325 return;
5326
5327 if (CurMethod->isInstanceMethod()) {
5328 // We are inside an instance method, which means that the message
5329 // send [super ...] is actually calling an instance method on the
Douglas Gregor392a84b2010-10-13 21:24:53 +00005330 // current object.
5331 return CodeCompleteObjCInstanceMessage(S, 0,
Douglas Gregor6fc04132010-08-27 15:10:57 +00005332 SelIdents, NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005333 AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005334 CDecl);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005335 }
5336
5337 // Fall through to send to the superclass in CDecl.
5338 } else {
5339 // "super" may be the name of a type or variable. Figure out which
5340 // it is.
Argyrios Kyrtzidis3e56dd42013-03-14 22:56:43 +00005341 IdentifierInfo *Super = getSuperIdentifier();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005342 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5343 LookupOrdinaryName);
5344 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5345 // "super" names an interface. Use it.
5346 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCall8b07ec22010-05-15 11:32:37 +00005347 if (const ObjCObjectType *Iface
5348 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5349 CDecl = Iface->getInterface();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005350 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5351 // "super" names an unresolved type; we can't be more specific.
5352 } else {
5353 // Assume that "super" names some kind of value and parse that way.
5354 CXXScopeSpec SS;
Abramo Bagnara7945c982012-01-27 09:46:47 +00005355 SourceLocation TemplateKWLoc;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005356 UnqualifiedId id;
5357 id.setIdentifier(Super, SuperLoc);
Abramo Bagnara7945c982012-01-27 09:46:47 +00005358 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5359 false, false);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005360 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005361 SelIdents, NumSelIdents,
5362 AtArgumentExpression);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005363 }
5364
5365 // Fall through
5366 }
5367
John McCallba7bf592010-08-24 05:47:05 +00005368 ParsedType Receiver;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005369 if (CDecl)
John McCallba7bf592010-08-24 05:47:05 +00005370 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005371 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005372 NumSelIdents, AtArgumentExpression,
5373 /*IsSuper=*/true);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005374}
5375
Douglas Gregor74661272010-09-21 00:03:25 +00005376/// \brief Given a set of code-completion results for the argument of a message
5377/// send, determine the preferred type (if any) for that argument expression.
5378static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5379 unsigned NumSelIdents) {
5380 typedef CodeCompletionResult Result;
5381 ASTContext &Context = Results.getSema().Context;
5382
5383 QualType PreferredType;
5384 unsigned BestPriority = CCP_Unlikely * 2;
5385 Result *ResultsData = Results.data();
5386 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5387 Result &R = ResultsData[I];
5388 if (R.Kind == Result::RK_Declaration &&
5389 isa<ObjCMethodDecl>(R.Declaration)) {
5390 if (R.Priority <= BestPriority) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00005391 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregor74661272010-09-21 00:03:25 +00005392 if (NumSelIdents <= Method->param_size()) {
5393 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
5394 ->getType();
5395 if (R.Priority < BestPriority || PreferredType.isNull()) {
5396 BestPriority = R.Priority;
5397 PreferredType = MyPreferredType;
5398 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5399 MyPreferredType)) {
5400 PreferredType = QualType();
5401 }
5402 }
5403 }
5404 }
5405 }
5406
5407 return PreferredType;
5408}
5409
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005410static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5411 ParsedType Receiver,
5412 IdentifierInfo **SelIdents,
5413 unsigned NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005414 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005415 bool IsSuper,
5416 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005417 typedef CodeCompletionResult Result;
Douglas Gregor8ce33212009-11-17 17:59:40 +00005418 ObjCInterfaceDecl *CDecl = 0;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005419
Douglas Gregor8ce33212009-11-17 17:59:40 +00005420 // If the given name refers to an interface type, retrieve the
5421 // corresponding declaration.
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005422 if (Receiver) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005423 QualType T = SemaRef.GetTypeFromParser(Receiver, 0);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005424 if (!T.isNull())
John McCall8b07ec22010-05-15 11:32:37 +00005425 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5426 CDecl = Interface->getInterface();
Douglas Gregor8ce33212009-11-17 17:59:40 +00005427 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005428
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005429 // Add all of the factory methods in this Objective-C class, its protocols,
5430 // superclasses, categories, implementation, etc.
Steve Naroffeae65032009-11-07 02:08:14 +00005431 Results.EnterNewScope();
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005432
Douglas Gregor6fc04132010-08-27 15:10:57 +00005433 // If this is a send-to-super, try to add the special "super" send
5434 // completion.
5435 if (IsSuper) {
5436 if (ObjCMethodDecl *SuperMethod
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005437 = AddSuperSendCompletion(SemaRef, false, SelIdents, NumSelIdents,
5438 Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005439 Results.Ignore(SuperMethod);
5440 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005441
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005442 // If we're inside an Objective-C method definition, prefer its selector to
5443 // others.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005444 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005445 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005446
Douglas Gregor1154e272010-09-16 16:06:31 +00005447 VisitedSelectorSet Selectors;
Douglas Gregor6285f752010-04-06 16:40:00 +00005448 if (CDecl)
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005449 AddObjCMethods(CDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005450 SemaRef.CurContext, Selectors, AtArgumentExpression,
5451 Results);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005452 else {
Douglas Gregor6285f752010-04-06 16:40:00 +00005453 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005454
Douglas Gregord720daf2010-04-06 17:30:22 +00005455 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005456 // pool from the AST file.
Axel Naumanndd433f02012-10-18 19:05:02 +00005457 if (SemaRef.getExternalSource()) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005458 for (uint32_t I = 0,
Axel Naumanndd433f02012-10-18 19:05:02 +00005459 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall75b960e2010-06-01 09:23:16 +00005460 I != N; ++I) {
Axel Naumanndd433f02012-10-18 19:05:02 +00005461 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005462 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005463 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005464
5465 SemaRef.ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005466 }
5467 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005468
5469 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5470 MEnd = SemaRef.MethodPool.end();
Sebastian Redl75d8a322010-08-02 23:18:59 +00005471 M != MEnd; ++M) {
5472 for (ObjCMethodList *MethList = &M->second.second;
5473 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005474 MethList = MethList->getNext()) {
Douglas Gregor6285f752010-04-06 16:40:00 +00005475 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5476 NumSelIdents))
5477 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005478
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005479 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Douglas Gregor6285f752010-04-06 16:40:00 +00005480 R.StartParameter = NumSelIdents;
5481 R.AllParametersAreInformative = false;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005482 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor6285f752010-04-06 16:40:00 +00005483 }
5484 }
5485 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005486
5487 Results.ExitScope();
5488}
Douglas Gregor6285f752010-04-06 16:40:00 +00005489
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005490void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
5491 IdentifierInfo **SelIdents,
5492 unsigned NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005493 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005494 bool IsSuper) {
Douglas Gregor63745d52011-07-21 01:05:26 +00005495
5496 QualType T = this->GetTypeFromParser(Receiver);
5497
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005498 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005499 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005500 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Douglas Gregorea777402011-07-26 15:24:30 +00005501 T, SelIdents, NumSelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005502
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005503 AddClassMessageCompletions(*this, S, Receiver, SelIdents, NumSelIdents,
5504 AtArgumentExpression, IsSuper, Results);
Douglas Gregor74661272010-09-21 00:03:25 +00005505
5506 // If we're actually at the argument expression (rather than prior to the
5507 // selector), we're actually performing code completion for an expression.
5508 // Determine whether we have a single, best method. If so, we can
5509 // code-complete the expression using the corresponding parameter type as
5510 // our preferred type, improving completion results.
5511 if (AtArgumentExpression) {
5512 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Douglas Gregor63745d52011-07-21 01:05:26 +00005513 NumSelIdents);
Douglas Gregor74661272010-09-21 00:03:25 +00005514 if (PreferredType.isNull())
5515 CodeCompleteOrdinaryName(S, PCC_Expression);
5516 else
5517 CodeCompleteExpression(S, PreferredType);
5518 return;
5519 }
5520
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005521 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005522 Results.getCompletionContext(),
Douglas Gregor6fc04132010-08-27 15:10:57 +00005523 Results.data(), Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005524}
5525
Richard Trieu2bd04012011-09-09 02:00:50 +00005526void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Douglas Gregor1b605f72009-11-19 01:08:35 +00005527 IdentifierInfo **SelIdents,
Douglas Gregor6fc04132010-08-27 15:10:57 +00005528 unsigned NumSelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005529 bool AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005530 ObjCInterfaceDecl *Super) {
John McCall276321a2010-08-25 06:19:51 +00005531 typedef CodeCompletionResult Result;
Steve Naroffeae65032009-11-07 02:08:14 +00005532
5533 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffeae65032009-11-07 02:08:14 +00005534
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005535 // If necessary, apply function/array conversion to the receiver.
5536 // C99 6.7.5.3p[7,8].
John Wiegley01296292011-04-08 18:41:53 +00005537 if (RecExpr) {
5538 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5539 if (Conv.isInvalid()) // conversion failed. bail.
5540 return;
5541 RecExpr = Conv.take();
5542 }
Douglas Gregor392a84b2010-10-13 21:24:53 +00005543 QualType ReceiverType = RecExpr? RecExpr->getType()
5544 : Super? Context.getObjCObjectPointerType(
5545 Context.getObjCInterfaceType(Super))
5546 : Context.getObjCIdType();
Steve Naroffeae65032009-11-07 02:08:14 +00005547
Douglas Gregordc520b02010-11-08 21:12:30 +00005548 // If we're messaging an expression with type "id" or "Class", check
5549 // whether we know something special about the receiver that allows
5550 // us to assume a more-specific receiver type.
5551 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType())
5552 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5553 if (ReceiverType->isObjCClassType())
5554 return CodeCompleteObjCClassMessage(S,
5555 ParsedType::make(Context.getObjCInterfaceType(IFace)),
5556 SelIdents, NumSelIdents,
5557 AtArgumentExpression, Super);
5558
5559 ReceiverType = Context.getObjCObjectPointerType(
5560 Context.getObjCInterfaceType(IFace));
5561 }
5562
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005563 // Build the set of methods we can see.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005564 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005565 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005566 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Douglas Gregorea777402011-07-26 15:24:30 +00005567 ReceiverType, SelIdents, NumSelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005568
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005569 Results.EnterNewScope();
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005570
Douglas Gregor6fc04132010-08-27 15:10:57 +00005571 // If this is a send-to-super, try to add the special "super" send
5572 // completion.
Douglas Gregor392a84b2010-10-13 21:24:53 +00005573 if (Super) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005574 if (ObjCMethodDecl *SuperMethod
5575 = AddSuperSendCompletion(*this, false, SelIdents, NumSelIdents,
5576 Results))
5577 Results.Ignore(SuperMethod);
5578 }
5579
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005580 // If we're inside an Objective-C method definition, prefer its selector to
5581 // others.
5582 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5583 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005584
Douglas Gregor1154e272010-09-16 16:06:31 +00005585 // Keep track of the selectors we've already added.
5586 VisitedSelectorSet Selectors;
5587
Douglas Gregora3329fa2009-11-18 00:06:18 +00005588 // Handle messages to Class. This really isn't a message to an instance
5589 // method, so we treat it the same way we would treat a message send to a
5590 // class method.
5591 if (ReceiverType->isObjCClassType() ||
5592 ReceiverType->isObjCQualifiedClassType()) {
5593 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5594 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Douglas Gregorc8537c52009-11-19 07:41:15 +00005595 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005596 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005597 }
5598 }
5599 // Handle messages to a qualified ID ("id<foo>").
5600 else if (const ObjCObjectPointerType *QualID
5601 = ReceiverType->getAsObjCQualifiedIdType()) {
5602 // Search protocols for instance methods.
5603 for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(),
5604 E = QualID->qual_end();
5605 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00005606 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005607 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005608 }
5609 // Handle messages to a pointer to interface type.
5610 else if (const ObjCObjectPointerType *IFacePtr
5611 = ReceiverType->getAsObjCInterfacePointerType()) {
5612 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregorc8537c52009-11-19 07:41:15 +00005613 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005614 NumSelIdents, CurContext, Selectors, AtArgumentExpression,
5615 Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005616
5617 // Search protocols for instance methods.
5618 for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(),
5619 E = IFacePtr->qual_end();
5620 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00005621 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005622 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005623 }
Douglas Gregor6285f752010-04-06 16:40:00 +00005624 // Handle messages to "id".
5625 else if (ReceiverType->isObjCIdType()) {
Douglas Gregord720daf2010-04-06 17:30:22 +00005626 // We're messaging "id", so provide all instance methods we know
5627 // about as code-completion results.
5628
5629 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005630 // pool from the AST file.
Douglas Gregord720daf2010-04-06 17:30:22 +00005631 if (ExternalSource) {
John McCall75b960e2010-06-01 09:23:16 +00005632 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5633 I != N; ++I) {
5634 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00005635 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005636 continue;
5637
Sebastian Redl75d8a322010-08-02 23:18:59 +00005638 ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005639 }
5640 }
5641
Sebastian Redl75d8a322010-08-02 23:18:59 +00005642 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5643 MEnd = MethodPool.end();
5644 M != MEnd; ++M) {
5645 for (ObjCMethodList *MethList = &M->second.first;
5646 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005647 MethList = MethList->getNext()) {
Douglas Gregor6285f752010-04-06 16:40:00 +00005648 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5649 NumSelIdents))
5650 continue;
Douglas Gregor1154e272010-09-16 16:06:31 +00005651
5652 if (!Selectors.insert(MethList->Method->getSelector()))
5653 continue;
5654
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005655 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Douglas Gregor6285f752010-04-06 16:40:00 +00005656 R.StartParameter = NumSelIdents;
5657 R.AllParametersAreInformative = false;
5658 Results.MaybeAddResult(R, CurContext);
5659 }
5660 }
5661 }
Steve Naroffeae65032009-11-07 02:08:14 +00005662 Results.ExitScope();
Douglas Gregor74661272010-09-21 00:03:25 +00005663
5664
5665 // If we're actually at the argument expression (rather than prior to the
5666 // selector), we're actually performing code completion for an expression.
5667 // Determine whether we have a single, best method. If so, we can
5668 // code-complete the expression using the corresponding parameter type as
5669 // our preferred type, improving completion results.
5670 if (AtArgumentExpression) {
5671 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
5672 NumSelIdents);
5673 if (PreferredType.isNull())
5674 CodeCompleteOrdinaryName(S, PCC_Expression);
5675 else
5676 CodeCompleteExpression(S, PreferredType);
5677 return;
5678 }
5679
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005680 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005681 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005682 Results.data(),Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005683}
Douglas Gregorbaf69612009-11-18 04:19:12 +00005684
Douglas Gregor68762e72010-08-23 21:17:50 +00005685void Sema::CodeCompleteObjCForCollection(Scope *S,
5686 DeclGroupPtrTy IterationVar) {
5687 CodeCompleteExpressionData Data;
5688 Data.ObjCCollection = true;
5689
5690 if (IterationVar.getAsOpaquePtr()) {
5691 DeclGroupRef DG = IterationVar.getAsVal<DeclGroupRef>();
5692 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5693 if (*I)
5694 Data.IgnoreDecls.push_back(*I);
5695 }
5696 }
5697
5698 CodeCompleteExpression(S, Data);
5699}
5700
Douglas Gregor67c692c2010-08-26 15:07:07 +00005701void Sema::CodeCompleteObjCSelector(Scope *S, IdentifierInfo **SelIdents,
5702 unsigned NumSelIdents) {
5703 // If we have an external source, load the entire class method
5704 // pool from the AST file.
5705 if (ExternalSource) {
5706 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5707 I != N; ++I) {
5708 Selector Sel = ExternalSource->GetExternalSelector(I);
5709 if (Sel.isNull() || MethodPool.count(Sel))
5710 continue;
5711
5712 ReadMethodPool(Sel);
5713 }
5714 }
5715
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005716 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005717 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005718 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor67c692c2010-08-26 15:07:07 +00005719 Results.EnterNewScope();
5720 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5721 MEnd = MethodPool.end();
5722 M != MEnd; ++M) {
5723
5724 Selector Sel = M->first;
5725 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents, NumSelIdents))
5726 continue;
5727
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005728 CodeCompletionBuilder Builder(Results.getAllocator(),
5729 Results.getCodeCompletionTUInfo());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005730 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005731 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005732 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005733 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005734 continue;
5735 }
5736
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005737 std::string Accumulator;
Douglas Gregor67c692c2010-08-26 15:07:07 +00005738 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005739 if (I == NumSelIdents) {
5740 if (!Accumulator.empty()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005741 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005742 Accumulator));
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005743 Accumulator.clear();
5744 }
5745 }
5746
Benjamin Kramer632500c2011-07-26 16:59:25 +00005747 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005748 Accumulator += ':';
Douglas Gregor67c692c2010-08-26 15:07:07 +00005749 }
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005750 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005751 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005752 }
5753 Results.ExitScope();
5754
5755 HandleCodeCompleteResults(this, CodeCompleter,
5756 CodeCompletionContext::CCC_SelectorName,
5757 Results.data(), Results.size());
5758}
5759
Douglas Gregorbaf69612009-11-18 04:19:12 +00005760/// \brief Add all of the protocol declarations that we find in the given
5761/// (translation unit) context.
5762static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005763 bool OnlyForwardDeclarations,
Douglas Gregorbaf69612009-11-18 04:19:12 +00005764 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005765 typedef CodeCompletionResult Result;
Douglas Gregorbaf69612009-11-18 04:19:12 +00005766
5767 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5768 DEnd = Ctx->decls_end();
5769 D != DEnd; ++D) {
5770 // Record any protocols we find.
5771 if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*D))
Douglas Gregore6e48b12012-01-01 19:29:29 +00005772 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005773 Results.AddResult(Result(Proto, Results.getBasePriority(Proto), 0),
5774 CurContext, 0, false);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005775 }
5776}
5777
5778void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5779 unsigned NumProtocols) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005780 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005781 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005782 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005783
Douglas Gregora3b23b02010-12-09 21:44:02 +00005784 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5785 Results.EnterNewScope();
5786
5787 // Tell the result set to ignore all of the protocols we have
5788 // already seen.
5789 // FIXME: This doesn't work when caching code-completion results.
5790 for (unsigned I = 0; I != NumProtocols; ++I)
5791 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5792 Protocols[I].second))
5793 Results.Ignore(Protocol);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005794
Douglas Gregora3b23b02010-12-09 21:44:02 +00005795 // Add all protocols.
5796 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5797 Results);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005798
Douglas Gregora3b23b02010-12-09 21:44:02 +00005799 Results.ExitScope();
5800 }
5801
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005802 HandleCodeCompleteResults(this, CodeCompleter,
5803 CodeCompletionContext::CCC_ObjCProtocolName,
5804 Results.data(),Results.size());
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005805}
5806
5807void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005808 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005809 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005810 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005811
Douglas Gregora3b23b02010-12-09 21:44:02 +00005812 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5813 Results.EnterNewScope();
5814
5815 // Add all protocols.
5816 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5817 Results);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005818
Douglas Gregora3b23b02010-12-09 21:44:02 +00005819 Results.ExitScope();
5820 }
5821
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005822 HandleCodeCompleteResults(this, CodeCompleter,
5823 CodeCompletionContext::CCC_ObjCProtocolName,
5824 Results.data(),Results.size());
Douglas Gregorbaf69612009-11-18 04:19:12 +00005825}
Douglas Gregor49c22a72009-11-18 16:26:39 +00005826
5827/// \brief Add all of the Objective-C interface declarations that we find in
5828/// the given (translation unit) context.
5829static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5830 bool OnlyForwardDeclarations,
5831 bool OnlyUnimplemented,
5832 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005833 typedef CodeCompletionResult Result;
Douglas Gregor49c22a72009-11-18 16:26:39 +00005834
5835 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5836 DEnd = Ctx->decls_end();
5837 D != DEnd; ++D) {
Douglas Gregor1c283312010-08-11 12:19:30 +00005838 // Record any interfaces we find.
5839 if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*D))
Douglas Gregordc9166c2011-12-15 20:29:51 +00005840 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregor1c283312010-08-11 12:19:30 +00005841 (!OnlyUnimplemented || !Class->getImplementation()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005842 Results.AddResult(Result(Class, Results.getBasePriority(Class), 0),
5843 CurContext, 0, false);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005844 }
5845}
5846
5847void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005848 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005849 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005850 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005851 Results.EnterNewScope();
5852
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005853 if (CodeCompleter->includeGlobals()) {
5854 // Add all classes.
5855 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5856 false, Results);
5857 }
5858
Douglas Gregor49c22a72009-11-18 16:26:39 +00005859 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005860
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005861 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005862 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005863 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005864}
5865
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005866void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5867 SourceLocation ClassNameLoc) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005868 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005869 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005870 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005871 Results.EnterNewScope();
5872
5873 // Make sure that we ignore the class we're currently defining.
5874 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005875 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005876 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor49c22a72009-11-18 16:26:39 +00005877 Results.Ignore(CurClass);
5878
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005879 if (CodeCompleter->includeGlobals()) {
5880 // Add all classes.
5881 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5882 false, Results);
5883 }
5884
Douglas Gregor49c22a72009-11-18 16:26:39 +00005885 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005886
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005887 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005888 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005889 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005890}
5891
5892void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005893 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005894 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005895 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005896 Results.EnterNewScope();
5897
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005898 if (CodeCompleter->includeGlobals()) {
5899 // Add all unimplemented classes.
5900 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5901 true, Results);
5902 }
5903
Douglas Gregor49c22a72009-11-18 16:26:39 +00005904 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005905
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005906 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005907 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005908 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005909}
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005910
5911void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005912 IdentifierInfo *ClassName,
5913 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005914 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005915
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005916 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005917 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005918 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005919
5920 // Ignore any categories we find that have already been implemented by this
5921 // interface.
5922 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5923 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005924 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005925 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
5926 for (ObjCInterfaceDecl::visible_categories_iterator
5927 Cat = Class->visible_categories_begin(),
5928 CatEnd = Class->visible_categories_end();
5929 Cat != CatEnd; ++Cat) {
5930 CategoryNames.insert(Cat->getIdentifier());
5931 }
5932 }
5933
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005934 // Add all of the categories we know about.
5935 Results.EnterNewScope();
5936 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
5937 for (DeclContext::decl_iterator D = TU->decls_begin(),
5938 DEnd = TU->decls_end();
5939 D != DEnd; ++D)
5940 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(*D))
5941 if (CategoryNames.insert(Category->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005942 Results.AddResult(Result(Category, Results.getBasePriority(Category),0),
5943 CurContext, 0, false);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005944 Results.ExitScope();
5945
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005946 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005947 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005948 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005949}
5950
5951void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005952 IdentifierInfo *ClassName,
5953 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005954 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005955
5956 // Find the corresponding interface. If we couldn't find the interface, the
5957 // program itself is ill-formed. However, we'll try to be helpful still by
5958 // providing the list of all of the categories we know about.
5959 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005960 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005961 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5962 if (!Class)
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005963 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005964
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005965 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005966 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005967 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005968
5969 // Add all of the categories that have have corresponding interface
5970 // declarations in this class and any of its superclasses, except for
5971 // already-implemented categories in the class itself.
5972 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5973 Results.EnterNewScope();
5974 bool IgnoreImplemented = true;
5975 while (Class) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005976 for (ObjCInterfaceDecl::visible_categories_iterator
5977 Cat = Class->visible_categories_begin(),
5978 CatEnd = Class->visible_categories_end();
5979 Cat != CatEnd; ++Cat) {
5980 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
5981 CategoryNames.insert(Cat->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005982 Results.AddResult(Result(*Cat, Results.getBasePriority(*Cat), 0),
5983 CurContext, 0, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005984 }
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005985
5986 Class = Class->getSuperClass();
5987 IgnoreImplemented = false;
5988 }
5989 Results.ExitScope();
5990
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005991 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005992 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005993 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005994}
Douglas Gregor5d649882009-11-18 22:32:06 +00005995
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005996void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005997 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005998 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005999 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00006000
6001 // Figure out where this @synthesize lives.
6002 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006003 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00006004 if (!Container ||
6005 (!isa<ObjCImplementationDecl>(Container) &&
6006 !isa<ObjCCategoryImplDecl>(Container)))
6007 return;
6008
6009 // Ignore any properties that have already been implemented.
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006010 Container = getContainerDef(Container);
6011 for (DeclContext::decl_iterator D = Container->decls_begin(),
Douglas Gregor5d649882009-11-18 22:32:06 +00006012 DEnd = Container->decls_end();
6013 D != DEnd; ++D)
6014 if (ObjCPropertyImplDecl *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(*D))
6015 Results.Ignore(PropertyImpl->getPropertyDecl());
6016
6017 // Add any properties that we find.
Douglas Gregorb888acf2010-12-09 23:01:55 +00006018 AddedPropertiesSet AddedProperties;
Douglas Gregor5d649882009-11-18 22:32:06 +00006019 Results.EnterNewScope();
6020 if (ObjCImplementationDecl *ClassImpl
6021 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor95147142011-05-05 15:50:42 +00006022 AddObjCProperties(ClassImpl->getClassInterface(), false,
6023 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00006024 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00006025 else
6026 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor95147142011-05-05 15:50:42 +00006027 false, /*AllowNullaryMethods=*/false, CurContext,
6028 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00006029 Results.ExitScope();
6030
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006031 HandleCodeCompleteResults(this, CodeCompleter,
6032 CodeCompletionContext::CCC_Other,
6033 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006034}
6035
6036void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006037 IdentifierInfo *PropertyName) {
John McCall276321a2010-08-25 06:19:51 +00006038 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006039 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006040 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006041 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00006042
6043 // Figure out where this @synthesize lives.
6044 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006045 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00006046 if (!Container ||
6047 (!isa<ObjCImplementationDecl>(Container) &&
6048 !isa<ObjCCategoryImplDecl>(Container)))
6049 return;
6050
6051 // Figure out which interface we're looking into.
6052 ObjCInterfaceDecl *Class = 0;
6053 if (ObjCImplementationDecl *ClassImpl
6054 = dyn_cast<ObjCImplementationDecl>(Container))
6055 Class = ClassImpl->getClassInterface();
6056 else
6057 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
6058 ->getClassInterface();
6059
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006060 // Determine the type of the property we're synthesizing.
6061 QualType PropertyType = Context.getObjCIdType();
6062 if (Class) {
6063 if (ObjCPropertyDecl *Property
6064 = Class->FindPropertyDeclaration(PropertyName)) {
6065 PropertyType
6066 = Property->getType().getNonReferenceType().getUnqualifiedType();
6067
6068 // Give preference to ivars
6069 Results.setPreferredType(PropertyType);
6070 }
6071 }
6072
Douglas Gregor5d649882009-11-18 22:32:06 +00006073 // Add all of the instance variables in this class and its superclasses.
6074 Results.EnterNewScope();
Douglas Gregor331faa02011-04-18 14:13:53 +00006075 bool SawSimilarlyNamedIvar = false;
6076 std::string NameWithPrefix;
6077 NameWithPrefix += '_';
Benjamin Kramer632500c2011-07-26 16:59:25 +00006078 NameWithPrefix += PropertyName->getName();
Douglas Gregor331faa02011-04-18 14:13:53 +00006079 std::string NameWithSuffix = PropertyName->getName().str();
6080 NameWithSuffix += '_';
Douglas Gregor5d649882009-11-18 22:32:06 +00006081 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006082 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
6083 Ivar = Ivar->getNextIvar()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00006084 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), 0),
6085 CurContext, 0, false);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006086
Douglas Gregor331faa02011-04-18 14:13:53 +00006087 // Determine whether we've seen an ivar with a name similar to the
6088 // property.
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006089 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregor331faa02011-04-18 14:13:53 +00006090 NameWithPrefix == Ivar->getName() ||
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006091 NameWithSuffix == Ivar->getName())) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006092 SawSimilarlyNamedIvar = true;
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006093
6094 // Reduce the priority of this result by one, to give it a slight
6095 // advantage over other results whose names don't match so closely.
6096 if (Results.size() &&
6097 Results.data()[Results.size() - 1].Kind
6098 == CodeCompletionResult::RK_Declaration &&
6099 Results.data()[Results.size() - 1].Declaration == Ivar)
6100 Results.data()[Results.size() - 1].Priority--;
6101 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006102 }
Douglas Gregor5d649882009-11-18 22:32:06 +00006103 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006104
6105 if (!SawSimilarlyNamedIvar) {
6106 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006107 // an ivar of the appropriate type.
6108 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregor331faa02011-04-18 14:13:53 +00006109 typedef CodeCompletionResult Result;
6110 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006111 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6112 Priority,CXAvailability_Available);
Douglas Gregor331faa02011-04-18 14:13:53 +00006113
Douglas Gregor75acd922011-09-27 23:30:47 +00006114 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006115 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006116 Policy, Allocator));
Douglas Gregor331faa02011-04-18 14:13:53 +00006117 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6118 Results.AddResult(Result(Builder.TakeString(), Priority,
6119 CXCursor_ObjCIvarDecl));
6120 }
6121
Douglas Gregor5d649882009-11-18 22:32:06 +00006122 Results.ExitScope();
6123
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006124 HandleCodeCompleteResults(this, CodeCompleter,
6125 CodeCompletionContext::CCC_Other,
6126 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006127}
Douglas Gregor636a61e2010-04-07 00:21:17 +00006128
Douglas Gregor416b5752010-08-25 01:08:01 +00006129// Mapping from selectors to the methods that implement that selector, along
6130// with the "in original class" flag.
6131typedef llvm::DenseMap<Selector, std::pair<ObjCMethodDecl *, bool> >
6132 KnownMethodsMap;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006133
6134/// \brief Find all of the methods that reside in the given container
6135/// (and its superclasses, protocols, etc.) that meet the given
6136/// criteria. Insert those methods into the map of known methods,
6137/// indexed by selector so they can be easily found.
6138static void FindImplementableMethods(ASTContext &Context,
6139 ObjCContainerDecl *Container,
6140 bool WantInstanceMethods,
6141 QualType ReturnType,
Douglas Gregor416b5752010-08-25 01:08:01 +00006142 KnownMethodsMap &KnownMethods,
6143 bool InOriginalClass = true) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006144 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006145 // Make sure we have a definition; that's what we'll walk.
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006146 if (!IFace->hasDefinition())
6147 return;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006148
6149 IFace = IFace->getDefinition();
6150 Container = IFace;
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006151
Douglas Gregor636a61e2010-04-07 00:21:17 +00006152 const ObjCList<ObjCProtocolDecl> &Protocols
6153 = IFace->getReferencedProtocols();
6154 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006155 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006156 I != E; ++I)
6157 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006158 KnownMethods, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006159
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006160 // Add methods from any class extensions and categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006161 for (ObjCInterfaceDecl::visible_categories_iterator
6162 Cat = IFace->visible_categories_begin(),
6163 CatEnd = IFace->visible_categories_end();
6164 Cat != CatEnd; ++Cat) {
6165 FindImplementableMethods(Context, *Cat, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006166 KnownMethods, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006167 }
6168
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006169 // Visit the superclass.
6170 if (IFace->getSuperClass())
6171 FindImplementableMethods(Context, IFace->getSuperClass(),
6172 WantInstanceMethods, ReturnType,
6173 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006174 }
6175
6176 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6177 // Recurse into protocols.
6178 const ObjCList<ObjCProtocolDecl> &Protocols
6179 = Category->getReferencedProtocols();
6180 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006181 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006182 I != E; ++I)
6183 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006184 KnownMethods, InOriginalClass);
6185
6186 // If this category is the original class, jump to the interface.
6187 if (InOriginalClass && Category->getClassInterface())
6188 FindImplementableMethods(Context, Category->getClassInterface(),
6189 WantInstanceMethods, ReturnType, KnownMethods,
6190 false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006191 }
6192
6193 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006194 // Make sure we have a definition; that's what we'll walk.
6195 if (!Protocol->hasDefinition())
6196 return;
6197 Protocol = Protocol->getDefinition();
6198 Container = Protocol;
6199
6200 // Recurse into protocols.
6201 const ObjCList<ObjCProtocolDecl> &Protocols
6202 = Protocol->getReferencedProtocols();
6203 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6204 E = Protocols.end();
6205 I != E; ++I)
6206 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6207 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006208 }
6209
6210 // Add methods in this container. This operation occurs last because
6211 // we want the methods from this container to override any methods
6212 // we've previously seen with the same selector.
6213 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
6214 MEnd = Container->meth_end();
6215 M != MEnd; ++M) {
David Blaikie2d7c57e2012-04-30 02:36:29 +00006216 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006217 if (!ReturnType.isNull() &&
David Blaikie2d7c57e2012-04-30 02:36:29 +00006218 !Context.hasSameUnqualifiedType(ReturnType, M->getResultType()))
Douglas Gregor636a61e2010-04-07 00:21:17 +00006219 continue;
6220
David Blaikie40ed2972012-06-06 20:45:41 +00006221 KnownMethods[M->getSelector()] = std::make_pair(*M, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006222 }
6223 }
6224}
6225
Douglas Gregor669a25a2011-02-17 00:22:45 +00006226/// \brief Add the parenthesized return or parameter type chunk to a code
6227/// completion string.
6228static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor29979142012-04-10 18:35:07 +00006229 unsigned ObjCDeclQuals,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006230 ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006231 const PrintingPolicy &Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006232 CodeCompletionBuilder &Builder) {
6233 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor29979142012-04-10 18:35:07 +00006234 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6235 if (!Quals.empty())
6236 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor75acd922011-09-27 23:30:47 +00006237 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006238 Builder.getAllocator()));
6239 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6240}
6241
6242/// \brief Determine whether the given class is or inherits from a class by
6243/// the given name.
6244static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006245 StringRef Name) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006246 if (!Class)
6247 return false;
6248
6249 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6250 return true;
6251
6252 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6253}
6254
6255/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6256/// Key-Value Observing (KVO).
6257static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6258 bool IsInstanceMethod,
6259 QualType ReturnType,
6260 ASTContext &Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006261 VisitedSelectorSet &KnownSelectors,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006262 ResultBuilder &Results) {
6263 IdentifierInfo *PropName = Property->getIdentifier();
6264 if (!PropName || PropName->getLength() == 0)
6265 return;
6266
Douglas Gregor75acd922011-09-27 23:30:47 +00006267 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6268
Douglas Gregor669a25a2011-02-17 00:22:45 +00006269 // Builder that will create each code completion.
6270 typedef CodeCompletionResult Result;
6271 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006272 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor669a25a2011-02-17 00:22:45 +00006273
6274 // The selector table.
6275 SelectorTable &Selectors = Context.Selectors;
6276
6277 // The property name, copied into the code completion allocation region
6278 // on demand.
6279 struct KeyHolder {
6280 CodeCompletionAllocator &Allocator;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006281 StringRef Key;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006282 const char *CopiedKey;
6283
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006284 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006285 : Allocator(Allocator), Key(Key), CopiedKey(0) { }
6286
6287 operator const char *() {
6288 if (CopiedKey)
6289 return CopiedKey;
6290
6291 return CopiedKey = Allocator.CopyString(Key);
6292 }
6293 } Key(Allocator, PropName->getName());
6294
6295 // The uppercased name of the property name.
6296 std::string UpperKey = PropName->getName();
6297 if (!UpperKey.empty())
Jordan Rose4938f272013-02-09 10:09:43 +00006298 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006299
6300 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6301 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6302 Property->getType());
6303 bool ReturnTypeMatchesVoid
6304 = ReturnType.isNull() || ReturnType->isVoidType();
6305
6306 // Add the normal accessor -(type)key.
6307 if (IsInstanceMethod &&
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006308 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor669a25a2011-02-17 00:22:45 +00006309 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6310 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006311 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6312 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006313
6314 Builder.AddTypedTextChunk(Key);
6315 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6316 CXCursor_ObjCInstanceMethodDecl));
6317 }
6318
6319 // If we have an integral or boolean property (or the user has provided
6320 // an integral or boolean return type), add the accessor -(type)isKey.
6321 if (IsInstanceMethod &&
6322 ((!ReturnType.isNull() &&
6323 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6324 (ReturnType.isNull() &&
6325 (Property->getType()->isIntegerType() ||
6326 Property->getType()->isBooleanType())))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006327 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006328 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006329 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006330 if (ReturnType.isNull()) {
6331 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6332 Builder.AddTextChunk("BOOL");
6333 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6334 }
6335
6336 Builder.AddTypedTextChunk(
6337 Allocator.CopyString(SelectorId->getName()));
6338 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6339 CXCursor_ObjCInstanceMethodDecl));
6340 }
6341 }
6342
6343 // Add the normal mutator.
6344 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6345 !Property->getSetterMethodDecl()) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006346 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006347 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006348 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006349 if (ReturnType.isNull()) {
6350 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6351 Builder.AddTextChunk("void");
6352 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6353 }
6354
6355 Builder.AddTypedTextChunk(
6356 Allocator.CopyString(SelectorId->getName()));
6357 Builder.AddTypedTextChunk(":");
Douglas Gregor29979142012-04-10 18:35:07 +00006358 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6359 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006360 Builder.AddTextChunk(Key);
6361 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6362 CXCursor_ObjCInstanceMethodDecl));
6363 }
6364 }
6365
6366 // Indexed and unordered accessors
6367 unsigned IndexedGetterPriority = CCP_CodePattern;
6368 unsigned IndexedSetterPriority = CCP_CodePattern;
6369 unsigned UnorderedGetterPriority = CCP_CodePattern;
6370 unsigned UnorderedSetterPriority = CCP_CodePattern;
6371 if (const ObjCObjectPointerType *ObjCPointer
6372 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6373 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6374 // If this interface type is not provably derived from a known
6375 // collection, penalize the corresponding completions.
6376 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6377 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6378 if (!InheritsFromClassNamed(IFace, "NSArray"))
6379 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6380 }
6381
6382 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6383 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6384 if (!InheritsFromClassNamed(IFace, "NSSet"))
6385 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6386 }
6387 }
6388 } else {
6389 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6390 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6391 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6392 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6393 }
6394
6395 // Add -(NSUInteger)countOf<key>
6396 if (IsInstanceMethod &&
6397 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006398 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006399 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006400 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006401 if (ReturnType.isNull()) {
6402 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6403 Builder.AddTextChunk("NSUInteger");
6404 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6405 }
6406
6407 Builder.AddTypedTextChunk(
6408 Allocator.CopyString(SelectorId->getName()));
6409 Results.AddResult(Result(Builder.TakeString(),
6410 std::min(IndexedGetterPriority,
6411 UnorderedGetterPriority),
6412 CXCursor_ObjCInstanceMethodDecl));
6413 }
6414 }
6415
6416 // Indexed getters
6417 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6418 if (IsInstanceMethod &&
6419 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006420 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006421 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006422 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006423 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006424 if (ReturnType.isNull()) {
6425 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6426 Builder.AddTextChunk("id");
6427 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6428 }
6429
6430 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6431 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6432 Builder.AddTextChunk("NSUInteger");
6433 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6434 Builder.AddTextChunk("index");
6435 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6436 CXCursor_ObjCInstanceMethodDecl));
6437 }
6438 }
6439
6440 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6441 if (IsInstanceMethod &&
6442 (ReturnType.isNull() ||
6443 (ReturnType->isObjCObjectPointerType() &&
6444 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6445 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6446 ->getName() == "NSArray"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006447 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006448 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006449 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006450 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006451 if (ReturnType.isNull()) {
6452 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6453 Builder.AddTextChunk("NSArray *");
6454 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6455 }
6456
6457 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6458 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6459 Builder.AddTextChunk("NSIndexSet *");
6460 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6461 Builder.AddTextChunk("indexes");
6462 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6463 CXCursor_ObjCInstanceMethodDecl));
6464 }
6465 }
6466
6467 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6468 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006469 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006470 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006471 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006472 &Context.Idents.get("range")
6473 };
6474
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006475 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006476 if (ReturnType.isNull()) {
6477 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6478 Builder.AddTextChunk("void");
6479 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6480 }
6481
6482 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6483 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6484 Builder.AddPlaceholderChunk("object-type");
6485 Builder.AddTextChunk(" **");
6486 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6487 Builder.AddTextChunk("buffer");
6488 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6489 Builder.AddTypedTextChunk("range:");
6490 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6491 Builder.AddTextChunk("NSRange");
6492 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6493 Builder.AddTextChunk("inRange");
6494 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6495 CXCursor_ObjCInstanceMethodDecl));
6496 }
6497 }
6498
6499 // Mutable indexed accessors
6500
6501 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6502 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006503 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006504 IdentifierInfo *SelectorIds[2] = {
6505 &Context.Idents.get("insertObject"),
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006506 &Context.Idents.get(SelectorName)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006507 };
6508
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006509 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006510 if (ReturnType.isNull()) {
6511 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6512 Builder.AddTextChunk("void");
6513 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6514 }
6515
6516 Builder.AddTypedTextChunk("insertObject:");
6517 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6518 Builder.AddPlaceholderChunk("object-type");
6519 Builder.AddTextChunk(" *");
6520 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6521 Builder.AddTextChunk("object");
6522 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6523 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6524 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6525 Builder.AddPlaceholderChunk("NSUInteger");
6526 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6527 Builder.AddTextChunk("index");
6528 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6529 CXCursor_ObjCInstanceMethodDecl));
6530 }
6531 }
6532
6533 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6534 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006535 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006536 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006537 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006538 &Context.Idents.get("atIndexes")
6539 };
6540
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006541 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006542 if (ReturnType.isNull()) {
6543 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6544 Builder.AddTextChunk("void");
6545 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6546 }
6547
6548 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6549 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6550 Builder.AddTextChunk("NSArray *");
6551 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6552 Builder.AddTextChunk("array");
6553 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6554 Builder.AddTypedTextChunk("atIndexes:");
6555 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6556 Builder.AddPlaceholderChunk("NSIndexSet *");
6557 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6558 Builder.AddTextChunk("indexes");
6559 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6560 CXCursor_ObjCInstanceMethodDecl));
6561 }
6562 }
6563
6564 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6565 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006566 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006567 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006568 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006569 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006570 if (ReturnType.isNull()) {
6571 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6572 Builder.AddTextChunk("void");
6573 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6574 }
6575
6576 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6577 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6578 Builder.AddTextChunk("NSUInteger");
6579 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6580 Builder.AddTextChunk("index");
6581 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6582 CXCursor_ObjCInstanceMethodDecl));
6583 }
6584 }
6585
6586 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6587 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006588 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006589 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006590 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006591 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006592 if (ReturnType.isNull()) {
6593 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6594 Builder.AddTextChunk("void");
6595 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6596 }
6597
6598 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6599 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6600 Builder.AddTextChunk("NSIndexSet *");
6601 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6602 Builder.AddTextChunk("indexes");
6603 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6604 CXCursor_ObjCInstanceMethodDecl));
6605 }
6606 }
6607
6608 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6609 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006610 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006611 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006612 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006613 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006614 &Context.Idents.get("withObject")
6615 };
6616
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006617 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006618 if (ReturnType.isNull()) {
6619 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6620 Builder.AddTextChunk("void");
6621 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6622 }
6623
6624 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6625 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6626 Builder.AddPlaceholderChunk("NSUInteger");
6627 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6628 Builder.AddTextChunk("index");
6629 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6630 Builder.AddTypedTextChunk("withObject:");
6631 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6632 Builder.AddTextChunk("id");
6633 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6634 Builder.AddTextChunk("object");
6635 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6636 CXCursor_ObjCInstanceMethodDecl));
6637 }
6638 }
6639
6640 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6641 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006642 std::string SelectorName1
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006643 = (Twine("replace") + UpperKey + "AtIndexes").str();
6644 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006645 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006646 &Context.Idents.get(SelectorName1),
6647 &Context.Idents.get(SelectorName2)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006648 };
6649
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006650 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006651 if (ReturnType.isNull()) {
6652 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6653 Builder.AddTextChunk("void");
6654 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6655 }
6656
6657 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6658 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6659 Builder.AddPlaceholderChunk("NSIndexSet *");
6660 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6661 Builder.AddTextChunk("indexes");
6662 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6663 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6664 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6665 Builder.AddTextChunk("NSArray *");
6666 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6667 Builder.AddTextChunk("array");
6668 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6669 CXCursor_ObjCInstanceMethodDecl));
6670 }
6671 }
6672
6673 // Unordered getters
6674 // - (NSEnumerator *)enumeratorOfKey
6675 if (IsInstanceMethod &&
6676 (ReturnType.isNull() ||
6677 (ReturnType->isObjCObjectPointerType() &&
6678 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6679 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6680 ->getName() == "NSEnumerator"))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006681 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006682 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006683 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006684 if (ReturnType.isNull()) {
6685 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6686 Builder.AddTextChunk("NSEnumerator *");
6687 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6688 }
6689
6690 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6691 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6692 CXCursor_ObjCInstanceMethodDecl));
6693 }
6694 }
6695
6696 // - (type *)memberOfKey:(type *)object
6697 if (IsInstanceMethod &&
6698 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006699 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006700 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006701 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006702 if (ReturnType.isNull()) {
6703 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6704 Builder.AddPlaceholderChunk("object-type");
6705 Builder.AddTextChunk(" *");
6706 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6707 }
6708
6709 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6710 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6711 if (ReturnType.isNull()) {
6712 Builder.AddPlaceholderChunk("object-type");
6713 Builder.AddTextChunk(" *");
6714 } else {
6715 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006716 Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006717 Builder.getAllocator()));
6718 }
6719 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6720 Builder.AddTextChunk("object");
6721 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6722 CXCursor_ObjCInstanceMethodDecl));
6723 }
6724 }
6725
6726 // Mutable unordered accessors
6727 // - (void)addKeyObject:(type *)object
6728 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006729 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006730 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006731 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006732 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006733 if (ReturnType.isNull()) {
6734 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6735 Builder.AddTextChunk("void");
6736 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6737 }
6738
6739 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6740 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6741 Builder.AddPlaceholderChunk("object-type");
6742 Builder.AddTextChunk(" *");
6743 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6744 Builder.AddTextChunk("object");
6745 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6746 CXCursor_ObjCInstanceMethodDecl));
6747 }
6748 }
6749
6750 // - (void)addKey:(NSSet *)objects
6751 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006752 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006753 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006754 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006755 if (ReturnType.isNull()) {
6756 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6757 Builder.AddTextChunk("void");
6758 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6759 }
6760
6761 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6762 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6763 Builder.AddTextChunk("NSSet *");
6764 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6765 Builder.AddTextChunk("objects");
6766 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6767 CXCursor_ObjCInstanceMethodDecl));
6768 }
6769 }
6770
6771 // - (void)removeKeyObject:(type *)object
6772 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006773 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006774 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006775 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006776 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006777 if (ReturnType.isNull()) {
6778 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6779 Builder.AddTextChunk("void");
6780 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6781 }
6782
6783 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6784 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6785 Builder.AddPlaceholderChunk("object-type");
6786 Builder.AddTextChunk(" *");
6787 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6788 Builder.AddTextChunk("object");
6789 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6790 CXCursor_ObjCInstanceMethodDecl));
6791 }
6792 }
6793
6794 // - (void)removeKey:(NSSet *)objects
6795 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006796 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006797 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006798 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006799 if (ReturnType.isNull()) {
6800 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6801 Builder.AddTextChunk("void");
6802 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6803 }
6804
6805 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6806 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6807 Builder.AddTextChunk("NSSet *");
6808 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6809 Builder.AddTextChunk("objects");
6810 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6811 CXCursor_ObjCInstanceMethodDecl));
6812 }
6813 }
6814
6815 // - (void)intersectKey:(NSSet *)objects
6816 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006817 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006818 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006819 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006820 if (ReturnType.isNull()) {
6821 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6822 Builder.AddTextChunk("void");
6823 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6824 }
6825
6826 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6827 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6828 Builder.AddTextChunk("NSSet *");
6829 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6830 Builder.AddTextChunk("objects");
6831 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6832 CXCursor_ObjCInstanceMethodDecl));
6833 }
6834 }
6835
6836 // Key-Value Observing
6837 // + (NSSet *)keyPathsForValuesAffectingKey
6838 if (!IsInstanceMethod &&
6839 (ReturnType.isNull() ||
6840 (ReturnType->isObjCObjectPointerType() &&
6841 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6842 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6843 ->getName() == "NSSet"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006844 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006845 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006846 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006847 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006848 if (ReturnType.isNull()) {
6849 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6850 Builder.AddTextChunk("NSSet *");
6851 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6852 }
6853
6854 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6855 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor857bcda2011-06-02 04:02:27 +00006856 CXCursor_ObjCClassMethodDecl));
6857 }
6858 }
6859
6860 // + (BOOL)automaticallyNotifiesObserversForKey
6861 if (!IsInstanceMethod &&
6862 (ReturnType.isNull() ||
6863 ReturnType->isIntegerType() ||
6864 ReturnType->isBooleanType())) {
6865 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006866 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor857bcda2011-06-02 04:02:27 +00006867 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6868 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6869 if (ReturnType.isNull()) {
6870 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6871 Builder.AddTextChunk("BOOL");
6872 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6873 }
6874
6875 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6876 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6877 CXCursor_ObjCClassMethodDecl));
Douglas Gregor669a25a2011-02-17 00:22:45 +00006878 }
6879 }
6880}
6881
Douglas Gregor636a61e2010-04-07 00:21:17 +00006882void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6883 bool IsInstanceMethod,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006884 ParsedType ReturnTy) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006885 // Determine the return type of the method we're declaring, if
6886 // provided.
6887 QualType ReturnType = GetTypeFromParser(ReturnTy);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006888 Decl *IDecl = 0;
6889 if (CurContext->isObjCContainer()) {
6890 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6891 IDecl = cast<Decl>(OCD);
6892 }
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006893 // Determine where we should start searching for methods.
6894 ObjCContainerDecl *SearchDecl = 0;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006895 bool IsInImplementation = false;
John McCall48871652010-08-21 09:40:31 +00006896 if (Decl *D = IDecl) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006897 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6898 SearchDecl = Impl->getClassInterface();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006899 IsInImplementation = true;
6900 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006901 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006902 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006903 IsInImplementation = true;
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006904 } else
Douglas Gregor636a61e2010-04-07 00:21:17 +00006905 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006906 }
6907
6908 if (!SearchDecl && S) {
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006909 if (DeclContext *DC = static_cast<DeclContext *>(S->getEntity()))
Douglas Gregor636a61e2010-04-07 00:21:17 +00006910 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006911 }
6912
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006913 if (!SearchDecl) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006914 HandleCodeCompleteResults(this, CodeCompleter,
6915 CodeCompletionContext::CCC_Other,
6916 0, 0);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006917 return;
6918 }
6919
6920 // Find all of the methods that we could declare/implement here.
6921 KnownMethodsMap KnownMethods;
6922 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006923 ReturnType, KnownMethods);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006924
Douglas Gregor636a61e2010-04-07 00:21:17 +00006925 // Add declarations or definitions for each of the known methods.
John McCall276321a2010-08-25 06:19:51 +00006926 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006927 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006928 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006929 CodeCompletionContext::CCC_Other);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006930 Results.EnterNewScope();
Douglas Gregor75acd922011-09-27 23:30:47 +00006931 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006932 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6933 MEnd = KnownMethods.end();
6934 M != MEnd; ++M) {
Douglas Gregor416b5752010-08-25 01:08:01 +00006935 ObjCMethodDecl *Method = M->second.first;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006936 CodeCompletionBuilder Builder(Results.getAllocator(),
6937 Results.getCodeCompletionTUInfo());
Douglas Gregor636a61e2010-04-07 00:21:17 +00006938
6939 // If the result type was not already provided, add it to the
6940 // pattern as (type).
Douglas Gregor669a25a2011-02-17 00:22:45 +00006941 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006942 AddObjCPassingTypeChunk(Method->getResultType(),
6943 Method->getObjCDeclQualifier(),
6944 Context, Policy,
6945 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006946
6947 Selector Sel = Method->getSelector();
6948
6949 // Add the first part of the selector to the pattern.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006950 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006951 Sel.getNameForSlot(0)));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006952
6953 // Add parameters to the pattern.
6954 unsigned I = 0;
6955 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6956 PEnd = Method->param_end();
6957 P != PEnd; (void)++P, ++I) {
6958 // Add the part of the selector name.
6959 if (I == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006960 Builder.AddTypedTextChunk(":");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006961 else if (I < Sel.getNumArgs()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006962 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6963 Builder.AddTypedTextChunk(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006964 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006965 } else
6966 break;
6967
6968 // Add the parameter type.
Douglas Gregor29979142012-04-10 18:35:07 +00006969 AddObjCPassingTypeChunk((*P)->getOriginalType(),
6970 (*P)->getObjCDeclQualifier(),
6971 Context, Policy,
Douglas Gregor75acd922011-09-27 23:30:47 +00006972 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006973
6974 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006975 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006976 }
6977
6978 if (Method->isVariadic()) {
6979 if (Method->param_size() > 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006980 Builder.AddChunk(CodeCompletionString::CK_Comma);
6981 Builder.AddTextChunk("...");
Douglas Gregor400f5972010-08-31 05:13:43 +00006982 }
Douglas Gregor636a61e2010-04-07 00:21:17 +00006983
Douglas Gregord37c59d2010-05-28 00:57:46 +00006984 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006985 // We will be defining the method here, so add a compound statement.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006986 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6987 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6988 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006989 if (!Method->getResultType()->isVoidType()) {
6990 // If the result type is not void, add a return clause.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006991 Builder.AddTextChunk("return");
6992 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6993 Builder.AddPlaceholderChunk("expression");
6994 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006995 } else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006996 Builder.AddPlaceholderChunk("statements");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006997
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006998 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6999 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00007000 }
7001
Douglas Gregor416b5752010-08-25 01:08:01 +00007002 unsigned Priority = CCP_CodePattern;
7003 if (!M->second.second)
7004 Priority += CCD_InBaseClass;
7005
Douglas Gregor78254c82012-03-27 23:34:16 +00007006 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregor636a61e2010-04-07 00:21:17 +00007007 }
7008
Douglas Gregor669a25a2011-02-17 00:22:45 +00007009 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
7010 // the properties in this class and its categories.
David Blaikiebbafb8a2012-03-11 07:00:24 +00007011 if (Context.getLangOpts().ObjC2) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00007012 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor669a25a2011-02-17 00:22:45 +00007013 Containers.push_back(SearchDecl);
7014
Douglas Gregord4a8ced2011-05-04 23:50:46 +00007015 VisitedSelectorSet KnownSelectors;
7016 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
7017 MEnd = KnownMethods.end();
7018 M != MEnd; ++M)
7019 KnownSelectors.insert(M->first);
7020
7021
Douglas Gregor669a25a2011-02-17 00:22:45 +00007022 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
7023 if (!IFace)
7024 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
7025 IFace = Category->getClassInterface();
7026
7027 if (IFace) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00007028 for (ObjCInterfaceDecl::visible_categories_iterator
7029 Cat = IFace->visible_categories_begin(),
7030 CatEnd = IFace->visible_categories_end();
7031 Cat != CatEnd; ++Cat) {
7032 Containers.push_back(*Cat);
7033 }
Douglas Gregor669a25a2011-02-17 00:22:45 +00007034 }
7035
7036 for (unsigned I = 0, N = Containers.size(); I != N; ++I) {
7037 for (ObjCContainerDecl::prop_iterator P = Containers[I]->prop_begin(),
7038 PEnd = Containers[I]->prop_end();
7039 P != PEnd; ++P) {
David Blaikie40ed2972012-06-06 20:45:41 +00007040 AddObjCKeyValueCompletions(*P, IsInstanceMethod, ReturnType, Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00007041 KnownSelectors, Results);
Douglas Gregor669a25a2011-02-17 00:22:45 +00007042 }
7043 }
7044 }
7045
Douglas Gregor636a61e2010-04-07 00:21:17 +00007046 Results.ExitScope();
7047
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007048 HandleCodeCompleteResults(this, CodeCompleter,
7049 CodeCompletionContext::CCC_Other,
7050 Results.data(),Results.size());
Douglas Gregor636a61e2010-04-07 00:21:17 +00007051}
Douglas Gregor95887f92010-07-08 23:20:03 +00007052
7053void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
7054 bool IsInstanceMethod,
Douglas Gregor45879692010-07-08 23:37:41 +00007055 bool AtParameterName,
John McCallba7bf592010-08-24 05:47:05 +00007056 ParsedType ReturnTy,
Douglas Gregor95887f92010-07-08 23:20:03 +00007057 IdentifierInfo **SelIdents,
7058 unsigned NumSelIdents) {
Douglas Gregor95887f92010-07-08 23:20:03 +00007059 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00007060 // pool from the AST file.
Douglas Gregor95887f92010-07-08 23:20:03 +00007061 if (ExternalSource) {
7062 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
7063 I != N; ++I) {
7064 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00007065 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor95887f92010-07-08 23:20:03 +00007066 continue;
Sebastian Redl75d8a322010-08-02 23:18:59 +00007067
7068 ReadMethodPool(Sel);
Douglas Gregor95887f92010-07-08 23:20:03 +00007069 }
7070 }
7071
7072 // Build the set of methods we can see.
John McCall276321a2010-08-25 06:19:51 +00007073 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007074 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007075 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007076 CodeCompletionContext::CCC_Other);
Douglas Gregor95887f92010-07-08 23:20:03 +00007077
7078 if (ReturnTy)
7079 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redl75d8a322010-08-02 23:18:59 +00007080
Douglas Gregor95887f92010-07-08 23:20:03 +00007081 Results.EnterNewScope();
Sebastian Redl75d8a322010-08-02 23:18:59 +00007082 for (GlobalMethodPool::iterator M = MethodPool.begin(),
7083 MEnd = MethodPool.end();
7084 M != MEnd; ++M) {
7085 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
7086 &M->second.second;
7087 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00007088 MethList = MethList->getNext()) {
Douglas Gregor95887f92010-07-08 23:20:03 +00007089 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
7090 NumSelIdents))
7091 continue;
7092
Douglas Gregor45879692010-07-08 23:37:41 +00007093 if (AtParameterName) {
7094 // Suggest parameter names we've seen before.
7095 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
7096 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
7097 if (Param->getIdentifier()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007098 CodeCompletionBuilder Builder(Results.getAllocator(),
7099 Results.getCodeCompletionTUInfo());
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007100 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007101 Param->getIdentifier()->getName()));
7102 Results.AddResult(Builder.TakeString());
Douglas Gregor45879692010-07-08 23:37:41 +00007103 }
7104 }
7105
7106 continue;
7107 }
7108
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00007109 Result R(MethList->Method, Results.getBasePriority(MethList->Method), 0);
Douglas Gregor95887f92010-07-08 23:20:03 +00007110 R.StartParameter = NumSelIdents;
7111 R.AllParametersAreInformative = false;
7112 R.DeclaringEntity = true;
7113 Results.MaybeAddResult(R, CurContext);
7114 }
7115 }
7116
7117 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007118 HandleCodeCompleteResults(this, CodeCompleter,
7119 CodeCompletionContext::CCC_Other,
7120 Results.data(),Results.size());
Douglas Gregor95887f92010-07-08 23:20:03 +00007121}
Douglas Gregorb14904c2010-08-13 22:48:40 +00007122
Douglas Gregorec00a262010-08-24 22:20:20 +00007123void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007124 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007125 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007126 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007127 Results.EnterNewScope();
7128
7129 // #if <condition>
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007130 CodeCompletionBuilder Builder(Results.getAllocator(),
7131 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007132 Builder.AddTypedTextChunk("if");
7133 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7134 Builder.AddPlaceholderChunk("condition");
7135 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007136
7137 // #ifdef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007138 Builder.AddTypedTextChunk("ifdef");
7139 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7140 Builder.AddPlaceholderChunk("macro");
7141 Results.AddResult(Builder.TakeString());
7142
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007143 // #ifndef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007144 Builder.AddTypedTextChunk("ifndef");
7145 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7146 Builder.AddPlaceholderChunk("macro");
7147 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007148
7149 if (InConditional) {
7150 // #elif <condition>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007151 Builder.AddTypedTextChunk("elif");
7152 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7153 Builder.AddPlaceholderChunk("condition");
7154 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007155
7156 // #else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007157 Builder.AddTypedTextChunk("else");
7158 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007159
7160 // #endif
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007161 Builder.AddTypedTextChunk("endif");
7162 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007163 }
7164
7165 // #include "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007166 Builder.AddTypedTextChunk("include");
7167 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7168 Builder.AddTextChunk("\"");
7169 Builder.AddPlaceholderChunk("header");
7170 Builder.AddTextChunk("\"");
7171 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007172
7173 // #include <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007174 Builder.AddTypedTextChunk("include");
7175 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7176 Builder.AddTextChunk("<");
7177 Builder.AddPlaceholderChunk("header");
7178 Builder.AddTextChunk(">");
7179 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007180
7181 // #define <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007182 Builder.AddTypedTextChunk("define");
7183 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7184 Builder.AddPlaceholderChunk("macro");
7185 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007186
7187 // #define <macro>(<args>)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007188 Builder.AddTypedTextChunk("define");
7189 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7190 Builder.AddPlaceholderChunk("macro");
7191 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7192 Builder.AddPlaceholderChunk("args");
7193 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7194 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007195
7196 // #undef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007197 Builder.AddTypedTextChunk("undef");
7198 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7199 Builder.AddPlaceholderChunk("macro");
7200 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007201
7202 // #line <number>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007203 Builder.AddTypedTextChunk("line");
7204 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7205 Builder.AddPlaceholderChunk("number");
7206 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007207
7208 // #line <number> "filename"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007209 Builder.AddTypedTextChunk("line");
7210 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7211 Builder.AddPlaceholderChunk("number");
7212 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7213 Builder.AddTextChunk("\"");
7214 Builder.AddPlaceholderChunk("filename");
7215 Builder.AddTextChunk("\"");
7216 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007217
7218 // #error <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007219 Builder.AddTypedTextChunk("error");
7220 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7221 Builder.AddPlaceholderChunk("message");
7222 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007223
7224 // #pragma <arguments>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007225 Builder.AddTypedTextChunk("pragma");
7226 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7227 Builder.AddPlaceholderChunk("arguments");
7228 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007229
David Blaikiebbafb8a2012-03-11 07:00:24 +00007230 if (getLangOpts().ObjC1) {
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007231 // #import "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007232 Builder.AddTypedTextChunk("import");
7233 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7234 Builder.AddTextChunk("\"");
7235 Builder.AddPlaceholderChunk("header");
7236 Builder.AddTextChunk("\"");
7237 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007238
7239 // #import <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007240 Builder.AddTypedTextChunk("import");
7241 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7242 Builder.AddTextChunk("<");
7243 Builder.AddPlaceholderChunk("header");
7244 Builder.AddTextChunk(">");
7245 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007246 }
7247
7248 // #include_next "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007249 Builder.AddTypedTextChunk("include_next");
7250 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7251 Builder.AddTextChunk("\"");
7252 Builder.AddPlaceholderChunk("header");
7253 Builder.AddTextChunk("\"");
7254 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007255
7256 // #include_next <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007257 Builder.AddTypedTextChunk("include_next");
7258 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7259 Builder.AddTextChunk("<");
7260 Builder.AddPlaceholderChunk("header");
7261 Builder.AddTextChunk(">");
7262 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007263
7264 // #warning <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007265 Builder.AddTypedTextChunk("warning");
7266 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7267 Builder.AddPlaceholderChunk("message");
7268 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007269
7270 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7271 // completions for them. And __include_macros is a Clang-internal extension
7272 // that we don't want to encourage anyone to use.
7273
7274 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7275 Results.ExitScope();
7276
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007277 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0de55ce2010-08-25 18:41:16 +00007278 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007279 Results.data(), Results.size());
7280}
7281
7282void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorec00a262010-08-24 22:20:20 +00007283 CodeCompleteOrdinaryName(S,
John McCallfaf5fb42010-08-26 23:41:50 +00007284 S->getFnParent()? Sema::PCC_RecoveryInFunction
7285 : Sema::PCC_Namespace);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007286}
7287
Douglas Gregorec00a262010-08-24 22:20:20 +00007288void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007289 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007290 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007291 IsDefinition? CodeCompletionContext::CCC_MacroName
7292 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor12785102010-08-24 20:21:13 +00007293 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7294 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007295 CodeCompletionBuilder Builder(Results.getAllocator(),
7296 Results.getCodeCompletionTUInfo());
Douglas Gregor12785102010-08-24 20:21:13 +00007297 Results.EnterNewScope();
7298 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7299 MEnd = PP.macro_end();
7300 M != MEnd; ++M) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007301 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007302 M->first->getName()));
Argyrios Kyrtzidis5c8b1cd2012-09-27 00:24:09 +00007303 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7304 CCP_CodePattern,
7305 CXCursor_MacroDefinition));
Douglas Gregor12785102010-08-24 20:21:13 +00007306 }
7307 Results.ExitScope();
7308 } else if (IsDefinition) {
7309 // FIXME: Can we detect when the user just wrote an include guard above?
7310 }
7311
Douglas Gregor0ac41382010-09-23 23:01:17 +00007312 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor12785102010-08-24 20:21:13 +00007313 Results.data(), Results.size());
7314}
7315
Douglas Gregorec00a262010-08-24 22:20:20 +00007316void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007317 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007318 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007319 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorec00a262010-08-24 22:20:20 +00007320
7321 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007322 AddMacroResults(PP, Results, true);
Douglas Gregorec00a262010-08-24 22:20:20 +00007323
7324 // defined (<macro>)
7325 Results.EnterNewScope();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007326 CodeCompletionBuilder Builder(Results.getAllocator(),
7327 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007328 Builder.AddTypedTextChunk("defined");
7329 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7330 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7331 Builder.AddPlaceholderChunk("macro");
7332 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7333 Results.AddResult(Builder.TakeString());
Douglas Gregorec00a262010-08-24 22:20:20 +00007334 Results.ExitScope();
7335
7336 HandleCodeCompleteResults(this, CodeCompleter,
7337 CodeCompletionContext::CCC_PreprocessorExpression,
7338 Results.data(), Results.size());
7339}
7340
7341void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7342 IdentifierInfo *Macro,
7343 MacroInfo *MacroInfo,
7344 unsigned Argument) {
7345 // FIXME: In the future, we could provide "overload" results, much like we
7346 // do for function calls.
7347
Argyrios Kyrtzidis75f6cd22011-08-18 19:41:28 +00007348 // Now just ignore this. There will be another code-completion callback
7349 // for the expanded tokens.
Douglas Gregorec00a262010-08-24 22:20:20 +00007350}
7351
Douglas Gregor11583702010-08-25 17:04:25 +00007352void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor11583702010-08-25 17:04:25 +00007353 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorea736372010-08-25 17:10:00 +00007354 CodeCompletionContext::CCC_NaturalLanguage,
Douglas Gregor11583702010-08-25 17:04:25 +00007355 0, 0);
7356}
7357
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007358void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007359 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00007360 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007361 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7362 CodeCompletionContext::CCC_Recovery);
Douglas Gregor39982192010-08-15 06:18:01 +00007363 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7364 CodeCompletionDeclConsumer Consumer(Builder,
7365 Context.getTranslationUnitDecl());
7366 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7367 Consumer);
7368 }
Douglas Gregorb14904c2010-08-13 22:48:40 +00007369
7370 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007371 AddMacroResults(PP, Builder, true);
Douglas Gregorb14904c2010-08-13 22:48:40 +00007372
7373 Results.clear();
7374 Results.insert(Results.end(),
7375 Builder.data(), Builder.data() + Builder.size());
7376}