blob: 6a0c4c4d04ad73b2e33fff9d7580abaa02e7a8b0 [file] [log] [blame]
Douglas Gregor2436e712009-09-17 21:32:03 +00001//===---------------- SemaCodeComplete.cpp - Code Completion ----*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file defines the code-completion semantic actions.
11//
12//===----------------------------------------------------------------------===//
John McCall83024632010-08-25 22:03:47 +000013#include "clang/Sema/SemaInternal.h"
John McCallde6836a2010-08-24 07:21:54 +000014#include "clang/AST/DeclObjC.h"
Douglas Gregorf2510672009-09-21 19:57:38 +000015#include "clang/AST/ExprCXX.h"
Douglas Gregor8ce33212009-11-17 17:59:40 +000016#include "clang/AST/ExprObjC.h"
Jordan Rose4938f272013-02-09 10:09:43 +000017#include "clang/Basic/CharInfo.h"
Douglas Gregor07f43572012-01-29 18:15:03 +000018#include "clang/Lex/HeaderSearch.h"
Douglas Gregorf329c7c2009-10-30 16:50:04 +000019#include "clang/Lex/MacroInfo.h"
20#include "clang/Lex/Preprocessor.h"
Chandler Carruth3a022472012-12-04 09:13:33 +000021#include "clang/Sema/CodeCompleteConsumer.h"
22#include "clang/Sema/ExternalSemaSource.h"
23#include "clang/Sema/Lookup.h"
24#include "clang/Sema/Overload.h"
25#include "clang/Sema/Scope.h"
26#include "clang/Sema/ScopeInfo.h"
Douglas Gregor1154e272010-09-16 16:06:31 +000027#include "llvm/ADT/DenseSet.h"
Benjamin Kramere0513cb2012-01-30 16:17:39 +000028#include "llvm/ADT/SmallBitVector.h"
Douglas Gregor3545ff42009-09-21 16:56:56 +000029#include "llvm/ADT/SmallPtrSet.h"
Benjamin Kramer49038022012-02-04 13:45:25 +000030#include "llvm/ADT/SmallString.h"
Douglas Gregore6688e62009-09-28 03:51:44 +000031#include "llvm/ADT/StringExtras.h"
Douglas Gregor9d2ddb22010-04-06 19:22:33 +000032#include "llvm/ADT/StringSwitch.h"
Douglas Gregor67c692c2010-08-26 15:07:07 +000033#include "llvm/ADT/Twine.h"
Douglas Gregor3545ff42009-09-21 16:56:56 +000034#include <list>
35#include <map>
36#include <vector>
Douglas Gregor2436e712009-09-17 21:32:03 +000037
38using namespace clang;
John McCallaab3e412010-08-25 08:40:02 +000039using namespace sema;
Douglas Gregor2436e712009-09-17 21:32:03 +000040
Douglas Gregor3545ff42009-09-21 16:56:56 +000041namespace {
42 /// \brief A container of code-completion results.
43 class ResultBuilder {
44 public:
45 /// \brief The type of a name-lookup filter, which can be provided to the
46 /// name-lookup routines to specify which declarations should be included in
47 /// the result set (when it returns true) and which declarations should be
48 /// filtered out (returns false).
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000049 typedef bool (ResultBuilder::*LookupFilter)(const NamedDecl *) const;
Douglas Gregor3545ff42009-09-21 16:56:56 +000050
John McCall276321a2010-08-25 06:19:51 +000051 typedef CodeCompletionResult Result;
Douglas Gregor3545ff42009-09-21 16:56:56 +000052
53 private:
54 /// \brief The actual results we have found.
55 std::vector<Result> Results;
56
57 /// \brief A record of all of the declarations we have found and placed
58 /// into the result set, used to ensure that no declaration ever gets into
59 /// the result set twice.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000060 llvm::SmallPtrSet<const Decl*, 16> AllDeclsFound;
Douglas Gregor3545ff42009-09-21 16:56:56 +000061
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000062 typedef std::pair<const NamedDecl *, unsigned> DeclIndexPair;
Douglas Gregor05e7ca32009-12-06 20:23:50 +000063
64 /// \brief An entry in the shadow map, which is optimized to store
65 /// a single (declaration, index) mapping (the common case) but
66 /// can also store a list of (declaration, index) mappings.
67 class ShadowMapEntry {
Chris Lattner0e62c1c2011-07-23 10:55:15 +000068 typedef SmallVector<DeclIndexPair, 4> DeclIndexPairVector;
Douglas Gregor05e7ca32009-12-06 20:23:50 +000069
70 /// \brief Contains either the solitary NamedDecl * or a vector
71 /// of (declaration, index) pairs.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000072 llvm::PointerUnion<const NamedDecl *, DeclIndexPairVector*> DeclOrVector;
Douglas Gregor05e7ca32009-12-06 20:23:50 +000073
74 /// \brief When the entry contains a single declaration, this is
75 /// the index associated with that entry.
76 unsigned SingleDeclIndex;
77
78 public:
79 ShadowMapEntry() : DeclOrVector(), SingleDeclIndex(0) { }
80
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000081 void Add(const NamedDecl *ND, unsigned Index) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +000082 if (DeclOrVector.isNull()) {
83 // 0 - > 1 elements: just set the single element information.
84 DeclOrVector = ND;
85 SingleDeclIndex = Index;
86 return;
87 }
88
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +000089 if (const NamedDecl *PrevND =
90 DeclOrVector.dyn_cast<const NamedDecl *>()) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +000091 // 1 -> 2 elements: create the vector of results and push in the
92 // existing declaration.
93 DeclIndexPairVector *Vec = new DeclIndexPairVector;
94 Vec->push_back(DeclIndexPair(PrevND, SingleDeclIndex));
95 DeclOrVector = Vec;
96 }
97
98 // Add the new element to the end of the vector.
99 DeclOrVector.get<DeclIndexPairVector*>()->push_back(
100 DeclIndexPair(ND, Index));
101 }
102
103 void Destroy() {
104 if (DeclIndexPairVector *Vec
105 = DeclOrVector.dyn_cast<DeclIndexPairVector *>()) {
106 delete Vec;
107 DeclOrVector = ((NamedDecl *)0);
108 }
109 }
110
111 // Iteration.
112 class iterator;
113 iterator begin() const;
114 iterator end() const;
115 };
116
Douglas Gregor3545ff42009-09-21 16:56:56 +0000117 /// \brief A mapping from declaration names to the declarations that have
118 /// this name within a particular scope and their index within the list of
119 /// results.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000120 typedef llvm::DenseMap<DeclarationName, ShadowMapEntry> ShadowMap;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000121
122 /// \brief The semantic analysis object for which results are being
123 /// produced.
124 Sema &SemaRef;
Douglas Gregorb278aaf2011-02-01 19:23:04 +0000125
126 /// \brief The allocator used to allocate new code-completion strings.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +0000127 CodeCompletionAllocator &Allocator;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +0000128
129 CodeCompletionTUInfo &CCTUInfo;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000130
131 /// \brief If non-NULL, a filter function used to remove any code-completion
132 /// results that are not desirable.
133 LookupFilter Filter;
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000134
135 /// \brief Whether we should allow declarations as
136 /// nested-name-specifiers that would otherwise be filtered out.
137 bool AllowNestedNameSpecifiers;
138
Douglas Gregor7aa6b222010-05-30 01:49:25 +0000139 /// \brief If set, the type that we would prefer our resulting value
140 /// declarations to have.
141 ///
142 /// Closely matching the preferred type gives a boost to a result's
143 /// priority.
144 CanQualType PreferredType;
145
Douglas Gregor3545ff42009-09-21 16:56:56 +0000146 /// \brief A list of shadow maps, which is used to model name hiding at
147 /// different levels of, e.g., the inheritance hierarchy.
148 std::list<ShadowMap> ShadowMaps;
149
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000150 /// \brief If we're potentially referring to a C++ member function, the set
151 /// of qualifiers applied to the object type.
152 Qualifiers ObjectTypeQualifiers;
153
154 /// \brief Whether the \p ObjectTypeQualifiers field is active.
155 bool HasObjectTypeQualifiers;
156
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000157 /// \brief The selector that we prefer.
158 Selector PreferredSelector;
159
Douglas Gregor05fcf842010-11-02 20:36:02 +0000160 /// \brief The completion context in which we are gathering results.
Douglas Gregor50832e02010-09-20 22:39:41 +0000161 CodeCompletionContext CompletionContext;
162
James Dennett596e4752012-06-14 03:11:41 +0000163 /// \brief If we are in an instance method definition, the \@implementation
Douglas Gregor05fcf842010-11-02 20:36:02 +0000164 /// object.
165 ObjCImplementationDecl *ObjCImplementation;
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000166
Douglas Gregor50832e02010-09-20 22:39:41 +0000167 void AdjustResultPriorityForDecl(Result &R);
Douglas Gregor95887f92010-07-08 23:20:03 +0000168
Douglas Gregor0212fd72010-09-21 16:06:22 +0000169 void MaybeAddConstructorResults(Result R);
170
Douglas Gregor3545ff42009-09-21 16:56:56 +0000171 public:
Douglas Gregorbcbf46c2011-02-01 22:57:45 +0000172 explicit ResultBuilder(Sema &SemaRef, CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +0000173 CodeCompletionTUInfo &CCTUInfo,
Douglas Gregor0ac41382010-09-23 23:01:17 +0000174 const CodeCompletionContext &CompletionContext,
175 LookupFilter Filter = 0)
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +0000176 : SemaRef(SemaRef), Allocator(Allocator), CCTUInfo(CCTUInfo),
177 Filter(Filter),
Douglas Gregorb278aaf2011-02-01 19:23:04 +0000178 AllowNestedNameSpecifiers(false), HasObjectTypeQualifiers(false),
Douglas Gregor05fcf842010-11-02 20:36:02 +0000179 CompletionContext(CompletionContext),
180 ObjCImplementation(0)
181 {
182 // If this is an Objective-C instance method definition, dig out the
183 // corresponding implementation.
184 switch (CompletionContext.getKind()) {
185 case CodeCompletionContext::CCC_Expression:
186 case CodeCompletionContext::CCC_ObjCMessageReceiver:
187 case CodeCompletionContext::CCC_ParenthesizedExpression:
188 case CodeCompletionContext::CCC_Statement:
189 case CodeCompletionContext::CCC_Recovery:
190 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl())
191 if (Method->isInstanceMethod())
192 if (ObjCInterfaceDecl *Interface = Method->getClassInterface())
193 ObjCImplementation = Interface->getImplementation();
194 break;
195
196 default:
197 break;
198 }
199 }
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000200
201 /// \brief Determine the priority for a reference to the given declaration.
202 unsigned getBasePriority(const NamedDecl *D);
203
Douglas Gregorf64acca2010-05-25 21:41:55 +0000204 /// \brief Whether we should include code patterns in the completion
205 /// results.
206 bool includeCodePatterns() const {
207 return SemaRef.CodeCompleter &&
Douglas Gregorac322ec2010-08-27 21:18:54 +0000208 SemaRef.CodeCompleter->includeCodePatterns();
Douglas Gregorf64acca2010-05-25 21:41:55 +0000209 }
210
Douglas Gregor3545ff42009-09-21 16:56:56 +0000211 /// \brief Set the filter used for code-completion results.
212 void setFilter(LookupFilter Filter) {
213 this->Filter = Filter;
214 }
215
Douglas Gregor3545ff42009-09-21 16:56:56 +0000216 Result *data() { return Results.empty()? 0 : &Results.front(); }
217 unsigned size() const { return Results.size(); }
218 bool empty() const { return Results.empty(); }
219
Douglas Gregor7aa6b222010-05-30 01:49:25 +0000220 /// \brief Specify the preferred type.
221 void setPreferredType(QualType T) {
222 PreferredType = SemaRef.Context.getCanonicalType(T);
223 }
224
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000225 /// \brief Set the cv-qualifiers on the object type, for us in filtering
226 /// calls to member functions.
227 ///
228 /// When there are qualifiers in this set, they will be used to filter
229 /// out member functions that aren't available (because there will be a
230 /// cv-qualifier mismatch) or prefer functions with an exact qualifier
231 /// match.
232 void setObjectTypeQualifiers(Qualifiers Quals) {
233 ObjectTypeQualifiers = Quals;
234 HasObjectTypeQualifiers = true;
235 }
236
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000237 /// \brief Set the preferred selector.
238 ///
239 /// When an Objective-C method declaration result is added, and that
240 /// method's selector matches this preferred selector, we give that method
241 /// a slight priority boost.
242 void setPreferredSelector(Selector Sel) {
243 PreferredSelector = Sel;
244 }
Douglas Gregor05fcf842010-11-02 20:36:02 +0000245
Douglas Gregor50832e02010-09-20 22:39:41 +0000246 /// \brief Retrieve the code-completion context for which results are
247 /// being collected.
248 const CodeCompletionContext &getCompletionContext() const {
249 return CompletionContext;
250 }
251
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000252 /// \brief Specify whether nested-name-specifiers are allowed.
253 void allowNestedNameSpecifiers(bool Allow = true) {
254 AllowNestedNameSpecifiers = Allow;
255 }
256
Douglas Gregor74661272010-09-21 00:03:25 +0000257 /// \brief Return the semantic analysis object for which we are collecting
258 /// code completion results.
259 Sema &getSema() const { return SemaRef; }
260
Douglas Gregorb278aaf2011-02-01 19:23:04 +0000261 /// \brief Retrieve the allocator used to allocate code completion strings.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +0000262 CodeCompletionAllocator &getAllocator() const { return Allocator; }
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +0000263
264 CodeCompletionTUInfo &getCodeCompletionTUInfo() const { return CCTUInfo; }
Douglas Gregorb278aaf2011-02-01 19:23:04 +0000265
Douglas Gregor7c208612010-01-14 00:20:49 +0000266 /// \brief Determine whether the given declaration is at all interesting
267 /// as a code-completion result.
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000268 ///
269 /// \param ND the declaration that we are inspecting.
270 ///
271 /// \param AsNestedNameSpecifier will be set true if this declaration is
272 /// only interesting when it is a nested-name-specifier.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000273 bool isInterestingDecl(const NamedDecl *ND,
274 bool &AsNestedNameSpecifier) const;
Douglas Gregore0717ab2010-01-14 00:41:07 +0000275
276 /// \brief Check whether the result is hidden by the Hiding declaration.
277 ///
278 /// \returns true if the result is hidden and cannot be found, false if
279 /// the hidden result could still be found. When false, \p R may be
280 /// modified to describe how the result can be found (e.g., via extra
281 /// qualification).
282 bool CheckHiddenResult(Result &R, DeclContext *CurContext,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000283 const NamedDecl *Hiding);
Douglas Gregore0717ab2010-01-14 00:41:07 +0000284
Douglas Gregor3545ff42009-09-21 16:56:56 +0000285 /// \brief Add a new result to this result set (if it isn't already in one
286 /// of the shadow maps), or replace an existing result (for, e.g., a
287 /// redeclaration).
Douglas Gregor2af2f672009-09-21 20:12:40 +0000288 ///
Douglas Gregord8c61782012-02-15 15:34:24 +0000289 /// \param R the result to add (if it is unique).
Douglas Gregor2af2f672009-09-21 20:12:40 +0000290 ///
Douglas Gregord8c61782012-02-15 15:34:24 +0000291 /// \param CurContext the context in which this result will be named.
Douglas Gregor2af2f672009-09-21 20:12:40 +0000292 void MaybeAddResult(Result R, DeclContext *CurContext = 0);
Douglas Gregor3545ff42009-09-21 16:56:56 +0000293
Douglas Gregorc580c522010-01-14 01:09:38 +0000294 /// \brief Add a new result to this result set, where we already know
295 /// the hiding declation (if any).
296 ///
297 /// \param R the result to add (if it is unique).
298 ///
299 /// \param CurContext the context in which this result will be named.
300 ///
301 /// \param Hiding the declaration that hides the result.
Douglas Gregor09bbc652010-01-14 15:47:35 +0000302 ///
303 /// \param InBaseClass whether the result was found in a base
304 /// class of the searched context.
305 void AddResult(Result R, DeclContext *CurContext, NamedDecl *Hiding,
306 bool InBaseClass);
Douglas Gregorc580c522010-01-14 01:09:38 +0000307
Douglas Gregor78a21012010-01-14 16:01:26 +0000308 /// \brief Add a new non-declaration result to this result set.
309 void AddResult(Result R);
310
Douglas Gregor3545ff42009-09-21 16:56:56 +0000311 /// \brief Enter into a new scope.
312 void EnterNewScope();
313
314 /// \brief Exit from the current scope.
315 void ExitScope();
316
Douglas Gregorbaf69612009-11-18 04:19:12 +0000317 /// \brief Ignore this declaration, if it is seen again.
Dmitri Gribenko6cfb1532013-02-14 13:53:30 +0000318 void Ignore(const Decl *D) { AllDeclsFound.insert(D->getCanonicalDecl()); }
Douglas Gregorbaf69612009-11-18 04:19:12 +0000319
Douglas Gregor3545ff42009-09-21 16:56:56 +0000320 /// \name Name lookup predicates
321 ///
322 /// These predicates can be passed to the name lookup functions to filter the
323 /// results of name lookup. All of the predicates have the same type, so that
324 ///
325 //@{
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000326 bool IsOrdinaryName(const NamedDecl *ND) const;
327 bool IsOrdinaryNonTypeName(const NamedDecl *ND) const;
328 bool IsIntegralConstantValue(const NamedDecl *ND) const;
329 bool IsOrdinaryNonValueName(const NamedDecl *ND) const;
330 bool IsNestedNameSpecifier(const NamedDecl *ND) const;
331 bool IsEnum(const NamedDecl *ND) const;
332 bool IsClassOrStruct(const NamedDecl *ND) const;
333 bool IsUnion(const NamedDecl *ND) const;
334 bool IsNamespace(const NamedDecl *ND) const;
335 bool IsNamespaceOrAlias(const NamedDecl *ND) const;
336 bool IsType(const NamedDecl *ND) const;
337 bool IsMember(const NamedDecl *ND) const;
338 bool IsObjCIvar(const NamedDecl *ND) const;
339 bool IsObjCMessageReceiver(const NamedDecl *ND) const;
340 bool IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const;
341 bool IsObjCCollection(const NamedDecl *ND) const;
342 bool IsImpossibleToSatisfy(const NamedDecl *ND) const;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000343 //@}
344 };
345}
346
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000347class ResultBuilder::ShadowMapEntry::iterator {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000348 llvm::PointerUnion<const NamedDecl *, const DeclIndexPair *> DeclOrIterator;
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000349 unsigned SingleDeclIndex;
350
351public:
352 typedef DeclIndexPair value_type;
353 typedef value_type reference;
354 typedef std::ptrdiff_t difference_type;
355 typedef std::input_iterator_tag iterator_category;
356
357 class pointer {
358 DeclIndexPair Value;
359
360 public:
361 pointer(const DeclIndexPair &Value) : Value(Value) { }
362
363 const DeclIndexPair *operator->() const {
364 return &Value;
365 }
366 };
367
368 iterator() : DeclOrIterator((NamedDecl *)0), SingleDeclIndex(0) { }
369
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000370 iterator(const NamedDecl *SingleDecl, unsigned Index)
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000371 : DeclOrIterator(SingleDecl), SingleDeclIndex(Index) { }
372
373 iterator(const DeclIndexPair *Iterator)
374 : DeclOrIterator(Iterator), SingleDeclIndex(0) { }
375
376 iterator &operator++() {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000377 if (DeclOrIterator.is<const NamedDecl *>()) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000378 DeclOrIterator = (NamedDecl *)0;
379 SingleDeclIndex = 0;
380 return *this;
381 }
382
383 const DeclIndexPair *I = DeclOrIterator.get<const DeclIndexPair*>();
384 ++I;
385 DeclOrIterator = I;
386 return *this;
387 }
388
Chris Lattner9795b392010-09-04 18:12:20 +0000389 /*iterator operator++(int) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000390 iterator tmp(*this);
391 ++(*this);
392 return tmp;
Chris Lattner9795b392010-09-04 18:12:20 +0000393 }*/
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000394
395 reference operator*() const {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000396 if (const NamedDecl *ND = DeclOrIterator.dyn_cast<const NamedDecl *>())
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000397 return reference(ND, SingleDeclIndex);
398
Douglas Gregor94bb5e82009-12-06 21:27:58 +0000399 return *DeclOrIterator.get<const DeclIndexPair*>();
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000400 }
401
402 pointer operator->() const {
403 return pointer(**this);
404 }
405
406 friend bool operator==(const iterator &X, const iterator &Y) {
Douglas Gregor94bb5e82009-12-06 21:27:58 +0000407 return X.DeclOrIterator.getOpaqueValue()
408 == Y.DeclOrIterator.getOpaqueValue() &&
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000409 X.SingleDeclIndex == Y.SingleDeclIndex;
410 }
411
412 friend bool operator!=(const iterator &X, const iterator &Y) {
Douglas Gregor94bb5e82009-12-06 21:27:58 +0000413 return !(X == Y);
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000414 }
415};
416
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000417ResultBuilder::ShadowMapEntry::iterator
418ResultBuilder::ShadowMapEntry::begin() const {
419 if (DeclOrVector.isNull())
420 return iterator();
421
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000422 if (const NamedDecl *ND = DeclOrVector.dyn_cast<const NamedDecl *>())
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000423 return iterator(ND, SingleDeclIndex);
424
425 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->begin());
426}
427
428ResultBuilder::ShadowMapEntry::iterator
429ResultBuilder::ShadowMapEntry::end() const {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000430 if (DeclOrVector.is<const NamedDecl *>() || DeclOrVector.isNull())
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000431 return iterator();
432
433 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->end());
434}
435
Douglas Gregor2af2f672009-09-21 20:12:40 +0000436/// \brief Compute the qualification required to get from the current context
437/// (\p CurContext) to the target context (\p TargetContext).
438///
439/// \param Context the AST context in which the qualification will be used.
440///
441/// \param CurContext the context where an entity is being named, which is
442/// typically based on the current scope.
443///
444/// \param TargetContext the context in which the named entity actually
445/// resides.
446///
447/// \returns a nested name specifier that refers into the target context, or
448/// NULL if no qualification is needed.
449static NestedNameSpecifier *
450getRequiredQualification(ASTContext &Context,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000451 const DeclContext *CurContext,
452 const DeclContext *TargetContext) {
453 SmallVector<const DeclContext *, 4> TargetParents;
Douglas Gregor2af2f672009-09-21 20:12:40 +0000454
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000455 for (const DeclContext *CommonAncestor = TargetContext;
Douglas Gregor2af2f672009-09-21 20:12:40 +0000456 CommonAncestor && !CommonAncestor->Encloses(CurContext);
457 CommonAncestor = CommonAncestor->getLookupParent()) {
458 if (CommonAncestor->isTransparentContext() ||
459 CommonAncestor->isFunctionOrMethod())
460 continue;
461
462 TargetParents.push_back(CommonAncestor);
463 }
464
465 NestedNameSpecifier *Result = 0;
466 while (!TargetParents.empty()) {
Robert Wilhelm25284cc2013-08-23 16:11:15 +0000467 const DeclContext *Parent = TargetParents.pop_back_val();
468
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000469 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Parent)) {
Douglas Gregor68762e72010-08-23 21:17:50 +0000470 if (!Namespace->getIdentifier())
471 continue;
472
Douglas Gregor2af2f672009-09-21 20:12:40 +0000473 Result = NestedNameSpecifier::Create(Context, Result, Namespace);
Douglas Gregor68762e72010-08-23 21:17:50 +0000474 }
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000475 else if (const TagDecl *TD = dyn_cast<TagDecl>(Parent))
Douglas Gregor2af2f672009-09-21 20:12:40 +0000476 Result = NestedNameSpecifier::Create(Context, Result,
477 false,
478 Context.getTypeDeclType(TD).getTypePtr());
Douglas Gregor9eb77012009-11-07 00:00:49 +0000479 }
Douglas Gregor2af2f672009-09-21 20:12:40 +0000480 return Result;
481}
482
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000483bool ResultBuilder::isInterestingDecl(const NamedDecl *ND,
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000484 bool &AsNestedNameSpecifier) const {
485 AsNestedNameSpecifier = false;
486
Douglas Gregor7c208612010-01-14 00:20:49 +0000487 ND = ND->getUnderlyingDecl();
488 unsigned IDNS = ND->getIdentifierNamespace();
Douglas Gregor58acf322009-10-09 22:16:47 +0000489
490 // Skip unnamed entities.
Douglas Gregor7c208612010-01-14 00:20:49 +0000491 if (!ND->getDeclName())
492 return false;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000493
494 // Friend declarations and declarations introduced due to friends are never
495 // added as results.
John McCallbbbbe4e2010-03-11 07:50:04 +0000496 if (IDNS & (Decl::IDNS_OrdinaryFriend | Decl::IDNS_TagFriend))
Douglas Gregor7c208612010-01-14 00:20:49 +0000497 return false;
498
Douglas Gregor99fe2ad2009-12-11 17:31:05 +0000499 // Class template (partial) specializations are never added as results.
Douglas Gregor7c208612010-01-14 00:20:49 +0000500 if (isa<ClassTemplateSpecializationDecl>(ND) ||
501 isa<ClassTemplatePartialSpecializationDecl>(ND))
502 return false;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000503
Douglas Gregor99fe2ad2009-12-11 17:31:05 +0000504 // Using declarations themselves are never added as results.
Douglas Gregor7c208612010-01-14 00:20:49 +0000505 if (isa<UsingDecl>(ND))
506 return false;
507
508 // Some declarations have reserved names that we don't want to ever show.
509 if (const IdentifierInfo *Id = ND->getIdentifier()) {
Douglas Gregor3545ff42009-09-21 16:56:56 +0000510 // __va_list_tag is a freak of nature. Find it and skip it.
511 if (Id->isStr("__va_list_tag") || Id->isStr("__builtin_va_list"))
Douglas Gregor7c208612010-01-14 00:20:49 +0000512 return false;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000513
Douglas Gregor58acf322009-10-09 22:16:47 +0000514 // Filter out names reserved for the implementation (C99 7.1.3,
Douglas Gregor9f1570d2010-07-14 17:44:04 +0000515 // C++ [lib.global.names]) if they come from a system header.
Daniel Dunbar2c422dc92009-10-18 20:26:12 +0000516 //
517 // FIXME: Add predicate for this.
Douglas Gregor58acf322009-10-09 22:16:47 +0000518 if (Id->getLength() >= 2) {
Daniel Dunbar2c422dc92009-10-18 20:26:12 +0000519 const char *Name = Id->getNameStart();
Douglas Gregor58acf322009-10-09 22:16:47 +0000520 if (Name[0] == '_' &&
Douglas Gregor9f1570d2010-07-14 17:44:04 +0000521 (Name[1] == '_' || (Name[1] >= 'A' && Name[1] <= 'Z')) &&
522 (ND->getLocation().isInvalid() ||
523 SemaRef.SourceMgr.isInSystemHeader(
524 SemaRef.SourceMgr.getSpellingLoc(ND->getLocation()))))
Douglas Gregor7c208612010-01-14 00:20:49 +0000525 return false;
Douglas Gregor58acf322009-10-09 22:16:47 +0000526 }
Douglas Gregor3545ff42009-09-21 16:56:56 +0000527 }
Douglas Gregor2927c0c2010-11-09 03:59:40 +0000528
Douglas Gregor59cab552010-08-16 23:05:20 +0000529 if (Filter == &ResultBuilder::IsNestedNameSpecifier ||
530 ((isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND)) &&
531 Filter != &ResultBuilder::IsNamespace &&
Douglas Gregor0ac41382010-09-23 23:01:17 +0000532 Filter != &ResultBuilder::IsNamespaceOrAlias &&
533 Filter != 0))
Douglas Gregor59cab552010-08-16 23:05:20 +0000534 AsNestedNameSpecifier = true;
535
Douglas Gregor3545ff42009-09-21 16:56:56 +0000536 // Filter out any unwanted results.
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000537 if (Filter && !(this->*Filter)(ND)) {
538 // Check whether it is interesting as a nested-name-specifier.
David Blaikiebbafb8a2012-03-11 07:00:24 +0000539 if (AllowNestedNameSpecifiers && SemaRef.getLangOpts().CPlusPlus &&
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000540 IsNestedNameSpecifier(ND) &&
541 (Filter != &ResultBuilder::IsMember ||
542 (isa<CXXRecordDecl>(ND) &&
543 cast<CXXRecordDecl>(ND)->isInjectedClassName()))) {
544 AsNestedNameSpecifier = true;
545 return true;
546 }
547
Douglas Gregor7c208612010-01-14 00:20:49 +0000548 return false;
Douglas Gregor59cab552010-08-16 23:05:20 +0000549 }
Douglas Gregor7c208612010-01-14 00:20:49 +0000550 // ... then it must be interesting!
551 return true;
552}
553
Douglas Gregore0717ab2010-01-14 00:41:07 +0000554bool ResultBuilder::CheckHiddenResult(Result &R, DeclContext *CurContext,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000555 const NamedDecl *Hiding) {
Douglas Gregore0717ab2010-01-14 00:41:07 +0000556 // In C, there is no way to refer to a hidden name.
557 // FIXME: This isn't true; we can find a tag name hidden by an ordinary
558 // name if we introduce the tag type.
David Blaikiebbafb8a2012-03-11 07:00:24 +0000559 if (!SemaRef.getLangOpts().CPlusPlus)
Douglas Gregore0717ab2010-01-14 00:41:07 +0000560 return true;
561
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000562 const DeclContext *HiddenCtx =
563 R.Declaration->getDeclContext()->getRedeclContext();
Douglas Gregore0717ab2010-01-14 00:41:07 +0000564
565 // There is no way to qualify a name declared in a function or method.
566 if (HiddenCtx->isFunctionOrMethod())
567 return true;
568
Sebastian Redl50c68252010-08-31 00:36:30 +0000569 if (HiddenCtx == Hiding->getDeclContext()->getRedeclContext())
Douglas Gregore0717ab2010-01-14 00:41:07 +0000570 return true;
571
572 // We can refer to the result with the appropriate qualification. Do it.
573 R.Hidden = true;
574 R.QualifierIsInformative = false;
575
576 if (!R.Qualifier)
577 R.Qualifier = getRequiredQualification(SemaRef.Context,
578 CurContext,
579 R.Declaration->getDeclContext());
580 return false;
581}
582
Douglas Gregor95887f92010-07-08 23:20:03 +0000583/// \brief A simplified classification of types used to determine whether two
584/// types are "similar enough" when adjusting priorities.
Douglas Gregor6e240332010-08-16 16:18:59 +0000585SimplifiedTypeClass clang::getSimplifiedTypeClass(CanQualType T) {
Douglas Gregor95887f92010-07-08 23:20:03 +0000586 switch (T->getTypeClass()) {
587 case Type::Builtin:
588 switch (cast<BuiltinType>(T)->getKind()) {
589 case BuiltinType::Void:
590 return STC_Void;
591
592 case BuiltinType::NullPtr:
593 return STC_Pointer;
594
595 case BuiltinType::Overload:
596 case BuiltinType::Dependent:
Douglas Gregor95887f92010-07-08 23:20:03 +0000597 return STC_Other;
598
599 case BuiltinType::ObjCId:
600 case BuiltinType::ObjCClass:
601 case BuiltinType::ObjCSel:
602 return STC_ObjectiveC;
603
604 default:
605 return STC_Arithmetic;
606 }
David Blaikie8a40f702012-01-17 06:56:22 +0000607
Douglas Gregor95887f92010-07-08 23:20:03 +0000608 case Type::Complex:
609 return STC_Arithmetic;
610
611 case Type::Pointer:
612 return STC_Pointer;
613
614 case Type::BlockPointer:
615 return STC_Block;
616
617 case Type::LValueReference:
618 case Type::RValueReference:
619 return getSimplifiedTypeClass(T->getAs<ReferenceType>()->getPointeeType());
620
621 case Type::ConstantArray:
622 case Type::IncompleteArray:
623 case Type::VariableArray:
624 case Type::DependentSizedArray:
625 return STC_Array;
626
627 case Type::DependentSizedExtVector:
628 case Type::Vector:
629 case Type::ExtVector:
630 return STC_Arithmetic;
631
632 case Type::FunctionProto:
633 case Type::FunctionNoProto:
634 return STC_Function;
635
636 case Type::Record:
637 return STC_Record;
638
639 case Type::Enum:
640 return STC_Arithmetic;
641
642 case Type::ObjCObject:
643 case Type::ObjCInterface:
644 case Type::ObjCObjectPointer:
645 return STC_ObjectiveC;
646
647 default:
648 return STC_Other;
649 }
650}
651
652/// \brief Get the type that a given expression will have if this declaration
653/// is used as an expression in its "typical" code-completion form.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000654QualType clang::getDeclUsageType(ASTContext &C, const NamedDecl *ND) {
Douglas Gregor95887f92010-07-08 23:20:03 +0000655 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
656
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000657 if (const TypeDecl *Type = dyn_cast<TypeDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000658 return C.getTypeDeclType(Type);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000659 if (const ObjCInterfaceDecl *Iface = dyn_cast<ObjCInterfaceDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000660 return C.getObjCInterfaceType(Iface);
661
662 QualType T;
Alp Tokera2794f92014-01-22 07:29:52 +0000663 if (const FunctionDecl *Function = ND->getAsFunction())
Douglas Gregor603d81b2010-07-13 08:18:22 +0000664 T = Function->getCallResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000665 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregor603d81b2010-07-13 08:18:22 +0000666 T = Method->getSendResultType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000667 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000668 T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext()));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000669 else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000670 T = Property->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000671 else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND))
Douglas Gregor95887f92010-07-08 23:20:03 +0000672 T = Value->getType();
673 else
674 return QualType();
Douglas Gregoraf670a82011-04-14 20:33:34 +0000675
676 // Dig through references, function pointers, and block pointers to
677 // get down to the likely type of an expression when the entity is
678 // used.
679 do {
680 if (const ReferenceType *Ref = T->getAs<ReferenceType>()) {
681 T = Ref->getPointeeType();
682 continue;
683 }
684
685 if (const PointerType *Pointer = T->getAs<PointerType>()) {
686 if (Pointer->getPointeeType()->isFunctionType()) {
687 T = Pointer->getPointeeType();
688 continue;
689 }
690
691 break;
692 }
693
694 if (const BlockPointerType *Block = T->getAs<BlockPointerType>()) {
695 T = Block->getPointeeType();
696 continue;
697 }
698
699 if (const FunctionType *Function = T->getAs<FunctionType>()) {
Alp Toker314cc812014-01-25 16:55:45 +0000700 T = Function->getReturnType();
Douglas Gregoraf670a82011-04-14 20:33:34 +0000701 continue;
702 }
703
704 break;
705 } while (true);
706
707 return T;
Douglas Gregor95887f92010-07-08 23:20:03 +0000708}
709
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000710unsigned ResultBuilder::getBasePriority(const NamedDecl *ND) {
711 if (!ND)
712 return CCP_Unlikely;
713
714 // Context-based decisions.
Richard Smith541b38b2013-09-20 01:15:31 +0000715 const DeclContext *LexicalDC = ND->getLexicalDeclContext();
716 if (LexicalDC->isFunctionOrMethod()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000717 // _cmd is relatively rare
718 if (const ImplicitParamDecl *ImplicitParam =
719 dyn_cast<ImplicitParamDecl>(ND))
720 if (ImplicitParam->getIdentifier() &&
721 ImplicitParam->getIdentifier()->isStr("_cmd"))
722 return CCP_ObjC_cmd;
723
724 return CCP_LocalDeclaration;
725 }
Richard Smith541b38b2013-09-20 01:15:31 +0000726
727 const DeclContext *DC = ND->getDeclContext()->getRedeclContext();
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000728 if (DC->isRecord() || isa<ObjCContainerDecl>(DC))
729 return CCP_MemberDeclaration;
730
731 // Content-based decisions.
732 if (isa<EnumConstantDecl>(ND))
733 return CCP_Constant;
734
Douglas Gregor52e0de42013-01-31 05:03:46 +0000735 // Use CCP_Type for type declarations unless we're in a statement, Objective-C
736 // message receiver, or parenthesized expression context. There, it's as
737 // likely that the user will want to write a type as other declarations.
738 if ((isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND)) &&
739 !(CompletionContext.getKind() == CodeCompletionContext::CCC_Statement ||
740 CompletionContext.getKind()
741 == CodeCompletionContext::CCC_ObjCMessageReceiver ||
742 CompletionContext.getKind()
743 == CodeCompletionContext::CCC_ParenthesizedExpression))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000744 return CCP_Type;
745
746 return CCP_Declaration;
747}
748
Douglas Gregor50832e02010-09-20 22:39:41 +0000749void ResultBuilder::AdjustResultPriorityForDecl(Result &R) {
750 // If this is an Objective-C method declaration whose selector matches our
751 // preferred selector, give it a priority boost.
752 if (!PreferredSelector.isNull())
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000753 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(R.Declaration))
Douglas Gregor50832e02010-09-20 22:39:41 +0000754 if (PreferredSelector == Method->getSelector())
755 R.Priority += CCD_SelectorMatch;
Douglas Gregor5fb901d2010-09-20 23:11:55 +0000756
Douglas Gregor50832e02010-09-20 22:39:41 +0000757 // If we have a preferred type, adjust the priority for results with exactly-
758 // matching or nearly-matching types.
759 if (!PreferredType.isNull()) {
760 QualType T = getDeclUsageType(SemaRef.Context, R.Declaration);
761 if (!T.isNull()) {
762 CanQualType TC = SemaRef.Context.getCanonicalType(T);
763 // Check for exactly-matching types (modulo qualifiers).
764 if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC))
765 R.Priority /= CCF_ExactTypeMatch;
766 // Check for nearly-matching types, based on classification of each.
767 else if ((getSimplifiedTypeClass(PreferredType)
Douglas Gregor95887f92010-07-08 23:20:03 +0000768 == getSimplifiedTypeClass(TC)) &&
Douglas Gregor50832e02010-09-20 22:39:41 +0000769 !(PreferredType->isEnumeralType() && TC->isEnumeralType()))
770 R.Priority /= CCF_SimilarTypeMatch;
771 }
772 }
Douglas Gregor95887f92010-07-08 23:20:03 +0000773}
774
Douglas Gregor0212fd72010-09-21 16:06:22 +0000775void ResultBuilder::MaybeAddConstructorResults(Result R) {
David Blaikiebbafb8a2012-03-11 07:00:24 +0000776 if (!SemaRef.getLangOpts().CPlusPlus || !R.Declaration ||
Douglas Gregor0212fd72010-09-21 16:06:22 +0000777 !CompletionContext.wantConstructorResults())
778 return;
779
780 ASTContext &Context = SemaRef.Context;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000781 const NamedDecl *D = R.Declaration;
782 const CXXRecordDecl *Record = 0;
783 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D))
Douglas Gregor0212fd72010-09-21 16:06:22 +0000784 Record = ClassTemplate->getTemplatedDecl();
785 else if ((Record = dyn_cast<CXXRecordDecl>(D))) {
786 // Skip specializations and partial specializations.
787 if (isa<ClassTemplateSpecializationDecl>(Record))
788 return;
789 } else {
790 // There are no constructors here.
791 return;
792 }
793
794 Record = Record->getDefinition();
795 if (!Record)
796 return;
797
798
799 QualType RecordTy = Context.getTypeDeclType(Record);
800 DeclarationName ConstructorName
801 = Context.DeclarationNames.getCXXConstructorName(
802 Context.getCanonicalType(RecordTy));
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000803 DeclContext::lookup_const_result Ctors = Record->lookup(ConstructorName);
804 for (DeclContext::lookup_const_iterator I = Ctors.begin(),
805 E = Ctors.end();
806 I != E; ++I) {
David Blaikieff7d47a2012-12-19 00:45:41 +0000807 R.Declaration = *I;
Douglas Gregor0212fd72010-09-21 16:06:22 +0000808 R.CursorKind = getCursorKindForDecl(R.Declaration);
809 Results.push_back(R);
810 }
811}
812
Douglas Gregor7c208612010-01-14 00:20:49 +0000813void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) {
814 assert(!ShadowMaps.empty() && "Must enter into a results scope");
815
816 if (R.Kind != Result::RK_Declaration) {
817 // For non-declaration results, just add the result.
818 Results.push_back(R);
819 return;
820 }
821
822 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000823 if (const UsingShadowDecl *Using =
824 dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000825 MaybeAddResult(Result(Using->getTargetDecl(),
826 getBasePriority(Using->getTargetDecl()),
827 R.Qualifier),
828 CurContext);
Douglas Gregor7c208612010-01-14 00:20:49 +0000829 return;
830 }
831
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000832 const Decl *CanonDecl = R.Declaration->getCanonicalDecl();
Douglas Gregor7c208612010-01-14 00:20:49 +0000833 unsigned IDNS = CanonDecl->getIdentifierNamespace();
834
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000835 bool AsNestedNameSpecifier = false;
836 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregor7c208612010-01-14 00:20:49 +0000837 return;
838
Douglas Gregor0212fd72010-09-21 16:06:22 +0000839 // C++ constructors are never found by name lookup.
840 if (isa<CXXConstructorDecl>(R.Declaration))
841 return;
842
Douglas Gregor3545ff42009-09-21 16:56:56 +0000843 ShadowMap &SMap = ShadowMaps.back();
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000844 ShadowMapEntry::iterator I, IEnd;
845 ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName());
846 if (NamePos != SMap.end()) {
847 I = NamePos->second.begin();
848 IEnd = NamePos->second.end();
849 }
850
851 for (; I != IEnd; ++I) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000852 const NamedDecl *ND = I->first;
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000853 unsigned Index = I->second;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000854 if (ND->getCanonicalDecl() == CanonDecl) {
855 // This is a redeclaration. Always pick the newer declaration.
Douglas Gregor3545ff42009-09-21 16:56:56 +0000856 Results[Index].Declaration = R.Declaration;
857
Douglas Gregor3545ff42009-09-21 16:56:56 +0000858 // We're done.
859 return;
860 }
861 }
862
863 // This is a new declaration in this scope. However, check whether this
864 // declaration name is hidden by a similarly-named declaration in an outer
865 // scope.
866 std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end();
867 --SMEnd;
868 for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) {
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000869 ShadowMapEntry::iterator I, IEnd;
870 ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName());
871 if (NamePos != SM->end()) {
872 I = NamePos->second.begin();
873 IEnd = NamePos->second.end();
874 }
875 for (; I != IEnd; ++I) {
Douglas Gregor3545ff42009-09-21 16:56:56 +0000876 // A tag declaration does not hide a non-tag declaration.
John McCalle87beb22010-04-23 18:46:30 +0000877 if (I->first->hasTagIdentifierNamespace() &&
Richard Smith541b38b2013-09-20 01:15:31 +0000878 (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary |
879 Decl::IDNS_LocalExtern | Decl::IDNS_ObjCProtocol)))
Douglas Gregor3545ff42009-09-21 16:56:56 +0000880 continue;
881
882 // Protocols are in distinct namespaces from everything else.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000883 if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000884 || (IDNS & Decl::IDNS_ObjCProtocol)) &&
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000885 I->first->getIdentifierNamespace() != IDNS)
Douglas Gregor3545ff42009-09-21 16:56:56 +0000886 continue;
887
888 // The newly-added result is hidden by an entry in the shadow map.
Douglas Gregore0717ab2010-01-14 00:41:07 +0000889 if (CheckHiddenResult(R, CurContext, I->first))
Douglas Gregor3545ff42009-09-21 16:56:56 +0000890 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +0000891
892 break;
893 }
894 }
895
896 // Make sure that any given declaration only shows up in the result set once.
897 if (!AllDeclsFound.insert(CanonDecl))
898 return;
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000899
Douglas Gregore412a5a2009-09-23 22:26:46 +0000900 // If the filter is for nested-name-specifiers, then this result starts a
901 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000902 if (AsNestedNameSpecifier) {
Douglas Gregore412a5a2009-09-23 22:26:46 +0000903 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000904 R.Priority = CCP_NestedNameSpecifier;
Douglas Gregor50832e02010-09-20 22:39:41 +0000905 } else
906 AdjustResultPriorityForDecl(R);
Douglas Gregorc2cb2e22010-08-27 15:29:55 +0000907
Douglas Gregor5bf52692009-09-22 23:15:58 +0000908 // If this result is supposed to have an informative qualifier, add one.
Douglas Gregore412a5a2009-09-23 22:26:46 +0000909 if (R.QualifierIsInformative && !R.Qualifier &&
910 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000911 const DeclContext *Ctx = R.Declaration->getDeclContext();
912 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregor5bf52692009-09-22 23:15:58 +0000913 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000914 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregor5bf52692009-09-22 23:15:58 +0000915 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
916 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
917 else
918 R.QualifierIsInformative = false;
919 }
Douglas Gregore412a5a2009-09-23 22:26:46 +0000920
Douglas Gregor3545ff42009-09-21 16:56:56 +0000921 // Insert this result into the set of results and into the current shadow
922 // map.
Douglas Gregor05e7ca32009-12-06 20:23:50 +0000923 SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size());
Douglas Gregor3545ff42009-09-21 16:56:56 +0000924 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +0000925
926 if (!AsNestedNameSpecifier)
927 MaybeAddConstructorResults(R);
Douglas Gregor3545ff42009-09-21 16:56:56 +0000928}
929
Douglas Gregorc580c522010-01-14 01:09:38 +0000930void ResultBuilder::AddResult(Result R, DeclContext *CurContext,
Douglas Gregor09bbc652010-01-14 15:47:35 +0000931 NamedDecl *Hiding, bool InBaseClass = false) {
Douglas Gregor78a21012010-01-14 16:01:26 +0000932 if (R.Kind != Result::RK_Declaration) {
933 // For non-declaration results, just add the result.
934 Results.push_back(R);
935 return;
936 }
937
Douglas Gregorc580c522010-01-14 01:09:38 +0000938 // Look through using declarations.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000939 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +0000940 AddResult(Result(Using->getTargetDecl(),
941 getBasePriority(Using->getTargetDecl()),
942 R.Qualifier),
943 CurContext, Hiding);
Douglas Gregorc580c522010-01-14 01:09:38 +0000944 return;
945 }
946
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000947 bool AsNestedNameSpecifier = false;
948 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregorc580c522010-01-14 01:09:38 +0000949 return;
950
Douglas Gregor0212fd72010-09-21 16:06:22 +0000951 // C++ constructors are never found by name lookup.
952 if (isa<CXXConstructorDecl>(R.Declaration))
953 return;
954
Douglas Gregorc580c522010-01-14 01:09:38 +0000955 if (Hiding && CheckHiddenResult(R, CurContext, Hiding))
956 return;
Nick Lewyckyc3921482012-04-03 21:44:08 +0000957
Douglas Gregorc580c522010-01-14 01:09:38 +0000958 // Make sure that any given declaration only shows up in the result set once.
959 if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl()))
960 return;
961
962 // If the filter is for nested-name-specifiers, then this result starts a
963 // nested-name-specifier.
Douglas Gregora2db7932010-05-26 22:00:08 +0000964 if (AsNestedNameSpecifier) {
Douglas Gregorc580c522010-01-14 01:09:38 +0000965 R.StartsNestedNameSpecifier = true;
Douglas Gregora2db7932010-05-26 22:00:08 +0000966 R.Priority = CCP_NestedNameSpecifier;
967 }
Douglas Gregor09bbc652010-01-14 15:47:35 +0000968 else if (Filter == &ResultBuilder::IsMember && !R.Qualifier && InBaseClass &&
969 isa<CXXRecordDecl>(R.Declaration->getDeclContext()
Sebastian Redl50c68252010-08-31 00:36:30 +0000970 ->getRedeclContext()))
Douglas Gregor09bbc652010-01-14 15:47:35 +0000971 R.QualifierIsInformative = true;
972
Douglas Gregorc580c522010-01-14 01:09:38 +0000973 // If this result is supposed to have an informative qualifier, add one.
974 if (R.QualifierIsInformative && !R.Qualifier &&
975 !R.StartsNestedNameSpecifier) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000976 const DeclContext *Ctx = R.Declaration->getDeclContext();
977 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregorc580c522010-01-14 01:09:38 +0000978 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000979 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregorc580c522010-01-14 01:09:38 +0000980 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
Douglas Gregor6ae4c522010-01-14 03:21:49 +0000981 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregorc580c522010-01-14 01:09:38 +0000982 else
983 R.QualifierIsInformative = false;
984 }
985
Douglas Gregora2db7932010-05-26 22:00:08 +0000986 // Adjust the priority if this result comes from a base class.
987 if (InBaseClass)
988 R.Priority += CCD_InBaseClass;
989
Douglas Gregor50832e02010-09-20 22:39:41 +0000990 AdjustResultPriorityForDecl(R);
Douglas Gregor7aa6b222010-05-30 01:49:25 +0000991
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000992 if (HasObjectTypeQualifiers)
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +0000993 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(R.Declaration))
Douglas Gregor9be0ed42010-08-26 16:36:48 +0000994 if (Method->isInstance()) {
995 Qualifiers MethodQuals
996 = Qualifiers::fromCVRMask(Method->getTypeQualifiers());
997 if (ObjectTypeQualifiers == MethodQuals)
998 R.Priority += CCD_ObjectQualifierMatch;
999 else if (ObjectTypeQualifiers - MethodQuals) {
1000 // The method cannot be invoked, because doing so would drop
1001 // qualifiers.
1002 return;
1003 }
1004 }
1005
Douglas Gregorc580c522010-01-14 01:09:38 +00001006 // Insert this result into the set of results.
1007 Results.push_back(R);
Douglas Gregor0212fd72010-09-21 16:06:22 +00001008
1009 if (!AsNestedNameSpecifier)
1010 MaybeAddConstructorResults(R);
Douglas Gregorc580c522010-01-14 01:09:38 +00001011}
1012
Douglas Gregor78a21012010-01-14 16:01:26 +00001013void ResultBuilder::AddResult(Result R) {
1014 assert(R.Kind != Result::RK_Declaration &&
1015 "Declaration results need more context");
1016 Results.push_back(R);
1017}
1018
Douglas Gregor3545ff42009-09-21 16:56:56 +00001019/// \brief Enter into a new scope.
1020void ResultBuilder::EnterNewScope() {
1021 ShadowMaps.push_back(ShadowMap());
1022}
1023
1024/// \brief Exit from the current scope.
1025void ResultBuilder::ExitScope() {
Douglas Gregor05e7ca32009-12-06 20:23:50 +00001026 for (ShadowMap::iterator E = ShadowMaps.back().begin(),
1027 EEnd = ShadowMaps.back().end();
1028 E != EEnd;
1029 ++E)
1030 E->second.Destroy();
1031
Douglas Gregor3545ff42009-09-21 16:56:56 +00001032 ShadowMaps.pop_back();
1033}
1034
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001035/// \brief Determines whether this given declaration will be found by
1036/// ordinary name lookup.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001037bool ResultBuilder::IsOrdinaryName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001038 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1039
Richard Smith541b38b2013-09-20 01:15:31 +00001040 // If name lookup finds a local extern declaration, then we are in a
1041 // context where it behaves like an ordinary name.
1042 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001043 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9858ed52010-06-15 20:26:51 +00001044 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001045 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor05fcf842010-11-02 20:36:02 +00001046 if (isa<ObjCIvarDecl>(ND))
1047 return true;
Douglas Gregor05fcf842010-11-02 20:36:02 +00001048 }
1049
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00001050 return ND->getIdentifierNamespace() & IDNS;
1051}
1052
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001053/// \brief Determines whether this given declaration will be found by
Douglas Gregor70febae2010-05-28 00:49:12 +00001054/// ordinary name lookup but is not a type name.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001055bool ResultBuilder::IsOrdinaryNonTypeName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001056 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1057 if (isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND))
1058 return false;
1059
Richard Smith541b38b2013-09-20 01:15:31 +00001060 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001061 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9858ed52010-06-15 20:26:51 +00001062 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001063 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor05fcf842010-11-02 20:36:02 +00001064 if (isa<ObjCIvarDecl>(ND))
1065 return true;
Douglas Gregor05fcf842010-11-02 20:36:02 +00001066 }
1067
Douglas Gregor70febae2010-05-28 00:49:12 +00001068 return ND->getIdentifierNamespace() & IDNS;
1069}
1070
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001071bool ResultBuilder::IsIntegralConstantValue(const NamedDecl *ND) const {
Douglas Gregor85b50632010-07-28 21:50:18 +00001072 if (!IsOrdinaryNonTypeName(ND))
1073 return 0;
1074
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001075 if (const ValueDecl *VD = dyn_cast<ValueDecl>(ND->getUnderlyingDecl()))
Douglas Gregor85b50632010-07-28 21:50:18 +00001076 if (VD->getType()->isIntegralOrEnumerationType())
1077 return true;
1078
1079 return false;
1080}
1081
Douglas Gregor70febae2010-05-28 00:49:12 +00001082/// \brief Determines whether this given declaration will be found by
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001083/// ordinary name lookup.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001084bool ResultBuilder::IsOrdinaryNonValueName(const NamedDecl *ND) const {
Douglas Gregor70febae2010-05-28 00:49:12 +00001085 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1086
Richard Smith541b38b2013-09-20 01:15:31 +00001087 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikiebbafb8a2012-03-11 07:00:24 +00001088 if (SemaRef.getLangOpts().CPlusPlus)
John McCalle87beb22010-04-23 18:46:30 +00001089 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001090
1091 return (ND->getIdentifierNamespace() & IDNS) &&
Douglas Gregor70febae2010-05-28 00:49:12 +00001092 !isa<ValueDecl>(ND) && !isa<FunctionTemplateDecl>(ND) &&
1093 !isa<ObjCPropertyDecl>(ND);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001094}
1095
Douglas Gregor3545ff42009-09-21 16:56:56 +00001096/// \brief Determines whether the given declaration is suitable as the
1097/// start of a C++ nested-name-specifier, e.g., a class or namespace.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001098bool ResultBuilder::IsNestedNameSpecifier(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001099 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001100 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001101 ND = ClassTemplate->getTemplatedDecl();
1102
1103 return SemaRef.isAcceptableNestedNameSpecifier(ND);
1104}
1105
1106/// \brief Determines whether the given declaration is an enumeration.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001107bool ResultBuilder::IsEnum(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001108 return isa<EnumDecl>(ND);
1109}
1110
1111/// \brief Determines whether the given declaration is a class or struct.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001112bool ResultBuilder::IsClassOrStruct(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001113 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001114 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001115 ND = ClassTemplate->getTemplatedDecl();
Joao Matosdc86f942012-08-31 18:45:21 +00001116
1117 // For purposes of this check, interfaces match too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001118 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara6150c882010-05-11 21:36:43 +00001119 return RD->getTagKind() == TTK_Class ||
Joao Matosdc86f942012-08-31 18:45:21 +00001120 RD->getTagKind() == TTK_Struct ||
1121 RD->getTagKind() == TTK_Interface;
Douglas Gregor3545ff42009-09-21 16:56:56 +00001122
1123 return false;
1124}
1125
1126/// \brief Determines whether the given declaration is a union.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001127bool ResultBuilder::IsUnion(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001128 // Allow us to find class templates, too.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001129 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor3545ff42009-09-21 16:56:56 +00001130 ND = ClassTemplate->getTemplatedDecl();
1131
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001132 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara6150c882010-05-11 21:36:43 +00001133 return RD->getTagKind() == TTK_Union;
Douglas Gregor3545ff42009-09-21 16:56:56 +00001134
1135 return false;
1136}
1137
1138/// \brief Determines whether the given declaration is a namespace.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001139bool ResultBuilder::IsNamespace(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001140 return isa<NamespaceDecl>(ND);
1141}
1142
1143/// \brief Determines whether the given declaration is a namespace or
1144/// namespace alias.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001145bool ResultBuilder::IsNamespaceOrAlias(const NamedDecl *ND) const {
Douglas Gregor3545ff42009-09-21 16:56:56 +00001146 return isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND);
1147}
1148
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001149/// \brief Determines whether the given declaration is a type.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001150bool ResultBuilder::IsType(const NamedDecl *ND) const {
1151 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor99fa2642010-08-24 01:06:58 +00001152 ND = Using->getTargetDecl();
1153
1154 return isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND);
Douglas Gregor3545ff42009-09-21 16:56:56 +00001155}
1156
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001157/// \brief Determines which members of a class should be visible via
1158/// "." or "->". Only value declarations, nested name specifiers, and
1159/// using declarations thereof should show up.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001160bool ResultBuilder::IsMember(const NamedDecl *ND) const {
1161 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor99fe2ad2009-12-11 17:31:05 +00001162 ND = Using->getTargetDecl();
1163
Douglas Gregor70788392009-12-11 18:14:22 +00001164 return isa<ValueDecl>(ND) || isa<FunctionTemplateDecl>(ND) ||
1165 isa<ObjCPropertyDecl>(ND);
Douglas Gregore412a5a2009-09-23 22:26:46 +00001166}
1167
Douglas Gregora817a192010-05-27 23:06:34 +00001168static bool isObjCReceiverType(ASTContext &C, QualType T) {
1169 T = C.getCanonicalType(T);
1170 switch (T->getTypeClass()) {
1171 case Type::ObjCObject:
1172 case Type::ObjCInterface:
1173 case Type::ObjCObjectPointer:
1174 return true;
1175
1176 case Type::Builtin:
1177 switch (cast<BuiltinType>(T)->getKind()) {
1178 case BuiltinType::ObjCId:
1179 case BuiltinType::ObjCClass:
1180 case BuiltinType::ObjCSel:
1181 return true;
1182
1183 default:
1184 break;
1185 }
1186 return false;
1187
1188 default:
1189 break;
1190 }
1191
David Blaikiebbafb8a2012-03-11 07:00:24 +00001192 if (!C.getLangOpts().CPlusPlus)
Douglas Gregora817a192010-05-27 23:06:34 +00001193 return false;
1194
1195 // FIXME: We could perform more analysis here to determine whether a
1196 // particular class type has any conversions to Objective-C types. For now,
1197 // just accept all class types.
1198 return T->isDependentType() || T->isRecordType();
1199}
1200
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001201bool ResultBuilder::IsObjCMessageReceiver(const NamedDecl *ND) const {
Douglas Gregora817a192010-05-27 23:06:34 +00001202 QualType T = getDeclUsageType(SemaRef.Context, ND);
1203 if (T.isNull())
1204 return false;
1205
1206 T = SemaRef.Context.getBaseElementType(T);
1207 return isObjCReceiverType(SemaRef.Context, T);
1208}
1209
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001210bool ResultBuilder::IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const {
Douglas Gregord8c61782012-02-15 15:34:24 +00001211 if (IsObjCMessageReceiver(ND))
1212 return true;
1213
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001214 const VarDecl *Var = dyn_cast<VarDecl>(ND);
Douglas Gregord8c61782012-02-15 15:34:24 +00001215 if (!Var)
1216 return false;
1217
1218 return Var->hasLocalStorage() && !Var->hasAttr<BlocksAttr>();
1219}
1220
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001221bool ResultBuilder::IsObjCCollection(const NamedDecl *ND) const {
David Blaikiebbafb8a2012-03-11 07:00:24 +00001222 if ((SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryName(ND)) ||
1223 (!SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryNonTypeName(ND)))
Douglas Gregor68762e72010-08-23 21:17:50 +00001224 return false;
1225
1226 QualType T = getDeclUsageType(SemaRef.Context, ND);
1227 if (T.isNull())
1228 return false;
1229
1230 T = SemaRef.Context.getBaseElementType(T);
1231 return T->isObjCObjectType() || T->isObjCObjectPointerType() ||
1232 T->isObjCIdType() ||
David Blaikiebbafb8a2012-03-11 07:00:24 +00001233 (SemaRef.getLangOpts().CPlusPlus && T->isRecordType());
Douglas Gregor68762e72010-08-23 21:17:50 +00001234}
Douglas Gregora817a192010-05-27 23:06:34 +00001235
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001236bool ResultBuilder::IsImpossibleToSatisfy(const NamedDecl *ND) const {
Douglas Gregor0ac41382010-09-23 23:01:17 +00001237 return false;
1238}
1239
James Dennettf1243872012-06-17 05:33:25 +00001240/// \brief Determines whether the given declaration is an Objective-C
Douglas Gregor2b8162b2010-01-14 16:08:12 +00001241/// instance variable.
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00001242bool ResultBuilder::IsObjCIvar(const NamedDecl *ND) const {
Douglas Gregor2b8162b2010-01-14 16:08:12 +00001243 return isa<ObjCIvarDecl>(ND);
1244}
1245
Douglas Gregorc580c522010-01-14 01:09:38 +00001246namespace {
1247 /// \brief Visible declaration consumer that adds a code-completion result
1248 /// for each visible declaration.
1249 class CodeCompletionDeclConsumer : public VisibleDeclConsumer {
1250 ResultBuilder &Results;
1251 DeclContext *CurContext;
1252
1253 public:
1254 CodeCompletionDeclConsumer(ResultBuilder &Results, DeclContext *CurContext)
1255 : Results(Results), CurContext(CurContext) { }
1256
Erik Verbruggen2e657ff2011-10-06 07:27:49 +00001257 virtual void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
1258 bool InBaseClass) {
1259 bool Accessible = true;
Douglas Gregor03ba1882011-11-03 16:51:37 +00001260 if (Ctx)
1261 Accessible = Results.getSema().IsSimplyAccessible(ND, Ctx);
1262
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00001263 ResultBuilder::Result Result(ND, Results.getBasePriority(ND), 0, false,
1264 Accessible);
Erik Verbruggen2e657ff2011-10-06 07:27:49 +00001265 Results.AddResult(Result, CurContext, Hiding, InBaseClass);
Douglas Gregorc580c522010-01-14 01:09:38 +00001266 }
1267 };
1268}
1269
Douglas Gregor3545ff42009-09-21 16:56:56 +00001270/// \brief Add type specifiers for the current language as keyword results.
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001271static void AddTypeSpecifierResults(const LangOptions &LangOpts,
Douglas Gregor3545ff42009-09-21 16:56:56 +00001272 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001273 typedef CodeCompletionResult Result;
Douglas Gregora2db7932010-05-26 22:00:08 +00001274 Results.AddResult(Result("short", CCP_Type));
1275 Results.AddResult(Result("long", CCP_Type));
1276 Results.AddResult(Result("signed", CCP_Type));
1277 Results.AddResult(Result("unsigned", CCP_Type));
1278 Results.AddResult(Result("void", CCP_Type));
1279 Results.AddResult(Result("char", CCP_Type));
1280 Results.AddResult(Result("int", CCP_Type));
1281 Results.AddResult(Result("float", CCP_Type));
1282 Results.AddResult(Result("double", CCP_Type));
1283 Results.AddResult(Result("enum", CCP_Type));
1284 Results.AddResult(Result("struct", CCP_Type));
1285 Results.AddResult(Result("union", CCP_Type));
1286 Results.AddResult(Result("const", CCP_Type));
1287 Results.AddResult(Result("volatile", CCP_Type));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001288
Douglas Gregor3545ff42009-09-21 16:56:56 +00001289 if (LangOpts.C99) {
1290 // C99-specific
Douglas Gregora2db7932010-05-26 22:00:08 +00001291 Results.AddResult(Result("_Complex", CCP_Type));
1292 Results.AddResult(Result("_Imaginary", CCP_Type));
1293 Results.AddResult(Result("_Bool", CCP_Type));
1294 Results.AddResult(Result("restrict", CCP_Type));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001295 }
1296
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001297 CodeCompletionBuilder Builder(Results.getAllocator(),
1298 Results.getCodeCompletionTUInfo());
Douglas Gregor3545ff42009-09-21 16:56:56 +00001299 if (LangOpts.CPlusPlus) {
1300 // C++-specific
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00001301 Results.AddResult(Result("bool", CCP_Type +
1302 (LangOpts.ObjC1? CCD_bool_in_ObjC : 0)));
Douglas Gregora2db7932010-05-26 22:00:08 +00001303 Results.AddResult(Result("class", CCP_Type));
1304 Results.AddResult(Result("wchar_t", CCP_Type));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001305
Douglas Gregorf4c33342010-05-28 00:22:41 +00001306 // typename qualified-id
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001307 Builder.AddTypedTextChunk("typename");
1308 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1309 Builder.AddPlaceholderChunk("qualifier");
1310 Builder.AddTextChunk("::");
1311 Builder.AddPlaceholderChunk("name");
1312 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001313
Richard Smith2bf7fdb2013-01-02 11:42:31 +00001314 if (LangOpts.CPlusPlus11) {
Douglas Gregora2db7932010-05-26 22:00:08 +00001315 Results.AddResult(Result("auto", CCP_Type));
1316 Results.AddResult(Result("char16_t", CCP_Type));
1317 Results.AddResult(Result("char32_t", CCP_Type));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001318
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001319 Builder.AddTypedTextChunk("decltype");
1320 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1321 Builder.AddPlaceholderChunk("expression");
1322 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1323 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001324 }
1325 }
1326
1327 // GNU extensions
1328 if (LangOpts.GNUMode) {
1329 // FIXME: Enable when we actually support decimal floating point.
Douglas Gregor78a21012010-01-14 16:01:26 +00001330 // Results.AddResult(Result("_Decimal32"));
1331 // Results.AddResult(Result("_Decimal64"));
1332 // Results.AddResult(Result("_Decimal128"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001333
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001334 Builder.AddTypedTextChunk("typeof");
1335 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1336 Builder.AddPlaceholderChunk("expression");
1337 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001338
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001339 Builder.AddTypedTextChunk("typeof");
1340 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1341 Builder.AddPlaceholderChunk("type");
1342 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1343 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor3545ff42009-09-21 16:56:56 +00001344 }
1345}
1346
John McCallfaf5fb42010-08-26 23:41:50 +00001347static void AddStorageSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001348 const LangOptions &LangOpts,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001349 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001350 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001351 // Note: we don't suggest either "auto" or "register", because both
1352 // are pointless as storage specifiers. Elsewhere, we suggest "auto"
1353 // in C++0x as a type specifier.
Douglas Gregor78a21012010-01-14 16:01:26 +00001354 Results.AddResult(Result("extern"));
1355 Results.AddResult(Result("static"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001356}
1357
John McCallfaf5fb42010-08-26 23:41:50 +00001358static void AddFunctionSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001359 const LangOptions &LangOpts,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001360 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00001361 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001362 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001363 case Sema::PCC_Class:
1364 case Sema::PCC_MemberTemplate:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001365 if (LangOpts.CPlusPlus) {
Douglas Gregor78a21012010-01-14 16:01:26 +00001366 Results.AddResult(Result("explicit"));
1367 Results.AddResult(Result("friend"));
1368 Results.AddResult(Result("mutable"));
1369 Results.AddResult(Result("virtual"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001370 }
1371 // Fall through
1372
John McCallfaf5fb42010-08-26 23:41:50 +00001373 case Sema::PCC_ObjCInterface:
1374 case Sema::PCC_ObjCImplementation:
1375 case Sema::PCC_Namespace:
1376 case Sema::PCC_Template:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001377 if (LangOpts.CPlusPlus || LangOpts.C99)
Douglas Gregor78a21012010-01-14 16:01:26 +00001378 Results.AddResult(Result("inline"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001379 break;
1380
John McCallfaf5fb42010-08-26 23:41:50 +00001381 case Sema::PCC_ObjCInstanceVariableList:
1382 case Sema::PCC_Expression:
1383 case Sema::PCC_Statement:
1384 case Sema::PCC_ForInit:
1385 case Sema::PCC_Condition:
1386 case Sema::PCC_RecoveryInFunction:
1387 case Sema::PCC_Type:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001388 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor80039242011-02-15 20:33:25 +00001389 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001390 break;
1391 }
1392}
1393
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001394static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt);
1395static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt);
1396static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00001397 ResultBuilder &Results,
1398 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001399static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00001400 ResultBuilder &Results,
1401 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001402static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00001403 ResultBuilder &Results,
1404 bool NeedAt);
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001405static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt);
Douglas Gregorf1934162010-01-13 21:24:21 +00001406
Douglas Gregorf4c33342010-05-28 00:22:41 +00001407static void AddTypedefResult(ResultBuilder &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001408 CodeCompletionBuilder Builder(Results.getAllocator(),
1409 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001410 Builder.AddTypedTextChunk("typedef");
1411 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1412 Builder.AddPlaceholderChunk("type");
1413 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1414 Builder.AddPlaceholderChunk("name");
1415 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001416}
1417
John McCallfaf5fb42010-08-26 23:41:50 +00001418static bool WantTypesInContext(Sema::ParserCompletionContext CCC,
Douglas Gregor70febae2010-05-28 00:49:12 +00001419 const LangOptions &LangOpts) {
Douglas Gregor70febae2010-05-28 00:49:12 +00001420 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001421 case Sema::PCC_Namespace:
1422 case Sema::PCC_Class:
1423 case Sema::PCC_ObjCInstanceVariableList:
1424 case Sema::PCC_Template:
1425 case Sema::PCC_MemberTemplate:
1426 case Sema::PCC_Statement:
1427 case Sema::PCC_RecoveryInFunction:
1428 case Sema::PCC_Type:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001429 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor80039242011-02-15 20:33:25 +00001430 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor70febae2010-05-28 00:49:12 +00001431 return true;
1432
John McCallfaf5fb42010-08-26 23:41:50 +00001433 case Sema::PCC_Expression:
1434 case Sema::PCC_Condition:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001435 return LangOpts.CPlusPlus;
1436
1437 case Sema::PCC_ObjCInterface:
1438 case Sema::PCC_ObjCImplementation:
Douglas Gregor70febae2010-05-28 00:49:12 +00001439 return false;
1440
John McCallfaf5fb42010-08-26 23:41:50 +00001441 case Sema::PCC_ForInit:
Douglas Gregor5e35d592010-09-14 23:59:36 +00001442 return LangOpts.CPlusPlus || LangOpts.ObjC1 || LangOpts.C99;
Douglas Gregor70febae2010-05-28 00:49:12 +00001443 }
David Blaikie8a40f702012-01-17 06:56:22 +00001444
1445 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor70febae2010-05-28 00:49:12 +00001446}
1447
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00001448static PrintingPolicy getCompletionPrintingPolicy(const ASTContext &Context,
1449 const Preprocessor &PP) {
1450 PrintingPolicy Policy = Sema::getPrintingPolicy(Context, PP);
Douglas Gregore5c79d52011-10-18 21:20:17 +00001451 Policy.AnonymousTagLocations = false;
1452 Policy.SuppressStrongLifetime = true;
Douglas Gregor2e10cf92011-11-03 00:16:13 +00001453 Policy.SuppressUnwrittenScope = true;
Douglas Gregore5c79d52011-10-18 21:20:17 +00001454 return Policy;
1455}
1456
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00001457/// \brief Retrieve a printing policy suitable for code completion.
1458static PrintingPolicy getCompletionPrintingPolicy(Sema &S) {
1459 return getCompletionPrintingPolicy(S.Context, S.PP);
1460}
1461
Douglas Gregore5c79d52011-10-18 21:20:17 +00001462/// \brief Retrieve the string representation of the given type as a string
1463/// that has the appropriate lifetime for code completion.
1464///
1465/// This routine provides a fast path where we provide constant strings for
1466/// common type names.
1467static const char *GetCompletionTypeString(QualType T,
1468 ASTContext &Context,
1469 const PrintingPolicy &Policy,
1470 CodeCompletionAllocator &Allocator) {
1471 if (!T.getLocalQualifiers()) {
1472 // Built-in type names are constant strings.
1473 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T))
Argyrios Kyrtzidisbbff3da2012-05-05 04:20:28 +00001474 return BT->getNameAsCString(Policy);
Douglas Gregore5c79d52011-10-18 21:20:17 +00001475
1476 // Anonymous tag types are constant strings.
1477 if (const TagType *TagT = dyn_cast<TagType>(T))
1478 if (TagDecl *Tag = TagT->getDecl())
John McCall5ea95772013-03-09 00:54:27 +00001479 if (!Tag->hasNameForLinkage()) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001480 switch (Tag->getTagKind()) {
1481 case TTK_Struct: return "struct <anonymous>";
Joao Matosdc86f942012-08-31 18:45:21 +00001482 case TTK_Interface: return "__interface <anonymous>";
1483 case TTK_Class: return "class <anonymous>";
Douglas Gregore5c79d52011-10-18 21:20:17 +00001484 case TTK_Union: return "union <anonymous>";
1485 case TTK_Enum: return "enum <anonymous>";
1486 }
1487 }
1488 }
1489
1490 // Slow path: format the type as a string.
1491 std::string Result;
1492 T.getAsStringInternal(Result, Policy);
1493 return Allocator.CopyString(Result);
1494}
1495
Douglas Gregord8c61782012-02-15 15:34:24 +00001496/// \brief Add a completion for "this", if we're in a member function.
1497static void addThisCompletion(Sema &S, ResultBuilder &Results) {
1498 QualType ThisTy = S.getCurrentThisType();
1499 if (ThisTy.isNull())
1500 return;
1501
1502 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001503 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregord8c61782012-02-15 15:34:24 +00001504 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
1505 Builder.AddResultTypeChunk(GetCompletionTypeString(ThisTy,
1506 S.Context,
1507 Policy,
1508 Allocator));
1509 Builder.AddTypedTextChunk("this");
Joao Matosdc86f942012-08-31 18:45:21 +00001510 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregord8c61782012-02-15 15:34:24 +00001511}
1512
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001513/// \brief Add language constructs that show up for "ordinary" names.
John McCallfaf5fb42010-08-26 23:41:50 +00001514static void AddOrdinaryNameResults(Sema::ParserCompletionContext CCC,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001515 Scope *S,
1516 Sema &SemaRef,
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001517 ResultBuilder &Results) {
Douglas Gregore5c79d52011-10-18 21:20:17 +00001518 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00001519 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregore5c79d52011-10-18 21:20:17 +00001520 PrintingPolicy Policy = getCompletionPrintingPolicy(SemaRef);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001521
John McCall276321a2010-08-25 06:19:51 +00001522 typedef CodeCompletionResult Result;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001523 switch (CCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00001524 case Sema::PCC_Namespace:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001525 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001526 if (Results.includeCodePatterns()) {
1527 // namespace <identifier> { declarations }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001528 Builder.AddTypedTextChunk("namespace");
1529 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1530 Builder.AddPlaceholderChunk("identifier");
1531 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1532 Builder.AddPlaceholderChunk("declarations");
1533 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1534 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1535 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001536 }
1537
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001538 // namespace identifier = identifier ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001539 Builder.AddTypedTextChunk("namespace");
1540 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1541 Builder.AddPlaceholderChunk("name");
1542 Builder.AddChunk(CodeCompletionString::CK_Equal);
1543 Builder.AddPlaceholderChunk("namespace");
1544 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001545
1546 // Using directives
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001547 Builder.AddTypedTextChunk("using");
1548 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1549 Builder.AddTextChunk("namespace");
1550 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1551 Builder.AddPlaceholderChunk("identifier");
1552 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001553
1554 // asm(string-literal)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001555 Builder.AddTypedTextChunk("asm");
1556 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1557 Builder.AddPlaceholderChunk("string-literal");
1558 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1559 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001560
Douglas Gregorf4c33342010-05-28 00:22:41 +00001561 if (Results.includeCodePatterns()) {
1562 // Explicit template instantiation
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001563 Builder.AddTypedTextChunk("template");
1564 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1565 Builder.AddPlaceholderChunk("declaration");
1566 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00001567 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001568 }
Douglas Gregorf1934162010-01-13 21:24:21 +00001569
David Blaikiebbafb8a2012-03-11 07:00:24 +00001570 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001571 AddObjCTopLevelResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001572
Douglas Gregorf4c33342010-05-28 00:22:41 +00001573 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001574 // Fall through
1575
John McCallfaf5fb42010-08-26 23:41:50 +00001576 case Sema::PCC_Class:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001577 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001578 // Using declaration
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001579 Builder.AddTypedTextChunk("using");
1580 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1581 Builder.AddPlaceholderChunk("qualifier");
1582 Builder.AddTextChunk("::");
1583 Builder.AddPlaceholderChunk("name");
1584 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001585
Douglas Gregorf4c33342010-05-28 00:22:41 +00001586 // using typename qualifier::name (only in a dependent context)
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001587 if (SemaRef.CurContext->isDependentContext()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001588 Builder.AddTypedTextChunk("using");
1589 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1590 Builder.AddTextChunk("typename");
1591 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1592 Builder.AddPlaceholderChunk("qualifier");
1593 Builder.AddTextChunk("::");
1594 Builder.AddPlaceholderChunk("name");
1595 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001596 }
1597
John McCallfaf5fb42010-08-26 23:41:50 +00001598 if (CCC == Sema::PCC_Class) {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001599 AddTypedefResult(Results);
1600
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001601 // public:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001602 Builder.AddTypedTextChunk("public");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001603 if (Results.includeCodePatterns())
1604 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001605 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001606
1607 // protected:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001608 Builder.AddTypedTextChunk("protected");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001609 if (Results.includeCodePatterns())
1610 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001611 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001612
1613 // private:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001614 Builder.AddTypedTextChunk("private");
Douglas Gregor9489cdf2012-04-10 17:56:28 +00001615 if (Results.includeCodePatterns())
1616 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001617 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001618 }
1619 }
1620 // Fall through
1621
John McCallfaf5fb42010-08-26 23:41:50 +00001622 case Sema::PCC_Template:
1623 case Sema::PCC_MemberTemplate:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001624 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001625 // template < parameters >
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001626 Builder.AddTypedTextChunk("template");
1627 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1628 Builder.AddPlaceholderChunk("parameters");
1629 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1630 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001631 }
1632
David Blaikiebbafb8a2012-03-11 07:00:24 +00001633 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1634 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001635 break;
1636
John McCallfaf5fb42010-08-26 23:41:50 +00001637 case Sema::PCC_ObjCInterface:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001638 AddObjCInterfaceResults(SemaRef.getLangOpts(), Results, true);
1639 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1640 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001641 break;
1642
John McCallfaf5fb42010-08-26 23:41:50 +00001643 case Sema::PCC_ObjCImplementation:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001644 AddObjCImplementationResults(SemaRef.getLangOpts(), Results, true);
1645 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1646 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorf1934162010-01-13 21:24:21 +00001647 break;
1648
John McCallfaf5fb42010-08-26 23:41:50 +00001649 case Sema::PCC_ObjCInstanceVariableList:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001650 AddObjCVisibilityResults(SemaRef.getLangOpts(), Results, true);
Douglas Gregor48d46252010-01-13 21:54:15 +00001651 break;
1652
John McCallfaf5fb42010-08-26 23:41:50 +00001653 case Sema::PCC_RecoveryInFunction:
1654 case Sema::PCC_Statement: {
Douglas Gregorf4c33342010-05-28 00:22:41 +00001655 AddTypedefResult(Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001656
David Blaikiebbafb8a2012-03-11 07:00:24 +00001657 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns() &&
1658 SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001659 Builder.AddTypedTextChunk("try");
1660 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1661 Builder.AddPlaceholderChunk("statements");
1662 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1663 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1664 Builder.AddTextChunk("catch");
1665 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1666 Builder.AddPlaceholderChunk("declaration");
1667 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1668 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1669 Builder.AddPlaceholderChunk("statements");
1670 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1671 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1672 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001673 }
David Blaikiebbafb8a2012-03-11 07:00:24 +00001674 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorf98e6a22010-01-13 23:51:12 +00001675 AddObjCStatementResults(Results, true);
Douglas Gregorf1934162010-01-13 21:24:21 +00001676
Douglas Gregorf64acca2010-05-25 21:41:55 +00001677 if (Results.includeCodePatterns()) {
1678 // if (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001679 Builder.AddTypedTextChunk("if");
1680 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001681 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001682 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001683 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001684 Builder.AddPlaceholderChunk("expression");
1685 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1686 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1687 Builder.AddPlaceholderChunk("statements");
1688 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1689 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1690 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001691
Douglas Gregorf64acca2010-05-25 21:41:55 +00001692 // switch (condition) { }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001693 Builder.AddTypedTextChunk("switch");
1694 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001695 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001696 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001697 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001698 Builder.AddPlaceholderChunk("expression");
1699 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1700 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1701 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1702 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1703 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001704 }
1705
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001706 // Switch-specific statements.
John McCallaab3e412010-08-25 08:40:02 +00001707 if (!SemaRef.getCurFunction()->SwitchStack.empty()) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001708 // case expression:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001709 Builder.AddTypedTextChunk("case");
1710 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1711 Builder.AddPlaceholderChunk("expression");
1712 Builder.AddChunk(CodeCompletionString::CK_Colon);
1713 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001714
1715 // default:
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001716 Builder.AddTypedTextChunk("default");
1717 Builder.AddChunk(CodeCompletionString::CK_Colon);
1718 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001719 }
1720
Douglas Gregorf64acca2010-05-25 21:41:55 +00001721 if (Results.includeCodePatterns()) {
1722 /// while (condition) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001723 Builder.AddTypedTextChunk("while");
1724 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001725 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001726 Builder.AddPlaceholderChunk("condition");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001727 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001728 Builder.AddPlaceholderChunk("expression");
1729 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1730 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1731 Builder.AddPlaceholderChunk("statements");
1732 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1733 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1734 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001735
1736 // do { statements } while ( expression );
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001737 Builder.AddTypedTextChunk("do");
1738 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1739 Builder.AddPlaceholderChunk("statements");
1740 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1741 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1742 Builder.AddTextChunk("while");
1743 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1744 Builder.AddPlaceholderChunk("expression");
1745 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1746 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001747
Douglas Gregorf64acca2010-05-25 21:41:55 +00001748 // for ( for-init-statement ; condition ; expression ) { statements }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001749 Builder.AddTypedTextChunk("for");
1750 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00001751 if (SemaRef.getLangOpts().CPlusPlus || SemaRef.getLangOpts().C99)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001752 Builder.AddPlaceholderChunk("init-statement");
Douglas Gregorf64acca2010-05-25 21:41:55 +00001753 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001754 Builder.AddPlaceholderChunk("init-expression");
1755 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1756 Builder.AddPlaceholderChunk("condition");
1757 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1758 Builder.AddPlaceholderChunk("inc-expression");
1759 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1760 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1761 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1762 Builder.AddPlaceholderChunk("statements");
1763 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1764 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1765 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf64acca2010-05-25 21:41:55 +00001766 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001767
1768 if (S->getContinueParent()) {
1769 // continue ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001770 Builder.AddTypedTextChunk("continue");
1771 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001772 }
1773
1774 if (S->getBreakParent()) {
1775 // break ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001776 Builder.AddTypedTextChunk("break");
1777 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001778 }
1779
1780 // "return expression ;" or "return ;", depending on whether we
1781 // know the function is void or not.
1782 bool isVoid = false;
1783 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00001784 isVoid = Function->getReturnType()->isVoidType();
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001785 else if (ObjCMethodDecl *Method
1786 = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00001787 isVoid = Method->getReturnType()->isVoidType();
Douglas Gregor9a28e842010-03-01 23:15:13 +00001788 else if (SemaRef.getCurBlock() &&
1789 !SemaRef.getCurBlock()->ReturnType.isNull())
1790 isVoid = SemaRef.getCurBlock()->ReturnType->isVoidType();
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001791 Builder.AddTypedTextChunk("return");
Douglas Gregor44272ca2010-02-18 04:06:48 +00001792 if (!isVoid) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001793 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1794 Builder.AddPlaceholderChunk("expression");
Douglas Gregor44272ca2010-02-18 04:06:48 +00001795 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001796 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001797
Douglas Gregorf4c33342010-05-28 00:22:41 +00001798 // goto identifier ;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001799 Builder.AddTypedTextChunk("goto");
1800 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1801 Builder.AddPlaceholderChunk("label");
1802 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001803
Douglas Gregorf4c33342010-05-28 00:22:41 +00001804 // Using directives
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001805 Builder.AddTypedTextChunk("using");
1806 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1807 Builder.AddTextChunk("namespace");
1808 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1809 Builder.AddPlaceholderChunk("identifier");
1810 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001811 }
1812
1813 // Fall through (for statement expressions).
John McCallfaf5fb42010-08-26 23:41:50 +00001814 case Sema::PCC_ForInit:
1815 case Sema::PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001816 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001817 // Fall through: conditions and statements can have expressions.
1818
Douglas Gregor5e35d592010-09-14 23:59:36 +00001819 case Sema::PCC_ParenthesizedExpression:
David Blaikiebbafb8a2012-03-11 07:00:24 +00001820 if (SemaRef.getLangOpts().ObjCAutoRefCount &&
John McCall31168b02011-06-15 23:02:42 +00001821 CCC == Sema::PCC_ParenthesizedExpression) {
1822 // (__bridge <type>)<expression>
1823 Builder.AddTypedTextChunk("__bridge");
1824 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1825 Builder.AddPlaceholderChunk("type");
1826 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1827 Builder.AddPlaceholderChunk("expression");
1828 Results.AddResult(Result(Builder.TakeString()));
1829
1830 // (__bridge_transfer <Objective-C type>)<expression>
1831 Builder.AddTypedTextChunk("__bridge_transfer");
1832 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1833 Builder.AddPlaceholderChunk("Objective-C type");
1834 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1835 Builder.AddPlaceholderChunk("expression");
1836 Results.AddResult(Result(Builder.TakeString()));
1837
1838 // (__bridge_retained <CF type>)<expression>
1839 Builder.AddTypedTextChunk("__bridge_retained");
1840 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1841 Builder.AddPlaceholderChunk("CF type");
1842 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1843 Builder.AddPlaceholderChunk("expression");
1844 Results.AddResult(Result(Builder.TakeString()));
1845 }
1846 // Fall through
1847
John McCallfaf5fb42010-08-26 23:41:50 +00001848 case Sema::PCC_Expression: {
David Blaikiebbafb8a2012-03-11 07:00:24 +00001849 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001850 // 'this', if we're in a non-static member function.
Douglas Gregord8c61782012-02-15 15:34:24 +00001851 addThisCompletion(SemaRef, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001852
Douglas Gregore5c79d52011-10-18 21:20:17 +00001853 // true
1854 Builder.AddResultTypeChunk("bool");
1855 Builder.AddTypedTextChunk("true");
1856 Results.AddResult(Result(Builder.TakeString()));
1857
1858 // false
1859 Builder.AddResultTypeChunk("bool");
1860 Builder.AddTypedTextChunk("false");
1861 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001862
David Blaikiebbafb8a2012-03-11 07:00:24 +00001863 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001864 // dynamic_cast < type-id > ( expression )
1865 Builder.AddTypedTextChunk("dynamic_cast");
1866 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1867 Builder.AddPlaceholderChunk("type");
1868 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1869 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1870 Builder.AddPlaceholderChunk("expression");
1871 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1872 Results.AddResult(Result(Builder.TakeString()));
1873 }
Douglas Gregorf4c33342010-05-28 00:22:41 +00001874
1875 // static_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001876 Builder.AddTypedTextChunk("static_cast");
1877 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1878 Builder.AddPlaceholderChunk("type");
1879 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1880 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1881 Builder.AddPlaceholderChunk("expression");
1882 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1883 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001884
Douglas Gregorf4c33342010-05-28 00:22:41 +00001885 // reinterpret_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001886 Builder.AddTypedTextChunk("reinterpret_cast");
1887 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1888 Builder.AddPlaceholderChunk("type");
1889 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1890 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1891 Builder.AddPlaceholderChunk("expression");
1892 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1893 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001894
Douglas Gregorf4c33342010-05-28 00:22:41 +00001895 // const_cast < type-id > ( expression )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001896 Builder.AddTypedTextChunk("const_cast");
1897 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1898 Builder.AddPlaceholderChunk("type");
1899 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1900 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1901 Builder.AddPlaceholderChunk("expression");
1902 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1903 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001904
David Blaikiebbafb8a2012-03-11 07:00:24 +00001905 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001906 // typeid ( expression-or-type )
Douglas Gregore5c79d52011-10-18 21:20:17 +00001907 Builder.AddResultTypeChunk("std::type_info");
Douglas Gregorc05f6572011-04-12 02:47:21 +00001908 Builder.AddTypedTextChunk("typeid");
1909 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1910 Builder.AddPlaceholderChunk("expression-or-type");
1911 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1912 Results.AddResult(Result(Builder.TakeString()));
1913 }
1914
Douglas Gregorf4c33342010-05-28 00:22:41 +00001915 // new T ( ... )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001916 Builder.AddTypedTextChunk("new");
1917 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1918 Builder.AddPlaceholderChunk("type");
1919 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1920 Builder.AddPlaceholderChunk("expressions");
1921 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1922 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001923
Douglas Gregorf4c33342010-05-28 00:22:41 +00001924 // new T [ ] ( ... )
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001925 Builder.AddTypedTextChunk("new");
1926 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1927 Builder.AddPlaceholderChunk("type");
1928 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1929 Builder.AddPlaceholderChunk("size");
1930 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1931 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1932 Builder.AddPlaceholderChunk("expressions");
1933 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1934 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001935
Douglas Gregorf4c33342010-05-28 00:22:41 +00001936 // delete expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001937 Builder.AddResultTypeChunk("void");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001938 Builder.AddTypedTextChunk("delete");
1939 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1940 Builder.AddPlaceholderChunk("expression");
1941 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001942
Douglas Gregorf4c33342010-05-28 00:22:41 +00001943 // delete [] expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001944 Builder.AddResultTypeChunk("void");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00001945 Builder.AddTypedTextChunk("delete");
1946 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1947 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1948 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1949 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1950 Builder.AddPlaceholderChunk("expression");
1951 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001952
David Blaikiebbafb8a2012-03-11 07:00:24 +00001953 if (SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorc05f6572011-04-12 02:47:21 +00001954 // throw expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001955 Builder.AddResultTypeChunk("void");
Douglas Gregorc05f6572011-04-12 02:47:21 +00001956 Builder.AddTypedTextChunk("throw");
1957 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1958 Builder.AddPlaceholderChunk("expression");
1959 Results.AddResult(Result(Builder.TakeString()));
1960 }
Douglas Gregor4205fef2011-10-18 16:29:03 +00001961
Douglas Gregora2db7932010-05-26 22:00:08 +00001962 // FIXME: Rethrow?
Douglas Gregor4205fef2011-10-18 16:29:03 +00001963
Richard Smith2bf7fdb2013-01-02 11:42:31 +00001964 if (SemaRef.getLangOpts().CPlusPlus11) {
Douglas Gregor4205fef2011-10-18 16:29:03 +00001965 // nullptr
Douglas Gregore5c79d52011-10-18 21:20:17 +00001966 Builder.AddResultTypeChunk("std::nullptr_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001967 Builder.AddTypedTextChunk("nullptr");
1968 Results.AddResult(Result(Builder.TakeString()));
1969
1970 // alignof
Douglas Gregore5c79d52011-10-18 21:20:17 +00001971 Builder.AddResultTypeChunk("size_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001972 Builder.AddTypedTextChunk("alignof");
1973 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1974 Builder.AddPlaceholderChunk("type");
1975 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1976 Results.AddResult(Result(Builder.TakeString()));
1977
1978 // noexcept
Douglas Gregore5c79d52011-10-18 21:20:17 +00001979 Builder.AddResultTypeChunk("bool");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001980 Builder.AddTypedTextChunk("noexcept");
1981 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1982 Builder.AddPlaceholderChunk("expression");
1983 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1984 Results.AddResult(Result(Builder.TakeString()));
1985
1986 // sizeof... expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00001987 Builder.AddResultTypeChunk("size_t");
Douglas Gregor4205fef2011-10-18 16:29:03 +00001988 Builder.AddTypedTextChunk("sizeof...");
1989 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1990 Builder.AddPlaceholderChunk("parameter-pack");
1991 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1992 Results.AddResult(Result(Builder.TakeString()));
1993 }
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001994 }
1995
David Blaikiebbafb8a2012-03-11 07:00:24 +00001996 if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor504a6ae2010-01-10 23:08:15 +00001997 // Add "super", if we're in an Objective-C class with a superclass.
Ted Kremenek305a0a72010-05-31 21:43:10 +00001998 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) {
1999 // The interface can be NULL.
2000 if (ObjCInterfaceDecl *ID = Method->getClassInterface())
Douglas Gregore5c79d52011-10-18 21:20:17 +00002001 if (ID->getSuperClass()) {
2002 std::string SuperType;
2003 SuperType = ID->getSuperClass()->getNameAsString();
2004 if (Method->isInstanceMethod())
2005 SuperType += " *";
2006
2007 Builder.AddResultTypeChunk(Allocator.CopyString(SuperType));
2008 Builder.AddTypedTextChunk("super");
2009 Results.AddResult(Result(Builder.TakeString()));
2010 }
Ted Kremenek305a0a72010-05-31 21:43:10 +00002011 }
2012
Douglas Gregorf98e6a22010-01-13 23:51:12 +00002013 AddObjCExpressionResults(Results, true);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002014 }
2015
Jordan Rose58d54722012-06-30 21:33:57 +00002016 if (SemaRef.getLangOpts().C11) {
2017 // _Alignof
2018 Builder.AddResultTypeChunk("size_t");
2019 if (SemaRef.getASTContext().Idents.get("alignof").hasMacroDefinition())
2020 Builder.AddTypedTextChunk("alignof");
2021 else
2022 Builder.AddTypedTextChunk("_Alignof");
2023 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2024 Builder.AddPlaceholderChunk("type");
2025 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2026 Results.AddResult(Result(Builder.TakeString()));
2027 }
2028
Douglas Gregorf4c33342010-05-28 00:22:41 +00002029 // sizeof expression
Douglas Gregore5c79d52011-10-18 21:20:17 +00002030 Builder.AddResultTypeChunk("size_t");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002031 Builder.AddTypedTextChunk("sizeof");
2032 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2033 Builder.AddPlaceholderChunk("expression-or-type");
2034 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2035 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002036 break;
2037 }
Douglas Gregor99fa2642010-08-24 01:06:58 +00002038
John McCallfaf5fb42010-08-26 23:41:50 +00002039 case Sema::PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00002040 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor99fa2642010-08-24 01:06:58 +00002041 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002042 }
2043
David Blaikiebbafb8a2012-03-11 07:00:24 +00002044 if (WantTypesInContext(CCC, SemaRef.getLangOpts()))
2045 AddTypeSpecifierResults(SemaRef.getLangOpts(), Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002046
David Blaikiebbafb8a2012-03-11 07:00:24 +00002047 if (SemaRef.getLangOpts().CPlusPlus && CCC != Sema::PCC_Type)
Douglas Gregor78a21012010-01-14 16:01:26 +00002048 Results.AddResult(Result("operator"));
Douglas Gregor504a6ae2010-01-10 23:08:15 +00002049}
2050
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002051/// \brief If the given declaration has an associated type, add it as a result
2052/// type chunk.
2053static void AddResultTypeChunk(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002054 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002055 const NamedDecl *ND,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002056 CodeCompletionBuilder &Result) {
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002057 if (!ND)
2058 return;
Douglas Gregor0212fd72010-09-21 16:06:22 +00002059
2060 // Skip constructors and conversion functions, which have their return types
2061 // built into their names.
2062 if (isa<CXXConstructorDecl>(ND) || isa<CXXConversionDecl>(ND))
2063 return;
2064
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002065 // Determine the type of the declaration (if it has a type).
Alp Tokera2794f92014-01-22 07:29:52 +00002066 QualType T;
2067 if (const FunctionDecl *Function = ND->getAsFunction())
Alp Toker314cc812014-01-25 16:55:45 +00002068 T = Function->getReturnType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002069 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Alp Toker314cc812014-01-25 16:55:45 +00002070 T = Method->getReturnType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002071 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002072 T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext()));
2073 else if (isa<UnresolvedUsingValueDecl>(ND)) {
2074 /* Do nothing: ignore unresolved using declarations*/
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002075 } else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND)) {
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002076 T = Value->getType();
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002077 } else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002078 T = Property->getType();
2079
2080 if (T.isNull() || Context.hasSameType(T, Context.DependentTy))
2081 return;
2082
Douglas Gregor75acd922011-09-27 23:30:47 +00002083 Result.AddResultTypeChunk(GetCompletionTypeString(T, Context, Policy,
Douglas Gregor304f9b02011-02-01 21:15:40 +00002084 Result.getAllocator()));
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002085}
2086
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002087static void MaybeAddSentinel(ASTContext &Context,
2088 const NamedDecl *FunctionOrMethod,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002089 CodeCompletionBuilder &Result) {
Douglas Gregordbb71db2010-08-23 23:51:41 +00002090 if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>())
2091 if (Sentinel->getSentinel() == 0) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00002092 if (Context.getLangOpts().ObjC1 &&
Douglas Gregordbb71db2010-08-23 23:51:41 +00002093 Context.Idents.get("nil").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002094 Result.AddTextChunk(", nil");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002095 else if (Context.Idents.get("NULL").hasMacroDefinition())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002096 Result.AddTextChunk(", NULL");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002097 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002098 Result.AddTextChunk(", (void*)0");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002099 }
2100}
2101
Douglas Gregor8f08d742011-07-30 07:55:26 +00002102static std::string formatObjCParamQualifiers(unsigned ObjCQuals) {
2103 std::string Result;
2104 if (ObjCQuals & Decl::OBJC_TQ_In)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002105 Result += "in ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002106 else if (ObjCQuals & Decl::OBJC_TQ_Inout)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002107 Result += "inout ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002108 else if (ObjCQuals & Decl::OBJC_TQ_Out)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002109 Result += "out ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002110 if (ObjCQuals & Decl::OBJC_TQ_Bycopy)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002111 Result += "bycopy ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002112 else if (ObjCQuals & Decl::OBJC_TQ_Byref)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002113 Result += "byref ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002114 if (ObjCQuals & Decl::OBJC_TQ_Oneway)
Douglas Gregor407d1f92011-11-09 02:13:45 +00002115 Result += "oneway ";
Douglas Gregor8f08d742011-07-30 07:55:26 +00002116 return Result;
2117}
2118
Douglas Gregore90dd002010-08-24 16:15:59 +00002119static std::string FormatFunctionParameter(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002120 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002121 const ParmVarDecl *Param,
Douglas Gregord793e7c2011-10-18 04:23:19 +00002122 bool SuppressName = false,
2123 bool SuppressBlock = false) {
Douglas Gregore90dd002010-08-24 16:15:59 +00002124 bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext());
2125 if (Param->getType()->isDependentType() ||
2126 !Param->getType()->isBlockPointerType()) {
2127 // The argument for a dependent or non-block parameter is a placeholder
2128 // containing that parameter's type.
2129 std::string Result;
2130
Douglas Gregor981a0c42010-08-29 19:47:46 +00002131 if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002132 Result = Param->getIdentifier()->getName();
2133
John McCall31168b02011-06-15 23:02:42 +00002134 Param->getType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002135
2136 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002137 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2138 + Result + ")";
Douglas Gregor981a0c42010-08-29 19:47:46 +00002139 if (Param->getIdentifier() && !SuppressName)
Douglas Gregore90dd002010-08-24 16:15:59 +00002140 Result += Param->getIdentifier()->getName();
2141 }
2142 return Result;
2143 }
2144
2145 // The argument for a block pointer parameter is a block literal with
2146 // the appropriate type.
David Blaikie6adc78e2013-02-18 22:06:02 +00002147 FunctionTypeLoc Block;
2148 FunctionProtoTypeLoc BlockProto;
Douglas Gregore90dd002010-08-24 16:15:59 +00002149 TypeLoc TL;
2150 if (TypeSourceInfo *TSInfo = Param->getTypeSourceInfo()) {
2151 TL = TSInfo->getTypeLoc().getUnqualifiedLoc();
2152 while (true) {
2153 // Look through typedefs.
Douglas Gregord793e7c2011-10-18 04:23:19 +00002154 if (!SuppressBlock) {
David Blaikie6adc78e2013-02-18 22:06:02 +00002155 if (TypedefTypeLoc TypedefTL = TL.getAs<TypedefTypeLoc>()) {
2156 if (TypeSourceInfo *InnerTSInfo =
2157 TypedefTL.getTypedefNameDecl()->getTypeSourceInfo()) {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002158 TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc();
2159 continue;
2160 }
2161 }
2162
2163 // Look through qualified types
David Blaikie6adc78e2013-02-18 22:06:02 +00002164 if (QualifiedTypeLoc QualifiedTL = TL.getAs<QualifiedTypeLoc>()) {
2165 TL = QualifiedTL.getUnqualifiedLoc();
Douglas Gregore90dd002010-08-24 16:15:59 +00002166 continue;
2167 }
2168 }
2169
Douglas Gregore90dd002010-08-24 16:15:59 +00002170 // Try to get the function prototype behind the block pointer type,
2171 // then we're done.
David Blaikie6adc78e2013-02-18 22:06:02 +00002172 if (BlockPointerTypeLoc BlockPtr = TL.getAs<BlockPointerTypeLoc>()) {
2173 TL = BlockPtr.getPointeeLoc().IgnoreParens();
2174 Block = TL.getAs<FunctionTypeLoc>();
2175 BlockProto = TL.getAs<FunctionProtoTypeLoc>();
Douglas Gregore90dd002010-08-24 16:15:59 +00002176 }
2177 break;
2178 }
2179 }
2180
2181 if (!Block) {
2182 // We were unable to find a FunctionProtoTypeLoc with parameter names
2183 // for the block; just use the parameter type as a placeholder.
2184 std::string Result;
Douglas Gregord793e7c2011-10-18 04:23:19 +00002185 if (!ObjCMethodParam && Param->getIdentifier())
2186 Result = Param->getIdentifier()->getName();
2187
John McCall31168b02011-06-15 23:02:42 +00002188 Param->getType().getUnqualifiedType().getAsStringInternal(Result, Policy);
Douglas Gregore90dd002010-08-24 16:15:59 +00002189
2190 if (ObjCMethodParam) {
Douglas Gregor8f08d742011-07-30 07:55:26 +00002191 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2192 + Result + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002193 if (Param->getIdentifier())
2194 Result += Param->getIdentifier()->getName();
2195 }
2196
2197 return Result;
2198 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002199
Douglas Gregore90dd002010-08-24 16:15:59 +00002200 // We have the function prototype behind the block pointer type, as it was
2201 // written in the source.
Douglas Gregor67da50e2010-09-08 22:47:51 +00002202 std::string Result;
Alp Toker314cc812014-01-25 16:55:45 +00002203 QualType ResultType = Block.getTypePtr()->getReturnType();
Douglas Gregord793e7c2011-10-18 04:23:19 +00002204 if (!ResultType->isVoidType() || SuppressBlock)
John McCall31168b02011-06-15 23:02:42 +00002205 ResultType.getAsStringInternal(Result, Policy);
Douglas Gregord793e7c2011-10-18 04:23:19 +00002206
2207 // Format the parameter list.
2208 std::string Params;
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002209 if (!BlockProto || Block.getNumParams() == 0) {
David Blaikie6adc78e2013-02-18 22:06:02 +00002210 if (BlockProto && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002211 Params = "(...)";
Douglas Gregoraf25cfa2010-10-02 23:49:58 +00002212 else
Douglas Gregord793e7c2011-10-18 04:23:19 +00002213 Params = "(void)";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002214 } else {
Douglas Gregord793e7c2011-10-18 04:23:19 +00002215 Params += "(";
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002216 for (unsigned I = 0, N = Block.getNumParams(); I != N; ++I) {
Douglas Gregor67da50e2010-09-08 22:47:51 +00002217 if (I)
Douglas Gregord793e7c2011-10-18 04:23:19 +00002218 Params += ", ";
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002219 Params += FormatFunctionParameter(Context, Policy, Block.getParam(I),
2220 /*SuppressName=*/false,
Douglas Gregord793e7c2011-10-18 04:23:19 +00002221 /*SuppressBlock=*/true);
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002222
David Blaikie6adc78e2013-02-18 22:06:02 +00002223 if (I == N - 1 && BlockProto.getTypePtr()->isVariadic())
Douglas Gregord793e7c2011-10-18 04:23:19 +00002224 Params += ", ...";
Douglas Gregor67da50e2010-09-08 22:47:51 +00002225 }
Douglas Gregord793e7c2011-10-18 04:23:19 +00002226 Params += ")";
Douglas Gregor400f5972010-08-31 05:13:43 +00002227 }
Douglas Gregor67da50e2010-09-08 22:47:51 +00002228
Douglas Gregord793e7c2011-10-18 04:23:19 +00002229 if (SuppressBlock) {
2230 // Format as a parameter.
2231 Result = Result + " (^";
2232 if (Param->getIdentifier())
2233 Result += Param->getIdentifier()->getName();
2234 Result += ")";
2235 Result += Params;
2236 } else {
2237 // Format as a block literal argument.
2238 Result = '^' + Result;
2239 Result += Params;
2240
2241 if (Param->getIdentifier())
2242 Result += Param->getIdentifier()->getName();
2243 }
2244
Douglas Gregore90dd002010-08-24 16:15:59 +00002245 return Result;
2246}
2247
Douglas Gregor3545ff42009-09-21 16:56:56 +00002248/// \brief Add function parameter chunks to the given code completion string.
2249static void AddFunctionParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002250 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002251 const FunctionDecl *Function,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002252 CodeCompletionBuilder &Result,
2253 unsigned Start = 0,
2254 bool InOptional = false) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002255 bool FirstParameter = true;
Douglas Gregor9eb77012009-11-07 00:00:49 +00002256
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002257 for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002258 const ParmVarDecl *Param = Function->getParamDecl(P);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002259
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002260 if (Param->hasDefaultArg() && !InOptional) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002261 // When we see an optional default argument, put that argument and
2262 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002263 CodeCompletionBuilder Opt(Result.getAllocator(),
2264 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002265 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002266 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002267 AddFunctionParameterChunks(Context, Policy, Function, Opt, P, true);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002268 Result.AddOptionalChunk(Opt.TakeString());
2269 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002270 }
2271
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002272 if (FirstParameter)
2273 FirstParameter = false;
2274 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002275 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002276
2277 InOptional = false;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002278
2279 // Format the placeholder string.
Douglas Gregor75acd922011-09-27 23:30:47 +00002280 std::string PlaceholderStr = FormatFunctionParameter(Context, Policy,
2281 Param);
Douglas Gregore90dd002010-08-24 16:15:59 +00002282
Douglas Gregor400f5972010-08-31 05:13:43 +00002283 if (Function->isVariadic() && P == N - 1)
2284 PlaceholderStr += ", ...";
2285
Douglas Gregor3545ff42009-09-21 16:56:56 +00002286 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002287 Result.AddPlaceholderChunk(
2288 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002289 }
Douglas Gregorba449032009-09-22 21:42:17 +00002290
2291 if (const FunctionProtoType *Proto
2292 = Function->getType()->getAs<FunctionProtoType>())
Douglas Gregordbb71db2010-08-23 23:51:41 +00002293 if (Proto->isVariadic()) {
Alp Toker9cacbab2014-01-20 20:26:09 +00002294 if (Proto->getNumParams() == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002295 Result.AddPlaceholderChunk("...");
Douglas Gregordbb71db2010-08-23 23:51:41 +00002296
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002297 MaybeAddSentinel(Context, Function, Result);
Douglas Gregordbb71db2010-08-23 23:51:41 +00002298 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002299}
2300
2301/// \brief Add template parameter chunks to the given code completion string.
2302static void AddTemplateParameterChunks(ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00002303 const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002304 const TemplateDecl *Template,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002305 CodeCompletionBuilder &Result,
2306 unsigned MaxParameters = 0,
2307 unsigned Start = 0,
2308 bool InDefaultArg = false) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002309 bool FirstParameter = true;
2310
2311 TemplateParameterList *Params = Template->getTemplateParameters();
2312 TemplateParameterList::iterator PEnd = Params->end();
2313 if (MaxParameters)
2314 PEnd = Params->begin() + MaxParameters;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002315 for (TemplateParameterList::iterator P = Params->begin() + Start;
2316 P != PEnd; ++P) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002317 bool HasDefaultArg = false;
2318 std::string PlaceholderStr;
2319 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) {
2320 if (TTP->wasDeclaredWithTypename())
2321 PlaceholderStr = "typename";
2322 else
2323 PlaceholderStr = "class";
2324
2325 if (TTP->getIdentifier()) {
2326 PlaceholderStr += ' ';
2327 PlaceholderStr += TTP->getIdentifier()->getName();
2328 }
2329
2330 HasDefaultArg = TTP->hasDefaultArgument();
2331 } else if (NonTypeTemplateParmDecl *NTTP
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002332 = dyn_cast<NonTypeTemplateParmDecl>(*P)) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002333 if (NTTP->getIdentifier())
2334 PlaceholderStr = NTTP->getIdentifier()->getName();
John McCall31168b02011-06-15 23:02:42 +00002335 NTTP->getType().getAsStringInternal(PlaceholderStr, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002336 HasDefaultArg = NTTP->hasDefaultArgument();
2337 } else {
2338 assert(isa<TemplateTemplateParmDecl>(*P));
2339 TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P);
2340
2341 // Since putting the template argument list into the placeholder would
2342 // be very, very long, we just use an abbreviation.
2343 PlaceholderStr = "template<...> class";
2344 if (TTP->getIdentifier()) {
2345 PlaceholderStr += ' ';
2346 PlaceholderStr += TTP->getIdentifier()->getName();
2347 }
2348
2349 HasDefaultArg = TTP->hasDefaultArgument();
2350 }
2351
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002352 if (HasDefaultArg && !InDefaultArg) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00002353 // When we see an optional default argument, put that argument and
2354 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002355 CodeCompletionBuilder Opt(Result.getAllocator(),
2356 Result.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002357 if (!FirstParameter)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002358 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor75acd922011-09-27 23:30:47 +00002359 AddTemplateParameterChunks(Context, Policy, Template, Opt, MaxParameters,
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002360 P - Params->begin(), true);
2361 Result.AddOptionalChunk(Opt.TakeString());
2362 break;
Douglas Gregor3545ff42009-09-21 16:56:56 +00002363 }
2364
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002365 InDefaultArg = false;
2366
Douglas Gregor3545ff42009-09-21 16:56:56 +00002367 if (FirstParameter)
2368 FirstParameter = false;
2369 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002370 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002371
2372 // Add the placeholder string.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002373 Result.AddPlaceholderChunk(
2374 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor3545ff42009-09-21 16:56:56 +00002375 }
2376}
2377
Douglas Gregorf2510672009-09-21 19:57:38 +00002378/// \brief Add a qualifier to the given code-completion string, if the
2379/// provided nested-name-specifier is non-NULL.
Douglas Gregor0f622362009-12-11 18:44:16 +00002380static void
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002381AddQualifierToCompletionString(CodeCompletionBuilder &Result,
Douglas Gregor0f622362009-12-11 18:44:16 +00002382 NestedNameSpecifier *Qualifier,
2383 bool QualifierIsInformative,
Douglas Gregor75acd922011-09-27 23:30:47 +00002384 ASTContext &Context,
2385 const PrintingPolicy &Policy) {
Douglas Gregorf2510672009-09-21 19:57:38 +00002386 if (!Qualifier)
2387 return;
2388
2389 std::string PrintedNNS;
2390 {
2391 llvm::raw_string_ostream OS(PrintedNNS);
Douglas Gregor75acd922011-09-27 23:30:47 +00002392 Qualifier->print(OS, Policy);
Douglas Gregorf2510672009-09-21 19:57:38 +00002393 }
Douglas Gregor5bf52692009-09-22 23:15:58 +00002394 if (QualifierIsInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002395 Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregor5bf52692009-09-22 23:15:58 +00002396 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002397 Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregorf2510672009-09-21 19:57:38 +00002398}
2399
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002400static void
2401AddFunctionTypeQualsToCompletionString(CodeCompletionBuilder &Result,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002402 const FunctionDecl *Function) {
Douglas Gregor0f622362009-12-11 18:44:16 +00002403 const FunctionProtoType *Proto
2404 = Function->getType()->getAs<FunctionProtoType>();
2405 if (!Proto || !Proto->getTypeQuals())
2406 return;
2407
Douglas Gregor304f9b02011-02-01 21:15:40 +00002408 // FIXME: Add ref-qualifier!
2409
2410 // Handle single qualifiers without copying
2411 if (Proto->getTypeQuals() == Qualifiers::Const) {
2412 Result.AddInformativeChunk(" const");
2413 return;
2414 }
2415
2416 if (Proto->getTypeQuals() == Qualifiers::Volatile) {
2417 Result.AddInformativeChunk(" volatile");
2418 return;
2419 }
2420
2421 if (Proto->getTypeQuals() == Qualifiers::Restrict) {
2422 Result.AddInformativeChunk(" restrict");
2423 return;
2424 }
2425
2426 // Handle multiple qualifiers.
Douglas Gregor0f622362009-12-11 18:44:16 +00002427 std::string QualsStr;
David Blaikief5697e52012-08-10 00:55:35 +00002428 if (Proto->isConst())
Douglas Gregor0f622362009-12-11 18:44:16 +00002429 QualsStr += " const";
David Blaikief5697e52012-08-10 00:55:35 +00002430 if (Proto->isVolatile())
Douglas Gregor0f622362009-12-11 18:44:16 +00002431 QualsStr += " volatile";
David Blaikief5697e52012-08-10 00:55:35 +00002432 if (Proto->isRestrict())
Douglas Gregor0f622362009-12-11 18:44:16 +00002433 QualsStr += " restrict";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002434 Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr));
Douglas Gregor0f622362009-12-11 18:44:16 +00002435}
2436
Douglas Gregor0212fd72010-09-21 16:06:22 +00002437/// \brief Add the name of the given declaration
Douglas Gregor75acd922011-09-27 23:30:47 +00002438static void AddTypedNameChunk(ASTContext &Context, const PrintingPolicy &Policy,
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002439 const NamedDecl *ND,
2440 CodeCompletionBuilder &Result) {
Douglas Gregor0212fd72010-09-21 16:06:22 +00002441 DeclarationName Name = ND->getDeclName();
2442 if (!Name)
2443 return;
2444
2445 switch (Name.getNameKind()) {
Douglas Gregor304f9b02011-02-01 21:15:40 +00002446 case DeclarationName::CXXOperatorName: {
2447 const char *OperatorName = 0;
2448 switch (Name.getCXXOverloadedOperator()) {
2449 case OO_None:
2450 case OO_Conditional:
2451 case NUM_OVERLOADED_OPERATORS:
2452 OperatorName = "operator";
2453 break;
2454
2455#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
2456 case OO_##Name: OperatorName = "operator" Spelling; break;
2457#define OVERLOADED_OPERATOR_MULTI(Name,Spelling,Unary,Binary,MemberOnly)
2458#include "clang/Basic/OperatorKinds.def"
2459
2460 case OO_New: OperatorName = "operator new"; break;
2461 case OO_Delete: OperatorName = "operator delete"; break;
2462 case OO_Array_New: OperatorName = "operator new[]"; break;
2463 case OO_Array_Delete: OperatorName = "operator delete[]"; break;
2464 case OO_Call: OperatorName = "operator()"; break;
2465 case OO_Subscript: OperatorName = "operator[]"; break;
2466 }
2467 Result.AddTypedTextChunk(OperatorName);
2468 break;
2469 }
2470
Douglas Gregor0212fd72010-09-21 16:06:22 +00002471 case DeclarationName::Identifier:
2472 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor0212fd72010-09-21 16:06:22 +00002473 case DeclarationName::CXXDestructorName:
2474 case DeclarationName::CXXLiteralOperatorName:
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002475 Result.AddTypedTextChunk(
2476 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002477 break;
2478
2479 case DeclarationName::CXXUsingDirective:
2480 case DeclarationName::ObjCZeroArgSelector:
2481 case DeclarationName::ObjCOneArgSelector:
2482 case DeclarationName::ObjCMultiArgSelector:
2483 break;
2484
2485 case DeclarationName::CXXConstructorName: {
2486 CXXRecordDecl *Record = 0;
2487 QualType Ty = Name.getCXXNameType();
2488 if (const RecordType *RecordTy = Ty->getAs<RecordType>())
2489 Record = cast<CXXRecordDecl>(RecordTy->getDecl());
2490 else if (const InjectedClassNameType *InjectedTy
2491 = Ty->getAs<InjectedClassNameType>())
2492 Record = InjectedTy->getDecl();
2493 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002494 Result.AddTypedTextChunk(
2495 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002496 break;
2497 }
2498
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002499 Result.AddTypedTextChunk(
2500 Result.getAllocator().CopyString(Record->getNameAsString()));
Douglas Gregor0212fd72010-09-21 16:06:22 +00002501 if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002502 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Douglas Gregor75acd922011-09-27 23:30:47 +00002503 AddTemplateParameterChunks(Context, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002504 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002505 }
2506 break;
2507 }
2508 }
2509}
2510
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002511CodeCompletionString *CodeCompletionResult::CreateCodeCompletionString(Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002512 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002513 CodeCompletionTUInfo &CCTUInfo,
2514 bool IncludeBriefComments) {
2515 return CreateCodeCompletionString(S.Context, S.PP, Allocator, CCTUInfo,
2516 IncludeBriefComments);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002517}
2518
Douglas Gregor3545ff42009-09-21 16:56:56 +00002519/// \brief If possible, create a new code completion string for the given
2520/// result.
2521///
2522/// \returns Either a new, heap-allocated code completion string describing
2523/// how to use this result, or NULL to indicate that the string or name of the
2524/// result is all that is needed.
2525CodeCompletionString *
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002526CodeCompletionResult::CreateCodeCompletionString(ASTContext &Ctx,
2527 Preprocessor &PP,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002528 CodeCompletionAllocator &Allocator,
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002529 CodeCompletionTUInfo &CCTUInfo,
2530 bool IncludeBriefComments) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002531 CodeCompletionBuilder Result(Allocator, CCTUInfo, Priority, Availability);
Douglas Gregor9eb77012009-11-07 00:00:49 +00002532
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002533 PrintingPolicy Policy = getCompletionPrintingPolicy(Ctx, PP);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002534 if (Kind == RK_Pattern) {
2535 Pattern->Priority = Priority;
2536 Pattern->Availability = Availability;
Douglas Gregor78254c82012-03-27 23:34:16 +00002537
2538 if (Declaration) {
2539 Result.addParentContext(Declaration->getDeclContext());
Douglas Gregor78254c82012-03-27 23:34:16 +00002540 Pattern->ParentName = Result.getParentName();
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002541 // Provide code completion comment for self.GetterName where
2542 // GetterName is the getter method for a property with name
2543 // different from the property name (declared via a property
2544 // getter attribute.
2545 const NamedDecl *ND = Declaration;
2546 if (const ObjCMethodDecl *M = dyn_cast<ObjCMethodDecl>(ND))
2547 if (M->isPropertyAccessor())
2548 if (const ObjCPropertyDecl *PDecl = M->findPropertyDecl())
2549 if (PDecl->getGetterName() == M->getSelector() &&
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002550 PDecl->getIdentifier() != M->getIdentifier()) {
2551 if (const RawComment *RC =
2552 Ctx.getRawCommentForAnyRedecl(M)) {
Fariborz Jahanian1fcf4922013-03-22 17:55:27 +00002553 Result.addBriefComment(RC->getBriefText(Ctx));
2554 Pattern->BriefComment = Result.getBriefComment();
2555 }
Fariborz Jahanianbe8bc672013-03-23 01:10:45 +00002556 else if (const RawComment *RC =
2557 Ctx.getRawCommentForAnyRedecl(PDecl)) {
2558 Result.addBriefComment(RC->getBriefText(Ctx));
2559 Pattern->BriefComment = Result.getBriefComment();
2560 }
2561 }
Douglas Gregor78254c82012-03-27 23:34:16 +00002562 }
2563
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002564 return Pattern;
2565 }
Douglas Gregorf09935f2009-12-01 05:55:20 +00002566
Douglas Gregorf09935f2009-12-01 05:55:20 +00002567 if (Kind == RK_Keyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002568 Result.AddTypedTextChunk(Keyword);
2569 return Result.TakeString();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002570 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00002571
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002572 if (Kind == RK_Macro) {
Argyrios Kyrtzidis09c9e812013-02-20 00:54:57 +00002573 const MacroDirective *MD = PP.getMacroDirectiveHistory(Macro);
2574 assert(MD && "Not a macro?");
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00002575 const MacroInfo *MI = MD->getMacroInfo();
Douglas Gregorf09935f2009-12-01 05:55:20 +00002576
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002577 Result.AddTypedTextChunk(
2578 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregorf09935f2009-12-01 05:55:20 +00002579
2580 if (!MI->isFunctionLike())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002581 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002582
2583 // Format a function-like macro with placeholders for the arguments.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002584 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor0c505312011-07-30 08:17:44 +00002585 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002586
2587 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
2588 if (MI->isC99Varargs()) {
2589 --AEnd;
2590
2591 if (A == AEnd) {
2592 Result.AddPlaceholderChunk("...");
2593 }
Douglas Gregor0c505312011-07-30 08:17:44 +00002594 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002595
Douglas Gregor0c505312011-07-30 08:17:44 +00002596 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002597 if (A != MI->arg_begin())
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002598 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor3aa55262012-01-21 00:43:38 +00002599
2600 if (MI->isVariadic() && (A+1) == AEnd) {
Dylan Noblesmith2c1dd272012-02-05 02:13:05 +00002601 SmallString<32> Arg = (*A)->getName();
Douglas Gregor3aa55262012-01-21 00:43:38 +00002602 if (MI->isC99Varargs())
2603 Arg += ", ...";
2604 else
2605 Arg += "...";
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002606 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor3aa55262012-01-21 00:43:38 +00002607 break;
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002608 }
Douglas Gregor3aa55262012-01-21 00:43:38 +00002609
2610 // Non-variadic macros are simple.
2611 Result.AddPlaceholderChunk(
2612 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregor0c505312011-07-30 08:17:44 +00002613 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002614 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002615 return Result.TakeString();
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002616 }
2617
Douglas Gregorf64acca2010-05-25 21:41:55 +00002618 assert(Kind == RK_Declaration && "Missed a result kind?");
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002619 const NamedDecl *ND = Declaration;
Douglas Gregor78254c82012-03-27 23:34:16 +00002620 Result.addParentContext(ND->getDeclContext());
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002621
2622 if (IncludeBriefComments) {
2623 // Add documentation comment, if it exists.
Dmitri Gribenkoa43ec182012-08-11 00:51:43 +00002624 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(ND)) {
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002625 Result.addBriefComment(RC->getBriefText(Ctx));
Fariborz Jahanian15a0b552013-02-28 17:47:14 +00002626 }
2627 else if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2628 if (OMD->isPropertyAccessor())
2629 if (const ObjCPropertyDecl *PDecl = OMD->findPropertyDecl())
2630 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(PDecl))
2631 Result.addBriefComment(RC->getBriefText(Ctx));
Dmitri Gribenko3292d062012-07-02 17:35:10 +00002632 }
2633
Douglas Gregor9eb77012009-11-07 00:00:49 +00002634 if (StartsNestedNameSpecifier) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002635 Result.AddTypedTextChunk(
2636 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002637 Result.AddTextChunk("::");
2638 return Result.TakeString();
Douglas Gregor9eb77012009-11-07 00:00:49 +00002639 }
Erik Verbruggen98ea7f62011-10-14 15:31:08 +00002640
Aaron Ballmanb97112e2014-03-08 22:19:01 +00002641 for (specific_attr_iterator<AnnotateAttr>
2642 i = ND->specific_attr_begin<AnnotateAttr>(),
2643 e = ND->specific_attr_end<AnnotateAttr>();
2644 i != e; ++i)
2645 Result.AddAnnotation(
2646 Result.getAllocator().CopyString((*i)->getAnnotation()));
2647
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002648 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002649
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002650 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002651 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002652 Ctx, Policy);
2653 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002654 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002655 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002656 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002657 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002658 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002659 }
2660
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002661 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002662 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002663 Ctx, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002664 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002665 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002666
Douglas Gregor3545ff42009-09-21 16:56:56 +00002667 // Figure out which template parameters are deduced (or have default
2668 // arguments).
Benjamin Kramere0513cb2012-01-30 16:17:39 +00002669 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002670 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002671 unsigned LastDeducibleArgument;
2672 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2673 --LastDeducibleArgument) {
2674 if (!Deduced[LastDeducibleArgument - 1]) {
2675 // C++0x: Figure out if the template argument has a default. If so,
2676 // the user doesn't need to type this argument.
2677 // FIXME: We need to abstract template parameters better!
2678 bool HasDefaultArg = false;
2679 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002680 LastDeducibleArgument - 1);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002681 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2682 HasDefaultArg = TTP->hasDefaultArgument();
2683 else if (NonTypeTemplateParmDecl *NTTP
2684 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2685 HasDefaultArg = NTTP->hasDefaultArgument();
2686 else {
2687 assert(isa<TemplateTemplateParmDecl>(Param));
2688 HasDefaultArg
Douglas Gregor9eb77012009-11-07 00:00:49 +00002689 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002690 }
2691
2692 if (!HasDefaultArg)
2693 break;
2694 }
2695 }
2696
2697 if (LastDeducibleArgument) {
2698 // Some of the function template arguments cannot be deduced from a
2699 // function call, so we introduce an explicit template argument list
2700 // containing all of the arguments up to the first deducible argument.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002701 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002702 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor3545ff42009-09-21 16:56:56 +00002703 LastDeducibleArgument);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002704 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002705 }
2706
2707 // Add the function parameters
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002708 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002709 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002710 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002711 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002712 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002713 }
2714
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002715 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002716 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002717 Ctx, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002718 Result.AddTypedTextChunk(
2719 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002720 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002721 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002722 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002723 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002724 }
2725
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002726 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregord3c5d792009-11-17 16:44:22 +00002727 Selector Sel = Method->getSelector();
2728 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002729 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002730 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002731 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002732 }
2733
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002734 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregor1b605f72009-11-19 01:08:35 +00002735 SelName += ':';
2736 if (StartParameter == 0)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002737 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002738 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002739 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002740
2741 // If there is only one parameter, and we're past it, add an empty
2742 // typed-text chunk since there is nothing to type.
2743 if (Method->param_size() == 1)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002744 Result.AddTypedTextChunk("");
Douglas Gregor1b605f72009-11-19 01:08:35 +00002745 }
Douglas Gregord3c5d792009-11-17 16:44:22 +00002746 unsigned Idx = 0;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002747 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2748 PEnd = Method->param_end();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002749 P != PEnd; (void)++P, ++Idx) {
2750 if (Idx > 0) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00002751 std::string Keyword;
2752 if (Idx > StartParameter)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002753 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregord3c5d792009-11-17 16:44:22 +00002754 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramer632500c2011-07-26 16:59:25 +00002755 Keyword += II->getName();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002756 Keyword += ":";
Douglas Gregor95887f92010-07-08 23:20:03 +00002757 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002758 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002759 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002760 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002761 }
Douglas Gregor1b605f72009-11-19 01:08:35 +00002762
2763 // If we're before the starting parameter, skip the placeholder.
2764 if (Idx < StartParameter)
2765 continue;
Douglas Gregord3c5d792009-11-17 16:44:22 +00002766
2767 std::string Arg;
Douglas Gregore90dd002010-08-24 16:15:59 +00002768
2769 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002770 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregore90dd002010-08-24 16:15:59 +00002771 else {
John McCall31168b02011-06-15 23:02:42 +00002772 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor8f08d742011-07-30 07:55:26 +00002773 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2774 + Arg + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002775 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregor981a0c42010-08-29 19:47:46 +00002776 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramer632500c2011-07-26 16:59:25 +00002777 Arg += II->getName();
Douglas Gregore90dd002010-08-24 16:15:59 +00002778 }
2779
Douglas Gregor400f5972010-08-31 05:13:43 +00002780 if (Method->isVariadic() && (P + 1) == PEnd)
2781 Arg += ", ...";
2782
Douglas Gregor95887f92010-07-08 23:20:03 +00002783 if (DeclaringEntity)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002784 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor95887f92010-07-08 23:20:03 +00002785 else if (AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002786 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8537c52009-11-19 07:41:15 +00002787 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002788 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002789 }
2790
Douglas Gregor04c5f972009-12-23 00:21:46 +00002791 if (Method->isVariadic()) {
Douglas Gregor400f5972010-08-31 05:13:43 +00002792 if (Method->param_size() == 0) {
2793 if (DeclaringEntity)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002794 Result.AddTextChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002795 else if (AllParametersAreInformative)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002796 Result.AddInformativeChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002797 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002798 Result.AddPlaceholderChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002799 }
Douglas Gregordbb71db2010-08-23 23:51:41 +00002800
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002801 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor04c5f972009-12-23 00:21:46 +00002802 }
2803
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002804 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002805 }
2806
Douglas Gregorf09935f2009-12-01 05:55:20 +00002807 if (Qualifier)
Douglas Gregor5bf52692009-09-22 23:15:58 +00002808 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002809 Ctx, Policy);
Douglas Gregorf09935f2009-12-01 05:55:20 +00002810
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002811 Result.AddTypedTextChunk(
2812 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002813 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002814}
2815
Douglas Gregorf0f51982009-09-23 00:34:09 +00002816CodeCompletionString *
2817CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2818 unsigned CurrentArg,
Douglas Gregor36e3b5c2010-10-11 21:37:58 +00002819 Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002820 CodeCompletionAllocator &Allocator,
2821 CodeCompletionTUInfo &CCTUInfo) const {
Douglas Gregor75acd922011-09-27 23:30:47 +00002822 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCall31168b02011-06-15 23:02:42 +00002823
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002824 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002825 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002826 FunctionDecl *FDecl = getFunction();
Douglas Gregor75acd922011-09-27 23:30:47 +00002827 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002828 const FunctionProtoType *Proto
2829 = dyn_cast<FunctionProtoType>(getFunctionType());
2830 if (!FDecl && !Proto) {
2831 // Function without a prototype. Just give the return type and a
2832 // highlighted ellipsis.
2833 const FunctionType *FT = getFunctionType();
Alp Toker314cc812014-01-25 16:55:45 +00002834 Result.AddTextChunk(GetCompletionTypeString(FT->getReturnType(), S.Context,
2835 Policy, Result.getAllocator()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002836 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2837 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2838 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002839 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002840 }
2841
2842 if (FDecl)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002843 Result.AddTextChunk(
2844 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002845 else
Alp Toker314cc812014-01-25 16:55:45 +00002846 Result.AddTextChunk(Result.getAllocator().CopyString(
2847 Proto->getReturnType().getAsString(Policy)));
2848
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002849 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Alp Toker9cacbab2014-01-20 20:26:09 +00002850 unsigned NumParams = FDecl ? FDecl->getNumParams() : Proto->getNumParams();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002851 for (unsigned I = 0; I != NumParams; ++I) {
2852 if (I)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002853 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002854
2855 std::string ArgString;
2856 QualType ArgType;
2857
2858 if (FDecl) {
2859 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2860 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2861 } else {
Alp Toker9cacbab2014-01-20 20:26:09 +00002862 ArgType = Proto->getParamType(I);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002863 }
2864
John McCall31168b02011-06-15 23:02:42 +00002865 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002866
2867 if (I == CurrentArg)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002868 Result.AddChunk(CodeCompletionString::CK_CurrentParameter,
2869 Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002870 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002871 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002872 }
2873
2874 if (Proto && Proto->isVariadic()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002875 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002876 if (CurrentArg < NumParams)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002877 Result.AddTextChunk("...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002878 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002879 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002880 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002881 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002882
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002883 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002884}
2885
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002886unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002887 const LangOptions &LangOpts,
Douglas Gregor6e240332010-08-16 16:18:59 +00002888 bool PreferredTypeIsPointer) {
2889 unsigned Priority = CCP_Macro;
2890
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002891 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2892 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2893 MacroName.equals("Nil")) {
Douglas Gregor6e240332010-08-16 16:18:59 +00002894 Priority = CCP_Constant;
2895 if (PreferredTypeIsPointer)
2896 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002897 }
2898 // Treat "YES", "NO", "true", and "false" as constants.
2899 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2900 MacroName.equals("true") || MacroName.equals("false"))
2901 Priority = CCP_Constant;
2902 // Treat "bool" as a type.
2903 else if (MacroName.equals("bool"))
2904 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2905
Douglas Gregor6e240332010-08-16 16:18:59 +00002906
2907 return Priority;
2908}
2909
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002910CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002911 if (!D)
2912 return CXCursor_UnexposedDecl;
2913
2914 switch (D->getKind()) {
2915 case Decl::Enum: return CXCursor_EnumDecl;
2916 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2917 case Decl::Field: return CXCursor_FieldDecl;
2918 case Decl::Function:
2919 return CXCursor_FunctionDecl;
2920 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2921 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002922 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregordeafd0b2011-12-27 22:43:10 +00002923
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002924 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002925 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2926 case Decl::ObjCMethod:
2927 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2928 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2929 case Decl::CXXMethod: return CXCursor_CXXMethod;
2930 case Decl::CXXConstructor: return CXCursor_Constructor;
2931 case Decl::CXXDestructor: return CXCursor_Destructor;
2932 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2933 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002934 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002935 case Decl::ParmVar: return CXCursor_ParmDecl;
2936 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smithdda56e42011-04-15 14:24:37 +00002937 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002938 case Decl::Var: return CXCursor_VarDecl;
2939 case Decl::Namespace: return CXCursor_Namespace;
2940 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2941 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2942 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2943 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2944 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2945 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis12afd702011-09-30 17:58:23 +00002946 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002947 case Decl::ClassTemplatePartialSpecialization:
2948 return CXCursor_ClassTemplatePartialSpecialization;
2949 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor3e653b32012-04-30 23:41:16 +00002950 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002951
2952 case Decl::Using:
2953 case Decl::UnresolvedUsingValue:
2954 case Decl::UnresolvedUsingTypename:
2955 return CXCursor_UsingDeclaration;
2956
Douglas Gregor4cd65962011-06-03 23:08:58 +00002957 case Decl::ObjCPropertyImpl:
2958 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2959 case ObjCPropertyImplDecl::Dynamic:
2960 return CXCursor_ObjCDynamicDecl;
2961
2962 case ObjCPropertyImplDecl::Synthesize:
2963 return CXCursor_ObjCSynthesizeDecl;
2964 }
Argyrios Kyrtzidis50e5b1d2012-10-05 00:22:24 +00002965
2966 case Decl::Import:
2967 return CXCursor_ModuleImportDecl;
Douglas Gregor4cd65962011-06-03 23:08:58 +00002968
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002969 default:
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002970 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002971 switch (TD->getTagKind()) {
Joao Matosdc86f942012-08-31 18:45:21 +00002972 case TTK_Interface: // fall through
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002973 case TTK_Struct: return CXCursor_StructDecl;
2974 case TTK_Class: return CXCursor_ClassDecl;
2975 case TTK_Union: return CXCursor_UnionDecl;
2976 case TTK_Enum: return CXCursor_EnumDecl;
2977 }
2978 }
2979 }
2980
2981 return CXCursor_UnexposedDecl;
2982}
2983
Douglas Gregor55b037b2010-07-08 20:55:51 +00002984static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor8cb17462012-10-09 16:01:50 +00002985 bool IncludeUndefined,
Douglas Gregor55b037b2010-07-08 20:55:51 +00002986 bool TargetTypeIsPointer = false) {
John McCall276321a2010-08-25 06:19:51 +00002987 typedef CodeCompletionResult Result;
Douglas Gregor55b037b2010-07-08 20:55:51 +00002988
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002989 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002990
Douglas Gregor9eb77012009-11-07 00:00:49 +00002991 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2992 MEnd = PP.macro_end();
Douglas Gregor55b037b2010-07-08 20:55:51 +00002993 M != MEnd; ++M) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00002994 if (IncludeUndefined || M->first->hasMacroDefinition())
2995 Results.AddResult(Result(M->first,
Douglas Gregor6e240332010-08-16 16:18:59 +00002996 getMacroUsagePriority(M->first->getName(),
David Blaikiebbafb8a2012-03-11 07:00:24 +00002997 PP.getLangOpts(),
Douglas Gregor6e240332010-08-16 16:18:59 +00002998 TargetTypeIsPointer)));
Douglas Gregor55b037b2010-07-08 20:55:51 +00002999 }
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003000
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003001 Results.ExitScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003002
Douglas Gregorf329c7c2009-10-30 16:50:04 +00003003}
3004
Douglas Gregorce0e8562010-08-23 21:54:33 +00003005static void AddPrettyFunctionResults(const LangOptions &LangOpts,
3006 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003007 typedef CodeCompletionResult Result;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003008
3009 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003010
Douglas Gregorce0e8562010-08-23 21:54:33 +00003011 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
3012 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith2bf7fdb2013-01-02 11:42:31 +00003013 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregorce0e8562010-08-23 21:54:33 +00003014 Results.AddResult(Result("__func__", CCP_Constant));
3015 Results.ExitScope();
3016}
3017
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003018static void HandleCodeCompleteResults(Sema *S,
3019 CodeCompleteConsumer *CodeCompleter,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003020 CodeCompletionContext Context,
John McCall276321a2010-08-25 06:19:51 +00003021 CodeCompletionResult *Results,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003022 unsigned NumResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00003023 if (CodeCompleter)
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003024 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor3545ff42009-09-21 16:56:56 +00003025}
3026
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003027static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3028 Sema::ParserCompletionContext PCC) {
3029 switch (PCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00003030 case Sema::PCC_Namespace:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003031 return CodeCompletionContext::CCC_TopLevel;
3032
John McCallfaf5fb42010-08-26 23:41:50 +00003033 case Sema::PCC_Class:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003034 return CodeCompletionContext::CCC_ClassStructUnion;
3035
John McCallfaf5fb42010-08-26 23:41:50 +00003036 case Sema::PCC_ObjCInterface:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003037 return CodeCompletionContext::CCC_ObjCInterface;
3038
John McCallfaf5fb42010-08-26 23:41:50 +00003039 case Sema::PCC_ObjCImplementation:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003040 return CodeCompletionContext::CCC_ObjCImplementation;
3041
John McCallfaf5fb42010-08-26 23:41:50 +00003042 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003043 return CodeCompletionContext::CCC_ObjCIvarList;
3044
John McCallfaf5fb42010-08-26 23:41:50 +00003045 case Sema::PCC_Template:
3046 case Sema::PCC_MemberTemplate:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003047 if (S.CurContext->isFileContext())
3048 return CodeCompletionContext::CCC_TopLevel;
David Blaikie8a40f702012-01-17 06:56:22 +00003049 if (S.CurContext->isRecord())
Douglas Gregor0ac41382010-09-23 23:01:17 +00003050 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie8a40f702012-01-17 06:56:22 +00003051 return CodeCompletionContext::CCC_Other;
Douglas Gregor0ac41382010-09-23 23:01:17 +00003052
John McCallfaf5fb42010-08-26 23:41:50 +00003053 case Sema::PCC_RecoveryInFunction:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003054 return CodeCompletionContext::CCC_Recovery;
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003055
John McCallfaf5fb42010-08-26 23:41:50 +00003056 case Sema::PCC_ForInit:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003057 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3058 S.getLangOpts().ObjC1)
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003059 return CodeCompletionContext::CCC_ParenthesizedExpression;
3060 else
3061 return CodeCompletionContext::CCC_Expression;
3062
3063 case Sema::PCC_Expression:
John McCallfaf5fb42010-08-26 23:41:50 +00003064 case Sema::PCC_Condition:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003065 return CodeCompletionContext::CCC_Expression;
3066
John McCallfaf5fb42010-08-26 23:41:50 +00003067 case Sema::PCC_Statement:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003068 return CodeCompletionContext::CCC_Statement;
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003069
John McCallfaf5fb42010-08-26 23:41:50 +00003070 case Sema::PCC_Type:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003071 return CodeCompletionContext::CCC_Type;
Douglas Gregor5e35d592010-09-14 23:59:36 +00003072
3073 case Sema::PCC_ParenthesizedExpression:
3074 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor80039242011-02-15 20:33:25 +00003075
3076 case Sema::PCC_LocalDeclarationSpecifiers:
3077 return CodeCompletionContext::CCC_Type;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003078 }
David Blaikie8a40f702012-01-17 06:56:22 +00003079
3080 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003081}
3082
Douglas Gregorac322ec2010-08-27 21:18:54 +00003083/// \brief If we're in a C++ virtual member function, add completion results
3084/// that invoke the functions we override, since it's common to invoke the
3085/// overridden function as well as adding new functionality.
3086///
3087/// \param S The semantic analysis object for which we are generating results.
3088///
3089/// \param InContext This context in which the nested-name-specifier preceding
3090/// the code-completion point
3091static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3092 ResultBuilder &Results) {
3093 // Look through blocks.
3094 DeclContext *CurContext = S.CurContext;
3095 while (isa<BlockDecl>(CurContext))
3096 CurContext = CurContext->getParent();
3097
3098
3099 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3100 if (!Method || !Method->isVirtual())
3101 return;
3102
3103 // We need to have names for all of the parameters, if we're going to
3104 // generate a forwarding call.
Aaron Ballman43b68be2014-03-07 17:50:17 +00003105 for (auto P : Method->params())
3106 if (!P->getDeclName())
Douglas Gregorac322ec2010-08-27 21:18:54 +00003107 return;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003108
Douglas Gregor75acd922011-09-27 23:30:47 +00003109 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003110 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3111 MEnd = Method->end_overridden_methods();
3112 M != MEnd; ++M) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003113 CodeCompletionBuilder Builder(Results.getAllocator(),
3114 Results.getCodeCompletionTUInfo());
Dmitri Gribenko6cfb1532013-02-14 13:53:30 +00003115 const CXXMethodDecl *Overridden = *M;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003116 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3117 continue;
3118
3119 // If we need a nested-name-specifier, add one now.
3120 if (!InContext) {
3121 NestedNameSpecifier *NNS
3122 = getRequiredQualification(S.Context, CurContext,
3123 Overridden->getDeclContext());
3124 if (NNS) {
3125 std::string Str;
3126 llvm::raw_string_ostream OS(Str);
Douglas Gregor75acd922011-09-27 23:30:47 +00003127 NNS->print(OS, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003128 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003129 }
3130 } else if (!InContext->Equals(Overridden->getDeclContext()))
3131 continue;
3132
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003133 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003134 Overridden->getNameAsString()));
3135 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003136 bool FirstParam = true;
Aaron Ballman43b68be2014-03-07 17:50:17 +00003137 for (auto P : Method->params()) {
Douglas Gregorac322ec2010-08-27 21:18:54 +00003138 if (FirstParam)
3139 FirstParam = false;
3140 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003141 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003142
Aaron Ballman43b68be2014-03-07 17:50:17 +00003143 Builder.AddPlaceholderChunk(
3144 Results.getAllocator().CopyString(P->getIdentifier()->getName()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003145 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003146 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3147 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorac322ec2010-08-27 21:18:54 +00003148 CCP_SuperCompletion,
Douglas Gregor78254c82012-03-27 23:34:16 +00003149 CXCursor_CXXMethod,
3150 CXAvailability_Available,
3151 Overridden));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003152 Results.Ignore(Overridden);
3153 }
3154}
3155
Douglas Gregor07f43572012-01-29 18:15:03 +00003156void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3157 ModuleIdPath Path) {
3158 typedef CodeCompletionResult Result;
3159 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003160 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor07f43572012-01-29 18:15:03 +00003161 CodeCompletionContext::CCC_Other);
3162 Results.EnterNewScope();
3163
3164 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003165 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor07f43572012-01-29 18:15:03 +00003166 typedef CodeCompletionResult Result;
3167 if (Path.empty()) {
3168 // Enumerate all top-level modules.
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003169 SmallVector<Module *, 8> Modules;
Douglas Gregor07f43572012-01-29 18:15:03 +00003170 PP.getHeaderSearchInfo().collectAllModules(Modules);
3171 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3172 Builder.AddTypedTextChunk(
3173 Builder.getAllocator().CopyString(Modules[I]->Name));
3174 Results.AddResult(Result(Builder.TakeString(),
3175 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003176 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003177 Modules[I]->isAvailable()
3178 ? CXAvailability_Available
3179 : CXAvailability_NotAvailable));
3180 }
Daniel Jasper07e6c402013-08-05 20:26:17 +00003181 } else if (getLangOpts().Modules) {
Douglas Gregor07f43572012-01-29 18:15:03 +00003182 // Load the named module.
3183 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3184 Module::AllVisible,
3185 /*IsInclusionDirective=*/false);
3186 // Enumerate submodules.
3187 if (Mod) {
3188 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3189 SubEnd = Mod->submodule_end();
3190 Sub != SubEnd; ++Sub) {
3191
3192 Builder.AddTypedTextChunk(
3193 Builder.getAllocator().CopyString((*Sub)->Name));
3194 Results.AddResult(Result(Builder.TakeString(),
3195 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003196 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003197 (*Sub)->isAvailable()
3198 ? CXAvailability_Available
3199 : CXAvailability_NotAvailable));
3200 }
3201 }
3202 }
3203 Results.ExitScope();
3204 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3205 Results.data(),Results.size());
3206}
3207
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003208void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003209 ParserCompletionContext CompletionContext) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003210 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003211 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003212 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003213 Results.EnterNewScope();
Douglas Gregor50832e02010-09-20 22:39:41 +00003214
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003215 // Determine how to filter results, e.g., so that the names of
3216 // values (functions, enumerators, function templates, etc.) are
3217 // only allowed where we can have an expression.
3218 switch (CompletionContext) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003219 case PCC_Namespace:
3220 case PCC_Class:
3221 case PCC_ObjCInterface:
3222 case PCC_ObjCImplementation:
3223 case PCC_ObjCInstanceVariableList:
3224 case PCC_Template:
3225 case PCC_MemberTemplate:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003226 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003227 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003228 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3229 break;
3230
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003231 case PCC_Statement:
Douglas Gregor5e35d592010-09-14 23:59:36 +00003232 case PCC_ParenthesizedExpression:
Douglas Gregor4d755e82010-08-24 23:58:17 +00003233 case PCC_Expression:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003234 case PCC_ForInit:
3235 case PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003236 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor70febae2010-05-28 00:49:12 +00003237 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3238 else
3239 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003240
David Blaikiebbafb8a2012-03-11 07:00:24 +00003241 if (getLangOpts().CPlusPlus)
Douglas Gregorac322ec2010-08-27 21:18:54 +00003242 MaybeAddOverrideCalls(*this, /*InContext=*/0, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003243 break;
Douglas Gregor6da3db42010-05-25 05:58:43 +00003244
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003245 case PCC_RecoveryInFunction:
Douglas Gregor6da3db42010-05-25 05:58:43 +00003246 // Unfiltered
3247 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003248 }
3249
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003250 // If we are in a C++ non-static member function, check the qualifiers on
3251 // the member function to filter/prioritize the results list.
3252 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3253 if (CurMethod->isInstance())
3254 Results.setObjectTypeQualifiers(
3255 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3256
Douglas Gregorc580c522010-01-14 01:09:38 +00003257 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003258 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3259 CodeCompleter->includeGlobals());
Douglas Gregor92253692009-12-07 09:54:55 +00003260
Douglas Gregorf98e6a22010-01-13 23:51:12 +00003261 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor92253692009-12-07 09:54:55 +00003262 Results.ExitScope();
3263
Douglas Gregorce0e8562010-08-23 21:54:33 +00003264 switch (CompletionContext) {
Douglas Gregor5e35d592010-09-14 23:59:36 +00003265 case PCC_ParenthesizedExpression:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003266 case PCC_Expression:
3267 case PCC_Statement:
3268 case PCC_RecoveryInFunction:
3269 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00003270 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003271 break;
3272
3273 case PCC_Namespace:
3274 case PCC_Class:
3275 case PCC_ObjCInterface:
3276 case PCC_ObjCImplementation:
3277 case PCC_ObjCInstanceVariableList:
3278 case PCC_Template:
3279 case PCC_MemberTemplate:
3280 case PCC_ForInit:
3281 case PCC_Condition:
3282 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003283 case PCC_LocalDeclarationSpecifiers:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003284 break;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003285 }
3286
Douglas Gregor9eb77012009-11-07 00:00:49 +00003287 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003288 AddMacroResults(PP, Results, false);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003289
Douglas Gregor50832e02010-09-20 22:39:41 +00003290 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003291 Results.data(),Results.size());
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00003292}
3293
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003294static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3295 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003296 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00003297 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003298 bool IsSuper,
3299 ResultBuilder &Results);
3300
3301void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3302 bool AllowNonIdentifiers,
3303 bool AllowNestedNameSpecifiers) {
John McCall276321a2010-08-25 06:19:51 +00003304 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003305 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003306 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003307 AllowNestedNameSpecifiers
3308 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3309 : CodeCompletionContext::CCC_Name);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003310 Results.EnterNewScope();
3311
3312 // Type qualifiers can come after names.
3313 Results.AddResult(Result("const"));
3314 Results.AddResult(Result("volatile"));
David Blaikiebbafb8a2012-03-11 07:00:24 +00003315 if (getLangOpts().C99)
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003316 Results.AddResult(Result("restrict"));
3317
David Blaikiebbafb8a2012-03-11 07:00:24 +00003318 if (getLangOpts().CPlusPlus) {
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003319 if (AllowNonIdentifiers) {
3320 Results.AddResult(Result("operator"));
3321 }
3322
3323 // Add nested-name-specifiers.
3324 if (AllowNestedNameSpecifiers) {
3325 Results.allowNestedNameSpecifiers();
Douglas Gregor0ac41382010-09-23 23:01:17 +00003326 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003327 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3328 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3329 CodeCompleter->includeGlobals());
Douglas Gregor0ac41382010-09-23 23:01:17 +00003330 Results.setFilter(0);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003331 }
3332 }
3333 Results.ExitScope();
3334
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003335 // If we're in a context where we might have an expression (rather than a
3336 // declaration), and what we've seen so far is an Objective-C type that could
3337 // be a receiver of a class message, this may be a class message send with
3338 // the initial opening bracket '[' missing. Add appropriate completions.
3339 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003340 DS.getParsedSpecifiers() == DeclSpec::PQ_TypeSpecifier &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003341 DS.getTypeSpecType() == DeclSpec::TST_typename &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003342 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3343 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003344 !DS.isTypeAltiVecVector() &&
3345 S &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003346 (S->getFlags() & Scope::DeclScope) != 0 &&
3347 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3348 Scope::FunctionPrototypeScope |
3349 Scope::AtCatchScope)) == 0) {
3350 ParsedType T = DS.getRepAsType();
3351 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003352 AddClassMessageCompletions(*this, S, T, None, false, false, Results);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003353 }
3354
Douglas Gregor56ccce02010-08-24 04:59:56 +00003355 // Note that we intentionally suppress macro results here, since we do not
3356 // encourage using macros to produce the names of entities.
3357
Douglas Gregor0ac41382010-09-23 23:01:17 +00003358 HandleCodeCompleteResults(this, CodeCompleter,
3359 Results.getCompletionContext(),
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003360 Results.data(), Results.size());
3361}
3362
Douglas Gregor68762e72010-08-23 21:17:50 +00003363struct Sema::CodeCompleteExpressionData {
3364 CodeCompleteExpressionData(QualType PreferredType = QualType())
3365 : PreferredType(PreferredType), IntegralConstantExpression(false),
3366 ObjCCollection(false) { }
3367
3368 QualType PreferredType;
3369 bool IntegralConstantExpression;
3370 bool ObjCCollection;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003371 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregor68762e72010-08-23 21:17:50 +00003372};
3373
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003374/// \brief Perform code-completion in an expression context when we know what
3375/// type we're looking for.
Douglas Gregor68762e72010-08-23 21:17:50 +00003376void Sema::CodeCompleteExpression(Scope *S,
3377 const CodeCompleteExpressionData &Data) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003378 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003379 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003380 CodeCompletionContext::CCC_Expression);
Douglas Gregor68762e72010-08-23 21:17:50 +00003381 if (Data.ObjCCollection)
3382 Results.setFilter(&ResultBuilder::IsObjCCollection);
3383 else if (Data.IntegralConstantExpression)
Douglas Gregor85b50632010-07-28 21:50:18 +00003384 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003385 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003386 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3387 else
3388 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregor68762e72010-08-23 21:17:50 +00003389
3390 if (!Data.PreferredType.isNull())
3391 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3392
3393 // Ignore any declarations that we were told that we don't care about.
3394 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3395 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003396
3397 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003398 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3399 CodeCompleter->includeGlobals());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003400
3401 Results.EnterNewScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003402 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003403 Results.ExitScope();
3404
Douglas Gregor55b037b2010-07-08 20:55:51 +00003405 bool PreferredTypeIsPointer = false;
Douglas Gregor68762e72010-08-23 21:17:50 +00003406 if (!Data.PreferredType.isNull())
3407 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3408 || Data.PreferredType->isMemberPointerType()
3409 || Data.PreferredType->isBlockPointerType();
Douglas Gregor55b037b2010-07-08 20:55:51 +00003410
Douglas Gregorce0e8562010-08-23 21:54:33 +00003411 if (S->getFnParent() &&
3412 !Data.ObjCCollection &&
3413 !Data.IntegralConstantExpression)
David Blaikiebbafb8a2012-03-11 07:00:24 +00003414 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003415
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003416 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003417 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003418 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor68762e72010-08-23 21:17:50 +00003419 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3420 Data.PreferredType),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003421 Results.data(),Results.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003422}
3423
Douglas Gregoreda7e542010-09-18 01:28:11 +00003424void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3425 if (E.isInvalid())
3426 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003427 else if (getLangOpts().ObjC1)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003428 CodeCompleteObjCInstanceMessage(S, E.take(), None, false);
Douglas Gregored0b69d2010-09-15 16:23:04 +00003429}
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003430
Douglas Gregorb888acf2010-12-09 23:01:55 +00003431/// \brief The set of properties that have already been added, referenced by
3432/// property name.
3433typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3434
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003435/// \brief Retrieve the container definition, if any?
3436static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3437 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3438 if (Interface->hasDefinition())
3439 return Interface->getDefinition();
3440
3441 return Interface;
3442 }
3443
3444 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3445 if (Protocol->hasDefinition())
3446 return Protocol->getDefinition();
3447
3448 return Protocol;
3449 }
3450 return Container;
3451}
3452
3453static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor5d649882009-11-18 22:32:06 +00003454 bool AllowCategories,
Douglas Gregor95147142011-05-05 15:50:42 +00003455 bool AllowNullaryMethods,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003456 DeclContext *CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003457 AddedPropertiesSet &AddedProperties,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003458 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003459 typedef CodeCompletionResult Result;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003460
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003461 // Retrieve the definition.
3462 Container = getContainerDef(Container);
3463
Douglas Gregor9291bad2009-11-18 01:29:26 +00003464 // Add properties in this container.
3465 for (ObjCContainerDecl::prop_iterator P = Container->prop_begin(),
3466 PEnd = Container->prop_end();
3467 P != PEnd;
Douglas Gregorb888acf2010-12-09 23:01:55 +00003468 ++P) {
3469 if (AddedProperties.insert(P->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003470 Results.MaybeAddResult(Result(*P, Results.getBasePriority(*P), 0),
3471 CurContext);
Douglas Gregorb888acf2010-12-09 23:01:55 +00003472 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003473
Douglas Gregor95147142011-05-05 15:50:42 +00003474 // Add nullary methods
3475 if (AllowNullaryMethods) {
3476 ASTContext &Context = Container->getASTContext();
Douglas Gregor75acd922011-09-27 23:30:47 +00003477 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Douglas Gregor95147142011-05-05 15:50:42 +00003478 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
3479 MEnd = Container->meth_end();
3480 M != MEnd; ++M) {
3481 if (M->getSelector().isUnarySelector())
3482 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3483 if (AddedProperties.insert(Name)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003484 CodeCompletionBuilder Builder(Results.getAllocator(),
3485 Results.getCodeCompletionTUInfo());
David Blaikie40ed2972012-06-06 20:45:41 +00003486 AddResultTypeChunk(Context, Policy, *M, Builder);
Douglas Gregor95147142011-05-05 15:50:42 +00003487 Builder.AddTypedTextChunk(
3488 Results.getAllocator().CopyString(Name->getName()));
3489
David Blaikie40ed2972012-06-06 20:45:41 +00003490 Results.MaybeAddResult(Result(Builder.TakeString(), *M,
Douglas Gregor78254c82012-03-27 23:34:16 +00003491 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor95147142011-05-05 15:50:42 +00003492 CurContext);
3493 }
3494 }
3495 }
3496
3497
Douglas Gregor9291bad2009-11-18 01:29:26 +00003498 // Add properties in referenced protocols.
3499 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3500 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
3501 PEnd = Protocol->protocol_end();
3502 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003503 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3504 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003505 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor5d649882009-11-18 22:32:06 +00003506 if (AllowCategories) {
3507 // Look through categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00003508 for (ObjCInterfaceDecl::known_categories_iterator
3509 Cat = IFace->known_categories_begin(),
3510 CatEnd = IFace->known_categories_end();
3511 Cat != CatEnd; ++Cat)
3512 AddObjCProperties(*Cat, AllowCategories, AllowNullaryMethods,
Douglas Gregor95147142011-05-05 15:50:42 +00003513 CurContext, AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00003514 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003515
3516 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003517 for (ObjCInterfaceDecl::all_protocol_iterator
3518 I = IFace->all_referenced_protocol_begin(),
3519 E = IFace->all_referenced_protocol_end(); I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003520 AddObjCProperties(*I, AllowCategories, AllowNullaryMethods, CurContext,
3521 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003522
3523 // Look in the superclass.
3524 if (IFace->getSuperClass())
Douglas Gregor95147142011-05-05 15:50:42 +00003525 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3526 AllowNullaryMethods, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003527 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003528 } else if (const ObjCCategoryDecl *Category
3529 = dyn_cast<ObjCCategoryDecl>(Container)) {
3530 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003531 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
3532 PEnd = Category->protocol_end();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003533 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003534 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3535 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003536 }
3537}
3538
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003539void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor2436e712009-09-17 21:32:03 +00003540 SourceLocation OpLoc,
3541 bool IsArrow) {
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003542 if (!Base || !CodeCompleter)
Douglas Gregor2436e712009-09-17 21:32:03 +00003543 return;
3544
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003545 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3546 if (ConvertedBase.isInvalid())
3547 return;
3548 Base = ConvertedBase.get();
3549
John McCall276321a2010-08-25 06:19:51 +00003550 typedef CodeCompletionResult Result;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003551
Douglas Gregor2436e712009-09-17 21:32:03 +00003552 QualType BaseType = Base->getType();
Douglas Gregor3545ff42009-09-21 16:56:56 +00003553
3554 if (IsArrow) {
3555 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3556 BaseType = Ptr->getPointeeType();
3557 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003558 /*Do nothing*/ ;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003559 else
3560 return;
3561 }
3562
Douglas Gregor21325842011-07-07 16:03:39 +00003563 enum CodeCompletionContext::Kind contextKind;
3564
3565 if (IsArrow) {
3566 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3567 }
3568 else {
3569 if (BaseType->isObjCObjectPointerType() ||
3570 BaseType->isObjCObjectOrInterfaceType()) {
3571 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3572 }
3573 else {
3574 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3575 }
3576 }
3577
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003578 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003579 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00003580 CodeCompletionContext(contextKind,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003581 BaseType),
3582 &ResultBuilder::IsMember);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003583 Results.EnterNewScope();
3584 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003585 // Indicate that we are performing a member access, and the cv-qualifiers
3586 // for the base object type.
3587 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3588
Douglas Gregor9291bad2009-11-18 01:29:26 +00003589 // Access to a C/C++ class, struct, or union.
Douglas Gregor6ae4c522010-01-14 03:21:49 +00003590 Results.allowNestedNameSpecifiers();
Douglas Gregor09bbc652010-01-14 15:47:35 +00003591 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003592 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3593 CodeCompleter->includeGlobals());
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003594
David Blaikiebbafb8a2012-03-11 07:00:24 +00003595 if (getLangOpts().CPlusPlus) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003596 if (!Results.empty()) {
3597 // The "template" keyword can follow "->" or "." in the grammar.
3598 // However, we only want to suggest the template keyword if something
3599 // is dependent.
3600 bool IsDependent = BaseType->isDependentType();
3601 if (!IsDependent) {
3602 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
Ted Kremenekc37877d2013-10-08 17:08:03 +00003603 if (DeclContext *Ctx = DepScope->getEntity()) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003604 IsDependent = Ctx->isDependentContext();
3605 break;
3606 }
3607 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003608
Douglas Gregor9291bad2009-11-18 01:29:26 +00003609 if (IsDependent)
Douglas Gregor78a21012010-01-14 16:01:26 +00003610 Results.AddResult(Result("template"));
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003611 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003612 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003613 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3614 // Objective-C property reference.
Douglas Gregorb888acf2010-12-09 23:01:55 +00003615 AddedPropertiesSet AddedProperties;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003616
3617 // Add property results based on our interface.
3618 const ObjCObjectPointerType *ObjCPtr
3619 = BaseType->getAsObjCInterfacePointerType();
3620 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor95147142011-05-05 15:50:42 +00003621 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3622 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003623 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003624
3625 // Add properties from the protocols in a qualified interface.
3626 for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(),
3627 E = ObjCPtr->qual_end();
3628 I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003629 AddObjCProperties(*I, true, /*AllowNullaryMethods=*/true, CurContext,
3630 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003631 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCall8b07ec22010-05-15 11:32:37 +00003632 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003633 // Objective-C instance variable access.
3634 ObjCInterfaceDecl *Class = 0;
3635 if (const ObjCObjectPointerType *ObjCPtr
3636 = BaseType->getAs<ObjCObjectPointerType>())
3637 Class = ObjCPtr->getInterfaceDecl();
3638 else
John McCall8b07ec22010-05-15 11:32:37 +00003639 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003640
3641 // Add all ivars from this class and its superclasses.
Douglas Gregor2b8162b2010-01-14 16:08:12 +00003642 if (Class) {
3643 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3644 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor39982192010-08-15 06:18:01 +00003645 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3646 CodeCompleter->includeGlobals());
Douglas Gregor9291bad2009-11-18 01:29:26 +00003647 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003648 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003649
3650 // FIXME: How do we cope with isa?
3651
3652 Results.ExitScope();
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003653
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003654 // Hand off the results found for code completion.
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003655 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003656 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003657 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00003658}
3659
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003660void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3661 if (!CodeCompleter)
3662 return;
3663
Douglas Gregor3545ff42009-09-21 16:56:56 +00003664 ResultBuilder::LookupFilter Filter = 0;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003665 enum CodeCompletionContext::Kind ContextKind
3666 = CodeCompletionContext::CCC_Other;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003667 switch ((DeclSpec::TST)TagSpec) {
3668 case DeclSpec::TST_enum:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003669 Filter = &ResultBuilder::IsEnum;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003670 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003671 break;
3672
3673 case DeclSpec::TST_union:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003674 Filter = &ResultBuilder::IsUnion;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003675 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003676 break;
3677
3678 case DeclSpec::TST_struct:
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003679 case DeclSpec::TST_class:
Joao Matosdc86f942012-08-31 18:45:21 +00003680 case DeclSpec::TST_interface:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003681 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003682 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003683 break;
3684
3685 default:
David Blaikie83d382b2011-09-23 05:06:16 +00003686 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003687 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003688
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003689 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3690 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00003691 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCalle87beb22010-04-23 18:46:30 +00003692
3693 // First pass: look for tags.
3694 Results.setFilter(Filter);
Douglas Gregor39982192010-08-15 06:18:01 +00003695 LookupVisibleDecls(S, LookupTagName, Consumer,
3696 CodeCompleter->includeGlobals());
John McCalle87beb22010-04-23 18:46:30 +00003697
Douglas Gregor39982192010-08-15 06:18:01 +00003698 if (CodeCompleter->includeGlobals()) {
3699 // Second pass: look for nested name specifiers.
3700 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3701 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3702 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003703
Douglas Gregor0ac41382010-09-23 23:01:17 +00003704 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003705 Results.data(),Results.size());
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003706}
3707
Douglas Gregor28c78432010-08-27 17:35:51 +00003708void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003709 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003710 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003711 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor28c78432010-08-27 17:35:51 +00003712 Results.EnterNewScope();
3713 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3714 Results.AddResult("const");
3715 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3716 Results.AddResult("volatile");
David Blaikiebbafb8a2012-03-11 07:00:24 +00003717 if (getLangOpts().C99 &&
Douglas Gregor28c78432010-08-27 17:35:51 +00003718 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3719 Results.AddResult("restrict");
Richard Smith8e1ac332013-03-28 01:55:44 +00003720 if (getLangOpts().C11 &&
3721 !(DS.getTypeQualifiers() & DeclSpec::TQ_atomic))
3722 Results.AddResult("_Atomic");
Douglas Gregor28c78432010-08-27 17:35:51 +00003723 Results.ExitScope();
3724 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003725 Results.getCompletionContext(),
Douglas Gregor28c78432010-08-27 17:35:51 +00003726 Results.data(), Results.size());
3727}
3728
Douglas Gregord328d572009-09-21 18:10:23 +00003729void Sema::CodeCompleteCase(Scope *S) {
John McCallaab3e412010-08-25 08:40:02 +00003730 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregord328d572009-09-21 18:10:23 +00003731 return;
John McCall5939b162011-08-06 07:30:58 +00003732
John McCallaab3e412010-08-25 08:40:02 +00003733 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCall5939b162011-08-06 07:30:58 +00003734 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3735 if (!type->isEnumeralType()) {
3736 CodeCompleteExpressionData Data(type);
Douglas Gregor68762e72010-08-23 21:17:50 +00003737 Data.IntegralConstantExpression = true;
3738 CodeCompleteExpression(S, Data);
Douglas Gregord328d572009-09-21 18:10:23 +00003739 return;
Douglas Gregor85b50632010-07-28 21:50:18 +00003740 }
Douglas Gregord328d572009-09-21 18:10:23 +00003741
3742 // Code-complete the cases of a switch statement over an enumeration type
3743 // by providing the list of
John McCall5939b162011-08-06 07:30:58 +00003744 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003745 if (EnumDecl *Def = Enum->getDefinition())
3746 Enum = Def;
Douglas Gregord328d572009-09-21 18:10:23 +00003747
3748 // Determine which enumerators we have already seen in the switch statement.
3749 // FIXME: Ideally, we would also be able to look *past* the code-completion
3750 // token, in case we are code-completing in the middle of the switch and not
3751 // at the end. However, we aren't able to do so at the moment.
3752 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Douglas Gregorf2510672009-09-21 19:57:38 +00003753 NestedNameSpecifier *Qualifier = 0;
Douglas Gregord328d572009-09-21 18:10:23 +00003754 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3755 SC = SC->getNextSwitchCase()) {
3756 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3757 if (!Case)
3758 continue;
3759
3760 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3761 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3762 if (EnumConstantDecl *Enumerator
3763 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3764 // We look into the AST of the case statement to determine which
3765 // enumerator was named. Alternatively, we could compute the value of
3766 // the integral constant expression, then compare it against the
3767 // values of each enumerator. However, value-based approach would not
3768 // work as well with C++ templates where enumerators declared within a
3769 // template are type- and value-dependent.
3770 EnumeratorsSeen.insert(Enumerator);
3771
Douglas Gregorf2510672009-09-21 19:57:38 +00003772 // If this is a qualified-id, keep track of the nested-name-specifier
3773 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregord328d572009-09-21 18:10:23 +00003774 //
3775 // switch (TagD.getKind()) {
3776 // case TagDecl::TK_enum:
3777 // break;
3778 // case XXX
3779 //
Douglas Gregorf2510672009-09-21 19:57:38 +00003780 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregord328d572009-09-21 18:10:23 +00003781 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3782 // TK_struct, and TK_class.
Douglas Gregor4bd90e52009-10-23 18:54:35 +00003783 Qualifier = DRE->getQualifier();
Douglas Gregord328d572009-09-21 18:10:23 +00003784 }
3785 }
3786
David Blaikiebbafb8a2012-03-11 07:00:24 +00003787 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorf2510672009-09-21 19:57:38 +00003788 // If there are no prior enumerators in C++, check whether we have to
3789 // qualify the names of the enumerators that we suggest, because they
3790 // may not be visible in this scope.
Douglas Gregord3cebdb2012-02-01 05:02:47 +00003791 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorf2510672009-09-21 19:57:38 +00003792 }
3793
Douglas Gregord328d572009-09-21 18:10:23 +00003794 // Add any enumerators that have not yet been mentioned.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003795 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003796 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003797 CodeCompletionContext::CCC_Expression);
Douglas Gregord328d572009-09-21 18:10:23 +00003798 Results.EnterNewScope();
Aaron Ballman23a6dcb2014-03-08 18:45:14 +00003799 for (auto *E : Enum->enumerators()) {
3800 if (EnumeratorsSeen.count(E))
Douglas Gregord328d572009-09-21 18:10:23 +00003801 continue;
3802
Aaron Ballman23a6dcb2014-03-08 18:45:14 +00003803 CodeCompletionResult R(E, CCP_EnumInCase, Qualifier);
Douglas Gregor3a69eaf2011-02-18 23:30:37 +00003804 Results.AddResult(R, CurContext, 0, false);
Douglas Gregord328d572009-09-21 18:10:23 +00003805 }
3806 Results.ExitScope();
Douglas Gregor285560922010-04-06 20:02:15 +00003807
Douglas Gregor21325842011-07-07 16:03:39 +00003808 //We need to make sure we're setting the right context,
3809 //so only say we include macros if the code completer says we do
3810 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3811 if (CodeCompleter->includeMacros()) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00003812 AddMacroResults(PP, Results, false);
Douglas Gregor21325842011-07-07 16:03:39 +00003813 kind = CodeCompletionContext::CCC_OtherWithMacros;
3814 }
3815
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003816 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00003817 kind,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003818 Results.data(),Results.size());
Douglas Gregord328d572009-09-21 18:10:23 +00003819}
3820
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003821static bool anyNullArguments(ArrayRef<Expr *> Args) {
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003822 if (Args.size() && !Args.data())
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003823 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003824
3825 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003826 if (!Args[I])
3827 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003828
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003829 return false;
3830}
3831
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003832void Sema::CodeCompleteCall(Scope *S, Expr *FnIn, ArrayRef<Expr *> Args) {
Douglas Gregorcabea402009-09-22 15:41:20 +00003833 if (!CodeCompleter)
3834 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003835
3836 // When we're code-completing for a call, we fall back to ordinary
3837 // name code-completion whenever we can't produce specific
3838 // results. We may want to revisit this strategy in the future,
3839 // e.g., by merging the two kinds of results.
3840
Douglas Gregorcabea402009-09-22 15:41:20 +00003841 Expr *Fn = (Expr *)FnIn;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003842
Douglas Gregorcabea402009-09-22 15:41:20 +00003843 // Ignore type-dependent call expressions entirely.
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003844 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3845 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003846 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregorcabea402009-09-22 15:41:20 +00003847 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003848 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003849
John McCall57500772009-12-16 12:17:52 +00003850 // Build an overload candidate set based on the functions we find.
John McCallbc077cf2010-02-08 23:07:23 +00003851 SourceLocation Loc = Fn->getExprLoc();
3852 OverloadCandidateSet CandidateSet(Loc);
John McCall57500772009-12-16 12:17:52 +00003853
Douglas Gregorcabea402009-09-22 15:41:20 +00003854 // FIXME: What if we're calling something that isn't a function declaration?
3855 // FIXME: What if we're calling a pseudo-destructor?
3856 // FIXME: What if we're calling a member function?
3857
Douglas Gregorff59f672010-01-21 15:46:19 +00003858 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003859 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorff59f672010-01-21 15:46:19 +00003860
John McCall57500772009-12-16 12:17:52 +00003861 Expr *NakedFn = Fn->IgnoreParenCasts();
3862 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003863 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
John McCall57500772009-12-16 12:17:52 +00003864 /*PartialOverloading=*/ true);
3865 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3866 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorff59f672010-01-21 15:46:19 +00003867 if (FDecl) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00003868 if (!getLangOpts().CPlusPlus ||
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003869 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorff59f672010-01-21 15:46:19 +00003870 Results.push_back(ResultCandidate(FDecl));
3871 else
John McCallb89836b2010-01-26 01:37:31 +00003872 // FIXME: access?
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003873 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), Args,
3874 CandidateSet, false, /*PartialOverloading*/true);
Douglas Gregorff59f672010-01-21 15:46:19 +00003875 }
John McCall57500772009-12-16 12:17:52 +00003876 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003877
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003878 QualType ParamType;
3879
Douglas Gregorff59f672010-01-21 15:46:19 +00003880 if (!CandidateSet.empty()) {
3881 // Sort the overload candidate set by placing the best overloads first.
Benjamin Kramerbbdd7642014-03-01 14:48:57 +00003882 std::stable_sort(
3883 CandidateSet.begin(), CandidateSet.end(),
3884 [&](const OverloadCandidate &X, const OverloadCandidate &Y) {
3885 return isBetterOverloadCandidate(*this, X, Y, Loc);
3886 });
3887
Douglas Gregorff59f672010-01-21 15:46:19 +00003888 // Add the remaining viable overload candidates as code-completion reslults.
3889 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3890 CandEnd = CandidateSet.end();
3891 Cand != CandEnd; ++Cand) {
3892 if (Cand->Viable)
3893 Results.push_back(ResultCandidate(Cand->Function));
3894 }
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003895
3896 // From the viable candidates, try to determine the type of this parameter.
3897 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3898 if (const FunctionType *FType = Results[I].getFunctionType())
3899 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
Alp Toker9cacbab2014-01-20 20:26:09 +00003900 if (Args.size() < Proto->getNumParams()) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003901 if (ParamType.isNull())
Alp Toker9cacbab2014-01-20 20:26:09 +00003902 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003903 else if (!Context.hasSameUnqualifiedType(
Alp Toker9cacbab2014-01-20 20:26:09 +00003904 ParamType.getNonReferenceType(),
3905 Proto->getParamType(Args.size())
3906 .getNonReferenceType())) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003907 ParamType = QualType();
3908 break;
3909 }
3910 }
3911 }
3912 } else {
3913 // Try to determine the parameter type from the type of the expression
3914 // being called.
3915 QualType FunctionType = Fn->getType();
3916 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3917 FunctionType = Ptr->getPointeeType();
3918 else if (const BlockPointerType *BlockPtr
3919 = FunctionType->getAs<BlockPointerType>())
3920 FunctionType = BlockPtr->getPointeeType();
3921 else if (const MemberPointerType *MemPtr
3922 = FunctionType->getAs<MemberPointerType>())
3923 FunctionType = MemPtr->getPointeeType();
3924
3925 if (const FunctionProtoType *Proto
3926 = FunctionType->getAs<FunctionProtoType>()) {
Alp Toker9cacbab2014-01-20 20:26:09 +00003927 if (Args.size() < Proto->getNumParams())
3928 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003929 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003930 }
Douglas Gregor3ef59522009-12-11 19:06:04 +00003931
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003932 if (ParamType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003933 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003934 else
3935 CodeCompleteExpression(S, ParamType);
3936
Douglas Gregorc01890e2010-04-06 20:19:47 +00003937 if (!Results.empty())
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003938 CodeCompleter->ProcessOverloadCandidates(*this, Args.size(), Results.data(),
Douglas Gregor3ef59522009-12-11 19:06:04 +00003939 Results.size());
Douglas Gregorcabea402009-09-22 15:41:20 +00003940}
3941
John McCall48871652010-08-21 09:40:31 +00003942void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3943 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003944 if (!VD) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003945 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003946 return;
3947 }
3948
3949 CodeCompleteExpression(S, VD->getType());
3950}
3951
3952void Sema::CodeCompleteReturn(Scope *S) {
3953 QualType ResultType;
3954 if (isa<BlockDecl>(CurContext)) {
3955 if (BlockScopeInfo *BSI = getCurBlock())
3956 ResultType = BSI->ReturnType;
3957 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00003958 ResultType = Function->getReturnType();
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003959 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00003960 ResultType = Method->getReturnType();
3961
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003962 if (ResultType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003963 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003964 else
3965 CodeCompleteExpression(S, ResultType);
3966}
3967
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003968void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003969 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003970 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003971 mapCodeCompletionContext(*this, PCC_Statement));
3972 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3973 Results.EnterNewScope();
3974
3975 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3976 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3977 CodeCompleter->includeGlobals());
3978
3979 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
3980
3981 // "else" block
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003982 CodeCompletionBuilder Builder(Results.getAllocator(),
3983 Results.getCodeCompletionTUInfo());
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003984 Builder.AddTypedTextChunk("else");
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00003985 if (Results.includeCodePatterns()) {
3986 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3987 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
3988 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3989 Builder.AddPlaceholderChunk("statements");
3990 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3991 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
3992 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003993 Results.AddResult(Builder.TakeString());
3994
3995 // "else if" block
3996 Builder.AddTypedTextChunk("else");
3997 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3998 Builder.AddTextChunk("if");
3999 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4000 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00004001 if (getLangOpts().CPlusPlus)
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004002 Builder.AddPlaceholderChunk("condition");
4003 else
4004 Builder.AddPlaceholderChunk("expression");
4005 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00004006 if (Results.includeCodePatterns()) {
4007 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4008 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4009 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4010 Builder.AddPlaceholderChunk("statements");
4011 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4012 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4013 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004014 Results.AddResult(Builder.TakeString());
4015
4016 Results.ExitScope();
4017
4018 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004019 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004020
4021 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00004022 AddMacroResults(PP, Results, false);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004023
4024 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4025 Results.data(),Results.size());
4026}
4027
Richard Trieu2bd04012011-09-09 02:00:50 +00004028void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004029 if (LHS)
4030 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4031 else
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004032 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004033}
4034
Jeffrey Yasskinc76498d2010-04-08 16:38:48 +00004035void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor2436e712009-09-17 21:32:03 +00004036 bool EnteringContext) {
4037 if (!SS.getScopeRep() || !CodeCompleter)
4038 return;
4039
Douglas Gregor3545ff42009-09-21 16:56:56 +00004040 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4041 if (!Ctx)
4042 return;
Douglas Gregor800f2f02009-12-11 18:28:39 +00004043
4044 // Try to instantiate any non-dependent declaration contexts before
4045 // we look in them.
John McCall0b66eb32010-05-01 00:40:08 +00004046 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregor800f2f02009-12-11 18:28:39 +00004047 return;
4048
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004049 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004050 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004051 CodeCompletionContext::CCC_Name);
Douglas Gregorac322ec2010-08-27 21:18:54 +00004052 Results.EnterNewScope();
Douglas Gregor0ac41382010-09-23 23:01:17 +00004053
Douglas Gregor3545ff42009-09-21 16:56:56 +00004054 // The "template" keyword can follow "::" in the grammar, but only
4055 // put it into the grammar if the nested-name-specifier is dependent.
Aaron Ballman4a979672014-01-03 13:56:08 +00004056 NestedNameSpecifier *NNS = SS.getScopeRep();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004057 if (!Results.empty() && NNS->isDependent())
Douglas Gregor78a21012010-01-14 16:01:26 +00004058 Results.AddResult("template");
Douglas Gregorac322ec2010-08-27 21:18:54 +00004059
4060 // Add calls to overridden virtual functions, if there are any.
4061 //
4062 // FIXME: This isn't wonderful, because we don't know whether we're actually
4063 // in a context that permits expressions. This is a general issue with
4064 // qualified-id completions.
4065 if (!EnteringContext)
4066 MaybeAddOverrideCalls(*this, Ctx, Results);
4067 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004068
Douglas Gregorac322ec2010-08-27 21:18:54 +00004069 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4070 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4071
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004072 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorc1679ec2011-07-25 17:48:11 +00004073 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004074 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00004075}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004076
4077void Sema::CodeCompleteUsing(Scope *S) {
4078 if (!CodeCompleter)
4079 return;
4080
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004081 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004082 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004083 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4084 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004085 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004086
4087 // If we aren't in class scope, we could see the "namespace" keyword.
4088 if (!S->isClassScope())
John McCall276321a2010-08-25 06:19:51 +00004089 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004090
4091 // After "using", we can see anything that would start a
4092 // nested-name-specifier.
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004093 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004094 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4095 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004096 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004097
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004098 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004099 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004100 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004101}
4102
4103void Sema::CodeCompleteUsingDirective(Scope *S) {
4104 if (!CodeCompleter)
4105 return;
4106
Douglas Gregor3545ff42009-09-21 16:56:56 +00004107 // After "using namespace", we expect to see a namespace name or namespace
4108 // alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004109 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004110 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004111 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004112 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004113 Results.EnterNewScope();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004114 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004115 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4116 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004117 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004118 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004119 CodeCompletionContext::CCC_Namespace,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004120 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004121}
4122
4123void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4124 if (!CodeCompleter)
4125 return;
4126
Ted Kremenekc37877d2013-10-08 17:08:03 +00004127 DeclContext *Ctx = S->getEntity();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004128 if (!S->getParent())
4129 Ctx = Context.getTranslationUnitDecl();
4130
Douglas Gregor0ac41382010-09-23 23:01:17 +00004131 bool SuppressedGlobalResults
4132 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4133
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004134 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004135 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004136 SuppressedGlobalResults
4137 ? CodeCompletionContext::CCC_Namespace
4138 : CodeCompletionContext::CCC_Other,
4139 &ResultBuilder::IsNamespace);
4140
4141 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00004142 // We only want to see those namespaces that have already been defined
4143 // within this scope, because its likely that the user is creating an
4144 // extended namespace declaration. Keep track of the most recent
4145 // definition of each namespace.
4146 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4147 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4148 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4149 NS != NSEnd; ++NS)
David Blaikie40ed2972012-06-06 20:45:41 +00004150 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004151
4152 // Add the most recent definition (or extended definition) of each
4153 // namespace to the list of results.
Douglas Gregor64b12b52009-09-22 23:31:26 +00004154 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004155 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregor78254c82012-03-27 23:34:16 +00004156 NS = OrigToLatest.begin(),
4157 NSEnd = OrigToLatest.end();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004158 NS != NSEnd; ++NS)
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004159 Results.AddResult(CodeCompletionResult(
4160 NS->second, Results.getBasePriority(NS->second), 0),
Douglas Gregorfc59ce12010-01-14 16:14:35 +00004161 CurContext, 0, false);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004162 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004163 }
4164
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004165 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004166 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004167 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004168}
4169
4170void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4171 if (!CodeCompleter)
4172 return;
4173
Douglas Gregor3545ff42009-09-21 16:56:56 +00004174 // After "namespace", we expect to see a namespace or alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004175 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004176 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004177 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004178 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004179 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004180 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4181 CodeCompleter->includeGlobals());
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004182 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004183 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004184 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004185}
4186
Douglas Gregorc811ede2009-09-18 20:05:18 +00004187void Sema::CodeCompleteOperatorName(Scope *S) {
4188 if (!CodeCompleter)
4189 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004190
John McCall276321a2010-08-25 06:19:51 +00004191 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004192 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004193 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004194 CodeCompletionContext::CCC_Type,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004195 &ResultBuilder::IsType);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004196 Results.EnterNewScope();
Douglas Gregorc811ede2009-09-18 20:05:18 +00004197
Douglas Gregor3545ff42009-09-21 16:56:56 +00004198 // Add the names of overloadable operators.
4199#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4200 if (std::strcmp(Spelling, "?")) \
Douglas Gregor78a21012010-01-14 16:01:26 +00004201 Results.AddResult(Result(Spelling));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004202#include "clang/Basic/OperatorKinds.def"
4203
4204 // Add any type names visible from the current scope
Douglas Gregor6ae4c522010-01-14 03:21:49 +00004205 Results.allowNestedNameSpecifiers();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004206 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004207 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4208 CodeCompleter->includeGlobals());
Douglas Gregor3545ff42009-09-21 16:56:56 +00004209
4210 // Add any type specifiers
David Blaikiebbafb8a2012-03-11 07:00:24 +00004211 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004212 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004213
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004214 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004215 CodeCompletionContext::CCC_Type,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004216 Results.data(),Results.size());
Douglas Gregorc811ede2009-09-18 20:05:18 +00004217}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004218
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004219void Sema::CodeCompleteConstructorInitializer(
4220 Decl *ConstructorD,
4221 ArrayRef <CXXCtorInitializer *> Initializers) {
Douglas Gregor75acd922011-09-27 23:30:47 +00004222 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004223 CXXConstructorDecl *Constructor
4224 = static_cast<CXXConstructorDecl *>(ConstructorD);
4225 if (!Constructor)
4226 return;
4227
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004228 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004229 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004230 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004231 Results.EnterNewScope();
4232
4233 // Fill in any already-initialized fields or base classes.
4234 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4235 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004236 for (unsigned I = 0, E = Initializers.size(); I != E; ++I) {
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004237 if (Initializers[I]->isBaseInitializer())
4238 InitializedBases.insert(
4239 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4240 else
Francois Pichetd583da02010-12-04 09:14:42 +00004241 InitializedFields.insert(cast<FieldDecl>(
4242 Initializers[I]->getAnyMember()));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004243 }
4244
4245 // Add completions for base classes.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004246 CodeCompletionBuilder Builder(Results.getAllocator(),
4247 Results.getCodeCompletionTUInfo());
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004248 bool SawLastInitializer = Initializers.empty();
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004249 CXXRecordDecl *ClassDecl = Constructor->getParent();
4250 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->bases_begin(),
4251 BaseEnd = ClassDecl->bases_end();
4252 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004253 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4254 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004255 = !Initializers.empty() &&
4256 Initializers.back()->isBaseInitializer() &&
Douglas Gregor99129ef2010-08-29 19:27:27 +00004257 Context.hasSameUnqualifiedType(Base->getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004258 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004259 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004260 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004261
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004262 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004263 Results.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004264 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004265 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4266 Builder.AddPlaceholderChunk("args");
4267 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4268 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004269 SawLastInitializer? CCP_NextInitializer
4270 : CCP_MemberDeclaration));
4271 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004272 }
4273
4274 // Add completions for virtual base classes.
4275 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->vbases_begin(),
4276 BaseEnd = ClassDecl->vbases_end();
4277 Base != BaseEnd; ++Base) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004278 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4279 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004280 = !Initializers.empty() &&
4281 Initializers.back()->isBaseInitializer() &&
Douglas Gregor99129ef2010-08-29 19:27:27 +00004282 Context.hasSameUnqualifiedType(Base->getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004283 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004284 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004285 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004286
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004287 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004288 Builder.getAllocator().CopyString(
John McCall31168b02011-06-15 23:02:42 +00004289 Base->getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004290 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4291 Builder.AddPlaceholderChunk("args");
4292 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4293 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004294 SawLastInitializer? CCP_NextInitializer
4295 : CCP_MemberDeclaration));
4296 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004297 }
4298
4299 // Add completions for members.
Aaron Ballmane8a8bae2014-03-08 20:12:42 +00004300 for (auto *Field : ClassDecl->fields()) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004301 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4302 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004303 = !Initializers.empty() &&
4304 Initializers.back()->isAnyMemberInitializer() &&
Aaron Ballmane8a8bae2014-03-08 20:12:42 +00004305 Initializers.back()->getAnyMember() == Field;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004306 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004307 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004308
4309 if (!Field->getDeclName())
4310 continue;
4311
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004312 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004313 Field->getIdentifier()->getName()));
4314 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4315 Builder.AddPlaceholderChunk("args");
4316 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4317 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004318 SawLastInitializer? CCP_NextInitializer
Douglas Gregorf3af3112010-09-09 21:42:20 +00004319 : CCP_MemberDeclaration,
Douglas Gregor78254c82012-03-27 23:34:16 +00004320 CXCursor_MemberRef,
4321 CXAvailability_Available,
Aaron Ballmane8a8bae2014-03-08 20:12:42 +00004322 Field));
Douglas Gregor99129ef2010-08-29 19:27:27 +00004323 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004324 }
4325 Results.ExitScope();
4326
Douglas Gregor0ac41382010-09-23 23:01:17 +00004327 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004328 Results.data(), Results.size());
4329}
4330
Douglas Gregord8c61782012-02-15 15:34:24 +00004331/// \brief Determine whether this scope denotes a namespace.
4332static bool isNamespaceScope(Scope *S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00004333 DeclContext *DC = S->getEntity();
Douglas Gregord8c61782012-02-15 15:34:24 +00004334 if (!DC)
4335 return false;
4336
4337 return DC->isFileContext();
4338}
4339
4340void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4341 bool AfterAmpersand) {
4342 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004343 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord8c61782012-02-15 15:34:24 +00004344 CodeCompletionContext::CCC_Other);
4345 Results.EnterNewScope();
4346
4347 // Note what has already been captured.
4348 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4349 bool IncludedThis = false;
4350 for (SmallVectorImpl<LambdaCapture>::iterator C = Intro.Captures.begin(),
4351 CEnd = Intro.Captures.end();
4352 C != CEnd; ++C) {
4353 if (C->Kind == LCK_This) {
4354 IncludedThis = true;
4355 continue;
4356 }
4357
4358 Known.insert(C->Id);
4359 }
4360
4361 // Look for other capturable variables.
4362 for (; S && !isNamespaceScope(S); S = S->getParent()) {
4363 for (Scope::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
4364 D != DEnd; ++D) {
4365 VarDecl *Var = dyn_cast<VarDecl>(*D);
4366 if (!Var ||
4367 !Var->hasLocalStorage() ||
4368 Var->hasAttr<BlocksAttr>())
4369 continue;
4370
4371 if (Known.insert(Var->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004372 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
4373 CurContext, 0, false);
Douglas Gregord8c61782012-02-15 15:34:24 +00004374 }
4375 }
4376
4377 // Add 'this', if it would be valid.
4378 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4379 addThisCompletion(*this, Results);
4380
4381 Results.ExitScope();
4382
4383 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4384 Results.data(), Results.size());
4385}
4386
James Dennett596e4752012-06-14 03:11:41 +00004387/// Macro that optionally prepends an "@" to the string literal passed in via
4388/// Keyword, depending on whether NeedAt is true or false.
4389#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4390
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004391static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004392 ResultBuilder &Results,
4393 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004394 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004395 // Since we have an implementation, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004396 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004397
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004398 CodeCompletionBuilder Builder(Results.getAllocator(),
4399 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004400 if (LangOpts.ObjC2) {
4401 // @dynamic
James Dennett596e4752012-06-14 03:11:41 +00004402 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004403 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4404 Builder.AddPlaceholderChunk("property");
4405 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004406
4407 // @synthesize
James Dennett596e4752012-06-14 03:11:41 +00004408 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004409 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4410 Builder.AddPlaceholderChunk("property");
4411 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004412 }
4413}
4414
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004415static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004416 ResultBuilder &Results,
4417 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004418 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004419
4420 // Since we have an interface or protocol, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004421 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004422
4423 if (LangOpts.ObjC2) {
4424 // @property
James Dennett596e4752012-06-14 03:11:41 +00004425 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004426
4427 // @required
James Dennett596e4752012-06-14 03:11:41 +00004428 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004429
4430 // @optional
James Dennett596e4752012-06-14 03:11:41 +00004431 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004432 }
4433}
4434
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004435static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004436 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004437 CodeCompletionBuilder Builder(Results.getAllocator(),
4438 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004439
4440 // @class name ;
James Dennett596e4752012-06-14 03:11:41 +00004441 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004442 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4443 Builder.AddPlaceholderChunk("name");
4444 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004445
Douglas Gregorf4c33342010-05-28 00:22:41 +00004446 if (Results.includeCodePatterns()) {
4447 // @interface name
4448 // FIXME: Could introduce the whole pattern, including superclasses and
4449 // such.
James Dennett596e4752012-06-14 03:11:41 +00004450 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004451 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4452 Builder.AddPlaceholderChunk("class");
4453 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004454
Douglas Gregorf4c33342010-05-28 00:22:41 +00004455 // @protocol name
James Dennett596e4752012-06-14 03:11:41 +00004456 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004457 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4458 Builder.AddPlaceholderChunk("protocol");
4459 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004460
4461 // @implementation name
James Dennett596e4752012-06-14 03:11:41 +00004462 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004463 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4464 Builder.AddPlaceholderChunk("class");
4465 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004466 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004467
4468 // @compatibility_alias name
James Dennett596e4752012-06-14 03:11:41 +00004469 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004470 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4471 Builder.AddPlaceholderChunk("alias");
4472 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4473 Builder.AddPlaceholderChunk("class");
4474 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor61e36812013-03-07 23:26:24 +00004475
4476 if (Results.getSema().getLangOpts().Modules) {
4477 // @import name
4478 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "import"));
4479 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4480 Builder.AddPlaceholderChunk("module");
4481 Results.AddResult(Result(Builder.TakeString()));
4482 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004483}
4484
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004485void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004486 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004487 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004488 CodeCompletionContext::CCC_Other);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004489 Results.EnterNewScope();
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004490 if (isa<ObjCImplDecl>(CurContext))
David Blaikiebbafb8a2012-03-11 07:00:24 +00004491 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004492 else if (CurContext->isObjCContainer())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004493 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorf1934162010-01-13 21:24:21 +00004494 else
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004495 AddObjCTopLevelResults(Results, false);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004496 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004497 HandleCodeCompleteResults(this, CodeCompleter,
4498 CodeCompletionContext::CCC_Other,
4499 Results.data(),Results.size());
Douglas Gregorf48706c2009-12-07 09:27:33 +00004500}
4501
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004502static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004503 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004504 CodeCompletionBuilder Builder(Results.getAllocator(),
4505 Results.getCodeCompletionTUInfo());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004506
4507 // @encode ( type-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004508 const char *EncodeType = "char[]";
David Blaikiebbafb8a2012-03-11 07:00:24 +00004509 if (Results.getSema().getLangOpts().CPlusPlus ||
4510 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose9da05852012-06-15 18:19:56 +00004511 EncodeType = "const char[]";
Douglas Gregore5c79d52011-10-18 21:20:17 +00004512 Builder.AddResultTypeChunk(EncodeType);
James Dennett596e4752012-06-14 03:11:41 +00004513 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004514 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4515 Builder.AddPlaceholderChunk("type-name");
4516 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4517 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004518
4519 // @protocol ( protocol-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004520 Builder.AddResultTypeChunk("Protocol *");
James Dennett596e4752012-06-14 03:11:41 +00004521 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004522 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4523 Builder.AddPlaceholderChunk("protocol-name");
4524 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4525 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004526
4527 // @selector ( selector )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004528 Builder.AddResultTypeChunk("SEL");
James Dennett596e4752012-06-14 03:11:41 +00004529 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004530 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4531 Builder.AddPlaceholderChunk("selector");
4532 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4533 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004534
4535 // @"string"
4536 Builder.AddResultTypeChunk("NSString *");
4537 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4538 Builder.AddPlaceholderChunk("string");
4539 Builder.AddTextChunk("\"");
4540 Results.AddResult(Result(Builder.TakeString()));
4541
Douglas Gregor951de302012-07-17 23:24:47 +00004542 // @[objects, ...]
Jordan Rose9da05852012-06-15 18:19:56 +00004543 Builder.AddResultTypeChunk("NSArray *");
James Dennett596e4752012-06-14 03:11:41 +00004544 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004545 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004546 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4547 Results.AddResult(Result(Builder.TakeString()));
4548
Douglas Gregor951de302012-07-17 23:24:47 +00004549 // @{key : object, ...}
Jordan Rose9da05852012-06-15 18:19:56 +00004550 Builder.AddResultTypeChunk("NSDictionary *");
James Dennett596e4752012-06-14 03:11:41 +00004551 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"{"));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004552 Builder.AddPlaceholderChunk("key");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004553 Builder.AddChunk(CodeCompletionString::CK_Colon);
4554 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4555 Builder.AddPlaceholderChunk("object, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004556 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4557 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004558
Douglas Gregor951de302012-07-17 23:24:47 +00004559 // @(expression)
Jordan Rose9da05852012-06-15 18:19:56 +00004560 Builder.AddResultTypeChunk("id");
4561 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose9da05852012-06-15 18:19:56 +00004562 Builder.AddPlaceholderChunk("expression");
Jordan Rose9da05852012-06-15 18:19:56 +00004563 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4564 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004565}
4566
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004567static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004568 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004569 CodeCompletionBuilder Builder(Results.getAllocator(),
4570 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004571
Douglas Gregorf4c33342010-05-28 00:22:41 +00004572 if (Results.includeCodePatterns()) {
4573 // @try { statements } @catch ( declaration ) { statements } @finally
4574 // { statements }
James Dennett596e4752012-06-14 03:11:41 +00004575 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004576 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4577 Builder.AddPlaceholderChunk("statements");
4578 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4579 Builder.AddTextChunk("@catch");
4580 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4581 Builder.AddPlaceholderChunk("parameter");
4582 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4583 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4584 Builder.AddPlaceholderChunk("statements");
4585 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4586 Builder.AddTextChunk("@finally");
4587 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4588 Builder.AddPlaceholderChunk("statements");
4589 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4590 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004591 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004592
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004593 // @throw
James Dennett596e4752012-06-14 03:11:41 +00004594 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004595 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4596 Builder.AddPlaceholderChunk("expression");
4597 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004598
Douglas Gregorf4c33342010-05-28 00:22:41 +00004599 if (Results.includeCodePatterns()) {
4600 // @synchronized ( expression ) { statements }
James Dennett596e4752012-06-14 03:11:41 +00004601 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004602 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4603 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4604 Builder.AddPlaceholderChunk("expression");
4605 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4606 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4607 Builder.AddPlaceholderChunk("statements");
4608 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4609 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004610 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004611}
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004612
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004613static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00004614 ResultBuilder &Results,
4615 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004616 typedef CodeCompletionResult Result;
James Dennett596e4752012-06-14 03:11:41 +00004617 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4618 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4619 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004620 if (LangOpts.ObjC2)
James Dennett596e4752012-06-14 03:11:41 +00004621 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004622}
4623
4624void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004625 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004626 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004627 CodeCompletionContext::CCC_Other);
Douglas Gregor48d46252010-01-13 21:54:15 +00004628 Results.EnterNewScope();
David Blaikiebbafb8a2012-03-11 07:00:24 +00004629 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregor48d46252010-01-13 21:54:15 +00004630 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004631 HandleCodeCompleteResults(this, CodeCompleter,
4632 CodeCompletionContext::CCC_Other,
4633 Results.data(),Results.size());
Douglas Gregor48d46252010-01-13 21:54:15 +00004634}
4635
4636void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004637 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004638 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004639 CodeCompletionContext::CCC_Other);
Douglas Gregorf1934162010-01-13 21:24:21 +00004640 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004641 AddObjCStatementResults(Results, false);
4642 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004643 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004644 HandleCodeCompleteResults(this, CodeCompleter,
4645 CodeCompletionContext::CCC_Other,
4646 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004647}
4648
4649void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004650 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004651 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004652 CodeCompletionContext::CCC_Other);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004653 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004654 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004655 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004656 HandleCodeCompleteResults(this, CodeCompleter,
4657 CodeCompletionContext::CCC_Other,
4658 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004659}
4660
Douglas Gregore6078da2009-11-19 00:14:45 +00004661/// \brief Determine whether the addition of the given flag to an Objective-C
4662/// property's attributes will cause a conflict.
Bill Wendling44426052012-12-20 19:22:21 +00004663static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregore6078da2009-11-19 00:14:45 +00004664 // Check if we've already added this flag.
Bill Wendling44426052012-12-20 19:22:21 +00004665 if (Attributes & NewFlag)
Douglas Gregore6078da2009-11-19 00:14:45 +00004666 return true;
4667
Bill Wendling44426052012-12-20 19:22:21 +00004668 Attributes |= NewFlag;
Douglas Gregore6078da2009-11-19 00:14:45 +00004669
4670 // Check for collisions with "readonly".
Bill Wendling44426052012-12-20 19:22:21 +00004671 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4672 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregore6078da2009-11-19 00:14:45 +00004673 return true;
4674
Jordan Rose53cb2f32012-08-20 20:01:13 +00004675 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendling44426052012-12-20 19:22:21 +00004676 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCall31168b02011-06-15 23:02:42 +00004677 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregore6078da2009-11-19 00:14:45 +00004678 ObjCDeclSpec::DQ_PR_copy |
Jordan Rose53cb2f32012-08-20 20:01:13 +00004679 ObjCDeclSpec::DQ_PR_retain |
4680 ObjCDeclSpec::DQ_PR_strong |
4681 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregore6078da2009-11-19 00:14:45 +00004682 if (AssignCopyRetMask &&
4683 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCall31168b02011-06-15 23:02:42 +00004684 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregore6078da2009-11-19 00:14:45 +00004685 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCall31168b02011-06-15 23:02:42 +00004686 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rose53cb2f32012-08-20 20:01:13 +00004687 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4688 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregore6078da2009-11-19 00:14:45 +00004689 return true;
4690
4691 return false;
4692}
4693
Douglas Gregor36029f42009-11-18 23:08:07 +00004694void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroff936354c2009-10-08 21:55:05 +00004695 if (!CodeCompleter)
4696 return;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004697
Bill Wendling44426052012-12-20 19:22:21 +00004698 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroff936354c2009-10-08 21:55:05 +00004699
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004700 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004701 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004702 CodeCompletionContext::CCC_Other);
Steve Naroff936354c2009-10-08 21:55:05 +00004703 Results.EnterNewScope();
Bill Wendling44426052012-12-20 19:22:21 +00004704 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall276321a2010-08-25 06:19:51 +00004705 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendling44426052012-12-20 19:22:21 +00004706 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall276321a2010-08-25 06:19:51 +00004707 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendling44426052012-12-20 19:22:21 +00004708 if (!ObjCPropertyFlagConflicts(Attributes,
John McCall31168b02011-06-15 23:02:42 +00004709 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4710 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendling44426052012-12-20 19:22:21 +00004711 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall276321a2010-08-25 06:19:51 +00004712 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendling44426052012-12-20 19:22:21 +00004713 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall276321a2010-08-25 06:19:51 +00004714 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendling44426052012-12-20 19:22:21 +00004715 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCall31168b02011-06-15 23:02:42 +00004716 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendling44426052012-12-20 19:22:21 +00004717 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall276321a2010-08-25 06:19:51 +00004718 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendling44426052012-12-20 19:22:21 +00004719 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall276321a2010-08-25 06:19:51 +00004720 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendling44426052012-12-20 19:22:21 +00004721 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian1c2d29e2011-06-11 17:14:27 +00004722 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rose53cb2f32012-08-20 20:01:13 +00004723
4724 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall3deb1ad2012-08-21 02:47:43 +00004725 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendling44426052012-12-20 19:22:21 +00004726 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rose53cb2f32012-08-20 20:01:13 +00004727 Results.AddResult(CodeCompletionResult("weak"));
4728
Bill Wendling44426052012-12-20 19:22:21 +00004729 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004730 CodeCompletionBuilder Setter(Results.getAllocator(),
4731 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004732 Setter.AddTypedTextChunk("setter");
Argyrios Kyrtzidis7bbb8812014-02-20 07:55:15 +00004733 Setter.AddTextChunk("=");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004734 Setter.AddPlaceholderChunk("method");
4735 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004736 }
Bill Wendling44426052012-12-20 19:22:21 +00004737 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004738 CodeCompletionBuilder Getter(Results.getAllocator(),
4739 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004740 Getter.AddTypedTextChunk("getter");
Argyrios Kyrtzidis7bbb8812014-02-20 07:55:15 +00004741 Getter.AddTextChunk("=");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004742 Getter.AddPlaceholderChunk("method");
4743 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004744 }
Steve Naroff936354c2009-10-08 21:55:05 +00004745 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004746 HandleCodeCompleteResults(this, CodeCompleter,
4747 CodeCompletionContext::CCC_Other,
4748 Results.data(),Results.size());
Steve Naroff936354c2009-10-08 21:55:05 +00004749}
Steve Naroffeae65032009-11-07 02:08:14 +00004750
James Dennettf1243872012-06-17 05:33:25 +00004751/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregorc8537c52009-11-19 07:41:15 +00004752/// via code completion.
4753enum ObjCMethodKind {
Dmitri Gribenko4280e5c2012-06-08 23:13:42 +00004754 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4755 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4756 MK_OneArgSelector ///< One-argument selector.
Douglas Gregorc8537c52009-11-19 07:41:15 +00004757};
4758
Douglas Gregor67c692c2010-08-26 15:07:07 +00004759static bool isAcceptableObjCSelector(Selector Sel,
4760 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004761 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004762 bool AllowSameLength = true) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004763 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor67c692c2010-08-26 15:07:07 +00004764 if (NumSelIdents > Sel.getNumArgs())
4765 return false;
4766
4767 switch (WantKind) {
4768 case MK_Any: break;
4769 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4770 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4771 }
4772
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004773 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4774 return false;
4775
Douglas Gregor67c692c2010-08-26 15:07:07 +00004776 for (unsigned I = 0; I != NumSelIdents; ++I)
4777 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4778 return false;
4779
4780 return true;
4781}
4782
Douglas Gregorc8537c52009-11-19 07:41:15 +00004783static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4784 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004785 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004786 bool AllowSameLength = true) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00004787 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004788 AllowSameLength);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004789}
Douglas Gregor1154e272010-09-16 16:06:31 +00004790
4791namespace {
4792 /// \brief A set of selectors, which is used to avoid introducing multiple
4793 /// completions with the same selector into the result set.
4794 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4795}
4796
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004797/// \brief Add all of the Objective-C methods in the given Objective-C
4798/// container to the set of results.
4799///
4800/// The container will be a class, protocol, category, or implementation of
4801/// any of the above. This mether will recurse to include methods from
4802/// the superclasses of classes along with their categories, protocols, and
4803/// implementations.
4804///
4805/// \param Container the container in which we'll look to find methods.
4806///
James Dennett596e4752012-06-14 03:11:41 +00004807/// \param WantInstanceMethods Whether to add instance methods (only); if
4808/// false, this routine will add factory methods (only).
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004809///
4810/// \param CurContext the context in which we're performing the lookup that
4811/// finds methods.
4812///
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004813/// \param AllowSameLength Whether we allow a method to be added to the list
4814/// when it has the same number of parameters as we have selector identifiers.
4815///
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004816/// \param Results the structure into which we'll add results.
4817static void AddObjCMethods(ObjCContainerDecl *Container,
4818 bool WantInstanceMethods,
Douglas Gregorc8537c52009-11-19 07:41:15 +00004819 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004820 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004821 DeclContext *CurContext,
Douglas Gregor1154e272010-09-16 16:06:31 +00004822 VisitedSelectorSet &Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004823 bool AllowSameLength,
Douglas Gregor416b5752010-08-25 01:08:01 +00004824 ResultBuilder &Results,
4825 bool InOriginalClass = true) {
John McCall276321a2010-08-25 06:19:51 +00004826 typedef CodeCompletionResult Result;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00004827 Container = getContainerDef(Container);
Douglas Gregor41778c32013-01-30 06:58:39 +00004828 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4829 bool isRootClass = IFace && !IFace->getSuperClass();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004830 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
4831 MEnd = Container->meth_end();
4832 M != MEnd; ++M) {
Douglas Gregor41778c32013-01-30 06:58:39 +00004833 // The instance methods on the root class can be messaged via the
4834 // metaclass.
4835 if (M->isInstanceMethod() == WantInstanceMethods ||
4836 (isRootClass && !WantInstanceMethods)) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00004837 // Check whether the selector identifiers we've been given are a
4838 // subset of the identifiers for this particular method.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004839 if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, AllowSameLength))
Douglas Gregor1b605f72009-11-19 01:08:35 +00004840 continue;
Douglas Gregorc8537c52009-11-19 07:41:15 +00004841
David Blaikie2d7c57e2012-04-30 02:36:29 +00004842 if (!Selectors.insert(M->getSelector()))
Douglas Gregor1154e272010-09-16 16:06:31 +00004843 continue;
4844
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004845 Result R = Result(*M, Results.getBasePriority(*M), 0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004846 R.StartParameter = SelIdents.size();
Douglas Gregorc8537c52009-11-19 07:41:15 +00004847 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor416b5752010-08-25 01:08:01 +00004848 if (!InOriginalClass)
4849 R.Priority += CCD_InBaseClass;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004850 Results.MaybeAddResult(R, CurContext);
4851 }
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004852 }
4853
Douglas Gregorf37c9492010-09-16 15:34:59 +00004854 // Visit the protocols of protocols.
4855 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregore6e48b12012-01-01 19:29:29 +00004856 if (Protocol->hasDefinition()) {
4857 const ObjCList<ObjCProtocolDecl> &Protocols
4858 = Protocol->getReferencedProtocols();
4859 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4860 E = Protocols.end();
4861 I != E; ++I)
4862 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004863 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregore6e48b12012-01-01 19:29:29 +00004864 }
Douglas Gregorf37c9492010-09-16 15:34:59 +00004865 }
4866
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00004867 if (!IFace || !IFace->hasDefinition())
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004868 return;
4869
4870 // Add methods in protocols.
Argyrios Kyrtzidise7f3ef32012-03-13 01:09:41 +00004871 for (ObjCInterfaceDecl::protocol_iterator I = IFace->protocol_begin(),
4872 E = IFace->protocol_end();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004873 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004874 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004875 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004876
4877 // Add methods in categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00004878 for (ObjCInterfaceDecl::known_categories_iterator
4879 Cat = IFace->known_categories_begin(),
4880 CatEnd = IFace->known_categories_end();
4881 Cat != CatEnd; ++Cat) {
4882 ObjCCategoryDecl *CatDecl = *Cat;
4883
4884 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004885 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004886 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004887
4888 // Add a categories protocol methods.
4889 const ObjCList<ObjCProtocolDecl> &Protocols
4890 = CatDecl->getReferencedProtocols();
4891 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4892 E = Protocols.end();
4893 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00004894 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004895 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004896 Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004897
4898 // Add methods in category implementations.
4899 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004900 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004901 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004902 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004903 }
4904
4905 // Add methods in superclass.
4906 if (IFace->getSuperClass())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004907 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004908 SelIdents, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004909 AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004910
4911 // Add methods in our implementation, if any.
4912 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004913 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004914 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004915 Results, InOriginalClass);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004916}
4917
4918
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004919void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004920 // Try to find the interface where getters might live.
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004921 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004922 if (!Class) {
4923 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004924 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004925 Class = Category->getClassInterface();
4926
4927 if (!Class)
4928 return;
4929 }
4930
4931 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004932 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004933 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004934 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004935 Results.EnterNewScope();
4936
Douglas Gregor1154e272010-09-16 16:06:31 +00004937 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004938 AddObjCMethods(Class, true, MK_ZeroArgSelector, None, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004939 /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004940 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004941 HandleCodeCompleteResults(this, CodeCompleter,
4942 CodeCompletionContext::CCC_Other,
4943 Results.data(),Results.size());
Douglas Gregorc8537c52009-11-19 07:41:15 +00004944}
4945
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004946void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004947 // Try to find the interface where setters might live.
4948 ObjCInterfaceDecl *Class
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004949 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004950 if (!Class) {
4951 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004952 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004953 Class = Category->getClassInterface();
4954
4955 if (!Class)
4956 return;
4957 }
4958
4959 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004960 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004961 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004962 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004963 Results.EnterNewScope();
4964
Douglas Gregor1154e272010-09-16 16:06:31 +00004965 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004966 AddObjCMethods(Class, true, MK_OneArgSelector, None, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004967 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004968
4969 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004970 HandleCodeCompleteResults(this, CodeCompleter,
4971 CodeCompletionContext::CCC_Other,
4972 Results.data(),Results.size());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004973}
4974
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004975void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
4976 bool IsParameter) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004977 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004978 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004979 CodeCompletionContext::CCC_Type);
Douglas Gregor99fa2642010-08-24 01:06:58 +00004980 Results.EnterNewScope();
4981
4982 // Add context-sensitive, Objective-C parameter-passing keywords.
4983 bool AddedInOut = false;
4984 if ((DS.getObjCDeclQualifier() &
4985 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
4986 Results.AddResult("in");
4987 Results.AddResult("inout");
4988 AddedInOut = true;
4989 }
4990 if ((DS.getObjCDeclQualifier() &
4991 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
4992 Results.AddResult("out");
4993 if (!AddedInOut)
4994 Results.AddResult("inout");
4995 }
4996 if ((DS.getObjCDeclQualifier() &
4997 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
4998 ObjCDeclSpec::DQ_Oneway)) == 0) {
4999 Results.AddResult("bycopy");
5000 Results.AddResult("byref");
5001 Results.AddResult("oneway");
5002 }
5003
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005004 // If we're completing the return type of an Objective-C method and the
5005 // identifier IBAction refers to a macro, provide a completion item for
5006 // an action, e.g.,
5007 // IBAction)<#selector#>:(id)sender
5008 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
5009 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005010 CodeCompletionBuilder Builder(Results.getAllocator(),
5011 Results.getCodeCompletionTUInfo(),
5012 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005013 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005014 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005015 Builder.AddPlaceholderChunk("selector");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005016 Builder.AddChunk(CodeCompletionString::CK_Colon);
5017 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005018 Builder.AddTextChunk("id");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005019 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005020 Builder.AddTextChunk("sender");
5021 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
5022 }
Douglas Gregored1f5972013-01-30 07:11:43 +00005023
5024 // If we're completing the return type, provide 'instancetype'.
5025 if (!IsParameter) {
5026 Results.AddResult(CodeCompletionResult("instancetype"));
5027 }
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005028
Douglas Gregor99fa2642010-08-24 01:06:58 +00005029 // Add various builtin type names and specifiers.
5030 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5031 Results.ExitScope();
5032
5033 // Add the various type names
5034 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5035 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5036 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5037 CodeCompleter->includeGlobals());
5038
5039 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005040 AddMacroResults(PP, Results, false);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005041
5042 HandleCodeCompleteResults(this, CodeCompleter,
5043 CodeCompletionContext::CCC_Type,
5044 Results.data(), Results.size());
5045}
5046
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005047/// \brief When we have an expression with type "id", we may assume
5048/// that it has some more-specific class type based on knowledge of
5049/// common uses of Objective-C. This routine returns that class type,
5050/// or NULL if no better result could be determined.
5051static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregored0b69d2010-09-15 16:23:04 +00005052 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005053 if (!Msg)
5054 return 0;
5055
5056 Selector Sel = Msg->getSelector();
5057 if (Sel.isNull())
5058 return 0;
5059
5060 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5061 if (!Id)
5062 return 0;
5063
5064 ObjCMethodDecl *Method = Msg->getMethodDecl();
5065 if (!Method)
5066 return 0;
5067
5068 // Determine the class that we're sending the message to.
Douglas Gregor9a129192010-04-21 00:45:42 +00005069 ObjCInterfaceDecl *IFace = 0;
5070 switch (Msg->getReceiverKind()) {
5071 case ObjCMessageExpr::Class:
John McCall8b07ec22010-05-15 11:32:37 +00005072 if (const ObjCObjectType *ObjType
5073 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5074 IFace = ObjType->getInterface();
Douglas Gregor9a129192010-04-21 00:45:42 +00005075 break;
5076
5077 case ObjCMessageExpr::Instance: {
5078 QualType T = Msg->getInstanceReceiver()->getType();
5079 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5080 IFace = Ptr->getInterfaceDecl();
5081 break;
5082 }
5083
5084 case ObjCMessageExpr::SuperInstance:
5085 case ObjCMessageExpr::SuperClass:
5086 break;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005087 }
5088
5089 if (!IFace)
5090 return 0;
5091
5092 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5093 if (Method->isInstanceMethod())
5094 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5095 .Case("retain", IFace)
John McCall31168b02011-06-15 23:02:42 +00005096 .Case("strong", IFace)
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005097 .Case("autorelease", IFace)
5098 .Case("copy", IFace)
5099 .Case("copyWithZone", IFace)
5100 .Case("mutableCopy", IFace)
5101 .Case("mutableCopyWithZone", IFace)
5102 .Case("awakeFromCoder", IFace)
5103 .Case("replacementObjectFromCoder", IFace)
5104 .Case("class", IFace)
5105 .Case("classForCoder", IFace)
5106 .Case("superclass", Super)
5107 .Default(0);
5108
5109 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5110 .Case("new", IFace)
5111 .Case("alloc", IFace)
5112 .Case("allocWithZone", IFace)
5113 .Case("class", IFace)
5114 .Case("superclass", Super)
5115 .Default(0);
5116}
5117
Douglas Gregor6fc04132010-08-27 15:10:57 +00005118// Add a special completion for a message send to "super", which fills in the
5119// most likely case of forwarding all of our arguments to the superclass
5120// function.
5121///
5122/// \param S The semantic analysis object.
5123///
Dmitri Gribenkoadba9be2012-08-23 17:58:28 +00005124/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor6fc04132010-08-27 15:10:57 +00005125/// the "super" keyword. Otherwise, we just need to provide the arguments.
5126///
5127/// \param SelIdents The identifiers in the selector that have already been
5128/// provided as arguments for a send to "super".
5129///
Douglas Gregor6fc04132010-08-27 15:10:57 +00005130/// \param Results The set of results to augment.
5131///
5132/// \returns the Objective-C method declaration that would be invoked by
5133/// this "super" completion. If NULL, no completion was added.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005134static ObjCMethodDecl *AddSuperSendCompletion(
5135 Sema &S, bool NeedSuperKeyword,
5136 ArrayRef<IdentifierInfo *> SelIdents,
5137 ResultBuilder &Results) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005138 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5139 if (!CurMethod)
5140 return 0;
5141
5142 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5143 if (!Class)
5144 return 0;
5145
5146 // Try to find a superclass method with the same selector.
5147 ObjCMethodDecl *SuperMethod = 0;
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005148 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5149 // Check in the class
Douglas Gregor6fc04132010-08-27 15:10:57 +00005150 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5151 CurMethod->isInstanceMethod());
5152
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005153 // Check in categories or class extensions.
5154 if (!SuperMethod) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005155 for (ObjCInterfaceDecl::known_categories_iterator
5156 Cat = Class->known_categories_begin(),
5157 CatEnd = Class->known_categories_end();
5158 Cat != CatEnd; ++Cat) {
5159 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005160 CurMethod->isInstanceMethod())))
5161 break;
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005162 }
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005163 }
5164 }
5165
Douglas Gregor6fc04132010-08-27 15:10:57 +00005166 if (!SuperMethod)
5167 return 0;
5168
5169 // Check whether the superclass method has the same signature.
5170 if (CurMethod->param_size() != SuperMethod->param_size() ||
5171 CurMethod->isVariadic() != SuperMethod->isVariadic())
5172 return 0;
5173
5174 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5175 CurPEnd = CurMethod->param_end(),
5176 SuperP = SuperMethod->param_begin();
5177 CurP != CurPEnd; ++CurP, ++SuperP) {
5178 // Make sure the parameter types are compatible.
5179 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5180 (*SuperP)->getType()))
5181 return 0;
5182
5183 // Make sure we have a parameter name to forward!
5184 if (!(*CurP)->getIdentifier())
5185 return 0;
5186 }
5187
5188 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005189 CodeCompletionBuilder Builder(Results.getAllocator(),
5190 Results.getCodeCompletionTUInfo());
Douglas Gregor6fc04132010-08-27 15:10:57 +00005191
5192 // Give this completion a return type.
Douglas Gregor75acd922011-09-27 23:30:47 +00005193 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5194 Builder);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005195
5196 // If we need the "super" keyword, add it (plus some spacing).
5197 if (NeedSuperKeyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005198 Builder.AddTypedTextChunk("super");
5199 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005200 }
5201
5202 Selector Sel = CurMethod->getSelector();
5203 if (Sel.isUnarySelector()) {
5204 if (NeedSuperKeyword)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005205 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005206 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005207 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005208 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005209 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005210 } else {
5211 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5212 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005213 if (I > SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005214 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005215
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005216 if (I < SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005217 Builder.AddInformativeChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005218 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005219 Sel.getNameForSlot(I) + ":"));
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005220 else if (NeedSuperKeyword || I > SelIdents.size()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005221 Builder.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005222 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005223 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005224 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005225 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005226 } else {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005227 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005228 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005229 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005230 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005231 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005232 }
5233 }
5234 }
5235
Douglas Gregor78254c82012-03-27 23:34:16 +00005236 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5237 CCP_SuperCompletion));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005238 return SuperMethod;
5239}
5240
Douglas Gregora817a192010-05-27 23:06:34 +00005241void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall276321a2010-08-25 06:19:51 +00005242 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005243 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005244 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005245 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005246 getLangOpts().CPlusPlus11
Douglas Gregord8c61782012-02-15 15:34:24 +00005247 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5248 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregora817a192010-05-27 23:06:34 +00005249
Douglas Gregora817a192010-05-27 23:06:34 +00005250 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5251 Results.EnterNewScope();
Douglas Gregor39982192010-08-15 06:18:01 +00005252 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5253 CodeCompleter->includeGlobals());
Douglas Gregora817a192010-05-27 23:06:34 +00005254
5255 // If we are in an Objective-C method inside a class that has a superclass,
5256 // add "super" as an option.
5257 if (ObjCMethodDecl *Method = getCurMethodDecl())
5258 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor6fc04132010-08-27 15:10:57 +00005259 if (Iface->getSuperClass()) {
Douglas Gregora817a192010-05-27 23:06:34 +00005260 Results.AddResult(Result("super"));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005261
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005262 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, None, Results);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005263 }
Douglas Gregora817a192010-05-27 23:06:34 +00005264
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005265 if (getLangOpts().CPlusPlus11)
Douglas Gregord8c61782012-02-15 15:34:24 +00005266 addThisCompletion(*this, Results);
5267
Douglas Gregora817a192010-05-27 23:06:34 +00005268 Results.ExitScope();
5269
5270 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005271 AddMacroResults(PP, Results, false);
Douglas Gregor50832e02010-09-20 22:39:41 +00005272 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005273 Results.data(), Results.size());
Douglas Gregora817a192010-05-27 23:06:34 +00005274
5275}
5276
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005277void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005278 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005279 bool AtArgumentExpression) {
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005280 ObjCInterfaceDecl *CDecl = 0;
5281 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5282 // Figure out which interface we're in.
5283 CDecl = CurMethod->getClassInterface();
5284 if (!CDecl)
5285 return;
5286
5287 // Find the superclass of this class.
5288 CDecl = CDecl->getSuperClass();
5289 if (!CDecl)
5290 return;
5291
5292 if (CurMethod->isInstanceMethod()) {
5293 // We are inside an instance method, which means that the message
5294 // send [super ...] is actually calling an instance method on the
Douglas Gregor392a84b2010-10-13 21:24:53 +00005295 // current object.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005296 return CodeCompleteObjCInstanceMessage(S, 0, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005297 AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005298 CDecl);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005299 }
5300
5301 // Fall through to send to the superclass in CDecl.
5302 } else {
5303 // "super" may be the name of a type or variable. Figure out which
5304 // it is.
Argyrios Kyrtzidis3e56dd42013-03-14 22:56:43 +00005305 IdentifierInfo *Super = getSuperIdentifier();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005306 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5307 LookupOrdinaryName);
5308 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5309 // "super" names an interface. Use it.
5310 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCall8b07ec22010-05-15 11:32:37 +00005311 if (const ObjCObjectType *Iface
5312 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5313 CDecl = Iface->getInterface();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005314 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5315 // "super" names an unresolved type; we can't be more specific.
5316 } else {
5317 // Assume that "super" names some kind of value and parse that way.
5318 CXXScopeSpec SS;
Abramo Bagnara7945c982012-01-27 09:46:47 +00005319 SourceLocation TemplateKWLoc;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005320 UnqualifiedId id;
5321 id.setIdentifier(Super, SuperLoc);
Abramo Bagnara7945c982012-01-27 09:46:47 +00005322 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5323 false, false);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005324 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005325 SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005326 AtArgumentExpression);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005327 }
5328
5329 // Fall through
5330 }
5331
John McCallba7bf592010-08-24 05:47:05 +00005332 ParsedType Receiver;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005333 if (CDecl)
John McCallba7bf592010-08-24 05:47:05 +00005334 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005335 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005336 AtArgumentExpression,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005337 /*IsSuper=*/true);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005338}
5339
Douglas Gregor74661272010-09-21 00:03:25 +00005340/// \brief Given a set of code-completion results for the argument of a message
5341/// send, determine the preferred type (if any) for that argument expression.
5342static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5343 unsigned NumSelIdents) {
5344 typedef CodeCompletionResult Result;
5345 ASTContext &Context = Results.getSema().Context;
5346
5347 QualType PreferredType;
5348 unsigned BestPriority = CCP_Unlikely * 2;
5349 Result *ResultsData = Results.data();
5350 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5351 Result &R = ResultsData[I];
5352 if (R.Kind == Result::RK_Declaration &&
5353 isa<ObjCMethodDecl>(R.Declaration)) {
5354 if (R.Priority <= BestPriority) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00005355 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregor74661272010-09-21 00:03:25 +00005356 if (NumSelIdents <= Method->param_size()) {
5357 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
5358 ->getType();
5359 if (R.Priority < BestPriority || PreferredType.isNull()) {
5360 BestPriority = R.Priority;
5361 PreferredType = MyPreferredType;
5362 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5363 MyPreferredType)) {
5364 PreferredType = QualType();
5365 }
5366 }
5367 }
5368 }
5369 }
5370
5371 return PreferredType;
5372}
5373
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005374static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5375 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005376 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005377 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005378 bool IsSuper,
5379 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005380 typedef CodeCompletionResult Result;
Douglas Gregor8ce33212009-11-17 17:59:40 +00005381 ObjCInterfaceDecl *CDecl = 0;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005382
Douglas Gregor8ce33212009-11-17 17:59:40 +00005383 // If the given name refers to an interface type, retrieve the
5384 // corresponding declaration.
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005385 if (Receiver) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005386 QualType T = SemaRef.GetTypeFromParser(Receiver, 0);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005387 if (!T.isNull())
John McCall8b07ec22010-05-15 11:32:37 +00005388 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5389 CDecl = Interface->getInterface();
Douglas Gregor8ce33212009-11-17 17:59:40 +00005390 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005391
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005392 // Add all of the factory methods in this Objective-C class, its protocols,
5393 // superclasses, categories, implementation, etc.
Steve Naroffeae65032009-11-07 02:08:14 +00005394 Results.EnterNewScope();
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005395
Douglas Gregor6fc04132010-08-27 15:10:57 +00005396 // If this is a send-to-super, try to add the special "super" send
5397 // completion.
5398 if (IsSuper) {
5399 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005400 = AddSuperSendCompletion(SemaRef, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005401 Results.Ignore(SuperMethod);
5402 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005403
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005404 // If we're inside an Objective-C method definition, prefer its selector to
5405 // others.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005406 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005407 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005408
Douglas Gregor1154e272010-09-16 16:06:31 +00005409 VisitedSelectorSet Selectors;
Douglas Gregor6285f752010-04-06 16:40:00 +00005410 if (CDecl)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005411 AddObjCMethods(CDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005412 SemaRef.CurContext, Selectors, AtArgumentExpression,
5413 Results);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005414 else {
Douglas Gregor6285f752010-04-06 16:40:00 +00005415 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005416
Douglas Gregord720daf2010-04-06 17:30:22 +00005417 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005418 // pool from the AST file.
Axel Naumanndd433f02012-10-18 19:05:02 +00005419 if (SemaRef.getExternalSource()) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005420 for (uint32_t I = 0,
Axel Naumanndd433f02012-10-18 19:05:02 +00005421 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall75b960e2010-06-01 09:23:16 +00005422 I != N; ++I) {
Axel Naumanndd433f02012-10-18 19:05:02 +00005423 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005424 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005425 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005426
5427 SemaRef.ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005428 }
5429 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005430
5431 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5432 MEnd = SemaRef.MethodPool.end();
Sebastian Redl75d8a322010-08-02 23:18:59 +00005433 M != MEnd; ++M) {
5434 for (ObjCMethodList *MethList = &M->second.second;
5435 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005436 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005437 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005438 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005439
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005440 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005441 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005442 R.AllParametersAreInformative = false;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005443 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor6285f752010-04-06 16:40:00 +00005444 }
5445 }
5446 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005447
5448 Results.ExitScope();
5449}
Douglas Gregor6285f752010-04-06 16:40:00 +00005450
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005451void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005452 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005453 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005454 bool IsSuper) {
Douglas Gregor63745d52011-07-21 01:05:26 +00005455
5456 QualType T = this->GetTypeFromParser(Receiver);
5457
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005458 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005459 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005460 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005461 T, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005462
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005463 AddClassMessageCompletions(*this, S, Receiver, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005464 AtArgumentExpression, IsSuper, Results);
Douglas Gregor74661272010-09-21 00:03:25 +00005465
5466 // If we're actually at the argument expression (rather than prior to the
5467 // selector), we're actually performing code completion for an expression.
5468 // Determine whether we have a single, best method. If so, we can
5469 // code-complete the expression using the corresponding parameter type as
5470 // our preferred type, improving completion results.
5471 if (AtArgumentExpression) {
5472 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005473 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005474 if (PreferredType.isNull())
5475 CodeCompleteOrdinaryName(S, PCC_Expression);
5476 else
5477 CodeCompleteExpression(S, PreferredType);
5478 return;
5479 }
5480
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005481 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005482 Results.getCompletionContext(),
Douglas Gregor6fc04132010-08-27 15:10:57 +00005483 Results.data(), Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005484}
5485
Richard Trieu2bd04012011-09-09 02:00:50 +00005486void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005487 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005488 bool AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005489 ObjCInterfaceDecl *Super) {
John McCall276321a2010-08-25 06:19:51 +00005490 typedef CodeCompletionResult Result;
Steve Naroffeae65032009-11-07 02:08:14 +00005491
5492 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffeae65032009-11-07 02:08:14 +00005493
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005494 // If necessary, apply function/array conversion to the receiver.
5495 // C99 6.7.5.3p[7,8].
John Wiegley01296292011-04-08 18:41:53 +00005496 if (RecExpr) {
5497 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5498 if (Conv.isInvalid()) // conversion failed. bail.
5499 return;
5500 RecExpr = Conv.take();
5501 }
Douglas Gregor392a84b2010-10-13 21:24:53 +00005502 QualType ReceiverType = RecExpr? RecExpr->getType()
5503 : Super? Context.getObjCObjectPointerType(
5504 Context.getObjCInterfaceType(Super))
5505 : Context.getObjCIdType();
Steve Naroffeae65032009-11-07 02:08:14 +00005506
Douglas Gregordc520b02010-11-08 21:12:30 +00005507 // If we're messaging an expression with type "id" or "Class", check
5508 // whether we know something special about the receiver that allows
5509 // us to assume a more-specific receiver type.
Anders Carlsson382ba412014-02-28 19:07:22 +00005510 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType()) {
Douglas Gregordc520b02010-11-08 21:12:30 +00005511 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5512 if (ReceiverType->isObjCClassType())
5513 return CodeCompleteObjCClassMessage(S,
5514 ParsedType::make(Context.getObjCInterfaceType(IFace)),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005515 SelIdents,
Douglas Gregordc520b02010-11-08 21:12:30 +00005516 AtArgumentExpression, Super);
5517
5518 ReceiverType = Context.getObjCObjectPointerType(
5519 Context.getObjCInterfaceType(IFace));
5520 }
Anders Carlsson382ba412014-02-28 19:07:22 +00005521 } else if (RecExpr && getLangOpts().CPlusPlus) {
5522 ExprResult Conv = PerformContextuallyConvertToObjCPointer(RecExpr);
5523 if (Conv.isUsable()) {
5524 RecExpr = Conv.take();
5525 ReceiverType = RecExpr->getType();
5526 }
5527 }
Douglas Gregordc520b02010-11-08 21:12:30 +00005528
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005529 // Build the set of methods we can see.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005530 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005531 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005532 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005533 ReceiverType, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005534
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005535 Results.EnterNewScope();
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005536
Douglas Gregor6fc04132010-08-27 15:10:57 +00005537 // If this is a send-to-super, try to add the special "super" send
5538 // completion.
Douglas Gregor392a84b2010-10-13 21:24:53 +00005539 if (Super) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005540 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005541 = AddSuperSendCompletion(*this, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005542 Results.Ignore(SuperMethod);
5543 }
5544
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005545 // If we're inside an Objective-C method definition, prefer its selector to
5546 // others.
5547 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5548 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005549
Douglas Gregor1154e272010-09-16 16:06:31 +00005550 // Keep track of the selectors we've already added.
5551 VisitedSelectorSet Selectors;
5552
Douglas Gregora3329fa2009-11-18 00:06:18 +00005553 // Handle messages to Class. This really isn't a message to an instance
5554 // method, so we treat it the same way we would treat a message send to a
5555 // class method.
5556 if (ReceiverType->isObjCClassType() ||
5557 ReceiverType->isObjCQualifiedClassType()) {
5558 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5559 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005560 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005561 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005562 }
5563 }
5564 // Handle messages to a qualified ID ("id<foo>").
5565 else if (const ObjCObjectPointerType *QualID
5566 = ReceiverType->getAsObjCQualifiedIdType()) {
5567 // Search protocols for instance methods.
5568 for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(),
5569 E = QualID->qual_end();
5570 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005571 AddObjCMethods(*I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005572 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005573 }
5574 // Handle messages to a pointer to interface type.
5575 else if (const ObjCObjectPointerType *IFacePtr
5576 = ReceiverType->getAsObjCInterfacePointerType()) {
5577 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregorc8537c52009-11-19 07:41:15 +00005578 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005579 CurContext, Selectors, AtArgumentExpression,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005580 Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005581
5582 // Search protocols for instance methods.
5583 for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(),
5584 E = IFacePtr->qual_end();
5585 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005586 AddObjCMethods(*I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005587 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005588 }
Douglas Gregor6285f752010-04-06 16:40:00 +00005589 // Handle messages to "id".
5590 else if (ReceiverType->isObjCIdType()) {
Douglas Gregord720daf2010-04-06 17:30:22 +00005591 // We're messaging "id", so provide all instance methods we know
5592 // about as code-completion results.
5593
5594 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005595 // pool from the AST file.
Douglas Gregord720daf2010-04-06 17:30:22 +00005596 if (ExternalSource) {
John McCall75b960e2010-06-01 09:23:16 +00005597 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5598 I != N; ++I) {
5599 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00005600 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005601 continue;
5602
Sebastian Redl75d8a322010-08-02 23:18:59 +00005603 ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005604 }
5605 }
5606
Sebastian Redl75d8a322010-08-02 23:18:59 +00005607 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5608 MEnd = MethodPool.end();
5609 M != MEnd; ++M) {
5610 for (ObjCMethodList *MethList = &M->second.first;
5611 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005612 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005613 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005614 continue;
Douglas Gregor1154e272010-09-16 16:06:31 +00005615
5616 if (!Selectors.insert(MethList->Method->getSelector()))
5617 continue;
5618
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005619 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005620 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005621 R.AllParametersAreInformative = false;
5622 Results.MaybeAddResult(R, CurContext);
5623 }
5624 }
5625 }
Steve Naroffeae65032009-11-07 02:08:14 +00005626 Results.ExitScope();
Douglas Gregor74661272010-09-21 00:03:25 +00005627
5628
5629 // If we're actually at the argument expression (rather than prior to the
5630 // selector), we're actually performing code completion for an expression.
5631 // Determine whether we have a single, best method. If so, we can
5632 // code-complete the expression using the corresponding parameter type as
5633 // our preferred type, improving completion results.
5634 if (AtArgumentExpression) {
5635 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005636 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005637 if (PreferredType.isNull())
5638 CodeCompleteOrdinaryName(S, PCC_Expression);
5639 else
5640 CodeCompleteExpression(S, PreferredType);
5641 return;
5642 }
5643
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005644 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005645 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005646 Results.data(),Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005647}
Douglas Gregorbaf69612009-11-18 04:19:12 +00005648
Douglas Gregor68762e72010-08-23 21:17:50 +00005649void Sema::CodeCompleteObjCForCollection(Scope *S,
5650 DeclGroupPtrTy IterationVar) {
5651 CodeCompleteExpressionData Data;
5652 Data.ObjCCollection = true;
5653
5654 if (IterationVar.getAsOpaquePtr()) {
Serge Pavlov9ddb76e2013-08-27 13:15:56 +00005655 DeclGroupRef DG = IterationVar.get();
Douglas Gregor68762e72010-08-23 21:17:50 +00005656 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5657 if (*I)
5658 Data.IgnoreDecls.push_back(*I);
5659 }
5660 }
5661
5662 CodeCompleteExpression(S, Data);
5663}
5664
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005665void Sema::CodeCompleteObjCSelector(Scope *S,
5666 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00005667 // If we have an external source, load the entire class method
5668 // pool from the AST file.
5669 if (ExternalSource) {
5670 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5671 I != N; ++I) {
5672 Selector Sel = ExternalSource->GetExternalSelector(I);
5673 if (Sel.isNull() || MethodPool.count(Sel))
5674 continue;
5675
5676 ReadMethodPool(Sel);
5677 }
5678 }
5679
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005680 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005681 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005682 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor67c692c2010-08-26 15:07:07 +00005683 Results.EnterNewScope();
5684 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5685 MEnd = MethodPool.end();
5686 M != MEnd; ++M) {
5687
5688 Selector Sel = M->first;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005689 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents))
Douglas Gregor67c692c2010-08-26 15:07:07 +00005690 continue;
5691
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005692 CodeCompletionBuilder Builder(Results.getAllocator(),
5693 Results.getCodeCompletionTUInfo());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005694 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005695 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005696 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005697 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005698 continue;
5699 }
5700
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005701 std::string Accumulator;
Douglas Gregor67c692c2010-08-26 15:07:07 +00005702 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005703 if (I == SelIdents.size()) {
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005704 if (!Accumulator.empty()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005705 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005706 Accumulator));
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005707 Accumulator.clear();
5708 }
5709 }
5710
Benjamin Kramer632500c2011-07-26 16:59:25 +00005711 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005712 Accumulator += ':';
Douglas Gregor67c692c2010-08-26 15:07:07 +00005713 }
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005714 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005715 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005716 }
5717 Results.ExitScope();
5718
5719 HandleCodeCompleteResults(this, CodeCompleter,
5720 CodeCompletionContext::CCC_SelectorName,
5721 Results.data(), Results.size());
5722}
5723
Douglas Gregorbaf69612009-11-18 04:19:12 +00005724/// \brief Add all of the protocol declarations that we find in the given
5725/// (translation unit) context.
5726static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005727 bool OnlyForwardDeclarations,
Douglas Gregorbaf69612009-11-18 04:19:12 +00005728 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005729 typedef CodeCompletionResult Result;
Douglas Gregorbaf69612009-11-18 04:19:12 +00005730
Aaron Ballman629afae2014-03-07 19:56:05 +00005731 for (const auto *D : Ctx->decls()) {
Douglas Gregorbaf69612009-11-18 04:19:12 +00005732 // Record any protocols we find.
Aaron Ballman629afae2014-03-07 19:56:05 +00005733 if (const auto *Proto = dyn_cast<ObjCProtocolDecl>(D))
Douglas Gregore6e48b12012-01-01 19:29:29 +00005734 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005735 Results.AddResult(Result(Proto, Results.getBasePriority(Proto), 0),
5736 CurContext, 0, false);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005737 }
5738}
5739
5740void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5741 unsigned NumProtocols) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005742 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005743 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005744 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005745
Douglas Gregora3b23b02010-12-09 21:44:02 +00005746 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5747 Results.EnterNewScope();
5748
5749 // Tell the result set to ignore all of the protocols we have
5750 // already seen.
5751 // FIXME: This doesn't work when caching code-completion results.
5752 for (unsigned I = 0; I != NumProtocols; ++I)
5753 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5754 Protocols[I].second))
5755 Results.Ignore(Protocol);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005756
Douglas Gregora3b23b02010-12-09 21:44:02 +00005757 // Add all protocols.
5758 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5759 Results);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005760
Douglas Gregora3b23b02010-12-09 21:44:02 +00005761 Results.ExitScope();
5762 }
5763
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005764 HandleCodeCompleteResults(this, CodeCompleter,
5765 CodeCompletionContext::CCC_ObjCProtocolName,
5766 Results.data(),Results.size());
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005767}
5768
5769void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005770 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005771 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005772 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005773
Douglas Gregora3b23b02010-12-09 21:44:02 +00005774 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5775 Results.EnterNewScope();
5776
5777 // Add all protocols.
5778 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5779 Results);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005780
Douglas Gregora3b23b02010-12-09 21:44:02 +00005781 Results.ExitScope();
5782 }
5783
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005784 HandleCodeCompleteResults(this, CodeCompleter,
5785 CodeCompletionContext::CCC_ObjCProtocolName,
5786 Results.data(),Results.size());
Douglas Gregorbaf69612009-11-18 04:19:12 +00005787}
Douglas Gregor49c22a72009-11-18 16:26:39 +00005788
5789/// \brief Add all of the Objective-C interface declarations that we find in
5790/// the given (translation unit) context.
5791static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5792 bool OnlyForwardDeclarations,
5793 bool OnlyUnimplemented,
5794 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005795 typedef CodeCompletionResult Result;
Douglas Gregor49c22a72009-11-18 16:26:39 +00005796
Aaron Ballman629afae2014-03-07 19:56:05 +00005797 for (const auto *D : Ctx->decls()) {
Douglas Gregor1c283312010-08-11 12:19:30 +00005798 // Record any interfaces we find.
Aaron Ballman629afae2014-03-07 19:56:05 +00005799 if (const auto *Class = dyn_cast<ObjCInterfaceDecl>(D))
Douglas Gregordc9166c2011-12-15 20:29:51 +00005800 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregor1c283312010-08-11 12:19:30 +00005801 (!OnlyUnimplemented || !Class->getImplementation()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005802 Results.AddResult(Result(Class, Results.getBasePriority(Class), 0),
5803 CurContext, 0, false);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005804 }
5805}
5806
5807void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
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_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005811 Results.EnterNewScope();
5812
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005813 if (CodeCompleter->includeGlobals()) {
5814 // Add all classes.
5815 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5816 false, Results);
5817 }
5818
Douglas Gregor49c22a72009-11-18 16:26:39 +00005819 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005820
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005821 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005822 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005823 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005824}
5825
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005826void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5827 SourceLocation ClassNameLoc) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005828 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005829 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005830 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005831 Results.EnterNewScope();
5832
5833 // Make sure that we ignore the class we're currently defining.
5834 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005835 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005836 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor49c22a72009-11-18 16:26:39 +00005837 Results.Ignore(CurClass);
5838
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005839 if (CodeCompleter->includeGlobals()) {
5840 // Add all classes.
5841 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5842 false, Results);
5843 }
5844
Douglas Gregor49c22a72009-11-18 16:26:39 +00005845 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005846
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005847 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005848 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005849 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005850}
5851
5852void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005853 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005854 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005855 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005856 Results.EnterNewScope();
5857
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005858 if (CodeCompleter->includeGlobals()) {
5859 // Add all unimplemented classes.
5860 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5861 true, Results);
5862 }
5863
Douglas Gregor49c22a72009-11-18 16:26:39 +00005864 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005865
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005866 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005867 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005868 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005869}
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005870
5871void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005872 IdentifierInfo *ClassName,
5873 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005874 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005875
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005876 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005877 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005878 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005879
5880 // Ignore any categories we find that have already been implemented by this
5881 // interface.
5882 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5883 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005884 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005885 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
5886 for (ObjCInterfaceDecl::visible_categories_iterator
5887 Cat = Class->visible_categories_begin(),
5888 CatEnd = Class->visible_categories_end();
5889 Cat != CatEnd; ++Cat) {
5890 CategoryNames.insert(Cat->getIdentifier());
5891 }
5892 }
5893
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005894 // Add all of the categories we know about.
5895 Results.EnterNewScope();
5896 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
Aaron Ballman629afae2014-03-07 19:56:05 +00005897 for (const auto *D : TU->decls())
5898 if (const auto *Category = dyn_cast<ObjCCategoryDecl>(D))
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005899 if (CategoryNames.insert(Category->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005900 Results.AddResult(Result(Category, Results.getBasePriority(Category),0),
5901 CurContext, 0, false);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005902 Results.ExitScope();
5903
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005904 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005905 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005906 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005907}
5908
5909void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005910 IdentifierInfo *ClassName,
5911 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005912 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005913
5914 // Find the corresponding interface. If we couldn't find the interface, the
5915 // program itself is ill-formed. However, we'll try to be helpful still by
5916 // providing the list of all of the categories we know about.
5917 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005918 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005919 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5920 if (!Class)
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005921 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005922
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005923 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005924 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005925 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005926
5927 // Add all of the categories that have have corresponding interface
5928 // declarations in this class and any of its superclasses, except for
5929 // already-implemented categories in the class itself.
5930 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5931 Results.EnterNewScope();
5932 bool IgnoreImplemented = true;
5933 while (Class) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005934 for (ObjCInterfaceDecl::visible_categories_iterator
5935 Cat = Class->visible_categories_begin(),
5936 CatEnd = Class->visible_categories_end();
5937 Cat != CatEnd; ++Cat) {
5938 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
5939 CategoryNames.insert(Cat->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005940 Results.AddResult(Result(*Cat, Results.getBasePriority(*Cat), 0),
5941 CurContext, 0, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005942 }
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005943
5944 Class = Class->getSuperClass();
5945 IgnoreImplemented = false;
5946 }
5947 Results.ExitScope();
5948
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005949 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005950 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005951 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005952}
Douglas Gregor5d649882009-11-18 22:32:06 +00005953
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005954void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005955 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005956 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005957 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00005958
5959 // Figure out where this @synthesize lives.
5960 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005961 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00005962 if (!Container ||
5963 (!isa<ObjCImplementationDecl>(Container) &&
5964 !isa<ObjCCategoryImplDecl>(Container)))
5965 return;
5966
5967 // Ignore any properties that have already been implemented.
Douglas Gregor9b4f3702012-06-12 13:44:08 +00005968 Container = getContainerDef(Container);
Aaron Ballman629afae2014-03-07 19:56:05 +00005969 for (const auto *D : Container->decls())
5970 if (const auto *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(D))
Douglas Gregor5d649882009-11-18 22:32:06 +00005971 Results.Ignore(PropertyImpl->getPropertyDecl());
5972
5973 // Add any properties that we find.
Douglas Gregorb888acf2010-12-09 23:01:55 +00005974 AddedPropertiesSet AddedProperties;
Douglas Gregor5d649882009-11-18 22:32:06 +00005975 Results.EnterNewScope();
5976 if (ObjCImplementationDecl *ClassImpl
5977 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor95147142011-05-05 15:50:42 +00005978 AddObjCProperties(ClassImpl->getClassInterface(), false,
5979 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00005980 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00005981 else
5982 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor95147142011-05-05 15:50:42 +00005983 false, /*AllowNullaryMethods=*/false, CurContext,
5984 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00005985 Results.ExitScope();
5986
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005987 HandleCodeCompleteResults(this, CodeCompleter,
5988 CodeCompletionContext::CCC_Other,
5989 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00005990}
5991
5992void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005993 IdentifierInfo *PropertyName) {
John McCall276321a2010-08-25 06:19:51 +00005994 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005995 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005996 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005997 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00005998
5999 // Figure out where this @synthesize lives.
6000 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006001 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00006002 if (!Container ||
6003 (!isa<ObjCImplementationDecl>(Container) &&
6004 !isa<ObjCCategoryImplDecl>(Container)))
6005 return;
6006
6007 // Figure out which interface we're looking into.
6008 ObjCInterfaceDecl *Class = 0;
6009 if (ObjCImplementationDecl *ClassImpl
6010 = dyn_cast<ObjCImplementationDecl>(Container))
6011 Class = ClassImpl->getClassInterface();
6012 else
6013 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
6014 ->getClassInterface();
6015
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006016 // Determine the type of the property we're synthesizing.
6017 QualType PropertyType = Context.getObjCIdType();
6018 if (Class) {
6019 if (ObjCPropertyDecl *Property
6020 = Class->FindPropertyDeclaration(PropertyName)) {
6021 PropertyType
6022 = Property->getType().getNonReferenceType().getUnqualifiedType();
6023
6024 // Give preference to ivars
6025 Results.setPreferredType(PropertyType);
6026 }
6027 }
6028
Douglas Gregor5d649882009-11-18 22:32:06 +00006029 // Add all of the instance variables in this class and its superclasses.
6030 Results.EnterNewScope();
Douglas Gregor331faa02011-04-18 14:13:53 +00006031 bool SawSimilarlyNamedIvar = false;
6032 std::string NameWithPrefix;
6033 NameWithPrefix += '_';
Benjamin Kramer632500c2011-07-26 16:59:25 +00006034 NameWithPrefix += PropertyName->getName();
Douglas Gregor331faa02011-04-18 14:13:53 +00006035 std::string NameWithSuffix = PropertyName->getName().str();
6036 NameWithSuffix += '_';
Douglas Gregor5d649882009-11-18 22:32:06 +00006037 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006038 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
6039 Ivar = Ivar->getNextIvar()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00006040 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), 0),
6041 CurContext, 0, false);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006042
Douglas Gregor331faa02011-04-18 14:13:53 +00006043 // Determine whether we've seen an ivar with a name similar to the
6044 // property.
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006045 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregor331faa02011-04-18 14:13:53 +00006046 NameWithPrefix == Ivar->getName() ||
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006047 NameWithSuffix == Ivar->getName())) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006048 SawSimilarlyNamedIvar = true;
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006049
6050 // Reduce the priority of this result by one, to give it a slight
6051 // advantage over other results whose names don't match so closely.
6052 if (Results.size() &&
6053 Results.data()[Results.size() - 1].Kind
6054 == CodeCompletionResult::RK_Declaration &&
6055 Results.data()[Results.size() - 1].Declaration == Ivar)
6056 Results.data()[Results.size() - 1].Priority--;
6057 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006058 }
Douglas Gregor5d649882009-11-18 22:32:06 +00006059 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006060
6061 if (!SawSimilarlyNamedIvar) {
6062 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006063 // an ivar of the appropriate type.
6064 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregor331faa02011-04-18 14:13:53 +00006065 typedef CodeCompletionResult Result;
6066 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006067 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6068 Priority,CXAvailability_Available);
Douglas Gregor331faa02011-04-18 14:13:53 +00006069
Douglas Gregor75acd922011-09-27 23:30:47 +00006070 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006071 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006072 Policy, Allocator));
Douglas Gregor331faa02011-04-18 14:13:53 +00006073 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6074 Results.AddResult(Result(Builder.TakeString(), Priority,
6075 CXCursor_ObjCIvarDecl));
6076 }
6077
Douglas Gregor5d649882009-11-18 22:32:06 +00006078 Results.ExitScope();
6079
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006080 HandleCodeCompleteResults(this, CodeCompleter,
6081 CodeCompletionContext::CCC_Other,
6082 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006083}
Douglas Gregor636a61e2010-04-07 00:21:17 +00006084
Douglas Gregor416b5752010-08-25 01:08:01 +00006085// Mapping from selectors to the methods that implement that selector, along
6086// with the "in original class" flag.
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006087typedef llvm::DenseMap<
6088 Selector, llvm::PointerIntPair<ObjCMethodDecl *, 1, bool> > KnownMethodsMap;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006089
6090/// \brief Find all of the methods that reside in the given container
6091/// (and its superclasses, protocols, etc.) that meet the given
6092/// criteria. Insert those methods into the map of known methods,
6093/// indexed by selector so they can be easily found.
6094static void FindImplementableMethods(ASTContext &Context,
6095 ObjCContainerDecl *Container,
6096 bool WantInstanceMethods,
6097 QualType ReturnType,
Douglas Gregor416b5752010-08-25 01:08:01 +00006098 KnownMethodsMap &KnownMethods,
6099 bool InOriginalClass = true) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006100 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006101 // Make sure we have a definition; that's what we'll walk.
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006102 if (!IFace->hasDefinition())
6103 return;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006104
6105 IFace = IFace->getDefinition();
6106 Container = IFace;
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006107
Douglas Gregor636a61e2010-04-07 00:21:17 +00006108 const ObjCList<ObjCProtocolDecl> &Protocols
6109 = IFace->getReferencedProtocols();
6110 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006111 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006112 I != E; ++I)
6113 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006114 KnownMethods, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006115
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006116 // Add methods from any class extensions and categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006117 for (ObjCInterfaceDecl::visible_categories_iterator
6118 Cat = IFace->visible_categories_begin(),
6119 CatEnd = IFace->visible_categories_end();
6120 Cat != CatEnd; ++Cat) {
6121 FindImplementableMethods(Context, *Cat, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006122 KnownMethods, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006123 }
6124
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006125 // Visit the superclass.
6126 if (IFace->getSuperClass())
6127 FindImplementableMethods(Context, IFace->getSuperClass(),
6128 WantInstanceMethods, ReturnType,
6129 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006130 }
6131
6132 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6133 // Recurse into protocols.
6134 const ObjCList<ObjCProtocolDecl> &Protocols
6135 = Category->getReferencedProtocols();
6136 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006137 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006138 I != E; ++I)
6139 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006140 KnownMethods, InOriginalClass);
6141
6142 // If this category is the original class, jump to the interface.
6143 if (InOriginalClass && Category->getClassInterface())
6144 FindImplementableMethods(Context, Category->getClassInterface(),
6145 WantInstanceMethods, ReturnType, KnownMethods,
6146 false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006147 }
6148
6149 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006150 // Make sure we have a definition; that's what we'll walk.
6151 if (!Protocol->hasDefinition())
6152 return;
6153 Protocol = Protocol->getDefinition();
6154 Container = Protocol;
6155
6156 // Recurse into protocols.
6157 const ObjCList<ObjCProtocolDecl> &Protocols
6158 = Protocol->getReferencedProtocols();
6159 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6160 E = Protocols.end();
6161 I != E; ++I)
6162 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6163 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006164 }
6165
6166 // Add methods in this container. This operation occurs last because
6167 // we want the methods from this container to override any methods
6168 // we've previously seen with the same selector.
6169 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
6170 MEnd = Container->meth_end();
6171 M != MEnd; ++M) {
David Blaikie2d7c57e2012-04-30 02:36:29 +00006172 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006173 if (!ReturnType.isNull() &&
Alp Toker314cc812014-01-25 16:55:45 +00006174 !Context.hasSameUnqualifiedType(ReturnType, M->getReturnType()))
Douglas Gregor636a61e2010-04-07 00:21:17 +00006175 continue;
6176
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006177 KnownMethods[M->getSelector()] =
6178 KnownMethodsMap::mapped_type(*M, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006179 }
6180 }
6181}
6182
Douglas Gregor669a25a2011-02-17 00:22:45 +00006183/// \brief Add the parenthesized return or parameter type chunk to a code
6184/// completion string.
6185static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor29979142012-04-10 18:35:07 +00006186 unsigned ObjCDeclQuals,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006187 ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006188 const PrintingPolicy &Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006189 CodeCompletionBuilder &Builder) {
6190 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor29979142012-04-10 18:35:07 +00006191 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6192 if (!Quals.empty())
6193 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor75acd922011-09-27 23:30:47 +00006194 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006195 Builder.getAllocator()));
6196 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6197}
6198
6199/// \brief Determine whether the given class is or inherits from a class by
6200/// the given name.
6201static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006202 StringRef Name) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006203 if (!Class)
6204 return false;
6205
6206 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6207 return true;
6208
6209 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6210}
6211
6212/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6213/// Key-Value Observing (KVO).
6214static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6215 bool IsInstanceMethod,
6216 QualType ReturnType,
6217 ASTContext &Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006218 VisitedSelectorSet &KnownSelectors,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006219 ResultBuilder &Results) {
6220 IdentifierInfo *PropName = Property->getIdentifier();
6221 if (!PropName || PropName->getLength() == 0)
6222 return;
6223
Douglas Gregor75acd922011-09-27 23:30:47 +00006224 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6225
Douglas Gregor669a25a2011-02-17 00:22:45 +00006226 // Builder that will create each code completion.
6227 typedef CodeCompletionResult Result;
6228 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006229 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor669a25a2011-02-17 00:22:45 +00006230
6231 // The selector table.
6232 SelectorTable &Selectors = Context.Selectors;
6233
6234 // The property name, copied into the code completion allocation region
6235 // on demand.
6236 struct KeyHolder {
6237 CodeCompletionAllocator &Allocator;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006238 StringRef Key;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006239 const char *CopiedKey;
6240
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006241 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006242 : Allocator(Allocator), Key(Key), CopiedKey(0) { }
6243
6244 operator const char *() {
6245 if (CopiedKey)
6246 return CopiedKey;
6247
6248 return CopiedKey = Allocator.CopyString(Key);
6249 }
6250 } Key(Allocator, PropName->getName());
6251
6252 // The uppercased name of the property name.
6253 std::string UpperKey = PropName->getName();
6254 if (!UpperKey.empty())
Jordan Rose4938f272013-02-09 10:09:43 +00006255 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006256
6257 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6258 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6259 Property->getType());
6260 bool ReturnTypeMatchesVoid
6261 = ReturnType.isNull() || ReturnType->isVoidType();
6262
6263 // Add the normal accessor -(type)key.
6264 if (IsInstanceMethod &&
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006265 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor669a25a2011-02-17 00:22:45 +00006266 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6267 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006268 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6269 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006270
6271 Builder.AddTypedTextChunk(Key);
6272 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6273 CXCursor_ObjCInstanceMethodDecl));
6274 }
6275
6276 // If we have an integral or boolean property (or the user has provided
6277 // an integral or boolean return type), add the accessor -(type)isKey.
6278 if (IsInstanceMethod &&
6279 ((!ReturnType.isNull() &&
6280 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6281 (ReturnType.isNull() &&
6282 (Property->getType()->isIntegerType() ||
6283 Property->getType()->isBooleanType())))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006284 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006285 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006286 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006287 if (ReturnType.isNull()) {
6288 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6289 Builder.AddTextChunk("BOOL");
6290 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6291 }
6292
6293 Builder.AddTypedTextChunk(
6294 Allocator.CopyString(SelectorId->getName()));
6295 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6296 CXCursor_ObjCInstanceMethodDecl));
6297 }
6298 }
6299
6300 // Add the normal mutator.
6301 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6302 !Property->getSetterMethodDecl()) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006303 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006304 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006305 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006306 if (ReturnType.isNull()) {
6307 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6308 Builder.AddTextChunk("void");
6309 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6310 }
6311
6312 Builder.AddTypedTextChunk(
6313 Allocator.CopyString(SelectorId->getName()));
6314 Builder.AddTypedTextChunk(":");
Douglas Gregor29979142012-04-10 18:35:07 +00006315 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6316 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006317 Builder.AddTextChunk(Key);
6318 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6319 CXCursor_ObjCInstanceMethodDecl));
6320 }
6321 }
6322
6323 // Indexed and unordered accessors
6324 unsigned IndexedGetterPriority = CCP_CodePattern;
6325 unsigned IndexedSetterPriority = CCP_CodePattern;
6326 unsigned UnorderedGetterPriority = CCP_CodePattern;
6327 unsigned UnorderedSetterPriority = CCP_CodePattern;
6328 if (const ObjCObjectPointerType *ObjCPointer
6329 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6330 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6331 // If this interface type is not provably derived from a known
6332 // collection, penalize the corresponding completions.
6333 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6334 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6335 if (!InheritsFromClassNamed(IFace, "NSArray"))
6336 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6337 }
6338
6339 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6340 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6341 if (!InheritsFromClassNamed(IFace, "NSSet"))
6342 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6343 }
6344 }
6345 } else {
6346 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6347 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6348 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6349 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6350 }
6351
6352 // Add -(NSUInteger)countOf<key>
6353 if (IsInstanceMethod &&
6354 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006355 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006356 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006357 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006358 if (ReturnType.isNull()) {
6359 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6360 Builder.AddTextChunk("NSUInteger");
6361 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6362 }
6363
6364 Builder.AddTypedTextChunk(
6365 Allocator.CopyString(SelectorId->getName()));
6366 Results.AddResult(Result(Builder.TakeString(),
6367 std::min(IndexedGetterPriority,
6368 UnorderedGetterPriority),
6369 CXCursor_ObjCInstanceMethodDecl));
6370 }
6371 }
6372
6373 // Indexed getters
6374 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6375 if (IsInstanceMethod &&
6376 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006377 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006378 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006379 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006380 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006381 if (ReturnType.isNull()) {
6382 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6383 Builder.AddTextChunk("id");
6384 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6385 }
6386
6387 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6388 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6389 Builder.AddTextChunk("NSUInteger");
6390 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6391 Builder.AddTextChunk("index");
6392 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6393 CXCursor_ObjCInstanceMethodDecl));
6394 }
6395 }
6396
6397 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6398 if (IsInstanceMethod &&
6399 (ReturnType.isNull() ||
6400 (ReturnType->isObjCObjectPointerType() &&
6401 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6402 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6403 ->getName() == "NSArray"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006404 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006405 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006406 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006407 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006408 if (ReturnType.isNull()) {
6409 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6410 Builder.AddTextChunk("NSArray *");
6411 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6412 }
6413
6414 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6415 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6416 Builder.AddTextChunk("NSIndexSet *");
6417 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6418 Builder.AddTextChunk("indexes");
6419 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6420 CXCursor_ObjCInstanceMethodDecl));
6421 }
6422 }
6423
6424 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6425 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006426 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006427 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006428 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006429 &Context.Idents.get("range")
6430 };
6431
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006432 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006433 if (ReturnType.isNull()) {
6434 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6435 Builder.AddTextChunk("void");
6436 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6437 }
6438
6439 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6440 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6441 Builder.AddPlaceholderChunk("object-type");
6442 Builder.AddTextChunk(" **");
6443 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6444 Builder.AddTextChunk("buffer");
6445 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6446 Builder.AddTypedTextChunk("range:");
6447 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6448 Builder.AddTextChunk("NSRange");
6449 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6450 Builder.AddTextChunk("inRange");
6451 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6452 CXCursor_ObjCInstanceMethodDecl));
6453 }
6454 }
6455
6456 // Mutable indexed accessors
6457
6458 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6459 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006460 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006461 IdentifierInfo *SelectorIds[2] = {
6462 &Context.Idents.get("insertObject"),
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006463 &Context.Idents.get(SelectorName)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006464 };
6465
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006466 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006467 if (ReturnType.isNull()) {
6468 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6469 Builder.AddTextChunk("void");
6470 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6471 }
6472
6473 Builder.AddTypedTextChunk("insertObject:");
6474 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6475 Builder.AddPlaceholderChunk("object-type");
6476 Builder.AddTextChunk(" *");
6477 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6478 Builder.AddTextChunk("object");
6479 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6480 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6481 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6482 Builder.AddPlaceholderChunk("NSUInteger");
6483 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6484 Builder.AddTextChunk("index");
6485 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6486 CXCursor_ObjCInstanceMethodDecl));
6487 }
6488 }
6489
6490 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6491 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006492 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006493 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006494 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006495 &Context.Idents.get("atIndexes")
6496 };
6497
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006498 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006499 if (ReturnType.isNull()) {
6500 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6501 Builder.AddTextChunk("void");
6502 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6503 }
6504
6505 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6506 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6507 Builder.AddTextChunk("NSArray *");
6508 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6509 Builder.AddTextChunk("array");
6510 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6511 Builder.AddTypedTextChunk("atIndexes:");
6512 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6513 Builder.AddPlaceholderChunk("NSIndexSet *");
6514 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6515 Builder.AddTextChunk("indexes");
6516 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6517 CXCursor_ObjCInstanceMethodDecl));
6518 }
6519 }
6520
6521 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6522 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006523 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006524 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006525 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006526 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006527 if (ReturnType.isNull()) {
6528 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6529 Builder.AddTextChunk("void");
6530 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6531 }
6532
6533 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6534 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6535 Builder.AddTextChunk("NSUInteger");
6536 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6537 Builder.AddTextChunk("index");
6538 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6539 CXCursor_ObjCInstanceMethodDecl));
6540 }
6541 }
6542
6543 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6544 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006545 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006546 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006547 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006548 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006549 if (ReturnType.isNull()) {
6550 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6551 Builder.AddTextChunk("void");
6552 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6553 }
6554
6555 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6556 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6557 Builder.AddTextChunk("NSIndexSet *");
6558 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6559 Builder.AddTextChunk("indexes");
6560 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6561 CXCursor_ObjCInstanceMethodDecl));
6562 }
6563 }
6564
6565 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6566 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006567 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006568 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006569 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006570 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006571 &Context.Idents.get("withObject")
6572 };
6573
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006574 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006575 if (ReturnType.isNull()) {
6576 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6577 Builder.AddTextChunk("void");
6578 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6579 }
6580
6581 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6582 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6583 Builder.AddPlaceholderChunk("NSUInteger");
6584 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6585 Builder.AddTextChunk("index");
6586 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6587 Builder.AddTypedTextChunk("withObject:");
6588 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6589 Builder.AddTextChunk("id");
6590 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6591 Builder.AddTextChunk("object");
6592 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6593 CXCursor_ObjCInstanceMethodDecl));
6594 }
6595 }
6596
6597 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6598 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006599 std::string SelectorName1
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006600 = (Twine("replace") + UpperKey + "AtIndexes").str();
6601 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006602 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006603 &Context.Idents.get(SelectorName1),
6604 &Context.Idents.get(SelectorName2)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006605 };
6606
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006607 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006608 if (ReturnType.isNull()) {
6609 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6610 Builder.AddTextChunk("void");
6611 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6612 }
6613
6614 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6615 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6616 Builder.AddPlaceholderChunk("NSIndexSet *");
6617 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6618 Builder.AddTextChunk("indexes");
6619 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6620 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6621 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6622 Builder.AddTextChunk("NSArray *");
6623 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6624 Builder.AddTextChunk("array");
6625 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6626 CXCursor_ObjCInstanceMethodDecl));
6627 }
6628 }
6629
6630 // Unordered getters
6631 // - (NSEnumerator *)enumeratorOfKey
6632 if (IsInstanceMethod &&
6633 (ReturnType.isNull() ||
6634 (ReturnType->isObjCObjectPointerType() &&
6635 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6636 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6637 ->getName() == "NSEnumerator"))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006638 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006639 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006640 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006641 if (ReturnType.isNull()) {
6642 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6643 Builder.AddTextChunk("NSEnumerator *");
6644 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6645 }
6646
6647 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6648 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6649 CXCursor_ObjCInstanceMethodDecl));
6650 }
6651 }
6652
6653 // - (type *)memberOfKey:(type *)object
6654 if (IsInstanceMethod &&
6655 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006656 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006657 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006658 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006659 if (ReturnType.isNull()) {
6660 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6661 Builder.AddPlaceholderChunk("object-type");
6662 Builder.AddTextChunk(" *");
6663 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6664 }
6665
6666 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6667 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6668 if (ReturnType.isNull()) {
6669 Builder.AddPlaceholderChunk("object-type");
6670 Builder.AddTextChunk(" *");
6671 } else {
6672 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006673 Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006674 Builder.getAllocator()));
6675 }
6676 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6677 Builder.AddTextChunk("object");
6678 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6679 CXCursor_ObjCInstanceMethodDecl));
6680 }
6681 }
6682
6683 // Mutable unordered accessors
6684 // - (void)addKeyObject:(type *)object
6685 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006686 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006687 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006688 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006689 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006690 if (ReturnType.isNull()) {
6691 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6692 Builder.AddTextChunk("void");
6693 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6694 }
6695
6696 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6697 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6698 Builder.AddPlaceholderChunk("object-type");
6699 Builder.AddTextChunk(" *");
6700 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6701 Builder.AddTextChunk("object");
6702 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6703 CXCursor_ObjCInstanceMethodDecl));
6704 }
6705 }
6706
6707 // - (void)addKey:(NSSet *)objects
6708 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006709 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006710 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006711 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006712 if (ReturnType.isNull()) {
6713 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6714 Builder.AddTextChunk("void");
6715 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6716 }
6717
6718 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6719 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6720 Builder.AddTextChunk("NSSet *");
6721 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6722 Builder.AddTextChunk("objects");
6723 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6724 CXCursor_ObjCInstanceMethodDecl));
6725 }
6726 }
6727
6728 // - (void)removeKeyObject:(type *)object
6729 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006730 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006731 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006732 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006733 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006734 if (ReturnType.isNull()) {
6735 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6736 Builder.AddTextChunk("void");
6737 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6738 }
6739
6740 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6741 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6742 Builder.AddPlaceholderChunk("object-type");
6743 Builder.AddTextChunk(" *");
6744 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6745 Builder.AddTextChunk("object");
6746 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6747 CXCursor_ObjCInstanceMethodDecl));
6748 }
6749 }
6750
6751 // - (void)removeKey:(NSSet *)objects
6752 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006753 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006754 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006755 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006756 if (ReturnType.isNull()) {
6757 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6758 Builder.AddTextChunk("void");
6759 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6760 }
6761
6762 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6763 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6764 Builder.AddTextChunk("NSSet *");
6765 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6766 Builder.AddTextChunk("objects");
6767 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6768 CXCursor_ObjCInstanceMethodDecl));
6769 }
6770 }
6771
6772 // - (void)intersectKey:(NSSet *)objects
6773 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006774 std::string SelectorName = (Twine("intersect") + UpperKey).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.AddTextChunk("NSSet *");
6786 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6787 Builder.AddTextChunk("objects");
6788 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6789 CXCursor_ObjCInstanceMethodDecl));
6790 }
6791 }
6792
6793 // Key-Value Observing
6794 // + (NSSet *)keyPathsForValuesAffectingKey
6795 if (!IsInstanceMethod &&
6796 (ReturnType.isNull() ||
6797 (ReturnType->isObjCObjectPointerType() &&
6798 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6799 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6800 ->getName() == "NSSet"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006801 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006802 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006803 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006804 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006805 if (ReturnType.isNull()) {
6806 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6807 Builder.AddTextChunk("NSSet *");
6808 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6809 }
6810
6811 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6812 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor857bcda2011-06-02 04:02:27 +00006813 CXCursor_ObjCClassMethodDecl));
6814 }
6815 }
6816
6817 // + (BOOL)automaticallyNotifiesObserversForKey
6818 if (!IsInstanceMethod &&
6819 (ReturnType.isNull() ||
6820 ReturnType->isIntegerType() ||
6821 ReturnType->isBooleanType())) {
6822 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006823 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor857bcda2011-06-02 04:02:27 +00006824 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6825 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6826 if (ReturnType.isNull()) {
6827 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6828 Builder.AddTextChunk("BOOL");
6829 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6830 }
6831
6832 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6833 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6834 CXCursor_ObjCClassMethodDecl));
Douglas Gregor669a25a2011-02-17 00:22:45 +00006835 }
6836 }
6837}
6838
Douglas Gregor636a61e2010-04-07 00:21:17 +00006839void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6840 bool IsInstanceMethod,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006841 ParsedType ReturnTy) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006842 // Determine the return type of the method we're declaring, if
6843 // provided.
6844 QualType ReturnType = GetTypeFromParser(ReturnTy);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006845 Decl *IDecl = 0;
6846 if (CurContext->isObjCContainer()) {
6847 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6848 IDecl = cast<Decl>(OCD);
6849 }
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006850 // Determine where we should start searching for methods.
6851 ObjCContainerDecl *SearchDecl = 0;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006852 bool IsInImplementation = false;
John McCall48871652010-08-21 09:40:31 +00006853 if (Decl *D = IDecl) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006854 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6855 SearchDecl = Impl->getClassInterface();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006856 IsInImplementation = true;
6857 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006858 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006859 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006860 IsInImplementation = true;
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006861 } else
Douglas Gregor636a61e2010-04-07 00:21:17 +00006862 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006863 }
6864
6865 if (!SearchDecl && S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00006866 if (DeclContext *DC = S->getEntity())
Douglas Gregor636a61e2010-04-07 00:21:17 +00006867 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006868 }
6869
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006870 if (!SearchDecl) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006871 HandleCodeCompleteResults(this, CodeCompleter,
6872 CodeCompletionContext::CCC_Other,
6873 0, 0);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006874 return;
6875 }
6876
6877 // Find all of the methods that we could declare/implement here.
6878 KnownMethodsMap KnownMethods;
6879 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006880 ReturnType, KnownMethods);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006881
Douglas Gregor636a61e2010-04-07 00:21:17 +00006882 // Add declarations or definitions for each of the known methods.
John McCall276321a2010-08-25 06:19:51 +00006883 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006884 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006885 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006886 CodeCompletionContext::CCC_Other);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006887 Results.EnterNewScope();
Douglas Gregor75acd922011-09-27 23:30:47 +00006888 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006889 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6890 MEnd = KnownMethods.end();
6891 M != MEnd; ++M) {
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006892 ObjCMethodDecl *Method = M->second.getPointer();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006893 CodeCompletionBuilder Builder(Results.getAllocator(),
6894 Results.getCodeCompletionTUInfo());
Douglas Gregor636a61e2010-04-07 00:21:17 +00006895
6896 // If the result type was not already provided, add it to the
6897 // pattern as (type).
Douglas Gregor669a25a2011-02-17 00:22:45 +00006898 if (ReturnType.isNull())
Alp Toker314cc812014-01-25 16:55:45 +00006899 AddObjCPassingTypeChunk(Method->getReturnType(),
6900 Method->getObjCDeclQualifier(), Context, Policy,
6901 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006902
6903 Selector Sel = Method->getSelector();
6904
6905 // Add the first part of the selector to the pattern.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006906 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006907 Sel.getNameForSlot(0)));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006908
6909 // Add parameters to the pattern.
6910 unsigned I = 0;
6911 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6912 PEnd = Method->param_end();
6913 P != PEnd; (void)++P, ++I) {
6914 // Add the part of the selector name.
6915 if (I == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006916 Builder.AddTypedTextChunk(":");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006917 else if (I < Sel.getNumArgs()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006918 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6919 Builder.AddTypedTextChunk(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006920 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006921 } else
6922 break;
6923
6924 // Add the parameter type.
Douglas Gregor29979142012-04-10 18:35:07 +00006925 AddObjCPassingTypeChunk((*P)->getOriginalType(),
6926 (*P)->getObjCDeclQualifier(),
6927 Context, Policy,
Douglas Gregor75acd922011-09-27 23:30:47 +00006928 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006929
6930 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006931 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006932 }
6933
6934 if (Method->isVariadic()) {
6935 if (Method->param_size() > 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006936 Builder.AddChunk(CodeCompletionString::CK_Comma);
6937 Builder.AddTextChunk("...");
Douglas Gregor400f5972010-08-31 05:13:43 +00006938 }
Douglas Gregor636a61e2010-04-07 00:21:17 +00006939
Douglas Gregord37c59d2010-05-28 00:57:46 +00006940 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006941 // We will be defining the method here, so add a compound statement.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006942 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6943 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6944 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Alp Toker314cc812014-01-25 16:55:45 +00006945 if (!Method->getReturnType()->isVoidType()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006946 // If the result type is not void, add a return clause.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006947 Builder.AddTextChunk("return");
6948 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6949 Builder.AddPlaceholderChunk("expression");
6950 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006951 } else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006952 Builder.AddPlaceholderChunk("statements");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006953
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006954 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6955 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006956 }
6957
Douglas Gregor416b5752010-08-25 01:08:01 +00006958 unsigned Priority = CCP_CodePattern;
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006959 if (!M->second.getInt())
Douglas Gregor416b5752010-08-25 01:08:01 +00006960 Priority += CCD_InBaseClass;
6961
Douglas Gregor78254c82012-03-27 23:34:16 +00006962 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006963 }
6964
Douglas Gregor669a25a2011-02-17 00:22:45 +00006965 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
6966 // the properties in this class and its categories.
David Blaikiebbafb8a2012-03-11 07:00:24 +00006967 if (Context.getLangOpts().ObjC2) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006968 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006969 Containers.push_back(SearchDecl);
6970
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006971 VisitedSelectorSet KnownSelectors;
6972 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6973 MEnd = KnownMethods.end();
6974 M != MEnd; ++M)
6975 KnownSelectors.insert(M->first);
6976
6977
Douglas Gregor669a25a2011-02-17 00:22:45 +00006978 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
6979 if (!IFace)
6980 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
6981 IFace = Category->getClassInterface();
6982
6983 if (IFace) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006984 for (ObjCInterfaceDecl::visible_categories_iterator
6985 Cat = IFace->visible_categories_begin(),
6986 CatEnd = IFace->visible_categories_end();
6987 Cat != CatEnd; ++Cat) {
6988 Containers.push_back(*Cat);
6989 }
Douglas Gregor669a25a2011-02-17 00:22:45 +00006990 }
6991
6992 for (unsigned I = 0, N = Containers.size(); I != N; ++I) {
6993 for (ObjCContainerDecl::prop_iterator P = Containers[I]->prop_begin(),
6994 PEnd = Containers[I]->prop_end();
6995 P != PEnd; ++P) {
David Blaikie40ed2972012-06-06 20:45:41 +00006996 AddObjCKeyValueCompletions(*P, IsInstanceMethod, ReturnType, Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006997 KnownSelectors, Results);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006998 }
6999 }
7000 }
7001
Douglas Gregor636a61e2010-04-07 00:21:17 +00007002 Results.ExitScope();
7003
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007004 HandleCodeCompleteResults(this, CodeCompleter,
7005 CodeCompletionContext::CCC_Other,
7006 Results.data(),Results.size());
Douglas Gregor636a61e2010-04-07 00:21:17 +00007007}
Douglas Gregor95887f92010-07-08 23:20:03 +00007008
7009void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
7010 bool IsInstanceMethod,
Douglas Gregor45879692010-07-08 23:37:41 +00007011 bool AtParameterName,
John McCallba7bf592010-08-24 05:47:05 +00007012 ParsedType ReturnTy,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007013 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor95887f92010-07-08 23:20:03 +00007014 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00007015 // pool from the AST file.
Douglas Gregor95887f92010-07-08 23:20:03 +00007016 if (ExternalSource) {
7017 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
7018 I != N; ++I) {
7019 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00007020 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor95887f92010-07-08 23:20:03 +00007021 continue;
Sebastian Redl75d8a322010-08-02 23:18:59 +00007022
7023 ReadMethodPool(Sel);
Douglas Gregor95887f92010-07-08 23:20:03 +00007024 }
7025 }
7026
7027 // Build the set of methods we can see.
John McCall276321a2010-08-25 06:19:51 +00007028 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007029 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007030 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007031 CodeCompletionContext::CCC_Other);
Douglas Gregor95887f92010-07-08 23:20:03 +00007032
7033 if (ReturnTy)
7034 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redl75d8a322010-08-02 23:18:59 +00007035
Douglas Gregor95887f92010-07-08 23:20:03 +00007036 Results.EnterNewScope();
Sebastian Redl75d8a322010-08-02 23:18:59 +00007037 for (GlobalMethodPool::iterator M = MethodPool.begin(),
7038 MEnd = MethodPool.end();
7039 M != MEnd; ++M) {
7040 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
7041 &M->second.second;
7042 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00007043 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007044 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor95887f92010-07-08 23:20:03 +00007045 continue;
7046
Douglas Gregor45879692010-07-08 23:37:41 +00007047 if (AtParameterName) {
7048 // Suggest parameter names we've seen before.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007049 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor45879692010-07-08 23:37:41 +00007050 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
7051 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
7052 if (Param->getIdentifier()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007053 CodeCompletionBuilder Builder(Results.getAllocator(),
7054 Results.getCodeCompletionTUInfo());
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007055 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007056 Param->getIdentifier()->getName()));
7057 Results.AddResult(Builder.TakeString());
Douglas Gregor45879692010-07-08 23:37:41 +00007058 }
7059 }
7060
7061 continue;
7062 }
7063
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00007064 Result R(MethList->Method, Results.getBasePriority(MethList->Method), 0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007065 R.StartParameter = SelIdents.size();
Douglas Gregor95887f92010-07-08 23:20:03 +00007066 R.AllParametersAreInformative = false;
7067 R.DeclaringEntity = true;
7068 Results.MaybeAddResult(R, CurContext);
7069 }
7070 }
7071
7072 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007073 HandleCodeCompleteResults(this, CodeCompleter,
7074 CodeCompletionContext::CCC_Other,
7075 Results.data(),Results.size());
Douglas Gregor95887f92010-07-08 23:20:03 +00007076}
Douglas Gregorb14904c2010-08-13 22:48:40 +00007077
Douglas Gregorec00a262010-08-24 22:20:20 +00007078void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007079 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007080 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007081 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007082 Results.EnterNewScope();
7083
7084 // #if <condition>
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007085 CodeCompletionBuilder Builder(Results.getAllocator(),
7086 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007087 Builder.AddTypedTextChunk("if");
7088 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7089 Builder.AddPlaceholderChunk("condition");
7090 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007091
7092 // #ifdef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007093 Builder.AddTypedTextChunk("ifdef");
7094 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7095 Builder.AddPlaceholderChunk("macro");
7096 Results.AddResult(Builder.TakeString());
7097
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007098 // #ifndef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007099 Builder.AddTypedTextChunk("ifndef");
7100 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7101 Builder.AddPlaceholderChunk("macro");
7102 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007103
7104 if (InConditional) {
7105 // #elif <condition>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007106 Builder.AddTypedTextChunk("elif");
7107 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7108 Builder.AddPlaceholderChunk("condition");
7109 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007110
7111 // #else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007112 Builder.AddTypedTextChunk("else");
7113 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007114
7115 // #endif
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007116 Builder.AddTypedTextChunk("endif");
7117 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007118 }
7119
7120 // #include "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007121 Builder.AddTypedTextChunk("include");
7122 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7123 Builder.AddTextChunk("\"");
7124 Builder.AddPlaceholderChunk("header");
7125 Builder.AddTextChunk("\"");
7126 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007127
7128 // #include <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007129 Builder.AddTypedTextChunk("include");
7130 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7131 Builder.AddTextChunk("<");
7132 Builder.AddPlaceholderChunk("header");
7133 Builder.AddTextChunk(">");
7134 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007135
7136 // #define <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007137 Builder.AddTypedTextChunk("define");
7138 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7139 Builder.AddPlaceholderChunk("macro");
7140 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007141
7142 // #define <macro>(<args>)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007143 Builder.AddTypedTextChunk("define");
7144 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7145 Builder.AddPlaceholderChunk("macro");
7146 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7147 Builder.AddPlaceholderChunk("args");
7148 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7149 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007150
7151 // #undef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007152 Builder.AddTypedTextChunk("undef");
7153 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7154 Builder.AddPlaceholderChunk("macro");
7155 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007156
7157 // #line <number>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007158 Builder.AddTypedTextChunk("line");
7159 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7160 Builder.AddPlaceholderChunk("number");
7161 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007162
7163 // #line <number> "filename"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007164 Builder.AddTypedTextChunk("line");
7165 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7166 Builder.AddPlaceholderChunk("number");
7167 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7168 Builder.AddTextChunk("\"");
7169 Builder.AddPlaceholderChunk("filename");
7170 Builder.AddTextChunk("\"");
7171 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007172
7173 // #error <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007174 Builder.AddTypedTextChunk("error");
7175 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7176 Builder.AddPlaceholderChunk("message");
7177 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007178
7179 // #pragma <arguments>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007180 Builder.AddTypedTextChunk("pragma");
7181 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7182 Builder.AddPlaceholderChunk("arguments");
7183 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007184
David Blaikiebbafb8a2012-03-11 07:00:24 +00007185 if (getLangOpts().ObjC1) {
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007186 // #import "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007187 Builder.AddTypedTextChunk("import");
7188 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7189 Builder.AddTextChunk("\"");
7190 Builder.AddPlaceholderChunk("header");
7191 Builder.AddTextChunk("\"");
7192 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007193
7194 // #import <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007195 Builder.AddTypedTextChunk("import");
7196 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7197 Builder.AddTextChunk("<");
7198 Builder.AddPlaceholderChunk("header");
7199 Builder.AddTextChunk(">");
7200 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007201 }
7202
7203 // #include_next "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007204 Builder.AddTypedTextChunk("include_next");
7205 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7206 Builder.AddTextChunk("\"");
7207 Builder.AddPlaceholderChunk("header");
7208 Builder.AddTextChunk("\"");
7209 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007210
7211 // #include_next <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007212 Builder.AddTypedTextChunk("include_next");
7213 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7214 Builder.AddTextChunk("<");
7215 Builder.AddPlaceholderChunk("header");
7216 Builder.AddTextChunk(">");
7217 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007218
7219 // #warning <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007220 Builder.AddTypedTextChunk("warning");
7221 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7222 Builder.AddPlaceholderChunk("message");
7223 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007224
7225 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7226 // completions for them. And __include_macros is a Clang-internal extension
7227 // that we don't want to encourage anyone to use.
7228
7229 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7230 Results.ExitScope();
7231
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007232 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0de55ce2010-08-25 18:41:16 +00007233 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007234 Results.data(), Results.size());
7235}
7236
7237void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorec00a262010-08-24 22:20:20 +00007238 CodeCompleteOrdinaryName(S,
John McCallfaf5fb42010-08-26 23:41:50 +00007239 S->getFnParent()? Sema::PCC_RecoveryInFunction
7240 : Sema::PCC_Namespace);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007241}
7242
Douglas Gregorec00a262010-08-24 22:20:20 +00007243void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007244 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007245 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007246 IsDefinition? CodeCompletionContext::CCC_MacroName
7247 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor12785102010-08-24 20:21:13 +00007248 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7249 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007250 CodeCompletionBuilder Builder(Results.getAllocator(),
7251 Results.getCodeCompletionTUInfo());
Douglas Gregor12785102010-08-24 20:21:13 +00007252 Results.EnterNewScope();
7253 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7254 MEnd = PP.macro_end();
7255 M != MEnd; ++M) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007256 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007257 M->first->getName()));
Argyrios Kyrtzidis5c8b1cd2012-09-27 00:24:09 +00007258 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7259 CCP_CodePattern,
7260 CXCursor_MacroDefinition));
Douglas Gregor12785102010-08-24 20:21:13 +00007261 }
7262 Results.ExitScope();
7263 } else if (IsDefinition) {
7264 // FIXME: Can we detect when the user just wrote an include guard above?
7265 }
7266
Douglas Gregor0ac41382010-09-23 23:01:17 +00007267 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor12785102010-08-24 20:21:13 +00007268 Results.data(), Results.size());
7269}
7270
Douglas Gregorec00a262010-08-24 22:20:20 +00007271void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007272 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007273 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007274 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorec00a262010-08-24 22:20:20 +00007275
7276 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007277 AddMacroResults(PP, Results, true);
Douglas Gregorec00a262010-08-24 22:20:20 +00007278
7279 // defined (<macro>)
7280 Results.EnterNewScope();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007281 CodeCompletionBuilder Builder(Results.getAllocator(),
7282 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007283 Builder.AddTypedTextChunk("defined");
7284 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7285 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7286 Builder.AddPlaceholderChunk("macro");
7287 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7288 Results.AddResult(Builder.TakeString());
Douglas Gregorec00a262010-08-24 22:20:20 +00007289 Results.ExitScope();
7290
7291 HandleCodeCompleteResults(this, CodeCompleter,
7292 CodeCompletionContext::CCC_PreprocessorExpression,
7293 Results.data(), Results.size());
7294}
7295
7296void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7297 IdentifierInfo *Macro,
7298 MacroInfo *MacroInfo,
7299 unsigned Argument) {
7300 // FIXME: In the future, we could provide "overload" results, much like we
7301 // do for function calls.
7302
Argyrios Kyrtzidis75f6cd22011-08-18 19:41:28 +00007303 // Now just ignore this. There will be another code-completion callback
7304 // for the expanded tokens.
Douglas Gregorec00a262010-08-24 22:20:20 +00007305}
7306
Douglas Gregor11583702010-08-25 17:04:25 +00007307void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor11583702010-08-25 17:04:25 +00007308 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorea736372010-08-25 17:10:00 +00007309 CodeCompletionContext::CCC_NaturalLanguage,
Douglas Gregor11583702010-08-25 17:04:25 +00007310 0, 0);
7311}
7312
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007313void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007314 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00007315 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007316 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7317 CodeCompletionContext::CCC_Recovery);
Douglas Gregor39982192010-08-15 06:18:01 +00007318 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7319 CodeCompletionDeclConsumer Consumer(Builder,
7320 Context.getTranslationUnitDecl());
7321 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7322 Consumer);
7323 }
Douglas Gregorb14904c2010-08-13 22:48:40 +00007324
7325 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007326 AddMacroResults(PP, Builder, true);
Douglas Gregorb14904c2010-08-13 22:48:40 +00007327
7328 Results.clear();
7329 Results.insert(Results.end(),
7330 Builder.data(), Builder.data() + Builder.size());
7331}