blob: f1da3b4ad17f62e130672600c706eeccefb4a0df [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) { }
Craig Toppere14c0f82014-03-12 04:55:44 +00001256
1257 void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
1258 bool InBaseClass) override {
Erik Verbruggen2e657ff2011-10-06 07:27:49 +00001259 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 Ballmanbe22bcb2014-03-10 17:08:28 +00002641 for (const auto *I : ND->specific_attrs<AnnotateAttr>())
2642 Result.AddAnnotation(Result.getAllocator().CopyString(I->getAnnotation()));
Aaron Ballmanb97112e2014-03-08 22:19:01 +00002643
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002644 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorb3fa9192009-12-18 18:53:37 +00002645
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002646 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002647 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002648 Ctx, Policy);
2649 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002650 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002651 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002652 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002653 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002654 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002655 }
2656
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002657 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002658 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002659 Ctx, Policy);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002660 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002661 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor0212fd72010-09-21 16:06:22 +00002662
Douglas Gregor3545ff42009-09-21 16:56:56 +00002663 // Figure out which template parameters are deduced (or have default
2664 // arguments).
Benjamin Kramere0513cb2012-01-30 16:17:39 +00002665 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002666 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002667 unsigned LastDeducibleArgument;
2668 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2669 --LastDeducibleArgument) {
2670 if (!Deduced[LastDeducibleArgument - 1]) {
2671 // C++0x: Figure out if the template argument has a default. If so,
2672 // the user doesn't need to type this argument.
2673 // FIXME: We need to abstract template parameters better!
2674 bool HasDefaultArg = false;
2675 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002676 LastDeducibleArgument - 1);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002677 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2678 HasDefaultArg = TTP->hasDefaultArgument();
2679 else if (NonTypeTemplateParmDecl *NTTP
2680 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2681 HasDefaultArg = NTTP->hasDefaultArgument();
2682 else {
2683 assert(isa<TemplateTemplateParmDecl>(Param));
2684 HasDefaultArg
Douglas Gregor9eb77012009-11-07 00:00:49 +00002685 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002686 }
2687
2688 if (!HasDefaultArg)
2689 break;
2690 }
2691 }
2692
2693 if (LastDeducibleArgument) {
2694 // Some of the function template arguments cannot be deduced from a
2695 // function call, so we introduce an explicit template argument list
2696 // containing all of the arguments up to the first deducible argument.
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002697 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002698 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor3545ff42009-09-21 16:56:56 +00002699 LastDeducibleArgument);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002700 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor3545ff42009-09-21 16:56:56 +00002701 }
2702
2703 // Add the function parameters
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002704 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002705 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002706 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor0f622362009-12-11 18:44:16 +00002707 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002708 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002709 }
2710
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002711 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor5bf52692009-09-22 23:15:58 +00002712 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002713 Ctx, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002714 Result.AddTypedTextChunk(
2715 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002716 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002717 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002718 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002719 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002720 }
2721
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002722 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregord3c5d792009-11-17 16:44:22 +00002723 Selector Sel = Method->getSelector();
2724 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002725 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002726 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002727 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002728 }
2729
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00002730 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregor1b605f72009-11-19 01:08:35 +00002731 SelName += ':';
2732 if (StartParameter == 0)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002733 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002734 else {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002735 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregor1b605f72009-11-19 01:08:35 +00002736
2737 // If there is only one parameter, and we're past it, add an empty
2738 // typed-text chunk since there is nothing to type.
2739 if (Method->param_size() == 1)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002740 Result.AddTypedTextChunk("");
Douglas Gregor1b605f72009-11-19 01:08:35 +00002741 }
Douglas Gregord3c5d792009-11-17 16:44:22 +00002742 unsigned Idx = 0;
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00002743 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2744 PEnd = Method->param_end();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002745 P != PEnd; (void)++P, ++Idx) {
2746 if (Idx > 0) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00002747 std::string Keyword;
2748 if (Idx > StartParameter)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002749 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregord3c5d792009-11-17 16:44:22 +00002750 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramer632500c2011-07-26 16:59:25 +00002751 Keyword += II->getName();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002752 Keyword += ":";
Douglas Gregor95887f92010-07-08 23:20:03 +00002753 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002754 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002755 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002756 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002757 }
Douglas Gregor1b605f72009-11-19 01:08:35 +00002758
2759 // If we're before the starting parameter, skip the placeholder.
2760 if (Idx < StartParameter)
2761 continue;
Douglas Gregord3c5d792009-11-17 16:44:22 +00002762
2763 std::string Arg;
Douglas Gregore90dd002010-08-24 16:15:59 +00002764
2765 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002766 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregore90dd002010-08-24 16:15:59 +00002767 else {
John McCall31168b02011-06-15 23:02:42 +00002768 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor8f08d742011-07-30 07:55:26 +00002769 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2770 + Arg + ")";
Douglas Gregore90dd002010-08-24 16:15:59 +00002771 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregor981a0c42010-08-29 19:47:46 +00002772 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramer632500c2011-07-26 16:59:25 +00002773 Arg += II->getName();
Douglas Gregore90dd002010-08-24 16:15:59 +00002774 }
2775
Douglas Gregor400f5972010-08-31 05:13:43 +00002776 if (Method->isVariadic() && (P + 1) == PEnd)
2777 Arg += ", ...";
2778
Douglas Gregor95887f92010-07-08 23:20:03 +00002779 if (DeclaringEntity)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002780 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor95887f92010-07-08 23:20:03 +00002781 else if (AllParametersAreInformative)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002782 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8537c52009-11-19 07:41:15 +00002783 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002784 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregord3c5d792009-11-17 16:44:22 +00002785 }
2786
Douglas Gregor04c5f972009-12-23 00:21:46 +00002787 if (Method->isVariadic()) {
Douglas Gregor400f5972010-08-31 05:13:43 +00002788 if (Method->param_size() == 0) {
2789 if (DeclaringEntity)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002790 Result.AddTextChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002791 else if (AllParametersAreInformative)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002792 Result.AddInformativeChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002793 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002794 Result.AddPlaceholderChunk(", ...");
Douglas Gregor400f5972010-08-31 05:13:43 +00002795 }
Douglas Gregordbb71db2010-08-23 23:51:41 +00002796
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002797 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor04c5f972009-12-23 00:21:46 +00002798 }
2799
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002800 return Result.TakeString();
Douglas Gregord3c5d792009-11-17 16:44:22 +00002801 }
2802
Douglas Gregorf09935f2009-12-01 05:55:20 +00002803 if (Qualifier)
Douglas Gregor5bf52692009-09-22 23:15:58 +00002804 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidis8d05ca72012-01-17 02:15:51 +00002805 Ctx, Policy);
Douglas Gregorf09935f2009-12-01 05:55:20 +00002806
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002807 Result.AddTypedTextChunk(
2808 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002809 return Result.TakeString();
Douglas Gregor3545ff42009-09-21 16:56:56 +00002810}
2811
Douglas Gregorf0f51982009-09-23 00:34:09 +00002812CodeCompletionString *
2813CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2814 unsigned CurrentArg,
Douglas Gregor36e3b5c2010-10-11 21:37:58 +00002815 Sema &S,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002816 CodeCompletionAllocator &Allocator,
2817 CodeCompletionTUInfo &CCTUInfo) const {
Douglas Gregor75acd922011-09-27 23:30:47 +00002818 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCall31168b02011-06-15 23:02:42 +00002819
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002820 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00002821 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002822 FunctionDecl *FDecl = getFunction();
Douglas Gregor75acd922011-09-27 23:30:47 +00002823 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002824 const FunctionProtoType *Proto
2825 = dyn_cast<FunctionProtoType>(getFunctionType());
2826 if (!FDecl && !Proto) {
2827 // Function without a prototype. Just give the return type and a
2828 // highlighted ellipsis.
2829 const FunctionType *FT = getFunctionType();
Alp Toker314cc812014-01-25 16:55:45 +00002830 Result.AddTextChunk(GetCompletionTypeString(FT->getReturnType(), S.Context,
2831 Policy, Result.getAllocator()));
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002832 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2833 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2834 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002835 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002836 }
2837
2838 if (FDecl)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002839 Result.AddTextChunk(
2840 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002841 else
Alp Toker314cc812014-01-25 16:55:45 +00002842 Result.AddTextChunk(Result.getAllocator().CopyString(
2843 Proto->getReturnType().getAsString(Policy)));
2844
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002845 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Alp Toker9cacbab2014-01-20 20:26:09 +00002846 unsigned NumParams = FDecl ? FDecl->getNumParams() : Proto->getNumParams();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002847 for (unsigned I = 0; I != NumParams; ++I) {
2848 if (I)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002849 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002850
2851 std::string ArgString;
2852 QualType ArgType;
2853
2854 if (FDecl) {
2855 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2856 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2857 } else {
Alp Toker9cacbab2014-01-20 20:26:09 +00002858 ArgType = Proto->getParamType(I);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002859 }
2860
John McCall31168b02011-06-15 23:02:42 +00002861 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002862
2863 if (I == CurrentArg)
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002864 Result.AddChunk(CodeCompletionString::CK_CurrentParameter,
2865 Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002866 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00002867 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregorf0f51982009-09-23 00:34:09 +00002868 }
2869
2870 if (Proto && Proto->isVariadic()) {
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002871 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002872 if (CurrentArg < NumParams)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002873 Result.AddTextChunk("...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002874 else
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002875 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
Douglas Gregorf0f51982009-09-23 00:34:09 +00002876 }
Benjamin Kramerdb534a42012-03-26 16:57:36 +00002877 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf0f51982009-09-23 00:34:09 +00002878
Douglas Gregorb278aaf2011-02-01 19:23:04 +00002879 return Result.TakeString();
Douglas Gregorf0f51982009-09-23 00:34:09 +00002880}
2881
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002882unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002883 const LangOptions &LangOpts,
Douglas Gregor6e240332010-08-16 16:18:59 +00002884 bool PreferredTypeIsPointer) {
2885 unsigned Priority = CCP_Macro;
2886
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002887 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2888 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2889 MacroName.equals("Nil")) {
Douglas Gregor6e240332010-08-16 16:18:59 +00002890 Priority = CCP_Constant;
2891 if (PreferredTypeIsPointer)
2892 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregor9dcf58a2010-09-20 21:11:48 +00002893 }
2894 // Treat "YES", "NO", "true", and "false" as constants.
2895 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2896 MacroName.equals("true") || MacroName.equals("false"))
2897 Priority = CCP_Constant;
2898 // Treat "bool" as a type.
2899 else if (MacroName.equals("bool"))
2900 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2901
Douglas Gregor6e240332010-08-16 16:18:59 +00002902
2903 return Priority;
2904}
2905
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002906CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002907 if (!D)
2908 return CXCursor_UnexposedDecl;
2909
2910 switch (D->getKind()) {
2911 case Decl::Enum: return CXCursor_EnumDecl;
2912 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2913 case Decl::Field: return CXCursor_FieldDecl;
2914 case Decl::Function:
2915 return CXCursor_FunctionDecl;
2916 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2917 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002918 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregordeafd0b2011-12-27 22:43:10 +00002919
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002920 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002921 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2922 case Decl::ObjCMethod:
2923 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2924 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2925 case Decl::CXXMethod: return CXCursor_CXXMethod;
2926 case Decl::CXXConstructor: return CXCursor_Constructor;
2927 case Decl::CXXDestructor: return CXCursor_Destructor;
2928 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2929 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidis3698cef2012-01-24 21:39:26 +00002930 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002931 case Decl::ParmVar: return CXCursor_ParmDecl;
2932 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smithdda56e42011-04-15 14:24:37 +00002933 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002934 case Decl::Var: return CXCursor_VarDecl;
2935 case Decl::Namespace: return CXCursor_Namespace;
2936 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2937 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2938 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2939 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2940 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2941 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis12afd702011-09-30 17:58:23 +00002942 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002943 case Decl::ClassTemplatePartialSpecialization:
2944 return CXCursor_ClassTemplatePartialSpecialization;
2945 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor3e653b32012-04-30 23:41:16 +00002946 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002947
2948 case Decl::Using:
2949 case Decl::UnresolvedUsingValue:
2950 case Decl::UnresolvedUsingTypename:
2951 return CXCursor_UsingDeclaration;
2952
Douglas Gregor4cd65962011-06-03 23:08:58 +00002953 case Decl::ObjCPropertyImpl:
2954 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2955 case ObjCPropertyImplDecl::Dynamic:
2956 return CXCursor_ObjCDynamicDecl;
2957
2958 case ObjCPropertyImplDecl::Synthesize:
2959 return CXCursor_ObjCSynthesizeDecl;
2960 }
Argyrios Kyrtzidis50e5b1d2012-10-05 00:22:24 +00002961
2962 case Decl::Import:
2963 return CXCursor_ModuleImportDecl;
Douglas Gregor4cd65962011-06-03 23:08:58 +00002964
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002965 default:
Dmitri Gribenkobdc80de2013-01-11 20:32:41 +00002966 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002967 switch (TD->getTagKind()) {
Joao Matosdc86f942012-08-31 18:45:21 +00002968 case TTK_Interface: // fall through
Douglas Gregor09c0eb12010-09-03 23:30:36 +00002969 case TTK_Struct: return CXCursor_StructDecl;
2970 case TTK_Class: return CXCursor_ClassDecl;
2971 case TTK_Union: return CXCursor_UnionDecl;
2972 case TTK_Enum: return CXCursor_EnumDecl;
2973 }
2974 }
2975 }
2976
2977 return CXCursor_UnexposedDecl;
2978}
2979
Douglas Gregor55b037b2010-07-08 20:55:51 +00002980static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor8cb17462012-10-09 16:01:50 +00002981 bool IncludeUndefined,
Douglas Gregor55b037b2010-07-08 20:55:51 +00002982 bool TargetTypeIsPointer = false) {
John McCall276321a2010-08-25 06:19:51 +00002983 typedef CodeCompletionResult Result;
Douglas Gregor55b037b2010-07-08 20:55:51 +00002984
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002985 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002986
Douglas Gregor9eb77012009-11-07 00:00:49 +00002987 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2988 MEnd = PP.macro_end();
Douglas Gregor55b037b2010-07-08 20:55:51 +00002989 M != MEnd; ++M) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00002990 if (IncludeUndefined || M->first->hasMacroDefinition())
2991 Results.AddResult(Result(M->first,
Douglas Gregor6e240332010-08-16 16:18:59 +00002992 getMacroUsagePriority(M->first->getName(),
David Blaikiebbafb8a2012-03-11 07:00:24 +00002993 PP.getLangOpts(),
Douglas Gregor6e240332010-08-16 16:18:59 +00002994 TargetTypeIsPointer)));
Douglas Gregor55b037b2010-07-08 20:55:51 +00002995 }
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002996
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002997 Results.ExitScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00002998
Douglas Gregorf329c7c2009-10-30 16:50:04 +00002999}
3000
Douglas Gregorce0e8562010-08-23 21:54:33 +00003001static void AddPrettyFunctionResults(const LangOptions &LangOpts,
3002 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003003 typedef CodeCompletionResult Result;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003004
3005 Results.EnterNewScope();
Douglas Gregor8e3e8742010-10-18 21:05:04 +00003006
Douglas Gregorce0e8562010-08-23 21:54:33 +00003007 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
3008 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith2bf7fdb2013-01-02 11:42:31 +00003009 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregorce0e8562010-08-23 21:54:33 +00003010 Results.AddResult(Result("__func__", CCP_Constant));
3011 Results.ExitScope();
3012}
3013
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003014static void HandleCodeCompleteResults(Sema *S,
3015 CodeCompleteConsumer *CodeCompleter,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003016 CodeCompletionContext Context,
John McCall276321a2010-08-25 06:19:51 +00003017 CodeCompletionResult *Results,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003018 unsigned NumResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00003019 if (CodeCompleter)
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003020 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor3545ff42009-09-21 16:56:56 +00003021}
3022
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003023static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3024 Sema::ParserCompletionContext PCC) {
3025 switch (PCC) {
John McCallfaf5fb42010-08-26 23:41:50 +00003026 case Sema::PCC_Namespace:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003027 return CodeCompletionContext::CCC_TopLevel;
3028
John McCallfaf5fb42010-08-26 23:41:50 +00003029 case Sema::PCC_Class:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003030 return CodeCompletionContext::CCC_ClassStructUnion;
3031
John McCallfaf5fb42010-08-26 23:41:50 +00003032 case Sema::PCC_ObjCInterface:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003033 return CodeCompletionContext::CCC_ObjCInterface;
3034
John McCallfaf5fb42010-08-26 23:41:50 +00003035 case Sema::PCC_ObjCImplementation:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003036 return CodeCompletionContext::CCC_ObjCImplementation;
3037
John McCallfaf5fb42010-08-26 23:41:50 +00003038 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003039 return CodeCompletionContext::CCC_ObjCIvarList;
3040
John McCallfaf5fb42010-08-26 23:41:50 +00003041 case Sema::PCC_Template:
3042 case Sema::PCC_MemberTemplate:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003043 if (S.CurContext->isFileContext())
3044 return CodeCompletionContext::CCC_TopLevel;
David Blaikie8a40f702012-01-17 06:56:22 +00003045 if (S.CurContext->isRecord())
Douglas Gregor0ac41382010-09-23 23:01:17 +00003046 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie8a40f702012-01-17 06:56:22 +00003047 return CodeCompletionContext::CCC_Other;
Douglas Gregor0ac41382010-09-23 23:01:17 +00003048
John McCallfaf5fb42010-08-26 23:41:50 +00003049 case Sema::PCC_RecoveryInFunction:
Douglas Gregor0ac41382010-09-23 23:01:17 +00003050 return CodeCompletionContext::CCC_Recovery;
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003051
John McCallfaf5fb42010-08-26 23:41:50 +00003052 case Sema::PCC_ForInit:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003053 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3054 S.getLangOpts().ObjC1)
Douglas Gregorc769d6e2010-10-18 22:01:46 +00003055 return CodeCompletionContext::CCC_ParenthesizedExpression;
3056 else
3057 return CodeCompletionContext::CCC_Expression;
3058
3059 case Sema::PCC_Expression:
John McCallfaf5fb42010-08-26 23:41:50 +00003060 case Sema::PCC_Condition:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003061 return CodeCompletionContext::CCC_Expression;
3062
John McCallfaf5fb42010-08-26 23:41:50 +00003063 case Sema::PCC_Statement:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003064 return CodeCompletionContext::CCC_Statement;
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003065
John McCallfaf5fb42010-08-26 23:41:50 +00003066 case Sema::PCC_Type:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003067 return CodeCompletionContext::CCC_Type;
Douglas Gregor5e35d592010-09-14 23:59:36 +00003068
3069 case Sema::PCC_ParenthesizedExpression:
3070 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor80039242011-02-15 20:33:25 +00003071
3072 case Sema::PCC_LocalDeclarationSpecifiers:
3073 return CodeCompletionContext::CCC_Type;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003074 }
David Blaikie8a40f702012-01-17 06:56:22 +00003075
3076 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003077}
3078
Douglas Gregorac322ec2010-08-27 21:18:54 +00003079/// \brief If we're in a C++ virtual member function, add completion results
3080/// that invoke the functions we override, since it's common to invoke the
3081/// overridden function as well as adding new functionality.
3082///
3083/// \param S The semantic analysis object for which we are generating results.
3084///
3085/// \param InContext This context in which the nested-name-specifier preceding
3086/// the code-completion point
3087static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3088 ResultBuilder &Results) {
3089 // Look through blocks.
3090 DeclContext *CurContext = S.CurContext;
3091 while (isa<BlockDecl>(CurContext))
3092 CurContext = CurContext->getParent();
3093
3094
3095 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3096 if (!Method || !Method->isVirtual())
3097 return;
3098
3099 // We need to have names for all of the parameters, if we're going to
3100 // generate a forwarding call.
Aaron Ballman43b68be2014-03-07 17:50:17 +00003101 for (auto P : Method->params())
3102 if (!P->getDeclName())
Douglas Gregorac322ec2010-08-27 21:18:54 +00003103 return;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003104
Douglas Gregor75acd922011-09-27 23:30:47 +00003105 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003106 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3107 MEnd = Method->end_overridden_methods();
3108 M != MEnd; ++M) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003109 CodeCompletionBuilder Builder(Results.getAllocator(),
3110 Results.getCodeCompletionTUInfo());
Dmitri Gribenko6cfb1532013-02-14 13:53:30 +00003111 const CXXMethodDecl *Overridden = *M;
Douglas Gregorac322ec2010-08-27 21:18:54 +00003112 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3113 continue;
3114
3115 // If we need a nested-name-specifier, add one now.
3116 if (!InContext) {
3117 NestedNameSpecifier *NNS
3118 = getRequiredQualification(S.Context, CurContext,
3119 Overridden->getDeclContext());
3120 if (NNS) {
3121 std::string Str;
3122 llvm::raw_string_ostream OS(Str);
Douglas Gregor75acd922011-09-27 23:30:47 +00003123 NNS->print(OS, Policy);
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003124 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003125 }
3126 } else if (!InContext->Equals(Overridden->getDeclContext()))
3127 continue;
3128
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00003129 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003130 Overridden->getNameAsString()));
3131 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003132 bool FirstParam = true;
Aaron Ballman43b68be2014-03-07 17:50:17 +00003133 for (auto P : Method->params()) {
Douglas Gregorac322ec2010-08-27 21:18:54 +00003134 if (FirstParam)
3135 FirstParam = false;
3136 else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003137 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003138
Aaron Ballman43b68be2014-03-07 17:50:17 +00003139 Builder.AddPlaceholderChunk(
3140 Results.getAllocator().CopyString(P->getIdentifier()->getName()));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003141 }
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003142 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3143 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorac322ec2010-08-27 21:18:54 +00003144 CCP_SuperCompletion,
Douglas Gregor78254c82012-03-27 23:34:16 +00003145 CXCursor_CXXMethod,
3146 CXAvailability_Available,
3147 Overridden));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003148 Results.Ignore(Overridden);
3149 }
3150}
3151
Douglas Gregor07f43572012-01-29 18:15:03 +00003152void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3153 ModuleIdPath Path) {
3154 typedef CodeCompletionResult Result;
3155 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003156 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor07f43572012-01-29 18:15:03 +00003157 CodeCompletionContext::CCC_Other);
3158 Results.EnterNewScope();
3159
3160 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003161 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor07f43572012-01-29 18:15:03 +00003162 typedef CodeCompletionResult Result;
3163 if (Path.empty()) {
3164 // Enumerate all top-level modules.
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003165 SmallVector<Module *, 8> Modules;
Douglas Gregor07f43572012-01-29 18:15:03 +00003166 PP.getHeaderSearchInfo().collectAllModules(Modules);
3167 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3168 Builder.AddTypedTextChunk(
3169 Builder.getAllocator().CopyString(Modules[I]->Name));
3170 Results.AddResult(Result(Builder.TakeString(),
3171 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003172 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003173 Modules[I]->isAvailable()
3174 ? CXAvailability_Available
3175 : CXAvailability_NotAvailable));
3176 }
Daniel Jasper07e6c402013-08-05 20:26:17 +00003177 } else if (getLangOpts().Modules) {
Douglas Gregor07f43572012-01-29 18:15:03 +00003178 // Load the named module.
3179 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3180 Module::AllVisible,
3181 /*IsInclusionDirective=*/false);
3182 // Enumerate submodules.
3183 if (Mod) {
3184 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3185 SubEnd = Mod->submodule_end();
3186 Sub != SubEnd; ++Sub) {
3187
3188 Builder.AddTypedTextChunk(
3189 Builder.getAllocator().CopyString((*Sub)->Name));
3190 Results.AddResult(Result(Builder.TakeString(),
3191 CCP_Declaration,
Argyrios Kyrtzidis345d05f2013-05-29 18:50:15 +00003192 CXCursor_ModuleImportDecl,
Douglas Gregor07f43572012-01-29 18:15:03 +00003193 (*Sub)->isAvailable()
3194 ? CXAvailability_Available
3195 : CXAvailability_NotAvailable));
3196 }
3197 }
3198 }
3199 Results.ExitScope();
3200 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3201 Results.data(),Results.size());
3202}
3203
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003204void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003205 ParserCompletionContext CompletionContext) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003206 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003207 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003208 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorac322ec2010-08-27 21:18:54 +00003209 Results.EnterNewScope();
Douglas Gregor50832e02010-09-20 22:39:41 +00003210
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003211 // Determine how to filter results, e.g., so that the names of
3212 // values (functions, enumerators, function templates, etc.) are
3213 // only allowed where we can have an expression.
3214 switch (CompletionContext) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003215 case PCC_Namespace:
3216 case PCC_Class:
3217 case PCC_ObjCInterface:
3218 case PCC_ObjCImplementation:
3219 case PCC_ObjCInstanceVariableList:
3220 case PCC_Template:
3221 case PCC_MemberTemplate:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003222 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003223 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003224 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3225 break;
3226
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003227 case PCC_Statement:
Douglas Gregor5e35d592010-09-14 23:59:36 +00003228 case PCC_ParenthesizedExpression:
Douglas Gregor4d755e82010-08-24 23:58:17 +00003229 case PCC_Expression:
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003230 case PCC_ForInit:
3231 case PCC_Condition:
David Blaikiebbafb8a2012-03-11 07:00:24 +00003232 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor70febae2010-05-28 00:49:12 +00003233 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3234 else
3235 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorac322ec2010-08-27 21:18:54 +00003236
David Blaikiebbafb8a2012-03-11 07:00:24 +00003237 if (getLangOpts().CPlusPlus)
Douglas Gregorac322ec2010-08-27 21:18:54 +00003238 MaybeAddOverrideCalls(*this, /*InContext=*/0, Results);
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003239 break;
Douglas Gregor6da3db42010-05-25 05:58:43 +00003240
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003241 case PCC_RecoveryInFunction:
Douglas Gregor6da3db42010-05-25 05:58:43 +00003242 // Unfiltered
3243 break;
Douglas Gregor504a6ae2010-01-10 23:08:15 +00003244 }
3245
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003246 // If we are in a C++ non-static member function, check the qualifiers on
3247 // the member function to filter/prioritize the results list.
3248 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3249 if (CurMethod->isInstance())
3250 Results.setObjectTypeQualifiers(
3251 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3252
Douglas Gregorc580c522010-01-14 01:09:38 +00003253 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003254 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3255 CodeCompleter->includeGlobals());
Douglas Gregor92253692009-12-07 09:54:55 +00003256
Douglas Gregorf98e6a22010-01-13 23:51:12 +00003257 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor92253692009-12-07 09:54:55 +00003258 Results.ExitScope();
3259
Douglas Gregorce0e8562010-08-23 21:54:33 +00003260 switch (CompletionContext) {
Douglas Gregor5e35d592010-09-14 23:59:36 +00003261 case PCC_ParenthesizedExpression:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003262 case PCC_Expression:
3263 case PCC_Statement:
3264 case PCC_RecoveryInFunction:
3265 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00003266 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003267 break;
3268
3269 case PCC_Namespace:
3270 case PCC_Class:
3271 case PCC_ObjCInterface:
3272 case PCC_ObjCImplementation:
3273 case PCC_ObjCInstanceVariableList:
3274 case PCC_Template:
3275 case PCC_MemberTemplate:
3276 case PCC_ForInit:
3277 case PCC_Condition:
3278 case PCC_Type:
Douglas Gregor80039242011-02-15 20:33:25 +00003279 case PCC_LocalDeclarationSpecifiers:
Douglas Gregorf02e5f32010-08-24 01:11:00 +00003280 break;
Douglas Gregorce0e8562010-08-23 21:54:33 +00003281 }
3282
Douglas Gregor9eb77012009-11-07 00:00:49 +00003283 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003284 AddMacroResults(PP, Results, false);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003285
Douglas Gregor50832e02010-09-20 22:39:41 +00003286 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003287 Results.data(),Results.size());
Douglas Gregor9d64c5e2009-09-21 20:51:25 +00003288}
3289
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003290static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3291 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003292 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00003293 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003294 bool IsSuper,
3295 ResultBuilder &Results);
3296
3297void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3298 bool AllowNonIdentifiers,
3299 bool AllowNestedNameSpecifiers) {
John McCall276321a2010-08-25 06:19:51 +00003300 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003301 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003302 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00003303 AllowNestedNameSpecifiers
3304 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3305 : CodeCompletionContext::CCC_Name);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003306 Results.EnterNewScope();
3307
3308 // Type qualifiers can come after names.
3309 Results.AddResult(Result("const"));
3310 Results.AddResult(Result("volatile"));
David Blaikiebbafb8a2012-03-11 07:00:24 +00003311 if (getLangOpts().C99)
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003312 Results.AddResult(Result("restrict"));
3313
David Blaikiebbafb8a2012-03-11 07:00:24 +00003314 if (getLangOpts().CPlusPlus) {
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003315 if (AllowNonIdentifiers) {
3316 Results.AddResult(Result("operator"));
3317 }
3318
3319 // Add nested-name-specifiers.
3320 if (AllowNestedNameSpecifiers) {
3321 Results.allowNestedNameSpecifiers();
Douglas Gregor0ac41382010-09-23 23:01:17 +00003322 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003323 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3324 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3325 CodeCompleter->includeGlobals());
Douglas Gregor0ac41382010-09-23 23:01:17 +00003326 Results.setFilter(0);
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003327 }
3328 }
3329 Results.ExitScope();
3330
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003331 // If we're in a context where we might have an expression (rather than a
3332 // declaration), and what we've seen so far is an Objective-C type that could
3333 // be a receiver of a class message, this may be a class message send with
3334 // the initial opening bracket '[' missing. Add appropriate completions.
3335 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003336 DS.getParsedSpecifiers() == DeclSpec::PQ_TypeSpecifier &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003337 DS.getTypeSpecType() == DeclSpec::TST_typename &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003338 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3339 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
Richard Smithb4a9e862013-04-12 22:46:28 +00003340 !DS.isTypeAltiVecVector() &&
3341 S &&
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003342 (S->getFlags() & Scope::DeclScope) != 0 &&
3343 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3344 Scope::FunctionPrototypeScope |
3345 Scope::AtCatchScope)) == 0) {
3346 ParsedType T = DS.getRepAsType();
3347 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003348 AddClassMessageCompletions(*this, S, T, None, false, false, Results);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00003349 }
3350
Douglas Gregor56ccce02010-08-24 04:59:56 +00003351 // Note that we intentionally suppress macro results here, since we do not
3352 // encourage using macros to produce the names of entities.
3353
Douglas Gregor0ac41382010-09-23 23:01:17 +00003354 HandleCodeCompleteResults(this, CodeCompleter,
3355 Results.getCompletionContext(),
Douglas Gregorc49f5b22010-08-23 18:23:48 +00003356 Results.data(), Results.size());
3357}
3358
Douglas Gregor68762e72010-08-23 21:17:50 +00003359struct Sema::CodeCompleteExpressionData {
3360 CodeCompleteExpressionData(QualType PreferredType = QualType())
3361 : PreferredType(PreferredType), IntegralConstantExpression(false),
3362 ObjCCollection(false) { }
3363
3364 QualType PreferredType;
3365 bool IntegralConstantExpression;
3366 bool ObjCCollection;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003367 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregor68762e72010-08-23 21:17:50 +00003368};
3369
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003370/// \brief Perform code-completion in an expression context when we know what
3371/// type we're looking for.
Douglas Gregor68762e72010-08-23 21:17:50 +00003372void Sema::CodeCompleteExpression(Scope *S,
3373 const CodeCompleteExpressionData &Data) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003374 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003375 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003376 CodeCompletionContext::CCC_Expression);
Douglas Gregor68762e72010-08-23 21:17:50 +00003377 if (Data.ObjCCollection)
3378 Results.setFilter(&ResultBuilder::IsObjCCollection);
3379 else if (Data.IntegralConstantExpression)
Douglas Gregor85b50632010-07-28 21:50:18 +00003380 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003381 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003382 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3383 else
3384 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregor68762e72010-08-23 21:17:50 +00003385
3386 if (!Data.PreferredType.isNull())
3387 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3388
3389 // Ignore any declarations that we were told that we don't care about.
3390 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3391 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003392
3393 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003394 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3395 CodeCompleter->includeGlobals());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003396
3397 Results.EnterNewScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003398 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003399 Results.ExitScope();
3400
Douglas Gregor55b037b2010-07-08 20:55:51 +00003401 bool PreferredTypeIsPointer = false;
Douglas Gregor68762e72010-08-23 21:17:50 +00003402 if (!Data.PreferredType.isNull())
3403 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3404 || Data.PreferredType->isMemberPointerType()
3405 || Data.PreferredType->isBlockPointerType();
Douglas Gregor55b037b2010-07-08 20:55:51 +00003406
Douglas Gregorce0e8562010-08-23 21:54:33 +00003407 if (S->getFnParent() &&
3408 !Data.ObjCCollection &&
3409 !Data.IntegralConstantExpression)
David Blaikiebbafb8a2012-03-11 07:00:24 +00003410 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregorce0e8562010-08-23 21:54:33 +00003411
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003412 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00003413 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003414 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor68762e72010-08-23 21:17:50 +00003415 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3416 Data.PreferredType),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003417 Results.data(),Results.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003418}
3419
Douglas Gregoreda7e542010-09-18 01:28:11 +00003420void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3421 if (E.isInvalid())
3422 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003423 else if (getLangOpts().ObjC1)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00003424 CodeCompleteObjCInstanceMessage(S, E.take(), None, false);
Douglas Gregored0b69d2010-09-15 16:23:04 +00003425}
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003426
Douglas Gregorb888acf2010-12-09 23:01:55 +00003427/// \brief The set of properties that have already been added, referenced by
3428/// property name.
3429typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3430
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003431/// \brief Retrieve the container definition, if any?
3432static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3433 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3434 if (Interface->hasDefinition())
3435 return Interface->getDefinition();
3436
3437 return Interface;
3438 }
3439
3440 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3441 if (Protocol->hasDefinition())
3442 return Protocol->getDefinition();
3443
3444 return Protocol;
3445 }
3446 return Container;
3447}
3448
3449static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor5d649882009-11-18 22:32:06 +00003450 bool AllowCategories,
Douglas Gregor95147142011-05-05 15:50:42 +00003451 bool AllowNullaryMethods,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003452 DeclContext *CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003453 AddedPropertiesSet &AddedProperties,
Douglas Gregor9291bad2009-11-18 01:29:26 +00003454 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00003455 typedef CodeCompletionResult Result;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003456
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003457 // Retrieve the definition.
3458 Container = getContainerDef(Container);
3459
Douglas Gregor9291bad2009-11-18 01:29:26 +00003460 // Add properties in this container.
Aaron Ballmandc4bea42014-03-13 18:47:37 +00003461 for (const auto *P : Container->props())
Douglas Gregorb888acf2010-12-09 23:01:55 +00003462 if (AddedProperties.insert(P->getIdentifier()))
Aaron Ballmandc4bea42014-03-13 18:47:37 +00003463 Results.MaybeAddResult(Result(P, Results.getBasePriority(P), 0),
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00003464 CurContext);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003465
Douglas Gregor95147142011-05-05 15:50:42 +00003466 // Add nullary methods
3467 if (AllowNullaryMethods) {
3468 ASTContext &Context = Container->getASTContext();
Douglas Gregor75acd922011-09-27 23:30:47 +00003469 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Douglas Gregor95147142011-05-05 15:50:42 +00003470 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
3471 MEnd = Container->meth_end();
3472 M != MEnd; ++M) {
3473 if (M->getSelector().isUnarySelector())
3474 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3475 if (AddedProperties.insert(Name)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003476 CodeCompletionBuilder Builder(Results.getAllocator(),
3477 Results.getCodeCompletionTUInfo());
David Blaikie40ed2972012-06-06 20:45:41 +00003478 AddResultTypeChunk(Context, Policy, *M, Builder);
Douglas Gregor95147142011-05-05 15:50:42 +00003479 Builder.AddTypedTextChunk(
3480 Results.getAllocator().CopyString(Name->getName()));
3481
David Blaikie40ed2972012-06-06 20:45:41 +00003482 Results.MaybeAddResult(Result(Builder.TakeString(), *M,
Douglas Gregor78254c82012-03-27 23:34:16 +00003483 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor95147142011-05-05 15:50:42 +00003484 CurContext);
3485 }
3486 }
3487 }
3488
3489
Douglas Gregor9291bad2009-11-18 01:29:26 +00003490 // Add properties in referenced protocols.
3491 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3492 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
3493 PEnd = Protocol->protocol_end();
3494 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003495 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3496 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003497 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor5d649882009-11-18 22:32:06 +00003498 if (AllowCategories) {
3499 // Look through categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00003500 for (ObjCInterfaceDecl::known_categories_iterator
3501 Cat = IFace->known_categories_begin(),
3502 CatEnd = IFace->known_categories_end();
3503 Cat != CatEnd; ++Cat)
3504 AddObjCProperties(*Cat, AllowCategories, AllowNullaryMethods,
Douglas Gregor95147142011-05-05 15:50:42 +00003505 CurContext, AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00003506 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003507
3508 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003509 for (ObjCInterfaceDecl::all_protocol_iterator
3510 I = IFace->all_referenced_protocol_begin(),
3511 E = IFace->all_referenced_protocol_end(); I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003512 AddObjCProperties(*I, AllowCategories, AllowNullaryMethods, CurContext,
3513 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003514
3515 // Look in the superclass.
3516 if (IFace->getSuperClass())
Douglas Gregor95147142011-05-05 15:50:42 +00003517 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3518 AllowNullaryMethods, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003519 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003520 } else if (const ObjCCategoryDecl *Category
3521 = dyn_cast<ObjCCategoryDecl>(Container)) {
3522 // Look through protocols.
Ted Kremenek0ef508d2010-09-01 01:21:15 +00003523 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
3524 PEnd = Category->protocol_end();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003525 P != PEnd; ++P)
Douglas Gregor95147142011-05-05 15:50:42 +00003526 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3527 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003528 }
3529}
3530
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003531void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor2436e712009-09-17 21:32:03 +00003532 SourceLocation OpLoc,
3533 bool IsArrow) {
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003534 if (!Base || !CodeCompleter)
Douglas Gregor2436e712009-09-17 21:32:03 +00003535 return;
3536
Douglas Gregor1cc88a92012-01-23 15:59:30 +00003537 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3538 if (ConvertedBase.isInvalid())
3539 return;
3540 Base = ConvertedBase.get();
3541
John McCall276321a2010-08-25 06:19:51 +00003542 typedef CodeCompletionResult Result;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003543
Douglas Gregor2436e712009-09-17 21:32:03 +00003544 QualType BaseType = Base->getType();
Douglas Gregor3545ff42009-09-21 16:56:56 +00003545
3546 if (IsArrow) {
3547 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3548 BaseType = Ptr->getPointeeType();
3549 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003550 /*Do nothing*/ ;
Douglas Gregor3545ff42009-09-21 16:56:56 +00003551 else
3552 return;
3553 }
3554
Douglas Gregor21325842011-07-07 16:03:39 +00003555 enum CodeCompletionContext::Kind contextKind;
3556
3557 if (IsArrow) {
3558 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3559 }
3560 else {
3561 if (BaseType->isObjCObjectPointerType() ||
3562 BaseType->isObjCObjectOrInterfaceType()) {
3563 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3564 }
3565 else {
3566 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3567 }
3568 }
3569
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003570 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003571 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00003572 CodeCompletionContext(contextKind,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003573 BaseType),
3574 &ResultBuilder::IsMember);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003575 Results.EnterNewScope();
3576 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor9be0ed42010-08-26 16:36:48 +00003577 // Indicate that we are performing a member access, and the cv-qualifiers
3578 // for the base object type.
3579 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3580
Douglas Gregor9291bad2009-11-18 01:29:26 +00003581 // Access to a C/C++ class, struct, or union.
Douglas Gregor6ae4c522010-01-14 03:21:49 +00003582 Results.allowNestedNameSpecifiers();
Douglas Gregor09bbc652010-01-14 15:47:35 +00003583 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00003584 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3585 CodeCompleter->includeGlobals());
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003586
David Blaikiebbafb8a2012-03-11 07:00:24 +00003587 if (getLangOpts().CPlusPlus) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003588 if (!Results.empty()) {
3589 // The "template" keyword can follow "->" or "." in the grammar.
3590 // However, we only want to suggest the template keyword if something
3591 // is dependent.
3592 bool IsDependent = BaseType->isDependentType();
3593 if (!IsDependent) {
3594 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
Ted Kremenekc37877d2013-10-08 17:08:03 +00003595 if (DeclContext *Ctx = DepScope->getEntity()) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003596 IsDependent = Ctx->isDependentContext();
3597 break;
3598 }
3599 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003600
Douglas Gregor9291bad2009-11-18 01:29:26 +00003601 if (IsDependent)
Douglas Gregor78a21012010-01-14 16:01:26 +00003602 Results.AddResult(Result("template"));
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003603 }
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003604 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003605 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3606 // Objective-C property reference.
Douglas Gregorb888acf2010-12-09 23:01:55 +00003607 AddedPropertiesSet AddedProperties;
Douglas Gregor9291bad2009-11-18 01:29:26 +00003608
3609 // Add property results based on our interface.
3610 const ObjCObjectPointerType *ObjCPtr
3611 = BaseType->getAsObjCInterfacePointerType();
3612 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor95147142011-05-05 15:50:42 +00003613 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3614 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00003615 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003616
3617 // Add properties from the protocols in a qualified interface.
3618 for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(),
3619 E = ObjCPtr->qual_end();
3620 I != E; ++I)
Douglas Gregor95147142011-05-05 15:50:42 +00003621 AddObjCProperties(*I, true, /*AllowNullaryMethods=*/true, CurContext,
3622 AddedProperties, Results);
Douglas Gregor9291bad2009-11-18 01:29:26 +00003623 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCall8b07ec22010-05-15 11:32:37 +00003624 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor9291bad2009-11-18 01:29:26 +00003625 // Objective-C instance variable access.
3626 ObjCInterfaceDecl *Class = 0;
3627 if (const ObjCObjectPointerType *ObjCPtr
3628 = BaseType->getAs<ObjCObjectPointerType>())
3629 Class = ObjCPtr->getInterfaceDecl();
3630 else
John McCall8b07ec22010-05-15 11:32:37 +00003631 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor9291bad2009-11-18 01:29:26 +00003632
3633 // Add all ivars from this class and its superclasses.
Douglas Gregor2b8162b2010-01-14 16:08:12 +00003634 if (Class) {
3635 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3636 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor39982192010-08-15 06:18:01 +00003637 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3638 CodeCompleter->includeGlobals());
Douglas Gregor9291bad2009-11-18 01:29:26 +00003639 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003640 }
Douglas Gregor9291bad2009-11-18 01:29:26 +00003641
3642 // FIXME: How do we cope with isa?
3643
3644 Results.ExitScope();
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003645
Daniel Dunbar242ea9a2009-11-13 08:58:20 +00003646 // Hand off the results found for code completion.
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003647 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003648 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003649 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00003650}
3651
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003652void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3653 if (!CodeCompleter)
3654 return;
3655
Douglas Gregor3545ff42009-09-21 16:56:56 +00003656 ResultBuilder::LookupFilter Filter = 0;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003657 enum CodeCompletionContext::Kind ContextKind
3658 = CodeCompletionContext::CCC_Other;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003659 switch ((DeclSpec::TST)TagSpec) {
3660 case DeclSpec::TST_enum:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003661 Filter = &ResultBuilder::IsEnum;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003662 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003663 break;
3664
3665 case DeclSpec::TST_union:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003666 Filter = &ResultBuilder::IsUnion;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003667 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003668 break;
3669
3670 case DeclSpec::TST_struct:
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003671 case DeclSpec::TST_class:
Joao Matosdc86f942012-08-31 18:45:21 +00003672 case DeclSpec::TST_interface:
Douglas Gregor3545ff42009-09-21 16:56:56 +00003673 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003674 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003675 break;
3676
3677 default:
David Blaikie83d382b2011-09-23 05:06:16 +00003678 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003679 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003680
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003681 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3682 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00003683 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCalle87beb22010-04-23 18:46:30 +00003684
3685 // First pass: look for tags.
3686 Results.setFilter(Filter);
Douglas Gregor39982192010-08-15 06:18:01 +00003687 LookupVisibleDecls(S, LookupTagName, Consumer,
3688 CodeCompleter->includeGlobals());
John McCalle87beb22010-04-23 18:46:30 +00003689
Douglas Gregor39982192010-08-15 06:18:01 +00003690 if (CodeCompleter->includeGlobals()) {
3691 // Second pass: look for nested name specifiers.
3692 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3693 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3694 }
Douglas Gregor3545ff42009-09-21 16:56:56 +00003695
Douglas Gregor0ac41382010-09-23 23:01:17 +00003696 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003697 Results.data(),Results.size());
Douglas Gregorf45b0cf2009-09-18 15:37:17 +00003698}
3699
Douglas Gregor28c78432010-08-27 17:35:51 +00003700void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003701 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003702 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003703 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor28c78432010-08-27 17:35:51 +00003704 Results.EnterNewScope();
3705 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3706 Results.AddResult("const");
3707 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3708 Results.AddResult("volatile");
David Blaikiebbafb8a2012-03-11 07:00:24 +00003709 if (getLangOpts().C99 &&
Douglas Gregor28c78432010-08-27 17:35:51 +00003710 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3711 Results.AddResult("restrict");
Richard Smith8e1ac332013-03-28 01:55:44 +00003712 if (getLangOpts().C11 &&
3713 !(DS.getTypeQualifiers() & DeclSpec::TQ_atomic))
3714 Results.AddResult("_Atomic");
Douglas Gregor28c78432010-08-27 17:35:51 +00003715 Results.ExitScope();
3716 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00003717 Results.getCompletionContext(),
Douglas Gregor28c78432010-08-27 17:35:51 +00003718 Results.data(), Results.size());
3719}
3720
Douglas Gregord328d572009-09-21 18:10:23 +00003721void Sema::CodeCompleteCase(Scope *S) {
John McCallaab3e412010-08-25 08:40:02 +00003722 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregord328d572009-09-21 18:10:23 +00003723 return;
John McCall5939b162011-08-06 07:30:58 +00003724
John McCallaab3e412010-08-25 08:40:02 +00003725 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCall5939b162011-08-06 07:30:58 +00003726 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3727 if (!type->isEnumeralType()) {
3728 CodeCompleteExpressionData Data(type);
Douglas Gregor68762e72010-08-23 21:17:50 +00003729 Data.IntegralConstantExpression = true;
3730 CodeCompleteExpression(S, Data);
Douglas Gregord328d572009-09-21 18:10:23 +00003731 return;
Douglas Gregor85b50632010-07-28 21:50:18 +00003732 }
Douglas Gregord328d572009-09-21 18:10:23 +00003733
3734 // Code-complete the cases of a switch statement over an enumeration type
3735 // by providing the list of
John McCall5939b162011-08-06 07:30:58 +00003736 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregor9b4f3702012-06-12 13:44:08 +00003737 if (EnumDecl *Def = Enum->getDefinition())
3738 Enum = Def;
Douglas Gregord328d572009-09-21 18:10:23 +00003739
3740 // Determine which enumerators we have already seen in the switch statement.
3741 // FIXME: Ideally, we would also be able to look *past* the code-completion
3742 // token, in case we are code-completing in the middle of the switch and not
3743 // at the end. However, we aren't able to do so at the moment.
3744 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Douglas Gregorf2510672009-09-21 19:57:38 +00003745 NestedNameSpecifier *Qualifier = 0;
Douglas Gregord328d572009-09-21 18:10:23 +00003746 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3747 SC = SC->getNextSwitchCase()) {
3748 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3749 if (!Case)
3750 continue;
3751
3752 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3753 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3754 if (EnumConstantDecl *Enumerator
3755 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3756 // We look into the AST of the case statement to determine which
3757 // enumerator was named. Alternatively, we could compute the value of
3758 // the integral constant expression, then compare it against the
3759 // values of each enumerator. However, value-based approach would not
3760 // work as well with C++ templates where enumerators declared within a
3761 // template are type- and value-dependent.
3762 EnumeratorsSeen.insert(Enumerator);
3763
Douglas Gregorf2510672009-09-21 19:57:38 +00003764 // If this is a qualified-id, keep track of the nested-name-specifier
3765 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregord328d572009-09-21 18:10:23 +00003766 //
3767 // switch (TagD.getKind()) {
3768 // case TagDecl::TK_enum:
3769 // break;
3770 // case XXX
3771 //
Douglas Gregorf2510672009-09-21 19:57:38 +00003772 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregord328d572009-09-21 18:10:23 +00003773 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3774 // TK_struct, and TK_class.
Douglas Gregor4bd90e52009-10-23 18:54:35 +00003775 Qualifier = DRE->getQualifier();
Douglas Gregord328d572009-09-21 18:10:23 +00003776 }
3777 }
3778
David Blaikiebbafb8a2012-03-11 07:00:24 +00003779 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorf2510672009-09-21 19:57:38 +00003780 // If there are no prior enumerators in C++, check whether we have to
3781 // qualify the names of the enumerators that we suggest, because they
3782 // may not be visible in this scope.
Douglas Gregord3cebdb2012-02-01 05:02:47 +00003783 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorf2510672009-09-21 19:57:38 +00003784 }
3785
Douglas Gregord328d572009-09-21 18:10:23 +00003786 // Add any enumerators that have not yet been mentioned.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003787 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003788 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00003789 CodeCompletionContext::CCC_Expression);
Douglas Gregord328d572009-09-21 18:10:23 +00003790 Results.EnterNewScope();
Aaron Ballman23a6dcb2014-03-08 18:45:14 +00003791 for (auto *E : Enum->enumerators()) {
3792 if (EnumeratorsSeen.count(E))
Douglas Gregord328d572009-09-21 18:10:23 +00003793 continue;
3794
Aaron Ballman23a6dcb2014-03-08 18:45:14 +00003795 CodeCompletionResult R(E, CCP_EnumInCase, Qualifier);
Douglas Gregor3a69eaf2011-02-18 23:30:37 +00003796 Results.AddResult(R, CurContext, 0, false);
Douglas Gregord328d572009-09-21 18:10:23 +00003797 }
3798 Results.ExitScope();
Douglas Gregor285560922010-04-06 20:02:15 +00003799
Douglas Gregor21325842011-07-07 16:03:39 +00003800 //We need to make sure we're setting the right context,
3801 //so only say we include macros if the code completer says we do
3802 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3803 if (CodeCompleter->includeMacros()) {
Douglas Gregor8cb17462012-10-09 16:01:50 +00003804 AddMacroResults(PP, Results, false);
Douglas Gregor21325842011-07-07 16:03:39 +00003805 kind = CodeCompletionContext::CCC_OtherWithMacros;
3806 }
3807
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003808 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00003809 kind,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003810 Results.data(),Results.size());
Douglas Gregord328d572009-09-21 18:10:23 +00003811}
3812
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003813static bool anyNullArguments(ArrayRef<Expr *> Args) {
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003814 if (Args.size() && !Args.data())
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003815 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003816
3817 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003818 if (!Args[I])
3819 return true;
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003820
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003821 return false;
3822}
3823
Robert Wilhelm16e94b92013-08-09 18:02:13 +00003824void Sema::CodeCompleteCall(Scope *S, Expr *FnIn, ArrayRef<Expr *> Args) {
Douglas Gregorcabea402009-09-22 15:41:20 +00003825 if (!CodeCompleter)
3826 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003827
3828 // When we're code-completing for a call, we fall back to ordinary
3829 // name code-completion whenever we can't produce specific
3830 // results. We may want to revisit this strategy in the future,
3831 // e.g., by merging the two kinds of results.
3832
Douglas Gregorcabea402009-09-22 15:41:20 +00003833 Expr *Fn = (Expr *)FnIn;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003834
Douglas Gregorcabea402009-09-22 15:41:20 +00003835 // Ignore type-dependent call expressions entirely.
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003836 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3837 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003838 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregorcabea402009-09-22 15:41:20 +00003839 return;
Douglas Gregor3ef59522009-12-11 19:06:04 +00003840 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003841
John McCall57500772009-12-16 12:17:52 +00003842 // Build an overload candidate set based on the functions we find.
John McCallbc077cf2010-02-08 23:07:23 +00003843 SourceLocation Loc = Fn->getExprLoc();
3844 OverloadCandidateSet CandidateSet(Loc);
John McCall57500772009-12-16 12:17:52 +00003845
Douglas Gregorcabea402009-09-22 15:41:20 +00003846 // FIXME: What if we're calling something that isn't a function declaration?
3847 // FIXME: What if we're calling a pseudo-destructor?
3848 // FIXME: What if we're calling a member function?
3849
Douglas Gregorff59f672010-01-21 15:46:19 +00003850 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00003851 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorff59f672010-01-21 15:46:19 +00003852
John McCall57500772009-12-16 12:17:52 +00003853 Expr *NakedFn = Fn->IgnoreParenCasts();
3854 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003855 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
John McCall57500772009-12-16 12:17:52 +00003856 /*PartialOverloading=*/ true);
3857 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3858 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorff59f672010-01-21 15:46:19 +00003859 if (FDecl) {
David Blaikiebbafb8a2012-03-11 07:00:24 +00003860 if (!getLangOpts().CPlusPlus ||
Douglas Gregor6ed3eb82010-05-30 06:10:08 +00003861 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorff59f672010-01-21 15:46:19 +00003862 Results.push_back(ResultCandidate(FDecl));
3863 else
John McCallb89836b2010-01-26 01:37:31 +00003864 // FIXME: access?
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003865 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), Args,
3866 CandidateSet, false, /*PartialOverloading*/true);
Douglas Gregorff59f672010-01-21 15:46:19 +00003867 }
John McCall57500772009-12-16 12:17:52 +00003868 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003869
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003870 QualType ParamType;
3871
Douglas Gregorff59f672010-01-21 15:46:19 +00003872 if (!CandidateSet.empty()) {
3873 // Sort the overload candidate set by placing the best overloads first.
Benjamin Kramerbbdd7642014-03-01 14:48:57 +00003874 std::stable_sort(
3875 CandidateSet.begin(), CandidateSet.end(),
3876 [&](const OverloadCandidate &X, const OverloadCandidate &Y) {
3877 return isBetterOverloadCandidate(*this, X, Y, Loc);
3878 });
3879
Douglas Gregorff59f672010-01-21 15:46:19 +00003880 // Add the remaining viable overload candidates as code-completion reslults.
3881 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3882 CandEnd = CandidateSet.end();
3883 Cand != CandEnd; ++Cand) {
3884 if (Cand->Viable)
3885 Results.push_back(ResultCandidate(Cand->Function));
3886 }
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003887
3888 // From the viable candidates, try to determine the type of this parameter.
3889 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3890 if (const FunctionType *FType = Results[I].getFunctionType())
3891 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
Alp Toker9cacbab2014-01-20 20:26:09 +00003892 if (Args.size() < Proto->getNumParams()) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003893 if (ParamType.isNull())
Alp Toker9cacbab2014-01-20 20:26:09 +00003894 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003895 else if (!Context.hasSameUnqualifiedType(
Alp Toker9cacbab2014-01-20 20:26:09 +00003896 ParamType.getNonReferenceType(),
3897 Proto->getParamType(Args.size())
3898 .getNonReferenceType())) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003899 ParamType = QualType();
3900 break;
3901 }
3902 }
3903 }
3904 } else {
3905 // Try to determine the parameter type from the type of the expression
3906 // being called.
3907 QualType FunctionType = Fn->getType();
3908 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3909 FunctionType = Ptr->getPointeeType();
3910 else if (const BlockPointerType *BlockPtr
3911 = FunctionType->getAs<BlockPointerType>())
3912 FunctionType = BlockPtr->getPointeeType();
3913 else if (const MemberPointerType *MemPtr
3914 = FunctionType->getAs<MemberPointerType>())
3915 FunctionType = MemPtr->getPointeeType();
3916
3917 if (const FunctionProtoType *Proto
3918 = FunctionType->getAs<FunctionProtoType>()) {
Alp Toker9cacbab2014-01-20 20:26:09 +00003919 if (Args.size() < Proto->getNumParams())
3920 ParamType = Proto->getParamType(Args.size());
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003921 }
Douglas Gregorcabea402009-09-22 15:41:20 +00003922 }
Douglas Gregor3ef59522009-12-11 19:06:04 +00003923
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003924 if (ParamType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003925 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003926 else
3927 CodeCompleteExpression(S, ParamType);
3928
Douglas Gregorc01890e2010-04-06 20:19:47 +00003929 if (!Results.empty())
Ahmed Charlesb24b9aa2012-02-25 11:00:22 +00003930 CodeCompleter->ProcessOverloadCandidates(*this, Args.size(), Results.data(),
Douglas Gregor3ef59522009-12-11 19:06:04 +00003931 Results.size());
Douglas Gregorcabea402009-09-22 15:41:20 +00003932}
3933
John McCall48871652010-08-21 09:40:31 +00003934void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3935 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003936 if (!VD) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003937 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003938 return;
3939 }
3940
3941 CodeCompleteExpression(S, VD->getType());
3942}
3943
3944void Sema::CodeCompleteReturn(Scope *S) {
3945 QualType ResultType;
3946 if (isa<BlockDecl>(CurContext)) {
3947 if (BlockScopeInfo *BSI = getCurBlock())
3948 ResultType = BSI->ReturnType;
3949 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00003950 ResultType = Function->getReturnType();
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003951 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
Alp Toker314cc812014-01-25 16:55:45 +00003952 ResultType = Method->getReturnType();
3953
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003954 if (ResultType.isNull())
Douglas Gregor00c37ef2010-08-11 21:23:17 +00003955 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00003956 else
3957 CodeCompleteExpression(S, ResultType);
3958}
3959
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003960void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003961 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003962 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003963 mapCodeCompletionContext(*this, PCC_Statement));
3964 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3965 Results.EnterNewScope();
3966
3967 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3968 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3969 CodeCompleter->includeGlobals());
3970
3971 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
3972
3973 // "else" block
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00003974 CodeCompletionBuilder Builder(Results.getAllocator(),
3975 Results.getCodeCompletionTUInfo());
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003976 Builder.AddTypedTextChunk("else");
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00003977 if (Results.includeCodePatterns()) {
3978 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3979 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
3980 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3981 Builder.AddPlaceholderChunk("statements");
3982 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3983 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
3984 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003985 Results.AddResult(Builder.TakeString());
3986
3987 // "else if" block
3988 Builder.AddTypedTextChunk("else");
3989 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3990 Builder.AddTextChunk("if");
3991 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3992 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikiebbafb8a2012-03-11 07:00:24 +00003993 if (getLangOpts().CPlusPlus)
Douglas Gregor4ecb7202011-07-30 08:36:53 +00003994 Builder.AddPlaceholderChunk("condition");
3995 else
3996 Builder.AddPlaceholderChunk("expression");
3997 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor3a5d6c22012-02-16 17:49:04 +00003998 if (Results.includeCodePatterns()) {
3999 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4000 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4001 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4002 Builder.AddPlaceholderChunk("statements");
4003 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4004 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4005 }
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004006 Results.AddResult(Builder.TakeString());
4007
4008 Results.ExitScope();
4009
4010 if (S->getFnParent())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004011 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004012
4013 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00004014 AddMacroResults(PP, Results, false);
Douglas Gregor4ecb7202011-07-30 08:36:53 +00004015
4016 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4017 Results.data(),Results.size());
4018}
4019
Richard Trieu2bd04012011-09-09 02:00:50 +00004020void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004021 if (LHS)
4022 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4023 else
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004024 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor7aa6b222010-05-30 01:49:25 +00004025}
4026
Jeffrey Yasskinc76498d2010-04-08 16:38:48 +00004027void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor2436e712009-09-17 21:32:03 +00004028 bool EnteringContext) {
4029 if (!SS.getScopeRep() || !CodeCompleter)
4030 return;
4031
Douglas Gregor3545ff42009-09-21 16:56:56 +00004032 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4033 if (!Ctx)
4034 return;
Douglas Gregor800f2f02009-12-11 18:28:39 +00004035
4036 // Try to instantiate any non-dependent declaration contexts before
4037 // we look in them.
John McCall0b66eb32010-05-01 00:40:08 +00004038 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregor800f2f02009-12-11 18:28:39 +00004039 return;
4040
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004041 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004042 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004043 CodeCompletionContext::CCC_Name);
Douglas Gregorac322ec2010-08-27 21:18:54 +00004044 Results.EnterNewScope();
Douglas Gregor0ac41382010-09-23 23:01:17 +00004045
Douglas Gregor3545ff42009-09-21 16:56:56 +00004046 // The "template" keyword can follow "::" in the grammar, but only
4047 // put it into the grammar if the nested-name-specifier is dependent.
Aaron Ballman4a979672014-01-03 13:56:08 +00004048 NestedNameSpecifier *NNS = SS.getScopeRep();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004049 if (!Results.empty() && NNS->isDependent())
Douglas Gregor78a21012010-01-14 16:01:26 +00004050 Results.AddResult("template");
Douglas Gregorac322ec2010-08-27 21:18:54 +00004051
4052 // Add calls to overridden virtual functions, if there are any.
4053 //
4054 // FIXME: This isn't wonderful, because we don't know whether we're actually
4055 // in a context that permits expressions. This is a general issue with
4056 // qualified-id completions.
4057 if (!EnteringContext)
4058 MaybeAddOverrideCalls(*this, Ctx, Results);
4059 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004060
Douglas Gregorac322ec2010-08-27 21:18:54 +00004061 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4062 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4063
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004064 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorc1679ec2011-07-25 17:48:11 +00004065 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004066 Results.data(),Results.size());
Douglas Gregor2436e712009-09-17 21:32:03 +00004067}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004068
4069void Sema::CodeCompleteUsing(Scope *S) {
4070 if (!CodeCompleter)
4071 return;
4072
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004073 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004074 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004075 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4076 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004077 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004078
4079 // If we aren't in class scope, we could see the "namespace" keyword.
4080 if (!S->isClassScope())
John McCall276321a2010-08-25 06:19:51 +00004081 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004082
4083 // After "using", we can see anything that would start a
4084 // nested-name-specifier.
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004085 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004086 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4087 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004088 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004089
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004090 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004091 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004092 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004093}
4094
4095void Sema::CodeCompleteUsingDirective(Scope *S) {
4096 if (!CodeCompleter)
4097 return;
4098
Douglas Gregor3545ff42009-09-21 16:56:56 +00004099 // After "using namespace", we expect to see a namespace name or namespace
4100 // alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004101 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004102 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004103 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004104 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004105 Results.EnterNewScope();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004106 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004107 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4108 CodeCompleter->includeGlobals());
Douglas Gregor64b12b52009-09-22 23:31:26 +00004109 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004110 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004111 CodeCompletionContext::CCC_Namespace,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004112 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004113}
4114
4115void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4116 if (!CodeCompleter)
4117 return;
4118
Ted Kremenekc37877d2013-10-08 17:08:03 +00004119 DeclContext *Ctx = S->getEntity();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004120 if (!S->getParent())
4121 Ctx = Context.getTranslationUnitDecl();
4122
Douglas Gregor0ac41382010-09-23 23:01:17 +00004123 bool SuppressedGlobalResults
4124 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4125
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004126 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004127 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004128 SuppressedGlobalResults
4129 ? CodeCompletionContext::CCC_Namespace
4130 : CodeCompletionContext::CCC_Other,
4131 &ResultBuilder::IsNamespace);
4132
4133 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor3545ff42009-09-21 16:56:56 +00004134 // We only want to see those namespaces that have already been defined
4135 // within this scope, because its likely that the user is creating an
4136 // extended namespace declaration. Keep track of the most recent
4137 // definition of each namespace.
4138 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4139 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4140 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4141 NS != NSEnd; ++NS)
David Blaikie40ed2972012-06-06 20:45:41 +00004142 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004143
4144 // Add the most recent definition (or extended definition) of each
4145 // namespace to the list of results.
Douglas Gregor64b12b52009-09-22 23:31:26 +00004146 Results.EnterNewScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004147 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregor78254c82012-03-27 23:34:16 +00004148 NS = OrigToLatest.begin(),
4149 NSEnd = OrigToLatest.end();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004150 NS != NSEnd; ++NS)
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004151 Results.AddResult(CodeCompletionResult(
4152 NS->second, Results.getBasePriority(NS->second), 0),
Douglas Gregorfc59ce12010-01-14 16:14:35 +00004153 CurContext, 0, false);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004154 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004155 }
4156
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004157 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004158 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004159 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004160}
4161
4162void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4163 if (!CodeCompleter)
4164 return;
4165
Douglas Gregor3545ff42009-09-21 16:56:56 +00004166 // After "namespace", we expect to see a namespace or alias.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004167 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004168 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004169 CodeCompletionContext::CCC_Namespace,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004170 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004171 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004172 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4173 CodeCompleter->includeGlobals());
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004174 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004175 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004176 Results.data(),Results.size());
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004177}
4178
Douglas Gregorc811ede2009-09-18 20:05:18 +00004179void Sema::CodeCompleteOperatorName(Scope *S) {
4180 if (!CodeCompleter)
4181 return;
Douglas Gregor3545ff42009-09-21 16:56:56 +00004182
John McCall276321a2010-08-25 06:19:51 +00004183 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004184 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004185 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004186 CodeCompletionContext::CCC_Type,
Douglas Gregor0ac41382010-09-23 23:01:17 +00004187 &ResultBuilder::IsType);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004188 Results.EnterNewScope();
Douglas Gregorc811ede2009-09-18 20:05:18 +00004189
Douglas Gregor3545ff42009-09-21 16:56:56 +00004190 // Add the names of overloadable operators.
4191#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4192 if (std::strcmp(Spelling, "?")) \
Douglas Gregor78a21012010-01-14 16:01:26 +00004193 Results.AddResult(Result(Spelling));
Douglas Gregor3545ff42009-09-21 16:56:56 +00004194#include "clang/Basic/OperatorKinds.def"
4195
4196 // Add any type names visible from the current scope
Douglas Gregor6ae4c522010-01-14 03:21:49 +00004197 Results.allowNestedNameSpecifiers();
Douglas Gregora6e2edc2010-01-14 03:27:13 +00004198 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor39982192010-08-15 06:18:01 +00004199 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4200 CodeCompleter->includeGlobals());
Douglas Gregor3545ff42009-09-21 16:56:56 +00004201
4202 // Add any type specifiers
David Blaikiebbafb8a2012-03-11 07:00:24 +00004203 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor64b12b52009-09-22 23:31:26 +00004204 Results.ExitScope();
Douglas Gregor3545ff42009-09-21 16:56:56 +00004205
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004206 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor39982192010-08-15 06:18:01 +00004207 CodeCompletionContext::CCC_Type,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004208 Results.data(),Results.size());
Douglas Gregorc811ede2009-09-18 20:05:18 +00004209}
Douglas Gregor7e90c6d2009-09-18 19:03:04 +00004210
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004211void Sema::CodeCompleteConstructorInitializer(
4212 Decl *ConstructorD,
4213 ArrayRef <CXXCtorInitializer *> Initializers) {
Douglas Gregor75acd922011-09-27 23:30:47 +00004214 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004215 CXXConstructorDecl *Constructor
4216 = static_cast<CXXConstructorDecl *>(ConstructorD);
4217 if (!Constructor)
4218 return;
4219
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004220 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004221 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00004222 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004223 Results.EnterNewScope();
4224
4225 // Fill in any already-initialized fields or base classes.
4226 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4227 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004228 for (unsigned I = 0, E = Initializers.size(); I != E; ++I) {
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004229 if (Initializers[I]->isBaseInitializer())
4230 InitializedBases.insert(
4231 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4232 else
Francois Pichetd583da02010-12-04 09:14:42 +00004233 InitializedFields.insert(cast<FieldDecl>(
4234 Initializers[I]->getAnyMember()));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004235 }
4236
4237 // Add completions for base classes.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004238 CodeCompletionBuilder Builder(Results.getAllocator(),
4239 Results.getCodeCompletionTUInfo());
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004240 bool SawLastInitializer = Initializers.empty();
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004241 CXXRecordDecl *ClassDecl = Constructor->getParent();
Aaron Ballman574705e2014-03-13 15:41:46 +00004242 for (const auto &Base : ClassDecl->bases()) {
4243 if (!InitializedBases.insert(Context.getCanonicalType(Base.getType()))) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004244 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004245 = !Initializers.empty() &&
4246 Initializers.back()->isBaseInitializer() &&
Aaron Ballman574705e2014-03-13 15:41:46 +00004247 Context.hasSameUnqualifiedType(Base.getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004248 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004249 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004250 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004251
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004252 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004253 Results.getAllocator().CopyString(
Aaron Ballman574705e2014-03-13 15:41:46 +00004254 Base.getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004255 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4256 Builder.AddPlaceholderChunk("args");
4257 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4258 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004259 SawLastInitializer? CCP_NextInitializer
4260 : CCP_MemberDeclaration));
4261 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004262 }
4263
4264 // Add completions for virtual base classes.
Aaron Ballman445a9392014-03-13 16:15:17 +00004265 for (const auto &Base : ClassDecl->vbases()) {
4266 if (!InitializedBases.insert(Context.getCanonicalType(Base.getType()))) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004267 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004268 = !Initializers.empty() &&
4269 Initializers.back()->isBaseInitializer() &&
Aaron Ballman445a9392014-03-13 16:15:17 +00004270 Context.hasSameUnqualifiedType(Base.getType(),
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004271 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004272 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004273 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004274
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004275 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004276 Builder.getAllocator().CopyString(
Aaron Ballman445a9392014-03-13 16:15:17 +00004277 Base.getType().getAsString(Policy)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004278 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4279 Builder.AddPlaceholderChunk("args");
4280 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4281 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004282 SawLastInitializer? CCP_NextInitializer
4283 : CCP_MemberDeclaration));
4284 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004285 }
4286
4287 // Add completions for members.
Aaron Ballmane8a8bae2014-03-08 20:12:42 +00004288 for (auto *Field : ClassDecl->fields()) {
Douglas Gregor99129ef2010-08-29 19:27:27 +00004289 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4290 SawLastInitializer
Dmitri Gribenko27cb3dd02013-06-23 22:58:02 +00004291 = !Initializers.empty() &&
4292 Initializers.back()->isAnyMemberInitializer() &&
Aaron Ballmane8a8bae2014-03-08 20:12:42 +00004293 Initializers.back()->getAnyMember() == Field;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004294 continue;
Douglas Gregor99129ef2010-08-29 19:27:27 +00004295 }
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004296
4297 if (!Field->getDeclName())
4298 continue;
4299
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00004300 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004301 Field->getIdentifier()->getName()));
4302 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4303 Builder.AddPlaceholderChunk("args");
4304 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4305 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor99129ef2010-08-29 19:27:27 +00004306 SawLastInitializer? CCP_NextInitializer
Douglas Gregorf3af3112010-09-09 21:42:20 +00004307 : CCP_MemberDeclaration,
Douglas Gregor78254c82012-03-27 23:34:16 +00004308 CXCursor_MemberRef,
4309 CXAvailability_Available,
Aaron Ballmane8a8bae2014-03-08 20:12:42 +00004310 Field));
Douglas Gregor99129ef2010-08-29 19:27:27 +00004311 SawLastInitializer = false;
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004312 }
4313 Results.ExitScope();
4314
Douglas Gregor0ac41382010-09-23 23:01:17 +00004315 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregoreaeeca92010-08-28 00:00:50 +00004316 Results.data(), Results.size());
4317}
4318
Douglas Gregord8c61782012-02-15 15:34:24 +00004319/// \brief Determine whether this scope denotes a namespace.
4320static bool isNamespaceScope(Scope *S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00004321 DeclContext *DC = S->getEntity();
Douglas Gregord8c61782012-02-15 15:34:24 +00004322 if (!DC)
4323 return false;
4324
4325 return DC->isFileContext();
4326}
4327
4328void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4329 bool AfterAmpersand) {
4330 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004331 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord8c61782012-02-15 15:34:24 +00004332 CodeCompletionContext::CCC_Other);
4333 Results.EnterNewScope();
4334
4335 // Note what has already been captured.
4336 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4337 bool IncludedThis = false;
4338 for (SmallVectorImpl<LambdaCapture>::iterator C = Intro.Captures.begin(),
4339 CEnd = Intro.Captures.end();
4340 C != CEnd; ++C) {
4341 if (C->Kind == LCK_This) {
4342 IncludedThis = true;
4343 continue;
4344 }
4345
4346 Known.insert(C->Id);
4347 }
4348
4349 // Look for other capturable variables.
4350 for (; S && !isNamespaceScope(S); S = S->getParent()) {
4351 for (Scope::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
4352 D != DEnd; ++D) {
4353 VarDecl *Var = dyn_cast<VarDecl>(*D);
4354 if (!Var ||
4355 !Var->hasLocalStorage() ||
4356 Var->hasAttr<BlocksAttr>())
4357 continue;
4358
4359 if (Known.insert(Var->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004360 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
4361 CurContext, 0, false);
Douglas Gregord8c61782012-02-15 15:34:24 +00004362 }
4363 }
4364
4365 // Add 'this', if it would be valid.
4366 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4367 addThisCompletion(*this, Results);
4368
4369 Results.ExitScope();
4370
4371 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4372 Results.data(), Results.size());
4373}
4374
James Dennett596e4752012-06-14 03:11:41 +00004375/// Macro that optionally prepends an "@" to the string literal passed in via
4376/// Keyword, depending on whether NeedAt is true or false.
4377#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4378
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004379static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004380 ResultBuilder &Results,
4381 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004382 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004383 // Since we have an implementation, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004384 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004385
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004386 CodeCompletionBuilder Builder(Results.getAllocator(),
4387 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004388 if (LangOpts.ObjC2) {
4389 // @dynamic
James Dennett596e4752012-06-14 03:11:41 +00004390 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004391 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4392 Builder.AddPlaceholderChunk("property");
4393 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004394
4395 // @synthesize
James Dennett596e4752012-06-14 03:11:41 +00004396 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004397 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4398 Builder.AddPlaceholderChunk("property");
4399 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004400 }
4401}
4402
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004403static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorf1934162010-01-13 21:24:21 +00004404 ResultBuilder &Results,
4405 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004406 typedef CodeCompletionResult Result;
Douglas Gregorf1934162010-01-13 21:24:21 +00004407
4408 // Since we have an interface or protocol, we can end it.
James Dennett596e4752012-06-14 03:11:41 +00004409 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004410
4411 if (LangOpts.ObjC2) {
4412 // @property
James Dennett596e4752012-06-14 03:11:41 +00004413 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004414
4415 // @required
James Dennett596e4752012-06-14 03:11:41 +00004416 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004417
4418 // @optional
James Dennett596e4752012-06-14 03:11:41 +00004419 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorf1934162010-01-13 21:24:21 +00004420 }
4421}
4422
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004423static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004424 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004425 CodeCompletionBuilder Builder(Results.getAllocator(),
4426 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004427
4428 // @class name ;
James Dennett596e4752012-06-14 03:11:41 +00004429 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004430 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4431 Builder.AddPlaceholderChunk("name");
4432 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004433
Douglas Gregorf4c33342010-05-28 00:22:41 +00004434 if (Results.includeCodePatterns()) {
4435 // @interface name
4436 // FIXME: Could introduce the whole pattern, including superclasses and
4437 // such.
James Dennett596e4752012-06-14 03:11:41 +00004438 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004439 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4440 Builder.AddPlaceholderChunk("class");
4441 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004442
Douglas Gregorf4c33342010-05-28 00:22:41 +00004443 // @protocol name
James Dennett596e4752012-06-14 03:11:41 +00004444 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004445 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4446 Builder.AddPlaceholderChunk("protocol");
4447 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004448
4449 // @implementation name
James Dennett596e4752012-06-14 03:11:41 +00004450 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004451 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4452 Builder.AddPlaceholderChunk("class");
4453 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004454 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004455
4456 // @compatibility_alias name
James Dennett596e4752012-06-14 03:11:41 +00004457 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004458 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4459 Builder.AddPlaceholderChunk("alias");
4460 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4461 Builder.AddPlaceholderChunk("class");
4462 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor61e36812013-03-07 23:26:24 +00004463
4464 if (Results.getSema().getLangOpts().Modules) {
4465 // @import name
4466 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "import"));
4467 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4468 Builder.AddPlaceholderChunk("module");
4469 Results.AddResult(Result(Builder.TakeString()));
4470 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004471}
4472
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004473void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004474 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004475 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004476 CodeCompletionContext::CCC_Other);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004477 Results.EnterNewScope();
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004478 if (isa<ObjCImplDecl>(CurContext))
David Blaikiebbafb8a2012-03-11 07:00:24 +00004479 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004480 else if (CurContext->isObjCContainer())
David Blaikiebbafb8a2012-03-11 07:00:24 +00004481 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorf1934162010-01-13 21:24:21 +00004482 else
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004483 AddObjCTopLevelResults(Results, false);
Douglas Gregorf48706c2009-12-07 09:27:33 +00004484 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004485 HandleCodeCompleteResults(this, CodeCompleter,
4486 CodeCompletionContext::CCC_Other,
4487 Results.data(),Results.size());
Douglas Gregorf48706c2009-12-07 09:27:33 +00004488}
4489
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004490static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004491 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004492 CodeCompletionBuilder Builder(Results.getAllocator(),
4493 Results.getCodeCompletionTUInfo());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004494
4495 // @encode ( type-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004496 const char *EncodeType = "char[]";
David Blaikiebbafb8a2012-03-11 07:00:24 +00004497 if (Results.getSema().getLangOpts().CPlusPlus ||
4498 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose9da05852012-06-15 18:19:56 +00004499 EncodeType = "const char[]";
Douglas Gregore5c79d52011-10-18 21:20:17 +00004500 Builder.AddResultTypeChunk(EncodeType);
James Dennett596e4752012-06-14 03:11:41 +00004501 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004502 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4503 Builder.AddPlaceholderChunk("type-name");
4504 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4505 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004506
4507 // @protocol ( protocol-name )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004508 Builder.AddResultTypeChunk("Protocol *");
James Dennett596e4752012-06-14 03:11:41 +00004509 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004510 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4511 Builder.AddPlaceholderChunk("protocol-name");
4512 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4513 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004514
4515 // @selector ( selector )
Douglas Gregore5c79d52011-10-18 21:20:17 +00004516 Builder.AddResultTypeChunk("SEL");
James Dennett596e4752012-06-14 03:11:41 +00004517 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004518 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4519 Builder.AddPlaceholderChunk("selector");
4520 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4521 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004522
4523 // @"string"
4524 Builder.AddResultTypeChunk("NSString *");
4525 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4526 Builder.AddPlaceholderChunk("string");
4527 Builder.AddTextChunk("\"");
4528 Results.AddResult(Result(Builder.TakeString()));
4529
Douglas Gregor951de302012-07-17 23:24:47 +00004530 // @[objects, ...]
Jordan Rose9da05852012-06-15 18:19:56 +00004531 Builder.AddResultTypeChunk("NSArray *");
James Dennett596e4752012-06-14 03:11:41 +00004532 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004533 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004534 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4535 Results.AddResult(Result(Builder.TakeString()));
4536
Douglas Gregor951de302012-07-17 23:24:47 +00004537 // @{key : object, ...}
Jordan Rose9da05852012-06-15 18:19:56 +00004538 Builder.AddResultTypeChunk("NSDictionary *");
James Dennett596e4752012-06-14 03:11:41 +00004539 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"{"));
Ted Kremeneke65b0862012-03-06 20:05:56 +00004540 Builder.AddPlaceholderChunk("key");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004541 Builder.AddChunk(CodeCompletionString::CK_Colon);
4542 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4543 Builder.AddPlaceholderChunk("object, ...");
Ted Kremeneke65b0862012-03-06 20:05:56 +00004544 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4545 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose9da05852012-06-15 18:19:56 +00004546
Douglas Gregor951de302012-07-17 23:24:47 +00004547 // @(expression)
Jordan Rose9da05852012-06-15 18:19:56 +00004548 Builder.AddResultTypeChunk("id");
4549 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose9da05852012-06-15 18:19:56 +00004550 Builder.AddPlaceholderChunk("expression");
Jordan Rose9da05852012-06-15 18:19:56 +00004551 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4552 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004553}
4554
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004555static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004556 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004557 CodeCompletionBuilder Builder(Results.getAllocator(),
4558 Results.getCodeCompletionTUInfo());
Douglas Gregorf1934162010-01-13 21:24:21 +00004559
Douglas Gregorf4c33342010-05-28 00:22:41 +00004560 if (Results.includeCodePatterns()) {
4561 // @try { statements } @catch ( declaration ) { statements } @finally
4562 // { statements }
James Dennett596e4752012-06-14 03:11:41 +00004563 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004564 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4565 Builder.AddPlaceholderChunk("statements");
4566 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4567 Builder.AddTextChunk("@catch");
4568 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4569 Builder.AddPlaceholderChunk("parameter");
4570 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4571 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4572 Builder.AddPlaceholderChunk("statements");
4573 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4574 Builder.AddTextChunk("@finally");
4575 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4576 Builder.AddPlaceholderChunk("statements");
4577 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4578 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004579 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004580
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004581 // @throw
James Dennett596e4752012-06-14 03:11:41 +00004582 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004583 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4584 Builder.AddPlaceholderChunk("expression");
4585 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf1934162010-01-13 21:24:21 +00004586
Douglas Gregorf4c33342010-05-28 00:22:41 +00004587 if (Results.includeCodePatterns()) {
4588 // @synchronized ( expression ) { statements }
James Dennett596e4752012-06-14 03:11:41 +00004589 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004590 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4591 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4592 Builder.AddPlaceholderChunk("expression");
4593 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4594 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4595 Builder.AddPlaceholderChunk("statements");
4596 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4597 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorf4c33342010-05-28 00:22:41 +00004598 }
Douglas Gregorf1934162010-01-13 21:24:21 +00004599}
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004600
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004601static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregor48d46252010-01-13 21:54:15 +00004602 ResultBuilder &Results,
4603 bool NeedAt) {
John McCall276321a2010-08-25 06:19:51 +00004604 typedef CodeCompletionResult Result;
James Dennett596e4752012-06-14 03:11:41 +00004605 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4606 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4607 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004608 if (LangOpts.ObjC2)
James Dennett596e4752012-06-14 03:11:41 +00004609 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregor48d46252010-01-13 21:54:15 +00004610}
4611
4612void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004613 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004614 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004615 CodeCompletionContext::CCC_Other);
Douglas Gregor48d46252010-01-13 21:54:15 +00004616 Results.EnterNewScope();
David Blaikiebbafb8a2012-03-11 07:00:24 +00004617 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregor48d46252010-01-13 21:54:15 +00004618 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004619 HandleCodeCompleteResults(this, CodeCompleter,
4620 CodeCompletionContext::CCC_Other,
4621 Results.data(),Results.size());
Douglas Gregor48d46252010-01-13 21:54:15 +00004622}
4623
4624void Sema::CodeCompleteObjCAtStatement(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 Gregorf1934162010-01-13 21:24:21 +00004628 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004629 AddObjCStatementResults(Results, false);
4630 AddObjCExpressionResults(Results, false);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004631 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004632 HandleCodeCompleteResults(this, CodeCompleter,
4633 CodeCompletionContext::CCC_Other,
4634 Results.data(),Results.size());
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004635}
4636
4637void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004638 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004639 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004640 CodeCompletionContext::CCC_Other);
Douglas Gregorbc7c5e42009-12-07 09:51:25 +00004641 Results.EnterNewScope();
Douglas Gregorf98e6a22010-01-13 23:51:12 +00004642 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
Douglas Gregore6078da2009-11-19 00:14:45 +00004649/// \brief Determine whether the addition of the given flag to an Objective-C
4650/// property's attributes will cause a conflict.
Bill Wendling44426052012-12-20 19:22:21 +00004651static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregore6078da2009-11-19 00:14:45 +00004652 // Check if we've already added this flag.
Bill Wendling44426052012-12-20 19:22:21 +00004653 if (Attributes & NewFlag)
Douglas Gregore6078da2009-11-19 00:14:45 +00004654 return true;
4655
Bill Wendling44426052012-12-20 19:22:21 +00004656 Attributes |= NewFlag;
Douglas Gregore6078da2009-11-19 00:14:45 +00004657
4658 // Check for collisions with "readonly".
Bill Wendling44426052012-12-20 19:22:21 +00004659 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4660 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregore6078da2009-11-19 00:14:45 +00004661 return true;
4662
Jordan Rose53cb2f32012-08-20 20:01:13 +00004663 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendling44426052012-12-20 19:22:21 +00004664 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCall31168b02011-06-15 23:02:42 +00004665 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregore6078da2009-11-19 00:14:45 +00004666 ObjCDeclSpec::DQ_PR_copy |
Jordan Rose53cb2f32012-08-20 20:01:13 +00004667 ObjCDeclSpec::DQ_PR_retain |
4668 ObjCDeclSpec::DQ_PR_strong |
4669 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregore6078da2009-11-19 00:14:45 +00004670 if (AssignCopyRetMask &&
4671 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCall31168b02011-06-15 23:02:42 +00004672 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregore6078da2009-11-19 00:14:45 +00004673 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCall31168b02011-06-15 23:02:42 +00004674 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rose53cb2f32012-08-20 20:01:13 +00004675 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4676 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregore6078da2009-11-19 00:14:45 +00004677 return true;
4678
4679 return false;
4680}
4681
Douglas Gregor36029f42009-11-18 23:08:07 +00004682void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroff936354c2009-10-08 21:55:05 +00004683 if (!CodeCompleter)
4684 return;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004685
Bill Wendling44426052012-12-20 19:22:21 +00004686 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroff936354c2009-10-08 21:55:05 +00004687
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004688 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004689 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004690 CodeCompletionContext::CCC_Other);
Steve Naroff936354c2009-10-08 21:55:05 +00004691 Results.EnterNewScope();
Bill Wendling44426052012-12-20 19:22:21 +00004692 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall276321a2010-08-25 06:19:51 +00004693 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendling44426052012-12-20 19:22:21 +00004694 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall276321a2010-08-25 06:19:51 +00004695 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendling44426052012-12-20 19:22:21 +00004696 if (!ObjCPropertyFlagConflicts(Attributes,
John McCall31168b02011-06-15 23:02:42 +00004697 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4698 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendling44426052012-12-20 19:22:21 +00004699 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall276321a2010-08-25 06:19:51 +00004700 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendling44426052012-12-20 19:22:21 +00004701 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall276321a2010-08-25 06:19:51 +00004702 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendling44426052012-12-20 19:22:21 +00004703 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCall31168b02011-06-15 23:02:42 +00004704 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendling44426052012-12-20 19:22:21 +00004705 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall276321a2010-08-25 06:19:51 +00004706 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendling44426052012-12-20 19:22:21 +00004707 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall276321a2010-08-25 06:19:51 +00004708 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendling44426052012-12-20 19:22:21 +00004709 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian1c2d29e2011-06-11 17:14:27 +00004710 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rose53cb2f32012-08-20 20:01:13 +00004711
4712 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall3deb1ad2012-08-21 02:47:43 +00004713 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendling44426052012-12-20 19:22:21 +00004714 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rose53cb2f32012-08-20 20:01:13 +00004715 Results.AddResult(CodeCompletionResult("weak"));
4716
Bill Wendling44426052012-12-20 19:22:21 +00004717 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004718 CodeCompletionBuilder Setter(Results.getAllocator(),
4719 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004720 Setter.AddTypedTextChunk("setter");
Argyrios Kyrtzidis7bbb8812014-02-20 07:55:15 +00004721 Setter.AddTextChunk("=");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004722 Setter.AddPlaceholderChunk("method");
4723 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004724 }
Bill Wendling44426052012-12-20 19:22:21 +00004725 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004726 CodeCompletionBuilder Getter(Results.getAllocator(),
4727 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004728 Getter.AddTypedTextChunk("getter");
Argyrios Kyrtzidis7bbb8812014-02-20 07:55:15 +00004729 Getter.AddTextChunk("=");
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004730 Getter.AddPlaceholderChunk("method");
4731 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor45f83ee2009-11-19 00:01:57 +00004732 }
Steve Naroff936354c2009-10-08 21:55:05 +00004733 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004734 HandleCodeCompleteResults(this, CodeCompleter,
4735 CodeCompletionContext::CCC_Other,
4736 Results.data(),Results.size());
Steve Naroff936354c2009-10-08 21:55:05 +00004737}
Steve Naroffeae65032009-11-07 02:08:14 +00004738
James Dennettf1243872012-06-17 05:33:25 +00004739/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregorc8537c52009-11-19 07:41:15 +00004740/// via code completion.
4741enum ObjCMethodKind {
Dmitri Gribenko4280e5c2012-06-08 23:13:42 +00004742 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4743 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4744 MK_OneArgSelector ///< One-argument selector.
Douglas Gregorc8537c52009-11-19 07:41:15 +00004745};
4746
Douglas Gregor67c692c2010-08-26 15:07:07 +00004747static bool isAcceptableObjCSelector(Selector Sel,
4748 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004749 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004750 bool AllowSameLength = true) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004751 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor67c692c2010-08-26 15:07:07 +00004752 if (NumSelIdents > Sel.getNumArgs())
4753 return false;
4754
4755 switch (WantKind) {
4756 case MK_Any: break;
4757 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4758 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4759 }
4760
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004761 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4762 return false;
4763
Douglas Gregor67c692c2010-08-26 15:07:07 +00004764 for (unsigned I = 0; I != NumSelIdents; ++I)
4765 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4766 return false;
4767
4768 return true;
4769}
4770
Douglas Gregorc8537c52009-11-19 07:41:15 +00004771static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4772 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004773 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004774 bool AllowSameLength = true) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00004775 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004776 AllowSameLength);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004777}
Douglas Gregor1154e272010-09-16 16:06:31 +00004778
4779namespace {
4780 /// \brief A set of selectors, which is used to avoid introducing multiple
4781 /// completions with the same selector into the result set.
4782 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4783}
4784
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004785/// \brief Add all of the Objective-C methods in the given Objective-C
4786/// container to the set of results.
4787///
4788/// The container will be a class, protocol, category, or implementation of
4789/// any of the above. This mether will recurse to include methods from
4790/// the superclasses of classes along with their categories, protocols, and
4791/// implementations.
4792///
4793/// \param Container the container in which we'll look to find methods.
4794///
James Dennett596e4752012-06-14 03:11:41 +00004795/// \param WantInstanceMethods Whether to add instance methods (only); if
4796/// false, this routine will add factory methods (only).
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004797///
4798/// \param CurContext the context in which we're performing the lookup that
4799/// finds methods.
4800///
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004801/// \param AllowSameLength Whether we allow a method to be added to the list
4802/// when it has the same number of parameters as we have selector identifiers.
4803///
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004804/// \param Results the structure into which we'll add results.
4805static void AddObjCMethods(ObjCContainerDecl *Container,
4806 bool WantInstanceMethods,
Douglas Gregorc8537c52009-11-19 07:41:15 +00004807 ObjCMethodKind WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004808 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004809 DeclContext *CurContext,
Douglas Gregor1154e272010-09-16 16:06:31 +00004810 VisitedSelectorSet &Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004811 bool AllowSameLength,
Douglas Gregor416b5752010-08-25 01:08:01 +00004812 ResultBuilder &Results,
4813 bool InOriginalClass = true) {
John McCall276321a2010-08-25 06:19:51 +00004814 typedef CodeCompletionResult Result;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00004815 Container = getContainerDef(Container);
Douglas Gregor41778c32013-01-30 06:58:39 +00004816 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4817 bool isRootClass = IFace && !IFace->getSuperClass();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004818 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
4819 MEnd = Container->meth_end();
4820 M != MEnd; ++M) {
Douglas Gregor41778c32013-01-30 06:58:39 +00004821 // The instance methods on the root class can be messaged via the
4822 // metaclass.
4823 if (M->isInstanceMethod() == WantInstanceMethods ||
4824 (isRootClass && !WantInstanceMethods)) {
Douglas Gregor1b605f72009-11-19 01:08:35 +00004825 // Check whether the selector identifiers we've been given are a
4826 // subset of the identifiers for this particular method.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004827 if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, AllowSameLength))
Douglas Gregor1b605f72009-11-19 01:08:35 +00004828 continue;
Douglas Gregorc8537c52009-11-19 07:41:15 +00004829
David Blaikie2d7c57e2012-04-30 02:36:29 +00004830 if (!Selectors.insert(M->getSelector()))
Douglas Gregor1154e272010-09-16 16:06:31 +00004831 continue;
4832
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00004833 Result R = Result(*M, Results.getBasePriority(*M), 0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004834 R.StartParameter = SelIdents.size();
Douglas Gregorc8537c52009-11-19 07:41:15 +00004835 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor416b5752010-08-25 01:08:01 +00004836 if (!InOriginalClass)
4837 R.Priority += CCD_InBaseClass;
Douglas Gregor1b605f72009-11-19 01:08:35 +00004838 Results.MaybeAddResult(R, CurContext);
4839 }
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004840 }
4841
Douglas Gregorf37c9492010-09-16 15:34:59 +00004842 // Visit the protocols of protocols.
4843 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregore6e48b12012-01-01 19:29:29 +00004844 if (Protocol->hasDefinition()) {
4845 const ObjCList<ObjCProtocolDecl> &Protocols
4846 = Protocol->getReferencedProtocols();
4847 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4848 E = Protocols.end();
4849 I != E; ++I)
4850 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004851 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregore6e48b12012-01-01 19:29:29 +00004852 }
Douglas Gregorf37c9492010-09-16 15:34:59 +00004853 }
4854
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00004855 if (!IFace || !IFace->hasDefinition())
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004856 return;
4857
4858 // Add methods in protocols.
Argyrios Kyrtzidise7f3ef32012-03-13 01:09:41 +00004859 for (ObjCInterfaceDecl::protocol_iterator I = IFace->protocol_begin(),
4860 E = IFace->protocol_end();
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004861 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004862 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004863 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004864
4865 // Add methods in categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00004866 for (ObjCInterfaceDecl::known_categories_iterator
4867 Cat = IFace->known_categories_begin(),
4868 CatEnd = IFace->known_categories_end();
4869 Cat != CatEnd; ++Cat) {
4870 ObjCCategoryDecl *CatDecl = *Cat;
4871
4872 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004873 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004874 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004875
4876 // Add a categories protocol methods.
4877 const ObjCList<ObjCProtocolDecl> &Protocols
4878 = CatDecl->getReferencedProtocols();
4879 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4880 E = Protocols.end();
4881 I != E; ++I)
Douglas Gregorc8537c52009-11-19 07:41:15 +00004882 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004883 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004884 Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004885
4886 // Add methods in category implementations.
4887 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004888 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004889 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004890 Results, InOriginalClass);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004891 }
4892
4893 // Add methods in superclass.
4894 if (IFace->getSuperClass())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004895 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004896 SelIdents, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004897 AllowSameLength, Results, false);
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004898
4899 // Add methods in our implementation, if any.
4900 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregorc8537c52009-11-19 07:41:15 +00004901 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004902 CurContext, Selectors, AllowSameLength,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004903 Results, InOriginalClass);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004904}
4905
4906
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004907void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004908 // Try to find the interface where getters might live.
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004909 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004910 if (!Class) {
4911 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004912 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004913 Class = Category->getClassInterface();
4914
4915 if (!Class)
4916 return;
4917 }
4918
4919 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004920 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004921 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004922 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004923 Results.EnterNewScope();
4924
Douglas Gregor1154e272010-09-16 16:06:31 +00004925 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004926 AddObjCMethods(Class, true, MK_ZeroArgSelector, None, CurContext, Selectors,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004927 /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004928 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004929 HandleCodeCompleteResults(this, CodeCompleter,
4930 CodeCompletionContext::CCC_Other,
4931 Results.data(),Results.size());
Douglas Gregorc8537c52009-11-19 07:41:15 +00004932}
4933
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004934void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregorc8537c52009-11-19 07:41:15 +00004935 // Try to find the interface where setters might live.
4936 ObjCInterfaceDecl *Class
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004937 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004938 if (!Class) {
4939 if (ObjCCategoryDecl *Category
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00004940 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregorc8537c52009-11-19 07:41:15 +00004941 Class = Category->getClassInterface();
4942
4943 if (!Class)
4944 return;
4945 }
4946
4947 // Find all of the potential getters.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004948 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004949 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004950 CodeCompletionContext::CCC_Other);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004951 Results.EnterNewScope();
4952
Douglas Gregor1154e272010-09-16 16:06:31 +00004953 VisitedSelectorSet Selectors;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00004954 AddObjCMethods(Class, true, MK_OneArgSelector, None, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00004955 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregorc8537c52009-11-19 07:41:15 +00004956
4957 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00004958 HandleCodeCompleteResults(this, CodeCompleter,
4959 CodeCompletionContext::CCC_Other,
4960 Results.data(),Results.size());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00004961}
4962
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004963void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
4964 bool IsParameter) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004965 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004966 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00004967 CodeCompletionContext::CCC_Type);
Douglas Gregor99fa2642010-08-24 01:06:58 +00004968 Results.EnterNewScope();
4969
4970 // Add context-sensitive, Objective-C parameter-passing keywords.
4971 bool AddedInOut = false;
4972 if ((DS.getObjCDeclQualifier() &
4973 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
4974 Results.AddResult("in");
4975 Results.AddResult("inout");
4976 AddedInOut = true;
4977 }
4978 if ((DS.getObjCDeclQualifier() &
4979 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
4980 Results.AddResult("out");
4981 if (!AddedInOut)
4982 Results.AddResult("inout");
4983 }
4984 if ((DS.getObjCDeclQualifier() &
4985 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
4986 ObjCDeclSpec::DQ_Oneway)) == 0) {
4987 Results.AddResult("bycopy");
4988 Results.AddResult("byref");
4989 Results.AddResult("oneway");
4990 }
4991
Douglas Gregorf34a6f02011-02-15 22:19:42 +00004992 // If we're completing the return type of an Objective-C method and the
4993 // identifier IBAction refers to a macro, provide a completion item for
4994 // an action, e.g.,
4995 // IBAction)<#selector#>:(id)sender
4996 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
4997 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00004998 CodeCompletionBuilder Builder(Results.getAllocator(),
4999 Results.getCodeCompletionTUInfo(),
5000 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005001 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005002 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005003 Builder.AddPlaceholderChunk("selector");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005004 Builder.AddChunk(CodeCompletionString::CK_Colon);
5005 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005006 Builder.AddTextChunk("id");
Benjamin Kramerdb534a42012-03-26 16:57:36 +00005007 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005008 Builder.AddTextChunk("sender");
5009 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
5010 }
Douglas Gregored1f5972013-01-30 07:11:43 +00005011
5012 // If we're completing the return type, provide 'instancetype'.
5013 if (!IsParameter) {
5014 Results.AddResult(CodeCompletionResult("instancetype"));
5015 }
Douglas Gregorf34a6f02011-02-15 22:19:42 +00005016
Douglas Gregor99fa2642010-08-24 01:06:58 +00005017 // Add various builtin type names and specifiers.
5018 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5019 Results.ExitScope();
5020
5021 // Add the various type names
5022 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5023 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5024 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5025 CodeCompleter->includeGlobals());
5026
5027 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005028 AddMacroResults(PP, Results, false);
Douglas Gregor99fa2642010-08-24 01:06:58 +00005029
5030 HandleCodeCompleteResults(this, CodeCompleter,
5031 CodeCompletionContext::CCC_Type,
5032 Results.data(), Results.size());
5033}
5034
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005035/// \brief When we have an expression with type "id", we may assume
5036/// that it has some more-specific class type based on knowledge of
5037/// common uses of Objective-C. This routine returns that class type,
5038/// or NULL if no better result could be determined.
5039static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregored0b69d2010-09-15 16:23:04 +00005040 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005041 if (!Msg)
5042 return 0;
5043
5044 Selector Sel = Msg->getSelector();
5045 if (Sel.isNull())
5046 return 0;
5047
5048 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5049 if (!Id)
5050 return 0;
5051
5052 ObjCMethodDecl *Method = Msg->getMethodDecl();
5053 if (!Method)
5054 return 0;
5055
5056 // Determine the class that we're sending the message to.
Douglas Gregor9a129192010-04-21 00:45:42 +00005057 ObjCInterfaceDecl *IFace = 0;
5058 switch (Msg->getReceiverKind()) {
5059 case ObjCMessageExpr::Class:
John McCall8b07ec22010-05-15 11:32:37 +00005060 if (const ObjCObjectType *ObjType
5061 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5062 IFace = ObjType->getInterface();
Douglas Gregor9a129192010-04-21 00:45:42 +00005063 break;
5064
5065 case ObjCMessageExpr::Instance: {
5066 QualType T = Msg->getInstanceReceiver()->getType();
5067 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5068 IFace = Ptr->getInterfaceDecl();
5069 break;
5070 }
5071
5072 case ObjCMessageExpr::SuperInstance:
5073 case ObjCMessageExpr::SuperClass:
5074 break;
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005075 }
5076
5077 if (!IFace)
5078 return 0;
5079
5080 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5081 if (Method->isInstanceMethod())
5082 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5083 .Case("retain", IFace)
John McCall31168b02011-06-15 23:02:42 +00005084 .Case("strong", IFace)
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005085 .Case("autorelease", IFace)
5086 .Case("copy", IFace)
5087 .Case("copyWithZone", IFace)
5088 .Case("mutableCopy", IFace)
5089 .Case("mutableCopyWithZone", IFace)
5090 .Case("awakeFromCoder", IFace)
5091 .Case("replacementObjectFromCoder", IFace)
5092 .Case("class", IFace)
5093 .Case("classForCoder", IFace)
5094 .Case("superclass", Super)
5095 .Default(0);
5096
5097 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5098 .Case("new", IFace)
5099 .Case("alloc", IFace)
5100 .Case("allocWithZone", IFace)
5101 .Case("class", IFace)
5102 .Case("superclass", Super)
5103 .Default(0);
5104}
5105
Douglas Gregor6fc04132010-08-27 15:10:57 +00005106// Add a special completion for a message send to "super", which fills in the
5107// most likely case of forwarding all of our arguments to the superclass
5108// function.
5109///
5110/// \param S The semantic analysis object.
5111///
Dmitri Gribenkoadba9be2012-08-23 17:58:28 +00005112/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor6fc04132010-08-27 15:10:57 +00005113/// the "super" keyword. Otherwise, we just need to provide the arguments.
5114///
5115/// \param SelIdents The identifiers in the selector that have already been
5116/// provided as arguments for a send to "super".
5117///
Douglas Gregor6fc04132010-08-27 15:10:57 +00005118/// \param Results The set of results to augment.
5119///
5120/// \returns the Objective-C method declaration that would be invoked by
5121/// this "super" completion. If NULL, no completion was added.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005122static ObjCMethodDecl *AddSuperSendCompletion(
5123 Sema &S, bool NeedSuperKeyword,
5124 ArrayRef<IdentifierInfo *> SelIdents,
5125 ResultBuilder &Results) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005126 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5127 if (!CurMethod)
5128 return 0;
5129
5130 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5131 if (!Class)
5132 return 0;
5133
5134 // Try to find a superclass method with the same selector.
5135 ObjCMethodDecl *SuperMethod = 0;
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005136 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5137 // Check in the class
Douglas Gregor6fc04132010-08-27 15:10:57 +00005138 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5139 CurMethod->isInstanceMethod());
5140
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005141 // Check in categories or class extensions.
5142 if (!SuperMethod) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005143 for (ObjCInterfaceDecl::known_categories_iterator
5144 Cat = Class->known_categories_begin(),
5145 CatEnd = Class->known_categories_end();
5146 Cat != CatEnd; ++Cat) {
5147 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005148 CurMethod->isInstanceMethod())))
5149 break;
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005150 }
Douglas Gregorb5f1e462011-02-16 00:51:18 +00005151 }
5152 }
5153
Douglas Gregor6fc04132010-08-27 15:10:57 +00005154 if (!SuperMethod)
5155 return 0;
5156
5157 // Check whether the superclass method has the same signature.
5158 if (CurMethod->param_size() != SuperMethod->param_size() ||
5159 CurMethod->isVariadic() != SuperMethod->isVariadic())
5160 return 0;
5161
5162 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5163 CurPEnd = CurMethod->param_end(),
5164 SuperP = SuperMethod->param_begin();
5165 CurP != CurPEnd; ++CurP, ++SuperP) {
5166 // Make sure the parameter types are compatible.
5167 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5168 (*SuperP)->getType()))
5169 return 0;
5170
5171 // Make sure we have a parameter name to forward!
5172 if (!(*CurP)->getIdentifier())
5173 return 0;
5174 }
5175
5176 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005177 CodeCompletionBuilder Builder(Results.getAllocator(),
5178 Results.getCodeCompletionTUInfo());
Douglas Gregor6fc04132010-08-27 15:10:57 +00005179
5180 // Give this completion a return type.
Douglas Gregor75acd922011-09-27 23:30:47 +00005181 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5182 Builder);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005183
5184 // If we need the "super" keyword, add it (plus some spacing).
5185 if (NeedSuperKeyword) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005186 Builder.AddTypedTextChunk("super");
5187 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005188 }
5189
5190 Selector Sel = CurMethod->getSelector();
5191 if (Sel.isUnarySelector()) {
5192 if (NeedSuperKeyword)
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005193 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005194 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005195 else
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005196 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005197 Sel.getNameForSlot(0)));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005198 } else {
5199 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5200 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005201 if (I > SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005202 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005203
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005204 if (I < SelIdents.size())
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005205 Builder.AddInformativeChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005206 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005207 Sel.getNameForSlot(I) + ":"));
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005208 else if (NeedSuperKeyword || I > SelIdents.size()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005209 Builder.AddTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005210 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005211 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005212 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005213 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005214 } else {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005215 Builder.AddTypedTextChunk(
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005216 Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005217 Sel.getNameForSlot(I) + ":"));
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005218 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005219 (*CurP)->getIdentifier()->getName()));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005220 }
5221 }
5222 }
5223
Douglas Gregor78254c82012-03-27 23:34:16 +00005224 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5225 CCP_SuperCompletion));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005226 return SuperMethod;
5227}
5228
Douglas Gregora817a192010-05-27 23:06:34 +00005229void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall276321a2010-08-25 06:19:51 +00005230 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005231 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005232 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005233 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005234 getLangOpts().CPlusPlus11
Douglas Gregord8c61782012-02-15 15:34:24 +00005235 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5236 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregora817a192010-05-27 23:06:34 +00005237
Douglas Gregora817a192010-05-27 23:06:34 +00005238 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5239 Results.EnterNewScope();
Douglas Gregor39982192010-08-15 06:18:01 +00005240 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5241 CodeCompleter->includeGlobals());
Douglas Gregora817a192010-05-27 23:06:34 +00005242
5243 // If we are in an Objective-C method inside a class that has a superclass,
5244 // add "super" as an option.
5245 if (ObjCMethodDecl *Method = getCurMethodDecl())
5246 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor6fc04132010-08-27 15:10:57 +00005247 if (Iface->getSuperClass()) {
Douglas Gregora817a192010-05-27 23:06:34 +00005248 Results.AddResult(Result("super"));
Douglas Gregor6fc04132010-08-27 15:10:57 +00005249
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005250 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, None, Results);
Douglas Gregor6fc04132010-08-27 15:10:57 +00005251 }
Douglas Gregora817a192010-05-27 23:06:34 +00005252
Richard Smith2bf7fdb2013-01-02 11:42:31 +00005253 if (getLangOpts().CPlusPlus11)
Douglas Gregord8c61782012-02-15 15:34:24 +00005254 addThisCompletion(*this, Results);
5255
Douglas Gregora817a192010-05-27 23:06:34 +00005256 Results.ExitScope();
5257
5258 if (CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00005259 AddMacroResults(PP, Results, false);
Douglas Gregor50832e02010-09-20 22:39:41 +00005260 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005261 Results.data(), Results.size());
Douglas Gregora817a192010-05-27 23:06:34 +00005262
5263}
5264
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005265void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005266 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005267 bool AtArgumentExpression) {
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005268 ObjCInterfaceDecl *CDecl = 0;
5269 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5270 // Figure out which interface we're in.
5271 CDecl = CurMethod->getClassInterface();
5272 if (!CDecl)
5273 return;
5274
5275 // Find the superclass of this class.
5276 CDecl = CDecl->getSuperClass();
5277 if (!CDecl)
5278 return;
5279
5280 if (CurMethod->isInstanceMethod()) {
5281 // We are inside an instance method, which means that the message
5282 // send [super ...] is actually calling an instance method on the
Douglas Gregor392a84b2010-10-13 21:24:53 +00005283 // current object.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005284 return CodeCompleteObjCInstanceMessage(S, 0, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005285 AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005286 CDecl);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005287 }
5288
5289 // Fall through to send to the superclass in CDecl.
5290 } else {
5291 // "super" may be the name of a type or variable. Figure out which
5292 // it is.
Argyrios Kyrtzidis3e56dd42013-03-14 22:56:43 +00005293 IdentifierInfo *Super = getSuperIdentifier();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005294 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5295 LookupOrdinaryName);
5296 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5297 // "super" names an interface. Use it.
5298 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCall8b07ec22010-05-15 11:32:37 +00005299 if (const ObjCObjectType *Iface
5300 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5301 CDecl = Iface->getInterface();
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005302 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5303 // "super" names an unresolved type; we can't be more specific.
5304 } else {
5305 // Assume that "super" names some kind of value and parse that way.
5306 CXXScopeSpec SS;
Abramo Bagnara7945c982012-01-27 09:46:47 +00005307 SourceLocation TemplateKWLoc;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005308 UnqualifiedId id;
5309 id.setIdentifier(Super, SuperLoc);
Abramo Bagnara7945c982012-01-27 09:46:47 +00005310 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5311 false, false);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005312 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005313 SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005314 AtArgumentExpression);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005315 }
5316
5317 // Fall through
5318 }
5319
John McCallba7bf592010-08-24 05:47:05 +00005320 ParsedType Receiver;
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005321 if (CDecl)
John McCallba7bf592010-08-24 05:47:05 +00005322 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005323 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005324 AtArgumentExpression,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005325 /*IsSuper=*/true);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005326}
5327
Douglas Gregor74661272010-09-21 00:03:25 +00005328/// \brief Given a set of code-completion results for the argument of a message
5329/// send, determine the preferred type (if any) for that argument expression.
5330static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5331 unsigned NumSelIdents) {
5332 typedef CodeCompletionResult Result;
5333 ASTContext &Context = Results.getSema().Context;
5334
5335 QualType PreferredType;
5336 unsigned BestPriority = CCP_Unlikely * 2;
5337 Result *ResultsData = Results.data();
5338 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5339 Result &R = ResultsData[I];
5340 if (R.Kind == Result::RK_Declaration &&
5341 isa<ObjCMethodDecl>(R.Declaration)) {
5342 if (R.Priority <= BestPriority) {
Dmitri Gribenkofe0483d2013-01-23 17:21:11 +00005343 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregor74661272010-09-21 00:03:25 +00005344 if (NumSelIdents <= Method->param_size()) {
5345 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
5346 ->getType();
5347 if (R.Priority < BestPriority || PreferredType.isNull()) {
5348 BestPriority = R.Priority;
5349 PreferredType = MyPreferredType;
5350 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5351 MyPreferredType)) {
5352 PreferredType = QualType();
5353 }
5354 }
5355 }
5356 }
5357 }
5358
5359 return PreferredType;
5360}
5361
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005362static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5363 ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005364 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005365 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005366 bool IsSuper,
5367 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005368 typedef CodeCompletionResult Result;
Douglas Gregor8ce33212009-11-17 17:59:40 +00005369 ObjCInterfaceDecl *CDecl = 0;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005370
Douglas Gregor8ce33212009-11-17 17:59:40 +00005371 // If the given name refers to an interface type, retrieve the
5372 // corresponding declaration.
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005373 if (Receiver) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005374 QualType T = SemaRef.GetTypeFromParser(Receiver, 0);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005375 if (!T.isNull())
John McCall8b07ec22010-05-15 11:32:37 +00005376 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5377 CDecl = Interface->getInterface();
Douglas Gregor8ce33212009-11-17 17:59:40 +00005378 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005379
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005380 // Add all of the factory methods in this Objective-C class, its protocols,
5381 // superclasses, categories, implementation, etc.
Steve Naroffeae65032009-11-07 02:08:14 +00005382 Results.EnterNewScope();
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005383
Douglas Gregor6fc04132010-08-27 15:10:57 +00005384 // If this is a send-to-super, try to add the special "super" send
5385 // completion.
5386 if (IsSuper) {
5387 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005388 = AddSuperSendCompletion(SemaRef, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005389 Results.Ignore(SuperMethod);
5390 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005391
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005392 // If we're inside an Objective-C method definition, prefer its selector to
5393 // others.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005394 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005395 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005396
Douglas Gregor1154e272010-09-16 16:06:31 +00005397 VisitedSelectorSet Selectors;
Douglas Gregor6285f752010-04-06 16:40:00 +00005398 if (CDecl)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005399 AddObjCMethods(CDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005400 SemaRef.CurContext, Selectors, AtArgumentExpression,
5401 Results);
Douglas Gregor0c78ad92010-04-21 19:57:20 +00005402 else {
Douglas Gregor6285f752010-04-06 16:40:00 +00005403 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005404
Douglas Gregord720daf2010-04-06 17:30:22 +00005405 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005406 // pool from the AST file.
Axel Naumanndd433f02012-10-18 19:05:02 +00005407 if (SemaRef.getExternalSource()) {
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005408 for (uint32_t I = 0,
Axel Naumanndd433f02012-10-18 19:05:02 +00005409 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall75b960e2010-06-01 09:23:16 +00005410 I != N; ++I) {
Axel Naumanndd433f02012-10-18 19:05:02 +00005411 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005412 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005413 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005414
5415 SemaRef.ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005416 }
5417 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005418
5419 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5420 MEnd = SemaRef.MethodPool.end();
Sebastian Redl75d8a322010-08-02 23:18:59 +00005421 M != MEnd; ++M) {
5422 for (ObjCMethodList *MethList = &M->second.second;
5423 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005424 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005425 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005426 continue;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005427
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005428 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005429 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005430 R.AllParametersAreInformative = false;
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005431 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor6285f752010-04-06 16:40:00 +00005432 }
5433 }
5434 }
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005435
5436 Results.ExitScope();
5437}
Douglas Gregor6285f752010-04-06 16:40:00 +00005438
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005439void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005440 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005441 bool AtArgumentExpression,
Douglas Gregorbfcea8b2010-09-16 15:14:18 +00005442 bool IsSuper) {
Douglas Gregor63745d52011-07-21 01:05:26 +00005443
5444 QualType T = this->GetTypeFromParser(Receiver);
5445
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005446 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005447 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005448 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005449 T, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005450
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005451 AddClassMessageCompletions(*this, S, Receiver, SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005452 AtArgumentExpression, IsSuper, Results);
Douglas Gregor74661272010-09-21 00:03:25 +00005453
5454 // If we're actually at the argument expression (rather than prior to the
5455 // selector), we're actually performing code completion for an expression.
5456 // Determine whether we have a single, best method. If so, we can
5457 // code-complete the expression using the corresponding parameter type as
5458 // our preferred type, improving completion results.
5459 if (AtArgumentExpression) {
5460 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005461 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005462 if (PreferredType.isNull())
5463 CodeCompleteOrdinaryName(S, PCC_Expression);
5464 else
5465 CodeCompleteExpression(S, PreferredType);
5466 return;
5467 }
5468
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005469 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005470 Results.getCompletionContext(),
Douglas Gregor6fc04132010-08-27 15:10:57 +00005471 Results.data(), Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005472}
5473
Richard Trieu2bd04012011-09-09 02:00:50 +00005474void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005475 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorf86e4da2010-09-20 23:34:21 +00005476 bool AtArgumentExpression,
Douglas Gregor392a84b2010-10-13 21:24:53 +00005477 ObjCInterfaceDecl *Super) {
John McCall276321a2010-08-25 06:19:51 +00005478 typedef CodeCompletionResult Result;
Steve Naroffeae65032009-11-07 02:08:14 +00005479
5480 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffeae65032009-11-07 02:08:14 +00005481
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005482 // If necessary, apply function/array conversion to the receiver.
5483 // C99 6.7.5.3p[7,8].
John Wiegley01296292011-04-08 18:41:53 +00005484 if (RecExpr) {
5485 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5486 if (Conv.isInvalid()) // conversion failed. bail.
5487 return;
5488 RecExpr = Conv.take();
5489 }
Douglas Gregor392a84b2010-10-13 21:24:53 +00005490 QualType ReceiverType = RecExpr? RecExpr->getType()
5491 : Super? Context.getObjCObjectPointerType(
5492 Context.getObjCInterfaceType(Super))
5493 : Context.getObjCIdType();
Steve Naroffeae65032009-11-07 02:08:14 +00005494
Douglas Gregordc520b02010-11-08 21:12:30 +00005495 // If we're messaging an expression with type "id" or "Class", check
5496 // whether we know something special about the receiver that allows
5497 // us to assume a more-specific receiver type.
Anders Carlsson382ba412014-02-28 19:07:22 +00005498 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType()) {
Douglas Gregordc520b02010-11-08 21:12:30 +00005499 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5500 if (ReceiverType->isObjCClassType())
5501 return CodeCompleteObjCClassMessage(S,
5502 ParsedType::make(Context.getObjCInterfaceType(IFace)),
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005503 SelIdents,
Douglas Gregordc520b02010-11-08 21:12:30 +00005504 AtArgumentExpression, Super);
5505
5506 ReceiverType = Context.getObjCObjectPointerType(
5507 Context.getObjCInterfaceType(IFace));
5508 }
Anders Carlsson382ba412014-02-28 19:07:22 +00005509 } else if (RecExpr && getLangOpts().CPlusPlus) {
5510 ExprResult Conv = PerformContextuallyConvertToObjCPointer(RecExpr);
5511 if (Conv.isUsable()) {
5512 RecExpr = Conv.take();
5513 ReceiverType = RecExpr->getType();
5514 }
5515 }
Douglas Gregordc520b02010-11-08 21:12:30 +00005516
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005517 // Build the set of methods we can see.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005518 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005519 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor63745d52011-07-21 01:05:26 +00005520 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005521 ReceiverType, SelIdents));
Douglas Gregor63745d52011-07-21 01:05:26 +00005522
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005523 Results.EnterNewScope();
Douglas Gregor9d2ddb22010-04-06 19:22:33 +00005524
Douglas Gregor6fc04132010-08-27 15:10:57 +00005525 // If this is a send-to-super, try to add the special "super" send
5526 // completion.
Douglas Gregor392a84b2010-10-13 21:24:53 +00005527 if (Super) {
Douglas Gregor6fc04132010-08-27 15:10:57 +00005528 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005529 = AddSuperSendCompletion(*this, false, SelIdents, Results))
Douglas Gregor6fc04132010-08-27 15:10:57 +00005530 Results.Ignore(SuperMethod);
5531 }
5532
Douglas Gregorc2cb2e22010-08-27 15:29:55 +00005533 // If we're inside an Objective-C method definition, prefer its selector to
5534 // others.
5535 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5536 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorbab2b3c2009-11-17 23:22:23 +00005537
Douglas Gregor1154e272010-09-16 16:06:31 +00005538 // Keep track of the selectors we've already added.
5539 VisitedSelectorSet Selectors;
5540
Douglas Gregora3329fa2009-11-18 00:06:18 +00005541 // Handle messages to Class. This really isn't a message to an instance
5542 // method, so we treat it the same way we would treat a message send to a
5543 // class method.
5544 if (ReceiverType->isObjCClassType() ||
5545 ReceiverType->isObjCQualifiedClassType()) {
5546 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5547 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005548 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005549 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005550 }
5551 }
5552 // Handle messages to a qualified ID ("id<foo>").
5553 else if (const ObjCObjectPointerType *QualID
5554 = ReceiverType->getAsObjCQualifiedIdType()) {
5555 // Search protocols for instance methods.
5556 for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(),
5557 E = QualID->qual_end();
5558 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005559 AddObjCMethods(*I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005560 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005561 }
5562 // Handle messages to a pointer to interface type.
5563 else if (const ObjCObjectPointerType *IFacePtr
5564 = ReceiverType->getAsObjCInterfacePointerType()) {
5565 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregorc8537c52009-11-19 07:41:15 +00005566 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005567 CurContext, Selectors, AtArgumentExpression,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005568 Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005569
5570 // Search protocols for instance methods.
5571 for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(),
5572 E = IFacePtr->qual_end();
5573 I != E; ++I)
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005574 AddObjCMethods(*I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorb4a7c032010-11-17 21:36:08 +00005575 Selectors, AtArgumentExpression, Results);
Douglas Gregora3329fa2009-11-18 00:06:18 +00005576 }
Douglas Gregor6285f752010-04-06 16:40:00 +00005577 // Handle messages to "id".
5578 else if (ReceiverType->isObjCIdType()) {
Douglas Gregord720daf2010-04-06 17:30:22 +00005579 // We're messaging "id", so provide all instance methods we know
5580 // about as code-completion results.
5581
5582 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00005583 // pool from the AST file.
Douglas Gregord720daf2010-04-06 17:30:22 +00005584 if (ExternalSource) {
John McCall75b960e2010-06-01 09:23:16 +00005585 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5586 I != N; ++I) {
5587 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00005588 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregord720daf2010-04-06 17:30:22 +00005589 continue;
5590
Sebastian Redl75d8a322010-08-02 23:18:59 +00005591 ReadMethodPool(Sel);
Douglas Gregord720daf2010-04-06 17:30:22 +00005592 }
5593 }
5594
Sebastian Redl75d8a322010-08-02 23:18:59 +00005595 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5596 MEnd = MethodPool.end();
5597 M != MEnd; ++M) {
5598 for (ObjCMethodList *MethList = &M->second.first;
5599 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00005600 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005601 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor6285f752010-04-06 16:40:00 +00005602 continue;
Douglas Gregor1154e272010-09-16 16:06:31 +00005603
5604 if (!Selectors.insert(MethList->Method->getSelector()))
5605 continue;
5606
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005607 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005608 R.StartParameter = SelIdents.size();
Douglas Gregor6285f752010-04-06 16:40:00 +00005609 R.AllParametersAreInformative = false;
5610 Results.MaybeAddResult(R, CurContext);
5611 }
5612 }
5613 }
Steve Naroffeae65032009-11-07 02:08:14 +00005614 Results.ExitScope();
Douglas Gregor74661272010-09-21 00:03:25 +00005615
5616
5617 // If we're actually at the argument expression (rather than prior to the
5618 // selector), we're actually performing code completion for an expression.
5619 // Determine whether we have a single, best method. If so, we can
5620 // code-complete the expression using the corresponding parameter type as
5621 // our preferred type, improving completion results.
5622 if (AtArgumentExpression) {
5623 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005624 SelIdents.size());
Douglas Gregor74661272010-09-21 00:03:25 +00005625 if (PreferredType.isNull())
5626 CodeCompleteOrdinaryName(S, PCC_Expression);
5627 else
5628 CodeCompleteExpression(S, PreferredType);
5629 return;
5630 }
5631
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005632 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor63745d52011-07-21 01:05:26 +00005633 Results.getCompletionContext(),
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005634 Results.data(),Results.size());
Steve Naroffeae65032009-11-07 02:08:14 +00005635}
Douglas Gregorbaf69612009-11-18 04:19:12 +00005636
Douglas Gregor68762e72010-08-23 21:17:50 +00005637void Sema::CodeCompleteObjCForCollection(Scope *S,
5638 DeclGroupPtrTy IterationVar) {
5639 CodeCompleteExpressionData Data;
5640 Data.ObjCCollection = true;
5641
5642 if (IterationVar.getAsOpaquePtr()) {
Serge Pavlov9ddb76e2013-08-27 13:15:56 +00005643 DeclGroupRef DG = IterationVar.get();
Douglas Gregor68762e72010-08-23 21:17:50 +00005644 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5645 if (*I)
5646 Data.IgnoreDecls.push_back(*I);
5647 }
5648 }
5649
5650 CodeCompleteExpression(S, Data);
5651}
5652
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005653void Sema::CodeCompleteObjCSelector(Scope *S,
5654 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor67c692c2010-08-26 15:07:07 +00005655 // If we have an external source, load the entire class method
5656 // pool from the AST file.
5657 if (ExternalSource) {
5658 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5659 I != N; ++I) {
5660 Selector Sel = ExternalSource->GetExternalSelector(I);
5661 if (Sel.isNull() || MethodPool.count(Sel))
5662 continue;
5663
5664 ReadMethodPool(Sel);
5665 }
5666 }
5667
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005668 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005669 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005670 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor67c692c2010-08-26 15:07:07 +00005671 Results.EnterNewScope();
5672 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5673 MEnd = MethodPool.end();
5674 M != MEnd; ++M) {
5675
5676 Selector Sel = M->first;
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005677 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents))
Douglas Gregor67c692c2010-08-26 15:07:07 +00005678 continue;
5679
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005680 CodeCompletionBuilder Builder(Results.getAllocator(),
5681 Results.getCodeCompletionTUInfo());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005682 if (Sel.isUnarySelector()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005683 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00005684 Sel.getNameForSlot(0)));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005685 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005686 continue;
5687 }
5688
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005689 std::string Accumulator;
Douglas Gregor67c692c2010-08-26 15:07:07 +00005690 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00005691 if (I == SelIdents.size()) {
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005692 if (!Accumulator.empty()) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005693 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005694 Accumulator));
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005695 Accumulator.clear();
5696 }
5697 }
5698
Benjamin Kramer632500c2011-07-26 16:59:25 +00005699 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor9ac1ad12010-08-26 16:46:39 +00005700 Accumulator += ':';
Douglas Gregor67c692c2010-08-26 15:07:07 +00005701 }
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00005702 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005703 Results.AddResult(Builder.TakeString());
Douglas Gregor67c692c2010-08-26 15:07:07 +00005704 }
5705 Results.ExitScope();
5706
5707 HandleCodeCompleteResults(this, CodeCompleter,
5708 CodeCompletionContext::CCC_SelectorName,
5709 Results.data(), Results.size());
5710}
5711
Douglas Gregorbaf69612009-11-18 04:19:12 +00005712/// \brief Add all of the protocol declarations that we find in the given
5713/// (translation unit) context.
5714static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005715 bool OnlyForwardDeclarations,
Douglas Gregorbaf69612009-11-18 04:19:12 +00005716 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005717 typedef CodeCompletionResult Result;
Douglas Gregorbaf69612009-11-18 04:19:12 +00005718
Aaron Ballman629afae2014-03-07 19:56:05 +00005719 for (const auto *D : Ctx->decls()) {
Douglas Gregorbaf69612009-11-18 04:19:12 +00005720 // Record any protocols we find.
Aaron Ballman629afae2014-03-07 19:56:05 +00005721 if (const auto *Proto = dyn_cast<ObjCProtocolDecl>(D))
Douglas Gregore6e48b12012-01-01 19:29:29 +00005722 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005723 Results.AddResult(Result(Proto, Results.getBasePriority(Proto), 0),
5724 CurContext, 0, false);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005725 }
5726}
5727
5728void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5729 unsigned NumProtocols) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005730 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005731 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005732 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005733
Douglas Gregora3b23b02010-12-09 21:44:02 +00005734 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5735 Results.EnterNewScope();
5736
5737 // Tell the result set to ignore all of the protocols we have
5738 // already seen.
5739 // FIXME: This doesn't work when caching code-completion results.
5740 for (unsigned I = 0; I != NumProtocols; ++I)
5741 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5742 Protocols[I].second))
5743 Results.Ignore(Protocol);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005744
Douglas Gregora3b23b02010-12-09 21:44:02 +00005745 // Add all protocols.
5746 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5747 Results);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005748
Douglas Gregora3b23b02010-12-09 21:44:02 +00005749 Results.ExitScope();
5750 }
5751
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005752 HandleCodeCompleteResults(this, CodeCompleter,
5753 CodeCompletionContext::CCC_ObjCProtocolName,
5754 Results.data(),Results.size());
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005755}
5756
5757void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005758 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005759 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005760 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor5b4671c2009-11-18 04:49:41 +00005761
Douglas Gregora3b23b02010-12-09 21:44:02 +00005762 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5763 Results.EnterNewScope();
5764
5765 // Add all protocols.
5766 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5767 Results);
Douglas Gregorbaf69612009-11-18 04:19:12 +00005768
Douglas Gregora3b23b02010-12-09 21:44:02 +00005769 Results.ExitScope();
5770 }
5771
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005772 HandleCodeCompleteResults(this, CodeCompleter,
5773 CodeCompletionContext::CCC_ObjCProtocolName,
5774 Results.data(),Results.size());
Douglas Gregorbaf69612009-11-18 04:19:12 +00005775}
Douglas Gregor49c22a72009-11-18 16:26:39 +00005776
5777/// \brief Add all of the Objective-C interface declarations that we find in
5778/// the given (translation unit) context.
5779static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5780 bool OnlyForwardDeclarations,
5781 bool OnlyUnimplemented,
5782 ResultBuilder &Results) {
John McCall276321a2010-08-25 06:19:51 +00005783 typedef CodeCompletionResult Result;
Douglas Gregor49c22a72009-11-18 16:26:39 +00005784
Aaron Ballman629afae2014-03-07 19:56:05 +00005785 for (const auto *D : Ctx->decls()) {
Douglas Gregor1c283312010-08-11 12:19:30 +00005786 // Record any interfaces we find.
Aaron Ballman629afae2014-03-07 19:56:05 +00005787 if (const auto *Class = dyn_cast<ObjCInterfaceDecl>(D))
Douglas Gregordc9166c2011-12-15 20:29:51 +00005788 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregor1c283312010-08-11 12:19:30 +00005789 (!OnlyUnimplemented || !Class->getImplementation()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005790 Results.AddResult(Result(Class, Results.getBasePriority(Class), 0),
5791 CurContext, 0, false);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005792 }
5793}
5794
5795void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005796 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005797 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005798 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005799 Results.EnterNewScope();
5800
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005801 if (CodeCompleter->includeGlobals()) {
5802 // Add all classes.
5803 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5804 false, Results);
5805 }
5806
Douglas Gregor49c22a72009-11-18 16:26:39 +00005807 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005808
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005809 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005810 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005811 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005812}
5813
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005814void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5815 SourceLocation ClassNameLoc) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005816 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005817 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005818 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005819 Results.EnterNewScope();
5820
5821 // Make sure that we ignore the class we're currently defining.
5822 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005823 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005824 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor49c22a72009-11-18 16:26:39 +00005825 Results.Ignore(CurClass);
5826
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005827 if (CodeCompleter->includeGlobals()) {
5828 // Add all classes.
5829 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5830 false, Results);
5831 }
5832
Douglas Gregor49c22a72009-11-18 16:26:39 +00005833 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005834
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005835 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005836 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005837 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005838}
5839
5840void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005841 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005842 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005843 CodeCompletionContext::CCC_Other);
Douglas Gregor49c22a72009-11-18 16:26:39 +00005844 Results.EnterNewScope();
5845
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005846 if (CodeCompleter->includeGlobals()) {
5847 // Add all unimplemented classes.
5848 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5849 true, Results);
5850 }
5851
Douglas Gregor49c22a72009-11-18 16:26:39 +00005852 Results.ExitScope();
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005853
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005854 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor2c595ad2011-07-30 06:55:39 +00005855 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005856 Results.data(),Results.size());
Douglas Gregor49c22a72009-11-18 16:26:39 +00005857}
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005858
5859void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005860 IdentifierInfo *ClassName,
5861 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005862 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005863
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005864 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005865 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005866 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005867
5868 // Ignore any categories we find that have already been implemented by this
5869 // interface.
5870 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5871 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005872 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005873 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
5874 for (ObjCInterfaceDecl::visible_categories_iterator
5875 Cat = Class->visible_categories_begin(),
5876 CatEnd = Class->visible_categories_end();
5877 Cat != CatEnd; ++Cat) {
5878 CategoryNames.insert(Cat->getIdentifier());
5879 }
5880 }
5881
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005882 // Add all of the categories we know about.
5883 Results.EnterNewScope();
5884 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
Aaron Ballman629afae2014-03-07 19:56:05 +00005885 for (const auto *D : TU->decls())
5886 if (const auto *Category = dyn_cast<ObjCCategoryDecl>(D))
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005887 if (CategoryNames.insert(Category->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005888 Results.AddResult(Result(Category, Results.getBasePriority(Category),0),
5889 CurContext, 0, false);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005890 Results.ExitScope();
5891
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005892 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005893 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005894 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005895}
5896
5897void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005898 IdentifierInfo *ClassName,
5899 SourceLocation ClassNameLoc) {
John McCall276321a2010-08-25 06:19:51 +00005900 typedef CodeCompletionResult Result;
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005901
5902 // Find the corresponding interface. If we couldn't find the interface, the
5903 // program itself is ill-formed. However, we'll try to be helpful still by
5904 // providing the list of all of the categories we know about.
5905 NamedDecl *CurClass
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005906 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005907 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5908 if (!Class)
Douglas Gregorb2ccf012010-04-15 22:33:43 +00005909 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005910
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005911 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005912 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor21325842011-07-07 16:03:39 +00005913 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005914
5915 // Add all of the categories that have have corresponding interface
5916 // declarations in this class and any of its superclasses, except for
5917 // already-implemented categories in the class itself.
5918 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5919 Results.EnterNewScope();
5920 bool IgnoreImplemented = true;
5921 while (Class) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005922 for (ObjCInterfaceDecl::visible_categories_iterator
5923 Cat = Class->visible_categories_begin(),
5924 CatEnd = Class->visible_categories_end();
5925 Cat != CatEnd; ++Cat) {
5926 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
5927 CategoryNames.insert(Cat->getIdentifier()))
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00005928 Results.AddResult(Result(*Cat, Results.getBasePriority(*Cat), 0),
5929 CurContext, 0, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00005930 }
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005931
5932 Class = Class->getSuperClass();
5933 IgnoreImplemented = false;
5934 }
5935 Results.ExitScope();
5936
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005937 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor21325842011-07-07 16:03:39 +00005938 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005939 Results.data(),Results.size());
Douglas Gregor5d34fd32009-11-18 19:08:43 +00005940}
Douglas Gregor5d649882009-11-18 22:32:06 +00005941
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005942void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005943 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005944 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005945 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00005946
5947 // Figure out where this @synthesize lives.
5948 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005949 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00005950 if (!Container ||
5951 (!isa<ObjCImplementationDecl>(Container) &&
5952 !isa<ObjCCategoryImplDecl>(Container)))
5953 return;
5954
5955 // Ignore any properties that have already been implemented.
Douglas Gregor9b4f3702012-06-12 13:44:08 +00005956 Container = getContainerDef(Container);
Aaron Ballman629afae2014-03-07 19:56:05 +00005957 for (const auto *D : Container->decls())
5958 if (const auto *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(D))
Douglas Gregor5d649882009-11-18 22:32:06 +00005959 Results.Ignore(PropertyImpl->getPropertyDecl());
5960
5961 // Add any properties that we find.
Douglas Gregorb888acf2010-12-09 23:01:55 +00005962 AddedPropertiesSet AddedProperties;
Douglas Gregor5d649882009-11-18 22:32:06 +00005963 Results.EnterNewScope();
5964 if (ObjCImplementationDecl *ClassImpl
5965 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor95147142011-05-05 15:50:42 +00005966 AddObjCProperties(ClassImpl->getClassInterface(), false,
5967 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregorb888acf2010-12-09 23:01:55 +00005968 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00005969 else
5970 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor95147142011-05-05 15:50:42 +00005971 false, /*AllowNullaryMethods=*/false, CurContext,
5972 AddedProperties, Results);
Douglas Gregor5d649882009-11-18 22:32:06 +00005973 Results.ExitScope();
5974
Douglas Gregor00c37ef2010-08-11 21:23:17 +00005975 HandleCodeCompleteResults(this, CodeCompleter,
5976 CodeCompletionContext::CCC_Other,
5977 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00005978}
5979
5980void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005981 IdentifierInfo *PropertyName) {
John McCall276321a2010-08-25 06:19:51 +00005982 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005983 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00005984 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00005985 CodeCompletionContext::CCC_Other);
Douglas Gregor5d649882009-11-18 22:32:06 +00005986
5987 // Figure out where this @synthesize lives.
5988 ObjCContainerDecl *Container
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00005989 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor5d649882009-11-18 22:32:06 +00005990 if (!Container ||
5991 (!isa<ObjCImplementationDecl>(Container) &&
5992 !isa<ObjCCategoryImplDecl>(Container)))
5993 return;
5994
5995 // Figure out which interface we're looking into.
5996 ObjCInterfaceDecl *Class = 0;
5997 if (ObjCImplementationDecl *ClassImpl
5998 = dyn_cast<ObjCImplementationDecl>(Container))
5999 Class = ClassImpl->getClassInterface();
6000 else
6001 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
6002 ->getClassInterface();
6003
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006004 // Determine the type of the property we're synthesizing.
6005 QualType PropertyType = Context.getObjCIdType();
6006 if (Class) {
6007 if (ObjCPropertyDecl *Property
6008 = Class->FindPropertyDeclaration(PropertyName)) {
6009 PropertyType
6010 = Property->getType().getNonReferenceType().getUnqualifiedType();
6011
6012 // Give preference to ivars
6013 Results.setPreferredType(PropertyType);
6014 }
6015 }
6016
Douglas Gregor5d649882009-11-18 22:32:06 +00006017 // Add all of the instance variables in this class and its superclasses.
6018 Results.EnterNewScope();
Douglas Gregor331faa02011-04-18 14:13:53 +00006019 bool SawSimilarlyNamedIvar = false;
6020 std::string NameWithPrefix;
6021 NameWithPrefix += '_';
Benjamin Kramer632500c2011-07-26 16:59:25 +00006022 NameWithPrefix += PropertyName->getName();
Douglas Gregor331faa02011-04-18 14:13:53 +00006023 std::string NameWithSuffix = PropertyName->getName().str();
6024 NameWithSuffix += '_';
Douglas Gregor5d649882009-11-18 22:32:06 +00006025 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006026 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
6027 Ivar = Ivar->getNextIvar()) {
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00006028 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), 0),
6029 CurContext, 0, false);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006030
Douglas Gregor331faa02011-04-18 14:13:53 +00006031 // Determine whether we've seen an ivar with a name similar to the
6032 // property.
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006033 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregor331faa02011-04-18 14:13:53 +00006034 NameWithPrefix == Ivar->getName() ||
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006035 NameWithSuffix == Ivar->getName())) {
Douglas Gregor331faa02011-04-18 14:13:53 +00006036 SawSimilarlyNamedIvar = true;
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006037
6038 // Reduce the priority of this result by one, to give it a slight
6039 // advantage over other results whose names don't match so closely.
6040 if (Results.size() &&
6041 Results.data()[Results.size() - 1].Kind
6042 == CodeCompletionResult::RK_Declaration &&
6043 Results.data()[Results.size() - 1].Declaration == Ivar)
6044 Results.data()[Results.size() - 1].Priority--;
6045 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006046 }
Douglas Gregor5d649882009-11-18 22:32:06 +00006047 }
Douglas Gregor331faa02011-04-18 14:13:53 +00006048
6049 if (!SawSimilarlyNamedIvar) {
6050 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006051 // an ivar of the appropriate type.
6052 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregor331faa02011-04-18 14:13:53 +00006053 typedef CodeCompletionResult Result;
6054 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006055 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6056 Priority,CXAvailability_Available);
Douglas Gregor331faa02011-04-18 14:13:53 +00006057
Douglas Gregor75acd922011-09-27 23:30:47 +00006058 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor6c7a9ee2011-04-18 14:40:46 +00006059 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006060 Policy, Allocator));
Douglas Gregor331faa02011-04-18 14:13:53 +00006061 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6062 Results.AddResult(Result(Builder.TakeString(), Priority,
6063 CXCursor_ObjCIvarDecl));
6064 }
6065
Douglas Gregor5d649882009-11-18 22:32:06 +00006066 Results.ExitScope();
6067
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006068 HandleCodeCompleteResults(this, CodeCompleter,
6069 CodeCompletionContext::CCC_Other,
6070 Results.data(),Results.size());
Douglas Gregor5d649882009-11-18 22:32:06 +00006071}
Douglas Gregor636a61e2010-04-07 00:21:17 +00006072
Douglas Gregor416b5752010-08-25 01:08:01 +00006073// Mapping from selectors to the methods that implement that selector, along
6074// with the "in original class" flag.
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006075typedef llvm::DenseMap<
6076 Selector, llvm::PointerIntPair<ObjCMethodDecl *, 1, bool> > KnownMethodsMap;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006077
6078/// \brief Find all of the methods that reside in the given container
6079/// (and its superclasses, protocols, etc.) that meet the given
6080/// criteria. Insert those methods into the map of known methods,
6081/// indexed by selector so they can be easily found.
6082static void FindImplementableMethods(ASTContext &Context,
6083 ObjCContainerDecl *Container,
6084 bool WantInstanceMethods,
6085 QualType ReturnType,
Douglas Gregor416b5752010-08-25 01:08:01 +00006086 KnownMethodsMap &KnownMethods,
6087 bool InOriginalClass = true) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006088 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006089 // Make sure we have a definition; that's what we'll walk.
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006090 if (!IFace->hasDefinition())
6091 return;
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006092
6093 IFace = IFace->getDefinition();
6094 Container = IFace;
Douglas Gregorc0ac7d62011-12-15 05:27:12 +00006095
Douglas Gregor636a61e2010-04-07 00:21:17 +00006096 const ObjCList<ObjCProtocolDecl> &Protocols
6097 = IFace->getReferencedProtocols();
6098 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006099 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006100 I != E; ++I)
6101 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006102 KnownMethods, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006103
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006104 // Add methods from any class extensions and categories.
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006105 for (ObjCInterfaceDecl::visible_categories_iterator
6106 Cat = IFace->visible_categories_begin(),
6107 CatEnd = IFace->visible_categories_end();
6108 Cat != CatEnd; ++Cat) {
6109 FindImplementableMethods(Context, *Cat, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006110 KnownMethods, false);
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006111 }
6112
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006113 // Visit the superclass.
6114 if (IFace->getSuperClass())
6115 FindImplementableMethods(Context, IFace->getSuperClass(),
6116 WantInstanceMethods, ReturnType,
6117 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006118 }
6119
6120 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6121 // Recurse into protocols.
6122 const ObjCList<ObjCProtocolDecl> &Protocols
6123 = Category->getReferencedProtocols();
6124 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006125 E = Protocols.end();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006126 I != E; ++I)
6127 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006128 KnownMethods, InOriginalClass);
6129
6130 // If this category is the original class, jump to the interface.
6131 if (InOriginalClass && Category->getClassInterface())
6132 FindImplementableMethods(Context, Category->getClassInterface(),
6133 WantInstanceMethods, ReturnType, KnownMethods,
6134 false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006135 }
6136
6137 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor9b4f3702012-06-12 13:44:08 +00006138 // Make sure we have a definition; that's what we'll walk.
6139 if (!Protocol->hasDefinition())
6140 return;
6141 Protocol = Protocol->getDefinition();
6142 Container = Protocol;
6143
6144 // Recurse into protocols.
6145 const ObjCList<ObjCProtocolDecl> &Protocols
6146 = Protocol->getReferencedProtocols();
6147 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6148 E = Protocols.end();
6149 I != E; ++I)
6150 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6151 KnownMethods, false);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006152 }
6153
6154 // Add methods in this container. This operation occurs last because
6155 // we want the methods from this container to override any methods
6156 // we've previously seen with the same selector.
6157 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
6158 MEnd = Container->meth_end();
6159 M != MEnd; ++M) {
David Blaikie2d7c57e2012-04-30 02:36:29 +00006160 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006161 if (!ReturnType.isNull() &&
Alp Toker314cc812014-01-25 16:55:45 +00006162 !Context.hasSameUnqualifiedType(ReturnType, M->getReturnType()))
Douglas Gregor636a61e2010-04-07 00:21:17 +00006163 continue;
6164
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006165 KnownMethods[M->getSelector()] =
6166 KnownMethodsMap::mapped_type(*M, InOriginalClass);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006167 }
6168 }
6169}
6170
Douglas Gregor669a25a2011-02-17 00:22:45 +00006171/// \brief Add the parenthesized return or parameter type chunk to a code
6172/// completion string.
6173static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor29979142012-04-10 18:35:07 +00006174 unsigned ObjCDeclQuals,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006175 ASTContext &Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006176 const PrintingPolicy &Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006177 CodeCompletionBuilder &Builder) {
6178 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor29979142012-04-10 18:35:07 +00006179 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6180 if (!Quals.empty())
6181 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor75acd922011-09-27 23:30:47 +00006182 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006183 Builder.getAllocator()));
6184 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6185}
6186
6187/// \brief Determine whether the given class is or inherits from a class by
6188/// the given name.
6189static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006190 StringRef Name) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006191 if (!Class)
6192 return false;
6193
6194 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6195 return true;
6196
6197 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6198}
6199
6200/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6201/// Key-Value Observing (KVO).
6202static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6203 bool IsInstanceMethod,
6204 QualType ReturnType,
6205 ASTContext &Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006206 VisitedSelectorSet &KnownSelectors,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006207 ResultBuilder &Results) {
6208 IdentifierInfo *PropName = Property->getIdentifier();
6209 if (!PropName || PropName->getLength() == 0)
6210 return;
6211
Douglas Gregor75acd922011-09-27 23:30:47 +00006212 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6213
Douglas Gregor669a25a2011-02-17 00:22:45 +00006214 // Builder that will create each code completion.
6215 typedef CodeCompletionResult Result;
6216 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006217 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor669a25a2011-02-17 00:22:45 +00006218
6219 // The selector table.
6220 SelectorTable &Selectors = Context.Selectors;
6221
6222 // The property name, copied into the code completion allocation region
6223 // on demand.
6224 struct KeyHolder {
6225 CodeCompletionAllocator &Allocator;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006226 StringRef Key;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006227 const char *CopiedKey;
6228
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006229 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006230 : Allocator(Allocator), Key(Key), CopiedKey(0) { }
6231
6232 operator const char *() {
6233 if (CopiedKey)
6234 return CopiedKey;
6235
6236 return CopiedKey = Allocator.CopyString(Key);
6237 }
6238 } Key(Allocator, PropName->getName());
6239
6240 // The uppercased name of the property name.
6241 std::string UpperKey = PropName->getName();
6242 if (!UpperKey.empty())
Jordan Rose4938f272013-02-09 10:09:43 +00006243 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006244
6245 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6246 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6247 Property->getType());
6248 bool ReturnTypeMatchesVoid
6249 = ReturnType.isNull() || ReturnType->isVoidType();
6250
6251 // Add the normal accessor -(type)key.
6252 if (IsInstanceMethod &&
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006253 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor669a25a2011-02-17 00:22:45 +00006254 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6255 if (ReturnType.isNull())
Douglas Gregor29979142012-04-10 18:35:07 +00006256 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6257 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006258
6259 Builder.AddTypedTextChunk(Key);
6260 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6261 CXCursor_ObjCInstanceMethodDecl));
6262 }
6263
6264 // If we have an integral or boolean property (or the user has provided
6265 // an integral or boolean return type), add the accessor -(type)isKey.
6266 if (IsInstanceMethod &&
6267 ((!ReturnType.isNull() &&
6268 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6269 (ReturnType.isNull() &&
6270 (Property->getType()->isIntegerType() ||
6271 Property->getType()->isBooleanType())))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006272 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006273 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006274 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006275 if (ReturnType.isNull()) {
6276 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6277 Builder.AddTextChunk("BOOL");
6278 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6279 }
6280
6281 Builder.AddTypedTextChunk(
6282 Allocator.CopyString(SelectorId->getName()));
6283 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6284 CXCursor_ObjCInstanceMethodDecl));
6285 }
6286 }
6287
6288 // Add the normal mutator.
6289 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6290 !Property->getSetterMethodDecl()) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006291 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006292 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006293 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006294 if (ReturnType.isNull()) {
6295 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6296 Builder.AddTextChunk("void");
6297 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6298 }
6299
6300 Builder.AddTypedTextChunk(
6301 Allocator.CopyString(SelectorId->getName()));
6302 Builder.AddTypedTextChunk(":");
Douglas Gregor29979142012-04-10 18:35:07 +00006303 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6304 Context, Policy, Builder);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006305 Builder.AddTextChunk(Key);
6306 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6307 CXCursor_ObjCInstanceMethodDecl));
6308 }
6309 }
6310
6311 // Indexed and unordered accessors
6312 unsigned IndexedGetterPriority = CCP_CodePattern;
6313 unsigned IndexedSetterPriority = CCP_CodePattern;
6314 unsigned UnorderedGetterPriority = CCP_CodePattern;
6315 unsigned UnorderedSetterPriority = CCP_CodePattern;
6316 if (const ObjCObjectPointerType *ObjCPointer
6317 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6318 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6319 // If this interface type is not provably derived from a known
6320 // collection, penalize the corresponding completions.
6321 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6322 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6323 if (!InheritsFromClassNamed(IFace, "NSArray"))
6324 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6325 }
6326
6327 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6328 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6329 if (!InheritsFromClassNamed(IFace, "NSSet"))
6330 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6331 }
6332 }
6333 } else {
6334 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6335 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6336 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6337 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6338 }
6339
6340 // Add -(NSUInteger)countOf<key>
6341 if (IsInstanceMethod &&
6342 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006343 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006344 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006345 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006346 if (ReturnType.isNull()) {
6347 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6348 Builder.AddTextChunk("NSUInteger");
6349 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6350 }
6351
6352 Builder.AddTypedTextChunk(
6353 Allocator.CopyString(SelectorId->getName()));
6354 Results.AddResult(Result(Builder.TakeString(),
6355 std::min(IndexedGetterPriority,
6356 UnorderedGetterPriority),
6357 CXCursor_ObjCInstanceMethodDecl));
6358 }
6359 }
6360
6361 // Indexed getters
6362 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6363 if (IsInstanceMethod &&
6364 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006365 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006366 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006367 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006368 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006369 if (ReturnType.isNull()) {
6370 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6371 Builder.AddTextChunk("id");
6372 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6373 }
6374
6375 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6376 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6377 Builder.AddTextChunk("NSUInteger");
6378 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6379 Builder.AddTextChunk("index");
6380 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6381 CXCursor_ObjCInstanceMethodDecl));
6382 }
6383 }
6384
6385 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6386 if (IsInstanceMethod &&
6387 (ReturnType.isNull() ||
6388 (ReturnType->isObjCObjectPointerType() &&
6389 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6390 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6391 ->getName() == "NSArray"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006392 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006393 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006394 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006395 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006396 if (ReturnType.isNull()) {
6397 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6398 Builder.AddTextChunk("NSArray *");
6399 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6400 }
6401
6402 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6403 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6404 Builder.AddTextChunk("NSIndexSet *");
6405 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6406 Builder.AddTextChunk("indexes");
6407 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6408 CXCursor_ObjCInstanceMethodDecl));
6409 }
6410 }
6411
6412 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6413 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006414 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006415 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006416 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006417 &Context.Idents.get("range")
6418 };
6419
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006420 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006421 if (ReturnType.isNull()) {
6422 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6423 Builder.AddTextChunk("void");
6424 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6425 }
6426
6427 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6428 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6429 Builder.AddPlaceholderChunk("object-type");
6430 Builder.AddTextChunk(" **");
6431 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6432 Builder.AddTextChunk("buffer");
6433 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6434 Builder.AddTypedTextChunk("range:");
6435 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6436 Builder.AddTextChunk("NSRange");
6437 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6438 Builder.AddTextChunk("inRange");
6439 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6440 CXCursor_ObjCInstanceMethodDecl));
6441 }
6442 }
6443
6444 // Mutable indexed accessors
6445
6446 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6447 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006448 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006449 IdentifierInfo *SelectorIds[2] = {
6450 &Context.Idents.get("insertObject"),
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006451 &Context.Idents.get(SelectorName)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006452 };
6453
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006454 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006455 if (ReturnType.isNull()) {
6456 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6457 Builder.AddTextChunk("void");
6458 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6459 }
6460
6461 Builder.AddTypedTextChunk("insertObject:");
6462 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6463 Builder.AddPlaceholderChunk("object-type");
6464 Builder.AddTextChunk(" *");
6465 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6466 Builder.AddTextChunk("object");
6467 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6468 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6469 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6470 Builder.AddPlaceholderChunk("NSUInteger");
6471 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6472 Builder.AddTextChunk("index");
6473 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6474 CXCursor_ObjCInstanceMethodDecl));
6475 }
6476 }
6477
6478 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6479 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006480 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006481 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006482 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006483 &Context.Idents.get("atIndexes")
6484 };
6485
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006486 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006487 if (ReturnType.isNull()) {
6488 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6489 Builder.AddTextChunk("void");
6490 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6491 }
6492
6493 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6494 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6495 Builder.AddTextChunk("NSArray *");
6496 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6497 Builder.AddTextChunk("array");
6498 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6499 Builder.AddTypedTextChunk("atIndexes:");
6500 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6501 Builder.AddPlaceholderChunk("NSIndexSet *");
6502 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6503 Builder.AddTextChunk("indexes");
6504 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6505 CXCursor_ObjCInstanceMethodDecl));
6506 }
6507 }
6508
6509 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6510 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006511 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006512 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006513 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006514 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006515 if (ReturnType.isNull()) {
6516 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6517 Builder.AddTextChunk("void");
6518 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6519 }
6520
6521 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6522 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6523 Builder.AddTextChunk("NSUInteger");
6524 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6525 Builder.AddTextChunk("index");
6526 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6527 CXCursor_ObjCInstanceMethodDecl));
6528 }
6529 }
6530
6531 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6532 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006533 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006534 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006535 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006536 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006537 if (ReturnType.isNull()) {
6538 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6539 Builder.AddTextChunk("void");
6540 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6541 }
6542
6543 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6544 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6545 Builder.AddTextChunk("NSIndexSet *");
6546 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6547 Builder.AddTextChunk("indexes");
6548 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6549 CXCursor_ObjCInstanceMethodDecl));
6550 }
6551 }
6552
6553 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6554 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006555 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006556 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006557 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006558 &Context.Idents.get(SelectorName),
Douglas Gregor669a25a2011-02-17 00:22:45 +00006559 &Context.Idents.get("withObject")
6560 };
6561
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006562 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006563 if (ReturnType.isNull()) {
6564 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6565 Builder.AddTextChunk("void");
6566 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6567 }
6568
6569 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6570 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6571 Builder.AddPlaceholderChunk("NSUInteger");
6572 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6573 Builder.AddTextChunk("index");
6574 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6575 Builder.AddTypedTextChunk("withObject:");
6576 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6577 Builder.AddTextChunk("id");
6578 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6579 Builder.AddTextChunk("object");
6580 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6581 CXCursor_ObjCInstanceMethodDecl));
6582 }
6583 }
6584
6585 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6586 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006587 std::string SelectorName1
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006588 = (Twine("replace") + UpperKey + "AtIndexes").str();
6589 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor669a25a2011-02-17 00:22:45 +00006590 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006591 &Context.Idents.get(SelectorName1),
6592 &Context.Idents.get(SelectorName2)
Douglas Gregor669a25a2011-02-17 00:22:45 +00006593 };
6594
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006595 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006596 if (ReturnType.isNull()) {
6597 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6598 Builder.AddTextChunk("void");
6599 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6600 }
6601
6602 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6603 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6604 Builder.AddPlaceholderChunk("NSIndexSet *");
6605 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6606 Builder.AddTextChunk("indexes");
6607 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6608 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6609 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6610 Builder.AddTextChunk("NSArray *");
6611 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6612 Builder.AddTextChunk("array");
6613 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6614 CXCursor_ObjCInstanceMethodDecl));
6615 }
6616 }
6617
6618 // Unordered getters
6619 // - (NSEnumerator *)enumeratorOfKey
6620 if (IsInstanceMethod &&
6621 (ReturnType.isNull() ||
6622 (ReturnType->isObjCObjectPointerType() &&
6623 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6624 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6625 ->getName() == "NSEnumerator"))) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006626 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006627 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006628 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006629 if (ReturnType.isNull()) {
6630 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6631 Builder.AddTextChunk("NSEnumerator *");
6632 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6633 }
6634
6635 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6636 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6637 CXCursor_ObjCInstanceMethodDecl));
6638 }
6639 }
6640
6641 // - (type *)memberOfKey:(type *)object
6642 if (IsInstanceMethod &&
6643 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006644 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006645 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006646 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006647 if (ReturnType.isNull()) {
6648 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6649 Builder.AddPlaceholderChunk("object-type");
6650 Builder.AddTextChunk(" *");
6651 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6652 }
6653
6654 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6655 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6656 if (ReturnType.isNull()) {
6657 Builder.AddPlaceholderChunk("object-type");
6658 Builder.AddTextChunk(" *");
6659 } else {
6660 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor75acd922011-09-27 23:30:47 +00006661 Policy,
Douglas Gregor669a25a2011-02-17 00:22:45 +00006662 Builder.getAllocator()));
6663 }
6664 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6665 Builder.AddTextChunk("object");
6666 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6667 CXCursor_ObjCInstanceMethodDecl));
6668 }
6669 }
6670
6671 // Mutable unordered accessors
6672 // - (void)addKeyObject:(type *)object
6673 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006674 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006675 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006676 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006677 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006678 if (ReturnType.isNull()) {
6679 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6680 Builder.AddTextChunk("void");
6681 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6682 }
6683
6684 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6685 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6686 Builder.AddPlaceholderChunk("object-type");
6687 Builder.AddTextChunk(" *");
6688 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6689 Builder.AddTextChunk("object");
6690 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6691 CXCursor_ObjCInstanceMethodDecl));
6692 }
6693 }
6694
6695 // - (void)addKey:(NSSet *)objects
6696 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006697 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006698 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006699 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006700 if (ReturnType.isNull()) {
6701 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6702 Builder.AddTextChunk("void");
6703 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6704 }
6705
6706 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6707 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6708 Builder.AddTextChunk("NSSet *");
6709 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6710 Builder.AddTextChunk("objects");
6711 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6712 CXCursor_ObjCInstanceMethodDecl));
6713 }
6714 }
6715
6716 // - (void)removeKeyObject:(type *)object
6717 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006718 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006719 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006720 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006721 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006722 if (ReturnType.isNull()) {
6723 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6724 Builder.AddTextChunk("void");
6725 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6726 }
6727
6728 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6729 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6730 Builder.AddPlaceholderChunk("object-type");
6731 Builder.AddTextChunk(" *");
6732 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6733 Builder.AddTextChunk("object");
6734 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6735 CXCursor_ObjCInstanceMethodDecl));
6736 }
6737 }
6738
6739 // - (void)removeKey:(NSSet *)objects
6740 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006741 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006742 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006743 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006744 if (ReturnType.isNull()) {
6745 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6746 Builder.AddTextChunk("void");
6747 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6748 }
6749
6750 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6751 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6752 Builder.AddTextChunk("NSSet *");
6753 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6754 Builder.AddTextChunk("objects");
6755 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6756 CXCursor_ObjCInstanceMethodDecl));
6757 }
6758 }
6759
6760 // - (void)intersectKey:(NSSet *)objects
6761 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006762 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006763 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006764 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006765 if (ReturnType.isNull()) {
6766 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6767 Builder.AddTextChunk("void");
6768 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6769 }
6770
6771 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6772 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6773 Builder.AddTextChunk("NSSet *");
6774 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6775 Builder.AddTextChunk("objects");
6776 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6777 CXCursor_ObjCInstanceMethodDecl));
6778 }
6779 }
6780
6781 // Key-Value Observing
6782 // + (NSSet *)keyPathsForValuesAffectingKey
6783 if (!IsInstanceMethod &&
6784 (ReturnType.isNull() ||
6785 (ReturnType->isObjCObjectPointerType() &&
6786 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6787 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6788 ->getName() == "NSSet"))) {
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006789 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006790 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor0e5d72f2011-02-17 03:19:26 +00006791 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006792 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor669a25a2011-02-17 00:22:45 +00006793 if (ReturnType.isNull()) {
6794 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6795 Builder.AddTextChunk("NSSet *");
6796 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6797 }
6798
6799 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6800 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor857bcda2011-06-02 04:02:27 +00006801 CXCursor_ObjCClassMethodDecl));
6802 }
6803 }
6804
6805 // + (BOOL)automaticallyNotifiesObserversForKey
6806 if (!IsInstanceMethod &&
6807 (ReturnType.isNull() ||
6808 ReturnType->isIntegerType() ||
6809 ReturnType->isBooleanType())) {
6810 std::string SelectorName
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006811 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor857bcda2011-06-02 04:02:27 +00006812 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6813 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6814 if (ReturnType.isNull()) {
6815 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6816 Builder.AddTextChunk("BOOL");
6817 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6818 }
6819
6820 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6821 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6822 CXCursor_ObjCClassMethodDecl));
Douglas Gregor669a25a2011-02-17 00:22:45 +00006823 }
6824 }
6825}
6826
Douglas Gregor636a61e2010-04-07 00:21:17 +00006827void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6828 bool IsInstanceMethod,
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006829 ParsedType ReturnTy) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006830 // Determine the return type of the method we're declaring, if
6831 // provided.
6832 QualType ReturnType = GetTypeFromParser(ReturnTy);
Fariborz Jahanian8d382dc2011-08-22 15:54:49 +00006833 Decl *IDecl = 0;
6834 if (CurContext->isObjCContainer()) {
6835 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6836 IDecl = cast<Decl>(OCD);
6837 }
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006838 // Determine where we should start searching for methods.
6839 ObjCContainerDecl *SearchDecl = 0;
Douglas Gregor636a61e2010-04-07 00:21:17 +00006840 bool IsInImplementation = false;
John McCall48871652010-08-21 09:40:31 +00006841 if (Decl *D = IDecl) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006842 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6843 SearchDecl = Impl->getClassInterface();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006844 IsInImplementation = true;
6845 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006846 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006847 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregor636a61e2010-04-07 00:21:17 +00006848 IsInImplementation = true;
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006849 } else
Douglas Gregor636a61e2010-04-07 00:21:17 +00006850 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006851 }
6852
6853 if (!SearchDecl && S) {
Ted Kremenekc37877d2013-10-08 17:08:03 +00006854 if (DeclContext *DC = S->getEntity())
Douglas Gregor636a61e2010-04-07 00:21:17 +00006855 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006856 }
6857
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006858 if (!SearchDecl) {
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006859 HandleCodeCompleteResults(this, CodeCompleter,
6860 CodeCompletionContext::CCC_Other,
6861 0, 0);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006862 return;
6863 }
6864
6865 // Find all of the methods that we could declare/implement here.
6866 KnownMethodsMap KnownMethods;
6867 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregor1b035bb2010-10-18 18:21:28 +00006868 ReturnType, KnownMethods);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006869
Douglas Gregor636a61e2010-04-07 00:21:17 +00006870 // Add declarations or definitions for each of the known methods.
John McCall276321a2010-08-25 06:19:51 +00006871 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006872 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006873 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006874 CodeCompletionContext::CCC_Other);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006875 Results.EnterNewScope();
Douglas Gregor75acd922011-09-27 23:30:47 +00006876 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006877 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6878 MEnd = KnownMethods.end();
6879 M != MEnd; ++M) {
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006880 ObjCMethodDecl *Method = M->second.getPointer();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00006881 CodeCompletionBuilder Builder(Results.getAllocator(),
6882 Results.getCodeCompletionTUInfo());
Douglas Gregor636a61e2010-04-07 00:21:17 +00006883
6884 // If the result type was not already provided, add it to the
6885 // pattern as (type).
Douglas Gregor669a25a2011-02-17 00:22:45 +00006886 if (ReturnType.isNull())
Alp Toker314cc812014-01-25 16:55:45 +00006887 AddObjCPassingTypeChunk(Method->getReturnType(),
6888 Method->getObjCDeclQualifier(), Context, Policy,
6889 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006890
6891 Selector Sel = Method->getSelector();
6892
6893 // Add the first part of the selector to the pattern.
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006894 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006895 Sel.getNameForSlot(0)));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006896
6897 // Add parameters to the pattern.
6898 unsigned I = 0;
6899 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6900 PEnd = Method->param_end();
6901 P != PEnd; (void)++P, ++I) {
6902 // Add the part of the selector name.
6903 if (I == 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006904 Builder.AddTypedTextChunk(":");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006905 else if (I < Sel.getNumArgs()) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006906 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6907 Builder.AddTypedTextChunk(
Douglas Gregoraf2a6ae2011-02-18 22:29:55 +00006908 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006909 } else
6910 break;
6911
6912 // Add the parameter type.
Douglas Gregor29979142012-04-10 18:35:07 +00006913 AddObjCPassingTypeChunk((*P)->getOriginalType(),
6914 (*P)->getObjCDeclQualifier(),
6915 Context, Policy,
Douglas Gregor75acd922011-09-27 23:30:47 +00006916 Builder);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006917
6918 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00006919 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006920 }
6921
6922 if (Method->isVariadic()) {
6923 if (Method->param_size() > 0)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006924 Builder.AddChunk(CodeCompletionString::CK_Comma);
6925 Builder.AddTextChunk("...");
Douglas Gregor400f5972010-08-31 05:13:43 +00006926 }
Douglas Gregor636a61e2010-04-07 00:21:17 +00006927
Douglas Gregord37c59d2010-05-28 00:57:46 +00006928 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006929 // We will be defining the method here, so add a compound statement.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006930 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6931 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6932 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Alp Toker314cc812014-01-25 16:55:45 +00006933 if (!Method->getReturnType()->isVoidType()) {
Douglas Gregor636a61e2010-04-07 00:21:17 +00006934 // If the result type is not void, add a return clause.
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006935 Builder.AddTextChunk("return");
6936 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6937 Builder.AddPlaceholderChunk("expression");
6938 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006939 } else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006940 Builder.AddPlaceholderChunk("statements");
Douglas Gregor636a61e2010-04-07 00:21:17 +00006941
Douglas Gregorb278aaf2011-02-01 19:23:04 +00006942 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6943 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregor636a61e2010-04-07 00:21:17 +00006944 }
6945
Douglas Gregor416b5752010-08-25 01:08:01 +00006946 unsigned Priority = CCP_CodePattern;
Benjamin Kramereb8c4462013-06-29 17:52:13 +00006947 if (!M->second.getInt())
Douglas Gregor416b5752010-08-25 01:08:01 +00006948 Priority += CCD_InBaseClass;
6949
Douglas Gregor78254c82012-03-27 23:34:16 +00006950 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregor636a61e2010-04-07 00:21:17 +00006951 }
6952
Douglas Gregor669a25a2011-02-17 00:22:45 +00006953 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
6954 // the properties in this class and its categories.
David Blaikiebbafb8a2012-03-11 07:00:24 +00006955 if (Context.getLangOpts().ObjC2) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00006956 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor669a25a2011-02-17 00:22:45 +00006957 Containers.push_back(SearchDecl);
6958
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006959 VisitedSelectorSet KnownSelectors;
6960 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6961 MEnd = KnownMethods.end();
6962 M != MEnd; ++M)
6963 KnownSelectors.insert(M->first);
6964
6965
Douglas Gregor669a25a2011-02-17 00:22:45 +00006966 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
6967 if (!IFace)
6968 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
6969 IFace = Category->getClassInterface();
6970
6971 if (IFace) {
Douglas Gregor048fbfa2013-01-16 23:00:23 +00006972 for (ObjCInterfaceDecl::visible_categories_iterator
6973 Cat = IFace->visible_categories_begin(),
6974 CatEnd = IFace->visible_categories_end();
6975 Cat != CatEnd; ++Cat) {
6976 Containers.push_back(*Cat);
6977 }
Douglas Gregor669a25a2011-02-17 00:22:45 +00006978 }
6979
Aaron Ballmandc4bea42014-03-13 18:47:37 +00006980 for (unsigned I = 0, N = Containers.size(); I != N; ++I)
6981 for (auto *P : Containers[I]->props())
6982 AddObjCKeyValueCompletions(P, IsInstanceMethod, ReturnType, Context,
Douglas Gregord4a8ced2011-05-04 23:50:46 +00006983 KnownSelectors, Results);
Douglas Gregor669a25a2011-02-17 00:22:45 +00006984 }
6985
Douglas Gregor636a61e2010-04-07 00:21:17 +00006986 Results.ExitScope();
6987
Douglas Gregor00c37ef2010-08-11 21:23:17 +00006988 HandleCodeCompleteResults(this, CodeCompleter,
6989 CodeCompletionContext::CCC_Other,
6990 Results.data(),Results.size());
Douglas Gregor636a61e2010-04-07 00:21:17 +00006991}
Douglas Gregor95887f92010-07-08 23:20:03 +00006992
6993void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
6994 bool IsInstanceMethod,
Douglas Gregor45879692010-07-08 23:37:41 +00006995 bool AtParameterName,
John McCallba7bf592010-08-24 05:47:05 +00006996 ParsedType ReturnTy,
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00006997 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor95887f92010-07-08 23:20:03 +00006998 // If we have an external source, load the entire class method
Sebastian Redld44cd6a2010-08-18 23:57:06 +00006999 // pool from the AST file.
Douglas Gregor95887f92010-07-08 23:20:03 +00007000 if (ExternalSource) {
7001 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
7002 I != N; ++I) {
7003 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redl75d8a322010-08-02 23:18:59 +00007004 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor95887f92010-07-08 23:20:03 +00007005 continue;
Sebastian Redl75d8a322010-08-02 23:18:59 +00007006
7007 ReadMethodPool(Sel);
Douglas Gregor95887f92010-07-08 23:20:03 +00007008 }
7009 }
7010
7011 // Build the set of methods we can see.
John McCall276321a2010-08-25 06:19:51 +00007012 typedef CodeCompletionResult Result;
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007013 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007014 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007015 CodeCompletionContext::CCC_Other);
Douglas Gregor95887f92010-07-08 23:20:03 +00007016
7017 if (ReturnTy)
7018 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redl75d8a322010-08-02 23:18:59 +00007019
Douglas Gregor95887f92010-07-08 23:20:03 +00007020 Results.EnterNewScope();
Sebastian Redl75d8a322010-08-02 23:18:59 +00007021 for (GlobalMethodPool::iterator M = MethodPool.begin(),
7022 MEnd = MethodPool.end();
7023 M != MEnd; ++M) {
7024 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
7025 &M->second.second;
7026 MethList && MethList->Method;
Argyrios Kyrtzidisd3da6e02013-04-17 00:08:58 +00007027 MethList = MethList->getNext()) {
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007028 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents))
Douglas Gregor95887f92010-07-08 23:20:03 +00007029 continue;
7030
Douglas Gregor45879692010-07-08 23:37:41 +00007031 if (AtParameterName) {
7032 // Suggest parameter names we've seen before.
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007033 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor45879692010-07-08 23:37:41 +00007034 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
7035 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
7036 if (Param->getIdentifier()) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007037 CodeCompletionBuilder Builder(Results.getAllocator(),
7038 Results.getCodeCompletionTUInfo());
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007039 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007040 Param->getIdentifier()->getName()));
7041 Results.AddResult(Builder.TakeString());
Douglas Gregor45879692010-07-08 23:37:41 +00007042 }
7043 }
7044
7045 continue;
7046 }
7047
Douglas Gregor0a0e2b32013-01-31 04:52:16 +00007048 Result R(MethList->Method, Results.getBasePriority(MethList->Method), 0);
Dmitri Gribenko070a10e2013-06-16 03:47:57 +00007049 R.StartParameter = SelIdents.size();
Douglas Gregor95887f92010-07-08 23:20:03 +00007050 R.AllParametersAreInformative = false;
7051 R.DeclaringEntity = true;
7052 Results.MaybeAddResult(R, CurContext);
7053 }
7054 }
7055
7056 Results.ExitScope();
Douglas Gregor00c37ef2010-08-11 21:23:17 +00007057 HandleCodeCompleteResults(this, CodeCompleter,
7058 CodeCompletionContext::CCC_Other,
7059 Results.data(),Results.size());
Douglas Gregor95887f92010-07-08 23:20:03 +00007060}
Douglas Gregorb14904c2010-08-13 22:48:40 +00007061
Douglas Gregorec00a262010-08-24 22:20:20 +00007062void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007063 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007064 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007065 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007066 Results.EnterNewScope();
7067
7068 // #if <condition>
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007069 CodeCompletionBuilder Builder(Results.getAllocator(),
7070 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007071 Builder.AddTypedTextChunk("if");
7072 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7073 Builder.AddPlaceholderChunk("condition");
7074 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007075
7076 // #ifdef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007077 Builder.AddTypedTextChunk("ifdef");
7078 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7079 Builder.AddPlaceholderChunk("macro");
7080 Results.AddResult(Builder.TakeString());
7081
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007082 // #ifndef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007083 Builder.AddTypedTextChunk("ifndef");
7084 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7085 Builder.AddPlaceholderChunk("macro");
7086 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007087
7088 if (InConditional) {
7089 // #elif <condition>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007090 Builder.AddTypedTextChunk("elif");
7091 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7092 Builder.AddPlaceholderChunk("condition");
7093 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007094
7095 // #else
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007096 Builder.AddTypedTextChunk("else");
7097 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007098
7099 // #endif
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007100 Builder.AddTypedTextChunk("endif");
7101 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007102 }
7103
7104 // #include "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007105 Builder.AddTypedTextChunk("include");
7106 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7107 Builder.AddTextChunk("\"");
7108 Builder.AddPlaceholderChunk("header");
7109 Builder.AddTextChunk("\"");
7110 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007111
7112 // #include <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007113 Builder.AddTypedTextChunk("include");
7114 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7115 Builder.AddTextChunk("<");
7116 Builder.AddPlaceholderChunk("header");
7117 Builder.AddTextChunk(">");
7118 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007119
7120 // #define <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007121 Builder.AddTypedTextChunk("define");
7122 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7123 Builder.AddPlaceholderChunk("macro");
7124 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007125
7126 // #define <macro>(<args>)
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007127 Builder.AddTypedTextChunk("define");
7128 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7129 Builder.AddPlaceholderChunk("macro");
7130 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7131 Builder.AddPlaceholderChunk("args");
7132 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7133 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007134
7135 // #undef <macro>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007136 Builder.AddTypedTextChunk("undef");
7137 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7138 Builder.AddPlaceholderChunk("macro");
7139 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007140
7141 // #line <number>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007142 Builder.AddTypedTextChunk("line");
7143 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7144 Builder.AddPlaceholderChunk("number");
7145 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007146
7147 // #line <number> "filename"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007148 Builder.AddTypedTextChunk("line");
7149 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7150 Builder.AddPlaceholderChunk("number");
7151 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7152 Builder.AddTextChunk("\"");
7153 Builder.AddPlaceholderChunk("filename");
7154 Builder.AddTextChunk("\"");
7155 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007156
7157 // #error <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007158 Builder.AddTypedTextChunk("error");
7159 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7160 Builder.AddPlaceholderChunk("message");
7161 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007162
7163 // #pragma <arguments>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007164 Builder.AddTypedTextChunk("pragma");
7165 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7166 Builder.AddPlaceholderChunk("arguments");
7167 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007168
David Blaikiebbafb8a2012-03-11 07:00:24 +00007169 if (getLangOpts().ObjC1) {
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007170 // #import "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007171 Builder.AddTypedTextChunk("import");
7172 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7173 Builder.AddTextChunk("\"");
7174 Builder.AddPlaceholderChunk("header");
7175 Builder.AddTextChunk("\"");
7176 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007177
7178 // #import <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007179 Builder.AddTypedTextChunk("import");
7180 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7181 Builder.AddTextChunk("<");
7182 Builder.AddPlaceholderChunk("header");
7183 Builder.AddTextChunk(">");
7184 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007185 }
7186
7187 // #include_next "header"
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007188 Builder.AddTypedTextChunk("include_next");
7189 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7190 Builder.AddTextChunk("\"");
7191 Builder.AddPlaceholderChunk("header");
7192 Builder.AddTextChunk("\"");
7193 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007194
7195 // #include_next <header>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007196 Builder.AddTypedTextChunk("include_next");
7197 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7198 Builder.AddTextChunk("<");
7199 Builder.AddPlaceholderChunk("header");
7200 Builder.AddTextChunk(">");
7201 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007202
7203 // #warning <message>
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007204 Builder.AddTypedTextChunk("warning");
7205 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7206 Builder.AddPlaceholderChunk("message");
7207 Results.AddResult(Builder.TakeString());
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007208
7209 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7210 // completions for them. And __include_macros is a Clang-internal extension
7211 // that we don't want to encourage anyone to use.
7212
7213 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7214 Results.ExitScope();
7215
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007216 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0de55ce2010-08-25 18:41:16 +00007217 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007218 Results.data(), Results.size());
7219}
7220
7221void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorec00a262010-08-24 22:20:20 +00007222 CodeCompleteOrdinaryName(S,
John McCallfaf5fb42010-08-26 23:41:50 +00007223 S->getFnParent()? Sema::PCC_RecoveryInFunction
7224 : Sema::PCC_Namespace);
Douglas Gregor3a7ad252010-08-24 19:08:16 +00007225}
7226
Douglas Gregorec00a262010-08-24 22:20:20 +00007227void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007228 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007229 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007230 IsDefinition? CodeCompletionContext::CCC_MacroName
7231 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor12785102010-08-24 20:21:13 +00007232 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7233 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007234 CodeCompletionBuilder Builder(Results.getAllocator(),
7235 Results.getCodeCompletionTUInfo());
Douglas Gregor12785102010-08-24 20:21:13 +00007236 Results.EnterNewScope();
7237 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7238 MEnd = PP.macro_end();
7239 M != MEnd; ++M) {
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007240 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007241 M->first->getName()));
Argyrios Kyrtzidis5c8b1cd2012-09-27 00:24:09 +00007242 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7243 CCP_CodePattern,
7244 CXCursor_MacroDefinition));
Douglas Gregor12785102010-08-24 20:21:13 +00007245 }
7246 Results.ExitScope();
7247 } else if (IsDefinition) {
7248 // FIXME: Can we detect when the user just wrote an include guard above?
7249 }
7250
Douglas Gregor0ac41382010-09-23 23:01:17 +00007251 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor12785102010-08-24 20:21:13 +00007252 Results.data(), Results.size());
7253}
7254
Douglas Gregorec00a262010-08-24 22:20:20 +00007255void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007256 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007257 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0ac41382010-09-23 23:01:17 +00007258 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorec00a262010-08-24 22:20:20 +00007259
7260 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007261 AddMacroResults(PP, Results, true);
Douglas Gregorec00a262010-08-24 22:20:20 +00007262
7263 // defined (<macro>)
7264 Results.EnterNewScope();
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007265 CodeCompletionBuilder Builder(Results.getAllocator(),
7266 Results.getCodeCompletionTUInfo());
Douglas Gregorb278aaf2011-02-01 19:23:04 +00007267 Builder.AddTypedTextChunk("defined");
7268 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7269 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7270 Builder.AddPlaceholderChunk("macro");
7271 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7272 Results.AddResult(Builder.TakeString());
Douglas Gregorec00a262010-08-24 22:20:20 +00007273 Results.ExitScope();
7274
7275 HandleCodeCompleteResults(this, CodeCompleter,
7276 CodeCompletionContext::CCC_PreprocessorExpression,
7277 Results.data(), Results.size());
7278}
7279
7280void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7281 IdentifierInfo *Macro,
7282 MacroInfo *MacroInfo,
7283 unsigned Argument) {
7284 // FIXME: In the future, we could provide "overload" results, much like we
7285 // do for function calls.
7286
Argyrios Kyrtzidis75f6cd22011-08-18 19:41:28 +00007287 // Now just ignore this. There will be another code-completion callback
7288 // for the expanded tokens.
Douglas Gregorec00a262010-08-24 22:20:20 +00007289}
7290
Douglas Gregor11583702010-08-25 17:04:25 +00007291void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor11583702010-08-25 17:04:25 +00007292 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorea736372010-08-25 17:10:00 +00007293 CodeCompletionContext::CCC_NaturalLanguage,
Douglas Gregor11583702010-08-25 17:04:25 +00007294 0, 0);
7295}
7296
Douglas Gregorbcbf46c2011-02-01 22:57:45 +00007297void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007298 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00007299 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis9d7c0fe2012-04-10 17:23:48 +00007300 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7301 CodeCompletionContext::CCC_Recovery);
Douglas Gregor39982192010-08-15 06:18:01 +00007302 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7303 CodeCompletionDeclConsumer Consumer(Builder,
7304 Context.getTranslationUnitDecl());
7305 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7306 Consumer);
7307 }
Douglas Gregorb14904c2010-08-13 22:48:40 +00007308
7309 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor8cb17462012-10-09 16:01:50 +00007310 AddMacroResults(PP, Builder, true);
Douglas Gregorb14904c2010-08-13 22:48:40 +00007311
7312 Results.clear();
7313 Results.insert(Results.end(),
7314 Builder.data(), Builder.data() + Builder.size());
7315}