blob: 5b7f251c4c9091ae26dd3448e2801716a0b8890d [file] [log] [blame]
Douglas Gregor81b747b2009-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 McCall2d887082010-08-25 22:03:47 +000013#include "clang/Sema/SemaInternal.h"
Douglas Gregore737f502010-08-12 20:07:10 +000014#include "clang/Sema/Lookup.h"
John McCall120d63c2010-08-24 20:38:10 +000015#include "clang/Sema/Overload.h"
Douglas Gregor81b747b2009-09-17 21:32:03 +000016#include "clang/Sema/CodeCompleteConsumer.h"
Douglas Gregor719770d2010-04-06 17:30:22 +000017#include "clang/Sema/ExternalSemaSource.h"
John McCall5f1e0942010-08-24 08:50:51 +000018#include "clang/Sema/Scope.h"
John McCall781472f2010-08-25 08:40:02 +000019#include "clang/Sema/ScopeInfo.h"
John McCall7cd088e2010-08-24 07:21:54 +000020#include "clang/AST/DeclObjC.h"
Douglas Gregorb9d0ef72009-09-21 19:57:38 +000021#include "clang/AST/ExprCXX.h"
Douglas Gregor24a069f2009-11-17 17:59:40 +000022#include "clang/AST/ExprObjC.h"
Douglas Gregor3f7c7f42009-10-30 16:50:04 +000023#include "clang/Lex/MacroInfo.h"
24#include "clang/Lex/Preprocessor.h"
Douglas Gregord36adf52010-09-16 16:06:31 +000025#include "llvm/ADT/DenseSet.h"
Douglas Gregor86d9a522009-09-21 16:56:56 +000026#include "llvm/ADT/SmallPtrSet.h"
Douglas Gregor6a684032009-09-28 03:51:44 +000027#include "llvm/ADT/StringExtras.h"
Douglas Gregor22f56992010-04-06 19:22:33 +000028#include "llvm/ADT/StringSwitch.h"
Douglas Gregor458433d2010-08-26 15:07:07 +000029#include "llvm/ADT/Twine.h"
Douglas Gregor86d9a522009-09-21 16:56:56 +000030#include <list>
31#include <map>
32#include <vector>
Douglas Gregor81b747b2009-09-17 21:32:03 +000033
34using namespace clang;
John McCall781472f2010-08-25 08:40:02 +000035using namespace sema;
Douglas Gregor81b747b2009-09-17 21:32:03 +000036
Douglas Gregor86d9a522009-09-21 16:56:56 +000037namespace {
38 /// \brief A container of code-completion results.
39 class ResultBuilder {
40 public:
41 /// \brief The type of a name-lookup filter, which can be provided to the
42 /// name-lookup routines to specify which declarations should be included in
43 /// the result set (when it returns true) and which declarations should be
44 /// filtered out (returns false).
45 typedef bool (ResultBuilder::*LookupFilter)(NamedDecl *) const;
46
John McCall0a2c5e22010-08-25 06:19:51 +000047 typedef CodeCompletionResult Result;
Douglas Gregor86d9a522009-09-21 16:56:56 +000048
49 private:
50 /// \brief The actual results we have found.
51 std::vector<Result> Results;
52
53 /// \brief A record of all of the declarations we have found and placed
54 /// into the result set, used to ensure that no declaration ever gets into
55 /// the result set twice.
56 llvm::SmallPtrSet<Decl*, 16> AllDeclsFound;
57
Douglas Gregorfbcb5d62009-12-06 20:23:50 +000058 typedef std::pair<NamedDecl *, unsigned> DeclIndexPair;
59
60 /// \brief An entry in the shadow map, which is optimized to store
61 /// a single (declaration, index) mapping (the common case) but
62 /// can also store a list of (declaration, index) mappings.
63 class ShadowMapEntry {
Chris Lattner5f9e2722011-07-23 10:55:15 +000064 typedef SmallVector<DeclIndexPair, 4> DeclIndexPairVector;
Douglas Gregorfbcb5d62009-12-06 20:23:50 +000065
66 /// \brief Contains either the solitary NamedDecl * or a vector
67 /// of (declaration, index) pairs.
68 llvm::PointerUnion<NamedDecl *, DeclIndexPairVector*> DeclOrVector;
69
70 /// \brief When the entry contains a single declaration, this is
71 /// the index associated with that entry.
72 unsigned SingleDeclIndex;
73
74 public:
75 ShadowMapEntry() : DeclOrVector(), SingleDeclIndex(0) { }
76
77 void Add(NamedDecl *ND, unsigned Index) {
78 if (DeclOrVector.isNull()) {
79 // 0 - > 1 elements: just set the single element information.
80 DeclOrVector = ND;
81 SingleDeclIndex = Index;
82 return;
83 }
84
85 if (NamedDecl *PrevND = DeclOrVector.dyn_cast<NamedDecl *>()) {
86 // 1 -> 2 elements: create the vector of results and push in the
87 // existing declaration.
88 DeclIndexPairVector *Vec = new DeclIndexPairVector;
89 Vec->push_back(DeclIndexPair(PrevND, SingleDeclIndex));
90 DeclOrVector = Vec;
91 }
92
93 // Add the new element to the end of the vector.
94 DeclOrVector.get<DeclIndexPairVector*>()->push_back(
95 DeclIndexPair(ND, Index));
96 }
97
98 void Destroy() {
99 if (DeclIndexPairVector *Vec
100 = DeclOrVector.dyn_cast<DeclIndexPairVector *>()) {
101 delete Vec;
102 DeclOrVector = ((NamedDecl *)0);
103 }
104 }
105
106 // Iteration.
107 class iterator;
108 iterator begin() const;
109 iterator end() const;
110 };
111
Douglas Gregor86d9a522009-09-21 16:56:56 +0000112 /// \brief A mapping from declaration names to the declarations that have
113 /// this name within a particular scope and their index within the list of
114 /// results.
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000115 typedef llvm::DenseMap<DeclarationName, ShadowMapEntry> ShadowMap;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000116
117 /// \brief The semantic analysis object for which results are being
118 /// produced.
119 Sema &SemaRef;
Douglas Gregor218937c2011-02-01 19:23:04 +0000120
121 /// \brief The allocator used to allocate new code-completion strings.
Douglas Gregordae68752011-02-01 22:57:45 +0000122 CodeCompletionAllocator &Allocator;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000123
124 /// \brief If non-NULL, a filter function used to remove any code-completion
125 /// results that are not desirable.
126 LookupFilter Filter;
Douglas Gregor45bcd432010-01-14 03:21:49 +0000127
128 /// \brief Whether we should allow declarations as
129 /// nested-name-specifiers that would otherwise be filtered out.
130 bool AllowNestedNameSpecifiers;
131
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +0000132 /// \brief If set, the type that we would prefer our resulting value
133 /// declarations to have.
134 ///
135 /// Closely matching the preferred type gives a boost to a result's
136 /// priority.
137 CanQualType PreferredType;
138
Douglas Gregor86d9a522009-09-21 16:56:56 +0000139 /// \brief A list of shadow maps, which is used to model name hiding at
140 /// different levels of, e.g., the inheritance hierarchy.
141 std::list<ShadowMap> ShadowMaps;
142
Douglas Gregor3cdee122010-08-26 16:36:48 +0000143 /// \brief If we're potentially referring to a C++ member function, the set
144 /// of qualifiers applied to the object type.
145 Qualifiers ObjectTypeQualifiers;
146
147 /// \brief Whether the \p ObjectTypeQualifiers field is active.
148 bool HasObjectTypeQualifiers;
149
Douglas Gregor265f7492010-08-27 15:29:55 +0000150 /// \brief The selector that we prefer.
151 Selector PreferredSelector;
152
Douglas Gregorca45da02010-11-02 20:36:02 +0000153 /// \brief The completion context in which we are gathering results.
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000154 CodeCompletionContext CompletionContext;
155
Douglas Gregorca45da02010-11-02 20:36:02 +0000156 /// \brief If we are in an instance method definition, the @implementation
157 /// object.
158 ObjCImplementationDecl *ObjCImplementation;
159
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000160 void AdjustResultPriorityForDecl(Result &R);
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000161
Douglas Gregor6f942b22010-09-21 16:06:22 +0000162 void MaybeAddConstructorResults(Result R);
163
Douglas Gregor86d9a522009-09-21 16:56:56 +0000164 public:
Douglas Gregordae68752011-02-01 22:57:45 +0000165 explicit ResultBuilder(Sema &SemaRef, CodeCompletionAllocator &Allocator,
Douglas Gregor52779fb2010-09-23 23:01:17 +0000166 const CodeCompletionContext &CompletionContext,
167 LookupFilter Filter = 0)
Douglas Gregor218937c2011-02-01 19:23:04 +0000168 : SemaRef(SemaRef), Allocator(Allocator), Filter(Filter),
169 AllowNestedNameSpecifiers(false), HasObjectTypeQualifiers(false),
Douglas Gregorca45da02010-11-02 20:36:02 +0000170 CompletionContext(CompletionContext),
171 ObjCImplementation(0)
172 {
173 // If this is an Objective-C instance method definition, dig out the
174 // corresponding implementation.
175 switch (CompletionContext.getKind()) {
176 case CodeCompletionContext::CCC_Expression:
177 case CodeCompletionContext::CCC_ObjCMessageReceiver:
178 case CodeCompletionContext::CCC_ParenthesizedExpression:
179 case CodeCompletionContext::CCC_Statement:
180 case CodeCompletionContext::CCC_Recovery:
181 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl())
182 if (Method->isInstanceMethod())
183 if (ObjCInterfaceDecl *Interface = Method->getClassInterface())
184 ObjCImplementation = Interface->getImplementation();
185 break;
186
187 default:
188 break;
189 }
190 }
Douglas Gregor86d9a522009-09-21 16:56:56 +0000191
Douglas Gregord8e8a582010-05-25 21:41:55 +0000192 /// \brief Whether we should include code patterns in the completion
193 /// results.
194 bool includeCodePatterns() const {
195 return SemaRef.CodeCompleter &&
Douglas Gregorf6961522010-08-27 21:18:54 +0000196 SemaRef.CodeCompleter->includeCodePatterns();
Douglas Gregord8e8a582010-05-25 21:41:55 +0000197 }
198
Douglas Gregor86d9a522009-09-21 16:56:56 +0000199 /// \brief Set the filter used for code-completion results.
200 void setFilter(LookupFilter Filter) {
201 this->Filter = Filter;
202 }
203
Douglas Gregor86d9a522009-09-21 16:56:56 +0000204 Result *data() { return Results.empty()? 0 : &Results.front(); }
205 unsigned size() const { return Results.size(); }
206 bool empty() const { return Results.empty(); }
207
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +0000208 /// \brief Specify the preferred type.
209 void setPreferredType(QualType T) {
210 PreferredType = SemaRef.Context.getCanonicalType(T);
211 }
212
Douglas Gregor3cdee122010-08-26 16:36:48 +0000213 /// \brief Set the cv-qualifiers on the object type, for us in filtering
214 /// calls to member functions.
215 ///
216 /// When there are qualifiers in this set, they will be used to filter
217 /// out member functions that aren't available (because there will be a
218 /// cv-qualifier mismatch) or prefer functions with an exact qualifier
219 /// match.
220 void setObjectTypeQualifiers(Qualifiers Quals) {
221 ObjectTypeQualifiers = Quals;
222 HasObjectTypeQualifiers = true;
223 }
224
Douglas Gregor265f7492010-08-27 15:29:55 +0000225 /// \brief Set the preferred selector.
226 ///
227 /// When an Objective-C method declaration result is added, and that
228 /// method's selector matches this preferred selector, we give that method
229 /// a slight priority boost.
230 void setPreferredSelector(Selector Sel) {
231 PreferredSelector = Sel;
232 }
Douglas Gregorca45da02010-11-02 20:36:02 +0000233
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000234 /// \brief Retrieve the code-completion context for which results are
235 /// being collected.
236 const CodeCompletionContext &getCompletionContext() const {
237 return CompletionContext;
238 }
239
Douglas Gregor45bcd432010-01-14 03:21:49 +0000240 /// \brief Specify whether nested-name-specifiers are allowed.
241 void allowNestedNameSpecifiers(bool Allow = true) {
242 AllowNestedNameSpecifiers = Allow;
243 }
244
Douglas Gregorb9d77572010-09-21 00:03:25 +0000245 /// \brief Return the semantic analysis object for which we are collecting
246 /// code completion results.
247 Sema &getSema() const { return SemaRef; }
248
Douglas Gregor218937c2011-02-01 19:23:04 +0000249 /// \brief Retrieve the allocator used to allocate code completion strings.
Douglas Gregordae68752011-02-01 22:57:45 +0000250 CodeCompletionAllocator &getAllocator() const { return Allocator; }
Douglas Gregor218937c2011-02-01 19:23:04 +0000251
Douglas Gregore495b7f2010-01-14 00:20:49 +0000252 /// \brief Determine whether the given declaration is at all interesting
253 /// as a code-completion result.
Douglas Gregor45bcd432010-01-14 03:21:49 +0000254 ///
255 /// \param ND the declaration that we are inspecting.
256 ///
257 /// \param AsNestedNameSpecifier will be set true if this declaration is
258 /// only interesting when it is a nested-name-specifier.
259 bool isInterestingDecl(NamedDecl *ND, bool &AsNestedNameSpecifier) const;
Douglas Gregor6660d842010-01-14 00:41:07 +0000260
261 /// \brief Check whether the result is hidden by the Hiding declaration.
262 ///
263 /// \returns true if the result is hidden and cannot be found, false if
264 /// the hidden result could still be found. When false, \p R may be
265 /// modified to describe how the result can be found (e.g., via extra
266 /// qualification).
267 bool CheckHiddenResult(Result &R, DeclContext *CurContext,
268 NamedDecl *Hiding);
269
Douglas Gregor86d9a522009-09-21 16:56:56 +0000270 /// \brief Add a new result to this result set (if it isn't already in one
271 /// of the shadow maps), or replace an existing result (for, e.g., a
272 /// redeclaration).
Douglas Gregor456c4a12009-09-21 20:12:40 +0000273 ///
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000274 /// \param CurContext the result to add (if it is unique).
Douglas Gregor456c4a12009-09-21 20:12:40 +0000275 ///
276 /// \param R the context in which this result will be named.
277 void MaybeAddResult(Result R, DeclContext *CurContext = 0);
Douglas Gregor86d9a522009-09-21 16:56:56 +0000278
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000279 /// \brief Add a new result to this result set, where we already know
280 /// the hiding declation (if any).
281 ///
282 /// \param R the result to add (if it is unique).
283 ///
284 /// \param CurContext the context in which this result will be named.
285 ///
286 /// \param Hiding the declaration that hides the result.
Douglas Gregor0cc84042010-01-14 15:47:35 +0000287 ///
288 /// \param InBaseClass whether the result was found in a base
289 /// class of the searched context.
290 void AddResult(Result R, DeclContext *CurContext, NamedDecl *Hiding,
291 bool InBaseClass);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000292
Douglas Gregora4477812010-01-14 16:01:26 +0000293 /// \brief Add a new non-declaration result to this result set.
294 void AddResult(Result R);
295
Douglas Gregor86d9a522009-09-21 16:56:56 +0000296 /// \brief Enter into a new scope.
297 void EnterNewScope();
298
299 /// \brief Exit from the current scope.
300 void ExitScope();
301
Douglas Gregor55385fe2009-11-18 04:19:12 +0000302 /// \brief Ignore this declaration, if it is seen again.
303 void Ignore(Decl *D) { AllDeclsFound.insert(D->getCanonicalDecl()); }
304
Douglas Gregor86d9a522009-09-21 16:56:56 +0000305 /// \name Name lookup predicates
306 ///
307 /// These predicates can be passed to the name lookup functions to filter the
308 /// results of name lookup. All of the predicates have the same type, so that
309 ///
310 //@{
Douglas Gregor791215b2009-09-21 20:51:25 +0000311 bool IsOrdinaryName(NamedDecl *ND) const;
Douglas Gregor4710e5b2010-05-28 00:49:12 +0000312 bool IsOrdinaryNonTypeName(NamedDecl *ND) const;
Douglas Gregorf9578432010-07-28 21:50:18 +0000313 bool IsIntegralConstantValue(NamedDecl *ND) const;
Douglas Gregor01dfea02010-01-10 23:08:15 +0000314 bool IsOrdinaryNonValueName(NamedDecl *ND) const;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000315 bool IsNestedNameSpecifier(NamedDecl *ND) const;
316 bool IsEnum(NamedDecl *ND) const;
317 bool IsClassOrStruct(NamedDecl *ND) const;
318 bool IsUnion(NamedDecl *ND) const;
319 bool IsNamespace(NamedDecl *ND) const;
320 bool IsNamespaceOrAlias(NamedDecl *ND) const;
321 bool IsType(NamedDecl *ND) const;
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000322 bool IsMember(NamedDecl *ND) const;
Douglas Gregor80f4f4c2010-01-14 16:08:12 +0000323 bool IsObjCIvar(NamedDecl *ND) const;
Douglas Gregor8e254cf2010-05-27 23:06:34 +0000324 bool IsObjCMessageReceiver(NamedDecl *ND) const;
Douglas Gregorfb629412010-08-23 21:17:50 +0000325 bool IsObjCCollection(NamedDecl *ND) const;
Douglas Gregor52779fb2010-09-23 23:01:17 +0000326 bool IsImpossibleToSatisfy(NamedDecl *ND) const;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000327 //@}
328 };
329}
330
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000331class ResultBuilder::ShadowMapEntry::iterator {
332 llvm::PointerUnion<NamedDecl*, const DeclIndexPair*> DeclOrIterator;
333 unsigned SingleDeclIndex;
334
335public:
336 typedef DeclIndexPair value_type;
337 typedef value_type reference;
338 typedef std::ptrdiff_t difference_type;
339 typedef std::input_iterator_tag iterator_category;
340
341 class pointer {
342 DeclIndexPair Value;
343
344 public:
345 pointer(const DeclIndexPair &Value) : Value(Value) { }
346
347 const DeclIndexPair *operator->() const {
348 return &Value;
349 }
350 };
351
352 iterator() : DeclOrIterator((NamedDecl *)0), SingleDeclIndex(0) { }
353
354 iterator(NamedDecl *SingleDecl, unsigned Index)
355 : DeclOrIterator(SingleDecl), SingleDeclIndex(Index) { }
356
357 iterator(const DeclIndexPair *Iterator)
358 : DeclOrIterator(Iterator), SingleDeclIndex(0) { }
359
360 iterator &operator++() {
361 if (DeclOrIterator.is<NamedDecl *>()) {
362 DeclOrIterator = (NamedDecl *)0;
363 SingleDeclIndex = 0;
364 return *this;
365 }
366
367 const DeclIndexPair *I = DeclOrIterator.get<const DeclIndexPair*>();
368 ++I;
369 DeclOrIterator = I;
370 return *this;
371 }
372
Chris Lattner66392d42010-09-04 18:12:20 +0000373 /*iterator operator++(int) {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000374 iterator tmp(*this);
375 ++(*this);
376 return tmp;
Chris Lattner66392d42010-09-04 18:12:20 +0000377 }*/
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000378
379 reference operator*() const {
380 if (NamedDecl *ND = DeclOrIterator.dyn_cast<NamedDecl *>())
381 return reference(ND, SingleDeclIndex);
382
Douglas Gregord490f952009-12-06 21:27:58 +0000383 return *DeclOrIterator.get<const DeclIndexPair*>();
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000384 }
385
386 pointer operator->() const {
387 return pointer(**this);
388 }
389
390 friend bool operator==(const iterator &X, const iterator &Y) {
Douglas Gregord490f952009-12-06 21:27:58 +0000391 return X.DeclOrIterator.getOpaqueValue()
392 == Y.DeclOrIterator.getOpaqueValue() &&
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000393 X.SingleDeclIndex == Y.SingleDeclIndex;
394 }
395
396 friend bool operator!=(const iterator &X, const iterator &Y) {
Douglas Gregord490f952009-12-06 21:27:58 +0000397 return !(X == Y);
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000398 }
399};
400
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000401ResultBuilder::ShadowMapEntry::iterator
402ResultBuilder::ShadowMapEntry::begin() const {
403 if (DeclOrVector.isNull())
404 return iterator();
405
406 if (NamedDecl *ND = DeclOrVector.dyn_cast<NamedDecl *>())
407 return iterator(ND, SingleDeclIndex);
408
409 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->begin());
410}
411
412ResultBuilder::ShadowMapEntry::iterator
413ResultBuilder::ShadowMapEntry::end() const {
414 if (DeclOrVector.is<NamedDecl *>() || DeclOrVector.isNull())
415 return iterator();
416
417 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->end());
418}
419
Douglas Gregor456c4a12009-09-21 20:12:40 +0000420/// \brief Compute the qualification required to get from the current context
421/// (\p CurContext) to the target context (\p TargetContext).
422///
423/// \param Context the AST context in which the qualification will be used.
424///
425/// \param CurContext the context where an entity is being named, which is
426/// typically based on the current scope.
427///
428/// \param TargetContext the context in which the named entity actually
429/// resides.
430///
431/// \returns a nested name specifier that refers into the target context, or
432/// NULL if no qualification is needed.
433static NestedNameSpecifier *
434getRequiredQualification(ASTContext &Context,
435 DeclContext *CurContext,
436 DeclContext *TargetContext) {
Chris Lattner5f9e2722011-07-23 10:55:15 +0000437 SmallVector<DeclContext *, 4> TargetParents;
Douglas Gregor456c4a12009-09-21 20:12:40 +0000438
439 for (DeclContext *CommonAncestor = TargetContext;
440 CommonAncestor && !CommonAncestor->Encloses(CurContext);
441 CommonAncestor = CommonAncestor->getLookupParent()) {
442 if (CommonAncestor->isTransparentContext() ||
443 CommonAncestor->isFunctionOrMethod())
444 continue;
445
446 TargetParents.push_back(CommonAncestor);
447 }
448
449 NestedNameSpecifier *Result = 0;
450 while (!TargetParents.empty()) {
451 DeclContext *Parent = TargetParents.back();
452 TargetParents.pop_back();
453
Douglas Gregorfb629412010-08-23 21:17:50 +0000454 if (NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Parent)) {
455 if (!Namespace->getIdentifier())
456 continue;
457
Douglas Gregor456c4a12009-09-21 20:12:40 +0000458 Result = NestedNameSpecifier::Create(Context, Result, Namespace);
Douglas Gregorfb629412010-08-23 21:17:50 +0000459 }
Douglas Gregor456c4a12009-09-21 20:12:40 +0000460 else if (TagDecl *TD = dyn_cast<TagDecl>(Parent))
461 Result = NestedNameSpecifier::Create(Context, Result,
462 false,
463 Context.getTypeDeclType(TD).getTypePtr());
Douglas Gregor0c8296d2009-11-07 00:00:49 +0000464 }
Douglas Gregor456c4a12009-09-21 20:12:40 +0000465 return Result;
466}
467
Douglas Gregor45bcd432010-01-14 03:21:49 +0000468bool ResultBuilder::isInterestingDecl(NamedDecl *ND,
469 bool &AsNestedNameSpecifier) const {
470 AsNestedNameSpecifier = false;
471
Douglas Gregore495b7f2010-01-14 00:20:49 +0000472 ND = ND->getUnderlyingDecl();
473 unsigned IDNS = ND->getIdentifierNamespace();
Douglas Gregorf52cede2009-10-09 22:16:47 +0000474
475 // Skip unnamed entities.
Douglas Gregore495b7f2010-01-14 00:20:49 +0000476 if (!ND->getDeclName())
477 return false;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000478
479 // Friend declarations and declarations introduced due to friends are never
480 // added as results.
John McCall92b7f702010-03-11 07:50:04 +0000481 if (IDNS & (Decl::IDNS_OrdinaryFriend | Decl::IDNS_TagFriend))
Douglas Gregore495b7f2010-01-14 00:20:49 +0000482 return false;
483
Douglas Gregor76282942009-12-11 17:31:05 +0000484 // Class template (partial) specializations are never added as results.
Douglas Gregore495b7f2010-01-14 00:20:49 +0000485 if (isa<ClassTemplateSpecializationDecl>(ND) ||
486 isa<ClassTemplatePartialSpecializationDecl>(ND))
487 return false;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000488
Douglas Gregor76282942009-12-11 17:31:05 +0000489 // Using declarations themselves are never added as results.
Douglas Gregore495b7f2010-01-14 00:20:49 +0000490 if (isa<UsingDecl>(ND))
491 return false;
492
493 // Some declarations have reserved names that we don't want to ever show.
494 if (const IdentifierInfo *Id = ND->getIdentifier()) {
Douglas Gregor86d9a522009-09-21 16:56:56 +0000495 // __va_list_tag is a freak of nature. Find it and skip it.
496 if (Id->isStr("__va_list_tag") || Id->isStr("__builtin_va_list"))
Douglas Gregore495b7f2010-01-14 00:20:49 +0000497 return false;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000498
Douglas Gregorf52cede2009-10-09 22:16:47 +0000499 // Filter out names reserved for the implementation (C99 7.1.3,
Douglas Gregor797efb52010-07-14 17:44:04 +0000500 // C++ [lib.global.names]) if they come from a system header.
Daniel Dunbare013d682009-10-18 20:26:12 +0000501 //
502 // FIXME: Add predicate for this.
Douglas Gregorf52cede2009-10-09 22:16:47 +0000503 if (Id->getLength() >= 2) {
Daniel Dunbare013d682009-10-18 20:26:12 +0000504 const char *Name = Id->getNameStart();
Douglas Gregorf52cede2009-10-09 22:16:47 +0000505 if (Name[0] == '_' &&
Douglas Gregor797efb52010-07-14 17:44:04 +0000506 (Name[1] == '_' || (Name[1] >= 'A' && Name[1] <= 'Z')) &&
507 (ND->getLocation().isInvalid() ||
508 SemaRef.SourceMgr.isInSystemHeader(
509 SemaRef.SourceMgr.getSpellingLoc(ND->getLocation()))))
Douglas Gregore495b7f2010-01-14 00:20:49 +0000510 return false;
Douglas Gregorf52cede2009-10-09 22:16:47 +0000511 }
Douglas Gregor86d9a522009-09-21 16:56:56 +0000512 }
Douglas Gregor6f942b22010-09-21 16:06:22 +0000513
Douglas Gregor9b0ba872010-11-09 03:59:40 +0000514 // Skip out-of-line declarations and definitions.
515 // NOTE: Unless it's an Objective-C property, method, or ivar, where
516 // the contexts can be messy.
517 if (!ND->getDeclContext()->Equals(ND->getLexicalDeclContext()) &&
518 !(isa<ObjCPropertyDecl>(ND) || isa<ObjCIvarDecl>(ND) ||
519 isa<ObjCMethodDecl>(ND)))
520 return false;
521
Douglas Gregora5fb7c32010-08-16 23:05:20 +0000522 if (Filter == &ResultBuilder::IsNestedNameSpecifier ||
523 ((isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND)) &&
524 Filter != &ResultBuilder::IsNamespace &&
Douglas Gregor52779fb2010-09-23 23:01:17 +0000525 Filter != &ResultBuilder::IsNamespaceOrAlias &&
526 Filter != 0))
Douglas Gregora5fb7c32010-08-16 23:05:20 +0000527 AsNestedNameSpecifier = true;
528
Douglas Gregor86d9a522009-09-21 16:56:56 +0000529 // Filter out any unwanted results.
Douglas Gregor45bcd432010-01-14 03:21:49 +0000530 if (Filter && !(this->*Filter)(ND)) {
531 // Check whether it is interesting as a nested-name-specifier.
532 if (AllowNestedNameSpecifiers && SemaRef.getLangOptions().CPlusPlus &&
533 IsNestedNameSpecifier(ND) &&
534 (Filter != &ResultBuilder::IsMember ||
535 (isa<CXXRecordDecl>(ND) &&
536 cast<CXXRecordDecl>(ND)->isInjectedClassName()))) {
537 AsNestedNameSpecifier = true;
538 return true;
539 }
540
Douglas Gregore495b7f2010-01-14 00:20:49 +0000541 return false;
Douglas Gregora5fb7c32010-08-16 23:05:20 +0000542 }
Douglas Gregore495b7f2010-01-14 00:20:49 +0000543 // ... then it must be interesting!
544 return true;
545}
546
Douglas Gregor6660d842010-01-14 00:41:07 +0000547bool ResultBuilder::CheckHiddenResult(Result &R, DeclContext *CurContext,
548 NamedDecl *Hiding) {
549 // In C, there is no way to refer to a hidden name.
550 // FIXME: This isn't true; we can find a tag name hidden by an ordinary
551 // name if we introduce the tag type.
552 if (!SemaRef.getLangOptions().CPlusPlus)
553 return true;
554
Sebastian Redl7a126a42010-08-31 00:36:30 +0000555 DeclContext *HiddenCtx = R.Declaration->getDeclContext()->getRedeclContext();
Douglas Gregor6660d842010-01-14 00:41:07 +0000556
557 // There is no way to qualify a name declared in a function or method.
558 if (HiddenCtx->isFunctionOrMethod())
559 return true;
560
Sebastian Redl7a126a42010-08-31 00:36:30 +0000561 if (HiddenCtx == Hiding->getDeclContext()->getRedeclContext())
Douglas Gregor6660d842010-01-14 00:41:07 +0000562 return true;
563
564 // We can refer to the result with the appropriate qualification. Do it.
565 R.Hidden = true;
566 R.QualifierIsInformative = false;
567
568 if (!R.Qualifier)
569 R.Qualifier = getRequiredQualification(SemaRef.Context,
570 CurContext,
571 R.Declaration->getDeclContext());
572 return false;
573}
574
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000575/// \brief A simplified classification of types used to determine whether two
576/// types are "similar enough" when adjusting priorities.
Douglas Gregor1827e102010-08-16 16:18:59 +0000577SimplifiedTypeClass clang::getSimplifiedTypeClass(CanQualType T) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000578 switch (T->getTypeClass()) {
579 case Type::Builtin:
580 switch (cast<BuiltinType>(T)->getKind()) {
581 case BuiltinType::Void:
582 return STC_Void;
583
584 case BuiltinType::NullPtr:
585 return STC_Pointer;
586
587 case BuiltinType::Overload:
588 case BuiltinType::Dependent:
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000589 return STC_Other;
590
591 case BuiltinType::ObjCId:
592 case BuiltinType::ObjCClass:
593 case BuiltinType::ObjCSel:
594 return STC_ObjectiveC;
595
596 default:
597 return STC_Arithmetic;
598 }
599 return STC_Other;
600
601 case Type::Complex:
602 return STC_Arithmetic;
603
604 case Type::Pointer:
605 return STC_Pointer;
606
607 case Type::BlockPointer:
608 return STC_Block;
609
610 case Type::LValueReference:
611 case Type::RValueReference:
612 return getSimplifiedTypeClass(T->getAs<ReferenceType>()->getPointeeType());
613
614 case Type::ConstantArray:
615 case Type::IncompleteArray:
616 case Type::VariableArray:
617 case Type::DependentSizedArray:
618 return STC_Array;
619
620 case Type::DependentSizedExtVector:
621 case Type::Vector:
622 case Type::ExtVector:
623 return STC_Arithmetic;
624
625 case Type::FunctionProto:
626 case Type::FunctionNoProto:
627 return STC_Function;
628
629 case Type::Record:
630 return STC_Record;
631
632 case Type::Enum:
633 return STC_Arithmetic;
634
635 case Type::ObjCObject:
636 case Type::ObjCInterface:
637 case Type::ObjCObjectPointer:
638 return STC_ObjectiveC;
639
640 default:
641 return STC_Other;
642 }
643}
644
645/// \brief Get the type that a given expression will have if this declaration
646/// is used as an expression in its "typical" code-completion form.
Douglas Gregor1827e102010-08-16 16:18:59 +0000647QualType clang::getDeclUsageType(ASTContext &C, NamedDecl *ND) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000648 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
649
650 if (TypeDecl *Type = dyn_cast<TypeDecl>(ND))
651 return C.getTypeDeclType(Type);
652 if (ObjCInterfaceDecl *Iface = dyn_cast<ObjCInterfaceDecl>(ND))
653 return C.getObjCInterfaceType(Iface);
654
655 QualType T;
656 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(ND))
Douglas Gregor5291c3c2010-07-13 08:18:22 +0000657 T = Function->getCallResultType();
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000658 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregor5291c3c2010-07-13 08:18:22 +0000659 T = Method->getSendResultType();
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000660 else if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND))
Douglas Gregor5291c3c2010-07-13 08:18:22 +0000661 T = FunTmpl->getTemplatedDecl()->getCallResultType();
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000662 else if (EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
663 T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext()));
664 else if (ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
665 T = Property->getType();
666 else if (ValueDecl *Value = dyn_cast<ValueDecl>(ND))
667 T = Value->getType();
668 else
669 return QualType();
Douglas Gregor3e64d562011-04-14 20:33:34 +0000670
671 // Dig through references, function pointers, and block pointers to
672 // get down to the likely type of an expression when the entity is
673 // used.
674 do {
675 if (const ReferenceType *Ref = T->getAs<ReferenceType>()) {
676 T = Ref->getPointeeType();
677 continue;
678 }
679
680 if (const PointerType *Pointer = T->getAs<PointerType>()) {
681 if (Pointer->getPointeeType()->isFunctionType()) {
682 T = Pointer->getPointeeType();
683 continue;
684 }
685
686 break;
687 }
688
689 if (const BlockPointerType *Block = T->getAs<BlockPointerType>()) {
690 T = Block->getPointeeType();
691 continue;
692 }
693
694 if (const FunctionType *Function = T->getAs<FunctionType>()) {
695 T = Function->getResultType();
696 continue;
697 }
698
699 break;
700 } while (true);
701
702 return T;
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000703}
704
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000705void ResultBuilder::AdjustResultPriorityForDecl(Result &R) {
706 // If this is an Objective-C method declaration whose selector matches our
707 // preferred selector, give it a priority boost.
708 if (!PreferredSelector.isNull())
709 if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(R.Declaration))
710 if (PreferredSelector == Method->getSelector())
711 R.Priority += CCD_SelectorMatch;
Douglas Gregor08f43cd2010-09-20 23:11:55 +0000712
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000713 // If we have a preferred type, adjust the priority for results with exactly-
714 // matching or nearly-matching types.
715 if (!PreferredType.isNull()) {
716 QualType T = getDeclUsageType(SemaRef.Context, R.Declaration);
717 if (!T.isNull()) {
718 CanQualType TC = SemaRef.Context.getCanonicalType(T);
719 // Check for exactly-matching types (modulo qualifiers).
720 if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC))
721 R.Priority /= CCF_ExactTypeMatch;
722 // Check for nearly-matching types, based on classification of each.
723 else if ((getSimplifiedTypeClass(PreferredType)
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000724 == getSimplifiedTypeClass(TC)) &&
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000725 !(PreferredType->isEnumeralType() && TC->isEnumeralType()))
726 R.Priority /= CCF_SimilarTypeMatch;
727 }
728 }
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000729}
730
Douglas Gregor6f942b22010-09-21 16:06:22 +0000731void ResultBuilder::MaybeAddConstructorResults(Result R) {
732 if (!SemaRef.getLangOptions().CPlusPlus || !R.Declaration ||
733 !CompletionContext.wantConstructorResults())
734 return;
735
736 ASTContext &Context = SemaRef.Context;
737 NamedDecl *D = R.Declaration;
738 CXXRecordDecl *Record = 0;
739 if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D))
740 Record = ClassTemplate->getTemplatedDecl();
741 else if ((Record = dyn_cast<CXXRecordDecl>(D))) {
742 // Skip specializations and partial specializations.
743 if (isa<ClassTemplateSpecializationDecl>(Record))
744 return;
745 } else {
746 // There are no constructors here.
747 return;
748 }
749
750 Record = Record->getDefinition();
751 if (!Record)
752 return;
753
754
755 QualType RecordTy = Context.getTypeDeclType(Record);
756 DeclarationName ConstructorName
757 = Context.DeclarationNames.getCXXConstructorName(
758 Context.getCanonicalType(RecordTy));
759 for (DeclContext::lookup_result Ctors = Record->lookup(ConstructorName);
760 Ctors.first != Ctors.second; ++Ctors.first) {
761 R.Declaration = *Ctors.first;
762 R.CursorKind = getCursorKindForDecl(R.Declaration);
763 Results.push_back(R);
764 }
765}
766
Douglas Gregore495b7f2010-01-14 00:20:49 +0000767void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) {
768 assert(!ShadowMaps.empty() && "Must enter into a results scope");
769
770 if (R.Kind != Result::RK_Declaration) {
771 // For non-declaration results, just add the result.
772 Results.push_back(R);
773 return;
774 }
775
776 // Look through using declarations.
777 if (UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
778 MaybeAddResult(Result(Using->getTargetDecl(), R.Qualifier), CurContext);
779 return;
780 }
781
782 Decl *CanonDecl = R.Declaration->getCanonicalDecl();
783 unsigned IDNS = CanonDecl->getIdentifierNamespace();
784
Douglas Gregor45bcd432010-01-14 03:21:49 +0000785 bool AsNestedNameSpecifier = false;
786 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregore495b7f2010-01-14 00:20:49 +0000787 return;
788
Douglas Gregor6f942b22010-09-21 16:06:22 +0000789 // C++ constructors are never found by name lookup.
790 if (isa<CXXConstructorDecl>(R.Declaration))
791 return;
792
Douglas Gregor86d9a522009-09-21 16:56:56 +0000793 ShadowMap &SMap = ShadowMaps.back();
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000794 ShadowMapEntry::iterator I, IEnd;
795 ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName());
796 if (NamePos != SMap.end()) {
797 I = NamePos->second.begin();
798 IEnd = NamePos->second.end();
799 }
800
801 for (; I != IEnd; ++I) {
802 NamedDecl *ND = I->first;
803 unsigned Index = I->second;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000804 if (ND->getCanonicalDecl() == CanonDecl) {
805 // This is a redeclaration. Always pick the newer declaration.
Douglas Gregor86d9a522009-09-21 16:56:56 +0000806 Results[Index].Declaration = R.Declaration;
807
Douglas Gregor86d9a522009-09-21 16:56:56 +0000808 // We're done.
809 return;
810 }
811 }
812
813 // This is a new declaration in this scope. However, check whether this
814 // declaration name is hidden by a similarly-named declaration in an outer
815 // scope.
816 std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end();
817 --SMEnd;
818 for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000819 ShadowMapEntry::iterator I, IEnd;
820 ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName());
821 if (NamePos != SM->end()) {
822 I = NamePos->second.begin();
823 IEnd = NamePos->second.end();
824 }
825 for (; I != IEnd; ++I) {
Douglas Gregor86d9a522009-09-21 16:56:56 +0000826 // A tag declaration does not hide a non-tag declaration.
John McCall0d6b1642010-04-23 18:46:30 +0000827 if (I->first->hasTagIdentifierNamespace() &&
Douglas Gregor86d9a522009-09-21 16:56:56 +0000828 (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary |
829 Decl::IDNS_ObjCProtocol)))
830 continue;
831
832 // Protocols are in distinct namespaces from everything else.
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000833 if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol)
Douglas Gregor86d9a522009-09-21 16:56:56 +0000834 || (IDNS & Decl::IDNS_ObjCProtocol)) &&
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000835 I->first->getIdentifierNamespace() != IDNS)
Douglas Gregor86d9a522009-09-21 16:56:56 +0000836 continue;
837
838 // The newly-added result is hidden by an entry in the shadow map.
Douglas Gregor6660d842010-01-14 00:41:07 +0000839 if (CheckHiddenResult(R, CurContext, I->first))
Douglas Gregor86d9a522009-09-21 16:56:56 +0000840 return;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000841
842 break;
843 }
844 }
845
846 // Make sure that any given declaration only shows up in the result set once.
847 if (!AllDeclsFound.insert(CanonDecl))
848 return;
Douglas Gregor265f7492010-08-27 15:29:55 +0000849
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000850 // If the filter is for nested-name-specifiers, then this result starts a
851 // nested-name-specifier.
Douglas Gregor12e13132010-05-26 22:00:08 +0000852 if (AsNestedNameSpecifier) {
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000853 R.StartsNestedNameSpecifier = true;
Douglas Gregor12e13132010-05-26 22:00:08 +0000854 R.Priority = CCP_NestedNameSpecifier;
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000855 } else
856 AdjustResultPriorityForDecl(R);
Douglas Gregor265f7492010-08-27 15:29:55 +0000857
Douglas Gregor0563c262009-09-22 23:15:58 +0000858 // If this result is supposed to have an informative qualifier, add one.
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000859 if (R.QualifierIsInformative && !R.Qualifier &&
860 !R.StartsNestedNameSpecifier) {
Douglas Gregor0563c262009-09-22 23:15:58 +0000861 DeclContext *Ctx = R.Declaration->getDeclContext();
862 if (NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
863 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
864 else if (TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
865 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
866 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
867 else
868 R.QualifierIsInformative = false;
869 }
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000870
Douglas Gregor86d9a522009-09-21 16:56:56 +0000871 // Insert this result into the set of results and into the current shadow
872 // map.
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000873 SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size());
Douglas Gregor86d9a522009-09-21 16:56:56 +0000874 Results.push_back(R);
Douglas Gregor6f942b22010-09-21 16:06:22 +0000875
876 if (!AsNestedNameSpecifier)
877 MaybeAddConstructorResults(R);
Douglas Gregor86d9a522009-09-21 16:56:56 +0000878}
879
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000880void ResultBuilder::AddResult(Result R, DeclContext *CurContext,
Douglas Gregor0cc84042010-01-14 15:47:35 +0000881 NamedDecl *Hiding, bool InBaseClass = false) {
Douglas Gregora4477812010-01-14 16:01:26 +0000882 if (R.Kind != Result::RK_Declaration) {
883 // For non-declaration results, just add the result.
884 Results.push_back(R);
885 return;
886 }
887
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000888 // Look through using declarations.
889 if (UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
890 AddResult(Result(Using->getTargetDecl(), R.Qualifier), CurContext, Hiding);
891 return;
892 }
893
Douglas Gregor45bcd432010-01-14 03:21:49 +0000894 bool AsNestedNameSpecifier = false;
895 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000896 return;
897
Douglas Gregor6f942b22010-09-21 16:06:22 +0000898 // C++ constructors are never found by name lookup.
899 if (isa<CXXConstructorDecl>(R.Declaration))
900 return;
901
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000902 if (Hiding && CheckHiddenResult(R, CurContext, Hiding))
903 return;
904
905 // Make sure that any given declaration only shows up in the result set once.
906 if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl()))
907 return;
908
909 // If the filter is for nested-name-specifiers, then this result starts a
910 // nested-name-specifier.
Douglas Gregor12e13132010-05-26 22:00:08 +0000911 if (AsNestedNameSpecifier) {
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000912 R.StartsNestedNameSpecifier = true;
Douglas Gregor12e13132010-05-26 22:00:08 +0000913 R.Priority = CCP_NestedNameSpecifier;
914 }
Douglas Gregor0cc84042010-01-14 15:47:35 +0000915 else if (Filter == &ResultBuilder::IsMember && !R.Qualifier && InBaseClass &&
916 isa<CXXRecordDecl>(R.Declaration->getDeclContext()
Sebastian Redl7a126a42010-08-31 00:36:30 +0000917 ->getRedeclContext()))
Douglas Gregor0cc84042010-01-14 15:47:35 +0000918 R.QualifierIsInformative = true;
919
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000920 // If this result is supposed to have an informative qualifier, add one.
921 if (R.QualifierIsInformative && !R.Qualifier &&
922 !R.StartsNestedNameSpecifier) {
923 DeclContext *Ctx = R.Declaration->getDeclContext();
924 if (NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
925 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
926 else if (TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
927 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
Douglas Gregor45bcd432010-01-14 03:21:49 +0000928 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000929 else
930 R.QualifierIsInformative = false;
931 }
932
Douglas Gregor12e13132010-05-26 22:00:08 +0000933 // Adjust the priority if this result comes from a base class.
934 if (InBaseClass)
935 R.Priority += CCD_InBaseClass;
936
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000937 AdjustResultPriorityForDecl(R);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +0000938
Douglas Gregor3cdee122010-08-26 16:36:48 +0000939 if (HasObjectTypeQualifiers)
940 if (CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(R.Declaration))
941 if (Method->isInstance()) {
942 Qualifiers MethodQuals
943 = Qualifiers::fromCVRMask(Method->getTypeQualifiers());
944 if (ObjectTypeQualifiers == MethodQuals)
945 R.Priority += CCD_ObjectQualifierMatch;
946 else if (ObjectTypeQualifiers - MethodQuals) {
947 // The method cannot be invoked, because doing so would drop
948 // qualifiers.
949 return;
950 }
951 }
952
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000953 // Insert this result into the set of results.
954 Results.push_back(R);
Douglas Gregor6f942b22010-09-21 16:06:22 +0000955
956 if (!AsNestedNameSpecifier)
957 MaybeAddConstructorResults(R);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000958}
959
Douglas Gregora4477812010-01-14 16:01:26 +0000960void ResultBuilder::AddResult(Result R) {
961 assert(R.Kind != Result::RK_Declaration &&
962 "Declaration results need more context");
963 Results.push_back(R);
964}
965
Douglas Gregor86d9a522009-09-21 16:56:56 +0000966/// \brief Enter into a new scope.
967void ResultBuilder::EnterNewScope() {
968 ShadowMaps.push_back(ShadowMap());
969}
970
971/// \brief Exit from the current scope.
972void ResultBuilder::ExitScope() {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000973 for (ShadowMap::iterator E = ShadowMaps.back().begin(),
974 EEnd = ShadowMaps.back().end();
975 E != EEnd;
976 ++E)
977 E->second.Destroy();
978
Douglas Gregor86d9a522009-09-21 16:56:56 +0000979 ShadowMaps.pop_back();
980}
981
Douglas Gregor791215b2009-09-21 20:51:25 +0000982/// \brief Determines whether this given declaration will be found by
983/// ordinary name lookup.
984bool ResultBuilder::IsOrdinaryName(NamedDecl *ND) const {
Douglas Gregor4710e5b2010-05-28 00:49:12 +0000985 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
986
Douglas Gregor791215b2009-09-21 20:51:25 +0000987 unsigned IDNS = Decl::IDNS_Ordinary;
988 if (SemaRef.getLangOptions().CPlusPlus)
Douglas Gregor9b30b262010-06-15 20:26:51 +0000989 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
Douglas Gregorca45da02010-11-02 20:36:02 +0000990 else if (SemaRef.getLangOptions().ObjC1) {
991 if (isa<ObjCIvarDecl>(ND))
992 return true;
Douglas Gregorca45da02010-11-02 20:36:02 +0000993 }
994
Douglas Gregor791215b2009-09-21 20:51:25 +0000995 return ND->getIdentifierNamespace() & IDNS;
996}
997
Douglas Gregor01dfea02010-01-10 23:08:15 +0000998/// \brief Determines whether this given declaration will be found by
Douglas Gregor4710e5b2010-05-28 00:49:12 +0000999/// ordinary name lookup but is not a type name.
1000bool ResultBuilder::IsOrdinaryNonTypeName(NamedDecl *ND) const {
1001 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1002 if (isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND))
1003 return false;
1004
1005 unsigned IDNS = Decl::IDNS_Ordinary;
1006 if (SemaRef.getLangOptions().CPlusPlus)
Douglas Gregor9b30b262010-06-15 20:26:51 +00001007 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
Douglas Gregorca45da02010-11-02 20:36:02 +00001008 else if (SemaRef.getLangOptions().ObjC1) {
1009 if (isa<ObjCIvarDecl>(ND))
1010 return true;
Douglas Gregorca45da02010-11-02 20:36:02 +00001011 }
1012
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001013 return ND->getIdentifierNamespace() & IDNS;
1014}
1015
Douglas Gregorf9578432010-07-28 21:50:18 +00001016bool ResultBuilder::IsIntegralConstantValue(NamedDecl *ND) const {
1017 if (!IsOrdinaryNonTypeName(ND))
1018 return 0;
1019
1020 if (ValueDecl *VD = dyn_cast<ValueDecl>(ND->getUnderlyingDecl()))
1021 if (VD->getType()->isIntegralOrEnumerationType())
1022 return true;
1023
1024 return false;
1025}
1026
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001027/// \brief Determines whether this given declaration will be found by
Douglas Gregor01dfea02010-01-10 23:08:15 +00001028/// ordinary name lookup.
1029bool ResultBuilder::IsOrdinaryNonValueName(NamedDecl *ND) const {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001030 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1031
Douglas Gregor01dfea02010-01-10 23:08:15 +00001032 unsigned IDNS = Decl::IDNS_Ordinary;
1033 if (SemaRef.getLangOptions().CPlusPlus)
John McCall0d6b1642010-04-23 18:46:30 +00001034 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001035
1036 return (ND->getIdentifierNamespace() & IDNS) &&
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001037 !isa<ValueDecl>(ND) && !isa<FunctionTemplateDecl>(ND) &&
1038 !isa<ObjCPropertyDecl>(ND);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001039}
1040
Douglas Gregor86d9a522009-09-21 16:56:56 +00001041/// \brief Determines whether the given declaration is suitable as the
1042/// start of a C++ nested-name-specifier, e.g., a class or namespace.
1043bool ResultBuilder::IsNestedNameSpecifier(NamedDecl *ND) const {
1044 // Allow us to find class templates, too.
1045 if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
1046 ND = ClassTemplate->getTemplatedDecl();
1047
1048 return SemaRef.isAcceptableNestedNameSpecifier(ND);
1049}
1050
1051/// \brief Determines whether the given declaration is an enumeration.
1052bool ResultBuilder::IsEnum(NamedDecl *ND) const {
1053 return isa<EnumDecl>(ND);
1054}
1055
1056/// \brief Determines whether the given declaration is a class or struct.
1057bool ResultBuilder::IsClassOrStruct(NamedDecl *ND) const {
1058 // Allow us to find class templates, too.
1059 if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
1060 ND = ClassTemplate->getTemplatedDecl();
1061
1062 if (RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara465d41b2010-05-11 21:36:43 +00001063 return RD->getTagKind() == TTK_Class ||
1064 RD->getTagKind() == TTK_Struct;
Douglas Gregor86d9a522009-09-21 16:56:56 +00001065
1066 return false;
1067}
1068
1069/// \brief Determines whether the given declaration is a union.
1070bool ResultBuilder::IsUnion(NamedDecl *ND) const {
1071 // Allow us to find class templates, too.
1072 if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
1073 ND = ClassTemplate->getTemplatedDecl();
1074
1075 if (RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara465d41b2010-05-11 21:36:43 +00001076 return RD->getTagKind() == TTK_Union;
Douglas Gregor86d9a522009-09-21 16:56:56 +00001077
1078 return false;
1079}
1080
1081/// \brief Determines whether the given declaration is a namespace.
1082bool ResultBuilder::IsNamespace(NamedDecl *ND) const {
1083 return isa<NamespaceDecl>(ND);
1084}
1085
1086/// \brief Determines whether the given declaration is a namespace or
1087/// namespace alias.
1088bool ResultBuilder::IsNamespaceOrAlias(NamedDecl *ND) const {
1089 return isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND);
1090}
1091
Douglas Gregor76282942009-12-11 17:31:05 +00001092/// \brief Determines whether the given declaration is a type.
Douglas Gregor86d9a522009-09-21 16:56:56 +00001093bool ResultBuilder::IsType(NamedDecl *ND) const {
Douglas Gregord32b0222010-08-24 01:06:58 +00001094 if (UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
1095 ND = Using->getTargetDecl();
1096
1097 return isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND);
Douglas Gregor86d9a522009-09-21 16:56:56 +00001098}
1099
Douglas Gregor76282942009-12-11 17:31:05 +00001100/// \brief Determines which members of a class should be visible via
1101/// "." or "->". Only value declarations, nested name specifiers, and
1102/// using declarations thereof should show up.
Douglas Gregoreb5758b2009-09-23 22:26:46 +00001103bool ResultBuilder::IsMember(NamedDecl *ND) const {
Douglas Gregor76282942009-12-11 17:31:05 +00001104 if (UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
1105 ND = Using->getTargetDecl();
1106
Douglas Gregorce821962009-12-11 18:14:22 +00001107 return isa<ValueDecl>(ND) || isa<FunctionTemplateDecl>(ND) ||
1108 isa<ObjCPropertyDecl>(ND);
Douglas Gregoreb5758b2009-09-23 22:26:46 +00001109}
1110
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001111static bool isObjCReceiverType(ASTContext &C, QualType T) {
1112 T = C.getCanonicalType(T);
1113 switch (T->getTypeClass()) {
1114 case Type::ObjCObject:
1115 case Type::ObjCInterface:
1116 case Type::ObjCObjectPointer:
1117 return true;
1118
1119 case Type::Builtin:
1120 switch (cast<BuiltinType>(T)->getKind()) {
1121 case BuiltinType::ObjCId:
1122 case BuiltinType::ObjCClass:
1123 case BuiltinType::ObjCSel:
1124 return true;
1125
1126 default:
1127 break;
1128 }
1129 return false;
1130
1131 default:
1132 break;
1133 }
1134
1135 if (!C.getLangOptions().CPlusPlus)
1136 return false;
1137
1138 // FIXME: We could perform more analysis here to determine whether a
1139 // particular class type has any conversions to Objective-C types. For now,
1140 // just accept all class types.
1141 return T->isDependentType() || T->isRecordType();
1142}
1143
1144bool ResultBuilder::IsObjCMessageReceiver(NamedDecl *ND) const {
1145 QualType T = getDeclUsageType(SemaRef.Context, ND);
1146 if (T.isNull())
1147 return false;
1148
1149 T = SemaRef.Context.getBaseElementType(T);
1150 return isObjCReceiverType(SemaRef.Context, T);
1151}
1152
Douglas Gregorfb629412010-08-23 21:17:50 +00001153bool ResultBuilder::IsObjCCollection(NamedDecl *ND) const {
1154 if ((SemaRef.getLangOptions().CPlusPlus && !IsOrdinaryName(ND)) ||
1155 (!SemaRef.getLangOptions().CPlusPlus && !IsOrdinaryNonTypeName(ND)))
1156 return false;
1157
1158 QualType T = getDeclUsageType(SemaRef.Context, ND);
1159 if (T.isNull())
1160 return false;
1161
1162 T = SemaRef.Context.getBaseElementType(T);
1163 return T->isObjCObjectType() || T->isObjCObjectPointerType() ||
1164 T->isObjCIdType() ||
1165 (SemaRef.getLangOptions().CPlusPlus && T->isRecordType());
1166}
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001167
Douglas Gregor52779fb2010-09-23 23:01:17 +00001168bool ResultBuilder::IsImpossibleToSatisfy(NamedDecl *ND) const {
1169 return false;
1170}
1171
Douglas Gregor80f4f4c2010-01-14 16:08:12 +00001172/// \rief Determines whether the given declaration is an Objective-C
1173/// instance variable.
1174bool ResultBuilder::IsObjCIvar(NamedDecl *ND) const {
1175 return isa<ObjCIvarDecl>(ND);
1176}
1177
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001178namespace {
1179 /// \brief Visible declaration consumer that adds a code-completion result
1180 /// for each visible declaration.
1181 class CodeCompletionDeclConsumer : public VisibleDeclConsumer {
1182 ResultBuilder &Results;
1183 DeclContext *CurContext;
1184
1185 public:
1186 CodeCompletionDeclConsumer(ResultBuilder &Results, DeclContext *CurContext)
1187 : Results(Results), CurContext(CurContext) { }
1188
Douglas Gregor0cc84042010-01-14 15:47:35 +00001189 virtual void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, bool InBaseClass) {
1190 Results.AddResult(ND, CurContext, Hiding, InBaseClass);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001191 }
1192 };
1193}
1194
Douglas Gregor86d9a522009-09-21 16:56:56 +00001195/// \brief Add type specifiers for the current language as keyword results.
Douglas Gregorbca403c2010-01-13 23:51:12 +00001196static void AddTypeSpecifierResults(const LangOptions &LangOpts,
Douglas Gregor86d9a522009-09-21 16:56:56 +00001197 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00001198 typedef CodeCompletionResult Result;
Douglas Gregor12e13132010-05-26 22:00:08 +00001199 Results.AddResult(Result("short", CCP_Type));
1200 Results.AddResult(Result("long", CCP_Type));
1201 Results.AddResult(Result("signed", CCP_Type));
1202 Results.AddResult(Result("unsigned", CCP_Type));
1203 Results.AddResult(Result("void", CCP_Type));
1204 Results.AddResult(Result("char", CCP_Type));
1205 Results.AddResult(Result("int", CCP_Type));
1206 Results.AddResult(Result("float", CCP_Type));
1207 Results.AddResult(Result("double", CCP_Type));
1208 Results.AddResult(Result("enum", CCP_Type));
1209 Results.AddResult(Result("struct", CCP_Type));
1210 Results.AddResult(Result("union", CCP_Type));
1211 Results.AddResult(Result("const", CCP_Type));
1212 Results.AddResult(Result("volatile", CCP_Type));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001213
Douglas Gregor86d9a522009-09-21 16:56:56 +00001214 if (LangOpts.C99) {
1215 // C99-specific
Douglas Gregor12e13132010-05-26 22:00:08 +00001216 Results.AddResult(Result("_Complex", CCP_Type));
1217 Results.AddResult(Result("_Imaginary", CCP_Type));
1218 Results.AddResult(Result("_Bool", CCP_Type));
1219 Results.AddResult(Result("restrict", CCP_Type));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001220 }
1221
Douglas Gregor218937c2011-02-01 19:23:04 +00001222 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregor86d9a522009-09-21 16:56:56 +00001223 if (LangOpts.CPlusPlus) {
1224 // C++-specific
Douglas Gregorb05496d2010-09-20 21:11:48 +00001225 Results.AddResult(Result("bool", CCP_Type +
1226 (LangOpts.ObjC1? CCD_bool_in_ObjC : 0)));
Douglas Gregor12e13132010-05-26 22:00:08 +00001227 Results.AddResult(Result("class", CCP_Type));
1228 Results.AddResult(Result("wchar_t", CCP_Type));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001229
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001230 // typename qualified-id
Douglas Gregor218937c2011-02-01 19:23:04 +00001231 Builder.AddTypedTextChunk("typename");
1232 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1233 Builder.AddPlaceholderChunk("qualifier");
1234 Builder.AddTextChunk("::");
1235 Builder.AddPlaceholderChunk("name");
1236 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001237
Douglas Gregor86d9a522009-09-21 16:56:56 +00001238 if (LangOpts.CPlusPlus0x) {
Douglas Gregor12e13132010-05-26 22:00:08 +00001239 Results.AddResult(Result("auto", CCP_Type));
1240 Results.AddResult(Result("char16_t", CCP_Type));
1241 Results.AddResult(Result("char32_t", CCP_Type));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001242
Douglas Gregor218937c2011-02-01 19:23:04 +00001243 Builder.AddTypedTextChunk("decltype");
1244 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1245 Builder.AddPlaceholderChunk("expression");
1246 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1247 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001248 }
1249 }
1250
1251 // GNU extensions
1252 if (LangOpts.GNUMode) {
1253 // FIXME: Enable when we actually support decimal floating point.
Douglas Gregora4477812010-01-14 16:01:26 +00001254 // Results.AddResult(Result("_Decimal32"));
1255 // Results.AddResult(Result("_Decimal64"));
1256 // Results.AddResult(Result("_Decimal128"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001257
Douglas Gregor218937c2011-02-01 19:23:04 +00001258 Builder.AddTypedTextChunk("typeof");
1259 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1260 Builder.AddPlaceholderChunk("expression");
1261 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001262
Douglas Gregor218937c2011-02-01 19:23:04 +00001263 Builder.AddTypedTextChunk("typeof");
1264 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1265 Builder.AddPlaceholderChunk("type");
1266 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1267 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001268 }
1269}
1270
John McCallf312b1e2010-08-26 23:41:50 +00001271static void AddStorageSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001272 const LangOptions &LangOpts,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001273 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00001274 typedef CodeCompletionResult Result;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001275 // Note: we don't suggest either "auto" or "register", because both
1276 // are pointless as storage specifiers. Elsewhere, we suggest "auto"
1277 // in C++0x as a type specifier.
Douglas Gregora4477812010-01-14 16:01:26 +00001278 Results.AddResult(Result("extern"));
1279 Results.AddResult(Result("static"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001280}
1281
John McCallf312b1e2010-08-26 23:41:50 +00001282static void AddFunctionSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001283 const LangOptions &LangOpts,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001284 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00001285 typedef CodeCompletionResult Result;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001286 switch (CCC) {
John McCallf312b1e2010-08-26 23:41:50 +00001287 case Sema::PCC_Class:
1288 case Sema::PCC_MemberTemplate:
Douglas Gregor01dfea02010-01-10 23:08:15 +00001289 if (LangOpts.CPlusPlus) {
Douglas Gregora4477812010-01-14 16:01:26 +00001290 Results.AddResult(Result("explicit"));
1291 Results.AddResult(Result("friend"));
1292 Results.AddResult(Result("mutable"));
1293 Results.AddResult(Result("virtual"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001294 }
1295 // Fall through
1296
John McCallf312b1e2010-08-26 23:41:50 +00001297 case Sema::PCC_ObjCInterface:
1298 case Sema::PCC_ObjCImplementation:
1299 case Sema::PCC_Namespace:
1300 case Sema::PCC_Template:
Douglas Gregor01dfea02010-01-10 23:08:15 +00001301 if (LangOpts.CPlusPlus || LangOpts.C99)
Douglas Gregora4477812010-01-14 16:01:26 +00001302 Results.AddResult(Result("inline"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001303 break;
1304
John McCallf312b1e2010-08-26 23:41:50 +00001305 case Sema::PCC_ObjCInstanceVariableList:
1306 case Sema::PCC_Expression:
1307 case Sema::PCC_Statement:
1308 case Sema::PCC_ForInit:
1309 case Sema::PCC_Condition:
1310 case Sema::PCC_RecoveryInFunction:
1311 case Sema::PCC_Type:
Douglas Gregor02688102010-09-14 23:59:36 +00001312 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00001313 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor01dfea02010-01-10 23:08:15 +00001314 break;
1315 }
1316}
1317
Douglas Gregorbca403c2010-01-13 23:51:12 +00001318static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt);
1319static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt);
1320static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregorc38c3e12010-01-13 21:54:15 +00001321 ResultBuilder &Results,
1322 bool NeedAt);
Douglas Gregorbca403c2010-01-13 23:51:12 +00001323static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001324 ResultBuilder &Results,
1325 bool NeedAt);
Douglas Gregorbca403c2010-01-13 23:51:12 +00001326static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001327 ResultBuilder &Results,
1328 bool NeedAt);
Douglas Gregorbca403c2010-01-13 23:51:12 +00001329static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001330
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001331static void AddTypedefResult(ResultBuilder &Results) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001332 CodeCompletionBuilder Builder(Results.getAllocator());
1333 Builder.AddTypedTextChunk("typedef");
1334 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1335 Builder.AddPlaceholderChunk("type");
1336 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1337 Builder.AddPlaceholderChunk("name");
1338 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001339}
1340
John McCallf312b1e2010-08-26 23:41:50 +00001341static bool WantTypesInContext(Sema::ParserCompletionContext CCC,
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001342 const LangOptions &LangOpts) {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001343 switch (CCC) {
John McCallf312b1e2010-08-26 23:41:50 +00001344 case Sema::PCC_Namespace:
1345 case Sema::PCC_Class:
1346 case Sema::PCC_ObjCInstanceVariableList:
1347 case Sema::PCC_Template:
1348 case Sema::PCC_MemberTemplate:
1349 case Sema::PCC_Statement:
1350 case Sema::PCC_RecoveryInFunction:
1351 case Sema::PCC_Type:
Douglas Gregor02688102010-09-14 23:59:36 +00001352 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00001353 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001354 return true;
1355
John McCallf312b1e2010-08-26 23:41:50 +00001356 case Sema::PCC_Expression:
1357 case Sema::PCC_Condition:
Douglas Gregor02688102010-09-14 23:59:36 +00001358 return LangOpts.CPlusPlus;
1359
1360 case Sema::PCC_ObjCInterface:
1361 case Sema::PCC_ObjCImplementation:
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001362 return false;
1363
John McCallf312b1e2010-08-26 23:41:50 +00001364 case Sema::PCC_ForInit:
Douglas Gregor02688102010-09-14 23:59:36 +00001365 return LangOpts.CPlusPlus || LangOpts.ObjC1 || LangOpts.C99;
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001366 }
1367
1368 return false;
1369}
1370
Douglas Gregor01dfea02010-01-10 23:08:15 +00001371/// \brief Add language constructs that show up for "ordinary" names.
John McCallf312b1e2010-08-26 23:41:50 +00001372static void AddOrdinaryNameResults(Sema::ParserCompletionContext CCC,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001373 Scope *S,
1374 Sema &SemaRef,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001375 ResultBuilder &Results) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001376 CodeCompletionBuilder Builder(Results.getAllocator());
1377
John McCall0a2c5e22010-08-25 06:19:51 +00001378 typedef CodeCompletionResult Result;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001379 switch (CCC) {
John McCallf312b1e2010-08-26 23:41:50 +00001380 case Sema::PCC_Namespace:
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001381 if (SemaRef.getLangOptions().CPlusPlus) {
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001382 if (Results.includeCodePatterns()) {
1383 // namespace <identifier> { declarations }
Douglas Gregor218937c2011-02-01 19:23:04 +00001384 Builder.AddTypedTextChunk("namespace");
1385 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1386 Builder.AddPlaceholderChunk("identifier");
1387 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1388 Builder.AddPlaceholderChunk("declarations");
1389 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1390 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1391 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001392 }
1393
Douglas Gregor01dfea02010-01-10 23:08:15 +00001394 // namespace identifier = identifier ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001395 Builder.AddTypedTextChunk("namespace");
1396 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1397 Builder.AddPlaceholderChunk("name");
1398 Builder.AddChunk(CodeCompletionString::CK_Equal);
1399 Builder.AddPlaceholderChunk("namespace");
1400 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001401
1402 // Using directives
Douglas Gregor218937c2011-02-01 19:23:04 +00001403 Builder.AddTypedTextChunk("using");
1404 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1405 Builder.AddTextChunk("namespace");
1406 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1407 Builder.AddPlaceholderChunk("identifier");
1408 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001409
1410 // asm(string-literal)
Douglas Gregor218937c2011-02-01 19:23:04 +00001411 Builder.AddTypedTextChunk("asm");
1412 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1413 Builder.AddPlaceholderChunk("string-literal");
1414 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1415 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001416
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001417 if (Results.includeCodePatterns()) {
1418 // Explicit template instantiation
Douglas Gregor218937c2011-02-01 19:23:04 +00001419 Builder.AddTypedTextChunk("template");
1420 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1421 Builder.AddPlaceholderChunk("declaration");
1422 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001423 }
Douglas Gregor01dfea02010-01-10 23:08:15 +00001424 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001425
1426 if (SemaRef.getLangOptions().ObjC1)
Douglas Gregorbca403c2010-01-13 23:51:12 +00001427 AddObjCTopLevelResults(Results, true);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001428
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001429 AddTypedefResult(Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001430 // Fall through
1431
John McCallf312b1e2010-08-26 23:41:50 +00001432 case Sema::PCC_Class:
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001433 if (SemaRef.getLangOptions().CPlusPlus) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001434 // Using declaration
Douglas Gregor218937c2011-02-01 19:23:04 +00001435 Builder.AddTypedTextChunk("using");
1436 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1437 Builder.AddPlaceholderChunk("qualifier");
1438 Builder.AddTextChunk("::");
1439 Builder.AddPlaceholderChunk("name");
1440 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001441
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001442 // using typename qualifier::name (only in a dependent context)
Douglas Gregor01dfea02010-01-10 23:08:15 +00001443 if (SemaRef.CurContext->isDependentContext()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001444 Builder.AddTypedTextChunk("using");
1445 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1446 Builder.AddTextChunk("typename");
1447 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1448 Builder.AddPlaceholderChunk("qualifier");
1449 Builder.AddTextChunk("::");
1450 Builder.AddPlaceholderChunk("name");
1451 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001452 }
1453
John McCallf312b1e2010-08-26 23:41:50 +00001454 if (CCC == Sema::PCC_Class) {
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001455 AddTypedefResult(Results);
1456
Douglas Gregor01dfea02010-01-10 23:08:15 +00001457 // public:
Douglas Gregor218937c2011-02-01 19:23:04 +00001458 Builder.AddTypedTextChunk("public");
1459 Builder.AddChunk(CodeCompletionString::CK_Colon);
1460 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001461
1462 // protected:
Douglas Gregor218937c2011-02-01 19:23:04 +00001463 Builder.AddTypedTextChunk("protected");
1464 Builder.AddChunk(CodeCompletionString::CK_Colon);
1465 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001466
1467 // private:
Douglas Gregor218937c2011-02-01 19:23:04 +00001468 Builder.AddTypedTextChunk("private");
1469 Builder.AddChunk(CodeCompletionString::CK_Colon);
1470 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001471 }
1472 }
1473 // Fall through
1474
John McCallf312b1e2010-08-26 23:41:50 +00001475 case Sema::PCC_Template:
1476 case Sema::PCC_MemberTemplate:
Douglas Gregord8e8a582010-05-25 21:41:55 +00001477 if (SemaRef.getLangOptions().CPlusPlus && Results.includeCodePatterns()) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001478 // template < parameters >
Douglas Gregor218937c2011-02-01 19:23:04 +00001479 Builder.AddTypedTextChunk("template");
1480 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1481 Builder.AddPlaceholderChunk("parameters");
1482 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1483 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001484 }
1485
Douglas Gregorbca403c2010-01-13 23:51:12 +00001486 AddStorageSpecifiers(CCC, SemaRef.getLangOptions(), Results);
1487 AddFunctionSpecifiers(CCC, SemaRef.getLangOptions(), Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001488 break;
1489
John McCallf312b1e2010-08-26 23:41:50 +00001490 case Sema::PCC_ObjCInterface:
Douglas Gregorbca403c2010-01-13 23:51:12 +00001491 AddObjCInterfaceResults(SemaRef.getLangOptions(), Results, true);
1492 AddStorageSpecifiers(CCC, SemaRef.getLangOptions(), Results);
1493 AddFunctionSpecifiers(CCC, SemaRef.getLangOptions(), Results);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001494 break;
1495
John McCallf312b1e2010-08-26 23:41:50 +00001496 case Sema::PCC_ObjCImplementation:
Douglas Gregorbca403c2010-01-13 23:51:12 +00001497 AddObjCImplementationResults(SemaRef.getLangOptions(), Results, true);
1498 AddStorageSpecifiers(CCC, SemaRef.getLangOptions(), Results);
1499 AddFunctionSpecifiers(CCC, SemaRef.getLangOptions(), Results);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001500 break;
1501
John McCallf312b1e2010-08-26 23:41:50 +00001502 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregorbca403c2010-01-13 23:51:12 +00001503 AddObjCVisibilityResults(SemaRef.getLangOptions(), Results, true);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00001504 break;
1505
John McCallf312b1e2010-08-26 23:41:50 +00001506 case Sema::PCC_RecoveryInFunction:
1507 case Sema::PCC_Statement: {
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001508 AddTypedefResult(Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001509
Douglas Gregorec3310a2011-04-12 02:47:21 +00001510 if (SemaRef.getLangOptions().CPlusPlus && Results.includeCodePatterns() &&
1511 SemaRef.getLangOptions().CXXExceptions) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001512 Builder.AddTypedTextChunk("try");
1513 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1514 Builder.AddPlaceholderChunk("statements");
1515 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1516 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1517 Builder.AddTextChunk("catch");
1518 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1519 Builder.AddPlaceholderChunk("declaration");
1520 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1521 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1522 Builder.AddPlaceholderChunk("statements");
1523 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1524 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1525 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001526 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001527 if (SemaRef.getLangOptions().ObjC1)
Douglas Gregorbca403c2010-01-13 23:51:12 +00001528 AddObjCStatementResults(Results, true);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001529
Douglas Gregord8e8a582010-05-25 21:41:55 +00001530 if (Results.includeCodePatterns()) {
1531 // if (condition) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00001532 Builder.AddTypedTextChunk("if");
1533 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregord8e8a582010-05-25 21:41:55 +00001534 if (SemaRef.getLangOptions().CPlusPlus)
Douglas Gregor218937c2011-02-01 19:23:04 +00001535 Builder.AddPlaceholderChunk("condition");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001536 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001537 Builder.AddPlaceholderChunk("expression");
1538 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1539 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1540 Builder.AddPlaceholderChunk("statements");
1541 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1542 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1543 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001544
Douglas Gregord8e8a582010-05-25 21:41:55 +00001545 // switch (condition) { }
Douglas Gregor218937c2011-02-01 19:23:04 +00001546 Builder.AddTypedTextChunk("switch");
1547 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregord8e8a582010-05-25 21:41:55 +00001548 if (SemaRef.getLangOptions().CPlusPlus)
Douglas Gregor218937c2011-02-01 19:23:04 +00001549 Builder.AddPlaceholderChunk("condition");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001550 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001551 Builder.AddPlaceholderChunk("expression");
1552 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1553 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1554 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1555 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1556 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001557 }
1558
Douglas Gregor01dfea02010-01-10 23:08:15 +00001559 // Switch-specific statements.
John McCall781472f2010-08-25 08:40:02 +00001560 if (!SemaRef.getCurFunction()->SwitchStack.empty()) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001561 // case expression:
Douglas Gregor218937c2011-02-01 19:23:04 +00001562 Builder.AddTypedTextChunk("case");
1563 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1564 Builder.AddPlaceholderChunk("expression");
1565 Builder.AddChunk(CodeCompletionString::CK_Colon);
1566 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001567
1568 // default:
Douglas Gregor218937c2011-02-01 19:23:04 +00001569 Builder.AddTypedTextChunk("default");
1570 Builder.AddChunk(CodeCompletionString::CK_Colon);
1571 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001572 }
1573
Douglas Gregord8e8a582010-05-25 21:41:55 +00001574 if (Results.includeCodePatterns()) {
1575 /// while (condition) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00001576 Builder.AddTypedTextChunk("while");
1577 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregord8e8a582010-05-25 21:41:55 +00001578 if (SemaRef.getLangOptions().CPlusPlus)
Douglas Gregor218937c2011-02-01 19:23:04 +00001579 Builder.AddPlaceholderChunk("condition");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001580 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001581 Builder.AddPlaceholderChunk("expression");
1582 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1583 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1584 Builder.AddPlaceholderChunk("statements");
1585 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1586 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1587 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001588
1589 // do { statements } while ( expression );
Douglas Gregor218937c2011-02-01 19:23:04 +00001590 Builder.AddTypedTextChunk("do");
1591 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1592 Builder.AddPlaceholderChunk("statements");
1593 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1594 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1595 Builder.AddTextChunk("while");
1596 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1597 Builder.AddPlaceholderChunk("expression");
1598 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1599 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001600
Douglas Gregord8e8a582010-05-25 21:41:55 +00001601 // for ( for-init-statement ; condition ; expression ) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00001602 Builder.AddTypedTextChunk("for");
1603 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregord8e8a582010-05-25 21:41:55 +00001604 if (SemaRef.getLangOptions().CPlusPlus || SemaRef.getLangOptions().C99)
Douglas Gregor218937c2011-02-01 19:23:04 +00001605 Builder.AddPlaceholderChunk("init-statement");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001606 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001607 Builder.AddPlaceholderChunk("init-expression");
1608 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1609 Builder.AddPlaceholderChunk("condition");
1610 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1611 Builder.AddPlaceholderChunk("inc-expression");
1612 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1613 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1614 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1615 Builder.AddPlaceholderChunk("statements");
1616 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1617 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1618 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001619 }
Douglas Gregor01dfea02010-01-10 23:08:15 +00001620
1621 if (S->getContinueParent()) {
1622 // continue ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001623 Builder.AddTypedTextChunk("continue");
1624 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001625 }
1626
1627 if (S->getBreakParent()) {
1628 // break ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001629 Builder.AddTypedTextChunk("break");
1630 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001631 }
1632
1633 // "return expression ;" or "return ;", depending on whether we
1634 // know the function is void or not.
1635 bool isVoid = false;
1636 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext))
1637 isVoid = Function->getResultType()->isVoidType();
1638 else if (ObjCMethodDecl *Method
1639 = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext))
1640 isVoid = Method->getResultType()->isVoidType();
Douglas Gregor9ea9bdb2010-03-01 23:15:13 +00001641 else if (SemaRef.getCurBlock() &&
1642 !SemaRef.getCurBlock()->ReturnType.isNull())
1643 isVoid = SemaRef.getCurBlock()->ReturnType->isVoidType();
Douglas Gregor218937c2011-02-01 19:23:04 +00001644 Builder.AddTypedTextChunk("return");
Douglas Gregor93298002010-02-18 04:06:48 +00001645 if (!isVoid) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001646 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1647 Builder.AddPlaceholderChunk("expression");
Douglas Gregor93298002010-02-18 04:06:48 +00001648 }
Douglas Gregor218937c2011-02-01 19:23:04 +00001649 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001650
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001651 // goto identifier ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001652 Builder.AddTypedTextChunk("goto");
1653 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1654 Builder.AddPlaceholderChunk("label");
1655 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001656
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001657 // Using directives
Douglas Gregor218937c2011-02-01 19:23:04 +00001658 Builder.AddTypedTextChunk("using");
1659 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1660 Builder.AddTextChunk("namespace");
1661 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1662 Builder.AddPlaceholderChunk("identifier");
1663 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001664 }
1665
1666 // Fall through (for statement expressions).
John McCallf312b1e2010-08-26 23:41:50 +00001667 case Sema::PCC_ForInit:
1668 case Sema::PCC_Condition:
Douglas Gregorbca403c2010-01-13 23:51:12 +00001669 AddStorageSpecifiers(CCC, SemaRef.getLangOptions(), Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001670 // Fall through: conditions and statements can have expressions.
1671
Douglas Gregor02688102010-09-14 23:59:36 +00001672 case Sema::PCC_ParenthesizedExpression:
John McCallf85e1932011-06-15 23:02:42 +00001673 if (SemaRef.getLangOptions().ObjCAutoRefCount &&
1674 CCC == Sema::PCC_ParenthesizedExpression) {
1675 // (__bridge <type>)<expression>
1676 Builder.AddTypedTextChunk("__bridge");
1677 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1678 Builder.AddPlaceholderChunk("type");
1679 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1680 Builder.AddPlaceholderChunk("expression");
1681 Results.AddResult(Result(Builder.TakeString()));
1682
1683 // (__bridge_transfer <Objective-C type>)<expression>
1684 Builder.AddTypedTextChunk("__bridge_transfer");
1685 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1686 Builder.AddPlaceholderChunk("Objective-C type");
1687 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1688 Builder.AddPlaceholderChunk("expression");
1689 Results.AddResult(Result(Builder.TakeString()));
1690
1691 // (__bridge_retained <CF type>)<expression>
1692 Builder.AddTypedTextChunk("__bridge_retained");
1693 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1694 Builder.AddPlaceholderChunk("CF type");
1695 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1696 Builder.AddPlaceholderChunk("expression");
1697 Results.AddResult(Result(Builder.TakeString()));
1698 }
1699 // Fall through
1700
John McCallf312b1e2010-08-26 23:41:50 +00001701 case Sema::PCC_Expression: {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001702 if (SemaRef.getLangOptions().CPlusPlus) {
1703 // 'this', if we're in a non-static member function.
1704 if (CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(SemaRef.CurContext))
1705 if (!Method->isStatic())
Douglas Gregora4477812010-01-14 16:01:26 +00001706 Results.AddResult(Result("this"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001707
1708 // true, false
Douglas Gregora4477812010-01-14 16:01:26 +00001709 Results.AddResult(Result("true"));
1710 Results.AddResult(Result("false"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001711
Douglas Gregorec3310a2011-04-12 02:47:21 +00001712 if (SemaRef.getLangOptions().RTTI) {
1713 // dynamic_cast < type-id > ( expression )
1714 Builder.AddTypedTextChunk("dynamic_cast");
1715 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1716 Builder.AddPlaceholderChunk("type");
1717 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1718 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1719 Builder.AddPlaceholderChunk("expression");
1720 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1721 Results.AddResult(Result(Builder.TakeString()));
1722 }
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001723
1724 // static_cast < type-id > ( expression )
Douglas Gregor218937c2011-02-01 19:23:04 +00001725 Builder.AddTypedTextChunk("static_cast");
1726 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1727 Builder.AddPlaceholderChunk("type");
1728 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1729 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1730 Builder.AddPlaceholderChunk("expression");
1731 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1732 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001733
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001734 // reinterpret_cast < type-id > ( expression )
Douglas Gregor218937c2011-02-01 19:23:04 +00001735 Builder.AddTypedTextChunk("reinterpret_cast");
1736 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1737 Builder.AddPlaceholderChunk("type");
1738 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1739 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1740 Builder.AddPlaceholderChunk("expression");
1741 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1742 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001743
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001744 // const_cast < type-id > ( expression )
Douglas Gregor218937c2011-02-01 19:23:04 +00001745 Builder.AddTypedTextChunk("const_cast");
1746 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1747 Builder.AddPlaceholderChunk("type");
1748 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1749 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1750 Builder.AddPlaceholderChunk("expression");
1751 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1752 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001753
Douglas Gregorec3310a2011-04-12 02:47:21 +00001754 if (SemaRef.getLangOptions().RTTI) {
1755 // typeid ( expression-or-type )
1756 Builder.AddTypedTextChunk("typeid");
1757 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1758 Builder.AddPlaceholderChunk("expression-or-type");
1759 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1760 Results.AddResult(Result(Builder.TakeString()));
1761 }
1762
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001763 // new T ( ... )
Douglas Gregor218937c2011-02-01 19:23:04 +00001764 Builder.AddTypedTextChunk("new");
1765 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1766 Builder.AddPlaceholderChunk("type");
1767 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1768 Builder.AddPlaceholderChunk("expressions");
1769 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1770 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001771
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001772 // new T [ ] ( ... )
Douglas Gregor218937c2011-02-01 19:23:04 +00001773 Builder.AddTypedTextChunk("new");
1774 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1775 Builder.AddPlaceholderChunk("type");
1776 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1777 Builder.AddPlaceholderChunk("size");
1778 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1779 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1780 Builder.AddPlaceholderChunk("expressions");
1781 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1782 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001783
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001784 // delete expression
Douglas Gregor218937c2011-02-01 19:23:04 +00001785 Builder.AddTypedTextChunk("delete");
1786 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1787 Builder.AddPlaceholderChunk("expression");
1788 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001789
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001790 // delete [] expression
Douglas Gregor218937c2011-02-01 19:23:04 +00001791 Builder.AddTypedTextChunk("delete");
1792 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1793 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1794 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1795 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1796 Builder.AddPlaceholderChunk("expression");
1797 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001798
Douglas Gregorec3310a2011-04-12 02:47:21 +00001799 if (SemaRef.getLangOptions().CXXExceptions) {
1800 // throw expression
1801 Builder.AddTypedTextChunk("throw");
1802 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1803 Builder.AddPlaceholderChunk("expression");
1804 Results.AddResult(Result(Builder.TakeString()));
1805 }
Douglas Gregor12e13132010-05-26 22:00:08 +00001806
1807 // FIXME: Rethrow?
Douglas Gregor01dfea02010-01-10 23:08:15 +00001808 }
1809
1810 if (SemaRef.getLangOptions().ObjC1) {
1811 // Add "super", if we're in an Objective-C class with a superclass.
Ted Kremenek681e2562010-05-31 21:43:10 +00001812 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) {
1813 // The interface can be NULL.
1814 if (ObjCInterfaceDecl *ID = Method->getClassInterface())
1815 if (ID->getSuperClass())
1816 Results.AddResult(Result("super"));
1817 }
1818
Douglas Gregorbca403c2010-01-13 23:51:12 +00001819 AddObjCExpressionResults(Results, true);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001820 }
1821
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001822 // sizeof expression
Douglas Gregor218937c2011-02-01 19:23:04 +00001823 Builder.AddTypedTextChunk("sizeof");
1824 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1825 Builder.AddPlaceholderChunk("expression-or-type");
1826 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1827 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001828 break;
1829 }
Douglas Gregord32b0222010-08-24 01:06:58 +00001830
John McCallf312b1e2010-08-26 23:41:50 +00001831 case Sema::PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00001832 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregord32b0222010-08-24 01:06:58 +00001833 break;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001834 }
1835
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001836 if (WantTypesInContext(CCC, SemaRef.getLangOptions()))
1837 AddTypeSpecifierResults(SemaRef.getLangOptions(), Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001838
John McCallf312b1e2010-08-26 23:41:50 +00001839 if (SemaRef.getLangOptions().CPlusPlus && CCC != Sema::PCC_Type)
Douglas Gregora4477812010-01-14 16:01:26 +00001840 Results.AddResult(Result("operator"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001841}
1842
Douglas Gregor30c42402011-09-27 22:38:19 +00001843/// \brief Retrieve a printing policy suitable for code completion.
Douglas Gregor8987b232011-09-27 23:30:47 +00001844static PrintingPolicy getCompletionPrintingPolicy(Sema &S) {
1845 PrintingPolicy Policy = S.getPrintingPolicy();
Douglas Gregor30c42402011-09-27 22:38:19 +00001846 Policy.AnonymousTagLocations = false;
1847 Policy.SuppressStrongLifetime = true;
1848 return Policy;
1849}
1850
Douglas Gregora63f6de2011-02-01 21:15:40 +00001851/// \brief Retrieve the string representation of the given type as a string
1852/// that has the appropriate lifetime for code completion.
1853///
1854/// This routine provides a fast path where we provide constant strings for
1855/// common type names.
Benjamin Kramerda57f3e2011-03-26 12:38:21 +00001856static const char *GetCompletionTypeString(QualType T,
1857 ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00001858 const PrintingPolicy &Policy,
Benjamin Kramerda57f3e2011-03-26 12:38:21 +00001859 CodeCompletionAllocator &Allocator) {
Douglas Gregora63f6de2011-02-01 21:15:40 +00001860 if (!T.getLocalQualifiers()) {
1861 // Built-in type names are constant strings.
1862 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T))
Douglas Gregor30c42402011-09-27 22:38:19 +00001863 return BT->getName(Policy);
Douglas Gregora63f6de2011-02-01 21:15:40 +00001864
1865 // Anonymous tag types are constant strings.
1866 if (const TagType *TagT = dyn_cast<TagType>(T))
1867 if (TagDecl *Tag = TagT->getDecl())
Richard Smith162e1c12011-04-15 14:24:37 +00001868 if (!Tag->getIdentifier() && !Tag->getTypedefNameForAnonDecl()) {
Douglas Gregora63f6de2011-02-01 21:15:40 +00001869 switch (Tag->getTagKind()) {
1870 case TTK_Struct: return "struct <anonymous>";
1871 case TTK_Class: return "class <anonymous>";
1872 case TTK_Union: return "union <anonymous>";
1873 case TTK_Enum: return "enum <anonymous>";
1874 }
1875 }
1876 }
1877
1878 // Slow path: format the type as a string.
1879 std::string Result;
1880 T.getAsStringInternal(Result, Policy);
Douglas Gregordae68752011-02-01 22:57:45 +00001881 return Allocator.CopyString(Result);
Douglas Gregora63f6de2011-02-01 21:15:40 +00001882}
1883
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00001884/// \brief If the given declaration has an associated type, add it as a result
1885/// type chunk.
1886static void AddResultTypeChunk(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00001887 const PrintingPolicy &Policy,
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00001888 NamedDecl *ND,
Douglas Gregor218937c2011-02-01 19:23:04 +00001889 CodeCompletionBuilder &Result) {
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00001890 if (!ND)
1891 return;
Douglas Gregor6f942b22010-09-21 16:06:22 +00001892
1893 // Skip constructors and conversion functions, which have their return types
1894 // built into their names.
1895 if (isa<CXXConstructorDecl>(ND) || isa<CXXConversionDecl>(ND))
1896 return;
1897
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00001898 // Determine the type of the declaration (if it has a type).
Douglas Gregor6f942b22010-09-21 16:06:22 +00001899 QualType T;
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00001900 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(ND))
1901 T = Function->getResultType();
1902 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
1903 T = Method->getResultType();
1904 else if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND))
1905 T = FunTmpl->getTemplatedDecl()->getResultType();
1906 else if (EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
1907 T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext()));
1908 else if (isa<UnresolvedUsingValueDecl>(ND)) {
1909 /* Do nothing: ignore unresolved using declarations*/
John McCallf85e1932011-06-15 23:02:42 +00001910 } else if (ValueDecl *Value = dyn_cast<ValueDecl>(ND)) {
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00001911 T = Value->getType();
John McCallf85e1932011-06-15 23:02:42 +00001912 } else if (ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00001913 T = Property->getType();
1914
1915 if (T.isNull() || Context.hasSameType(T, Context.DependentTy))
1916 return;
1917
Douglas Gregor8987b232011-09-27 23:30:47 +00001918 Result.AddResultTypeChunk(GetCompletionTypeString(T, Context, Policy,
Douglas Gregora63f6de2011-02-01 21:15:40 +00001919 Result.getAllocator()));
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00001920}
1921
Douglas Gregoraaa107a2010-08-23 23:51:41 +00001922static void MaybeAddSentinel(ASTContext &Context, NamedDecl *FunctionOrMethod,
Douglas Gregor218937c2011-02-01 19:23:04 +00001923 CodeCompletionBuilder &Result) {
Douglas Gregoraaa107a2010-08-23 23:51:41 +00001924 if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>())
1925 if (Sentinel->getSentinel() == 0) {
1926 if (Context.getLangOptions().ObjC1 &&
1927 Context.Idents.get("nil").hasMacroDefinition())
Douglas Gregor218937c2011-02-01 19:23:04 +00001928 Result.AddTextChunk(", nil");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00001929 else if (Context.Idents.get("NULL").hasMacroDefinition())
Douglas Gregor218937c2011-02-01 19:23:04 +00001930 Result.AddTextChunk(", NULL");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00001931 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001932 Result.AddTextChunk(", (void*)0");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00001933 }
1934}
1935
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00001936static void appendWithSpace(std::string &Result, StringRef Text) {
1937 if (!Result.empty())
1938 Result += ' ';
1939 Result += Text.str();
1940}
1941static std::string formatObjCParamQualifiers(unsigned ObjCQuals) {
1942 std::string Result;
1943 if (ObjCQuals & Decl::OBJC_TQ_In)
1944 appendWithSpace(Result, "in");
1945 else if (ObjCQuals & Decl::OBJC_TQ_Inout)
1946 appendWithSpace(Result, "inout");
1947 else if (ObjCQuals & Decl::OBJC_TQ_Out)
1948 appendWithSpace(Result, "out");
1949 if (ObjCQuals & Decl::OBJC_TQ_Bycopy)
1950 appendWithSpace(Result, "bycopy");
1951 else if (ObjCQuals & Decl::OBJC_TQ_Byref)
1952 appendWithSpace(Result, "byref");
1953 if (ObjCQuals & Decl::OBJC_TQ_Oneway)
1954 appendWithSpace(Result, "oneway");
1955 return Result;
1956}
1957
Douglas Gregor83482d12010-08-24 16:15:59 +00001958static std::string FormatFunctionParameter(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00001959 const PrintingPolicy &Policy,
Douglas Gregoraba48082010-08-29 19:47:46 +00001960 ParmVarDecl *Param,
1961 bool SuppressName = false) {
Douglas Gregor83482d12010-08-24 16:15:59 +00001962 bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext());
1963 if (Param->getType()->isDependentType() ||
1964 !Param->getType()->isBlockPointerType()) {
1965 // The argument for a dependent or non-block parameter is a placeholder
1966 // containing that parameter's type.
1967 std::string Result;
1968
Douglas Gregoraba48082010-08-29 19:47:46 +00001969 if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName)
Douglas Gregor83482d12010-08-24 16:15:59 +00001970 Result = Param->getIdentifier()->getName();
1971
John McCallf85e1932011-06-15 23:02:42 +00001972 Param->getType().getAsStringInternal(Result, Policy);
Douglas Gregor83482d12010-08-24 16:15:59 +00001973
1974 if (ObjCMethodParam) {
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00001975 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
1976 + Result + ")";
Douglas Gregoraba48082010-08-29 19:47:46 +00001977 if (Param->getIdentifier() && !SuppressName)
Douglas Gregor83482d12010-08-24 16:15:59 +00001978 Result += Param->getIdentifier()->getName();
1979 }
1980 return Result;
1981 }
1982
1983 // The argument for a block pointer parameter is a block literal with
1984 // the appropriate type.
Douglas Gregor830072c2011-02-15 22:37:09 +00001985 FunctionTypeLoc *Block = 0;
1986 FunctionProtoTypeLoc *BlockProto = 0;
Douglas Gregor83482d12010-08-24 16:15:59 +00001987 TypeLoc TL;
1988 if (TypeSourceInfo *TSInfo = Param->getTypeSourceInfo()) {
1989 TL = TSInfo->getTypeLoc().getUnqualifiedLoc();
1990 while (true) {
1991 // Look through typedefs.
1992 if (TypedefTypeLoc *TypedefTL = dyn_cast<TypedefTypeLoc>(&TL)) {
1993 if (TypeSourceInfo *InnerTSInfo
Richard Smith162e1c12011-04-15 14:24:37 +00001994 = TypedefTL->getTypedefNameDecl()->getTypeSourceInfo()) {
Douglas Gregor83482d12010-08-24 16:15:59 +00001995 TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc();
1996 continue;
1997 }
1998 }
1999
2000 // Look through qualified types
2001 if (QualifiedTypeLoc *QualifiedTL = dyn_cast<QualifiedTypeLoc>(&TL)) {
2002 TL = QualifiedTL->getUnqualifiedLoc();
2003 continue;
2004 }
2005
2006 // Try to get the function prototype behind the block pointer type,
2007 // then we're done.
2008 if (BlockPointerTypeLoc *BlockPtr
2009 = dyn_cast<BlockPointerTypeLoc>(&TL)) {
Abramo Bagnara723df242010-12-14 22:11:44 +00002010 TL = BlockPtr->getPointeeLoc().IgnoreParens();
Douglas Gregor830072c2011-02-15 22:37:09 +00002011 Block = dyn_cast<FunctionTypeLoc>(&TL);
2012 BlockProto = dyn_cast<FunctionProtoTypeLoc>(&TL);
Douglas Gregor83482d12010-08-24 16:15:59 +00002013 }
2014 break;
2015 }
2016 }
2017
2018 if (!Block) {
2019 // We were unable to find a FunctionProtoTypeLoc with parameter names
2020 // for the block; just use the parameter type as a placeholder.
2021 std::string Result;
John McCallf85e1932011-06-15 23:02:42 +00002022 Param->getType().getUnqualifiedType().getAsStringInternal(Result, Policy);
Douglas Gregor83482d12010-08-24 16:15:59 +00002023
2024 if (ObjCMethodParam) {
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002025 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2026 + Result + ")";
Douglas Gregor83482d12010-08-24 16:15:59 +00002027 if (Param->getIdentifier())
2028 Result += Param->getIdentifier()->getName();
2029 }
2030
2031 return Result;
2032 }
2033
2034 // We have the function prototype behind the block pointer type, as it was
2035 // written in the source.
Douglas Gregor38276252010-09-08 22:47:51 +00002036 std::string Result;
2037 QualType ResultType = Block->getTypePtr()->getResultType();
2038 if (!ResultType->isVoidType())
John McCallf85e1932011-06-15 23:02:42 +00002039 ResultType.getAsStringInternal(Result, Policy);
Douglas Gregor38276252010-09-08 22:47:51 +00002040
2041 Result = '^' + Result;
Douglas Gregor830072c2011-02-15 22:37:09 +00002042 if (!BlockProto || Block->getNumArgs() == 0) {
2043 if (BlockProto && BlockProto->getTypePtr()->isVariadic())
Douglas Gregor38276252010-09-08 22:47:51 +00002044 Result += "(...)";
Douglas Gregorc2760bc2010-10-02 23:49:58 +00002045 else
2046 Result += "(void)";
Douglas Gregor38276252010-09-08 22:47:51 +00002047 } else {
2048 Result += "(";
2049 for (unsigned I = 0, N = Block->getNumArgs(); I != N; ++I) {
2050 if (I)
2051 Result += ", ";
Douglas Gregor8987b232011-09-27 23:30:47 +00002052 Result += FormatFunctionParameter(Context, Policy, Block->getArg(I));
Douglas Gregor38276252010-09-08 22:47:51 +00002053
Douglas Gregor830072c2011-02-15 22:37:09 +00002054 if (I == N - 1 && BlockProto->getTypePtr()->isVariadic())
Douglas Gregor38276252010-09-08 22:47:51 +00002055 Result += ", ...";
2056 }
2057 Result += ")";
Douglas Gregore17794f2010-08-31 05:13:43 +00002058 }
Douglas Gregor38276252010-09-08 22:47:51 +00002059
Douglas Gregorc2760bc2010-10-02 23:49:58 +00002060 if (Param->getIdentifier())
2061 Result += Param->getIdentifier()->getName();
2062
Douglas Gregor83482d12010-08-24 16:15:59 +00002063 return Result;
2064}
2065
Douglas Gregor86d9a522009-09-21 16:56:56 +00002066/// \brief Add function parameter chunks to the given code completion string.
2067static void AddFunctionParameterChunks(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002068 const PrintingPolicy &Policy,
Douglas Gregor86d9a522009-09-21 16:56:56 +00002069 FunctionDecl *Function,
Douglas Gregor218937c2011-02-01 19:23:04 +00002070 CodeCompletionBuilder &Result,
2071 unsigned Start = 0,
2072 bool InOptional = false) {
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002073 typedef CodeCompletionString::Chunk Chunk;
Douglas Gregor218937c2011-02-01 19:23:04 +00002074 bool FirstParameter = true;
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002075
Douglas Gregor218937c2011-02-01 19:23:04 +00002076 for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002077 ParmVarDecl *Param = Function->getParamDecl(P);
2078
Douglas Gregor218937c2011-02-01 19:23:04 +00002079 if (Param->hasDefaultArg() && !InOptional) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002080 // When we see an optional default argument, put that argument and
2081 // the remaining default arguments into a new, optional string.
Douglas Gregor218937c2011-02-01 19:23:04 +00002082 CodeCompletionBuilder Opt(Result.getAllocator());
2083 if (!FirstParameter)
2084 Opt.AddChunk(Chunk(CodeCompletionString::CK_Comma));
Douglas Gregor8987b232011-09-27 23:30:47 +00002085 AddFunctionParameterChunks(Context, Policy, Function, Opt, P, true);
Douglas Gregor218937c2011-02-01 19:23:04 +00002086 Result.AddOptionalChunk(Opt.TakeString());
2087 break;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002088 }
2089
Douglas Gregor218937c2011-02-01 19:23:04 +00002090 if (FirstParameter)
2091 FirstParameter = false;
2092 else
2093 Result.AddChunk(Chunk(CodeCompletionString::CK_Comma));
2094
2095 InOptional = false;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002096
2097 // Format the placeholder string.
Douglas Gregor8987b232011-09-27 23:30:47 +00002098 std::string PlaceholderStr = FormatFunctionParameter(Context, Policy,
2099 Param);
Douglas Gregor83482d12010-08-24 16:15:59 +00002100
Douglas Gregore17794f2010-08-31 05:13:43 +00002101 if (Function->isVariadic() && P == N - 1)
2102 PlaceholderStr += ", ...";
2103
Douglas Gregor86d9a522009-09-21 16:56:56 +00002104 // Add the placeholder string.
Douglas Gregordae68752011-02-01 22:57:45 +00002105 Result.AddPlaceholderChunk(
2106 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor86d9a522009-09-21 16:56:56 +00002107 }
Douglas Gregorb3d45252009-09-22 21:42:17 +00002108
2109 if (const FunctionProtoType *Proto
2110 = Function->getType()->getAs<FunctionProtoType>())
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002111 if (Proto->isVariadic()) {
Douglas Gregore17794f2010-08-31 05:13:43 +00002112 if (Proto->getNumArgs() == 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00002113 Result.AddPlaceholderChunk("...");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002114
Douglas Gregor218937c2011-02-01 19:23:04 +00002115 MaybeAddSentinel(Context, Function, Result);
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002116 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00002117}
2118
2119/// \brief Add template parameter chunks to the given code completion string.
2120static void AddTemplateParameterChunks(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002121 const PrintingPolicy &Policy,
Douglas Gregor86d9a522009-09-21 16:56:56 +00002122 TemplateDecl *Template,
Douglas Gregor218937c2011-02-01 19:23:04 +00002123 CodeCompletionBuilder &Result,
2124 unsigned MaxParameters = 0,
2125 unsigned Start = 0,
2126 bool InDefaultArg = false) {
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002127 typedef CodeCompletionString::Chunk Chunk;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002128 bool FirstParameter = true;
2129
2130 TemplateParameterList *Params = Template->getTemplateParameters();
2131 TemplateParameterList::iterator PEnd = Params->end();
2132 if (MaxParameters)
2133 PEnd = Params->begin() + MaxParameters;
Douglas Gregor218937c2011-02-01 19:23:04 +00002134 for (TemplateParameterList::iterator P = Params->begin() + Start;
2135 P != PEnd; ++P) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002136 bool HasDefaultArg = false;
2137 std::string PlaceholderStr;
2138 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) {
2139 if (TTP->wasDeclaredWithTypename())
2140 PlaceholderStr = "typename";
2141 else
2142 PlaceholderStr = "class";
2143
2144 if (TTP->getIdentifier()) {
2145 PlaceholderStr += ' ';
2146 PlaceholderStr += TTP->getIdentifier()->getName();
2147 }
2148
2149 HasDefaultArg = TTP->hasDefaultArgument();
2150 } else if (NonTypeTemplateParmDecl *NTTP
Douglas Gregor218937c2011-02-01 19:23:04 +00002151 = dyn_cast<NonTypeTemplateParmDecl>(*P)) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002152 if (NTTP->getIdentifier())
2153 PlaceholderStr = NTTP->getIdentifier()->getName();
John McCallf85e1932011-06-15 23:02:42 +00002154 NTTP->getType().getAsStringInternal(PlaceholderStr, Policy);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002155 HasDefaultArg = NTTP->hasDefaultArgument();
2156 } else {
2157 assert(isa<TemplateTemplateParmDecl>(*P));
2158 TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P);
2159
2160 // Since putting the template argument list into the placeholder would
2161 // be very, very long, we just use an abbreviation.
2162 PlaceholderStr = "template<...> class";
2163 if (TTP->getIdentifier()) {
2164 PlaceholderStr += ' ';
2165 PlaceholderStr += TTP->getIdentifier()->getName();
2166 }
2167
2168 HasDefaultArg = TTP->hasDefaultArgument();
2169 }
2170
Douglas Gregor218937c2011-02-01 19:23:04 +00002171 if (HasDefaultArg && !InDefaultArg) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002172 // When we see an optional default argument, put that argument and
2173 // the remaining default arguments into a new, optional string.
Douglas Gregor218937c2011-02-01 19:23:04 +00002174 CodeCompletionBuilder Opt(Result.getAllocator());
2175 if (!FirstParameter)
2176 Opt.AddChunk(Chunk(CodeCompletionString::CK_Comma));
Douglas Gregor8987b232011-09-27 23:30:47 +00002177 AddTemplateParameterChunks(Context, Policy, Template, Opt, MaxParameters,
Douglas Gregor218937c2011-02-01 19:23:04 +00002178 P - Params->begin(), true);
2179 Result.AddOptionalChunk(Opt.TakeString());
2180 break;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002181 }
2182
Douglas Gregor218937c2011-02-01 19:23:04 +00002183 InDefaultArg = false;
2184
Douglas Gregor86d9a522009-09-21 16:56:56 +00002185 if (FirstParameter)
2186 FirstParameter = false;
2187 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002188 Result.AddChunk(Chunk(CodeCompletionString::CK_Comma));
Douglas Gregor86d9a522009-09-21 16:56:56 +00002189
2190 // Add the placeholder string.
Douglas Gregordae68752011-02-01 22:57:45 +00002191 Result.AddPlaceholderChunk(
2192 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor86d9a522009-09-21 16:56:56 +00002193 }
2194}
2195
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002196/// \brief Add a qualifier to the given code-completion string, if the
2197/// provided nested-name-specifier is non-NULL.
Douglas Gregora61a8792009-12-11 18:44:16 +00002198static void
Douglas Gregor218937c2011-02-01 19:23:04 +00002199AddQualifierToCompletionString(CodeCompletionBuilder &Result,
Douglas Gregora61a8792009-12-11 18:44:16 +00002200 NestedNameSpecifier *Qualifier,
2201 bool QualifierIsInformative,
Douglas Gregor8987b232011-09-27 23:30:47 +00002202 ASTContext &Context,
2203 const PrintingPolicy &Policy) {
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002204 if (!Qualifier)
2205 return;
2206
2207 std::string PrintedNNS;
2208 {
2209 llvm::raw_string_ostream OS(PrintedNNS);
Douglas Gregor8987b232011-09-27 23:30:47 +00002210 Qualifier->print(OS, Policy);
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002211 }
Douglas Gregor0563c262009-09-22 23:15:58 +00002212 if (QualifierIsInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002213 Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregor0563c262009-09-22 23:15:58 +00002214 else
Douglas Gregordae68752011-02-01 22:57:45 +00002215 Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002216}
2217
Douglas Gregor218937c2011-02-01 19:23:04 +00002218static void
2219AddFunctionTypeQualsToCompletionString(CodeCompletionBuilder &Result,
2220 FunctionDecl *Function) {
Douglas Gregora61a8792009-12-11 18:44:16 +00002221 const FunctionProtoType *Proto
2222 = Function->getType()->getAs<FunctionProtoType>();
2223 if (!Proto || !Proto->getTypeQuals())
2224 return;
2225
Douglas Gregora63f6de2011-02-01 21:15:40 +00002226 // FIXME: Add ref-qualifier!
2227
2228 // Handle single qualifiers without copying
2229 if (Proto->getTypeQuals() == Qualifiers::Const) {
2230 Result.AddInformativeChunk(" const");
2231 return;
2232 }
2233
2234 if (Proto->getTypeQuals() == Qualifiers::Volatile) {
2235 Result.AddInformativeChunk(" volatile");
2236 return;
2237 }
2238
2239 if (Proto->getTypeQuals() == Qualifiers::Restrict) {
2240 Result.AddInformativeChunk(" restrict");
2241 return;
2242 }
2243
2244 // Handle multiple qualifiers.
Douglas Gregora61a8792009-12-11 18:44:16 +00002245 std::string QualsStr;
2246 if (Proto->getTypeQuals() & Qualifiers::Const)
2247 QualsStr += " const";
2248 if (Proto->getTypeQuals() & Qualifiers::Volatile)
2249 QualsStr += " volatile";
2250 if (Proto->getTypeQuals() & Qualifiers::Restrict)
2251 QualsStr += " restrict";
Douglas Gregordae68752011-02-01 22:57:45 +00002252 Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr));
Douglas Gregora61a8792009-12-11 18:44:16 +00002253}
2254
Douglas Gregor6f942b22010-09-21 16:06:22 +00002255/// \brief Add the name of the given declaration
Douglas Gregor8987b232011-09-27 23:30:47 +00002256static void AddTypedNameChunk(ASTContext &Context, const PrintingPolicy &Policy,
2257 NamedDecl *ND, CodeCompletionBuilder &Result) {
Douglas Gregor6f942b22010-09-21 16:06:22 +00002258 typedef CodeCompletionString::Chunk Chunk;
2259
2260 DeclarationName Name = ND->getDeclName();
2261 if (!Name)
2262 return;
2263
2264 switch (Name.getNameKind()) {
Douglas Gregora63f6de2011-02-01 21:15:40 +00002265 case DeclarationName::CXXOperatorName: {
2266 const char *OperatorName = 0;
2267 switch (Name.getCXXOverloadedOperator()) {
2268 case OO_None:
2269 case OO_Conditional:
2270 case NUM_OVERLOADED_OPERATORS:
2271 OperatorName = "operator";
2272 break;
2273
2274#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
2275 case OO_##Name: OperatorName = "operator" Spelling; break;
2276#define OVERLOADED_OPERATOR_MULTI(Name,Spelling,Unary,Binary,MemberOnly)
2277#include "clang/Basic/OperatorKinds.def"
2278
2279 case OO_New: OperatorName = "operator new"; break;
2280 case OO_Delete: OperatorName = "operator delete"; break;
2281 case OO_Array_New: OperatorName = "operator new[]"; break;
2282 case OO_Array_Delete: OperatorName = "operator delete[]"; break;
2283 case OO_Call: OperatorName = "operator()"; break;
2284 case OO_Subscript: OperatorName = "operator[]"; break;
2285 }
2286 Result.AddTypedTextChunk(OperatorName);
2287 break;
2288 }
2289
Douglas Gregor6f942b22010-09-21 16:06:22 +00002290 case DeclarationName::Identifier:
2291 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor6f942b22010-09-21 16:06:22 +00002292 case DeclarationName::CXXDestructorName:
2293 case DeclarationName::CXXLiteralOperatorName:
Douglas Gregordae68752011-02-01 22:57:45 +00002294 Result.AddTypedTextChunk(
2295 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002296 break;
2297
2298 case DeclarationName::CXXUsingDirective:
2299 case DeclarationName::ObjCZeroArgSelector:
2300 case DeclarationName::ObjCOneArgSelector:
2301 case DeclarationName::ObjCMultiArgSelector:
2302 break;
2303
2304 case DeclarationName::CXXConstructorName: {
2305 CXXRecordDecl *Record = 0;
2306 QualType Ty = Name.getCXXNameType();
2307 if (const RecordType *RecordTy = Ty->getAs<RecordType>())
2308 Record = cast<CXXRecordDecl>(RecordTy->getDecl());
2309 else if (const InjectedClassNameType *InjectedTy
2310 = Ty->getAs<InjectedClassNameType>())
2311 Record = InjectedTy->getDecl();
2312 else {
Douglas Gregordae68752011-02-01 22:57:45 +00002313 Result.AddTypedTextChunk(
2314 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002315 break;
2316 }
2317
Douglas Gregordae68752011-02-01 22:57:45 +00002318 Result.AddTypedTextChunk(
2319 Result.getAllocator().CopyString(Record->getNameAsString()));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002320 if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00002321 Result.AddChunk(Chunk(CodeCompletionString::CK_LeftAngle));
Douglas Gregor8987b232011-09-27 23:30:47 +00002322 AddTemplateParameterChunks(Context, Policy, Template, Result);
Douglas Gregor218937c2011-02-01 19:23:04 +00002323 Result.AddChunk(Chunk(CodeCompletionString::CK_RightAngle));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002324 }
2325 break;
2326 }
2327 }
2328}
2329
Douglas Gregor86d9a522009-09-21 16:56:56 +00002330/// \brief If possible, create a new code completion string for the given
2331/// result.
2332///
2333/// \returns Either a new, heap-allocated code completion string describing
2334/// how to use this result, or NULL to indicate that the string or name of the
2335/// result is all that is needed.
2336CodeCompletionString *
John McCall0a2c5e22010-08-25 06:19:51 +00002337CodeCompletionResult::CreateCodeCompletionString(Sema &S,
Douglas Gregordae68752011-02-01 22:57:45 +00002338 CodeCompletionAllocator &Allocator) {
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002339 typedef CodeCompletionString::Chunk Chunk;
Douglas Gregor218937c2011-02-01 19:23:04 +00002340 CodeCompletionBuilder Result(Allocator, Priority, Availability);
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002341
Douglas Gregor8987b232011-09-27 23:30:47 +00002342 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregor218937c2011-02-01 19:23:04 +00002343 if (Kind == RK_Pattern) {
2344 Pattern->Priority = Priority;
2345 Pattern->Availability = Availability;
2346 return Pattern;
2347 }
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002348
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002349 if (Kind == RK_Keyword) {
Douglas Gregor218937c2011-02-01 19:23:04 +00002350 Result.AddTypedTextChunk(Keyword);
2351 return Result.TakeString();
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002352 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00002353
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002354 if (Kind == RK_Macro) {
2355 MacroInfo *MI = S.PP.getMacroInfo(Macro);
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002356 assert(MI && "Not a macro?");
2357
Douglas Gregordae68752011-02-01 22:57:45 +00002358 Result.AddTypedTextChunk(
2359 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002360
2361 if (!MI->isFunctionLike())
Douglas Gregor218937c2011-02-01 19:23:04 +00002362 return Result.TakeString();
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002363
2364 // Format a function-like macro with placeholders for the arguments.
Douglas Gregor218937c2011-02-01 19:23:04 +00002365 Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen));
Douglas Gregore4244702011-07-30 08:17:44 +00002366 bool CombineVariadicArgument = false;
2367 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
2368 if (MI->isVariadic() && AEnd - A > 1) {
2369 AEnd -= 2;
2370 CombineVariadicArgument = true;
2371 }
2372 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002373 if (A != MI->arg_begin())
Douglas Gregor218937c2011-02-01 19:23:04 +00002374 Result.AddChunk(Chunk(CodeCompletionString::CK_Comma));
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002375
Douglas Gregore4244702011-07-30 08:17:44 +00002376 if (!MI->isVariadic() || A + 1 != AEnd) {
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002377 // Non-variadic argument.
Douglas Gregordae68752011-02-01 22:57:45 +00002378 Result.AddPlaceholderChunk(
2379 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002380 continue;
2381 }
2382
Douglas Gregore4244702011-07-30 08:17:44 +00002383 // Variadic argument; cope with the difference between GNU and C99
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002384 // variadic macros, providing a single placeholder for the rest of the
2385 // arguments.
2386 if ((*A)->isStr("__VA_ARGS__"))
Douglas Gregor218937c2011-02-01 19:23:04 +00002387 Result.AddPlaceholderChunk("...");
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002388 else {
2389 std::string Arg = (*A)->getName();
2390 Arg += "...";
Douglas Gregordae68752011-02-01 22:57:45 +00002391 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002392 }
2393 }
Douglas Gregore4244702011-07-30 08:17:44 +00002394
2395 if (CombineVariadicArgument) {
2396 // Handle the next-to-last argument, combining it with the variadic
2397 // argument.
2398 std::string LastArg = (*A)->getName();
2399 ++A;
2400 if ((*A)->isStr("__VA_ARGS__"))
2401 LastArg += ", ...";
2402 else
2403 LastArg += ", " + (*A)->getName().str() + "...";
2404 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(LastArg));
2405 }
Douglas Gregor218937c2011-02-01 19:23:04 +00002406 Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen));
2407 return Result.TakeString();
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002408 }
2409
Douglas Gregord8e8a582010-05-25 21:41:55 +00002410 assert(Kind == RK_Declaration && "Missed a result kind?");
Douglas Gregor86d9a522009-09-21 16:56:56 +00002411 NamedDecl *ND = Declaration;
2412
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002413 if (StartsNestedNameSpecifier) {
Douglas Gregordae68752011-02-01 22:57:45 +00002414 Result.AddTypedTextChunk(
2415 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002416 Result.AddTextChunk("::");
2417 return Result.TakeString();
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002418 }
2419
Douglas Gregor8987b232011-09-27 23:30:47 +00002420 AddResultTypeChunk(S.Context, Policy, ND, Result);
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002421
Douglas Gregor86d9a522009-09-21 16:56:56 +00002422 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002423 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Douglas Gregor8987b232011-09-27 23:30:47 +00002424 S.Context, Policy);
2425 AddTypedNameChunk(S.Context, Policy, ND, Result);
Douglas Gregor218937c2011-02-01 19:23:04 +00002426 Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen));
Douglas Gregor8987b232011-09-27 23:30:47 +00002427 AddFunctionParameterChunks(S.Context, Policy, Function, Result);
Douglas Gregor218937c2011-02-01 19:23:04 +00002428 Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen));
Douglas Gregora61a8792009-12-11 18:44:16 +00002429 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregor218937c2011-02-01 19:23:04 +00002430 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002431 }
2432
2433 if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002434 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Douglas Gregor8987b232011-09-27 23:30:47 +00002435 S.Context, Policy);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002436 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Douglas Gregor8987b232011-09-27 23:30:47 +00002437 AddTypedNameChunk(S.Context, Policy, Function, Result);
Douglas Gregor6f942b22010-09-21 16:06:22 +00002438
Douglas Gregor86d9a522009-09-21 16:56:56 +00002439 // Figure out which template parameters are deduced (or have default
2440 // arguments).
Chris Lattner5f9e2722011-07-23 10:55:15 +00002441 SmallVector<bool, 16> Deduced;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002442 S.MarkDeducedTemplateParameters(FunTmpl, Deduced);
2443 unsigned LastDeducibleArgument;
2444 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2445 --LastDeducibleArgument) {
2446 if (!Deduced[LastDeducibleArgument - 1]) {
2447 // C++0x: Figure out if the template argument has a default. If so,
2448 // the user doesn't need to type this argument.
2449 // FIXME: We need to abstract template parameters better!
2450 bool HasDefaultArg = false;
2451 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregor218937c2011-02-01 19:23:04 +00002452 LastDeducibleArgument - 1);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002453 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2454 HasDefaultArg = TTP->hasDefaultArgument();
2455 else if (NonTypeTemplateParmDecl *NTTP
2456 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2457 HasDefaultArg = NTTP->hasDefaultArgument();
2458 else {
2459 assert(isa<TemplateTemplateParmDecl>(Param));
2460 HasDefaultArg
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002461 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002462 }
2463
2464 if (!HasDefaultArg)
2465 break;
2466 }
2467 }
2468
2469 if (LastDeducibleArgument) {
2470 // Some of the function template arguments cannot be deduced from a
2471 // function call, so we introduce an explicit template argument list
2472 // containing all of the arguments up to the first deducible argument.
Douglas Gregor218937c2011-02-01 19:23:04 +00002473 Result.AddChunk(Chunk(CodeCompletionString::CK_LeftAngle));
Douglas Gregor8987b232011-09-27 23:30:47 +00002474 AddTemplateParameterChunks(S.Context, Policy, FunTmpl, Result,
Douglas Gregor86d9a522009-09-21 16:56:56 +00002475 LastDeducibleArgument);
Douglas Gregor218937c2011-02-01 19:23:04 +00002476 Result.AddChunk(Chunk(CodeCompletionString::CK_RightAngle));
Douglas Gregor86d9a522009-09-21 16:56:56 +00002477 }
2478
2479 // Add the function parameters
Douglas Gregor218937c2011-02-01 19:23:04 +00002480 Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen));
Douglas Gregor8987b232011-09-27 23:30:47 +00002481 AddFunctionParameterChunks(S.Context, Policy, Function, Result);
Douglas Gregor218937c2011-02-01 19:23:04 +00002482 Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen));
Douglas Gregora61a8792009-12-11 18:44:16 +00002483 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregor218937c2011-02-01 19:23:04 +00002484 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002485 }
2486
2487 if (TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002488 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Douglas Gregor8987b232011-09-27 23:30:47 +00002489 S.Context, Policy);
Douglas Gregordae68752011-02-01 22:57:45 +00002490 Result.AddTypedTextChunk(
2491 Result.getAllocator().CopyString(Template->getNameAsString()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002492 Result.AddChunk(Chunk(CodeCompletionString::CK_LeftAngle));
Douglas Gregor8987b232011-09-27 23:30:47 +00002493 AddTemplateParameterChunks(S.Context, Policy, Template, Result);
Douglas Gregor218937c2011-02-01 19:23:04 +00002494 Result.AddChunk(Chunk(CodeCompletionString::CK_RightAngle));
2495 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002496 }
2497
Douglas Gregor9630eb62009-11-17 16:44:22 +00002498 if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregor9630eb62009-11-17 16:44:22 +00002499 Selector Sel = Method->getSelector();
2500 if (Sel.isUnarySelector()) {
Douglas Gregordae68752011-02-01 22:57:45 +00002501 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00002502 Sel.getNameForSlot(0)));
Douglas Gregor218937c2011-02-01 19:23:04 +00002503 return Result.TakeString();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002504 }
2505
Douglas Gregor813d8342011-02-18 22:29:55 +00002506 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregord3c68542009-11-19 01:08:35 +00002507 SelName += ':';
2508 if (StartParameter == 0)
Douglas Gregordae68752011-02-01 22:57:45 +00002509 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregord3c68542009-11-19 01:08:35 +00002510 else {
Douglas Gregordae68752011-02-01 22:57:45 +00002511 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregord3c68542009-11-19 01:08:35 +00002512
2513 // If there is only one parameter, and we're past it, add an empty
2514 // typed-text chunk since there is nothing to type.
2515 if (Method->param_size() == 1)
Douglas Gregor218937c2011-02-01 19:23:04 +00002516 Result.AddTypedTextChunk("");
Douglas Gregord3c68542009-11-19 01:08:35 +00002517 }
Douglas Gregor9630eb62009-11-17 16:44:22 +00002518 unsigned Idx = 0;
2519 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
2520 PEnd = Method->param_end();
2521 P != PEnd; (void)++P, ++Idx) {
2522 if (Idx > 0) {
Douglas Gregord3c68542009-11-19 01:08:35 +00002523 std::string Keyword;
2524 if (Idx > StartParameter)
Douglas Gregor218937c2011-02-01 19:23:04 +00002525 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor9630eb62009-11-17 16:44:22 +00002526 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramera0651c52011-07-26 16:59:25 +00002527 Keyword += II->getName();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002528 Keyword += ":";
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002529 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002530 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002531 else
Douglas Gregordae68752011-02-01 22:57:45 +00002532 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor9630eb62009-11-17 16:44:22 +00002533 }
Douglas Gregord3c68542009-11-19 01:08:35 +00002534
2535 // If we're before the starting parameter, skip the placeholder.
2536 if (Idx < StartParameter)
2537 continue;
Douglas Gregor9630eb62009-11-17 16:44:22 +00002538
2539 std::string Arg;
Douglas Gregor83482d12010-08-24 16:15:59 +00002540
2541 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Douglas Gregor8987b232011-09-27 23:30:47 +00002542 Arg = FormatFunctionParameter(S.Context, Policy, *P, true);
Douglas Gregor83482d12010-08-24 16:15:59 +00002543 else {
John McCallf85e1932011-06-15 23:02:42 +00002544 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002545 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2546 + Arg + ")";
Douglas Gregor83482d12010-08-24 16:15:59 +00002547 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregoraba48082010-08-29 19:47:46 +00002548 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramera0651c52011-07-26 16:59:25 +00002549 Arg += II->getName();
Douglas Gregor83482d12010-08-24 16:15:59 +00002550 }
2551
Douglas Gregore17794f2010-08-31 05:13:43 +00002552 if (Method->isVariadic() && (P + 1) == PEnd)
2553 Arg += ", ...";
2554
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002555 if (DeclaringEntity)
Douglas Gregordae68752011-02-01 22:57:45 +00002556 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002557 else if (AllParametersAreInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002558 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor4ad96852009-11-19 07:41:15 +00002559 else
Douglas Gregordae68752011-02-01 22:57:45 +00002560 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor9630eb62009-11-17 16:44:22 +00002561 }
2562
Douglas Gregor2a17af02009-12-23 00:21:46 +00002563 if (Method->isVariadic()) {
Douglas Gregore17794f2010-08-31 05:13:43 +00002564 if (Method->param_size() == 0) {
2565 if (DeclaringEntity)
Douglas Gregor218937c2011-02-01 19:23:04 +00002566 Result.AddTextChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002567 else if (AllParametersAreInformative)
Douglas Gregor218937c2011-02-01 19:23:04 +00002568 Result.AddInformativeChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002569 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002570 Result.AddPlaceholderChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002571 }
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002572
2573 MaybeAddSentinel(S.Context, Method, Result);
Douglas Gregor2a17af02009-12-23 00:21:46 +00002574 }
2575
Douglas Gregor218937c2011-02-01 19:23:04 +00002576 return Result.TakeString();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002577 }
2578
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002579 if (Qualifier)
Douglas Gregor0563c262009-09-22 23:15:58 +00002580 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Douglas Gregor8987b232011-09-27 23:30:47 +00002581 S.Context, Policy);
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002582
Douglas Gregordae68752011-02-01 22:57:45 +00002583 Result.AddTypedTextChunk(
2584 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002585 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002586}
2587
Douglas Gregor86d802e2009-09-23 00:34:09 +00002588CodeCompletionString *
2589CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2590 unsigned CurrentArg,
Douglas Gregor32be4a52010-10-11 21:37:58 +00002591 Sema &S,
Douglas Gregordae68752011-02-01 22:57:45 +00002592 CodeCompletionAllocator &Allocator) const {
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002593 typedef CodeCompletionString::Chunk Chunk;
Douglas Gregor8987b232011-09-27 23:30:47 +00002594 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCallf85e1932011-06-15 23:02:42 +00002595
Douglas Gregor218937c2011-02-01 19:23:04 +00002596 // FIXME: Set priority, availability appropriately.
2597 CodeCompletionBuilder Result(Allocator, 1, CXAvailability_Available);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002598 FunctionDecl *FDecl = getFunction();
Douglas Gregor8987b232011-09-27 23:30:47 +00002599 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002600 const FunctionProtoType *Proto
2601 = dyn_cast<FunctionProtoType>(getFunctionType());
2602 if (!FDecl && !Proto) {
2603 // Function without a prototype. Just give the return type and a
2604 // highlighted ellipsis.
2605 const FunctionType *FT = getFunctionType();
Douglas Gregora63f6de2011-02-01 21:15:40 +00002606 Result.AddTextChunk(GetCompletionTypeString(FT->getResultType(),
Douglas Gregor8987b232011-09-27 23:30:47 +00002607 S.Context, Policy,
Douglas Gregora63f6de2011-02-01 21:15:40 +00002608 Result.getAllocator()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002609 Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen));
2610 Result.AddChunk(Chunk(CodeCompletionString::CK_CurrentParameter, "..."));
2611 Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen));
2612 return Result.TakeString();
Douglas Gregor86d802e2009-09-23 00:34:09 +00002613 }
2614
2615 if (FDecl)
Douglas Gregordae68752011-02-01 22:57:45 +00002616 Result.AddTextChunk(
2617 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002618 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002619 Result.AddTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00002620 Result.getAllocator().CopyString(
John McCallf85e1932011-06-15 23:02:42 +00002621 Proto->getResultType().getAsString(Policy)));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002622
Douglas Gregor218937c2011-02-01 19:23:04 +00002623 Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002624 unsigned NumParams = FDecl? FDecl->getNumParams() : Proto->getNumArgs();
2625 for (unsigned I = 0; I != NumParams; ++I) {
2626 if (I)
Douglas Gregor218937c2011-02-01 19:23:04 +00002627 Result.AddChunk(Chunk(CodeCompletionString::CK_Comma));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002628
2629 std::string ArgString;
2630 QualType ArgType;
2631
2632 if (FDecl) {
2633 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2634 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2635 } else {
2636 ArgType = Proto->getArgType(I);
2637 }
2638
John McCallf85e1932011-06-15 23:02:42 +00002639 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002640
2641 if (I == CurrentArg)
Douglas Gregor218937c2011-02-01 19:23:04 +00002642 Result.AddChunk(Chunk(CodeCompletionString::CK_CurrentParameter,
Douglas Gregordae68752011-02-01 22:57:45 +00002643 Result.getAllocator().CopyString(ArgString)));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002644 else
Douglas Gregordae68752011-02-01 22:57:45 +00002645 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002646 }
2647
2648 if (Proto && Proto->isVariadic()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00002649 Result.AddChunk(Chunk(CodeCompletionString::CK_Comma));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002650 if (CurrentArg < NumParams)
Douglas Gregor218937c2011-02-01 19:23:04 +00002651 Result.AddTextChunk("...");
Douglas Gregor86d802e2009-09-23 00:34:09 +00002652 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002653 Result.AddChunk(Chunk(CodeCompletionString::CK_CurrentParameter, "..."));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002654 }
Douglas Gregor218937c2011-02-01 19:23:04 +00002655 Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002656
Douglas Gregor218937c2011-02-01 19:23:04 +00002657 return Result.TakeString();
Douglas Gregor86d802e2009-09-23 00:34:09 +00002658}
2659
Chris Lattner5f9e2722011-07-23 10:55:15 +00002660unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregorb05496d2010-09-20 21:11:48 +00002661 const LangOptions &LangOpts,
Douglas Gregor1827e102010-08-16 16:18:59 +00002662 bool PreferredTypeIsPointer) {
2663 unsigned Priority = CCP_Macro;
2664
Douglas Gregorb05496d2010-09-20 21:11:48 +00002665 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2666 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2667 MacroName.equals("Nil")) {
Douglas Gregor1827e102010-08-16 16:18:59 +00002668 Priority = CCP_Constant;
2669 if (PreferredTypeIsPointer)
2670 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregorb05496d2010-09-20 21:11:48 +00002671 }
2672 // Treat "YES", "NO", "true", and "false" as constants.
2673 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2674 MacroName.equals("true") || MacroName.equals("false"))
2675 Priority = CCP_Constant;
2676 // Treat "bool" as a type.
2677 else if (MacroName.equals("bool"))
2678 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2679
Douglas Gregor1827e102010-08-16 16:18:59 +00002680
2681 return Priority;
2682}
2683
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002684CXCursorKind clang::getCursorKindForDecl(Decl *D) {
2685 if (!D)
2686 return CXCursor_UnexposedDecl;
2687
2688 switch (D->getKind()) {
2689 case Decl::Enum: return CXCursor_EnumDecl;
2690 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2691 case Decl::Field: return CXCursor_FieldDecl;
2692 case Decl::Function:
2693 return CXCursor_FunctionDecl;
2694 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2695 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
2696 case Decl::ObjCClass:
2697 // FIXME
2698 return CXCursor_UnexposedDecl;
2699 case Decl::ObjCForwardProtocol:
2700 // FIXME
2701 return CXCursor_UnexposedDecl;
2702 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
2703 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
2704 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2705 case Decl::ObjCMethod:
2706 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2707 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2708 case Decl::CXXMethod: return CXCursor_CXXMethod;
2709 case Decl::CXXConstructor: return CXCursor_Constructor;
2710 case Decl::CXXDestructor: return CXCursor_Destructor;
2711 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2712 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
2713 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
2714 case Decl::ParmVar: return CXCursor_ParmDecl;
2715 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smith162e1c12011-04-15 14:24:37 +00002716 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002717 case Decl::Var: return CXCursor_VarDecl;
2718 case Decl::Namespace: return CXCursor_Namespace;
2719 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2720 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2721 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2722 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2723 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2724 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis2dfdb942011-09-30 17:58:23 +00002725 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002726 case Decl::ClassTemplatePartialSpecialization:
2727 return CXCursor_ClassTemplatePartialSpecialization;
2728 case Decl::UsingDirective: return CXCursor_UsingDirective;
2729
2730 case Decl::Using:
2731 case Decl::UnresolvedUsingValue:
2732 case Decl::UnresolvedUsingTypename:
2733 return CXCursor_UsingDeclaration;
2734
Douglas Gregor352697a2011-06-03 23:08:58 +00002735 case Decl::ObjCPropertyImpl:
2736 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2737 case ObjCPropertyImplDecl::Dynamic:
2738 return CXCursor_ObjCDynamicDecl;
2739
2740 case ObjCPropertyImplDecl::Synthesize:
2741 return CXCursor_ObjCSynthesizeDecl;
2742 }
2743 break;
2744
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002745 default:
2746 if (TagDecl *TD = dyn_cast<TagDecl>(D)) {
2747 switch (TD->getTagKind()) {
2748 case TTK_Struct: return CXCursor_StructDecl;
2749 case TTK_Class: return CXCursor_ClassDecl;
2750 case TTK_Union: return CXCursor_UnionDecl;
2751 case TTK_Enum: return CXCursor_EnumDecl;
2752 }
2753 }
2754 }
2755
2756 return CXCursor_UnexposedDecl;
2757}
2758
Douglas Gregor590c7d52010-07-08 20:55:51 +00002759static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
2760 bool TargetTypeIsPointer = false) {
John McCall0a2c5e22010-08-25 06:19:51 +00002761 typedef CodeCompletionResult Result;
Douglas Gregor590c7d52010-07-08 20:55:51 +00002762
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002763 Results.EnterNewScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002764
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002765 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2766 MEnd = PP.macro_end();
Douglas Gregor590c7d52010-07-08 20:55:51 +00002767 M != MEnd; ++M) {
Douglas Gregor1827e102010-08-16 16:18:59 +00002768 Results.AddResult(Result(M->first,
2769 getMacroUsagePriority(M->first->getName(),
Douglas Gregorb05496d2010-09-20 21:11:48 +00002770 PP.getLangOptions(),
Douglas Gregor1827e102010-08-16 16:18:59 +00002771 TargetTypeIsPointer)));
Douglas Gregor590c7d52010-07-08 20:55:51 +00002772 }
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002773
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002774 Results.ExitScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002775
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002776}
2777
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00002778static void AddPrettyFunctionResults(const LangOptions &LangOpts,
2779 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00002780 typedef CodeCompletionResult Result;
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00002781
2782 Results.EnterNewScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002783
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00002784 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
2785 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
2786 if (LangOpts.C99 || LangOpts.CPlusPlus0x)
2787 Results.AddResult(Result("__func__", CCP_Constant));
2788 Results.ExitScope();
2789}
2790
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00002791static void HandleCodeCompleteResults(Sema *S,
2792 CodeCompleteConsumer *CodeCompleter,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002793 CodeCompletionContext Context,
John McCall0a2c5e22010-08-25 06:19:51 +00002794 CodeCompletionResult *Results,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002795 unsigned NumResults) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002796 if (CodeCompleter)
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002797 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002798}
2799
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002800static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
2801 Sema::ParserCompletionContext PCC) {
2802 switch (PCC) {
John McCallf312b1e2010-08-26 23:41:50 +00002803 case Sema::PCC_Namespace:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002804 return CodeCompletionContext::CCC_TopLevel;
2805
John McCallf312b1e2010-08-26 23:41:50 +00002806 case Sema::PCC_Class:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002807 return CodeCompletionContext::CCC_ClassStructUnion;
2808
John McCallf312b1e2010-08-26 23:41:50 +00002809 case Sema::PCC_ObjCInterface:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002810 return CodeCompletionContext::CCC_ObjCInterface;
2811
John McCallf312b1e2010-08-26 23:41:50 +00002812 case Sema::PCC_ObjCImplementation:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002813 return CodeCompletionContext::CCC_ObjCImplementation;
2814
John McCallf312b1e2010-08-26 23:41:50 +00002815 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002816 return CodeCompletionContext::CCC_ObjCIvarList;
2817
John McCallf312b1e2010-08-26 23:41:50 +00002818 case Sema::PCC_Template:
2819 case Sema::PCC_MemberTemplate:
Douglas Gregor52779fb2010-09-23 23:01:17 +00002820 if (S.CurContext->isFileContext())
2821 return CodeCompletionContext::CCC_TopLevel;
2822 else if (S.CurContext->isRecord())
2823 return CodeCompletionContext::CCC_ClassStructUnion;
2824 else
2825 return CodeCompletionContext::CCC_Other;
2826
John McCallf312b1e2010-08-26 23:41:50 +00002827 case Sema::PCC_RecoveryInFunction:
Douglas Gregor52779fb2010-09-23 23:01:17 +00002828 return CodeCompletionContext::CCC_Recovery;
Douglas Gregora5450a02010-10-18 22:01:46 +00002829
John McCallf312b1e2010-08-26 23:41:50 +00002830 case Sema::PCC_ForInit:
Douglas Gregora5450a02010-10-18 22:01:46 +00002831 if (S.getLangOptions().CPlusPlus || S.getLangOptions().C99 ||
2832 S.getLangOptions().ObjC1)
2833 return CodeCompletionContext::CCC_ParenthesizedExpression;
2834 else
2835 return CodeCompletionContext::CCC_Expression;
2836
2837 case Sema::PCC_Expression:
John McCallf312b1e2010-08-26 23:41:50 +00002838 case Sema::PCC_Condition:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002839 return CodeCompletionContext::CCC_Expression;
2840
John McCallf312b1e2010-08-26 23:41:50 +00002841 case Sema::PCC_Statement:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002842 return CodeCompletionContext::CCC_Statement;
Douglas Gregor72db1082010-08-24 01:11:00 +00002843
John McCallf312b1e2010-08-26 23:41:50 +00002844 case Sema::PCC_Type:
Douglas Gregor72db1082010-08-24 01:11:00 +00002845 return CodeCompletionContext::CCC_Type;
Douglas Gregor02688102010-09-14 23:59:36 +00002846
2847 case Sema::PCC_ParenthesizedExpression:
2848 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00002849
2850 case Sema::PCC_LocalDeclarationSpecifiers:
2851 return CodeCompletionContext::CCC_Type;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002852 }
2853
2854 return CodeCompletionContext::CCC_Other;
2855}
2856
Douglas Gregorf6961522010-08-27 21:18:54 +00002857/// \brief If we're in a C++ virtual member function, add completion results
2858/// that invoke the functions we override, since it's common to invoke the
2859/// overridden function as well as adding new functionality.
2860///
2861/// \param S The semantic analysis object for which we are generating results.
2862///
2863/// \param InContext This context in which the nested-name-specifier preceding
2864/// the code-completion point
2865static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
2866 ResultBuilder &Results) {
2867 // Look through blocks.
2868 DeclContext *CurContext = S.CurContext;
2869 while (isa<BlockDecl>(CurContext))
2870 CurContext = CurContext->getParent();
2871
2872
2873 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
2874 if (!Method || !Method->isVirtual())
2875 return;
2876
2877 // We need to have names for all of the parameters, if we're going to
2878 // generate a forwarding call.
2879 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
2880 PEnd = Method->param_end();
2881 P != PEnd;
2882 ++P) {
2883 if (!(*P)->getDeclName())
2884 return;
2885 }
2886
Douglas Gregor8987b232011-09-27 23:30:47 +00002887 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorf6961522010-08-27 21:18:54 +00002888 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
2889 MEnd = Method->end_overridden_methods();
2890 M != MEnd; ++M) {
Douglas Gregor218937c2011-02-01 19:23:04 +00002891 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregorf6961522010-08-27 21:18:54 +00002892 CXXMethodDecl *Overridden = const_cast<CXXMethodDecl *>(*M);
2893 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
2894 continue;
2895
2896 // If we need a nested-name-specifier, add one now.
2897 if (!InContext) {
2898 NestedNameSpecifier *NNS
2899 = getRequiredQualification(S.Context, CurContext,
2900 Overridden->getDeclContext());
2901 if (NNS) {
2902 std::string Str;
2903 llvm::raw_string_ostream OS(Str);
Douglas Gregor8987b232011-09-27 23:30:47 +00002904 NNS->print(OS, Policy);
Douglas Gregordae68752011-02-01 22:57:45 +00002905 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorf6961522010-08-27 21:18:54 +00002906 }
2907 } else if (!InContext->Equals(Overridden->getDeclContext()))
2908 continue;
2909
Douglas Gregordae68752011-02-01 22:57:45 +00002910 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00002911 Overridden->getNameAsString()));
2912 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf6961522010-08-27 21:18:54 +00002913 bool FirstParam = true;
2914 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
2915 PEnd = Method->param_end();
2916 P != PEnd; ++P) {
2917 if (FirstParam)
2918 FirstParam = false;
2919 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002920 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf6961522010-08-27 21:18:54 +00002921
Douglas Gregordae68752011-02-01 22:57:45 +00002922 Builder.AddPlaceholderChunk(Results.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00002923 (*P)->getIdentifier()->getName()));
Douglas Gregorf6961522010-08-27 21:18:54 +00002924 }
Douglas Gregor218937c2011-02-01 19:23:04 +00002925 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2926 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorf6961522010-08-27 21:18:54 +00002927 CCP_SuperCompletion,
2928 CXCursor_CXXMethod));
2929 Results.Ignore(Overridden);
2930 }
2931}
2932
Douglas Gregor01dfea02010-01-10 23:08:15 +00002933void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002934 ParserCompletionContext CompletionContext) {
John McCall0a2c5e22010-08-25 06:19:51 +00002935 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00002936 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00002937 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorf6961522010-08-27 21:18:54 +00002938 Results.EnterNewScope();
Douglas Gregorcee9ff12010-09-20 22:39:41 +00002939
Douglas Gregor01dfea02010-01-10 23:08:15 +00002940 // Determine how to filter results, e.g., so that the names of
2941 // values (functions, enumerators, function templates, etc.) are
2942 // only allowed where we can have an expression.
2943 switch (CompletionContext) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002944 case PCC_Namespace:
2945 case PCC_Class:
2946 case PCC_ObjCInterface:
2947 case PCC_ObjCImplementation:
2948 case PCC_ObjCInstanceVariableList:
2949 case PCC_Template:
2950 case PCC_MemberTemplate:
Douglas Gregor72db1082010-08-24 01:11:00 +00002951 case PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00002952 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor01dfea02010-01-10 23:08:15 +00002953 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
2954 break;
2955
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002956 case PCC_Statement:
Douglas Gregor02688102010-09-14 23:59:36 +00002957 case PCC_ParenthesizedExpression:
Douglas Gregoreb0d0142010-08-24 23:58:17 +00002958 case PCC_Expression:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002959 case PCC_ForInit:
2960 case PCC_Condition:
Douglas Gregor4710e5b2010-05-28 00:49:12 +00002961 if (WantTypesInContext(CompletionContext, getLangOptions()))
2962 Results.setFilter(&ResultBuilder::IsOrdinaryName);
2963 else
2964 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorf6961522010-08-27 21:18:54 +00002965
2966 if (getLangOptions().CPlusPlus)
2967 MaybeAddOverrideCalls(*this, /*InContext=*/0, Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00002968 break;
Douglas Gregordc845342010-05-25 05:58:43 +00002969
Douglas Gregore6b1bb62010-08-11 21:23:17 +00002970 case PCC_RecoveryInFunction:
Douglas Gregordc845342010-05-25 05:58:43 +00002971 // Unfiltered
2972 break;
Douglas Gregor01dfea02010-01-10 23:08:15 +00002973 }
2974
Douglas Gregor3cdee122010-08-26 16:36:48 +00002975 // If we are in a C++ non-static member function, check the qualifiers on
2976 // the member function to filter/prioritize the results list.
2977 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
2978 if (CurMethod->isInstance())
2979 Results.setObjectTypeQualifiers(
2980 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
2981
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00002982 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00002983 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
2984 CodeCompleter->includeGlobals());
Douglas Gregor2a7925c2009-12-07 09:54:55 +00002985
Douglas Gregorbca403c2010-01-13 23:51:12 +00002986 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor2a7925c2009-12-07 09:54:55 +00002987 Results.ExitScope();
2988
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00002989 switch (CompletionContext) {
Douglas Gregor02688102010-09-14 23:59:36 +00002990 case PCC_ParenthesizedExpression:
Douglas Gregor72db1082010-08-24 01:11:00 +00002991 case PCC_Expression:
2992 case PCC_Statement:
2993 case PCC_RecoveryInFunction:
2994 if (S->getFnParent())
2995 AddPrettyFunctionResults(PP.getLangOptions(), Results);
2996 break;
2997
2998 case PCC_Namespace:
2999 case PCC_Class:
3000 case PCC_ObjCInterface:
3001 case PCC_ObjCImplementation:
3002 case PCC_ObjCInstanceVariableList:
3003 case PCC_Template:
3004 case PCC_MemberTemplate:
3005 case PCC_ForInit:
3006 case PCC_Condition:
3007 case PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003008 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor72db1082010-08-24 01:11:00 +00003009 break;
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003010 }
3011
Douglas Gregor0c8296d2009-11-07 00:00:49 +00003012 if (CodeCompleter->includeMacros())
Douglas Gregorbca403c2010-01-13 23:51:12 +00003013 AddMacroResults(PP, Results);
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003014
Douglas Gregorcee9ff12010-09-20 22:39:41 +00003015 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003016 Results.data(),Results.size());
Douglas Gregor791215b2009-09-21 20:51:25 +00003017}
3018
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003019static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3020 ParsedType Receiver,
3021 IdentifierInfo **SelIdents,
3022 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00003023 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003024 bool IsSuper,
3025 ResultBuilder &Results);
3026
3027void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3028 bool AllowNonIdentifiers,
3029 bool AllowNestedNameSpecifiers) {
John McCall0a2c5e22010-08-25 06:19:51 +00003030 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00003031 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003032 AllowNestedNameSpecifiers
3033 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3034 : CodeCompletionContext::CCC_Name);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003035 Results.EnterNewScope();
3036
3037 // Type qualifiers can come after names.
3038 Results.AddResult(Result("const"));
3039 Results.AddResult(Result("volatile"));
3040 if (getLangOptions().C99)
3041 Results.AddResult(Result("restrict"));
3042
3043 if (getLangOptions().CPlusPlus) {
3044 if (AllowNonIdentifiers) {
3045 Results.AddResult(Result("operator"));
3046 }
3047
3048 // Add nested-name-specifiers.
3049 if (AllowNestedNameSpecifiers) {
3050 Results.allowNestedNameSpecifiers();
Douglas Gregor52779fb2010-09-23 23:01:17 +00003051 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003052 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3053 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3054 CodeCompleter->includeGlobals());
Douglas Gregor52779fb2010-09-23 23:01:17 +00003055 Results.setFilter(0);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003056 }
3057 }
3058 Results.ExitScope();
3059
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003060 // If we're in a context where we might have an expression (rather than a
3061 // declaration), and what we've seen so far is an Objective-C type that could
3062 // be a receiver of a class message, this may be a class message send with
3063 // the initial opening bracket '[' missing. Add appropriate completions.
3064 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
3065 DS.getTypeSpecType() == DeclSpec::TST_typename &&
3066 DS.getStorageClassSpecAsWritten() == DeclSpec::SCS_unspecified &&
3067 !DS.isThreadSpecified() && !DS.isExternInLinkageSpec() &&
3068 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3069 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
3070 DS.getTypeQualifiers() == 0 &&
3071 S &&
3072 (S->getFlags() & Scope::DeclScope) != 0 &&
3073 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3074 Scope::FunctionPrototypeScope |
3075 Scope::AtCatchScope)) == 0) {
3076 ParsedType T = DS.getRepAsType();
3077 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Douglas Gregor70c5ac72010-09-20 23:34:21 +00003078 AddClassMessageCompletions(*this, S, T, 0, 0, false, false, Results);
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003079 }
3080
Douglas Gregor4497dd42010-08-24 04:59:56 +00003081 // Note that we intentionally suppress macro results here, since we do not
3082 // encourage using macros to produce the names of entities.
3083
Douglas Gregor52779fb2010-09-23 23:01:17 +00003084 HandleCodeCompleteResults(this, CodeCompleter,
3085 Results.getCompletionContext(),
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003086 Results.data(), Results.size());
3087}
3088
Douglas Gregorfb629412010-08-23 21:17:50 +00003089struct Sema::CodeCompleteExpressionData {
3090 CodeCompleteExpressionData(QualType PreferredType = QualType())
3091 : PreferredType(PreferredType), IntegralConstantExpression(false),
3092 ObjCCollection(false) { }
3093
3094 QualType PreferredType;
3095 bool IntegralConstantExpression;
3096 bool ObjCCollection;
Chris Lattner5f9e2722011-07-23 10:55:15 +00003097 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregorfb629412010-08-23 21:17:50 +00003098};
3099
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003100/// \brief Perform code-completion in an expression context when we know what
3101/// type we're looking for.
Douglas Gregorf9578432010-07-28 21:50:18 +00003102///
3103/// \param IntegralConstantExpression Only permit integral constant
3104/// expressions.
Douglas Gregorfb629412010-08-23 21:17:50 +00003105void Sema::CodeCompleteExpression(Scope *S,
3106 const CodeCompleteExpressionData &Data) {
John McCall0a2c5e22010-08-25 06:19:51 +00003107 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00003108 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3109 CodeCompletionContext::CCC_Expression);
Douglas Gregorfb629412010-08-23 21:17:50 +00003110 if (Data.ObjCCollection)
3111 Results.setFilter(&ResultBuilder::IsObjCCollection);
3112 else if (Data.IntegralConstantExpression)
Douglas Gregorf9578432010-07-28 21:50:18 +00003113 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003114 else if (WantTypesInContext(PCC_Expression, getLangOptions()))
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003115 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3116 else
3117 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorfb629412010-08-23 21:17:50 +00003118
3119 if (!Data.PreferredType.isNull())
3120 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3121
3122 // Ignore any declarations that we were told that we don't care about.
3123 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3124 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003125
3126 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003127 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3128 CodeCompleter->includeGlobals());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003129
3130 Results.EnterNewScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003131 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003132 Results.ExitScope();
3133
Douglas Gregor590c7d52010-07-08 20:55:51 +00003134 bool PreferredTypeIsPointer = false;
Douglas Gregorfb629412010-08-23 21:17:50 +00003135 if (!Data.PreferredType.isNull())
3136 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3137 || Data.PreferredType->isMemberPointerType()
3138 || Data.PreferredType->isBlockPointerType();
Douglas Gregor590c7d52010-07-08 20:55:51 +00003139
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003140 if (S->getFnParent() &&
3141 !Data.ObjCCollection &&
3142 !Data.IntegralConstantExpression)
3143 AddPrettyFunctionResults(PP.getLangOptions(), Results);
3144
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003145 if (CodeCompleter->includeMacros())
Douglas Gregor590c7d52010-07-08 20:55:51 +00003146 AddMacroResults(PP, Results, PreferredTypeIsPointer);
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003147 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorfb629412010-08-23 21:17:50 +00003148 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3149 Data.PreferredType),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003150 Results.data(),Results.size());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003151}
3152
Douglas Gregorac5fd842010-09-18 01:28:11 +00003153void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3154 if (E.isInvalid())
3155 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
3156 else if (getLangOptions().ObjC1)
3157 CodeCompleteObjCInstanceMessage(S, E.take(), 0, 0, false);
Douglas Gregor78edf512010-09-15 16:23:04 +00003158}
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003159
Douglas Gregor73449212010-12-09 23:01:55 +00003160/// \brief The set of properties that have already been added, referenced by
3161/// property name.
3162typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3163
Douglas Gregor95ac6552009-11-18 01:29:26 +00003164static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor322328b2009-11-18 22:32:06 +00003165 bool AllowCategories,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003166 bool AllowNullaryMethods,
Douglas Gregor95ac6552009-11-18 01:29:26 +00003167 DeclContext *CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003168 AddedPropertiesSet &AddedProperties,
Douglas Gregor95ac6552009-11-18 01:29:26 +00003169 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00003170 typedef CodeCompletionResult Result;
Douglas Gregor95ac6552009-11-18 01:29:26 +00003171
3172 // Add properties in this container.
3173 for (ObjCContainerDecl::prop_iterator P = Container->prop_begin(),
3174 PEnd = Container->prop_end();
3175 P != PEnd;
Douglas Gregor73449212010-12-09 23:01:55 +00003176 ++P) {
3177 if (AddedProperties.insert(P->getIdentifier()))
3178 Results.MaybeAddResult(Result(*P, 0), CurContext);
3179 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003180
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003181 // Add nullary methods
3182 if (AllowNullaryMethods) {
3183 ASTContext &Context = Container->getASTContext();
Douglas Gregor8987b232011-09-27 23:30:47 +00003184 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003185 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
3186 MEnd = Container->meth_end();
3187 M != MEnd; ++M) {
3188 if (M->getSelector().isUnarySelector())
3189 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3190 if (AddedProperties.insert(Name)) {
3191 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregor8987b232011-09-27 23:30:47 +00003192 AddResultTypeChunk(Context, Policy, *M, Builder);
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003193 Builder.AddTypedTextChunk(
3194 Results.getAllocator().CopyString(Name->getName()));
3195
3196 CXAvailabilityKind Availability = CXAvailability_Available;
3197 switch (M->getAvailability()) {
3198 case AR_Available:
3199 case AR_NotYetIntroduced:
3200 Availability = CXAvailability_Available;
3201 break;
3202
3203 case AR_Deprecated:
3204 Availability = CXAvailability_Deprecated;
3205 break;
3206
3207 case AR_Unavailable:
3208 Availability = CXAvailability_NotAvailable;
3209 break;
3210 }
3211
3212 Results.MaybeAddResult(Result(Builder.TakeString(),
3213 CCP_MemberDeclaration + CCD_MethodAsProperty,
3214 M->isInstanceMethod()
3215 ? CXCursor_ObjCInstanceMethodDecl
3216 : CXCursor_ObjCClassMethodDecl,
3217 Availability),
3218 CurContext);
3219 }
3220 }
3221 }
3222
3223
Douglas Gregor95ac6552009-11-18 01:29:26 +00003224 // Add properties in referenced protocols.
3225 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3226 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
3227 PEnd = Protocol->protocol_end();
3228 P != PEnd; ++P)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003229 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3230 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003231 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor322328b2009-11-18 22:32:06 +00003232 if (AllowCategories) {
3233 // Look through categories.
3234 for (ObjCCategoryDecl *Category = IFace->getCategoryList();
3235 Category; Category = Category->getNextClassCategory())
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003236 AddObjCProperties(Category, AllowCategories, AllowNullaryMethods,
3237 CurContext, AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00003238 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003239
3240 // Look through protocols.
Ted Kremenek53b94412010-09-01 01:21:15 +00003241 for (ObjCInterfaceDecl::all_protocol_iterator
3242 I = IFace->all_referenced_protocol_begin(),
3243 E = IFace->all_referenced_protocol_end(); I != E; ++I)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003244 AddObjCProperties(*I, AllowCategories, AllowNullaryMethods, CurContext,
3245 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003246
3247 // Look in the superclass.
3248 if (IFace->getSuperClass())
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003249 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3250 AllowNullaryMethods, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003251 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003252 } else if (const ObjCCategoryDecl *Category
3253 = dyn_cast<ObjCCategoryDecl>(Container)) {
3254 // Look through protocols.
Ted Kremenek53b94412010-09-01 01:21:15 +00003255 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
3256 PEnd = Category->protocol_end();
Douglas Gregor95ac6552009-11-18 01:29:26 +00003257 P != PEnd; ++P)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003258 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3259 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003260 }
3261}
3262
Richard Trieuf81e5a92011-09-09 02:00:50 +00003263void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *BaseE,
Douglas Gregor81b747b2009-09-17 21:32:03 +00003264 SourceLocation OpLoc,
3265 bool IsArrow) {
3266 if (!BaseE || !CodeCompleter)
3267 return;
3268
John McCall0a2c5e22010-08-25 06:19:51 +00003269 typedef CodeCompletionResult Result;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003270
Douglas Gregor81b747b2009-09-17 21:32:03 +00003271 Expr *Base = static_cast<Expr *>(BaseE);
3272 QualType BaseType = Base->getType();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003273
3274 if (IsArrow) {
3275 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3276 BaseType = Ptr->getPointeeType();
3277 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor3cdee122010-08-26 16:36:48 +00003278 /*Do nothing*/ ;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003279 else
3280 return;
3281 }
3282
Douglas Gregor3da626b2011-07-07 16:03:39 +00003283 enum CodeCompletionContext::Kind contextKind;
3284
3285 if (IsArrow) {
3286 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3287 }
3288 else {
3289 if (BaseType->isObjCObjectPointerType() ||
3290 BaseType->isObjCObjectOrInterfaceType()) {
3291 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3292 }
3293 else {
3294 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3295 }
3296 }
3297
Douglas Gregor218937c2011-02-01 19:23:04 +00003298 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00003299 CodeCompletionContext(contextKind,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003300 BaseType),
3301 &ResultBuilder::IsMember);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003302 Results.EnterNewScope();
3303 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor3cdee122010-08-26 16:36:48 +00003304 // Indicate that we are performing a member access, and the cv-qualifiers
3305 // for the base object type.
3306 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3307
Douglas Gregor95ac6552009-11-18 01:29:26 +00003308 // Access to a C/C++ class, struct, or union.
Douglas Gregor45bcd432010-01-14 03:21:49 +00003309 Results.allowNestedNameSpecifiers();
Douglas Gregor0cc84042010-01-14 15:47:35 +00003310 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003311 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3312 CodeCompleter->includeGlobals());
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003313
Douglas Gregor95ac6552009-11-18 01:29:26 +00003314 if (getLangOptions().CPlusPlus) {
3315 if (!Results.empty()) {
3316 // The "template" keyword can follow "->" or "." in the grammar.
3317 // However, we only want to suggest the template keyword if something
3318 // is dependent.
3319 bool IsDependent = BaseType->isDependentType();
3320 if (!IsDependent) {
3321 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
3322 if (DeclContext *Ctx = (DeclContext *)DepScope->getEntity()) {
3323 IsDependent = Ctx->isDependentContext();
3324 break;
3325 }
3326 }
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003327
Douglas Gregor95ac6552009-11-18 01:29:26 +00003328 if (IsDependent)
Douglas Gregora4477812010-01-14 16:01:26 +00003329 Results.AddResult(Result("template"));
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003330 }
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003331 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003332 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3333 // Objective-C property reference.
Douglas Gregor73449212010-12-09 23:01:55 +00003334 AddedPropertiesSet AddedProperties;
Douglas Gregor95ac6552009-11-18 01:29:26 +00003335
3336 // Add property results based on our interface.
3337 const ObjCObjectPointerType *ObjCPtr
3338 = BaseType->getAsObjCInterfacePointerType();
3339 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003340 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3341 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003342 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003343
3344 // Add properties from the protocols in a qualified interface.
3345 for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(),
3346 E = ObjCPtr->qual_end();
3347 I != E; ++I)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003348 AddObjCProperties(*I, true, /*AllowNullaryMethods=*/true, CurContext,
3349 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003350 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCallc12c5bb2010-05-15 11:32:37 +00003351 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor95ac6552009-11-18 01:29:26 +00003352 // Objective-C instance variable access.
3353 ObjCInterfaceDecl *Class = 0;
3354 if (const ObjCObjectPointerType *ObjCPtr
3355 = BaseType->getAs<ObjCObjectPointerType>())
3356 Class = ObjCPtr->getInterfaceDecl();
3357 else
John McCallc12c5bb2010-05-15 11:32:37 +00003358 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor95ac6552009-11-18 01:29:26 +00003359
3360 // Add all ivars from this class and its superclasses.
Douglas Gregor80f4f4c2010-01-14 16:08:12 +00003361 if (Class) {
3362 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3363 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor8071e422010-08-15 06:18:01 +00003364 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3365 CodeCompleter->includeGlobals());
Douglas Gregor95ac6552009-11-18 01:29:26 +00003366 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003367 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003368
3369 // FIXME: How do we cope with isa?
3370
3371 Results.ExitScope();
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003372
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003373 // Hand off the results found for code completion.
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003374 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003375 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003376 Results.data(),Results.size());
Douglas Gregor81b747b2009-09-17 21:32:03 +00003377}
3378
Douglas Gregor374929f2009-09-18 15:37:17 +00003379void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3380 if (!CodeCompleter)
3381 return;
3382
John McCall0a2c5e22010-08-25 06:19:51 +00003383 typedef CodeCompletionResult Result;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003384 ResultBuilder::LookupFilter Filter = 0;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003385 enum CodeCompletionContext::Kind ContextKind
3386 = CodeCompletionContext::CCC_Other;
Douglas Gregor374929f2009-09-18 15:37:17 +00003387 switch ((DeclSpec::TST)TagSpec) {
3388 case DeclSpec::TST_enum:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003389 Filter = &ResultBuilder::IsEnum;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003390 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003391 break;
3392
3393 case DeclSpec::TST_union:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003394 Filter = &ResultBuilder::IsUnion;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003395 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003396 break;
3397
3398 case DeclSpec::TST_struct:
Douglas Gregor374929f2009-09-18 15:37:17 +00003399 case DeclSpec::TST_class:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003400 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003401 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003402 break;
3403
3404 default:
David Blaikieb219cfc2011-09-23 05:06:16 +00003405 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregor374929f2009-09-18 15:37:17 +00003406 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003407
Douglas Gregor218937c2011-02-01 19:23:04 +00003408 ResultBuilder Results(*this, CodeCompleter->getAllocator(), ContextKind);
Douglas Gregor5d2fc402010-01-14 03:27:13 +00003409 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCall0d6b1642010-04-23 18:46:30 +00003410
3411 // First pass: look for tags.
3412 Results.setFilter(Filter);
Douglas Gregor8071e422010-08-15 06:18:01 +00003413 LookupVisibleDecls(S, LookupTagName, Consumer,
3414 CodeCompleter->includeGlobals());
John McCall0d6b1642010-04-23 18:46:30 +00003415
Douglas Gregor8071e422010-08-15 06:18:01 +00003416 if (CodeCompleter->includeGlobals()) {
3417 // Second pass: look for nested name specifiers.
3418 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3419 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3420 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003421
Douglas Gregor52779fb2010-09-23 23:01:17 +00003422 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003423 Results.data(),Results.size());
Douglas Gregor374929f2009-09-18 15:37:17 +00003424}
3425
Douglas Gregor1a480c42010-08-27 17:35:51 +00003426void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003427 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3428 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor1a480c42010-08-27 17:35:51 +00003429 Results.EnterNewScope();
3430 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3431 Results.AddResult("const");
3432 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3433 Results.AddResult("volatile");
3434 if (getLangOptions().C99 &&
3435 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3436 Results.AddResult("restrict");
3437 Results.ExitScope();
3438 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003439 Results.getCompletionContext(),
Douglas Gregor1a480c42010-08-27 17:35:51 +00003440 Results.data(), Results.size());
3441}
3442
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003443void Sema::CodeCompleteCase(Scope *S) {
John McCall781472f2010-08-25 08:40:02 +00003444 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003445 return;
John McCalla8e0cd82011-08-06 07:30:58 +00003446
John McCall781472f2010-08-25 08:40:02 +00003447 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCalla8e0cd82011-08-06 07:30:58 +00003448 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3449 if (!type->isEnumeralType()) {
3450 CodeCompleteExpressionData Data(type);
Douglas Gregorfb629412010-08-23 21:17:50 +00003451 Data.IntegralConstantExpression = true;
3452 CodeCompleteExpression(S, Data);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003453 return;
Douglas Gregorf9578432010-07-28 21:50:18 +00003454 }
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003455
3456 // Code-complete the cases of a switch statement over an enumeration type
3457 // by providing the list of
John McCalla8e0cd82011-08-06 07:30:58 +00003458 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003459
3460 // Determine which enumerators we have already seen in the switch statement.
3461 // FIXME: Ideally, we would also be able to look *past* the code-completion
3462 // token, in case we are code-completing in the middle of the switch and not
3463 // at the end. However, we aren't able to do so at the moment.
3464 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003465 NestedNameSpecifier *Qualifier = 0;
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003466 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3467 SC = SC->getNextSwitchCase()) {
3468 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3469 if (!Case)
3470 continue;
3471
3472 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3473 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3474 if (EnumConstantDecl *Enumerator
3475 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3476 // We look into the AST of the case statement to determine which
3477 // enumerator was named. Alternatively, we could compute the value of
3478 // the integral constant expression, then compare it against the
3479 // values of each enumerator. However, value-based approach would not
3480 // work as well with C++ templates where enumerators declared within a
3481 // template are type- and value-dependent.
3482 EnumeratorsSeen.insert(Enumerator);
3483
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003484 // If this is a qualified-id, keep track of the nested-name-specifier
3485 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003486 //
3487 // switch (TagD.getKind()) {
3488 // case TagDecl::TK_enum:
3489 // break;
3490 // case XXX
3491 //
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003492 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003493 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3494 // TK_struct, and TK_class.
Douglas Gregora2813ce2009-10-23 18:54:35 +00003495 Qualifier = DRE->getQualifier();
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003496 }
3497 }
3498
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003499 if (getLangOptions().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
3500 // If there are no prior enumerators in C++, check whether we have to
3501 // qualify the names of the enumerators that we suggest, because they
3502 // may not be visible in this scope.
3503 Qualifier = getRequiredQualification(Context, CurContext,
3504 Enum->getDeclContext());
3505
3506 // FIXME: Scoped enums need to start with "EnumDecl" as the context!
3507 }
3508
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003509 // Add any enumerators that have not yet been mentioned.
Douglas Gregor218937c2011-02-01 19:23:04 +00003510 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3511 CodeCompletionContext::CCC_Expression);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003512 Results.EnterNewScope();
3513 for (EnumDecl::enumerator_iterator E = Enum->enumerator_begin(),
3514 EEnd = Enum->enumerator_end();
3515 E != EEnd; ++E) {
3516 if (EnumeratorsSeen.count(*E))
3517 continue;
3518
Douglas Gregor5c722c702011-02-18 23:30:37 +00003519 CodeCompletionResult R(*E, Qualifier);
3520 R.Priority = CCP_EnumInCase;
3521 Results.AddResult(R, CurContext, 0, false);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003522 }
3523 Results.ExitScope();
Douglas Gregor2f880e42010-04-06 20:02:15 +00003524
Douglas Gregor3da626b2011-07-07 16:03:39 +00003525 //We need to make sure we're setting the right context,
3526 //so only say we include macros if the code completer says we do
3527 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3528 if (CodeCompleter->includeMacros()) {
Douglas Gregorbca403c2010-01-13 23:51:12 +00003529 AddMacroResults(PP, Results);
Douglas Gregor3da626b2011-07-07 16:03:39 +00003530 kind = CodeCompletionContext::CCC_OtherWithMacros;
3531 }
3532
3533
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003534 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00003535 kind,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003536 Results.data(),Results.size());
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003537}
3538
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003539namespace {
3540 struct IsBetterOverloadCandidate {
3541 Sema &S;
John McCall5769d612010-02-08 23:07:23 +00003542 SourceLocation Loc;
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003543
3544 public:
John McCall5769d612010-02-08 23:07:23 +00003545 explicit IsBetterOverloadCandidate(Sema &S, SourceLocation Loc)
3546 : S(S), Loc(Loc) { }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003547
3548 bool
3549 operator()(const OverloadCandidate &X, const OverloadCandidate &Y) const {
John McCall120d63c2010-08-24 20:38:10 +00003550 return isBetterOverloadCandidate(S, X, Y, Loc);
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003551 }
3552 };
3553}
3554
Douglas Gregord28dcd72010-05-30 06:10:08 +00003555static bool anyNullArguments(Expr **Args, unsigned NumArgs) {
3556 if (NumArgs && !Args)
3557 return true;
3558
3559 for (unsigned I = 0; I != NumArgs; ++I)
3560 if (!Args[I])
3561 return true;
3562
3563 return false;
3564}
3565
Richard Trieuf81e5a92011-09-09 02:00:50 +00003566void Sema::CodeCompleteCall(Scope *S, Expr *FnIn,
3567 Expr **ArgsIn, unsigned NumArgs) {
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003568 if (!CodeCompleter)
3569 return;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003570
3571 // When we're code-completing for a call, we fall back to ordinary
3572 // name code-completion whenever we can't produce specific
3573 // results. We may want to revisit this strategy in the future,
3574 // e.g., by merging the two kinds of results.
3575
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003576 Expr *Fn = (Expr *)FnIn;
3577 Expr **Args = (Expr **)ArgsIn;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003578
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003579 // Ignore type-dependent call expressions entirely.
Douglas Gregord28dcd72010-05-30 06:10:08 +00003580 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args, NumArgs) ||
Douglas Gregoref96eac2009-12-11 19:06:04 +00003581 Expr::hasAnyTypeDependentArguments(Args, NumArgs)) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003582 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003583 return;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003584 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003585
John McCall3b4294e2009-12-16 12:17:52 +00003586 // Build an overload candidate set based on the functions we find.
John McCall5769d612010-02-08 23:07:23 +00003587 SourceLocation Loc = Fn->getExprLoc();
3588 OverloadCandidateSet CandidateSet(Loc);
John McCall3b4294e2009-12-16 12:17:52 +00003589
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003590 // FIXME: What if we're calling something that isn't a function declaration?
3591 // FIXME: What if we're calling a pseudo-destructor?
3592 // FIXME: What if we're calling a member function?
3593
Douglas Gregorc0265402010-01-21 15:46:19 +00003594 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner5f9e2722011-07-23 10:55:15 +00003595 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorc0265402010-01-21 15:46:19 +00003596
John McCall3b4294e2009-12-16 12:17:52 +00003597 Expr *NakedFn = Fn->IgnoreParenCasts();
3598 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
3599 AddOverloadedCallCandidates(ULE, Args, NumArgs, CandidateSet,
3600 /*PartialOverloading=*/ true);
3601 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3602 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorc0265402010-01-21 15:46:19 +00003603 if (FDecl) {
Douglas Gregord28dcd72010-05-30 06:10:08 +00003604 if (!getLangOptions().CPlusPlus ||
3605 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorc0265402010-01-21 15:46:19 +00003606 Results.push_back(ResultCandidate(FDecl));
3607 else
John McCall86820f52010-01-26 01:37:31 +00003608 // FIXME: access?
John McCall9aa472c2010-03-19 07:35:19 +00003609 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none),
3610 Args, NumArgs, CandidateSet,
Douglas Gregorc27d6c52010-04-16 17:41:49 +00003611 false, /*PartialOverloading*/true);
Douglas Gregorc0265402010-01-21 15:46:19 +00003612 }
John McCall3b4294e2009-12-16 12:17:52 +00003613 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003614
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003615 QualType ParamType;
3616
Douglas Gregorc0265402010-01-21 15:46:19 +00003617 if (!CandidateSet.empty()) {
3618 // Sort the overload candidate set by placing the best overloads first.
3619 std::stable_sort(CandidateSet.begin(), CandidateSet.end(),
John McCall5769d612010-02-08 23:07:23 +00003620 IsBetterOverloadCandidate(*this, Loc));
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003621
Douglas Gregorc0265402010-01-21 15:46:19 +00003622 // Add the remaining viable overload candidates as code-completion reslults.
3623 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3624 CandEnd = CandidateSet.end();
3625 Cand != CandEnd; ++Cand) {
3626 if (Cand->Viable)
3627 Results.push_back(ResultCandidate(Cand->Function));
3628 }
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003629
3630 // From the viable candidates, try to determine the type of this parameter.
3631 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3632 if (const FunctionType *FType = Results[I].getFunctionType())
3633 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
3634 if (NumArgs < Proto->getNumArgs()) {
3635 if (ParamType.isNull())
3636 ParamType = Proto->getArgType(NumArgs);
3637 else if (!Context.hasSameUnqualifiedType(
3638 ParamType.getNonReferenceType(),
3639 Proto->getArgType(NumArgs).getNonReferenceType())) {
3640 ParamType = QualType();
3641 break;
3642 }
3643 }
3644 }
3645 } else {
3646 // Try to determine the parameter type from the type of the expression
3647 // being called.
3648 QualType FunctionType = Fn->getType();
3649 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3650 FunctionType = Ptr->getPointeeType();
3651 else if (const BlockPointerType *BlockPtr
3652 = FunctionType->getAs<BlockPointerType>())
3653 FunctionType = BlockPtr->getPointeeType();
3654 else if (const MemberPointerType *MemPtr
3655 = FunctionType->getAs<MemberPointerType>())
3656 FunctionType = MemPtr->getPointeeType();
3657
3658 if (const FunctionProtoType *Proto
3659 = FunctionType->getAs<FunctionProtoType>()) {
3660 if (NumArgs < Proto->getNumArgs())
3661 ParamType = Proto->getArgType(NumArgs);
3662 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003663 }
Douglas Gregoref96eac2009-12-11 19:06:04 +00003664
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003665 if (ParamType.isNull())
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003666 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003667 else
3668 CodeCompleteExpression(S, ParamType);
3669
Douglas Gregor2e4c7a52010-04-06 20:19:47 +00003670 if (!Results.empty())
Douglas Gregoref96eac2009-12-11 19:06:04 +00003671 CodeCompleter->ProcessOverloadCandidates(*this, NumArgs, Results.data(),
3672 Results.size());
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003673}
3674
John McCalld226f652010-08-21 09:40:31 +00003675void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3676 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003677 if (!VD) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003678 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003679 return;
3680 }
3681
3682 CodeCompleteExpression(S, VD->getType());
3683}
3684
3685void Sema::CodeCompleteReturn(Scope *S) {
3686 QualType ResultType;
3687 if (isa<BlockDecl>(CurContext)) {
3688 if (BlockScopeInfo *BSI = getCurBlock())
3689 ResultType = BSI->ReturnType;
3690 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
3691 ResultType = Function->getResultType();
3692 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
3693 ResultType = Method->getResultType();
3694
3695 if (ResultType.isNull())
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003696 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003697 else
3698 CodeCompleteExpression(S, ResultType);
3699}
3700
Douglas Gregord2d8be62011-07-30 08:36:53 +00003701void Sema::CodeCompleteAfterIf(Scope *S) {
3702 typedef CodeCompletionResult Result;
3703 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3704 mapCodeCompletionContext(*this, PCC_Statement));
3705 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3706 Results.EnterNewScope();
3707
3708 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3709 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3710 CodeCompleter->includeGlobals());
3711
3712 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
3713
3714 // "else" block
3715 CodeCompletionBuilder Builder(Results.getAllocator());
3716 Builder.AddTypedTextChunk("else");
3717 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3718 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
3719 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3720 Builder.AddPlaceholderChunk("statements");
3721 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3722 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
3723 Results.AddResult(Builder.TakeString());
3724
3725 // "else if" block
3726 Builder.AddTypedTextChunk("else");
3727 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3728 Builder.AddTextChunk("if");
3729 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3730 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
3731 if (getLangOptions().CPlusPlus)
3732 Builder.AddPlaceholderChunk("condition");
3733 else
3734 Builder.AddPlaceholderChunk("expression");
3735 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3736 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
3737 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
3738 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3739 Builder.AddPlaceholderChunk("statements");
3740 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
3741 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
3742 Results.AddResult(Builder.TakeString());
3743
3744 Results.ExitScope();
3745
3746 if (S->getFnParent())
3747 AddPrettyFunctionResults(PP.getLangOptions(), Results);
3748
3749 if (CodeCompleter->includeMacros())
3750 AddMacroResults(PP, Results);
3751
3752 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3753 Results.data(),Results.size());
3754}
3755
Richard Trieuf81e5a92011-09-09 02:00:50 +00003756void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003757 if (LHS)
3758 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
3759 else
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003760 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003761}
3762
Jeffrey Yasskin9ab14542010-04-08 16:38:48 +00003763void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor81b747b2009-09-17 21:32:03 +00003764 bool EnteringContext) {
3765 if (!SS.getScopeRep() || !CodeCompleter)
3766 return;
3767
Douglas Gregor86d9a522009-09-21 16:56:56 +00003768 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
3769 if (!Ctx)
3770 return;
Douglas Gregord1cd31a2009-12-11 18:28:39 +00003771
3772 // Try to instantiate any non-dependent declaration contexts before
3773 // we look in them.
John McCall77bb1aa2010-05-01 00:40:08 +00003774 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregord1cd31a2009-12-11 18:28:39 +00003775 return;
3776
Douglas Gregor218937c2011-02-01 19:23:04 +00003777 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3778 CodeCompletionContext::CCC_Name);
Douglas Gregorf6961522010-08-27 21:18:54 +00003779 Results.EnterNewScope();
Douglas Gregor52779fb2010-09-23 23:01:17 +00003780
Douglas Gregor86d9a522009-09-21 16:56:56 +00003781 // The "template" keyword can follow "::" in the grammar, but only
3782 // put it into the grammar if the nested-name-specifier is dependent.
3783 NestedNameSpecifier *NNS = (NestedNameSpecifier *)SS.getScopeRep();
3784 if (!Results.empty() && NNS->isDependent())
Douglas Gregora4477812010-01-14 16:01:26 +00003785 Results.AddResult("template");
Douglas Gregorf6961522010-08-27 21:18:54 +00003786
3787 // Add calls to overridden virtual functions, if there are any.
3788 //
3789 // FIXME: This isn't wonderful, because we don't know whether we're actually
3790 // in a context that permits expressions. This is a general issue with
3791 // qualified-id completions.
3792 if (!EnteringContext)
3793 MaybeAddOverrideCalls(*this, Ctx, Results);
3794 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003795
Douglas Gregorf6961522010-08-27 21:18:54 +00003796 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3797 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
3798
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003799 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor430d7a12011-07-25 17:48:11 +00003800 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003801 Results.data(),Results.size());
Douglas Gregor81b747b2009-09-17 21:32:03 +00003802}
Douglas Gregor49f40bd2009-09-18 19:03:04 +00003803
3804void Sema::CodeCompleteUsing(Scope *S) {
3805 if (!CodeCompleter)
3806 return;
3807
Douglas Gregor218937c2011-02-01 19:23:04 +00003808 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003809 CodeCompletionContext::CCC_PotentiallyQualifiedName,
3810 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00003811 Results.EnterNewScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003812
3813 // If we aren't in class scope, we could see the "namespace" keyword.
3814 if (!S->isClassScope())
John McCall0a2c5e22010-08-25 06:19:51 +00003815 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor86d9a522009-09-21 16:56:56 +00003816
3817 // After "using", we can see anything that would start a
3818 // nested-name-specifier.
Douglas Gregor5d2fc402010-01-14 03:27:13 +00003819 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003820 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3821 CodeCompleter->includeGlobals());
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00003822 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003823
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003824 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003825 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003826 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00003827}
3828
3829void Sema::CodeCompleteUsingDirective(Scope *S) {
3830 if (!CodeCompleter)
3831 return;
3832
Douglas Gregor86d9a522009-09-21 16:56:56 +00003833 // After "using namespace", we expect to see a namespace name or namespace
3834 // alias.
Douglas Gregor218937c2011-02-01 19:23:04 +00003835 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3836 CodeCompletionContext::CCC_Namespace,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003837 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00003838 Results.EnterNewScope();
Douglas Gregor5d2fc402010-01-14 03:27:13 +00003839 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003840 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3841 CodeCompleter->includeGlobals());
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00003842 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003843 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor8071e422010-08-15 06:18:01 +00003844 CodeCompletionContext::CCC_Namespace,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003845 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00003846}
3847
3848void Sema::CodeCompleteNamespaceDecl(Scope *S) {
3849 if (!CodeCompleter)
3850 return;
3851
Douglas Gregor86d9a522009-09-21 16:56:56 +00003852 DeclContext *Ctx = (DeclContext *)S->getEntity();
3853 if (!S->getParent())
3854 Ctx = Context.getTranslationUnitDecl();
3855
Douglas Gregor52779fb2010-09-23 23:01:17 +00003856 bool SuppressedGlobalResults
3857 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
3858
Douglas Gregor218937c2011-02-01 19:23:04 +00003859 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003860 SuppressedGlobalResults
3861 ? CodeCompletionContext::CCC_Namespace
3862 : CodeCompletionContext::CCC_Other,
3863 &ResultBuilder::IsNamespace);
3864
3865 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00003866 // We only want to see those namespaces that have already been defined
3867 // within this scope, because its likely that the user is creating an
3868 // extended namespace declaration. Keep track of the most recent
3869 // definition of each namespace.
3870 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
3871 for (DeclContext::specific_decl_iterator<NamespaceDecl>
3872 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
3873 NS != NSEnd; ++NS)
3874 OrigToLatest[NS->getOriginalNamespace()] = *NS;
3875
3876 // Add the most recent definition (or extended definition) of each
3877 // namespace to the list of results.
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00003878 Results.EnterNewScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003879 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
3880 NS = OrigToLatest.begin(), NSEnd = OrigToLatest.end();
3881 NS != NSEnd; ++NS)
John McCall0a2c5e22010-08-25 06:19:51 +00003882 Results.AddResult(CodeCompletionResult(NS->second, 0),
Douglas Gregor608300b2010-01-14 16:14:35 +00003883 CurContext, 0, false);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00003884 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003885 }
3886
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003887 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003888 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003889 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00003890}
3891
3892void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
3893 if (!CodeCompleter)
3894 return;
3895
Douglas Gregor86d9a522009-09-21 16:56:56 +00003896 // After "namespace", we expect to see a namespace or alias.
Douglas Gregor218937c2011-02-01 19:23:04 +00003897 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3898 CodeCompletionContext::CCC_Namespace,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003899 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor5d2fc402010-01-14 03:27:13 +00003900 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003901 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3902 CodeCompleter->includeGlobals());
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003903 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003904 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003905 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00003906}
3907
Douglas Gregored8d3222009-09-18 20:05:18 +00003908void Sema::CodeCompleteOperatorName(Scope *S) {
3909 if (!CodeCompleter)
3910 return;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003911
John McCall0a2c5e22010-08-25 06:19:51 +00003912 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00003913 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3914 CodeCompletionContext::CCC_Type,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003915 &ResultBuilder::IsType);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00003916 Results.EnterNewScope();
Douglas Gregored8d3222009-09-18 20:05:18 +00003917
Douglas Gregor86d9a522009-09-21 16:56:56 +00003918 // Add the names of overloadable operators.
3919#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
3920 if (std::strcmp(Spelling, "?")) \
Douglas Gregora4477812010-01-14 16:01:26 +00003921 Results.AddResult(Result(Spelling));
Douglas Gregor86d9a522009-09-21 16:56:56 +00003922#include "clang/Basic/OperatorKinds.def"
3923
3924 // Add any type names visible from the current scope
Douglas Gregor45bcd432010-01-14 03:21:49 +00003925 Results.allowNestedNameSpecifiers();
Douglas Gregor5d2fc402010-01-14 03:27:13 +00003926 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003927 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3928 CodeCompleter->includeGlobals());
Douglas Gregor86d9a522009-09-21 16:56:56 +00003929
3930 // Add any type specifiers
Douglas Gregorbca403c2010-01-13 23:51:12 +00003931 AddTypeSpecifierResults(getLangOptions(), Results);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00003932 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003933
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003934 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor8071e422010-08-15 06:18:01 +00003935 CodeCompletionContext::CCC_Type,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003936 Results.data(),Results.size());
Douglas Gregored8d3222009-09-18 20:05:18 +00003937}
Douglas Gregor49f40bd2009-09-18 19:03:04 +00003938
Douglas Gregor0133f522010-08-28 00:00:50 +00003939void Sema::CodeCompleteConstructorInitializer(Decl *ConstructorD,
Sean Huntcbb67482011-01-08 20:30:50 +00003940 CXXCtorInitializer** Initializers,
Douglas Gregor0133f522010-08-28 00:00:50 +00003941 unsigned NumInitializers) {
Douglas Gregor8987b232011-09-27 23:30:47 +00003942 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor0133f522010-08-28 00:00:50 +00003943 CXXConstructorDecl *Constructor
3944 = static_cast<CXXConstructorDecl *>(ConstructorD);
3945 if (!Constructor)
3946 return;
3947
Douglas Gregor218937c2011-02-01 19:23:04 +00003948 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003949 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregor0133f522010-08-28 00:00:50 +00003950 Results.EnterNewScope();
3951
3952 // Fill in any already-initialized fields or base classes.
3953 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
3954 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
3955 for (unsigned I = 0; I != NumInitializers; ++I) {
3956 if (Initializers[I]->isBaseInitializer())
3957 InitializedBases.insert(
3958 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
3959 else
Francois Pichet00eb3f92010-12-04 09:14:42 +00003960 InitializedFields.insert(cast<FieldDecl>(
3961 Initializers[I]->getAnyMember()));
Douglas Gregor0133f522010-08-28 00:00:50 +00003962 }
3963
3964 // Add completions for base classes.
Douglas Gregor218937c2011-02-01 19:23:04 +00003965 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregor0c431c82010-08-29 19:27:27 +00003966 bool SawLastInitializer = (NumInitializers == 0);
Douglas Gregor0133f522010-08-28 00:00:50 +00003967 CXXRecordDecl *ClassDecl = Constructor->getParent();
3968 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->bases_begin(),
3969 BaseEnd = ClassDecl->bases_end();
3970 Base != BaseEnd; ++Base) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00003971 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
3972 SawLastInitializer
3973 = NumInitializers > 0 &&
3974 Initializers[NumInitializers - 1]->isBaseInitializer() &&
3975 Context.hasSameUnqualifiedType(Base->getType(),
3976 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregor0133f522010-08-28 00:00:50 +00003977 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00003978 }
Douglas Gregor0133f522010-08-28 00:00:50 +00003979
Douglas Gregor218937c2011-02-01 19:23:04 +00003980 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00003981 Results.getAllocator().CopyString(
John McCallf85e1932011-06-15 23:02:42 +00003982 Base->getType().getAsString(Policy)));
Douglas Gregor218937c2011-02-01 19:23:04 +00003983 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
3984 Builder.AddPlaceholderChunk("args");
3985 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3986 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00003987 SawLastInitializer? CCP_NextInitializer
3988 : CCP_MemberDeclaration));
3989 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00003990 }
3991
3992 // Add completions for virtual base classes.
3993 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->vbases_begin(),
3994 BaseEnd = ClassDecl->vbases_end();
3995 Base != BaseEnd; ++Base) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00003996 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
3997 SawLastInitializer
3998 = NumInitializers > 0 &&
3999 Initializers[NumInitializers - 1]->isBaseInitializer() &&
4000 Context.hasSameUnqualifiedType(Base->getType(),
4001 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregor0133f522010-08-28 00:00:50 +00004002 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004003 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004004
Douglas Gregor218937c2011-02-01 19:23:04 +00004005 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004006 Builder.getAllocator().CopyString(
John McCallf85e1932011-06-15 23:02:42 +00004007 Base->getType().getAsString(Policy)));
Douglas Gregor218937c2011-02-01 19:23:04 +00004008 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4009 Builder.AddPlaceholderChunk("args");
4010 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4011 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004012 SawLastInitializer? CCP_NextInitializer
4013 : CCP_MemberDeclaration));
4014 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004015 }
4016
4017 // Add completions for members.
4018 for (CXXRecordDecl::field_iterator Field = ClassDecl->field_begin(),
4019 FieldEnd = ClassDecl->field_end();
4020 Field != FieldEnd; ++Field) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004021 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4022 SawLastInitializer
4023 = NumInitializers > 0 &&
Francois Pichet00eb3f92010-12-04 09:14:42 +00004024 Initializers[NumInitializers - 1]->isAnyMemberInitializer() &&
4025 Initializers[NumInitializers - 1]->getAnyMember() == *Field;
Douglas Gregor0133f522010-08-28 00:00:50 +00004026 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004027 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004028
4029 if (!Field->getDeclName())
4030 continue;
4031
Douglas Gregordae68752011-02-01 22:57:45 +00004032 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00004033 Field->getIdentifier()->getName()));
4034 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4035 Builder.AddPlaceholderChunk("args");
4036 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4037 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004038 SawLastInitializer? CCP_NextInitializer
Douglas Gregora67e03f2010-09-09 21:42:20 +00004039 : CCP_MemberDeclaration,
4040 CXCursor_MemberRef));
Douglas Gregor0c431c82010-08-29 19:27:27 +00004041 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004042 }
4043 Results.ExitScope();
4044
Douglas Gregor52779fb2010-09-23 23:01:17 +00004045 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor0133f522010-08-28 00:00:50 +00004046 Results.data(), Results.size());
4047}
4048
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004049// Macro that expands to @Keyword or Keyword, depending on whether NeedAt is
4050// true or false.
4051#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) NeedAt? "@" #Keyword : #Keyword
Douglas Gregorbca403c2010-01-13 23:51:12 +00004052static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004053 ResultBuilder &Results,
4054 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004055 typedef CodeCompletionResult Result;
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004056 // Since we have an implementation, we can end it.
Douglas Gregora4477812010-01-14 16:01:26 +00004057 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,end)));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004058
Douglas Gregor218937c2011-02-01 19:23:04 +00004059 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004060 if (LangOpts.ObjC2) {
4061 // @dynamic
Douglas Gregor218937c2011-02-01 19:23:04 +00004062 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,dynamic));
4063 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4064 Builder.AddPlaceholderChunk("property");
4065 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004066
4067 // @synthesize
Douglas Gregor218937c2011-02-01 19:23:04 +00004068 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,synthesize));
4069 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4070 Builder.AddPlaceholderChunk("property");
4071 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004072 }
4073}
4074
Douglas Gregorbca403c2010-01-13 23:51:12 +00004075static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004076 ResultBuilder &Results,
4077 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004078 typedef CodeCompletionResult Result;
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004079
4080 // Since we have an interface or protocol, we can end it.
Douglas Gregora4477812010-01-14 16:01:26 +00004081 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,end)));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004082
4083 if (LangOpts.ObjC2) {
4084 // @property
Douglas Gregora4477812010-01-14 16:01:26 +00004085 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,property)));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004086
4087 // @required
Douglas Gregora4477812010-01-14 16:01:26 +00004088 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,required)));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004089
4090 // @optional
Douglas Gregora4477812010-01-14 16:01:26 +00004091 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,optional)));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004092 }
4093}
4094
Douglas Gregorbca403c2010-01-13 23:51:12 +00004095static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004096 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004097 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004098
4099 // @class name ;
Douglas Gregor218937c2011-02-01 19:23:04 +00004100 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,class));
4101 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4102 Builder.AddPlaceholderChunk("name");
4103 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004104
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004105 if (Results.includeCodePatterns()) {
4106 // @interface name
4107 // FIXME: Could introduce the whole pattern, including superclasses and
4108 // such.
Douglas Gregor218937c2011-02-01 19:23:04 +00004109 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,interface));
4110 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4111 Builder.AddPlaceholderChunk("class");
4112 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004113
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004114 // @protocol name
Douglas Gregor218937c2011-02-01 19:23:04 +00004115 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,protocol));
4116 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4117 Builder.AddPlaceholderChunk("protocol");
4118 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004119
4120 // @implementation name
Douglas Gregor218937c2011-02-01 19:23:04 +00004121 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,implementation));
4122 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4123 Builder.AddPlaceholderChunk("class");
4124 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004125 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004126
4127 // @compatibility_alias name
Douglas Gregor218937c2011-02-01 19:23:04 +00004128 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,compatibility_alias));
4129 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4130 Builder.AddPlaceholderChunk("alias");
4131 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4132 Builder.AddPlaceholderChunk("class");
4133 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004134}
4135
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004136void Sema::CodeCompleteObjCAtDirective(Scope *S) {
John McCall0a2c5e22010-08-25 06:19:51 +00004137 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004138 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4139 CodeCompletionContext::CCC_Other);
Douglas Gregorc464ae82009-12-07 09:27:33 +00004140 Results.EnterNewScope();
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004141 if (isa<ObjCImplDecl>(CurContext))
Douglas Gregorbca403c2010-01-13 23:51:12 +00004142 AddObjCImplementationResults(getLangOptions(), Results, false);
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004143 else if (CurContext->isObjCContainer())
Douglas Gregorbca403c2010-01-13 23:51:12 +00004144 AddObjCInterfaceResults(getLangOptions(), Results, false);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004145 else
Douglas Gregorbca403c2010-01-13 23:51:12 +00004146 AddObjCTopLevelResults(Results, false);
Douglas Gregorc464ae82009-12-07 09:27:33 +00004147 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004148 HandleCodeCompleteResults(this, CodeCompleter,
4149 CodeCompletionContext::CCC_Other,
4150 Results.data(),Results.size());
Douglas Gregorc464ae82009-12-07 09:27:33 +00004151}
4152
Douglas Gregorbca403c2010-01-13 23:51:12 +00004153static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004154 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004155 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004156
4157 // @encode ( type-name )
Douglas Gregor218937c2011-02-01 19:23:04 +00004158 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,encode));
4159 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4160 Builder.AddPlaceholderChunk("type-name");
4161 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4162 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004163
4164 // @protocol ( protocol-name )
Douglas Gregor218937c2011-02-01 19:23:04 +00004165 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,protocol));
4166 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4167 Builder.AddPlaceholderChunk("protocol-name");
4168 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4169 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004170
4171 // @selector ( selector )
Douglas Gregor218937c2011-02-01 19:23:04 +00004172 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,selector));
4173 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4174 Builder.AddPlaceholderChunk("selector");
4175 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4176 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004177}
4178
Douglas Gregorbca403c2010-01-13 23:51:12 +00004179static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004180 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004181 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004182
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004183 if (Results.includeCodePatterns()) {
4184 // @try { statements } @catch ( declaration ) { statements } @finally
4185 // { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00004186 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,try));
4187 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4188 Builder.AddPlaceholderChunk("statements");
4189 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4190 Builder.AddTextChunk("@catch");
4191 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4192 Builder.AddPlaceholderChunk("parameter");
4193 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4194 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4195 Builder.AddPlaceholderChunk("statements");
4196 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4197 Builder.AddTextChunk("@finally");
4198 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4199 Builder.AddPlaceholderChunk("statements");
4200 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4201 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004202 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004203
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004204 // @throw
Douglas Gregor218937c2011-02-01 19:23:04 +00004205 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,throw));
4206 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4207 Builder.AddPlaceholderChunk("expression");
4208 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004209
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004210 if (Results.includeCodePatterns()) {
4211 // @synchronized ( expression ) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00004212 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,synchronized));
4213 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4214 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4215 Builder.AddPlaceholderChunk("expression");
4216 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4217 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4218 Builder.AddPlaceholderChunk("statements");
4219 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4220 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004221 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004222}
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004223
Douglas Gregorbca403c2010-01-13 23:51:12 +00004224static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004225 ResultBuilder &Results,
4226 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004227 typedef CodeCompletionResult Result;
Douglas Gregora4477812010-01-14 16:01:26 +00004228 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,private)));
4229 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,protected)));
4230 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,public)));
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004231 if (LangOpts.ObjC2)
Douglas Gregora4477812010-01-14 16:01:26 +00004232 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,package)));
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004233}
4234
4235void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004236 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4237 CodeCompletionContext::CCC_Other);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004238 Results.EnterNewScope();
Douglas Gregorbca403c2010-01-13 23:51:12 +00004239 AddObjCVisibilityResults(getLangOptions(), Results, false);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004240 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004241 HandleCodeCompleteResults(this, CodeCompleter,
4242 CodeCompletionContext::CCC_Other,
4243 Results.data(),Results.size());
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004244}
4245
4246void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004247 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4248 CodeCompletionContext::CCC_Other);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004249 Results.EnterNewScope();
Douglas Gregorbca403c2010-01-13 23:51:12 +00004250 AddObjCStatementResults(Results, false);
4251 AddObjCExpressionResults(Results, false);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004252 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004253 HandleCodeCompleteResults(this, CodeCompleter,
4254 CodeCompletionContext::CCC_Other,
4255 Results.data(),Results.size());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004256}
4257
4258void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004259 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4260 CodeCompletionContext::CCC_Other);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004261 Results.EnterNewScope();
Douglas Gregorbca403c2010-01-13 23:51:12 +00004262 AddObjCExpressionResults(Results, false);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004263 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004264 HandleCodeCompleteResults(this, CodeCompleter,
4265 CodeCompletionContext::CCC_Other,
4266 Results.data(),Results.size());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004267}
4268
Douglas Gregor988358f2009-11-19 00:14:45 +00004269/// \brief Determine whether the addition of the given flag to an Objective-C
4270/// property's attributes will cause a conflict.
4271static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
4272 // Check if we've already added this flag.
4273 if (Attributes & NewFlag)
4274 return true;
4275
4276 Attributes |= NewFlag;
4277
4278 // Check for collisions with "readonly".
4279 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4280 (Attributes & (ObjCDeclSpec::DQ_PR_readwrite |
4281 ObjCDeclSpec::DQ_PR_assign |
John McCallf85e1932011-06-15 23:02:42 +00004282 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregor988358f2009-11-19 00:14:45 +00004283 ObjCDeclSpec::DQ_PR_copy |
John McCallf85e1932011-06-15 23:02:42 +00004284 ObjCDeclSpec::DQ_PR_retain |
4285 ObjCDeclSpec::DQ_PR_strong)))
Douglas Gregor988358f2009-11-19 00:14:45 +00004286 return true;
4287
John McCallf85e1932011-06-15 23:02:42 +00004288 // Check for more than one of { assign, copy, retain, strong }.
Douglas Gregor988358f2009-11-19 00:14:45 +00004289 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCallf85e1932011-06-15 23:02:42 +00004290 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregor988358f2009-11-19 00:14:45 +00004291 ObjCDeclSpec::DQ_PR_copy |
John McCallf85e1932011-06-15 23:02:42 +00004292 ObjCDeclSpec::DQ_PR_retain|
4293 ObjCDeclSpec::DQ_PR_strong);
Douglas Gregor988358f2009-11-19 00:14:45 +00004294 if (AssignCopyRetMask &&
4295 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCallf85e1932011-06-15 23:02:42 +00004296 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregor988358f2009-11-19 00:14:45 +00004297 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCallf85e1932011-06-15 23:02:42 +00004298 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
4299 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong)
Douglas Gregor988358f2009-11-19 00:14:45 +00004300 return true;
4301
4302 return false;
4303}
4304
Douglas Gregora93b1082009-11-18 23:08:07 +00004305void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroffece8e712009-10-08 21:55:05 +00004306 if (!CodeCompleter)
4307 return;
Douglas Gregord3c68542009-11-19 01:08:35 +00004308
Steve Naroffece8e712009-10-08 21:55:05 +00004309 unsigned Attributes = ODS.getPropertyAttributes();
4310
John McCall0a2c5e22010-08-25 06:19:51 +00004311 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004312 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4313 CodeCompletionContext::CCC_Other);
Steve Naroffece8e712009-10-08 21:55:05 +00004314 Results.EnterNewScope();
Douglas Gregor988358f2009-11-19 00:14:45 +00004315 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall0a2c5e22010-08-25 06:19:51 +00004316 Results.AddResult(CodeCompletionResult("readonly"));
Douglas Gregor988358f2009-11-19 00:14:45 +00004317 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall0a2c5e22010-08-25 06:19:51 +00004318 Results.AddResult(CodeCompletionResult("assign"));
John McCallf85e1932011-06-15 23:02:42 +00004319 if (!ObjCPropertyFlagConflicts(Attributes,
4320 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4321 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Douglas Gregor988358f2009-11-19 00:14:45 +00004322 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall0a2c5e22010-08-25 06:19:51 +00004323 Results.AddResult(CodeCompletionResult("readwrite"));
Douglas Gregor988358f2009-11-19 00:14:45 +00004324 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall0a2c5e22010-08-25 06:19:51 +00004325 Results.AddResult(CodeCompletionResult("retain"));
John McCallf85e1932011-06-15 23:02:42 +00004326 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
4327 Results.AddResult(CodeCompletionResult("strong"));
Douglas Gregor988358f2009-11-19 00:14:45 +00004328 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall0a2c5e22010-08-25 06:19:51 +00004329 Results.AddResult(CodeCompletionResult("copy"));
Douglas Gregor988358f2009-11-19 00:14:45 +00004330 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall0a2c5e22010-08-25 06:19:51 +00004331 Results.AddResult(CodeCompletionResult("nonatomic"));
Fariborz Jahanian27f45232011-06-11 17:14:27 +00004332 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
4333 Results.AddResult(CodeCompletionResult("atomic"));
Douglas Gregor988358f2009-11-19 00:14:45 +00004334 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004335 CodeCompletionBuilder Setter(Results.getAllocator());
4336 Setter.AddTypedTextChunk("setter");
4337 Setter.AddTextChunk(" = ");
4338 Setter.AddPlaceholderChunk("method");
4339 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor54f01612009-11-19 00:01:57 +00004340 }
Douglas Gregor988358f2009-11-19 00:14:45 +00004341 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004342 CodeCompletionBuilder Getter(Results.getAllocator());
4343 Getter.AddTypedTextChunk("getter");
4344 Getter.AddTextChunk(" = ");
4345 Getter.AddPlaceholderChunk("method");
4346 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor54f01612009-11-19 00:01:57 +00004347 }
Steve Naroffece8e712009-10-08 21:55:05 +00004348 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004349 HandleCodeCompleteResults(this, CodeCompleter,
4350 CodeCompletionContext::CCC_Other,
4351 Results.data(),Results.size());
Steve Naroffece8e712009-10-08 21:55:05 +00004352}
Steve Naroffc4df6d22009-11-07 02:08:14 +00004353
Douglas Gregor4ad96852009-11-19 07:41:15 +00004354/// \brief Descripts the kind of Objective-C method that we want to find
4355/// via code completion.
4356enum ObjCMethodKind {
4357 MK_Any, //< Any kind of method, provided it means other specified criteria.
4358 MK_ZeroArgSelector, //< Zero-argument (unary) selector.
4359 MK_OneArgSelector //< One-argument selector.
4360};
4361
Douglas Gregor458433d2010-08-26 15:07:07 +00004362static bool isAcceptableObjCSelector(Selector Sel,
4363 ObjCMethodKind WantKind,
4364 IdentifierInfo **SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004365 unsigned NumSelIdents,
4366 bool AllowSameLength = true) {
Douglas Gregor458433d2010-08-26 15:07:07 +00004367 if (NumSelIdents > Sel.getNumArgs())
4368 return false;
4369
4370 switch (WantKind) {
4371 case MK_Any: break;
4372 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4373 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4374 }
4375
Douglas Gregorcf544262010-11-17 21:36:08 +00004376 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4377 return false;
4378
Douglas Gregor458433d2010-08-26 15:07:07 +00004379 for (unsigned I = 0; I != NumSelIdents; ++I)
4380 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4381 return false;
4382
4383 return true;
4384}
4385
Douglas Gregor4ad96852009-11-19 07:41:15 +00004386static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4387 ObjCMethodKind WantKind,
4388 IdentifierInfo **SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004389 unsigned NumSelIdents,
4390 bool AllowSameLength = true) {
Douglas Gregor458433d2010-08-26 15:07:07 +00004391 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004392 NumSelIdents, AllowSameLength);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004393}
Douglas Gregord36adf52010-09-16 16:06:31 +00004394
4395namespace {
4396 /// \brief A set of selectors, which is used to avoid introducing multiple
4397 /// completions with the same selector into the result set.
4398 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4399}
4400
Douglas Gregor36ecb042009-11-17 23:22:23 +00004401/// \brief Add all of the Objective-C methods in the given Objective-C
4402/// container to the set of results.
4403///
4404/// The container will be a class, protocol, category, or implementation of
4405/// any of the above. This mether will recurse to include methods from
4406/// the superclasses of classes along with their categories, protocols, and
4407/// implementations.
4408///
4409/// \param Container the container in which we'll look to find methods.
4410///
4411/// \param WantInstance whether to add instance methods (only); if false, this
4412/// routine will add factory methods (only).
4413///
4414/// \param CurContext the context in which we're performing the lookup that
4415/// finds methods.
4416///
Douglas Gregorcf544262010-11-17 21:36:08 +00004417/// \param AllowSameLength Whether we allow a method to be added to the list
4418/// when it has the same number of parameters as we have selector identifiers.
4419///
Douglas Gregor36ecb042009-11-17 23:22:23 +00004420/// \param Results the structure into which we'll add results.
4421static void AddObjCMethods(ObjCContainerDecl *Container,
4422 bool WantInstanceMethods,
Douglas Gregor4ad96852009-11-19 07:41:15 +00004423 ObjCMethodKind WantKind,
Douglas Gregord3c68542009-11-19 01:08:35 +00004424 IdentifierInfo **SelIdents,
4425 unsigned NumSelIdents,
Douglas Gregor36ecb042009-11-17 23:22:23 +00004426 DeclContext *CurContext,
Douglas Gregord36adf52010-09-16 16:06:31 +00004427 VisitedSelectorSet &Selectors,
Douglas Gregorcf544262010-11-17 21:36:08 +00004428 bool AllowSameLength,
Douglas Gregor408be5a2010-08-25 01:08:01 +00004429 ResultBuilder &Results,
4430 bool InOriginalClass = true) {
John McCall0a2c5e22010-08-25 06:19:51 +00004431 typedef CodeCompletionResult Result;
Douglas Gregor36ecb042009-11-17 23:22:23 +00004432 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
4433 MEnd = Container->meth_end();
4434 M != MEnd; ++M) {
Douglas Gregord3c68542009-11-19 01:08:35 +00004435 if ((*M)->isInstanceMethod() == WantInstanceMethods) {
4436 // Check whether the selector identifiers we've been given are a
4437 // subset of the identifiers for this particular method.
Douglas Gregorcf544262010-11-17 21:36:08 +00004438 if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, NumSelIdents,
4439 AllowSameLength))
Douglas Gregord3c68542009-11-19 01:08:35 +00004440 continue;
Douglas Gregor4ad96852009-11-19 07:41:15 +00004441
Douglas Gregord36adf52010-09-16 16:06:31 +00004442 if (!Selectors.insert((*M)->getSelector()))
4443 continue;
4444
Douglas Gregord3c68542009-11-19 01:08:35 +00004445 Result R = Result(*M, 0);
4446 R.StartParameter = NumSelIdents;
Douglas Gregor4ad96852009-11-19 07:41:15 +00004447 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor408be5a2010-08-25 01:08:01 +00004448 if (!InOriginalClass)
4449 R.Priority += CCD_InBaseClass;
Douglas Gregord3c68542009-11-19 01:08:35 +00004450 Results.MaybeAddResult(R, CurContext);
4451 }
Douglas Gregor36ecb042009-11-17 23:22:23 +00004452 }
4453
Douglas Gregore396c7b2010-09-16 15:34:59 +00004454 // Visit the protocols of protocols.
4455 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
4456 const ObjCList<ObjCProtocolDecl> &Protocols
4457 = Protocol->getReferencedProtocols();
4458 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4459 E = Protocols.end();
4460 I != E; ++I)
4461 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004462 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregore396c7b2010-09-16 15:34:59 +00004463 }
4464
Douglas Gregor36ecb042009-11-17 23:22:23 +00004465 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4466 if (!IFace)
4467 return;
4468
4469 // Add methods in protocols.
4470 const ObjCList<ObjCProtocolDecl> &Protocols= IFace->getReferencedProtocols();
4471 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4472 E = Protocols.end();
4473 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00004474 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004475 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004476
4477 // Add methods in categories.
4478 for (ObjCCategoryDecl *CatDecl = IFace->getCategoryList(); CatDecl;
4479 CatDecl = CatDecl->getNextClassCategory()) {
Douglas Gregor4ad96852009-11-19 07:41:15 +00004480 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004481 NumSelIdents, CurContext, Selectors, AllowSameLength,
4482 Results, InOriginalClass);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004483
4484 // Add a categories protocol methods.
4485 const ObjCList<ObjCProtocolDecl> &Protocols
4486 = CatDecl->getReferencedProtocols();
4487 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4488 E = Protocols.end();
4489 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00004490 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004491 NumSelIdents, CurContext, Selectors, AllowSameLength,
4492 Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004493
4494 // Add methods in category implementations.
4495 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregor4ad96852009-11-19 07:41:15 +00004496 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004497 NumSelIdents, CurContext, Selectors, AllowSameLength,
4498 Results, InOriginalClass);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004499 }
4500
4501 // Add methods in superclass.
4502 if (IFace->getSuperClass())
Douglas Gregor4ad96852009-11-19 07:41:15 +00004503 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Douglas Gregorcf544262010-11-17 21:36:08 +00004504 SelIdents, NumSelIdents, CurContext, Selectors,
4505 AllowSameLength, Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004506
4507 // Add methods in our implementation, if any.
4508 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregor4ad96852009-11-19 07:41:15 +00004509 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004510 NumSelIdents, CurContext, Selectors, AllowSameLength,
4511 Results, InOriginalClass);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004512}
4513
4514
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004515void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
John McCall0a2c5e22010-08-25 06:19:51 +00004516 typedef CodeCompletionResult Result;
Douglas Gregor4ad96852009-11-19 07:41:15 +00004517
4518 // Try to find the interface where getters might live.
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004519 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004520 if (!Class) {
4521 if (ObjCCategoryDecl *Category
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004522 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregor4ad96852009-11-19 07:41:15 +00004523 Class = Category->getClassInterface();
4524
4525 if (!Class)
4526 return;
4527 }
4528
4529 // Find all of the potential getters.
Douglas Gregor218937c2011-02-01 19:23:04 +00004530 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4531 CodeCompletionContext::CCC_Other);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004532 Results.EnterNewScope();
4533
Douglas Gregord36adf52010-09-16 16:06:31 +00004534 VisitedSelectorSet Selectors;
4535 AddObjCMethods(Class, true, MK_ZeroArgSelector, 0, 0, CurContext, Selectors,
Douglas Gregorcf544262010-11-17 21:36:08 +00004536 /*AllowSameLength=*/true, Results);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004537 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004538 HandleCodeCompleteResults(this, CodeCompleter,
4539 CodeCompletionContext::CCC_Other,
4540 Results.data(),Results.size());
Douglas Gregor4ad96852009-11-19 07:41:15 +00004541}
4542
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004543void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
John McCall0a2c5e22010-08-25 06:19:51 +00004544 typedef CodeCompletionResult Result;
Douglas Gregor4ad96852009-11-19 07:41:15 +00004545
4546 // Try to find the interface where setters might live.
4547 ObjCInterfaceDecl *Class
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004548 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004549 if (!Class) {
4550 if (ObjCCategoryDecl *Category
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004551 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregor4ad96852009-11-19 07:41:15 +00004552 Class = Category->getClassInterface();
4553
4554 if (!Class)
4555 return;
4556 }
4557
4558 // Find all of the potential getters.
Douglas Gregor218937c2011-02-01 19:23:04 +00004559 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4560 CodeCompletionContext::CCC_Other);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004561 Results.EnterNewScope();
4562
Douglas Gregord36adf52010-09-16 16:06:31 +00004563 VisitedSelectorSet Selectors;
4564 AddObjCMethods(Class, true, MK_OneArgSelector, 0, 0, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00004565 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004566
4567 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004568 HandleCodeCompleteResults(this, CodeCompleter,
4569 CodeCompletionContext::CCC_Other,
4570 Results.data(),Results.size());
Douglas Gregor36ecb042009-11-17 23:22:23 +00004571}
4572
Douglas Gregorafc45782011-02-15 22:19:42 +00004573void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
4574 bool IsParameter) {
John McCall0a2c5e22010-08-25 06:19:51 +00004575 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004576 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4577 CodeCompletionContext::CCC_Type);
Douglas Gregord32b0222010-08-24 01:06:58 +00004578 Results.EnterNewScope();
4579
4580 // Add context-sensitive, Objective-C parameter-passing keywords.
4581 bool AddedInOut = false;
4582 if ((DS.getObjCDeclQualifier() &
4583 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
4584 Results.AddResult("in");
4585 Results.AddResult("inout");
4586 AddedInOut = true;
4587 }
4588 if ((DS.getObjCDeclQualifier() &
4589 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
4590 Results.AddResult("out");
4591 if (!AddedInOut)
4592 Results.AddResult("inout");
4593 }
4594 if ((DS.getObjCDeclQualifier() &
4595 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
4596 ObjCDeclSpec::DQ_Oneway)) == 0) {
4597 Results.AddResult("bycopy");
4598 Results.AddResult("byref");
4599 Results.AddResult("oneway");
4600 }
4601
Douglas Gregorafc45782011-02-15 22:19:42 +00004602 // If we're completing the return type of an Objective-C method and the
4603 // identifier IBAction refers to a macro, provide a completion item for
4604 // an action, e.g.,
4605 // IBAction)<#selector#>:(id)sender
4606 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
4607 Context.Idents.get("IBAction").hasMacroDefinition()) {
4608 typedef CodeCompletionString::Chunk Chunk;
4609 CodeCompletionBuilder Builder(Results.getAllocator(), CCP_CodePattern,
4610 CXAvailability_Available);
4611 Builder.AddTypedTextChunk("IBAction");
4612 Builder.AddChunk(Chunk(CodeCompletionString::CK_RightParen));
4613 Builder.AddPlaceholderChunk("selector");
4614 Builder.AddChunk(Chunk(CodeCompletionString::CK_Colon));
4615 Builder.AddChunk(Chunk(CodeCompletionString::CK_LeftParen));
4616 Builder.AddTextChunk("id");
4617 Builder.AddChunk(Chunk(CodeCompletionString::CK_RightParen));
4618 Builder.AddTextChunk("sender");
4619 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
4620 }
4621
Douglas Gregord32b0222010-08-24 01:06:58 +00004622 // Add various builtin type names and specifiers.
4623 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
4624 Results.ExitScope();
4625
4626 // Add the various type names
4627 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
4628 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4629 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4630 CodeCompleter->includeGlobals());
4631
4632 if (CodeCompleter->includeMacros())
4633 AddMacroResults(PP, Results);
4634
4635 HandleCodeCompleteResults(this, CodeCompleter,
4636 CodeCompletionContext::CCC_Type,
4637 Results.data(), Results.size());
4638}
4639
Douglas Gregor22f56992010-04-06 19:22:33 +00004640/// \brief When we have an expression with type "id", we may assume
4641/// that it has some more-specific class type based on knowledge of
4642/// common uses of Objective-C. This routine returns that class type,
4643/// or NULL if no better result could be determined.
4644static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregor78edf512010-09-15 16:23:04 +00004645 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor22f56992010-04-06 19:22:33 +00004646 if (!Msg)
4647 return 0;
4648
4649 Selector Sel = Msg->getSelector();
4650 if (Sel.isNull())
4651 return 0;
4652
4653 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
4654 if (!Id)
4655 return 0;
4656
4657 ObjCMethodDecl *Method = Msg->getMethodDecl();
4658 if (!Method)
4659 return 0;
4660
4661 // Determine the class that we're sending the message to.
Douglas Gregor04badcf2010-04-21 00:45:42 +00004662 ObjCInterfaceDecl *IFace = 0;
4663 switch (Msg->getReceiverKind()) {
4664 case ObjCMessageExpr::Class:
John McCallc12c5bb2010-05-15 11:32:37 +00004665 if (const ObjCObjectType *ObjType
4666 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
4667 IFace = ObjType->getInterface();
Douglas Gregor04badcf2010-04-21 00:45:42 +00004668 break;
4669
4670 case ObjCMessageExpr::Instance: {
4671 QualType T = Msg->getInstanceReceiver()->getType();
4672 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
4673 IFace = Ptr->getInterfaceDecl();
4674 break;
4675 }
4676
4677 case ObjCMessageExpr::SuperInstance:
4678 case ObjCMessageExpr::SuperClass:
4679 break;
Douglas Gregor22f56992010-04-06 19:22:33 +00004680 }
4681
4682 if (!IFace)
4683 return 0;
4684
4685 ObjCInterfaceDecl *Super = IFace->getSuperClass();
4686 if (Method->isInstanceMethod())
4687 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
4688 .Case("retain", IFace)
John McCallf85e1932011-06-15 23:02:42 +00004689 .Case("strong", IFace)
Douglas Gregor22f56992010-04-06 19:22:33 +00004690 .Case("autorelease", IFace)
4691 .Case("copy", IFace)
4692 .Case("copyWithZone", IFace)
4693 .Case("mutableCopy", IFace)
4694 .Case("mutableCopyWithZone", IFace)
4695 .Case("awakeFromCoder", IFace)
4696 .Case("replacementObjectFromCoder", IFace)
4697 .Case("class", IFace)
4698 .Case("classForCoder", IFace)
4699 .Case("superclass", Super)
4700 .Default(0);
4701
4702 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
4703 .Case("new", IFace)
4704 .Case("alloc", IFace)
4705 .Case("allocWithZone", IFace)
4706 .Case("class", IFace)
4707 .Case("superclass", Super)
4708 .Default(0);
4709}
4710
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004711// Add a special completion for a message send to "super", which fills in the
4712// most likely case of forwarding all of our arguments to the superclass
4713// function.
4714///
4715/// \param S The semantic analysis object.
4716///
4717/// \param S NeedSuperKeyword Whether we need to prefix this completion with
4718/// the "super" keyword. Otherwise, we just need to provide the arguments.
4719///
4720/// \param SelIdents The identifiers in the selector that have already been
4721/// provided as arguments for a send to "super".
4722///
4723/// \param NumSelIdents The number of identifiers in \p SelIdents.
4724///
4725/// \param Results The set of results to augment.
4726///
4727/// \returns the Objective-C method declaration that would be invoked by
4728/// this "super" completion. If NULL, no completion was added.
4729static ObjCMethodDecl *AddSuperSendCompletion(Sema &S, bool NeedSuperKeyword,
4730 IdentifierInfo **SelIdents,
4731 unsigned NumSelIdents,
4732 ResultBuilder &Results) {
4733 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
4734 if (!CurMethod)
4735 return 0;
4736
4737 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
4738 if (!Class)
4739 return 0;
4740
4741 // Try to find a superclass method with the same selector.
4742 ObjCMethodDecl *SuperMethod = 0;
Douglas Gregor78bcd912011-02-16 00:51:18 +00004743 while ((Class = Class->getSuperClass()) && !SuperMethod) {
4744 // Check in the class
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004745 SuperMethod = Class->getMethod(CurMethod->getSelector(),
4746 CurMethod->isInstanceMethod());
4747
Douglas Gregor78bcd912011-02-16 00:51:18 +00004748 // Check in categories or class extensions.
4749 if (!SuperMethod) {
4750 for (ObjCCategoryDecl *Category = Class->getCategoryList(); Category;
4751 Category = Category->getNextClassCategory())
4752 if ((SuperMethod = Category->getMethod(CurMethod->getSelector(),
4753 CurMethod->isInstanceMethod())))
4754 break;
4755 }
4756 }
4757
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004758 if (!SuperMethod)
4759 return 0;
4760
4761 // Check whether the superclass method has the same signature.
4762 if (CurMethod->param_size() != SuperMethod->param_size() ||
4763 CurMethod->isVariadic() != SuperMethod->isVariadic())
4764 return 0;
4765
4766 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
4767 CurPEnd = CurMethod->param_end(),
4768 SuperP = SuperMethod->param_begin();
4769 CurP != CurPEnd; ++CurP, ++SuperP) {
4770 // Make sure the parameter types are compatible.
4771 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
4772 (*SuperP)->getType()))
4773 return 0;
4774
4775 // Make sure we have a parameter name to forward!
4776 if (!(*CurP)->getIdentifier())
4777 return 0;
4778 }
4779
4780 // We have a superclass method. Now, form the send-to-super completion.
Douglas Gregor218937c2011-02-01 19:23:04 +00004781 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004782
4783 // Give this completion a return type.
Douglas Gregor8987b232011-09-27 23:30:47 +00004784 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
4785 Builder);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004786
4787 // If we need the "super" keyword, add it (plus some spacing).
4788 if (NeedSuperKeyword) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004789 Builder.AddTypedTextChunk("super");
4790 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004791 }
4792
4793 Selector Sel = CurMethod->getSelector();
4794 if (Sel.isUnarySelector()) {
4795 if (NeedSuperKeyword)
Douglas Gregordae68752011-02-01 22:57:45 +00004796 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00004797 Sel.getNameForSlot(0)));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004798 else
Douglas Gregordae68752011-02-01 22:57:45 +00004799 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00004800 Sel.getNameForSlot(0)));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004801 } else {
4802 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
4803 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
4804 if (I > NumSelIdents)
Douglas Gregor218937c2011-02-01 19:23:04 +00004805 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004806
4807 if (I < NumSelIdents)
Douglas Gregor218937c2011-02-01 19:23:04 +00004808 Builder.AddInformativeChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004809 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00004810 Sel.getNameForSlot(I) + ":"));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004811 else if (NeedSuperKeyword || I > NumSelIdents) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004812 Builder.AddTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004813 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00004814 Sel.getNameForSlot(I) + ":"));
Douglas Gregordae68752011-02-01 22:57:45 +00004815 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00004816 (*CurP)->getIdentifier()->getName()));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004817 } else {
Douglas Gregor218937c2011-02-01 19:23:04 +00004818 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004819 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00004820 Sel.getNameForSlot(I) + ":"));
Douglas Gregordae68752011-02-01 22:57:45 +00004821 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00004822 (*CurP)->getIdentifier()->getName()));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004823 }
4824 }
4825 }
4826
Douglas Gregor218937c2011-02-01 19:23:04 +00004827 Results.AddResult(CodeCompletionResult(Builder.TakeString(), CCP_SuperCompletion,
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004828 SuperMethod->isInstanceMethod()
4829 ? CXCursor_ObjCInstanceMethodDecl
4830 : CXCursor_ObjCClassMethodDecl));
4831 return SuperMethod;
4832}
4833
Douglas Gregor8e254cf2010-05-27 23:06:34 +00004834void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall0a2c5e22010-08-25 06:19:51 +00004835 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004836 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
4837 CodeCompletionContext::CCC_ObjCMessageReceiver,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004838 &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregor8e254cf2010-05-27 23:06:34 +00004839
Douglas Gregor8e254cf2010-05-27 23:06:34 +00004840 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4841 Results.EnterNewScope();
Douglas Gregor8071e422010-08-15 06:18:01 +00004842 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4843 CodeCompleter->includeGlobals());
Douglas Gregor8e254cf2010-05-27 23:06:34 +00004844
4845 // If we are in an Objective-C method inside a class that has a superclass,
4846 // add "super" as an option.
4847 if (ObjCMethodDecl *Method = getCurMethodDecl())
4848 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004849 if (Iface->getSuperClass()) {
Douglas Gregor8e254cf2010-05-27 23:06:34 +00004850 Results.AddResult(Result("super"));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004851
4852 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, 0, 0, Results);
4853 }
Douglas Gregor8e254cf2010-05-27 23:06:34 +00004854
4855 Results.ExitScope();
4856
4857 if (CodeCompleter->includeMacros())
4858 AddMacroResults(PP, Results);
Douglas Gregorcee9ff12010-09-20 22:39:41 +00004859 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004860 Results.data(), Results.size());
Douglas Gregor8e254cf2010-05-27 23:06:34 +00004861
4862}
4863
Douglas Gregor2725ca82010-04-21 19:57:20 +00004864void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
4865 IdentifierInfo **SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00004866 unsigned NumSelIdents,
4867 bool AtArgumentExpression) {
Douglas Gregor2725ca82010-04-21 19:57:20 +00004868 ObjCInterfaceDecl *CDecl = 0;
4869 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
4870 // Figure out which interface we're in.
4871 CDecl = CurMethod->getClassInterface();
4872 if (!CDecl)
4873 return;
4874
4875 // Find the superclass of this class.
4876 CDecl = CDecl->getSuperClass();
4877 if (!CDecl)
4878 return;
4879
4880 if (CurMethod->isInstanceMethod()) {
4881 // We are inside an instance method, which means that the message
4882 // send [super ...] is actually calling an instance method on the
Douglas Gregor6b0656a2010-10-13 21:24:53 +00004883 // current object.
4884 return CodeCompleteObjCInstanceMessage(S, 0,
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004885 SelIdents, NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00004886 AtArgumentExpression,
Douglas Gregor6b0656a2010-10-13 21:24:53 +00004887 CDecl);
Douglas Gregor2725ca82010-04-21 19:57:20 +00004888 }
4889
4890 // Fall through to send to the superclass in CDecl.
4891 } else {
4892 // "super" may be the name of a type or variable. Figure out which
4893 // it is.
4894 IdentifierInfo *Super = &Context.Idents.get("super");
4895 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
4896 LookupOrdinaryName);
4897 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
4898 // "super" names an interface. Use it.
4899 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCallc12c5bb2010-05-15 11:32:37 +00004900 if (const ObjCObjectType *Iface
4901 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
4902 CDecl = Iface->getInterface();
Douglas Gregor2725ca82010-04-21 19:57:20 +00004903 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
4904 // "super" names an unresolved type; we can't be more specific.
4905 } else {
4906 // Assume that "super" names some kind of value and parse that way.
4907 CXXScopeSpec SS;
4908 UnqualifiedId id;
4909 id.setIdentifier(Super, SuperLoc);
John McCall60d7b3a2010-08-24 06:29:42 +00004910 ExprResult SuperExpr = ActOnIdExpression(S, SS, id, false, false);
Douglas Gregor2725ca82010-04-21 19:57:20 +00004911 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Douglas Gregor70c5ac72010-09-20 23:34:21 +00004912 SelIdents, NumSelIdents,
4913 AtArgumentExpression);
Douglas Gregor2725ca82010-04-21 19:57:20 +00004914 }
4915
4916 // Fall through
4917 }
4918
John McCallb3d87482010-08-24 05:47:05 +00004919 ParsedType Receiver;
Douglas Gregor2725ca82010-04-21 19:57:20 +00004920 if (CDecl)
John McCallb3d87482010-08-24 05:47:05 +00004921 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor2725ca82010-04-21 19:57:20 +00004922 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00004923 NumSelIdents, AtArgumentExpression,
4924 /*IsSuper=*/true);
Douglas Gregor2725ca82010-04-21 19:57:20 +00004925}
4926
Douglas Gregorb9d77572010-09-21 00:03:25 +00004927/// \brief Given a set of code-completion results for the argument of a message
4928/// send, determine the preferred type (if any) for that argument expression.
4929static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
4930 unsigned NumSelIdents) {
4931 typedef CodeCompletionResult Result;
4932 ASTContext &Context = Results.getSema().Context;
4933
4934 QualType PreferredType;
4935 unsigned BestPriority = CCP_Unlikely * 2;
4936 Result *ResultsData = Results.data();
4937 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
4938 Result &R = ResultsData[I];
4939 if (R.Kind == Result::RK_Declaration &&
4940 isa<ObjCMethodDecl>(R.Declaration)) {
4941 if (R.Priority <= BestPriority) {
4942 ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
4943 if (NumSelIdents <= Method->param_size()) {
4944 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
4945 ->getType();
4946 if (R.Priority < BestPriority || PreferredType.isNull()) {
4947 BestPriority = R.Priority;
4948 PreferredType = MyPreferredType;
4949 } else if (!Context.hasSameUnqualifiedType(PreferredType,
4950 MyPreferredType)) {
4951 PreferredType = QualType();
4952 }
4953 }
4954 }
4955 }
4956 }
4957
4958 return PreferredType;
4959}
4960
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004961static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
4962 ParsedType Receiver,
4963 IdentifierInfo **SelIdents,
4964 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00004965 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004966 bool IsSuper,
4967 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00004968 typedef CodeCompletionResult Result;
Douglas Gregor24a069f2009-11-17 17:59:40 +00004969 ObjCInterfaceDecl *CDecl = 0;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004970
Douglas Gregor24a069f2009-11-17 17:59:40 +00004971 // If the given name refers to an interface type, retrieve the
4972 // corresponding declaration.
Douglas Gregor2725ca82010-04-21 19:57:20 +00004973 if (Receiver) {
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004974 QualType T = SemaRef.GetTypeFromParser(Receiver, 0);
Douglas Gregor2725ca82010-04-21 19:57:20 +00004975 if (!T.isNull())
John McCallc12c5bb2010-05-15 11:32:37 +00004976 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
4977 CDecl = Interface->getInterface();
Douglas Gregor24a069f2009-11-17 17:59:40 +00004978 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004979
Douglas Gregor36ecb042009-11-17 23:22:23 +00004980 // Add all of the factory methods in this Objective-C class, its protocols,
4981 // superclasses, categories, implementation, etc.
Steve Naroffc4df6d22009-11-07 02:08:14 +00004982 Results.EnterNewScope();
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004983
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004984 // If this is a send-to-super, try to add the special "super" send
4985 // completion.
4986 if (IsSuper) {
4987 if (ObjCMethodDecl *SuperMethod
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004988 = AddSuperSendCompletion(SemaRef, false, SelIdents, NumSelIdents,
4989 Results))
Douglas Gregor03d8aec2010-08-27 15:10:57 +00004990 Results.Ignore(SuperMethod);
4991 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004992
Douglas Gregor265f7492010-08-27 15:29:55 +00004993 // If we're inside an Objective-C method definition, prefer its selector to
4994 // others.
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004995 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregor265f7492010-08-27 15:29:55 +00004996 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorc7b6d882010-09-16 15:14:18 +00004997
Douglas Gregord36adf52010-09-16 16:06:31 +00004998 VisitedSelectorSet Selectors;
Douglas Gregor13438f92010-04-06 16:40:00 +00004999 if (CDecl)
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005000 AddObjCMethods(CDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005001 SemaRef.CurContext, Selectors, AtArgumentExpression,
5002 Results);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005003 else {
Douglas Gregor13438f92010-04-06 16:40:00 +00005004 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005005
Douglas Gregor719770d2010-04-06 17:30:22 +00005006 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00005007 // pool from the AST file.
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005008 if (SemaRef.ExternalSource) {
5009 for (uint32_t I = 0,
5010 N = SemaRef.ExternalSource->GetNumExternalSelectors();
John McCall76bd1f32010-06-01 09:23:16 +00005011 I != N; ++I) {
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005012 Selector Sel = SemaRef.ExternalSource->GetExternalSelector(I);
5013 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregor719770d2010-04-06 17:30:22 +00005014 continue;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005015
5016 SemaRef.ReadMethodPool(Sel);
Douglas Gregor719770d2010-04-06 17:30:22 +00005017 }
5018 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005019
5020 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5021 MEnd = SemaRef.MethodPool.end();
Sebastian Redldb9d2142010-08-02 23:18:59 +00005022 M != MEnd; ++M) {
5023 for (ObjCMethodList *MethList = &M->second.second;
5024 MethList && MethList->Method;
Douglas Gregor13438f92010-04-06 16:40:00 +00005025 MethList = MethList->Next) {
5026 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5027 NumSelIdents))
5028 continue;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005029
Douglas Gregor13438f92010-04-06 16:40:00 +00005030 Result R(MethList->Method, 0);
5031 R.StartParameter = NumSelIdents;
5032 R.AllParametersAreInformative = false;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005033 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor13438f92010-04-06 16:40:00 +00005034 }
5035 }
5036 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005037
5038 Results.ExitScope();
5039}
Douglas Gregor13438f92010-04-06 16:40:00 +00005040
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005041void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
5042 IdentifierInfo **SelIdents,
5043 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005044 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005045 bool IsSuper) {
Douglas Gregore081a612011-07-21 01:05:26 +00005046
5047 QualType T = this->GetTypeFromParser(Receiver);
5048
Douglas Gregor218937c2011-02-01 19:23:04 +00005049 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregore081a612011-07-21 01:05:26 +00005050 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Douglas Gregor0a47d692011-07-26 15:24:30 +00005051 T, SelIdents, NumSelIdents));
Douglas Gregore081a612011-07-21 01:05:26 +00005052
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005053 AddClassMessageCompletions(*this, S, Receiver, SelIdents, NumSelIdents,
5054 AtArgumentExpression, IsSuper, Results);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005055
5056 // If we're actually at the argument expression (rather than prior to the
5057 // selector), we're actually performing code completion for an expression.
5058 // Determine whether we have a single, best method. If so, we can
5059 // code-complete the expression using the corresponding parameter type as
5060 // our preferred type, improving completion results.
5061 if (AtArgumentExpression) {
5062 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Douglas Gregore081a612011-07-21 01:05:26 +00005063 NumSelIdents);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005064 if (PreferredType.isNull())
5065 CodeCompleteOrdinaryName(S, PCC_Expression);
5066 else
5067 CodeCompleteExpression(S, PreferredType);
5068 return;
5069 }
5070
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005071 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregore081a612011-07-21 01:05:26 +00005072 Results.getCompletionContext(),
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005073 Results.data(), Results.size());
Steve Naroffc4df6d22009-11-07 02:08:14 +00005074}
5075
Richard Trieuf81e5a92011-09-09 02:00:50 +00005076void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Douglas Gregord3c68542009-11-19 01:08:35 +00005077 IdentifierInfo **SelIdents,
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005078 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005079 bool AtArgumentExpression,
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005080 ObjCInterfaceDecl *Super) {
John McCall0a2c5e22010-08-25 06:19:51 +00005081 typedef CodeCompletionResult Result;
Steve Naroffc4df6d22009-11-07 02:08:14 +00005082
5083 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffc4df6d22009-11-07 02:08:14 +00005084
Douglas Gregor36ecb042009-11-17 23:22:23 +00005085 // If necessary, apply function/array conversion to the receiver.
5086 // C99 6.7.5.3p[7,8].
John Wiegley429bb272011-04-08 18:41:53 +00005087 if (RecExpr) {
5088 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5089 if (Conv.isInvalid()) // conversion failed. bail.
5090 return;
5091 RecExpr = Conv.take();
5092 }
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005093 QualType ReceiverType = RecExpr? RecExpr->getType()
5094 : Super? Context.getObjCObjectPointerType(
5095 Context.getObjCInterfaceType(Super))
5096 : Context.getObjCIdType();
Steve Naroffc4df6d22009-11-07 02:08:14 +00005097
Douglas Gregorda892642010-11-08 21:12:30 +00005098 // If we're messaging an expression with type "id" or "Class", check
5099 // whether we know something special about the receiver that allows
5100 // us to assume a more-specific receiver type.
5101 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType())
5102 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5103 if (ReceiverType->isObjCClassType())
5104 return CodeCompleteObjCClassMessage(S,
5105 ParsedType::make(Context.getObjCInterfaceType(IFace)),
5106 SelIdents, NumSelIdents,
5107 AtArgumentExpression, Super);
5108
5109 ReceiverType = Context.getObjCObjectPointerType(
5110 Context.getObjCInterfaceType(IFace));
5111 }
5112
Douglas Gregor36ecb042009-11-17 23:22:23 +00005113 // Build the set of methods we can see.
Douglas Gregor218937c2011-02-01 19:23:04 +00005114 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregore081a612011-07-21 01:05:26 +00005115 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Douglas Gregor0a47d692011-07-26 15:24:30 +00005116 ReceiverType, SelIdents, NumSelIdents));
Douglas Gregore081a612011-07-21 01:05:26 +00005117
Douglas Gregor36ecb042009-11-17 23:22:23 +00005118 Results.EnterNewScope();
Douglas Gregor22f56992010-04-06 19:22:33 +00005119
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005120 // If this is a send-to-super, try to add the special "super" send
5121 // completion.
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005122 if (Super) {
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005123 if (ObjCMethodDecl *SuperMethod
5124 = AddSuperSendCompletion(*this, false, SelIdents, NumSelIdents,
5125 Results))
5126 Results.Ignore(SuperMethod);
5127 }
5128
Douglas Gregor265f7492010-08-27 15:29:55 +00005129 // If we're inside an Objective-C method definition, prefer its selector to
5130 // others.
5131 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5132 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregor36ecb042009-11-17 23:22:23 +00005133
Douglas Gregord36adf52010-09-16 16:06:31 +00005134 // Keep track of the selectors we've already added.
5135 VisitedSelectorSet Selectors;
5136
Douglas Gregorf74a4192009-11-18 00:06:18 +00005137 // Handle messages to Class. This really isn't a message to an instance
5138 // method, so we treat it the same way we would treat a message send to a
5139 // class method.
5140 if (ReceiverType->isObjCClassType() ||
5141 ReceiverType->isObjCQualifiedClassType()) {
5142 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5143 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Douglas Gregor4ad96852009-11-19 07:41:15 +00005144 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005145 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005146 }
5147 }
5148 // Handle messages to a qualified ID ("id<foo>").
5149 else if (const ObjCObjectPointerType *QualID
5150 = ReceiverType->getAsObjCQualifiedIdType()) {
5151 // Search protocols for instance methods.
5152 for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(),
5153 E = QualID->qual_end();
5154 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00005155 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00005156 Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005157 }
5158 // Handle messages to a pointer to interface type.
5159 else if (const ObjCObjectPointerType *IFacePtr
5160 = ReceiverType->getAsObjCInterfacePointerType()) {
5161 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregor4ad96852009-11-19 07:41:15 +00005162 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005163 NumSelIdents, CurContext, Selectors, AtArgumentExpression,
5164 Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005165
5166 // Search protocols for instance methods.
5167 for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(),
5168 E = IFacePtr->qual_end();
5169 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00005170 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00005171 Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005172 }
Douglas Gregor13438f92010-04-06 16:40:00 +00005173 // Handle messages to "id".
5174 else if (ReceiverType->isObjCIdType()) {
Douglas Gregor719770d2010-04-06 17:30:22 +00005175 // We're messaging "id", so provide all instance methods we know
5176 // about as code-completion results.
5177
5178 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00005179 // pool from the AST file.
Douglas Gregor719770d2010-04-06 17:30:22 +00005180 if (ExternalSource) {
John McCall76bd1f32010-06-01 09:23:16 +00005181 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5182 I != N; ++I) {
5183 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redldb9d2142010-08-02 23:18:59 +00005184 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor719770d2010-04-06 17:30:22 +00005185 continue;
5186
Sebastian Redldb9d2142010-08-02 23:18:59 +00005187 ReadMethodPool(Sel);
Douglas Gregor719770d2010-04-06 17:30:22 +00005188 }
5189 }
5190
Sebastian Redldb9d2142010-08-02 23:18:59 +00005191 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5192 MEnd = MethodPool.end();
5193 M != MEnd; ++M) {
5194 for (ObjCMethodList *MethList = &M->second.first;
5195 MethList && MethList->Method;
Douglas Gregor13438f92010-04-06 16:40:00 +00005196 MethList = MethList->Next) {
5197 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5198 NumSelIdents))
5199 continue;
Douglas Gregord36adf52010-09-16 16:06:31 +00005200
5201 if (!Selectors.insert(MethList->Method->getSelector()))
5202 continue;
5203
Douglas Gregor13438f92010-04-06 16:40:00 +00005204 Result R(MethList->Method, 0);
5205 R.StartParameter = NumSelIdents;
5206 R.AllParametersAreInformative = false;
5207 Results.MaybeAddResult(R, CurContext);
5208 }
5209 }
5210 }
Steve Naroffc4df6d22009-11-07 02:08:14 +00005211 Results.ExitScope();
Douglas Gregorb9d77572010-09-21 00:03:25 +00005212
5213
5214 // If we're actually at the argument expression (rather than prior to the
5215 // selector), we're actually performing code completion for an expression.
5216 // Determine whether we have a single, best method. If so, we can
5217 // code-complete the expression using the corresponding parameter type as
5218 // our preferred type, improving completion results.
5219 if (AtArgumentExpression) {
5220 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
5221 NumSelIdents);
5222 if (PreferredType.isNull())
5223 CodeCompleteOrdinaryName(S, PCC_Expression);
5224 else
5225 CodeCompleteExpression(S, PreferredType);
5226 return;
5227 }
5228
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005229 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregore081a612011-07-21 01:05:26 +00005230 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005231 Results.data(),Results.size());
Steve Naroffc4df6d22009-11-07 02:08:14 +00005232}
Douglas Gregor55385fe2009-11-18 04:19:12 +00005233
Douglas Gregorfb629412010-08-23 21:17:50 +00005234void Sema::CodeCompleteObjCForCollection(Scope *S,
5235 DeclGroupPtrTy IterationVar) {
5236 CodeCompleteExpressionData Data;
5237 Data.ObjCCollection = true;
5238
5239 if (IterationVar.getAsOpaquePtr()) {
5240 DeclGroupRef DG = IterationVar.getAsVal<DeclGroupRef>();
5241 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5242 if (*I)
5243 Data.IgnoreDecls.push_back(*I);
5244 }
5245 }
5246
5247 CodeCompleteExpression(S, Data);
5248}
5249
Douglas Gregor458433d2010-08-26 15:07:07 +00005250void Sema::CodeCompleteObjCSelector(Scope *S, IdentifierInfo **SelIdents,
5251 unsigned NumSelIdents) {
5252 // If we have an external source, load the entire class method
5253 // pool from the AST file.
5254 if (ExternalSource) {
5255 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5256 I != N; ++I) {
5257 Selector Sel = ExternalSource->GetExternalSelector(I);
5258 if (Sel.isNull() || MethodPool.count(Sel))
5259 continue;
5260
5261 ReadMethodPool(Sel);
5262 }
5263 }
5264
Douglas Gregor218937c2011-02-01 19:23:04 +00005265 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
5266 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor458433d2010-08-26 15:07:07 +00005267 Results.EnterNewScope();
5268 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5269 MEnd = MethodPool.end();
5270 M != MEnd; ++M) {
5271
5272 Selector Sel = M->first;
5273 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents, NumSelIdents))
5274 continue;
5275
Douglas Gregor218937c2011-02-01 19:23:04 +00005276 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregor458433d2010-08-26 15:07:07 +00005277 if (Sel.isUnarySelector()) {
Douglas Gregordae68752011-02-01 22:57:45 +00005278 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005279 Sel.getNameForSlot(0)));
Douglas Gregor218937c2011-02-01 19:23:04 +00005280 Results.AddResult(Builder.TakeString());
Douglas Gregor458433d2010-08-26 15:07:07 +00005281 continue;
5282 }
5283
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005284 std::string Accumulator;
Douglas Gregor458433d2010-08-26 15:07:07 +00005285 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005286 if (I == NumSelIdents) {
5287 if (!Accumulator.empty()) {
Douglas Gregordae68752011-02-01 22:57:45 +00005288 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005289 Accumulator));
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005290 Accumulator.clear();
5291 }
5292 }
5293
Benjamin Kramera0651c52011-07-26 16:59:25 +00005294 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005295 Accumulator += ':';
Douglas Gregor458433d2010-08-26 15:07:07 +00005296 }
Douglas Gregordae68752011-02-01 22:57:45 +00005297 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregor218937c2011-02-01 19:23:04 +00005298 Results.AddResult(Builder.TakeString());
Douglas Gregor458433d2010-08-26 15:07:07 +00005299 }
5300 Results.ExitScope();
5301
5302 HandleCodeCompleteResults(this, CodeCompleter,
5303 CodeCompletionContext::CCC_SelectorName,
5304 Results.data(), Results.size());
5305}
5306
Douglas Gregor55385fe2009-11-18 04:19:12 +00005307/// \brief Add all of the protocol declarations that we find in the given
5308/// (translation unit) context.
5309static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor083128f2009-11-18 04:49:41 +00005310 bool OnlyForwardDeclarations,
Douglas Gregor55385fe2009-11-18 04:19:12 +00005311 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005312 typedef CodeCompletionResult Result;
Douglas Gregor55385fe2009-11-18 04:19:12 +00005313
5314 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5315 DEnd = Ctx->decls_end();
5316 D != DEnd; ++D) {
5317 // Record any protocols we find.
5318 if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*D))
Douglas Gregor083128f2009-11-18 04:49:41 +00005319 if (!OnlyForwardDeclarations || Proto->isForwardDecl())
Douglas Gregor608300b2010-01-14 16:14:35 +00005320 Results.AddResult(Result(Proto, 0), CurContext, 0, false);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005321
5322 // Record any forward-declared protocols we find.
5323 if (ObjCForwardProtocolDecl *Forward
5324 = dyn_cast<ObjCForwardProtocolDecl>(*D)) {
5325 for (ObjCForwardProtocolDecl::protocol_iterator
5326 P = Forward->protocol_begin(),
5327 PEnd = Forward->protocol_end();
5328 P != PEnd; ++P)
Douglas Gregor083128f2009-11-18 04:49:41 +00005329 if (!OnlyForwardDeclarations || (*P)->isForwardDecl())
Douglas Gregor608300b2010-01-14 16:14:35 +00005330 Results.AddResult(Result(*P, 0), CurContext, 0, false);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005331 }
5332 }
5333}
5334
5335void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5336 unsigned NumProtocols) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005337 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
5338 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005339
Douglas Gregor70c23352010-12-09 21:44:02 +00005340 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5341 Results.EnterNewScope();
5342
5343 // Tell the result set to ignore all of the protocols we have
5344 // already seen.
5345 // FIXME: This doesn't work when caching code-completion results.
5346 for (unsigned I = 0; I != NumProtocols; ++I)
5347 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5348 Protocols[I].second))
5349 Results.Ignore(Protocol);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005350
Douglas Gregor70c23352010-12-09 21:44:02 +00005351 // Add all protocols.
5352 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5353 Results);
Douglas Gregor083128f2009-11-18 04:49:41 +00005354
Douglas Gregor70c23352010-12-09 21:44:02 +00005355 Results.ExitScope();
5356 }
5357
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005358 HandleCodeCompleteResults(this, CodeCompleter,
5359 CodeCompletionContext::CCC_ObjCProtocolName,
5360 Results.data(),Results.size());
Douglas Gregor083128f2009-11-18 04:49:41 +00005361}
5362
5363void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005364 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
5365 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor083128f2009-11-18 04:49:41 +00005366
Douglas Gregor70c23352010-12-09 21:44:02 +00005367 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5368 Results.EnterNewScope();
5369
5370 // Add all protocols.
5371 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5372 Results);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005373
Douglas Gregor70c23352010-12-09 21:44:02 +00005374 Results.ExitScope();
5375 }
5376
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005377 HandleCodeCompleteResults(this, CodeCompleter,
5378 CodeCompletionContext::CCC_ObjCProtocolName,
5379 Results.data(),Results.size());
Douglas Gregor55385fe2009-11-18 04:19:12 +00005380}
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005381
5382/// \brief Add all of the Objective-C interface declarations that we find in
5383/// the given (translation unit) context.
5384static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5385 bool OnlyForwardDeclarations,
5386 bool OnlyUnimplemented,
5387 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005388 typedef CodeCompletionResult Result;
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005389
5390 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5391 DEnd = Ctx->decls_end();
5392 D != DEnd; ++D) {
Douglas Gregordeacbdc2010-08-11 12:19:30 +00005393 // Record any interfaces we find.
5394 if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*D))
5395 if ((!OnlyForwardDeclarations || Class->isForwardDecl()) &&
5396 (!OnlyUnimplemented || !Class->getImplementation()))
5397 Results.AddResult(Result(Class, 0), CurContext, 0, false);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005398
5399 // Record any forward-declared interfaces we find.
5400 if (ObjCClassDecl *Forward = dyn_cast<ObjCClassDecl>(*D)) {
Fariborz Jahanian95ed7782011-08-27 20:50:59 +00005401 ObjCInterfaceDecl *IDecl = Forward->getForwardInterfaceDecl();
5402 if ((!OnlyForwardDeclarations || IDecl->isForwardDecl()) &&
5403 (!OnlyUnimplemented || !IDecl->getImplementation()))
5404 Results.AddResult(Result(IDecl, 0), CurContext,
5405 0, false);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005406 }
5407 }
5408}
5409
5410void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005411 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
5412 CodeCompletionContext::CCC_Other);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005413 Results.EnterNewScope();
5414
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005415 if (CodeCompleter->includeGlobals()) {
5416 // Add all classes.
5417 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5418 false, Results);
5419 }
5420
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005421 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005422
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005423 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005424 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005425 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005426}
5427
Douglas Gregorc83c6872010-04-15 22:33:43 +00005428void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5429 SourceLocation ClassNameLoc) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005430 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005431 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005432 Results.EnterNewScope();
5433
5434 // Make sure that we ignore the class we're currently defining.
5435 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005436 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005437 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005438 Results.Ignore(CurClass);
5439
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005440 if (CodeCompleter->includeGlobals()) {
5441 // Add all classes.
5442 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5443 false, Results);
5444 }
5445
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005446 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005447
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005448 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005449 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005450 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005451}
5452
5453void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005454 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
5455 CodeCompletionContext::CCC_Other);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005456 Results.EnterNewScope();
5457
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005458 if (CodeCompleter->includeGlobals()) {
5459 // Add all unimplemented classes.
5460 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5461 true, Results);
5462 }
5463
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005464 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005465
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005466 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005467 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005468 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005469}
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005470
5471void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorc83c6872010-04-15 22:33:43 +00005472 IdentifierInfo *ClassName,
5473 SourceLocation ClassNameLoc) {
John McCall0a2c5e22010-08-25 06:19:51 +00005474 typedef CodeCompletionResult Result;
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005475
Douglas Gregor218937c2011-02-01 19:23:04 +00005476 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00005477 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005478
5479 // Ignore any categories we find that have already been implemented by this
5480 // interface.
5481 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5482 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005483 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005484 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass))
5485 for (ObjCCategoryDecl *Category = Class->getCategoryList(); Category;
5486 Category = Category->getNextClassCategory())
5487 CategoryNames.insert(Category->getIdentifier());
5488
5489 // Add all of the categories we know about.
5490 Results.EnterNewScope();
5491 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
5492 for (DeclContext::decl_iterator D = TU->decls_begin(),
5493 DEnd = TU->decls_end();
5494 D != DEnd; ++D)
5495 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(*D))
5496 if (CategoryNames.insert(Category->getIdentifier()))
Douglas Gregor608300b2010-01-14 16:14:35 +00005497 Results.AddResult(Result(Category, 0), CurContext, 0, false);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005498 Results.ExitScope();
5499
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005500 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00005501 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005502 Results.data(),Results.size());
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005503}
5504
5505void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorc83c6872010-04-15 22:33:43 +00005506 IdentifierInfo *ClassName,
5507 SourceLocation ClassNameLoc) {
John McCall0a2c5e22010-08-25 06:19:51 +00005508 typedef CodeCompletionResult Result;
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005509
5510 // Find the corresponding interface. If we couldn't find the interface, the
5511 // program itself is ill-formed. However, we'll try to be helpful still by
5512 // providing the list of all of the categories we know about.
5513 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005514 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005515 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5516 if (!Class)
Douglas Gregorc83c6872010-04-15 22:33:43 +00005517 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005518
Douglas Gregor218937c2011-02-01 19:23:04 +00005519 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00005520 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005521
5522 // Add all of the categories that have have corresponding interface
5523 // declarations in this class and any of its superclasses, except for
5524 // already-implemented categories in the class itself.
5525 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5526 Results.EnterNewScope();
5527 bool IgnoreImplemented = true;
5528 while (Class) {
5529 for (ObjCCategoryDecl *Category = Class->getCategoryList(); Category;
5530 Category = Category->getNextClassCategory())
5531 if ((!IgnoreImplemented || !Category->getImplementation()) &&
5532 CategoryNames.insert(Category->getIdentifier()))
Douglas Gregor608300b2010-01-14 16:14:35 +00005533 Results.AddResult(Result(Category, 0), CurContext, 0, false);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005534
5535 Class = Class->getSuperClass();
5536 IgnoreImplemented = false;
5537 }
5538 Results.ExitScope();
5539
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005540 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00005541 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005542 Results.data(),Results.size());
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005543}
Douglas Gregor322328b2009-11-18 22:32:06 +00005544
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005545void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
John McCall0a2c5e22010-08-25 06:19:51 +00005546 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00005547 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
5548 CodeCompletionContext::CCC_Other);
Douglas Gregor322328b2009-11-18 22:32:06 +00005549
5550 // Figure out where this @synthesize lives.
5551 ObjCContainerDecl *Container
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005552 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor322328b2009-11-18 22:32:06 +00005553 if (!Container ||
5554 (!isa<ObjCImplementationDecl>(Container) &&
5555 !isa<ObjCCategoryImplDecl>(Container)))
5556 return;
5557
5558 // Ignore any properties that have already been implemented.
5559 for (DeclContext::decl_iterator D = Container->decls_begin(),
5560 DEnd = Container->decls_end();
5561 D != DEnd; ++D)
5562 if (ObjCPropertyImplDecl *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(*D))
5563 Results.Ignore(PropertyImpl->getPropertyDecl());
5564
5565 // Add any properties that we find.
Douglas Gregor73449212010-12-09 23:01:55 +00005566 AddedPropertiesSet AddedProperties;
Douglas Gregor322328b2009-11-18 22:32:06 +00005567 Results.EnterNewScope();
5568 if (ObjCImplementationDecl *ClassImpl
5569 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor4b81cde2011-05-05 15:50:42 +00005570 AddObjCProperties(ClassImpl->getClassInterface(), false,
5571 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00005572 AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00005573 else
5574 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor4b81cde2011-05-05 15:50:42 +00005575 false, /*AllowNullaryMethods=*/false, CurContext,
5576 AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00005577 Results.ExitScope();
5578
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005579 HandleCodeCompleteResults(this, CodeCompleter,
5580 CodeCompletionContext::CCC_Other,
5581 Results.data(),Results.size());
Douglas Gregor322328b2009-11-18 22:32:06 +00005582}
5583
5584void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005585 IdentifierInfo *PropertyName) {
John McCall0a2c5e22010-08-25 06:19:51 +00005586 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00005587 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
5588 CodeCompletionContext::CCC_Other);
Douglas Gregor322328b2009-11-18 22:32:06 +00005589
5590 // Figure out where this @synthesize lives.
5591 ObjCContainerDecl *Container
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005592 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor322328b2009-11-18 22:32:06 +00005593 if (!Container ||
5594 (!isa<ObjCImplementationDecl>(Container) &&
5595 !isa<ObjCCategoryImplDecl>(Container)))
5596 return;
5597
5598 // Figure out which interface we're looking into.
5599 ObjCInterfaceDecl *Class = 0;
5600 if (ObjCImplementationDecl *ClassImpl
5601 = dyn_cast<ObjCImplementationDecl>(Container))
5602 Class = ClassImpl->getClassInterface();
5603 else
5604 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
5605 ->getClassInterface();
5606
Douglas Gregore8426052011-04-18 14:40:46 +00005607 // Determine the type of the property we're synthesizing.
5608 QualType PropertyType = Context.getObjCIdType();
5609 if (Class) {
5610 if (ObjCPropertyDecl *Property
5611 = Class->FindPropertyDeclaration(PropertyName)) {
5612 PropertyType
5613 = Property->getType().getNonReferenceType().getUnqualifiedType();
5614
5615 // Give preference to ivars
5616 Results.setPreferredType(PropertyType);
5617 }
5618 }
5619
Douglas Gregor322328b2009-11-18 22:32:06 +00005620 // Add all of the instance variables in this class and its superclasses.
5621 Results.EnterNewScope();
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005622 bool SawSimilarlyNamedIvar = false;
5623 std::string NameWithPrefix;
5624 NameWithPrefix += '_';
Benjamin Kramera0651c52011-07-26 16:59:25 +00005625 NameWithPrefix += PropertyName->getName();
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005626 std::string NameWithSuffix = PropertyName->getName().str();
5627 NameWithSuffix += '_';
Douglas Gregor322328b2009-11-18 22:32:06 +00005628 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005629 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
5630 Ivar = Ivar->getNextIvar()) {
Douglas Gregore8426052011-04-18 14:40:46 +00005631 Results.AddResult(Result(Ivar, 0), CurContext, 0, false);
5632
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005633 // Determine whether we've seen an ivar with a name similar to the
5634 // property.
Douglas Gregore8426052011-04-18 14:40:46 +00005635 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005636 NameWithPrefix == Ivar->getName() ||
Douglas Gregore8426052011-04-18 14:40:46 +00005637 NameWithSuffix == Ivar->getName())) {
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005638 SawSimilarlyNamedIvar = true;
Douglas Gregore8426052011-04-18 14:40:46 +00005639
5640 // Reduce the priority of this result by one, to give it a slight
5641 // advantage over other results whose names don't match so closely.
5642 if (Results.size() &&
5643 Results.data()[Results.size() - 1].Kind
5644 == CodeCompletionResult::RK_Declaration &&
5645 Results.data()[Results.size() - 1].Declaration == Ivar)
5646 Results.data()[Results.size() - 1].Priority--;
5647 }
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005648 }
Douglas Gregor322328b2009-11-18 22:32:06 +00005649 }
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005650
5651 if (!SawSimilarlyNamedIvar) {
5652 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregore8426052011-04-18 14:40:46 +00005653 // an ivar of the appropriate type.
5654 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005655 typedef CodeCompletionResult Result;
5656 CodeCompletionAllocator &Allocator = Results.getAllocator();
5657 CodeCompletionBuilder Builder(Allocator, Priority,CXAvailability_Available);
5658
Douglas Gregor8987b232011-09-27 23:30:47 +00005659 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregore8426052011-04-18 14:40:46 +00005660 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00005661 Policy, Allocator));
Douglas Gregoraa490cb2011-04-18 14:13:53 +00005662 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
5663 Results.AddResult(Result(Builder.TakeString(), Priority,
5664 CXCursor_ObjCIvarDecl));
5665 }
5666
Douglas Gregor322328b2009-11-18 22:32:06 +00005667 Results.ExitScope();
5668
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005669 HandleCodeCompleteResults(this, CodeCompleter,
5670 CodeCompletionContext::CCC_Other,
5671 Results.data(),Results.size());
Douglas Gregor322328b2009-11-18 22:32:06 +00005672}
Douglas Gregore8f5a172010-04-07 00:21:17 +00005673
Douglas Gregor408be5a2010-08-25 01:08:01 +00005674// Mapping from selectors to the methods that implement that selector, along
5675// with the "in original class" flag.
5676typedef llvm::DenseMap<Selector, std::pair<ObjCMethodDecl *, bool> >
5677 KnownMethodsMap;
Douglas Gregore8f5a172010-04-07 00:21:17 +00005678
5679/// \brief Find all of the methods that reside in the given container
5680/// (and its superclasses, protocols, etc.) that meet the given
5681/// criteria. Insert those methods into the map of known methods,
5682/// indexed by selector so they can be easily found.
5683static void FindImplementableMethods(ASTContext &Context,
5684 ObjCContainerDecl *Container,
5685 bool WantInstanceMethods,
5686 QualType ReturnType,
Douglas Gregor408be5a2010-08-25 01:08:01 +00005687 KnownMethodsMap &KnownMethods,
5688 bool InOriginalClass = true) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00005689 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
5690 // Recurse into protocols.
5691 const ObjCList<ObjCProtocolDecl> &Protocols
5692 = IFace->getReferencedProtocols();
5693 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregorea766182010-10-18 18:21:28 +00005694 E = Protocols.end();
Douglas Gregore8f5a172010-04-07 00:21:17 +00005695 I != E; ++I)
5696 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00005697 KnownMethods, InOriginalClass);
Douglas Gregore8f5a172010-04-07 00:21:17 +00005698
Douglas Gregorea766182010-10-18 18:21:28 +00005699 // Add methods from any class extensions and categories.
5700 for (const ObjCCategoryDecl *Cat = IFace->getCategoryList(); Cat;
5701 Cat = Cat->getNextClassCategory())
Fariborz Jahanian80aa1cd2010-06-22 23:20:40 +00005702 FindImplementableMethods(Context, const_cast<ObjCCategoryDecl*>(Cat),
5703 WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00005704 KnownMethods, false);
5705
5706 // Visit the superclass.
5707 if (IFace->getSuperClass())
5708 FindImplementableMethods(Context, IFace->getSuperClass(),
5709 WantInstanceMethods, ReturnType,
5710 KnownMethods, false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00005711 }
5712
5713 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
5714 // Recurse into protocols.
5715 const ObjCList<ObjCProtocolDecl> &Protocols
5716 = Category->getReferencedProtocols();
5717 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregorea766182010-10-18 18:21:28 +00005718 E = Protocols.end();
Douglas Gregore8f5a172010-04-07 00:21:17 +00005719 I != E; ++I)
5720 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00005721 KnownMethods, InOriginalClass);
5722
5723 // If this category is the original class, jump to the interface.
5724 if (InOriginalClass && Category->getClassInterface())
5725 FindImplementableMethods(Context, Category->getClassInterface(),
5726 WantInstanceMethods, ReturnType, KnownMethods,
5727 false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00005728 }
5729
5730 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
5731 // Recurse into protocols.
5732 const ObjCList<ObjCProtocolDecl> &Protocols
5733 = Protocol->getReferencedProtocols();
5734 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
5735 E = Protocols.end();
5736 I != E; ++I)
5737 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00005738 KnownMethods, false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00005739 }
5740
5741 // Add methods in this container. This operation occurs last because
5742 // we want the methods from this container to override any methods
5743 // we've previously seen with the same selector.
5744 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
5745 MEnd = Container->meth_end();
5746 M != MEnd; ++M) {
5747 if ((*M)->isInstanceMethod() == WantInstanceMethods) {
5748 if (!ReturnType.isNull() &&
5749 !Context.hasSameUnqualifiedType(ReturnType, (*M)->getResultType()))
5750 continue;
5751
Douglas Gregor408be5a2010-08-25 01:08:01 +00005752 KnownMethods[(*M)->getSelector()] = std::make_pair(*M, InOriginalClass);
Douglas Gregore8f5a172010-04-07 00:21:17 +00005753 }
5754 }
5755}
5756
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005757/// \brief Add the parenthesized return or parameter type chunk to a code
5758/// completion string.
5759static void AddObjCPassingTypeChunk(QualType Type,
5760 ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00005761 const PrintingPolicy &Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005762 CodeCompletionBuilder &Builder) {
5763 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor8987b232011-09-27 23:30:47 +00005764 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005765 Builder.getAllocator()));
5766 Builder.AddChunk(CodeCompletionString::CK_RightParen);
5767}
5768
5769/// \brief Determine whether the given class is or inherits from a class by
5770/// the given name.
5771static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner5f9e2722011-07-23 10:55:15 +00005772 StringRef Name) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005773 if (!Class)
5774 return false;
5775
5776 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
5777 return true;
5778
5779 return InheritsFromClassNamed(Class->getSuperClass(), Name);
5780}
5781
5782/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
5783/// Key-Value Observing (KVO).
5784static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
5785 bool IsInstanceMethod,
5786 QualType ReturnType,
5787 ASTContext &Context,
Douglas Gregore74c25c2011-05-04 23:50:46 +00005788 VisitedSelectorSet &KnownSelectors,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005789 ResultBuilder &Results) {
5790 IdentifierInfo *PropName = Property->getIdentifier();
5791 if (!PropName || PropName->getLength() == 0)
5792 return;
5793
Douglas Gregor8987b232011-09-27 23:30:47 +00005794 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
5795
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005796 // Builder that will create each code completion.
5797 typedef CodeCompletionResult Result;
5798 CodeCompletionAllocator &Allocator = Results.getAllocator();
5799 CodeCompletionBuilder Builder(Allocator);
5800
5801 // The selector table.
5802 SelectorTable &Selectors = Context.Selectors;
5803
5804 // The property name, copied into the code completion allocation region
5805 // on demand.
5806 struct KeyHolder {
5807 CodeCompletionAllocator &Allocator;
Chris Lattner5f9e2722011-07-23 10:55:15 +00005808 StringRef Key;
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005809 const char *CopiedKey;
5810
Chris Lattner5f9e2722011-07-23 10:55:15 +00005811 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005812 : Allocator(Allocator), Key(Key), CopiedKey(0) { }
5813
5814 operator const char *() {
5815 if (CopiedKey)
5816 return CopiedKey;
5817
5818 return CopiedKey = Allocator.CopyString(Key);
5819 }
5820 } Key(Allocator, PropName->getName());
5821
5822 // The uppercased name of the property name.
5823 std::string UpperKey = PropName->getName();
5824 if (!UpperKey.empty())
5825 UpperKey[0] = toupper(UpperKey[0]);
5826
5827 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
5828 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
5829 Property->getType());
5830 bool ReturnTypeMatchesVoid
5831 = ReturnType.isNull() || ReturnType->isVoidType();
5832
5833 // Add the normal accessor -(type)key.
5834 if (IsInstanceMethod &&
Douglas Gregore74c25c2011-05-04 23:50:46 +00005835 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005836 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
5837 if (ReturnType.isNull())
Douglas Gregor8987b232011-09-27 23:30:47 +00005838 AddObjCPassingTypeChunk(Property->getType(), Context, Policy, Builder);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005839
5840 Builder.AddTypedTextChunk(Key);
5841 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
5842 CXCursor_ObjCInstanceMethodDecl));
5843 }
5844
5845 // If we have an integral or boolean property (or the user has provided
5846 // an integral or boolean return type), add the accessor -(type)isKey.
5847 if (IsInstanceMethod &&
5848 ((!ReturnType.isNull() &&
5849 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
5850 (ReturnType.isNull() &&
5851 (Property->getType()->isIntegerType() ||
5852 Property->getType()->isBooleanType())))) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00005853 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00005854 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00005855 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005856 if (ReturnType.isNull()) {
5857 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
5858 Builder.AddTextChunk("BOOL");
5859 Builder.AddChunk(CodeCompletionString::CK_RightParen);
5860 }
5861
5862 Builder.AddTypedTextChunk(
5863 Allocator.CopyString(SelectorId->getName()));
5864 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
5865 CXCursor_ObjCInstanceMethodDecl));
5866 }
5867 }
5868
5869 // Add the normal mutator.
5870 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
5871 !Property->getSetterMethodDecl()) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00005872 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00005873 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00005874 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005875 if (ReturnType.isNull()) {
5876 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
5877 Builder.AddTextChunk("void");
5878 Builder.AddChunk(CodeCompletionString::CK_RightParen);
5879 }
5880
5881 Builder.AddTypedTextChunk(
5882 Allocator.CopyString(SelectorId->getName()));
5883 Builder.AddTypedTextChunk(":");
Douglas Gregor8987b232011-09-27 23:30:47 +00005884 AddObjCPassingTypeChunk(Property->getType(), Context, Policy, Builder);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005885 Builder.AddTextChunk(Key);
5886 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
5887 CXCursor_ObjCInstanceMethodDecl));
5888 }
5889 }
5890
5891 // Indexed and unordered accessors
5892 unsigned IndexedGetterPriority = CCP_CodePattern;
5893 unsigned IndexedSetterPriority = CCP_CodePattern;
5894 unsigned UnorderedGetterPriority = CCP_CodePattern;
5895 unsigned UnorderedSetterPriority = CCP_CodePattern;
5896 if (const ObjCObjectPointerType *ObjCPointer
5897 = Property->getType()->getAs<ObjCObjectPointerType>()) {
5898 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
5899 // If this interface type is not provably derived from a known
5900 // collection, penalize the corresponding completions.
5901 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
5902 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
5903 if (!InheritsFromClassNamed(IFace, "NSArray"))
5904 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
5905 }
5906
5907 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
5908 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
5909 if (!InheritsFromClassNamed(IFace, "NSSet"))
5910 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
5911 }
5912 }
5913 } else {
5914 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
5915 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
5916 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
5917 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
5918 }
5919
5920 // Add -(NSUInteger)countOf<key>
5921 if (IsInstanceMethod &&
5922 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00005923 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00005924 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00005925 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005926 if (ReturnType.isNull()) {
5927 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
5928 Builder.AddTextChunk("NSUInteger");
5929 Builder.AddChunk(CodeCompletionString::CK_RightParen);
5930 }
5931
5932 Builder.AddTypedTextChunk(
5933 Allocator.CopyString(SelectorId->getName()));
5934 Results.AddResult(Result(Builder.TakeString(),
5935 std::min(IndexedGetterPriority,
5936 UnorderedGetterPriority),
5937 CXCursor_ObjCInstanceMethodDecl));
5938 }
5939 }
5940
5941 // Indexed getters
5942 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
5943 if (IsInstanceMethod &&
5944 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor62041592011-02-17 03:19:26 +00005945 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00005946 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor62041592011-02-17 03:19:26 +00005947 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00005948 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005949 if (ReturnType.isNull()) {
5950 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
5951 Builder.AddTextChunk("id");
5952 Builder.AddChunk(CodeCompletionString::CK_RightParen);
5953 }
5954
5955 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
5956 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
5957 Builder.AddTextChunk("NSUInteger");
5958 Builder.AddChunk(CodeCompletionString::CK_RightParen);
5959 Builder.AddTextChunk("index");
5960 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
5961 CXCursor_ObjCInstanceMethodDecl));
5962 }
5963 }
5964
5965 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
5966 if (IsInstanceMethod &&
5967 (ReturnType.isNull() ||
5968 (ReturnType->isObjCObjectPointerType() &&
5969 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
5970 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
5971 ->getName() == "NSArray"))) {
Douglas Gregor62041592011-02-17 03:19:26 +00005972 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00005973 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor62041592011-02-17 03:19:26 +00005974 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00005975 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005976 if (ReturnType.isNull()) {
5977 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
5978 Builder.AddTextChunk("NSArray *");
5979 Builder.AddChunk(CodeCompletionString::CK_RightParen);
5980 }
5981
5982 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
5983 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
5984 Builder.AddTextChunk("NSIndexSet *");
5985 Builder.AddChunk(CodeCompletionString::CK_RightParen);
5986 Builder.AddTextChunk("indexes");
5987 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
5988 CXCursor_ObjCInstanceMethodDecl));
5989 }
5990 }
5991
5992 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
5993 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00005994 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005995 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00005996 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00005997 &Context.Idents.get("range")
5998 };
5999
Douglas Gregore74c25c2011-05-04 23:50:46 +00006000 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006001 if (ReturnType.isNull()) {
6002 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6003 Builder.AddTextChunk("void");
6004 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6005 }
6006
6007 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6008 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6009 Builder.AddPlaceholderChunk("object-type");
6010 Builder.AddTextChunk(" **");
6011 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6012 Builder.AddTextChunk("buffer");
6013 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6014 Builder.AddTypedTextChunk("range:");
6015 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6016 Builder.AddTextChunk("NSRange");
6017 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6018 Builder.AddTextChunk("inRange");
6019 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6020 CXCursor_ObjCInstanceMethodDecl));
6021 }
6022 }
6023
6024 // Mutable indexed accessors
6025
6026 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6027 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006028 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006029 IdentifierInfo *SelectorIds[2] = {
6030 &Context.Idents.get("insertObject"),
Douglas Gregor62041592011-02-17 03:19:26 +00006031 &Context.Idents.get(SelectorName)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006032 };
6033
Douglas Gregore74c25c2011-05-04 23:50:46 +00006034 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006035 if (ReturnType.isNull()) {
6036 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6037 Builder.AddTextChunk("void");
6038 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6039 }
6040
6041 Builder.AddTypedTextChunk("insertObject:");
6042 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6043 Builder.AddPlaceholderChunk("object-type");
6044 Builder.AddTextChunk(" *");
6045 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6046 Builder.AddTextChunk("object");
6047 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6048 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6049 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6050 Builder.AddPlaceholderChunk("NSUInteger");
6051 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6052 Builder.AddTextChunk("index");
6053 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6054 CXCursor_ObjCInstanceMethodDecl));
6055 }
6056 }
6057
6058 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6059 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006060 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006061 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006062 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006063 &Context.Idents.get("atIndexes")
6064 };
6065
Douglas Gregore74c25c2011-05-04 23:50:46 +00006066 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006067 if (ReturnType.isNull()) {
6068 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6069 Builder.AddTextChunk("void");
6070 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6071 }
6072
6073 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6074 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6075 Builder.AddTextChunk("NSArray *");
6076 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6077 Builder.AddTextChunk("array");
6078 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6079 Builder.AddTypedTextChunk("atIndexes:");
6080 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6081 Builder.AddPlaceholderChunk("NSIndexSet *");
6082 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6083 Builder.AddTextChunk("indexes");
6084 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6085 CXCursor_ObjCInstanceMethodDecl));
6086 }
6087 }
6088
6089 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6090 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006091 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006092 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006093 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006094 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006095 if (ReturnType.isNull()) {
6096 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6097 Builder.AddTextChunk("void");
6098 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6099 }
6100
6101 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6102 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6103 Builder.AddTextChunk("NSUInteger");
6104 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6105 Builder.AddTextChunk("index");
6106 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6107 CXCursor_ObjCInstanceMethodDecl));
6108 }
6109 }
6110
6111 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6112 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006113 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006114 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006115 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006116 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006117 if (ReturnType.isNull()) {
6118 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6119 Builder.AddTextChunk("void");
6120 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6121 }
6122
6123 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6124 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6125 Builder.AddTextChunk("NSIndexSet *");
6126 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6127 Builder.AddTextChunk("indexes");
6128 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6129 CXCursor_ObjCInstanceMethodDecl));
6130 }
6131 }
6132
6133 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6134 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006135 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006136 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006137 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006138 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006139 &Context.Idents.get("withObject")
6140 };
6141
Douglas Gregore74c25c2011-05-04 23:50:46 +00006142 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006143 if (ReturnType.isNull()) {
6144 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6145 Builder.AddTextChunk("void");
6146 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6147 }
6148
6149 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6150 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6151 Builder.AddPlaceholderChunk("NSUInteger");
6152 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6153 Builder.AddTextChunk("index");
6154 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6155 Builder.AddTypedTextChunk("withObject:");
6156 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6157 Builder.AddTextChunk("id");
6158 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6159 Builder.AddTextChunk("object");
6160 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6161 CXCursor_ObjCInstanceMethodDecl));
6162 }
6163 }
6164
6165 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6166 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006167 std::string SelectorName1
Chris Lattner5f9e2722011-07-23 10:55:15 +00006168 = (Twine("replace") + UpperKey + "AtIndexes").str();
6169 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006170 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006171 &Context.Idents.get(SelectorName1),
6172 &Context.Idents.get(SelectorName2)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006173 };
6174
Douglas Gregore74c25c2011-05-04 23:50:46 +00006175 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006176 if (ReturnType.isNull()) {
6177 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6178 Builder.AddTextChunk("void");
6179 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6180 }
6181
6182 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6183 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6184 Builder.AddPlaceholderChunk("NSIndexSet *");
6185 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6186 Builder.AddTextChunk("indexes");
6187 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6188 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6189 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6190 Builder.AddTextChunk("NSArray *");
6191 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6192 Builder.AddTextChunk("array");
6193 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6194 CXCursor_ObjCInstanceMethodDecl));
6195 }
6196 }
6197
6198 // Unordered getters
6199 // - (NSEnumerator *)enumeratorOfKey
6200 if (IsInstanceMethod &&
6201 (ReturnType.isNull() ||
6202 (ReturnType->isObjCObjectPointerType() &&
6203 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6204 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6205 ->getName() == "NSEnumerator"))) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006206 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006207 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006208 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006209 if (ReturnType.isNull()) {
6210 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6211 Builder.AddTextChunk("NSEnumerator *");
6212 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6213 }
6214
6215 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6216 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6217 CXCursor_ObjCInstanceMethodDecl));
6218 }
6219 }
6220
6221 // - (type *)memberOfKey:(type *)object
6222 if (IsInstanceMethod &&
6223 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006224 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006225 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006226 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006227 if (ReturnType.isNull()) {
6228 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6229 Builder.AddPlaceholderChunk("object-type");
6230 Builder.AddTextChunk(" *");
6231 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6232 }
6233
6234 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6235 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6236 if (ReturnType.isNull()) {
6237 Builder.AddPlaceholderChunk("object-type");
6238 Builder.AddTextChunk(" *");
6239 } else {
6240 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006241 Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006242 Builder.getAllocator()));
6243 }
6244 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6245 Builder.AddTextChunk("object");
6246 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6247 CXCursor_ObjCInstanceMethodDecl));
6248 }
6249 }
6250
6251 // Mutable unordered accessors
6252 // - (void)addKeyObject:(type *)object
6253 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006254 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006255 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006256 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006257 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006258 if (ReturnType.isNull()) {
6259 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6260 Builder.AddTextChunk("void");
6261 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6262 }
6263
6264 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6265 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6266 Builder.AddPlaceholderChunk("object-type");
6267 Builder.AddTextChunk(" *");
6268 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6269 Builder.AddTextChunk("object");
6270 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6271 CXCursor_ObjCInstanceMethodDecl));
6272 }
6273 }
6274
6275 // - (void)addKey:(NSSet *)objects
6276 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006277 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006278 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006279 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006280 if (ReturnType.isNull()) {
6281 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6282 Builder.AddTextChunk("void");
6283 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6284 }
6285
6286 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6287 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6288 Builder.AddTextChunk("NSSet *");
6289 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6290 Builder.AddTextChunk("objects");
6291 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6292 CXCursor_ObjCInstanceMethodDecl));
6293 }
6294 }
6295
6296 // - (void)removeKeyObject:(type *)object
6297 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006298 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006299 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006300 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006301 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006302 if (ReturnType.isNull()) {
6303 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6304 Builder.AddTextChunk("void");
6305 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6306 }
6307
6308 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6309 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6310 Builder.AddPlaceholderChunk("object-type");
6311 Builder.AddTextChunk(" *");
6312 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6313 Builder.AddTextChunk("object");
6314 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6315 CXCursor_ObjCInstanceMethodDecl));
6316 }
6317 }
6318
6319 // - (void)removeKey:(NSSet *)objects
6320 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006321 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006322 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006323 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006324 if (ReturnType.isNull()) {
6325 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6326 Builder.AddTextChunk("void");
6327 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6328 }
6329
6330 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6331 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6332 Builder.AddTextChunk("NSSet *");
6333 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6334 Builder.AddTextChunk("objects");
6335 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6336 CXCursor_ObjCInstanceMethodDecl));
6337 }
6338 }
6339
6340 // - (void)intersectKey:(NSSet *)objects
6341 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006342 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006343 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006344 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006345 if (ReturnType.isNull()) {
6346 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6347 Builder.AddTextChunk("void");
6348 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6349 }
6350
6351 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6352 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6353 Builder.AddTextChunk("NSSet *");
6354 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6355 Builder.AddTextChunk("objects");
6356 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6357 CXCursor_ObjCInstanceMethodDecl));
6358 }
6359 }
6360
6361 // Key-Value Observing
6362 // + (NSSet *)keyPathsForValuesAffectingKey
6363 if (!IsInstanceMethod &&
6364 (ReturnType.isNull() ||
6365 (ReturnType->isObjCObjectPointerType() &&
6366 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6367 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6368 ->getName() == "NSSet"))) {
Douglas Gregor62041592011-02-17 03:19:26 +00006369 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006370 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006371 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006372 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006373 if (ReturnType.isNull()) {
6374 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6375 Builder.AddTextChunk("NSSet *");
6376 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6377 }
6378
6379 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6380 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor3f828d12011-06-02 04:02:27 +00006381 CXCursor_ObjCClassMethodDecl));
6382 }
6383 }
6384
6385 // + (BOOL)automaticallyNotifiesObserversForKey
6386 if (!IsInstanceMethod &&
6387 (ReturnType.isNull() ||
6388 ReturnType->isIntegerType() ||
6389 ReturnType->isBooleanType())) {
6390 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006391 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor3f828d12011-06-02 04:02:27 +00006392 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6393 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6394 if (ReturnType.isNull()) {
6395 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6396 Builder.AddTextChunk("BOOL");
6397 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6398 }
6399
6400 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6401 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6402 CXCursor_ObjCClassMethodDecl));
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006403 }
6404 }
6405}
6406
Douglas Gregore8f5a172010-04-07 00:21:17 +00006407void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6408 bool IsInstanceMethod,
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006409 ParsedType ReturnTy) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006410 // Determine the return type of the method we're declaring, if
6411 // provided.
6412 QualType ReturnType = GetTypeFromParser(ReturnTy);
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006413 Decl *IDecl = 0;
6414 if (CurContext->isObjCContainer()) {
6415 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6416 IDecl = cast<Decl>(OCD);
6417 }
Douglas Gregorea766182010-10-18 18:21:28 +00006418 // Determine where we should start searching for methods.
6419 ObjCContainerDecl *SearchDecl = 0;
Douglas Gregore8f5a172010-04-07 00:21:17 +00006420 bool IsInImplementation = false;
John McCalld226f652010-08-21 09:40:31 +00006421 if (Decl *D = IDecl) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006422 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6423 SearchDecl = Impl->getClassInterface();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006424 IsInImplementation = true;
6425 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregorea766182010-10-18 18:21:28 +00006426 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006427 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006428 IsInImplementation = true;
Douglas Gregorea766182010-10-18 18:21:28 +00006429 } else
Douglas Gregore8f5a172010-04-07 00:21:17 +00006430 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006431 }
6432
6433 if (!SearchDecl && S) {
Douglas Gregorea766182010-10-18 18:21:28 +00006434 if (DeclContext *DC = static_cast<DeclContext *>(S->getEntity()))
Douglas Gregore8f5a172010-04-07 00:21:17 +00006435 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006436 }
6437
Douglas Gregorea766182010-10-18 18:21:28 +00006438 if (!SearchDecl) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006439 HandleCodeCompleteResults(this, CodeCompleter,
6440 CodeCompletionContext::CCC_Other,
6441 0, 0);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006442 return;
6443 }
6444
6445 // Find all of the methods that we could declare/implement here.
6446 KnownMethodsMap KnownMethods;
6447 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregorea766182010-10-18 18:21:28 +00006448 ReturnType, KnownMethods);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006449
Douglas Gregore8f5a172010-04-07 00:21:17 +00006450 // Add declarations or definitions for each of the known methods.
John McCall0a2c5e22010-08-25 06:19:51 +00006451 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00006452 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
6453 CodeCompletionContext::CCC_Other);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006454 Results.EnterNewScope();
Douglas Gregor8987b232011-09-27 23:30:47 +00006455 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006456 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6457 MEnd = KnownMethods.end();
6458 M != MEnd; ++M) {
Douglas Gregor408be5a2010-08-25 01:08:01 +00006459 ObjCMethodDecl *Method = M->second.first;
Douglas Gregor218937c2011-02-01 19:23:04 +00006460 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregore8f5a172010-04-07 00:21:17 +00006461
6462 // If the result type was not already provided, add it to the
6463 // pattern as (type).
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006464 if (ReturnType.isNull())
Douglas Gregor8987b232011-09-27 23:30:47 +00006465 AddObjCPassingTypeChunk(Method->getResultType(), Context, Policy,
6466 Builder);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006467
6468 Selector Sel = Method->getSelector();
6469
6470 // Add the first part of the selector to the pattern.
Douglas Gregordae68752011-02-01 22:57:45 +00006471 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00006472 Sel.getNameForSlot(0)));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006473
6474 // Add parameters to the pattern.
6475 unsigned I = 0;
6476 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6477 PEnd = Method->param_end();
6478 P != PEnd; (void)++P, ++I) {
6479 // Add the part of the selector name.
6480 if (I == 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00006481 Builder.AddTypedTextChunk(":");
Douglas Gregore8f5a172010-04-07 00:21:17 +00006482 else if (I < Sel.getNumArgs()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00006483 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6484 Builder.AddTypedTextChunk(
Douglas Gregor813d8342011-02-18 22:29:55 +00006485 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006486 } else
6487 break;
6488
6489 // Add the parameter type.
Douglas Gregor8987b232011-09-27 23:30:47 +00006490 AddObjCPassingTypeChunk((*P)->getOriginalType(), Context, Policy,
6491 Builder);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006492
6493 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregordae68752011-02-01 22:57:45 +00006494 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006495 }
6496
6497 if (Method->isVariadic()) {
6498 if (Method->param_size() > 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00006499 Builder.AddChunk(CodeCompletionString::CK_Comma);
6500 Builder.AddTextChunk("...");
Douglas Gregore17794f2010-08-31 05:13:43 +00006501 }
Douglas Gregore8f5a172010-04-07 00:21:17 +00006502
Douglas Gregor447107d2010-05-28 00:57:46 +00006503 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006504 // We will be defining the method here, so add a compound statement.
Douglas Gregor218937c2011-02-01 19:23:04 +00006505 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6506 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6507 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006508 if (!Method->getResultType()->isVoidType()) {
6509 // If the result type is not void, add a return clause.
Douglas Gregor218937c2011-02-01 19:23:04 +00006510 Builder.AddTextChunk("return");
6511 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6512 Builder.AddPlaceholderChunk("expression");
6513 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006514 } else
Douglas Gregor218937c2011-02-01 19:23:04 +00006515 Builder.AddPlaceholderChunk("statements");
Douglas Gregore8f5a172010-04-07 00:21:17 +00006516
Douglas Gregor218937c2011-02-01 19:23:04 +00006517 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6518 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006519 }
6520
Douglas Gregor408be5a2010-08-25 01:08:01 +00006521 unsigned Priority = CCP_CodePattern;
6522 if (!M->second.second)
6523 Priority += CCD_InBaseClass;
6524
Douglas Gregor218937c2011-02-01 19:23:04 +00006525 Results.AddResult(Result(Builder.TakeString(), Priority,
Douglas Gregor16ed9ad2010-08-17 16:06:07 +00006526 Method->isInstanceMethod()
6527 ? CXCursor_ObjCInstanceMethodDecl
6528 : CXCursor_ObjCClassMethodDecl));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006529 }
6530
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006531 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
6532 // the properties in this class and its categories.
6533 if (Context.getLangOptions().ObjC2) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006534 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006535 Containers.push_back(SearchDecl);
6536
Douglas Gregore74c25c2011-05-04 23:50:46 +00006537 VisitedSelectorSet KnownSelectors;
6538 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6539 MEnd = KnownMethods.end();
6540 M != MEnd; ++M)
6541 KnownSelectors.insert(M->first);
6542
6543
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006544 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
6545 if (!IFace)
6546 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
6547 IFace = Category->getClassInterface();
6548
6549 if (IFace) {
6550 for (ObjCCategoryDecl *Category = IFace->getCategoryList(); Category;
6551 Category = Category->getNextClassCategory())
6552 Containers.push_back(Category);
6553 }
6554
6555 for (unsigned I = 0, N = Containers.size(); I != N; ++I) {
6556 for (ObjCContainerDecl::prop_iterator P = Containers[I]->prop_begin(),
6557 PEnd = Containers[I]->prop_end();
6558 P != PEnd; ++P) {
6559 AddObjCKeyValueCompletions(*P, IsInstanceMethod, ReturnType, Context,
Douglas Gregore74c25c2011-05-04 23:50:46 +00006560 KnownSelectors, Results);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006561 }
6562 }
6563 }
6564
Douglas Gregore8f5a172010-04-07 00:21:17 +00006565 Results.ExitScope();
6566
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006567 HandleCodeCompleteResults(this, CodeCompleter,
6568 CodeCompletionContext::CCC_Other,
6569 Results.data(),Results.size());
Douglas Gregore8f5a172010-04-07 00:21:17 +00006570}
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006571
6572void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
6573 bool IsInstanceMethod,
Douglas Gregor40ed9a12010-07-08 23:37:41 +00006574 bool AtParameterName,
John McCallb3d87482010-08-24 05:47:05 +00006575 ParsedType ReturnTy,
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006576 IdentifierInfo **SelIdents,
6577 unsigned NumSelIdents) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006578 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00006579 // pool from the AST file.
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006580 if (ExternalSource) {
6581 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
6582 I != N; ++I) {
6583 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redldb9d2142010-08-02 23:18:59 +00006584 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006585 continue;
Sebastian Redldb9d2142010-08-02 23:18:59 +00006586
6587 ReadMethodPool(Sel);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006588 }
6589 }
6590
6591 // Build the set of methods we can see.
John McCall0a2c5e22010-08-25 06:19:51 +00006592 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00006593 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
6594 CodeCompletionContext::CCC_Other);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006595
6596 if (ReturnTy)
6597 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redldb9d2142010-08-02 23:18:59 +00006598
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006599 Results.EnterNewScope();
Sebastian Redldb9d2142010-08-02 23:18:59 +00006600 for (GlobalMethodPool::iterator M = MethodPool.begin(),
6601 MEnd = MethodPool.end();
6602 M != MEnd; ++M) {
6603 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
6604 &M->second.second;
6605 MethList && MethList->Method;
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006606 MethList = MethList->Next) {
6607 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
6608 NumSelIdents))
6609 continue;
6610
Douglas Gregor40ed9a12010-07-08 23:37:41 +00006611 if (AtParameterName) {
6612 // Suggest parameter names we've seen before.
6613 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
6614 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
6615 if (Param->getIdentifier()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00006616 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregordae68752011-02-01 22:57:45 +00006617 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00006618 Param->getIdentifier()->getName()));
6619 Results.AddResult(Builder.TakeString());
Douglas Gregor40ed9a12010-07-08 23:37:41 +00006620 }
6621 }
6622
6623 continue;
6624 }
6625
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006626 Result R(MethList->Method, 0);
6627 R.StartParameter = NumSelIdents;
6628 R.AllParametersAreInformative = false;
6629 R.DeclaringEntity = true;
6630 Results.MaybeAddResult(R, CurContext);
6631 }
6632 }
6633
6634 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006635 HandleCodeCompleteResults(this, CodeCompleter,
6636 CodeCompletionContext::CCC_Other,
6637 Results.data(),Results.size());
Douglas Gregor1f5537a2010-07-08 23:20:03 +00006638}
Douglas Gregor87c08a52010-08-13 22:48:40 +00006639
Douglas Gregorf29c5232010-08-24 22:20:20 +00006640void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregor218937c2011-02-01 19:23:04 +00006641 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00006642 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregorf44e8542010-08-24 19:08:16 +00006643 Results.EnterNewScope();
6644
6645 // #if <condition>
Douglas Gregor218937c2011-02-01 19:23:04 +00006646 CodeCompletionBuilder Builder(Results.getAllocator());
6647 Builder.AddTypedTextChunk("if");
6648 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6649 Builder.AddPlaceholderChunk("condition");
6650 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006651
6652 // #ifdef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00006653 Builder.AddTypedTextChunk("ifdef");
6654 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6655 Builder.AddPlaceholderChunk("macro");
6656 Results.AddResult(Builder.TakeString());
6657
Douglas Gregorf44e8542010-08-24 19:08:16 +00006658 // #ifndef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00006659 Builder.AddTypedTextChunk("ifndef");
6660 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6661 Builder.AddPlaceholderChunk("macro");
6662 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006663
6664 if (InConditional) {
6665 // #elif <condition>
Douglas Gregor218937c2011-02-01 19:23:04 +00006666 Builder.AddTypedTextChunk("elif");
6667 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6668 Builder.AddPlaceholderChunk("condition");
6669 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006670
6671 // #else
Douglas Gregor218937c2011-02-01 19:23:04 +00006672 Builder.AddTypedTextChunk("else");
6673 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006674
6675 // #endif
Douglas Gregor218937c2011-02-01 19:23:04 +00006676 Builder.AddTypedTextChunk("endif");
6677 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006678 }
6679
6680 // #include "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00006681 Builder.AddTypedTextChunk("include");
6682 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6683 Builder.AddTextChunk("\"");
6684 Builder.AddPlaceholderChunk("header");
6685 Builder.AddTextChunk("\"");
6686 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006687
6688 // #include <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00006689 Builder.AddTypedTextChunk("include");
6690 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6691 Builder.AddTextChunk("<");
6692 Builder.AddPlaceholderChunk("header");
6693 Builder.AddTextChunk(">");
6694 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006695
6696 // #define <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00006697 Builder.AddTypedTextChunk("define");
6698 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6699 Builder.AddPlaceholderChunk("macro");
6700 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006701
6702 // #define <macro>(<args>)
Douglas Gregor218937c2011-02-01 19:23:04 +00006703 Builder.AddTypedTextChunk("define");
6704 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6705 Builder.AddPlaceholderChunk("macro");
6706 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6707 Builder.AddPlaceholderChunk("args");
6708 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6709 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006710
6711 // #undef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00006712 Builder.AddTypedTextChunk("undef");
6713 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6714 Builder.AddPlaceholderChunk("macro");
6715 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006716
6717 // #line <number>
Douglas Gregor218937c2011-02-01 19:23:04 +00006718 Builder.AddTypedTextChunk("line");
6719 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6720 Builder.AddPlaceholderChunk("number");
6721 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006722
6723 // #line <number> "filename"
Douglas Gregor218937c2011-02-01 19:23:04 +00006724 Builder.AddTypedTextChunk("line");
6725 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6726 Builder.AddPlaceholderChunk("number");
6727 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6728 Builder.AddTextChunk("\"");
6729 Builder.AddPlaceholderChunk("filename");
6730 Builder.AddTextChunk("\"");
6731 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006732
6733 // #error <message>
Douglas Gregor218937c2011-02-01 19:23:04 +00006734 Builder.AddTypedTextChunk("error");
6735 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6736 Builder.AddPlaceholderChunk("message");
6737 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006738
6739 // #pragma <arguments>
Douglas Gregor218937c2011-02-01 19:23:04 +00006740 Builder.AddTypedTextChunk("pragma");
6741 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6742 Builder.AddPlaceholderChunk("arguments");
6743 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006744
6745 if (getLangOptions().ObjC1) {
6746 // #import "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00006747 Builder.AddTypedTextChunk("import");
6748 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6749 Builder.AddTextChunk("\"");
6750 Builder.AddPlaceholderChunk("header");
6751 Builder.AddTextChunk("\"");
6752 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006753
6754 // #import <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00006755 Builder.AddTypedTextChunk("import");
6756 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6757 Builder.AddTextChunk("<");
6758 Builder.AddPlaceholderChunk("header");
6759 Builder.AddTextChunk(">");
6760 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006761 }
6762
6763 // #include_next "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00006764 Builder.AddTypedTextChunk("include_next");
6765 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6766 Builder.AddTextChunk("\"");
6767 Builder.AddPlaceholderChunk("header");
6768 Builder.AddTextChunk("\"");
6769 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006770
6771 // #include_next <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00006772 Builder.AddTypedTextChunk("include_next");
6773 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6774 Builder.AddTextChunk("<");
6775 Builder.AddPlaceholderChunk("header");
6776 Builder.AddTextChunk(">");
6777 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006778
6779 // #warning <message>
Douglas Gregor218937c2011-02-01 19:23:04 +00006780 Builder.AddTypedTextChunk("warning");
6781 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6782 Builder.AddPlaceholderChunk("message");
6783 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00006784
6785 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
6786 // completions for them. And __include_macros is a Clang-internal extension
6787 // that we don't want to encourage anyone to use.
6788
6789 // FIXME: we don't support #assert or #unassert, so don't suggest them.
6790 Results.ExitScope();
6791
Douglas Gregorf44e8542010-08-24 19:08:16 +00006792 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor721f3592010-08-25 18:41:16 +00006793 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregorf44e8542010-08-24 19:08:16 +00006794 Results.data(), Results.size());
6795}
6796
6797void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorf29c5232010-08-24 22:20:20 +00006798 CodeCompleteOrdinaryName(S,
John McCallf312b1e2010-08-26 23:41:50 +00006799 S->getFnParent()? Sema::PCC_RecoveryInFunction
6800 : Sema::PCC_Namespace);
Douglas Gregorf44e8542010-08-24 19:08:16 +00006801}
6802
Douglas Gregorf29c5232010-08-24 22:20:20 +00006803void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregor218937c2011-02-01 19:23:04 +00006804 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00006805 IsDefinition? CodeCompletionContext::CCC_MacroName
6806 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor1fbb4472010-08-24 20:21:13 +00006807 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
6808 // Add just the names of macros, not their arguments.
Douglas Gregor218937c2011-02-01 19:23:04 +00006809 CodeCompletionBuilder Builder(Results.getAllocator());
Douglas Gregor1fbb4472010-08-24 20:21:13 +00006810 Results.EnterNewScope();
6811 for (Preprocessor::macro_iterator M = PP.macro_begin(),
6812 MEnd = PP.macro_end();
6813 M != MEnd; ++M) {
Douglas Gregordae68752011-02-01 22:57:45 +00006814 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00006815 M->first->getName()));
6816 Results.AddResult(Builder.TakeString());
Douglas Gregor1fbb4472010-08-24 20:21:13 +00006817 }
6818 Results.ExitScope();
6819 } else if (IsDefinition) {
6820 // FIXME: Can we detect when the user just wrote an include guard above?
6821 }
6822
Douglas Gregor52779fb2010-09-23 23:01:17 +00006823 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor1fbb4472010-08-24 20:21:13 +00006824 Results.data(), Results.size());
6825}
6826
Douglas Gregorf29c5232010-08-24 22:20:20 +00006827void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregor218937c2011-02-01 19:23:04 +00006828 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00006829 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorf29c5232010-08-24 22:20:20 +00006830
6831 if (!CodeCompleter || CodeCompleter->includeMacros())
6832 AddMacroResults(PP, Results);
6833
6834 // defined (<macro>)
6835 Results.EnterNewScope();
Douglas Gregor218937c2011-02-01 19:23:04 +00006836 CodeCompletionBuilder Builder(Results.getAllocator());
6837 Builder.AddTypedTextChunk("defined");
6838 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6839 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6840 Builder.AddPlaceholderChunk("macro");
6841 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6842 Results.AddResult(Builder.TakeString());
Douglas Gregorf29c5232010-08-24 22:20:20 +00006843 Results.ExitScope();
6844
6845 HandleCodeCompleteResults(this, CodeCompleter,
6846 CodeCompletionContext::CCC_PreprocessorExpression,
6847 Results.data(), Results.size());
6848}
6849
6850void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
6851 IdentifierInfo *Macro,
6852 MacroInfo *MacroInfo,
6853 unsigned Argument) {
6854 // FIXME: In the future, we could provide "overload" results, much like we
6855 // do for function calls.
6856
Argyrios Kyrtzidis5c5f03e2011-08-18 19:41:28 +00006857 // Now just ignore this. There will be another code-completion callback
6858 // for the expanded tokens.
Douglas Gregorf29c5232010-08-24 22:20:20 +00006859}
6860
Douglas Gregor55817af2010-08-25 17:04:25 +00006861void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor55817af2010-08-25 17:04:25 +00006862 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregoraf1c6b52010-08-25 17:10:00 +00006863 CodeCompletionContext::CCC_NaturalLanguage,
Douglas Gregor55817af2010-08-25 17:04:25 +00006864 0, 0);
6865}
6866
Douglas Gregordae68752011-02-01 22:57:45 +00006867void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Chris Lattner5f9e2722011-07-23 10:55:15 +00006868 SmallVectorImpl<CodeCompletionResult> &Results) {
Douglas Gregor218937c2011-02-01 19:23:04 +00006869 ResultBuilder Builder(*this, Allocator, CodeCompletionContext::CCC_Recovery);
Douglas Gregor8071e422010-08-15 06:18:01 +00006870 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
6871 CodeCompletionDeclConsumer Consumer(Builder,
6872 Context.getTranslationUnitDecl());
6873 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
6874 Consumer);
6875 }
Douglas Gregor87c08a52010-08-13 22:48:40 +00006876
6877 if (!CodeCompleter || CodeCompleter->includeMacros())
6878 AddMacroResults(PP, Builder);
6879
6880 Results.clear();
6881 Results.insert(Results.end(),
6882 Builder.data(), Builder.data() + Builder.size());
6883}