blob: 34c83edcde7caacf6cccd74e162b51c17ccf96ec [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"
John McCall7cd088e2010-08-24 07:21:54 +000014#include "clang/AST/DeclObjC.h"
Douglas Gregorb9d0ef72009-09-21 19:57:38 +000015#include "clang/AST/ExprCXX.h"
Douglas Gregor24a069f2009-11-17 17:59:40 +000016#include "clang/AST/ExprObjC.h"
Jordan Rose223f0ff2013-02-09 10:09:43 +000017#include "clang/Basic/CharInfo.h"
Douglas Gregorc5b2e582012-01-29 18:15:03 +000018#include "clang/Lex/HeaderSearch.h"
Douglas Gregor3f7c7f42009-10-30 16:50:04 +000019#include "clang/Lex/MacroInfo.h"
20#include "clang/Lex/Preprocessor.h"
Chandler Carruth55fc8732012-12-04 09:13:33 +000021#include "clang/Sema/CodeCompleteConsumer.h"
22#include "clang/Sema/ExternalSemaSource.h"
23#include "clang/Sema/Lookup.h"
24#include "clang/Sema/Overload.h"
25#include "clang/Sema/Scope.h"
26#include "clang/Sema/ScopeInfo.h"
Douglas Gregord36adf52010-09-16 16:06:31 +000027#include "llvm/ADT/DenseSet.h"
Benjamin Kramer013b3662012-01-30 16:17:39 +000028#include "llvm/ADT/SmallBitVector.h"
Douglas Gregor86d9a522009-09-21 16:56:56 +000029#include "llvm/ADT/SmallPtrSet.h"
Benjamin Kramer8fe83e12012-02-04 13:45:25 +000030#include "llvm/ADT/SmallString.h"
Douglas Gregor6a684032009-09-28 03:51:44 +000031#include "llvm/ADT/StringExtras.h"
Douglas Gregor22f56992010-04-06 19:22:33 +000032#include "llvm/ADT/StringSwitch.h"
Douglas Gregor458433d2010-08-26 15:07:07 +000033#include "llvm/ADT/Twine.h"
Douglas Gregor86d9a522009-09-21 16:56:56 +000034#include <list>
35#include <map>
36#include <vector>
Douglas Gregor81b747b2009-09-17 21:32:03 +000037
38using namespace clang;
John McCall781472f2010-08-25 08:40:02 +000039using namespace sema;
Douglas Gregor81b747b2009-09-17 21:32:03 +000040
Douglas Gregor86d9a522009-09-21 16:56:56 +000041namespace {
42 /// \brief A container of code-completion results.
43 class ResultBuilder {
44 public:
45 /// \brief The type of a name-lookup filter, which can be provided to the
46 /// name-lookup routines to specify which declarations should be included in
47 /// the result set (when it returns true) and which declarations should be
48 /// filtered out (returns false).
Dmitri Gribenko89cf4252013-01-23 17:21:11 +000049 typedef bool (ResultBuilder::*LookupFilter)(const NamedDecl *) const;
Douglas Gregor86d9a522009-09-21 16:56:56 +000050
John McCall0a2c5e22010-08-25 06:19:51 +000051 typedef CodeCompletionResult Result;
Douglas Gregor86d9a522009-09-21 16:56:56 +000052
53 private:
54 /// \brief The actual results we have found.
55 std::vector<Result> Results;
56
57 /// \brief A record of all of the declarations we have found and placed
58 /// into the result set, used to ensure that no declaration ever gets into
59 /// the result set twice.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +000060 llvm::SmallPtrSet<const Decl*, 16> AllDeclsFound;
Douglas Gregor86d9a522009-09-21 16:56:56 +000061
Dmitri Gribenko89cf4252013-01-23 17:21:11 +000062 typedef std::pair<const NamedDecl *, unsigned> DeclIndexPair;
Douglas Gregorfbcb5d62009-12-06 20:23:50 +000063
64 /// \brief An entry in the shadow map, which is optimized to store
65 /// a single (declaration, index) mapping (the common case) but
66 /// can also store a list of (declaration, index) mappings.
67 class ShadowMapEntry {
Chris Lattner5f9e2722011-07-23 10:55:15 +000068 typedef SmallVector<DeclIndexPair, 4> DeclIndexPairVector;
Douglas Gregorfbcb5d62009-12-06 20:23:50 +000069
70 /// \brief Contains either the solitary NamedDecl * or a vector
71 /// of (declaration, index) pairs.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +000072 llvm::PointerUnion<const NamedDecl *, DeclIndexPairVector*> DeclOrVector;
Douglas Gregorfbcb5d62009-12-06 20:23:50 +000073
74 /// \brief When the entry contains a single declaration, this is
75 /// the index associated with that entry.
76 unsigned SingleDeclIndex;
77
78 public:
79 ShadowMapEntry() : DeclOrVector(), SingleDeclIndex(0) { }
80
Dmitri Gribenko89cf4252013-01-23 17:21:11 +000081 void Add(const NamedDecl *ND, unsigned Index) {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +000082 if (DeclOrVector.isNull()) {
83 // 0 - > 1 elements: just set the single element information.
84 DeclOrVector = ND;
85 SingleDeclIndex = Index;
86 return;
87 }
88
Dmitri Gribenko89cf4252013-01-23 17:21:11 +000089 if (const NamedDecl *PrevND =
90 DeclOrVector.dyn_cast<const NamedDecl *>()) {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +000091 // 1 -> 2 elements: create the vector of results and push in the
92 // existing declaration.
93 DeclIndexPairVector *Vec = new DeclIndexPairVector;
94 Vec->push_back(DeclIndexPair(PrevND, SingleDeclIndex));
95 DeclOrVector = Vec;
96 }
97
98 // Add the new element to the end of the vector.
99 DeclOrVector.get<DeclIndexPairVector*>()->push_back(
100 DeclIndexPair(ND, Index));
101 }
102
103 void Destroy() {
104 if (DeclIndexPairVector *Vec
105 = DeclOrVector.dyn_cast<DeclIndexPairVector *>()) {
106 delete Vec;
107 DeclOrVector = ((NamedDecl *)0);
108 }
109 }
110
111 // Iteration.
112 class iterator;
113 iterator begin() const;
114 iterator end() const;
115 };
116
Douglas Gregor86d9a522009-09-21 16:56:56 +0000117 /// \brief A mapping from declaration names to the declarations that have
118 /// this name within a particular scope and their index within the list of
119 /// results.
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000120 typedef llvm::DenseMap<DeclarationName, ShadowMapEntry> ShadowMap;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000121
122 /// \brief The semantic analysis object for which results are being
123 /// produced.
124 Sema &SemaRef;
Douglas Gregor218937c2011-02-01 19:23:04 +0000125
126 /// \brief The allocator used to allocate new code-completion strings.
Douglas Gregordae68752011-02-01 22:57:45 +0000127 CodeCompletionAllocator &Allocator;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +0000128
129 CodeCompletionTUInfo &CCTUInfo;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000130
131 /// \brief If non-NULL, a filter function used to remove any code-completion
132 /// results that are not desirable.
133 LookupFilter Filter;
Douglas Gregor45bcd432010-01-14 03:21:49 +0000134
135 /// \brief Whether we should allow declarations as
136 /// nested-name-specifiers that would otherwise be filtered out.
137 bool AllowNestedNameSpecifiers;
138
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +0000139 /// \brief If set, the type that we would prefer our resulting value
140 /// declarations to have.
141 ///
142 /// Closely matching the preferred type gives a boost to a result's
143 /// priority.
144 CanQualType PreferredType;
145
Douglas Gregor86d9a522009-09-21 16:56:56 +0000146 /// \brief A list of shadow maps, which is used to model name hiding at
147 /// different levels of, e.g., the inheritance hierarchy.
148 std::list<ShadowMap> ShadowMaps;
149
Douglas Gregor3cdee122010-08-26 16:36:48 +0000150 /// \brief If we're potentially referring to a C++ member function, the set
151 /// of qualifiers applied to the object type.
152 Qualifiers ObjectTypeQualifiers;
153
154 /// \brief Whether the \p ObjectTypeQualifiers field is active.
155 bool HasObjectTypeQualifiers;
156
Douglas Gregor265f7492010-08-27 15:29:55 +0000157 /// \brief The selector that we prefer.
158 Selector PreferredSelector;
159
Douglas Gregorca45da02010-11-02 20:36:02 +0000160 /// \brief The completion context in which we are gathering results.
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000161 CodeCompletionContext CompletionContext;
162
James Dennetta40f7922012-06-14 03:11:41 +0000163 /// \brief If we are in an instance method definition, the \@implementation
Douglas Gregorca45da02010-11-02 20:36:02 +0000164 /// object.
165 ObjCImplementationDecl *ObjCImplementation;
Douglas Gregord1f09b42013-01-31 04:52:16 +0000166
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000167 void AdjustResultPriorityForDecl(Result &R);
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000168
Douglas Gregor6f942b22010-09-21 16:06:22 +0000169 void MaybeAddConstructorResults(Result R);
170
Douglas Gregor86d9a522009-09-21 16:56:56 +0000171 public:
Douglas Gregordae68752011-02-01 22:57:45 +0000172 explicit ResultBuilder(Sema &SemaRef, CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +0000173 CodeCompletionTUInfo &CCTUInfo,
Douglas Gregor52779fb2010-09-23 23:01:17 +0000174 const CodeCompletionContext &CompletionContext,
175 LookupFilter Filter = 0)
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +0000176 : SemaRef(SemaRef), Allocator(Allocator), CCTUInfo(CCTUInfo),
177 Filter(Filter),
Douglas Gregor218937c2011-02-01 19:23:04 +0000178 AllowNestedNameSpecifiers(false), HasObjectTypeQualifiers(false),
Douglas Gregorca45da02010-11-02 20:36:02 +0000179 CompletionContext(CompletionContext),
180 ObjCImplementation(0)
181 {
182 // If this is an Objective-C instance method definition, dig out the
183 // corresponding implementation.
184 switch (CompletionContext.getKind()) {
185 case CodeCompletionContext::CCC_Expression:
186 case CodeCompletionContext::CCC_ObjCMessageReceiver:
187 case CodeCompletionContext::CCC_ParenthesizedExpression:
188 case CodeCompletionContext::CCC_Statement:
189 case CodeCompletionContext::CCC_Recovery:
190 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl())
191 if (Method->isInstanceMethod())
192 if (ObjCInterfaceDecl *Interface = Method->getClassInterface())
193 ObjCImplementation = Interface->getImplementation();
194 break;
195
196 default:
197 break;
198 }
199 }
Douglas Gregord1f09b42013-01-31 04:52:16 +0000200
201 /// \brief Determine the priority for a reference to the given declaration.
202 unsigned getBasePriority(const NamedDecl *D);
203
Douglas Gregord8e8a582010-05-25 21:41:55 +0000204 /// \brief Whether we should include code patterns in the completion
205 /// results.
206 bool includeCodePatterns() const {
207 return SemaRef.CodeCompleter &&
Douglas Gregorf6961522010-08-27 21:18:54 +0000208 SemaRef.CodeCompleter->includeCodePatterns();
Douglas Gregord8e8a582010-05-25 21:41:55 +0000209 }
210
Douglas Gregor86d9a522009-09-21 16:56:56 +0000211 /// \brief Set the filter used for code-completion results.
212 void setFilter(LookupFilter Filter) {
213 this->Filter = Filter;
214 }
215
Douglas Gregor86d9a522009-09-21 16:56:56 +0000216 Result *data() { return Results.empty()? 0 : &Results.front(); }
217 unsigned size() const { return Results.size(); }
218 bool empty() const { return Results.empty(); }
219
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +0000220 /// \brief Specify the preferred type.
221 void setPreferredType(QualType T) {
222 PreferredType = SemaRef.Context.getCanonicalType(T);
223 }
224
Douglas Gregor3cdee122010-08-26 16:36:48 +0000225 /// \brief Set the cv-qualifiers on the object type, for us in filtering
226 /// calls to member functions.
227 ///
228 /// When there are qualifiers in this set, they will be used to filter
229 /// out member functions that aren't available (because there will be a
230 /// cv-qualifier mismatch) or prefer functions with an exact qualifier
231 /// match.
232 void setObjectTypeQualifiers(Qualifiers Quals) {
233 ObjectTypeQualifiers = Quals;
234 HasObjectTypeQualifiers = true;
235 }
236
Douglas Gregor265f7492010-08-27 15:29:55 +0000237 /// \brief Set the preferred selector.
238 ///
239 /// When an Objective-C method declaration result is added, and that
240 /// method's selector matches this preferred selector, we give that method
241 /// a slight priority boost.
242 void setPreferredSelector(Selector Sel) {
243 PreferredSelector = Sel;
244 }
Douglas Gregorca45da02010-11-02 20:36:02 +0000245
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000246 /// \brief Retrieve the code-completion context for which results are
247 /// being collected.
248 const CodeCompletionContext &getCompletionContext() const {
249 return CompletionContext;
250 }
251
Douglas Gregor45bcd432010-01-14 03:21:49 +0000252 /// \brief Specify whether nested-name-specifiers are allowed.
253 void allowNestedNameSpecifiers(bool Allow = true) {
254 AllowNestedNameSpecifiers = Allow;
255 }
256
Douglas Gregorb9d77572010-09-21 00:03:25 +0000257 /// \brief Return the semantic analysis object for which we are collecting
258 /// code completion results.
259 Sema &getSema() const { return SemaRef; }
260
Douglas Gregor218937c2011-02-01 19:23:04 +0000261 /// \brief Retrieve the allocator used to allocate code completion strings.
Douglas Gregordae68752011-02-01 22:57:45 +0000262 CodeCompletionAllocator &getAllocator() const { return Allocator; }
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +0000263
264 CodeCompletionTUInfo &getCodeCompletionTUInfo() const { return CCTUInfo; }
Douglas Gregor218937c2011-02-01 19:23:04 +0000265
Douglas Gregore495b7f2010-01-14 00:20:49 +0000266 /// \brief Determine whether the given declaration is at all interesting
267 /// as a code-completion result.
Douglas Gregor45bcd432010-01-14 03:21:49 +0000268 ///
269 /// \param ND the declaration that we are inspecting.
270 ///
271 /// \param AsNestedNameSpecifier will be set true if this declaration is
272 /// only interesting when it is a nested-name-specifier.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000273 bool isInterestingDecl(const NamedDecl *ND,
274 bool &AsNestedNameSpecifier) const;
Douglas Gregor6660d842010-01-14 00:41:07 +0000275
276 /// \brief Check whether the result is hidden by the Hiding declaration.
277 ///
278 /// \returns true if the result is hidden and cannot be found, false if
279 /// the hidden result could still be found. When false, \p R may be
280 /// modified to describe how the result can be found (e.g., via extra
281 /// qualification).
282 bool CheckHiddenResult(Result &R, DeclContext *CurContext,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000283 const NamedDecl *Hiding);
Douglas Gregor6660d842010-01-14 00:41:07 +0000284
Douglas Gregor86d9a522009-09-21 16:56:56 +0000285 /// \brief Add a new result to this result set (if it isn't already in one
286 /// of the shadow maps), or replace an existing result (for, e.g., a
287 /// redeclaration).
Douglas Gregor456c4a12009-09-21 20:12:40 +0000288 ///
Douglas Gregor81f3bff2012-02-15 15:34:24 +0000289 /// \param R the result to add (if it is unique).
Douglas Gregor456c4a12009-09-21 20:12:40 +0000290 ///
Douglas Gregor81f3bff2012-02-15 15:34:24 +0000291 /// \param CurContext the context in which this result will be named.
Douglas Gregor456c4a12009-09-21 20:12:40 +0000292 void MaybeAddResult(Result R, DeclContext *CurContext = 0);
Douglas Gregor86d9a522009-09-21 16:56:56 +0000293
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000294 /// \brief Add a new result to this result set, where we already know
295 /// the hiding declation (if any).
296 ///
297 /// \param R the result to add (if it is unique).
298 ///
299 /// \param CurContext the context in which this result will be named.
300 ///
301 /// \param Hiding the declaration that hides the result.
Douglas Gregor0cc84042010-01-14 15:47:35 +0000302 ///
303 /// \param InBaseClass whether the result was found in a base
304 /// class of the searched context.
305 void AddResult(Result R, DeclContext *CurContext, NamedDecl *Hiding,
306 bool InBaseClass);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000307
Douglas Gregora4477812010-01-14 16:01:26 +0000308 /// \brief Add a new non-declaration result to this result set.
309 void AddResult(Result R);
310
Douglas Gregor86d9a522009-09-21 16:56:56 +0000311 /// \brief Enter into a new scope.
312 void EnterNewScope();
313
314 /// \brief Exit from the current scope.
315 void ExitScope();
316
Douglas Gregor55385fe2009-11-18 04:19:12 +0000317 /// \brief Ignore this declaration, if it is seen again.
Dmitri Gribenko68a932d2013-02-14 13:53:30 +0000318 void Ignore(const Decl *D) { AllDeclsFound.insert(D->getCanonicalDecl()); }
Douglas Gregor55385fe2009-11-18 04:19:12 +0000319
Douglas Gregor86d9a522009-09-21 16:56:56 +0000320 /// \name Name lookup predicates
321 ///
322 /// These predicates can be passed to the name lookup functions to filter the
323 /// results of name lookup. All of the predicates have the same type, so that
324 ///
325 //@{
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000326 bool IsOrdinaryName(const NamedDecl *ND) const;
327 bool IsOrdinaryNonTypeName(const NamedDecl *ND) const;
328 bool IsIntegralConstantValue(const NamedDecl *ND) const;
329 bool IsOrdinaryNonValueName(const NamedDecl *ND) const;
330 bool IsNestedNameSpecifier(const NamedDecl *ND) const;
331 bool IsEnum(const NamedDecl *ND) const;
332 bool IsClassOrStruct(const NamedDecl *ND) const;
333 bool IsUnion(const NamedDecl *ND) const;
334 bool IsNamespace(const NamedDecl *ND) const;
335 bool IsNamespaceOrAlias(const NamedDecl *ND) const;
336 bool IsType(const NamedDecl *ND) const;
337 bool IsMember(const NamedDecl *ND) const;
338 bool IsObjCIvar(const NamedDecl *ND) const;
339 bool IsObjCMessageReceiver(const NamedDecl *ND) const;
340 bool IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const;
341 bool IsObjCCollection(const NamedDecl *ND) const;
342 bool IsImpossibleToSatisfy(const NamedDecl *ND) const;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000343 //@}
344 };
345}
346
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000347class ResultBuilder::ShadowMapEntry::iterator {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000348 llvm::PointerUnion<const NamedDecl *, const DeclIndexPair *> DeclOrIterator;
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000349 unsigned SingleDeclIndex;
350
351public:
352 typedef DeclIndexPair value_type;
353 typedef value_type reference;
354 typedef std::ptrdiff_t difference_type;
355 typedef std::input_iterator_tag iterator_category;
356
357 class pointer {
358 DeclIndexPair Value;
359
360 public:
361 pointer(const DeclIndexPair &Value) : Value(Value) { }
362
363 const DeclIndexPair *operator->() const {
364 return &Value;
365 }
366 };
367
368 iterator() : DeclOrIterator((NamedDecl *)0), SingleDeclIndex(0) { }
369
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000370 iterator(const NamedDecl *SingleDecl, unsigned Index)
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000371 : DeclOrIterator(SingleDecl), SingleDeclIndex(Index) { }
372
373 iterator(const DeclIndexPair *Iterator)
374 : DeclOrIterator(Iterator), SingleDeclIndex(0) { }
375
376 iterator &operator++() {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000377 if (DeclOrIterator.is<const NamedDecl *>()) {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000378 DeclOrIterator = (NamedDecl *)0;
379 SingleDeclIndex = 0;
380 return *this;
381 }
382
383 const DeclIndexPair *I = DeclOrIterator.get<const DeclIndexPair*>();
384 ++I;
385 DeclOrIterator = I;
386 return *this;
387 }
388
Chris Lattner66392d42010-09-04 18:12:20 +0000389 /*iterator operator++(int) {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000390 iterator tmp(*this);
391 ++(*this);
392 return tmp;
Chris Lattner66392d42010-09-04 18:12:20 +0000393 }*/
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000394
395 reference operator*() const {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000396 if (const NamedDecl *ND = DeclOrIterator.dyn_cast<const NamedDecl *>())
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000397 return reference(ND, SingleDeclIndex);
398
Douglas Gregord490f952009-12-06 21:27:58 +0000399 return *DeclOrIterator.get<const DeclIndexPair*>();
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000400 }
401
402 pointer operator->() const {
403 return pointer(**this);
404 }
405
406 friend bool operator==(const iterator &X, const iterator &Y) {
Douglas Gregord490f952009-12-06 21:27:58 +0000407 return X.DeclOrIterator.getOpaqueValue()
408 == Y.DeclOrIterator.getOpaqueValue() &&
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000409 X.SingleDeclIndex == Y.SingleDeclIndex;
410 }
411
412 friend bool operator!=(const iterator &X, const iterator &Y) {
Douglas Gregord490f952009-12-06 21:27:58 +0000413 return !(X == Y);
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000414 }
415};
416
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000417ResultBuilder::ShadowMapEntry::iterator
418ResultBuilder::ShadowMapEntry::begin() const {
419 if (DeclOrVector.isNull())
420 return iterator();
421
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000422 if (const NamedDecl *ND = DeclOrVector.dyn_cast<const NamedDecl *>())
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000423 return iterator(ND, SingleDeclIndex);
424
425 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->begin());
426}
427
428ResultBuilder::ShadowMapEntry::iterator
429ResultBuilder::ShadowMapEntry::end() const {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000430 if (DeclOrVector.is<const NamedDecl *>() || DeclOrVector.isNull())
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000431 return iterator();
432
433 return iterator(DeclOrVector.get<DeclIndexPairVector *>()->end());
434}
435
Douglas Gregor456c4a12009-09-21 20:12:40 +0000436/// \brief Compute the qualification required to get from the current context
437/// (\p CurContext) to the target context (\p TargetContext).
438///
439/// \param Context the AST context in which the qualification will be used.
440///
441/// \param CurContext the context where an entity is being named, which is
442/// typically based on the current scope.
443///
444/// \param TargetContext the context in which the named entity actually
445/// resides.
446///
447/// \returns a nested name specifier that refers into the target context, or
448/// NULL if no qualification is needed.
449static NestedNameSpecifier *
450getRequiredQualification(ASTContext &Context,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000451 const DeclContext *CurContext,
452 const DeclContext *TargetContext) {
453 SmallVector<const DeclContext *, 4> TargetParents;
Douglas Gregor456c4a12009-09-21 20:12:40 +0000454
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000455 for (const DeclContext *CommonAncestor = TargetContext;
Douglas Gregor456c4a12009-09-21 20:12:40 +0000456 CommonAncestor && !CommonAncestor->Encloses(CurContext);
457 CommonAncestor = CommonAncestor->getLookupParent()) {
458 if (CommonAncestor->isTransparentContext() ||
459 CommonAncestor->isFunctionOrMethod())
460 continue;
461
462 TargetParents.push_back(CommonAncestor);
463 }
464
465 NestedNameSpecifier *Result = 0;
466 while (!TargetParents.empty()) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000467 const DeclContext *Parent = TargetParents.back();
Douglas Gregor456c4a12009-09-21 20:12:40 +0000468 TargetParents.pop_back();
469
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000470 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Parent)) {
Douglas Gregorfb629412010-08-23 21:17:50 +0000471 if (!Namespace->getIdentifier())
472 continue;
473
Douglas Gregor456c4a12009-09-21 20:12:40 +0000474 Result = NestedNameSpecifier::Create(Context, Result, Namespace);
Douglas Gregorfb629412010-08-23 21:17:50 +0000475 }
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000476 else if (const TagDecl *TD = dyn_cast<TagDecl>(Parent))
Douglas Gregor456c4a12009-09-21 20:12:40 +0000477 Result = NestedNameSpecifier::Create(Context, Result,
478 false,
479 Context.getTypeDeclType(TD).getTypePtr());
Douglas Gregor0c8296d2009-11-07 00:00:49 +0000480 }
Douglas Gregor456c4a12009-09-21 20:12:40 +0000481 return Result;
482}
483
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000484bool ResultBuilder::isInterestingDecl(const NamedDecl *ND,
Douglas Gregor45bcd432010-01-14 03:21:49 +0000485 bool &AsNestedNameSpecifier) const {
486 AsNestedNameSpecifier = false;
487
Douglas Gregore495b7f2010-01-14 00:20:49 +0000488 ND = ND->getUnderlyingDecl();
489 unsigned IDNS = ND->getIdentifierNamespace();
Douglas Gregorf52cede2009-10-09 22:16:47 +0000490
491 // Skip unnamed entities.
Douglas Gregore495b7f2010-01-14 00:20:49 +0000492 if (!ND->getDeclName())
493 return false;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000494
495 // Friend declarations and declarations introduced due to friends are never
496 // added as results.
John McCall92b7f702010-03-11 07:50:04 +0000497 if (IDNS & (Decl::IDNS_OrdinaryFriend | Decl::IDNS_TagFriend))
Douglas Gregore495b7f2010-01-14 00:20:49 +0000498 return false;
499
Douglas Gregor76282942009-12-11 17:31:05 +0000500 // Class template (partial) specializations are never added as results.
Douglas Gregore495b7f2010-01-14 00:20:49 +0000501 if (isa<ClassTemplateSpecializationDecl>(ND) ||
502 isa<ClassTemplatePartialSpecializationDecl>(ND))
503 return false;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000504
Douglas Gregor76282942009-12-11 17:31:05 +0000505 // Using declarations themselves are never added as results.
Douglas Gregore495b7f2010-01-14 00:20:49 +0000506 if (isa<UsingDecl>(ND))
507 return false;
508
509 // Some declarations have reserved names that we don't want to ever show.
510 if (const IdentifierInfo *Id = ND->getIdentifier()) {
Douglas Gregor86d9a522009-09-21 16:56:56 +0000511 // __va_list_tag is a freak of nature. Find it and skip it.
512 if (Id->isStr("__va_list_tag") || Id->isStr("__builtin_va_list"))
Douglas Gregore495b7f2010-01-14 00:20:49 +0000513 return false;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000514
Douglas Gregorf52cede2009-10-09 22:16:47 +0000515 // Filter out names reserved for the implementation (C99 7.1.3,
Douglas Gregor797efb52010-07-14 17:44:04 +0000516 // C++ [lib.global.names]) if they come from a system header.
Daniel Dunbare013d682009-10-18 20:26:12 +0000517 //
518 // FIXME: Add predicate for this.
Douglas Gregorf52cede2009-10-09 22:16:47 +0000519 if (Id->getLength() >= 2) {
Daniel Dunbare013d682009-10-18 20:26:12 +0000520 const char *Name = Id->getNameStart();
Douglas Gregorf52cede2009-10-09 22:16:47 +0000521 if (Name[0] == '_' &&
Douglas Gregor797efb52010-07-14 17:44:04 +0000522 (Name[1] == '_' || (Name[1] >= 'A' && Name[1] <= 'Z')) &&
523 (ND->getLocation().isInvalid() ||
524 SemaRef.SourceMgr.isInSystemHeader(
525 SemaRef.SourceMgr.getSpellingLoc(ND->getLocation()))))
Douglas Gregore495b7f2010-01-14 00:20:49 +0000526 return false;
Douglas Gregorf52cede2009-10-09 22:16:47 +0000527 }
Douglas Gregor86d9a522009-09-21 16:56:56 +0000528 }
Douglas Gregor9b0ba872010-11-09 03:59:40 +0000529
Douglas Gregora5fb7c32010-08-16 23:05:20 +0000530 if (Filter == &ResultBuilder::IsNestedNameSpecifier ||
531 ((isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND)) &&
532 Filter != &ResultBuilder::IsNamespace &&
Douglas Gregor52779fb2010-09-23 23:01:17 +0000533 Filter != &ResultBuilder::IsNamespaceOrAlias &&
534 Filter != 0))
Douglas Gregora5fb7c32010-08-16 23:05:20 +0000535 AsNestedNameSpecifier = true;
536
Douglas Gregor86d9a522009-09-21 16:56:56 +0000537 // Filter out any unwanted results.
Douglas Gregor45bcd432010-01-14 03:21:49 +0000538 if (Filter && !(this->*Filter)(ND)) {
539 // Check whether it is interesting as a nested-name-specifier.
David Blaikie4e4d0842012-03-11 07:00:24 +0000540 if (AllowNestedNameSpecifiers && SemaRef.getLangOpts().CPlusPlus &&
Douglas Gregor45bcd432010-01-14 03:21:49 +0000541 IsNestedNameSpecifier(ND) &&
542 (Filter != &ResultBuilder::IsMember ||
543 (isa<CXXRecordDecl>(ND) &&
544 cast<CXXRecordDecl>(ND)->isInjectedClassName()))) {
545 AsNestedNameSpecifier = true;
546 return true;
547 }
548
Douglas Gregore495b7f2010-01-14 00:20:49 +0000549 return false;
Douglas Gregora5fb7c32010-08-16 23:05:20 +0000550 }
Douglas Gregore495b7f2010-01-14 00:20:49 +0000551 // ... then it must be interesting!
552 return true;
553}
554
Douglas Gregor6660d842010-01-14 00:41:07 +0000555bool ResultBuilder::CheckHiddenResult(Result &R, DeclContext *CurContext,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000556 const NamedDecl *Hiding) {
Douglas Gregor6660d842010-01-14 00:41:07 +0000557 // In C, there is no way to refer to a hidden name.
558 // FIXME: This isn't true; we can find a tag name hidden by an ordinary
559 // name if we introduce the tag type.
David Blaikie4e4d0842012-03-11 07:00:24 +0000560 if (!SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor6660d842010-01-14 00:41:07 +0000561 return true;
562
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000563 const DeclContext *HiddenCtx =
564 R.Declaration->getDeclContext()->getRedeclContext();
Douglas Gregor6660d842010-01-14 00:41:07 +0000565
566 // There is no way to qualify a name declared in a function or method.
567 if (HiddenCtx->isFunctionOrMethod())
568 return true;
569
Sebastian Redl7a126a42010-08-31 00:36:30 +0000570 if (HiddenCtx == Hiding->getDeclContext()->getRedeclContext())
Douglas Gregor6660d842010-01-14 00:41:07 +0000571 return true;
572
573 // We can refer to the result with the appropriate qualification. Do it.
574 R.Hidden = true;
575 R.QualifierIsInformative = false;
576
577 if (!R.Qualifier)
578 R.Qualifier = getRequiredQualification(SemaRef.Context,
579 CurContext,
580 R.Declaration->getDeclContext());
581 return false;
582}
583
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000584/// \brief A simplified classification of types used to determine whether two
585/// types are "similar enough" when adjusting priorities.
Douglas Gregor1827e102010-08-16 16:18:59 +0000586SimplifiedTypeClass clang::getSimplifiedTypeClass(CanQualType T) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000587 switch (T->getTypeClass()) {
588 case Type::Builtin:
589 switch (cast<BuiltinType>(T)->getKind()) {
590 case BuiltinType::Void:
591 return STC_Void;
592
593 case BuiltinType::NullPtr:
594 return STC_Pointer;
595
596 case BuiltinType::Overload:
597 case BuiltinType::Dependent:
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000598 return STC_Other;
599
600 case BuiltinType::ObjCId:
601 case BuiltinType::ObjCClass:
602 case BuiltinType::ObjCSel:
603 return STC_ObjectiveC;
604
605 default:
606 return STC_Arithmetic;
607 }
David Blaikie7530c032012-01-17 06:56:22 +0000608
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000609 case Type::Complex:
610 return STC_Arithmetic;
611
612 case Type::Pointer:
613 return STC_Pointer;
614
615 case Type::BlockPointer:
616 return STC_Block;
617
618 case Type::LValueReference:
619 case Type::RValueReference:
620 return getSimplifiedTypeClass(T->getAs<ReferenceType>()->getPointeeType());
621
622 case Type::ConstantArray:
623 case Type::IncompleteArray:
624 case Type::VariableArray:
625 case Type::DependentSizedArray:
626 return STC_Array;
627
628 case Type::DependentSizedExtVector:
629 case Type::Vector:
630 case Type::ExtVector:
631 return STC_Arithmetic;
632
633 case Type::FunctionProto:
634 case Type::FunctionNoProto:
635 return STC_Function;
636
637 case Type::Record:
638 return STC_Record;
639
640 case Type::Enum:
641 return STC_Arithmetic;
642
643 case Type::ObjCObject:
644 case Type::ObjCInterface:
645 case Type::ObjCObjectPointer:
646 return STC_ObjectiveC;
647
648 default:
649 return STC_Other;
650 }
651}
652
653/// \brief Get the type that a given expression will have if this declaration
654/// is used as an expression in its "typical" code-completion form.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000655QualType clang::getDeclUsageType(ASTContext &C, const NamedDecl *ND) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000656 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
657
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000658 if (const TypeDecl *Type = dyn_cast<TypeDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000659 return C.getTypeDeclType(Type);
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000660 if (const ObjCInterfaceDecl *Iface = dyn_cast<ObjCInterfaceDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000661 return C.getObjCInterfaceType(Iface);
662
663 QualType T;
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000664 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND))
Douglas Gregor5291c3c2010-07-13 08:18:22 +0000665 T = Function->getCallResultType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000666 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregor5291c3c2010-07-13 08:18:22 +0000667 T = Method->getSendResultType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000668 else if (const FunctionTemplateDecl *FunTmpl =
669 dyn_cast<FunctionTemplateDecl>(ND))
Douglas Gregor5291c3c2010-07-13 08:18:22 +0000670 T = FunTmpl->getTemplatedDecl()->getCallResultType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000671 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000672 T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext()));
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000673 else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000674 T = Property->getType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000675 else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000676 T = Value->getType();
677 else
678 return QualType();
Douglas Gregor3e64d562011-04-14 20:33:34 +0000679
680 // Dig through references, function pointers, and block pointers to
681 // get down to the likely type of an expression when the entity is
682 // used.
683 do {
684 if (const ReferenceType *Ref = T->getAs<ReferenceType>()) {
685 T = Ref->getPointeeType();
686 continue;
687 }
688
689 if (const PointerType *Pointer = T->getAs<PointerType>()) {
690 if (Pointer->getPointeeType()->isFunctionType()) {
691 T = Pointer->getPointeeType();
692 continue;
693 }
694
695 break;
696 }
697
698 if (const BlockPointerType *Block = T->getAs<BlockPointerType>()) {
699 T = Block->getPointeeType();
700 continue;
701 }
702
703 if (const FunctionType *Function = T->getAs<FunctionType>()) {
704 T = Function->getResultType();
705 continue;
706 }
707
708 break;
709 } while (true);
710
711 return T;
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000712}
713
Douglas Gregord1f09b42013-01-31 04:52:16 +0000714unsigned ResultBuilder::getBasePriority(const NamedDecl *ND) {
715 if (!ND)
716 return CCP_Unlikely;
717
718 // Context-based decisions.
719 const DeclContext *DC = ND->getDeclContext()->getRedeclContext();
720 if (DC->isFunctionOrMethod() || isa<BlockDecl>(DC)) {
721 // _cmd is relatively rare
722 if (const ImplicitParamDecl *ImplicitParam =
723 dyn_cast<ImplicitParamDecl>(ND))
724 if (ImplicitParam->getIdentifier() &&
725 ImplicitParam->getIdentifier()->isStr("_cmd"))
726 return CCP_ObjC_cmd;
727
728 return CCP_LocalDeclaration;
729 }
730 if (DC->isRecord() || isa<ObjCContainerDecl>(DC))
731 return CCP_MemberDeclaration;
732
733 // Content-based decisions.
734 if (isa<EnumConstantDecl>(ND))
735 return CCP_Constant;
736
Douglas Gregor626799b2013-01-31 05:03:46 +0000737 // Use CCP_Type for type declarations unless we're in a statement, Objective-C
738 // message receiver, or parenthesized expression context. There, it's as
739 // likely that the user will want to write a type as other declarations.
740 if ((isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND)) &&
741 !(CompletionContext.getKind() == CodeCompletionContext::CCC_Statement ||
742 CompletionContext.getKind()
743 == CodeCompletionContext::CCC_ObjCMessageReceiver ||
744 CompletionContext.getKind()
745 == CodeCompletionContext::CCC_ParenthesizedExpression))
Douglas Gregord1f09b42013-01-31 04:52:16 +0000746 return CCP_Type;
747
748 return CCP_Declaration;
749}
750
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000751void ResultBuilder::AdjustResultPriorityForDecl(Result &R) {
752 // If this is an Objective-C method declaration whose selector matches our
753 // preferred selector, give it a priority boost.
754 if (!PreferredSelector.isNull())
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000755 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(R.Declaration))
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000756 if (PreferredSelector == Method->getSelector())
757 R.Priority += CCD_SelectorMatch;
Douglas Gregor08f43cd2010-09-20 23:11:55 +0000758
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000759 // If we have a preferred type, adjust the priority for results with exactly-
760 // matching or nearly-matching types.
761 if (!PreferredType.isNull()) {
762 QualType T = getDeclUsageType(SemaRef.Context, R.Declaration);
763 if (!T.isNull()) {
764 CanQualType TC = SemaRef.Context.getCanonicalType(T);
765 // Check for exactly-matching types (modulo qualifiers).
766 if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC))
767 R.Priority /= CCF_ExactTypeMatch;
768 // Check for nearly-matching types, based on classification of each.
769 else if ((getSimplifiedTypeClass(PreferredType)
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000770 == getSimplifiedTypeClass(TC)) &&
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000771 !(PreferredType->isEnumeralType() && TC->isEnumeralType()))
772 R.Priority /= CCF_SimilarTypeMatch;
773 }
774 }
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000775}
776
Douglas Gregor6f942b22010-09-21 16:06:22 +0000777void ResultBuilder::MaybeAddConstructorResults(Result R) {
David Blaikie4e4d0842012-03-11 07:00:24 +0000778 if (!SemaRef.getLangOpts().CPlusPlus || !R.Declaration ||
Douglas Gregor6f942b22010-09-21 16:06:22 +0000779 !CompletionContext.wantConstructorResults())
780 return;
781
782 ASTContext &Context = SemaRef.Context;
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000783 const NamedDecl *D = R.Declaration;
784 const CXXRecordDecl *Record = 0;
785 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D))
Douglas Gregor6f942b22010-09-21 16:06:22 +0000786 Record = ClassTemplate->getTemplatedDecl();
787 else if ((Record = dyn_cast<CXXRecordDecl>(D))) {
788 // Skip specializations and partial specializations.
789 if (isa<ClassTemplateSpecializationDecl>(Record))
790 return;
791 } else {
792 // There are no constructors here.
793 return;
794 }
795
796 Record = Record->getDefinition();
797 if (!Record)
798 return;
799
800
801 QualType RecordTy = Context.getTypeDeclType(Record);
802 DeclarationName ConstructorName
803 = Context.DeclarationNames.getCXXConstructorName(
804 Context.getCanonicalType(RecordTy));
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000805 DeclContext::lookup_const_result Ctors = Record->lookup(ConstructorName);
806 for (DeclContext::lookup_const_iterator I = Ctors.begin(),
807 E = Ctors.end();
808 I != E; ++I) {
David Blaikie3bc93e32012-12-19 00:45:41 +0000809 R.Declaration = *I;
Douglas Gregor6f942b22010-09-21 16:06:22 +0000810 R.CursorKind = getCursorKindForDecl(R.Declaration);
811 Results.push_back(R);
812 }
813}
814
Douglas Gregore495b7f2010-01-14 00:20:49 +0000815void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) {
816 assert(!ShadowMaps.empty() && "Must enter into a results scope");
817
818 if (R.Kind != Result::RK_Declaration) {
819 // For non-declaration results, just add the result.
820 Results.push_back(R);
821 return;
822 }
823
824 // Look through using declarations.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000825 if (const UsingShadowDecl *Using =
826 dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregord1f09b42013-01-31 04:52:16 +0000827 MaybeAddResult(Result(Using->getTargetDecl(),
828 getBasePriority(Using->getTargetDecl()),
829 R.Qualifier),
830 CurContext);
Douglas Gregore495b7f2010-01-14 00:20:49 +0000831 return;
832 }
833
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000834 const Decl *CanonDecl = R.Declaration->getCanonicalDecl();
Douglas Gregore495b7f2010-01-14 00:20:49 +0000835 unsigned IDNS = CanonDecl->getIdentifierNamespace();
836
Douglas Gregor45bcd432010-01-14 03:21:49 +0000837 bool AsNestedNameSpecifier = false;
838 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregore495b7f2010-01-14 00:20:49 +0000839 return;
840
Douglas Gregor6f942b22010-09-21 16:06:22 +0000841 // C++ constructors are never found by name lookup.
842 if (isa<CXXConstructorDecl>(R.Declaration))
843 return;
844
Douglas Gregor86d9a522009-09-21 16:56:56 +0000845 ShadowMap &SMap = ShadowMaps.back();
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000846 ShadowMapEntry::iterator I, IEnd;
847 ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName());
848 if (NamePos != SMap.end()) {
849 I = NamePos->second.begin();
850 IEnd = NamePos->second.end();
851 }
852
853 for (; I != IEnd; ++I) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000854 const NamedDecl *ND = I->first;
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000855 unsigned Index = I->second;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000856 if (ND->getCanonicalDecl() == CanonDecl) {
857 // This is a redeclaration. Always pick the newer declaration.
Douglas Gregor86d9a522009-09-21 16:56:56 +0000858 Results[Index].Declaration = R.Declaration;
859
Douglas Gregor86d9a522009-09-21 16:56:56 +0000860 // We're done.
861 return;
862 }
863 }
864
865 // This is a new declaration in this scope. However, check whether this
866 // declaration name is hidden by a similarly-named declaration in an outer
867 // scope.
868 std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end();
869 --SMEnd;
870 for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000871 ShadowMapEntry::iterator I, IEnd;
872 ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName());
873 if (NamePos != SM->end()) {
874 I = NamePos->second.begin();
875 IEnd = NamePos->second.end();
876 }
877 for (; I != IEnd; ++I) {
Douglas Gregor86d9a522009-09-21 16:56:56 +0000878 // A tag declaration does not hide a non-tag declaration.
John McCall0d6b1642010-04-23 18:46:30 +0000879 if (I->first->hasTagIdentifierNamespace() &&
Douglas Gregor86d9a522009-09-21 16:56:56 +0000880 (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary |
881 Decl::IDNS_ObjCProtocol)))
882 continue;
883
884 // Protocols are in distinct namespaces from everything else.
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000885 if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol)
Douglas Gregor86d9a522009-09-21 16:56:56 +0000886 || (IDNS & Decl::IDNS_ObjCProtocol)) &&
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000887 I->first->getIdentifierNamespace() != IDNS)
Douglas Gregor86d9a522009-09-21 16:56:56 +0000888 continue;
889
890 // The newly-added result is hidden by an entry in the shadow map.
Douglas Gregor6660d842010-01-14 00:41:07 +0000891 if (CheckHiddenResult(R, CurContext, I->first))
Douglas Gregor86d9a522009-09-21 16:56:56 +0000892 return;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000893
894 break;
895 }
896 }
897
898 // Make sure that any given declaration only shows up in the result set once.
899 if (!AllDeclsFound.insert(CanonDecl))
900 return;
Douglas Gregor265f7492010-08-27 15:29:55 +0000901
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000902 // If the filter is for nested-name-specifiers, then this result starts a
903 // nested-name-specifier.
Douglas Gregor12e13132010-05-26 22:00:08 +0000904 if (AsNestedNameSpecifier) {
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000905 R.StartsNestedNameSpecifier = true;
Douglas Gregor12e13132010-05-26 22:00:08 +0000906 R.Priority = CCP_NestedNameSpecifier;
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000907 } else
908 AdjustResultPriorityForDecl(R);
Douglas Gregor265f7492010-08-27 15:29:55 +0000909
Douglas Gregor0563c262009-09-22 23:15:58 +0000910 // If this result is supposed to have an informative qualifier, add one.
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000911 if (R.QualifierIsInformative && !R.Qualifier &&
912 !R.StartsNestedNameSpecifier) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000913 const DeclContext *Ctx = R.Declaration->getDeclContext();
914 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregor0563c262009-09-22 23:15:58 +0000915 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000916 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregor0563c262009-09-22 23:15:58 +0000917 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
918 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
919 else
920 R.QualifierIsInformative = false;
921 }
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000922
Douglas Gregor86d9a522009-09-21 16:56:56 +0000923 // Insert this result into the set of results and into the current shadow
924 // map.
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000925 SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size());
Douglas Gregor86d9a522009-09-21 16:56:56 +0000926 Results.push_back(R);
Douglas Gregor6f942b22010-09-21 16:06:22 +0000927
928 if (!AsNestedNameSpecifier)
929 MaybeAddConstructorResults(R);
Douglas Gregor86d9a522009-09-21 16:56:56 +0000930}
931
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000932void ResultBuilder::AddResult(Result R, DeclContext *CurContext,
Douglas Gregor0cc84042010-01-14 15:47:35 +0000933 NamedDecl *Hiding, bool InBaseClass = false) {
Douglas Gregora4477812010-01-14 16:01:26 +0000934 if (R.Kind != Result::RK_Declaration) {
935 // For non-declaration results, just add the result.
936 Results.push_back(R);
937 return;
938 }
939
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000940 // Look through using declarations.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000941 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregord1f09b42013-01-31 04:52:16 +0000942 AddResult(Result(Using->getTargetDecl(),
943 getBasePriority(Using->getTargetDecl()),
944 R.Qualifier),
945 CurContext, Hiding);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000946 return;
947 }
948
Douglas Gregor45bcd432010-01-14 03:21:49 +0000949 bool AsNestedNameSpecifier = false;
950 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000951 return;
952
Douglas Gregor6f942b22010-09-21 16:06:22 +0000953 // C++ constructors are never found by name lookup.
954 if (isa<CXXConstructorDecl>(R.Declaration))
955 return;
956
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000957 if (Hiding && CheckHiddenResult(R, CurContext, Hiding))
958 return;
Nick Lewycky173a37a2012-04-03 21:44:08 +0000959
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000960 // Make sure that any given declaration only shows up in the result set once.
961 if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl()))
962 return;
963
964 // If the filter is for nested-name-specifiers, then this result starts a
965 // nested-name-specifier.
Douglas Gregor12e13132010-05-26 22:00:08 +0000966 if (AsNestedNameSpecifier) {
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000967 R.StartsNestedNameSpecifier = true;
Douglas Gregor12e13132010-05-26 22:00:08 +0000968 R.Priority = CCP_NestedNameSpecifier;
969 }
Douglas Gregor0cc84042010-01-14 15:47:35 +0000970 else if (Filter == &ResultBuilder::IsMember && !R.Qualifier && InBaseClass &&
971 isa<CXXRecordDecl>(R.Declaration->getDeclContext()
Sebastian Redl7a126a42010-08-31 00:36:30 +0000972 ->getRedeclContext()))
Douglas Gregor0cc84042010-01-14 15:47:35 +0000973 R.QualifierIsInformative = true;
974
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000975 // If this result is supposed to have an informative qualifier, add one.
976 if (R.QualifierIsInformative && !R.Qualifier &&
977 !R.StartsNestedNameSpecifier) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000978 const DeclContext *Ctx = R.Declaration->getDeclContext();
979 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000980 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace);
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000981 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000982 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false,
Douglas Gregor45bcd432010-01-14 03:21:49 +0000983 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000984 else
985 R.QualifierIsInformative = false;
986 }
987
Douglas Gregor12e13132010-05-26 22:00:08 +0000988 // Adjust the priority if this result comes from a base class.
989 if (InBaseClass)
990 R.Priority += CCD_InBaseClass;
991
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000992 AdjustResultPriorityForDecl(R);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +0000993
Douglas Gregor3cdee122010-08-26 16:36:48 +0000994 if (HasObjectTypeQualifiers)
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000995 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(R.Declaration))
Douglas Gregor3cdee122010-08-26 16:36:48 +0000996 if (Method->isInstance()) {
997 Qualifiers MethodQuals
998 = Qualifiers::fromCVRMask(Method->getTypeQualifiers());
999 if (ObjectTypeQualifiers == MethodQuals)
1000 R.Priority += CCD_ObjectQualifierMatch;
1001 else if (ObjectTypeQualifiers - MethodQuals) {
1002 // The method cannot be invoked, because doing so would drop
1003 // qualifiers.
1004 return;
1005 }
1006 }
1007
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001008 // Insert this result into the set of results.
1009 Results.push_back(R);
Douglas Gregor6f942b22010-09-21 16:06:22 +00001010
1011 if (!AsNestedNameSpecifier)
1012 MaybeAddConstructorResults(R);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001013}
1014
Douglas Gregora4477812010-01-14 16:01:26 +00001015void ResultBuilder::AddResult(Result R) {
1016 assert(R.Kind != Result::RK_Declaration &&
1017 "Declaration results need more context");
1018 Results.push_back(R);
1019}
1020
Douglas Gregor86d9a522009-09-21 16:56:56 +00001021/// \brief Enter into a new scope.
1022void ResultBuilder::EnterNewScope() {
1023 ShadowMaps.push_back(ShadowMap());
1024}
1025
1026/// \brief Exit from the current scope.
1027void ResultBuilder::ExitScope() {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +00001028 for (ShadowMap::iterator E = ShadowMaps.back().begin(),
1029 EEnd = ShadowMaps.back().end();
1030 E != EEnd;
1031 ++E)
1032 E->second.Destroy();
1033
Douglas Gregor86d9a522009-09-21 16:56:56 +00001034 ShadowMaps.pop_back();
1035}
1036
Douglas Gregor791215b2009-09-21 20:51:25 +00001037/// \brief Determines whether this given declaration will be found by
1038/// ordinary name lookup.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001039bool ResultBuilder::IsOrdinaryName(const NamedDecl *ND) const {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001040 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1041
Douglas Gregor791215b2009-09-21 20:51:25 +00001042 unsigned IDNS = Decl::IDNS_Ordinary;
David Blaikie4e4d0842012-03-11 07:00:24 +00001043 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9b30b262010-06-15 20:26:51 +00001044 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikie4e4d0842012-03-11 07:00:24 +00001045 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregorca45da02010-11-02 20:36:02 +00001046 if (isa<ObjCIvarDecl>(ND))
1047 return true;
Douglas Gregorca45da02010-11-02 20:36:02 +00001048 }
1049
Douglas Gregor791215b2009-09-21 20:51:25 +00001050 return ND->getIdentifierNamespace() & IDNS;
1051}
1052
Douglas Gregor01dfea02010-01-10 23:08:15 +00001053/// \brief Determines whether this given declaration will be found by
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001054/// ordinary name lookup but is not a type name.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001055bool ResultBuilder::IsOrdinaryNonTypeName(const NamedDecl *ND) const {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001056 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1057 if (isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND))
1058 return false;
1059
1060 unsigned IDNS = Decl::IDNS_Ordinary;
David Blaikie4e4d0842012-03-11 07:00:24 +00001061 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9b30b262010-06-15 20:26:51 +00001062 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikie4e4d0842012-03-11 07:00:24 +00001063 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregorca45da02010-11-02 20:36:02 +00001064 if (isa<ObjCIvarDecl>(ND))
1065 return true;
Douglas Gregorca45da02010-11-02 20:36:02 +00001066 }
1067
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001068 return ND->getIdentifierNamespace() & IDNS;
1069}
1070
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001071bool ResultBuilder::IsIntegralConstantValue(const NamedDecl *ND) const {
Douglas Gregorf9578432010-07-28 21:50:18 +00001072 if (!IsOrdinaryNonTypeName(ND))
1073 return 0;
1074
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001075 if (const ValueDecl *VD = dyn_cast<ValueDecl>(ND->getUnderlyingDecl()))
Douglas Gregorf9578432010-07-28 21:50:18 +00001076 if (VD->getType()->isIntegralOrEnumerationType())
1077 return true;
1078
1079 return false;
1080}
1081
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001082/// \brief Determines whether this given declaration will be found by
Douglas Gregor01dfea02010-01-10 23:08:15 +00001083/// ordinary name lookup.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001084bool ResultBuilder::IsOrdinaryNonValueName(const NamedDecl *ND) const {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001085 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1086
Douglas Gregor01dfea02010-01-10 23:08:15 +00001087 unsigned IDNS = Decl::IDNS_Ordinary;
David Blaikie4e4d0842012-03-11 07:00:24 +00001088 if (SemaRef.getLangOpts().CPlusPlus)
John McCall0d6b1642010-04-23 18:46:30 +00001089 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001090
1091 return (ND->getIdentifierNamespace() & IDNS) &&
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001092 !isa<ValueDecl>(ND) && !isa<FunctionTemplateDecl>(ND) &&
1093 !isa<ObjCPropertyDecl>(ND);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001094}
1095
Douglas Gregor86d9a522009-09-21 16:56:56 +00001096/// \brief Determines whether the given declaration is suitable as the
1097/// start of a C++ nested-name-specifier, e.g., a class or namespace.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001098bool ResultBuilder::IsNestedNameSpecifier(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001099 // Allow us to find class templates, too.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001100 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor86d9a522009-09-21 16:56:56 +00001101 ND = ClassTemplate->getTemplatedDecl();
1102
1103 return SemaRef.isAcceptableNestedNameSpecifier(ND);
1104}
1105
1106/// \brief Determines whether the given declaration is an enumeration.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001107bool ResultBuilder::IsEnum(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001108 return isa<EnumDecl>(ND);
1109}
1110
1111/// \brief Determines whether the given declaration is a class or struct.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001112bool ResultBuilder::IsClassOrStruct(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001113 // Allow us to find class templates, too.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001114 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor86d9a522009-09-21 16:56:56 +00001115 ND = ClassTemplate->getTemplatedDecl();
Joao Matos6666ed42012-08-31 18:45:21 +00001116
1117 // For purposes of this check, interfaces match too.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001118 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara465d41b2010-05-11 21:36:43 +00001119 return RD->getTagKind() == TTK_Class ||
Joao Matos6666ed42012-08-31 18:45:21 +00001120 RD->getTagKind() == TTK_Struct ||
1121 RD->getTagKind() == TTK_Interface;
Douglas Gregor86d9a522009-09-21 16:56:56 +00001122
1123 return false;
1124}
1125
1126/// \brief Determines whether the given declaration is a union.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001127bool ResultBuilder::IsUnion(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001128 // Allow us to find class templates, too.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001129 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor86d9a522009-09-21 16:56:56 +00001130 ND = ClassTemplate->getTemplatedDecl();
1131
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001132 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara465d41b2010-05-11 21:36:43 +00001133 return RD->getTagKind() == TTK_Union;
Douglas Gregor86d9a522009-09-21 16:56:56 +00001134
1135 return false;
1136}
1137
1138/// \brief Determines whether the given declaration is a namespace.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001139bool ResultBuilder::IsNamespace(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001140 return isa<NamespaceDecl>(ND);
1141}
1142
1143/// \brief Determines whether the given declaration is a namespace or
1144/// namespace alias.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001145bool ResultBuilder::IsNamespaceOrAlias(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001146 return isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND);
1147}
1148
Douglas Gregor76282942009-12-11 17:31:05 +00001149/// \brief Determines whether the given declaration is a type.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001150bool ResultBuilder::IsType(const NamedDecl *ND) const {
1151 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregord32b0222010-08-24 01:06:58 +00001152 ND = Using->getTargetDecl();
1153
1154 return isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND);
Douglas Gregor86d9a522009-09-21 16:56:56 +00001155}
1156
Douglas Gregor76282942009-12-11 17:31:05 +00001157/// \brief Determines which members of a class should be visible via
1158/// "." or "->". Only value declarations, nested name specifiers, and
1159/// using declarations thereof should show up.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001160bool ResultBuilder::IsMember(const NamedDecl *ND) const {
1161 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor76282942009-12-11 17:31:05 +00001162 ND = Using->getTargetDecl();
1163
Douglas Gregorce821962009-12-11 18:14:22 +00001164 return isa<ValueDecl>(ND) || isa<FunctionTemplateDecl>(ND) ||
1165 isa<ObjCPropertyDecl>(ND);
Douglas Gregoreb5758b2009-09-23 22:26:46 +00001166}
1167
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001168static bool isObjCReceiverType(ASTContext &C, QualType T) {
1169 T = C.getCanonicalType(T);
1170 switch (T->getTypeClass()) {
1171 case Type::ObjCObject:
1172 case Type::ObjCInterface:
1173 case Type::ObjCObjectPointer:
1174 return true;
1175
1176 case Type::Builtin:
1177 switch (cast<BuiltinType>(T)->getKind()) {
1178 case BuiltinType::ObjCId:
1179 case BuiltinType::ObjCClass:
1180 case BuiltinType::ObjCSel:
1181 return true;
1182
1183 default:
1184 break;
1185 }
1186 return false;
1187
1188 default:
1189 break;
1190 }
1191
David Blaikie4e4d0842012-03-11 07:00:24 +00001192 if (!C.getLangOpts().CPlusPlus)
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001193 return false;
1194
1195 // FIXME: We could perform more analysis here to determine whether a
1196 // particular class type has any conversions to Objective-C types. For now,
1197 // just accept all class types.
1198 return T->isDependentType() || T->isRecordType();
1199}
1200
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001201bool ResultBuilder::IsObjCMessageReceiver(const NamedDecl *ND) const {
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001202 QualType T = getDeclUsageType(SemaRef.Context, ND);
1203 if (T.isNull())
1204 return false;
1205
1206 T = SemaRef.Context.getBaseElementType(T);
1207 return isObjCReceiverType(SemaRef.Context, T);
1208}
1209
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001210bool ResultBuilder::IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const {
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001211 if (IsObjCMessageReceiver(ND))
1212 return true;
1213
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001214 const VarDecl *Var = dyn_cast<VarDecl>(ND);
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001215 if (!Var)
1216 return false;
1217
1218 return Var->hasLocalStorage() && !Var->hasAttr<BlocksAttr>();
1219}
1220
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001221bool ResultBuilder::IsObjCCollection(const NamedDecl *ND) const {
David Blaikie4e4d0842012-03-11 07:00:24 +00001222 if ((SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryName(ND)) ||
1223 (!SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryNonTypeName(ND)))
Douglas Gregorfb629412010-08-23 21:17:50 +00001224 return false;
1225
1226 QualType T = getDeclUsageType(SemaRef.Context, ND);
1227 if (T.isNull())
1228 return false;
1229
1230 T = SemaRef.Context.getBaseElementType(T);
1231 return T->isObjCObjectType() || T->isObjCObjectPointerType() ||
1232 T->isObjCIdType() ||
David Blaikie4e4d0842012-03-11 07:00:24 +00001233 (SemaRef.getLangOpts().CPlusPlus && T->isRecordType());
Douglas Gregorfb629412010-08-23 21:17:50 +00001234}
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001235
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001236bool ResultBuilder::IsImpossibleToSatisfy(const NamedDecl *ND) const {
Douglas Gregor52779fb2010-09-23 23:01:17 +00001237 return false;
1238}
1239
James Dennettde23c7e2012-06-17 05:33:25 +00001240/// \brief Determines whether the given declaration is an Objective-C
Douglas Gregor80f4f4c2010-01-14 16:08:12 +00001241/// instance variable.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001242bool ResultBuilder::IsObjCIvar(const NamedDecl *ND) const {
Douglas Gregor80f4f4c2010-01-14 16:08:12 +00001243 return isa<ObjCIvarDecl>(ND);
1244}
1245
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001246namespace {
1247 /// \brief Visible declaration consumer that adds a code-completion result
1248 /// for each visible declaration.
1249 class CodeCompletionDeclConsumer : public VisibleDeclConsumer {
1250 ResultBuilder &Results;
1251 DeclContext *CurContext;
1252
1253 public:
1254 CodeCompletionDeclConsumer(ResultBuilder &Results, DeclContext *CurContext)
1255 : Results(Results), CurContext(CurContext) { }
1256
Erik Verbruggend1205962011-10-06 07:27:49 +00001257 virtual void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
1258 bool InBaseClass) {
1259 bool Accessible = true;
Douglas Gregor17015ef2011-11-03 16:51:37 +00001260 if (Ctx)
1261 Accessible = Results.getSema().IsSimplyAccessible(ND, Ctx);
1262
Douglas Gregord1f09b42013-01-31 04:52:16 +00001263 ResultBuilder::Result Result(ND, Results.getBasePriority(ND), 0, false,
1264 Accessible);
Erik Verbruggend1205962011-10-06 07:27:49 +00001265 Results.AddResult(Result, CurContext, Hiding, InBaseClass);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001266 }
1267 };
1268}
1269
Douglas Gregor86d9a522009-09-21 16:56:56 +00001270/// \brief Add type specifiers for the current language as keyword results.
Douglas Gregorbca403c2010-01-13 23:51:12 +00001271static void AddTypeSpecifierResults(const LangOptions &LangOpts,
Douglas Gregor86d9a522009-09-21 16:56:56 +00001272 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00001273 typedef CodeCompletionResult Result;
Douglas Gregor12e13132010-05-26 22:00:08 +00001274 Results.AddResult(Result("short", CCP_Type));
1275 Results.AddResult(Result("long", CCP_Type));
1276 Results.AddResult(Result("signed", CCP_Type));
1277 Results.AddResult(Result("unsigned", CCP_Type));
1278 Results.AddResult(Result("void", CCP_Type));
1279 Results.AddResult(Result("char", CCP_Type));
1280 Results.AddResult(Result("int", CCP_Type));
1281 Results.AddResult(Result("float", CCP_Type));
1282 Results.AddResult(Result("double", CCP_Type));
1283 Results.AddResult(Result("enum", CCP_Type));
1284 Results.AddResult(Result("struct", CCP_Type));
1285 Results.AddResult(Result("union", CCP_Type));
1286 Results.AddResult(Result("const", CCP_Type));
1287 Results.AddResult(Result("volatile", CCP_Type));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001288
Douglas Gregor86d9a522009-09-21 16:56:56 +00001289 if (LangOpts.C99) {
1290 // C99-specific
Douglas Gregor12e13132010-05-26 22:00:08 +00001291 Results.AddResult(Result("_Complex", CCP_Type));
1292 Results.AddResult(Result("_Imaginary", CCP_Type));
1293 Results.AddResult(Result("_Bool", CCP_Type));
1294 Results.AddResult(Result("restrict", CCP_Type));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001295 }
1296
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001297 CodeCompletionBuilder Builder(Results.getAllocator(),
1298 Results.getCodeCompletionTUInfo());
Douglas Gregor86d9a522009-09-21 16:56:56 +00001299 if (LangOpts.CPlusPlus) {
1300 // C++-specific
Douglas Gregorb05496d2010-09-20 21:11:48 +00001301 Results.AddResult(Result("bool", CCP_Type +
1302 (LangOpts.ObjC1? CCD_bool_in_ObjC : 0)));
Douglas Gregor12e13132010-05-26 22:00:08 +00001303 Results.AddResult(Result("class", CCP_Type));
1304 Results.AddResult(Result("wchar_t", CCP_Type));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001305
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001306 // typename qualified-id
Douglas Gregor218937c2011-02-01 19:23:04 +00001307 Builder.AddTypedTextChunk("typename");
1308 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1309 Builder.AddPlaceholderChunk("qualifier");
1310 Builder.AddTextChunk("::");
1311 Builder.AddPlaceholderChunk("name");
1312 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001313
Richard Smith80ad52f2013-01-02 11:42:31 +00001314 if (LangOpts.CPlusPlus11) {
Douglas Gregor12e13132010-05-26 22:00:08 +00001315 Results.AddResult(Result("auto", CCP_Type));
1316 Results.AddResult(Result("char16_t", CCP_Type));
1317 Results.AddResult(Result("char32_t", CCP_Type));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001318
Douglas Gregor218937c2011-02-01 19:23:04 +00001319 Builder.AddTypedTextChunk("decltype");
1320 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1321 Builder.AddPlaceholderChunk("expression");
1322 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1323 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001324 }
1325 }
1326
1327 // GNU extensions
1328 if (LangOpts.GNUMode) {
1329 // FIXME: Enable when we actually support decimal floating point.
Douglas Gregora4477812010-01-14 16:01:26 +00001330 // Results.AddResult(Result("_Decimal32"));
1331 // Results.AddResult(Result("_Decimal64"));
1332 // Results.AddResult(Result("_Decimal128"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001333
Douglas Gregor218937c2011-02-01 19:23:04 +00001334 Builder.AddTypedTextChunk("typeof");
1335 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1336 Builder.AddPlaceholderChunk("expression");
1337 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001338
Douglas Gregor218937c2011-02-01 19:23:04 +00001339 Builder.AddTypedTextChunk("typeof");
1340 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1341 Builder.AddPlaceholderChunk("type");
1342 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1343 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001344 }
1345}
1346
John McCallf312b1e2010-08-26 23:41:50 +00001347static void AddStorageSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001348 const LangOptions &LangOpts,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001349 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00001350 typedef CodeCompletionResult Result;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001351 // Note: we don't suggest either "auto" or "register", because both
1352 // are pointless as storage specifiers. Elsewhere, we suggest "auto"
1353 // in C++0x as a type specifier.
Douglas Gregora4477812010-01-14 16:01:26 +00001354 Results.AddResult(Result("extern"));
1355 Results.AddResult(Result("static"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001356}
1357
John McCallf312b1e2010-08-26 23:41:50 +00001358static void AddFunctionSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001359 const LangOptions &LangOpts,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001360 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00001361 typedef CodeCompletionResult Result;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001362 switch (CCC) {
John McCallf312b1e2010-08-26 23:41:50 +00001363 case Sema::PCC_Class:
1364 case Sema::PCC_MemberTemplate:
Douglas Gregor01dfea02010-01-10 23:08:15 +00001365 if (LangOpts.CPlusPlus) {
Douglas Gregora4477812010-01-14 16:01:26 +00001366 Results.AddResult(Result("explicit"));
1367 Results.AddResult(Result("friend"));
1368 Results.AddResult(Result("mutable"));
1369 Results.AddResult(Result("virtual"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001370 }
1371 // Fall through
1372
John McCallf312b1e2010-08-26 23:41:50 +00001373 case Sema::PCC_ObjCInterface:
1374 case Sema::PCC_ObjCImplementation:
1375 case Sema::PCC_Namespace:
1376 case Sema::PCC_Template:
Douglas Gregor01dfea02010-01-10 23:08:15 +00001377 if (LangOpts.CPlusPlus || LangOpts.C99)
Douglas Gregora4477812010-01-14 16:01:26 +00001378 Results.AddResult(Result("inline"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001379 break;
1380
John McCallf312b1e2010-08-26 23:41:50 +00001381 case Sema::PCC_ObjCInstanceVariableList:
1382 case Sema::PCC_Expression:
1383 case Sema::PCC_Statement:
1384 case Sema::PCC_ForInit:
1385 case Sema::PCC_Condition:
1386 case Sema::PCC_RecoveryInFunction:
1387 case Sema::PCC_Type:
Douglas Gregor02688102010-09-14 23:59:36 +00001388 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00001389 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor01dfea02010-01-10 23:08:15 +00001390 break;
1391 }
1392}
1393
Douglas Gregorbca403c2010-01-13 23:51:12 +00001394static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt);
1395static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt);
1396static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregorc38c3e12010-01-13 21:54:15 +00001397 ResultBuilder &Results,
1398 bool NeedAt);
Douglas Gregorbca403c2010-01-13 23:51:12 +00001399static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001400 ResultBuilder &Results,
1401 bool NeedAt);
Douglas Gregorbca403c2010-01-13 23:51:12 +00001402static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001403 ResultBuilder &Results,
1404 bool NeedAt);
Douglas Gregorbca403c2010-01-13 23:51:12 +00001405static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001406
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001407static void AddTypedefResult(ResultBuilder &Results) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001408 CodeCompletionBuilder Builder(Results.getAllocator(),
1409 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00001410 Builder.AddTypedTextChunk("typedef");
1411 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1412 Builder.AddPlaceholderChunk("type");
1413 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1414 Builder.AddPlaceholderChunk("name");
1415 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001416}
1417
John McCallf312b1e2010-08-26 23:41:50 +00001418static bool WantTypesInContext(Sema::ParserCompletionContext CCC,
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001419 const LangOptions &LangOpts) {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001420 switch (CCC) {
John McCallf312b1e2010-08-26 23:41:50 +00001421 case Sema::PCC_Namespace:
1422 case Sema::PCC_Class:
1423 case Sema::PCC_ObjCInstanceVariableList:
1424 case Sema::PCC_Template:
1425 case Sema::PCC_MemberTemplate:
1426 case Sema::PCC_Statement:
1427 case Sema::PCC_RecoveryInFunction:
1428 case Sema::PCC_Type:
Douglas Gregor02688102010-09-14 23:59:36 +00001429 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00001430 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001431 return true;
1432
John McCallf312b1e2010-08-26 23:41:50 +00001433 case Sema::PCC_Expression:
1434 case Sema::PCC_Condition:
Douglas Gregor02688102010-09-14 23:59:36 +00001435 return LangOpts.CPlusPlus;
1436
1437 case Sema::PCC_ObjCInterface:
1438 case Sema::PCC_ObjCImplementation:
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001439 return false;
1440
John McCallf312b1e2010-08-26 23:41:50 +00001441 case Sema::PCC_ForInit:
Douglas Gregor02688102010-09-14 23:59:36 +00001442 return LangOpts.CPlusPlus || LangOpts.ObjC1 || LangOpts.C99;
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001443 }
David Blaikie7530c032012-01-17 06:56:22 +00001444
1445 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001446}
1447
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00001448static PrintingPolicy getCompletionPrintingPolicy(const ASTContext &Context,
1449 const Preprocessor &PP) {
1450 PrintingPolicy Policy = Sema::getPrintingPolicy(Context, PP);
Douglas Gregor8ca72082011-10-18 21:20:17 +00001451 Policy.AnonymousTagLocations = false;
1452 Policy.SuppressStrongLifetime = true;
Douglas Gregor25270b62011-11-03 00:16:13 +00001453 Policy.SuppressUnwrittenScope = true;
Douglas Gregor8ca72082011-10-18 21:20:17 +00001454 return Policy;
1455}
1456
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00001457/// \brief Retrieve a printing policy suitable for code completion.
1458static PrintingPolicy getCompletionPrintingPolicy(Sema &S) {
1459 return getCompletionPrintingPolicy(S.Context, S.PP);
1460}
1461
Douglas Gregor8ca72082011-10-18 21:20:17 +00001462/// \brief Retrieve the string representation of the given type as a string
1463/// that has the appropriate lifetime for code completion.
1464///
1465/// This routine provides a fast path where we provide constant strings for
1466/// common type names.
1467static const char *GetCompletionTypeString(QualType T,
1468 ASTContext &Context,
1469 const PrintingPolicy &Policy,
1470 CodeCompletionAllocator &Allocator) {
1471 if (!T.getLocalQualifiers()) {
1472 // Built-in type names are constant strings.
1473 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T))
Argyrios Kyrtzidis27a00972012-05-05 04:20:28 +00001474 return BT->getNameAsCString(Policy);
Douglas Gregor8ca72082011-10-18 21:20:17 +00001475
1476 // Anonymous tag types are constant strings.
1477 if (const TagType *TagT = dyn_cast<TagType>(T))
1478 if (TagDecl *Tag = TagT->getDecl())
John McCall83972f12013-03-09 00:54:27 +00001479 if (!Tag->hasNameForLinkage()) {
Douglas Gregor8ca72082011-10-18 21:20:17 +00001480 switch (Tag->getTagKind()) {
1481 case TTK_Struct: return "struct <anonymous>";
Joao Matos6666ed42012-08-31 18:45:21 +00001482 case TTK_Interface: return "__interface <anonymous>";
1483 case TTK_Class: return "class <anonymous>";
Douglas Gregor8ca72082011-10-18 21:20:17 +00001484 case TTK_Union: return "union <anonymous>";
1485 case TTK_Enum: return "enum <anonymous>";
1486 }
1487 }
1488 }
1489
1490 // Slow path: format the type as a string.
1491 std::string Result;
1492 T.getAsStringInternal(Result, Policy);
1493 return Allocator.CopyString(Result);
1494}
1495
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001496/// \brief Add a completion for "this", if we're in a member function.
1497static void addThisCompletion(Sema &S, ResultBuilder &Results) {
1498 QualType ThisTy = S.getCurrentThisType();
1499 if (ThisTy.isNull())
1500 return;
1501
1502 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001503 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001504 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
1505 Builder.AddResultTypeChunk(GetCompletionTypeString(ThisTy,
1506 S.Context,
1507 Policy,
1508 Allocator));
1509 Builder.AddTypedTextChunk("this");
Joao Matos6666ed42012-08-31 18:45:21 +00001510 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001511}
1512
Douglas Gregor01dfea02010-01-10 23:08:15 +00001513/// \brief Add language constructs that show up for "ordinary" names.
John McCallf312b1e2010-08-26 23:41:50 +00001514static void AddOrdinaryNameResults(Sema::ParserCompletionContext CCC,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001515 Scope *S,
1516 Sema &SemaRef,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001517 ResultBuilder &Results) {
Douglas Gregor8ca72082011-10-18 21:20:17 +00001518 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001519 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor8ca72082011-10-18 21:20:17 +00001520 PrintingPolicy Policy = getCompletionPrintingPolicy(SemaRef);
Douglas Gregor218937c2011-02-01 19:23:04 +00001521
John McCall0a2c5e22010-08-25 06:19:51 +00001522 typedef CodeCompletionResult Result;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001523 switch (CCC) {
John McCallf312b1e2010-08-26 23:41:50 +00001524 case Sema::PCC_Namespace:
David Blaikie4e4d0842012-03-11 07:00:24 +00001525 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001526 if (Results.includeCodePatterns()) {
1527 // namespace <identifier> { declarations }
Douglas Gregor218937c2011-02-01 19:23:04 +00001528 Builder.AddTypedTextChunk("namespace");
1529 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1530 Builder.AddPlaceholderChunk("identifier");
1531 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1532 Builder.AddPlaceholderChunk("declarations");
1533 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1534 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1535 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001536 }
1537
Douglas Gregor01dfea02010-01-10 23:08:15 +00001538 // namespace identifier = identifier ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001539 Builder.AddTypedTextChunk("namespace");
1540 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1541 Builder.AddPlaceholderChunk("name");
1542 Builder.AddChunk(CodeCompletionString::CK_Equal);
1543 Builder.AddPlaceholderChunk("namespace");
1544 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001545
1546 // Using directives
Douglas Gregor218937c2011-02-01 19:23:04 +00001547 Builder.AddTypedTextChunk("using");
1548 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1549 Builder.AddTextChunk("namespace");
1550 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1551 Builder.AddPlaceholderChunk("identifier");
1552 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001553
1554 // asm(string-literal)
Douglas Gregor218937c2011-02-01 19:23:04 +00001555 Builder.AddTypedTextChunk("asm");
1556 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1557 Builder.AddPlaceholderChunk("string-literal");
1558 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1559 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001560
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001561 if (Results.includeCodePatterns()) {
1562 // Explicit template instantiation
Douglas Gregor218937c2011-02-01 19:23:04 +00001563 Builder.AddTypedTextChunk("template");
1564 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1565 Builder.AddPlaceholderChunk("declaration");
1566 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001567 }
Douglas Gregor01dfea02010-01-10 23:08:15 +00001568 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001569
David Blaikie4e4d0842012-03-11 07:00:24 +00001570 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorbca403c2010-01-13 23:51:12 +00001571 AddObjCTopLevelResults(Results, true);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001572
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001573 AddTypedefResult(Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001574 // Fall through
1575
John McCallf312b1e2010-08-26 23:41:50 +00001576 case Sema::PCC_Class:
David Blaikie4e4d0842012-03-11 07:00:24 +00001577 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001578 // Using declaration
Douglas Gregor218937c2011-02-01 19:23:04 +00001579 Builder.AddTypedTextChunk("using");
1580 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1581 Builder.AddPlaceholderChunk("qualifier");
1582 Builder.AddTextChunk("::");
1583 Builder.AddPlaceholderChunk("name");
1584 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001585
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001586 // using typename qualifier::name (only in a dependent context)
Douglas Gregor01dfea02010-01-10 23:08:15 +00001587 if (SemaRef.CurContext->isDependentContext()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001588 Builder.AddTypedTextChunk("using");
1589 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1590 Builder.AddTextChunk("typename");
1591 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1592 Builder.AddPlaceholderChunk("qualifier");
1593 Builder.AddTextChunk("::");
1594 Builder.AddPlaceholderChunk("name");
1595 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001596 }
1597
John McCallf312b1e2010-08-26 23:41:50 +00001598 if (CCC == Sema::PCC_Class) {
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001599 AddTypedefResult(Results);
1600
Douglas Gregor01dfea02010-01-10 23:08:15 +00001601 // public:
Douglas Gregor218937c2011-02-01 19:23:04 +00001602 Builder.AddTypedTextChunk("public");
Douglas Gregor10ccf122012-04-10 17:56:28 +00001603 if (Results.includeCodePatterns())
1604 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregor218937c2011-02-01 19:23:04 +00001605 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001606
1607 // protected:
Douglas Gregor218937c2011-02-01 19:23:04 +00001608 Builder.AddTypedTextChunk("protected");
Douglas Gregor10ccf122012-04-10 17:56:28 +00001609 if (Results.includeCodePatterns())
1610 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregor218937c2011-02-01 19:23:04 +00001611 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001612
1613 // private:
Douglas Gregor218937c2011-02-01 19:23:04 +00001614 Builder.AddTypedTextChunk("private");
Douglas Gregor10ccf122012-04-10 17:56:28 +00001615 if (Results.includeCodePatterns())
1616 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregor218937c2011-02-01 19:23:04 +00001617 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001618 }
1619 }
1620 // Fall through
1621
John McCallf312b1e2010-08-26 23:41:50 +00001622 case Sema::PCC_Template:
1623 case Sema::PCC_MemberTemplate:
David Blaikie4e4d0842012-03-11 07:00:24 +00001624 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns()) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001625 // template < parameters >
Douglas Gregor218937c2011-02-01 19:23:04 +00001626 Builder.AddTypedTextChunk("template");
1627 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1628 Builder.AddPlaceholderChunk("parameters");
1629 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1630 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001631 }
1632
David Blaikie4e4d0842012-03-11 07:00:24 +00001633 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1634 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001635 break;
1636
John McCallf312b1e2010-08-26 23:41:50 +00001637 case Sema::PCC_ObjCInterface:
David Blaikie4e4d0842012-03-11 07:00:24 +00001638 AddObjCInterfaceResults(SemaRef.getLangOpts(), Results, true);
1639 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1640 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001641 break;
1642
John McCallf312b1e2010-08-26 23:41:50 +00001643 case Sema::PCC_ObjCImplementation:
David Blaikie4e4d0842012-03-11 07:00:24 +00001644 AddObjCImplementationResults(SemaRef.getLangOpts(), Results, true);
1645 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1646 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001647 break;
1648
John McCallf312b1e2010-08-26 23:41:50 +00001649 case Sema::PCC_ObjCInstanceVariableList:
David Blaikie4e4d0842012-03-11 07:00:24 +00001650 AddObjCVisibilityResults(SemaRef.getLangOpts(), Results, true);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00001651 break;
1652
John McCallf312b1e2010-08-26 23:41:50 +00001653 case Sema::PCC_RecoveryInFunction:
1654 case Sema::PCC_Statement: {
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001655 AddTypedefResult(Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001656
David Blaikie4e4d0842012-03-11 07:00:24 +00001657 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns() &&
1658 SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001659 Builder.AddTypedTextChunk("try");
1660 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1661 Builder.AddPlaceholderChunk("statements");
1662 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1663 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1664 Builder.AddTextChunk("catch");
1665 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1666 Builder.AddPlaceholderChunk("declaration");
1667 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1668 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1669 Builder.AddPlaceholderChunk("statements");
1670 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1671 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1672 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001673 }
David Blaikie4e4d0842012-03-11 07:00:24 +00001674 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorbca403c2010-01-13 23:51:12 +00001675 AddObjCStatementResults(Results, true);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001676
Douglas Gregord8e8a582010-05-25 21:41:55 +00001677 if (Results.includeCodePatterns()) {
1678 // if (condition) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00001679 Builder.AddTypedTextChunk("if");
1680 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00001681 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor218937c2011-02-01 19:23:04 +00001682 Builder.AddPlaceholderChunk("condition");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001683 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001684 Builder.AddPlaceholderChunk("expression");
1685 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1686 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1687 Builder.AddPlaceholderChunk("statements");
1688 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1689 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1690 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001691
Douglas Gregord8e8a582010-05-25 21:41:55 +00001692 // switch (condition) { }
Douglas Gregor218937c2011-02-01 19:23:04 +00001693 Builder.AddTypedTextChunk("switch");
1694 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00001695 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor218937c2011-02-01 19:23:04 +00001696 Builder.AddPlaceholderChunk("condition");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001697 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001698 Builder.AddPlaceholderChunk("expression");
1699 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1700 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1701 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1702 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1703 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001704 }
1705
Douglas Gregor01dfea02010-01-10 23:08:15 +00001706 // Switch-specific statements.
John McCall781472f2010-08-25 08:40:02 +00001707 if (!SemaRef.getCurFunction()->SwitchStack.empty()) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001708 // case expression:
Douglas Gregor218937c2011-02-01 19:23:04 +00001709 Builder.AddTypedTextChunk("case");
1710 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1711 Builder.AddPlaceholderChunk("expression");
1712 Builder.AddChunk(CodeCompletionString::CK_Colon);
1713 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001714
1715 // default:
Douglas Gregor218937c2011-02-01 19:23:04 +00001716 Builder.AddTypedTextChunk("default");
1717 Builder.AddChunk(CodeCompletionString::CK_Colon);
1718 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001719 }
1720
Douglas Gregord8e8a582010-05-25 21:41:55 +00001721 if (Results.includeCodePatterns()) {
1722 /// while (condition) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00001723 Builder.AddTypedTextChunk("while");
1724 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00001725 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor218937c2011-02-01 19:23:04 +00001726 Builder.AddPlaceholderChunk("condition");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001727 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001728 Builder.AddPlaceholderChunk("expression");
1729 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1730 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1731 Builder.AddPlaceholderChunk("statements");
1732 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1733 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1734 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001735
1736 // do { statements } while ( expression );
Douglas Gregor218937c2011-02-01 19:23:04 +00001737 Builder.AddTypedTextChunk("do");
1738 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1739 Builder.AddPlaceholderChunk("statements");
1740 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1741 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1742 Builder.AddTextChunk("while");
1743 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1744 Builder.AddPlaceholderChunk("expression");
1745 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1746 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001747
Douglas Gregord8e8a582010-05-25 21:41:55 +00001748 // for ( for-init-statement ; condition ; expression ) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00001749 Builder.AddTypedTextChunk("for");
1750 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00001751 if (SemaRef.getLangOpts().CPlusPlus || SemaRef.getLangOpts().C99)
Douglas Gregor218937c2011-02-01 19:23:04 +00001752 Builder.AddPlaceholderChunk("init-statement");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001753 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001754 Builder.AddPlaceholderChunk("init-expression");
1755 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1756 Builder.AddPlaceholderChunk("condition");
1757 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1758 Builder.AddPlaceholderChunk("inc-expression");
1759 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1760 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1761 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1762 Builder.AddPlaceholderChunk("statements");
1763 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1764 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1765 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001766 }
Douglas Gregor01dfea02010-01-10 23:08:15 +00001767
1768 if (S->getContinueParent()) {
1769 // continue ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001770 Builder.AddTypedTextChunk("continue");
1771 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001772 }
1773
1774 if (S->getBreakParent()) {
1775 // break ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001776 Builder.AddTypedTextChunk("break");
1777 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001778 }
1779
1780 // "return expression ;" or "return ;", depending on whether we
1781 // know the function is void or not.
1782 bool isVoid = false;
1783 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext))
1784 isVoid = Function->getResultType()->isVoidType();
1785 else if (ObjCMethodDecl *Method
1786 = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext))
1787 isVoid = Method->getResultType()->isVoidType();
Douglas Gregor9ea9bdb2010-03-01 23:15:13 +00001788 else if (SemaRef.getCurBlock() &&
1789 !SemaRef.getCurBlock()->ReturnType.isNull())
1790 isVoid = SemaRef.getCurBlock()->ReturnType->isVoidType();
Douglas Gregor218937c2011-02-01 19:23:04 +00001791 Builder.AddTypedTextChunk("return");
Douglas Gregor93298002010-02-18 04:06:48 +00001792 if (!isVoid) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001793 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1794 Builder.AddPlaceholderChunk("expression");
Douglas Gregor93298002010-02-18 04:06:48 +00001795 }
Douglas Gregor218937c2011-02-01 19:23:04 +00001796 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001797
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001798 // goto identifier ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001799 Builder.AddTypedTextChunk("goto");
1800 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1801 Builder.AddPlaceholderChunk("label");
1802 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001803
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001804 // Using directives
Douglas Gregor218937c2011-02-01 19:23:04 +00001805 Builder.AddTypedTextChunk("using");
1806 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1807 Builder.AddTextChunk("namespace");
1808 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1809 Builder.AddPlaceholderChunk("identifier");
1810 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001811 }
1812
1813 // Fall through (for statement expressions).
John McCallf312b1e2010-08-26 23:41:50 +00001814 case Sema::PCC_ForInit:
1815 case Sema::PCC_Condition:
David Blaikie4e4d0842012-03-11 07:00:24 +00001816 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001817 // Fall through: conditions and statements can have expressions.
1818
Douglas Gregor02688102010-09-14 23:59:36 +00001819 case Sema::PCC_ParenthesizedExpression:
David Blaikie4e4d0842012-03-11 07:00:24 +00001820 if (SemaRef.getLangOpts().ObjCAutoRefCount &&
John McCallf85e1932011-06-15 23:02:42 +00001821 CCC == Sema::PCC_ParenthesizedExpression) {
1822 // (__bridge <type>)<expression>
1823 Builder.AddTypedTextChunk("__bridge");
1824 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1825 Builder.AddPlaceholderChunk("type");
1826 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1827 Builder.AddPlaceholderChunk("expression");
1828 Results.AddResult(Result(Builder.TakeString()));
1829
1830 // (__bridge_transfer <Objective-C type>)<expression>
1831 Builder.AddTypedTextChunk("__bridge_transfer");
1832 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1833 Builder.AddPlaceholderChunk("Objective-C type");
1834 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1835 Builder.AddPlaceholderChunk("expression");
1836 Results.AddResult(Result(Builder.TakeString()));
1837
1838 // (__bridge_retained <CF type>)<expression>
1839 Builder.AddTypedTextChunk("__bridge_retained");
1840 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1841 Builder.AddPlaceholderChunk("CF type");
1842 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1843 Builder.AddPlaceholderChunk("expression");
1844 Results.AddResult(Result(Builder.TakeString()));
1845 }
1846 // Fall through
1847
John McCallf312b1e2010-08-26 23:41:50 +00001848 case Sema::PCC_Expression: {
David Blaikie4e4d0842012-03-11 07:00:24 +00001849 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001850 // 'this', if we're in a non-static member function.
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001851 addThisCompletion(SemaRef, Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001852
Douglas Gregor8ca72082011-10-18 21:20:17 +00001853 // true
1854 Builder.AddResultTypeChunk("bool");
1855 Builder.AddTypedTextChunk("true");
1856 Results.AddResult(Result(Builder.TakeString()));
1857
1858 // false
1859 Builder.AddResultTypeChunk("bool");
1860 Builder.AddTypedTextChunk("false");
1861 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001862
David Blaikie4e4d0842012-03-11 07:00:24 +00001863 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorec3310a2011-04-12 02:47:21 +00001864 // dynamic_cast < type-id > ( expression )
1865 Builder.AddTypedTextChunk("dynamic_cast");
1866 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1867 Builder.AddPlaceholderChunk("type");
1868 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1869 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1870 Builder.AddPlaceholderChunk("expression");
1871 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1872 Results.AddResult(Result(Builder.TakeString()));
1873 }
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001874
1875 // static_cast < type-id > ( expression )
Douglas Gregor218937c2011-02-01 19:23:04 +00001876 Builder.AddTypedTextChunk("static_cast");
1877 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1878 Builder.AddPlaceholderChunk("type");
1879 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1880 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1881 Builder.AddPlaceholderChunk("expression");
1882 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1883 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001884
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001885 // reinterpret_cast < type-id > ( expression )
Douglas Gregor218937c2011-02-01 19:23:04 +00001886 Builder.AddTypedTextChunk("reinterpret_cast");
1887 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1888 Builder.AddPlaceholderChunk("type");
1889 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1890 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1891 Builder.AddPlaceholderChunk("expression");
1892 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1893 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001894
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001895 // const_cast < type-id > ( expression )
Douglas Gregor218937c2011-02-01 19:23:04 +00001896 Builder.AddTypedTextChunk("const_cast");
1897 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1898 Builder.AddPlaceholderChunk("type");
1899 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1900 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1901 Builder.AddPlaceholderChunk("expression");
1902 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1903 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001904
David Blaikie4e4d0842012-03-11 07:00:24 +00001905 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorec3310a2011-04-12 02:47:21 +00001906 // typeid ( expression-or-type )
Douglas Gregor8ca72082011-10-18 21:20:17 +00001907 Builder.AddResultTypeChunk("std::type_info");
Douglas Gregorec3310a2011-04-12 02:47:21 +00001908 Builder.AddTypedTextChunk("typeid");
1909 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1910 Builder.AddPlaceholderChunk("expression-or-type");
1911 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1912 Results.AddResult(Result(Builder.TakeString()));
1913 }
1914
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001915 // new T ( ... )
Douglas Gregor218937c2011-02-01 19:23:04 +00001916 Builder.AddTypedTextChunk("new");
1917 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1918 Builder.AddPlaceholderChunk("type");
1919 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1920 Builder.AddPlaceholderChunk("expressions");
1921 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1922 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001923
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001924 // new T [ ] ( ... )
Douglas Gregor218937c2011-02-01 19:23:04 +00001925 Builder.AddTypedTextChunk("new");
1926 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1927 Builder.AddPlaceholderChunk("type");
1928 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1929 Builder.AddPlaceholderChunk("size");
1930 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1931 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1932 Builder.AddPlaceholderChunk("expressions");
1933 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1934 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001935
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001936 // delete expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00001937 Builder.AddResultTypeChunk("void");
Douglas Gregor218937c2011-02-01 19:23:04 +00001938 Builder.AddTypedTextChunk("delete");
1939 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1940 Builder.AddPlaceholderChunk("expression");
1941 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001942
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001943 // delete [] expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00001944 Builder.AddResultTypeChunk("void");
Douglas Gregor218937c2011-02-01 19:23:04 +00001945 Builder.AddTypedTextChunk("delete");
1946 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1947 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1948 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1949 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1950 Builder.AddPlaceholderChunk("expression");
1951 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001952
David Blaikie4e4d0842012-03-11 07:00:24 +00001953 if (SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorec3310a2011-04-12 02:47:21 +00001954 // throw expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00001955 Builder.AddResultTypeChunk("void");
Douglas Gregorec3310a2011-04-12 02:47:21 +00001956 Builder.AddTypedTextChunk("throw");
1957 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1958 Builder.AddPlaceholderChunk("expression");
1959 Results.AddResult(Result(Builder.TakeString()));
1960 }
Douglas Gregora50216c2011-10-18 16:29:03 +00001961
Douglas Gregor12e13132010-05-26 22:00:08 +00001962 // FIXME: Rethrow?
Douglas Gregora50216c2011-10-18 16:29:03 +00001963
Richard Smith80ad52f2013-01-02 11:42:31 +00001964 if (SemaRef.getLangOpts().CPlusPlus11) {
Douglas Gregora50216c2011-10-18 16:29:03 +00001965 // nullptr
Douglas Gregor8ca72082011-10-18 21:20:17 +00001966 Builder.AddResultTypeChunk("std::nullptr_t");
Douglas Gregora50216c2011-10-18 16:29:03 +00001967 Builder.AddTypedTextChunk("nullptr");
1968 Results.AddResult(Result(Builder.TakeString()));
1969
1970 // alignof
Douglas Gregor8ca72082011-10-18 21:20:17 +00001971 Builder.AddResultTypeChunk("size_t");
Douglas Gregora50216c2011-10-18 16:29:03 +00001972 Builder.AddTypedTextChunk("alignof");
1973 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1974 Builder.AddPlaceholderChunk("type");
1975 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1976 Results.AddResult(Result(Builder.TakeString()));
1977
1978 // noexcept
Douglas Gregor8ca72082011-10-18 21:20:17 +00001979 Builder.AddResultTypeChunk("bool");
Douglas Gregora50216c2011-10-18 16:29:03 +00001980 Builder.AddTypedTextChunk("noexcept");
1981 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1982 Builder.AddPlaceholderChunk("expression");
1983 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1984 Results.AddResult(Result(Builder.TakeString()));
1985
1986 // sizeof... expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00001987 Builder.AddResultTypeChunk("size_t");
Douglas Gregora50216c2011-10-18 16:29:03 +00001988 Builder.AddTypedTextChunk("sizeof...");
1989 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1990 Builder.AddPlaceholderChunk("parameter-pack");
1991 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1992 Results.AddResult(Result(Builder.TakeString()));
1993 }
Douglas Gregor01dfea02010-01-10 23:08:15 +00001994 }
1995
David Blaikie4e4d0842012-03-11 07:00:24 +00001996 if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001997 // Add "super", if we're in an Objective-C class with a superclass.
Ted Kremenek681e2562010-05-31 21:43:10 +00001998 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) {
1999 // The interface can be NULL.
2000 if (ObjCInterfaceDecl *ID = Method->getClassInterface())
Douglas Gregor8ca72082011-10-18 21:20:17 +00002001 if (ID->getSuperClass()) {
2002 std::string SuperType;
2003 SuperType = ID->getSuperClass()->getNameAsString();
2004 if (Method->isInstanceMethod())
2005 SuperType += " *";
2006
2007 Builder.AddResultTypeChunk(Allocator.CopyString(SuperType));
2008 Builder.AddTypedTextChunk("super");
2009 Results.AddResult(Result(Builder.TakeString()));
2010 }
Ted Kremenek681e2562010-05-31 21:43:10 +00002011 }
2012
Douglas Gregorbca403c2010-01-13 23:51:12 +00002013 AddObjCExpressionResults(Results, true);
Douglas Gregor01dfea02010-01-10 23:08:15 +00002014 }
2015
Jordan Rosef70a8862012-06-30 21:33:57 +00002016 if (SemaRef.getLangOpts().C11) {
2017 // _Alignof
2018 Builder.AddResultTypeChunk("size_t");
2019 if (SemaRef.getASTContext().Idents.get("alignof").hasMacroDefinition())
2020 Builder.AddTypedTextChunk("alignof");
2021 else
2022 Builder.AddTypedTextChunk("_Alignof");
2023 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2024 Builder.AddPlaceholderChunk("type");
2025 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2026 Results.AddResult(Result(Builder.TakeString()));
2027 }
2028
Douglas Gregorc8bddde2010-05-28 00:22:41 +00002029 // sizeof expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00002030 Builder.AddResultTypeChunk("size_t");
Douglas Gregor218937c2011-02-01 19:23:04 +00002031 Builder.AddTypedTextChunk("sizeof");
2032 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2033 Builder.AddPlaceholderChunk("expression-or-type");
2034 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2035 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00002036 break;
2037 }
Douglas Gregord32b0222010-08-24 01:06:58 +00002038
John McCallf312b1e2010-08-26 23:41:50 +00002039 case Sema::PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00002040 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregord32b0222010-08-24 01:06:58 +00002041 break;
Douglas Gregor01dfea02010-01-10 23:08:15 +00002042 }
2043
David Blaikie4e4d0842012-03-11 07:00:24 +00002044 if (WantTypesInContext(CCC, SemaRef.getLangOpts()))
2045 AddTypeSpecifierResults(SemaRef.getLangOpts(), Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00002046
David Blaikie4e4d0842012-03-11 07:00:24 +00002047 if (SemaRef.getLangOpts().CPlusPlus && CCC != Sema::PCC_Type)
Douglas Gregora4477812010-01-14 16:01:26 +00002048 Results.AddResult(Result("operator"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00002049}
2050
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002051/// \brief If the given declaration has an associated type, add it as a result
2052/// type chunk.
2053static void AddResultTypeChunk(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002054 const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002055 const NamedDecl *ND,
Douglas Gregor218937c2011-02-01 19:23:04 +00002056 CodeCompletionBuilder &Result) {
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002057 if (!ND)
2058 return;
Douglas Gregor6f942b22010-09-21 16:06:22 +00002059
2060 // Skip constructors and conversion functions, which have their return types
2061 // built into their names.
2062 if (isa<CXXConstructorDecl>(ND) || isa<CXXConversionDecl>(ND))
2063 return;
2064
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002065 // Determine the type of the declaration (if it has a type).
Douglas Gregor6f942b22010-09-21 16:06:22 +00002066 QualType T;
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002067 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND))
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002068 T = Function->getResultType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002069 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002070 T = Method->getResultType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002071 else if (const FunctionTemplateDecl *FunTmpl =
2072 dyn_cast<FunctionTemplateDecl>(ND))
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002073 T = FunTmpl->getTemplatedDecl()->getResultType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002074 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002075 T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext()));
2076 else if (isa<UnresolvedUsingValueDecl>(ND)) {
2077 /* Do nothing: ignore unresolved using declarations*/
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002078 } else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND)) {
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002079 T = Value->getType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002080 } else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002081 T = Property->getType();
2082
2083 if (T.isNull() || Context.hasSameType(T, Context.DependentTy))
2084 return;
2085
Douglas Gregor8987b232011-09-27 23:30:47 +00002086 Result.AddResultTypeChunk(GetCompletionTypeString(T, Context, Policy,
Douglas Gregora63f6de2011-02-01 21:15:40 +00002087 Result.getAllocator()));
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002088}
2089
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002090static void MaybeAddSentinel(ASTContext &Context,
2091 const NamedDecl *FunctionOrMethod,
Douglas Gregor218937c2011-02-01 19:23:04 +00002092 CodeCompletionBuilder &Result) {
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002093 if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>())
2094 if (Sentinel->getSentinel() == 0) {
David Blaikie4e4d0842012-03-11 07:00:24 +00002095 if (Context.getLangOpts().ObjC1 &&
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002096 Context.Idents.get("nil").hasMacroDefinition())
Douglas Gregor218937c2011-02-01 19:23:04 +00002097 Result.AddTextChunk(", nil");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002098 else if (Context.Idents.get("NULL").hasMacroDefinition())
Douglas Gregor218937c2011-02-01 19:23:04 +00002099 Result.AddTextChunk(", NULL");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002100 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002101 Result.AddTextChunk(", (void*)0");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002102 }
2103}
2104
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002105static std::string formatObjCParamQualifiers(unsigned ObjCQuals) {
2106 std::string Result;
2107 if (ObjCQuals & Decl::OBJC_TQ_In)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002108 Result += "in ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002109 else if (ObjCQuals & Decl::OBJC_TQ_Inout)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002110 Result += "inout ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002111 else if (ObjCQuals & Decl::OBJC_TQ_Out)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002112 Result += "out ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002113 if (ObjCQuals & Decl::OBJC_TQ_Bycopy)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002114 Result += "bycopy ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002115 else if (ObjCQuals & Decl::OBJC_TQ_Byref)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002116 Result += "byref ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002117 if (ObjCQuals & Decl::OBJC_TQ_Oneway)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002118 Result += "oneway ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002119 return Result;
2120}
2121
Douglas Gregor83482d12010-08-24 16:15:59 +00002122static std::string FormatFunctionParameter(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002123 const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002124 const ParmVarDecl *Param,
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002125 bool SuppressName = false,
2126 bool SuppressBlock = false) {
Douglas Gregor83482d12010-08-24 16:15:59 +00002127 bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext());
2128 if (Param->getType()->isDependentType() ||
2129 !Param->getType()->isBlockPointerType()) {
2130 // The argument for a dependent or non-block parameter is a placeholder
2131 // containing that parameter's type.
2132 std::string Result;
2133
Douglas Gregoraba48082010-08-29 19:47:46 +00002134 if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName)
Douglas Gregor83482d12010-08-24 16:15:59 +00002135 Result = Param->getIdentifier()->getName();
2136
John McCallf85e1932011-06-15 23:02:42 +00002137 Param->getType().getAsStringInternal(Result, Policy);
Douglas Gregor83482d12010-08-24 16:15:59 +00002138
2139 if (ObjCMethodParam) {
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002140 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2141 + Result + ")";
Douglas Gregoraba48082010-08-29 19:47:46 +00002142 if (Param->getIdentifier() && !SuppressName)
Douglas Gregor83482d12010-08-24 16:15:59 +00002143 Result += Param->getIdentifier()->getName();
2144 }
2145 return Result;
2146 }
2147
2148 // The argument for a block pointer parameter is a block literal with
2149 // the appropriate type.
David Blaikie39e6ab42013-02-18 22:06:02 +00002150 FunctionTypeLoc Block;
2151 FunctionProtoTypeLoc BlockProto;
Douglas Gregor83482d12010-08-24 16:15:59 +00002152 TypeLoc TL;
2153 if (TypeSourceInfo *TSInfo = Param->getTypeSourceInfo()) {
2154 TL = TSInfo->getTypeLoc().getUnqualifiedLoc();
2155 while (true) {
2156 // Look through typedefs.
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002157 if (!SuppressBlock) {
David Blaikie39e6ab42013-02-18 22:06:02 +00002158 if (TypedefTypeLoc TypedefTL = TL.getAs<TypedefTypeLoc>()) {
2159 if (TypeSourceInfo *InnerTSInfo =
2160 TypedefTL.getTypedefNameDecl()->getTypeSourceInfo()) {
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002161 TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc();
2162 continue;
2163 }
2164 }
2165
2166 // Look through qualified types
David Blaikie39e6ab42013-02-18 22:06:02 +00002167 if (QualifiedTypeLoc QualifiedTL = TL.getAs<QualifiedTypeLoc>()) {
2168 TL = QualifiedTL.getUnqualifiedLoc();
Douglas Gregor83482d12010-08-24 16:15:59 +00002169 continue;
2170 }
2171 }
2172
Douglas Gregor83482d12010-08-24 16:15:59 +00002173 // Try to get the function prototype behind the block pointer type,
2174 // then we're done.
David Blaikie39e6ab42013-02-18 22:06:02 +00002175 if (BlockPointerTypeLoc BlockPtr = TL.getAs<BlockPointerTypeLoc>()) {
2176 TL = BlockPtr.getPointeeLoc().IgnoreParens();
2177 Block = TL.getAs<FunctionTypeLoc>();
2178 BlockProto = TL.getAs<FunctionProtoTypeLoc>();
Douglas Gregor83482d12010-08-24 16:15:59 +00002179 }
2180 break;
2181 }
2182 }
2183
2184 if (!Block) {
2185 // We were unable to find a FunctionProtoTypeLoc with parameter names
2186 // for the block; just use the parameter type as a placeholder.
2187 std::string Result;
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002188 if (!ObjCMethodParam && Param->getIdentifier())
2189 Result = Param->getIdentifier()->getName();
2190
John McCallf85e1932011-06-15 23:02:42 +00002191 Param->getType().getUnqualifiedType().getAsStringInternal(Result, Policy);
Douglas Gregor83482d12010-08-24 16:15:59 +00002192
2193 if (ObjCMethodParam) {
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002194 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2195 + Result + ")";
Douglas Gregor83482d12010-08-24 16:15:59 +00002196 if (Param->getIdentifier())
2197 Result += Param->getIdentifier()->getName();
2198 }
2199
2200 return Result;
2201 }
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002202
Douglas Gregor83482d12010-08-24 16:15:59 +00002203 // We have the function prototype behind the block pointer type, as it was
2204 // written in the source.
Douglas Gregor38276252010-09-08 22:47:51 +00002205 std::string Result;
David Blaikie39e6ab42013-02-18 22:06:02 +00002206 QualType ResultType = Block.getTypePtr()->getResultType();
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002207 if (!ResultType->isVoidType() || SuppressBlock)
John McCallf85e1932011-06-15 23:02:42 +00002208 ResultType.getAsStringInternal(Result, Policy);
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002209
2210 // Format the parameter list.
2211 std::string Params;
David Blaikie39e6ab42013-02-18 22:06:02 +00002212 if (!BlockProto || Block.getNumArgs() == 0) {
2213 if (BlockProto && BlockProto.getTypePtr()->isVariadic())
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002214 Params = "(...)";
Douglas Gregorc2760bc2010-10-02 23:49:58 +00002215 else
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002216 Params = "(void)";
Douglas Gregor38276252010-09-08 22:47:51 +00002217 } else {
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002218 Params += "(";
David Blaikie39e6ab42013-02-18 22:06:02 +00002219 for (unsigned I = 0, N = Block.getNumArgs(); I != N; ++I) {
Douglas Gregor38276252010-09-08 22:47:51 +00002220 if (I)
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002221 Params += ", ";
David Blaikie39e6ab42013-02-18 22:06:02 +00002222 Params += FormatFunctionParameter(Context, Policy, Block.getArg(I),
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002223 /*SuppressName=*/false,
2224 /*SuppressBlock=*/true);
Douglas Gregor38276252010-09-08 22:47:51 +00002225
David Blaikie39e6ab42013-02-18 22:06:02 +00002226 if (I == N - 1 && BlockProto.getTypePtr()->isVariadic())
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002227 Params += ", ...";
Douglas Gregor38276252010-09-08 22:47:51 +00002228 }
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002229 Params += ")";
Douglas Gregore17794f2010-08-31 05:13:43 +00002230 }
Douglas Gregor38276252010-09-08 22:47:51 +00002231
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002232 if (SuppressBlock) {
2233 // Format as a parameter.
2234 Result = Result + " (^";
2235 if (Param->getIdentifier())
2236 Result += Param->getIdentifier()->getName();
2237 Result += ")";
2238 Result += Params;
2239 } else {
2240 // Format as a block literal argument.
2241 Result = '^' + Result;
2242 Result += Params;
2243
2244 if (Param->getIdentifier())
2245 Result += Param->getIdentifier()->getName();
2246 }
2247
Douglas Gregor83482d12010-08-24 16:15:59 +00002248 return Result;
2249}
2250
Douglas Gregor86d9a522009-09-21 16:56:56 +00002251/// \brief Add function parameter chunks to the given code completion string.
2252static void AddFunctionParameterChunks(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002253 const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002254 const FunctionDecl *Function,
Douglas Gregor218937c2011-02-01 19:23:04 +00002255 CodeCompletionBuilder &Result,
2256 unsigned Start = 0,
2257 bool InOptional = false) {
Douglas Gregor218937c2011-02-01 19:23:04 +00002258 bool FirstParameter = true;
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002259
Douglas Gregor218937c2011-02-01 19:23:04 +00002260 for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002261 const ParmVarDecl *Param = Function->getParamDecl(P);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002262
Douglas Gregor218937c2011-02-01 19:23:04 +00002263 if (Param->hasDefaultArg() && !InOptional) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002264 // When we see an optional default argument, put that argument and
2265 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002266 CodeCompletionBuilder Opt(Result.getAllocator(),
2267 Result.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00002268 if (!FirstParameter)
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002269 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor8987b232011-09-27 23:30:47 +00002270 AddFunctionParameterChunks(Context, Policy, Function, Opt, P, true);
Douglas Gregor218937c2011-02-01 19:23:04 +00002271 Result.AddOptionalChunk(Opt.TakeString());
2272 break;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002273 }
2274
Douglas Gregor218937c2011-02-01 19:23:04 +00002275 if (FirstParameter)
2276 FirstParameter = false;
2277 else
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002278 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor218937c2011-02-01 19:23:04 +00002279
2280 InOptional = false;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002281
2282 // Format the placeholder string.
Douglas Gregor8987b232011-09-27 23:30:47 +00002283 std::string PlaceholderStr = FormatFunctionParameter(Context, Policy,
2284 Param);
Douglas Gregor83482d12010-08-24 16:15:59 +00002285
Douglas Gregore17794f2010-08-31 05:13:43 +00002286 if (Function->isVariadic() && P == N - 1)
2287 PlaceholderStr += ", ...";
2288
Douglas Gregor86d9a522009-09-21 16:56:56 +00002289 // Add the placeholder string.
Douglas Gregordae68752011-02-01 22:57:45 +00002290 Result.AddPlaceholderChunk(
2291 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor86d9a522009-09-21 16:56:56 +00002292 }
Douglas Gregorb3d45252009-09-22 21:42:17 +00002293
2294 if (const FunctionProtoType *Proto
2295 = Function->getType()->getAs<FunctionProtoType>())
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002296 if (Proto->isVariadic()) {
Douglas Gregore17794f2010-08-31 05:13:43 +00002297 if (Proto->getNumArgs() == 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00002298 Result.AddPlaceholderChunk("...");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002299
Douglas Gregor218937c2011-02-01 19:23:04 +00002300 MaybeAddSentinel(Context, Function, Result);
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002301 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00002302}
2303
2304/// \brief Add template parameter chunks to the given code completion string.
2305static void AddTemplateParameterChunks(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002306 const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002307 const TemplateDecl *Template,
Douglas Gregor218937c2011-02-01 19:23:04 +00002308 CodeCompletionBuilder &Result,
2309 unsigned MaxParameters = 0,
2310 unsigned Start = 0,
2311 bool InDefaultArg = false) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002312 bool FirstParameter = true;
2313
2314 TemplateParameterList *Params = Template->getTemplateParameters();
2315 TemplateParameterList::iterator PEnd = Params->end();
2316 if (MaxParameters)
2317 PEnd = Params->begin() + MaxParameters;
Douglas Gregor218937c2011-02-01 19:23:04 +00002318 for (TemplateParameterList::iterator P = Params->begin() + Start;
2319 P != PEnd; ++P) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002320 bool HasDefaultArg = false;
2321 std::string PlaceholderStr;
2322 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) {
2323 if (TTP->wasDeclaredWithTypename())
2324 PlaceholderStr = "typename";
2325 else
2326 PlaceholderStr = "class";
2327
2328 if (TTP->getIdentifier()) {
2329 PlaceholderStr += ' ';
2330 PlaceholderStr += TTP->getIdentifier()->getName();
2331 }
2332
2333 HasDefaultArg = TTP->hasDefaultArgument();
2334 } else if (NonTypeTemplateParmDecl *NTTP
Douglas Gregor218937c2011-02-01 19:23:04 +00002335 = dyn_cast<NonTypeTemplateParmDecl>(*P)) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002336 if (NTTP->getIdentifier())
2337 PlaceholderStr = NTTP->getIdentifier()->getName();
John McCallf85e1932011-06-15 23:02:42 +00002338 NTTP->getType().getAsStringInternal(PlaceholderStr, Policy);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002339 HasDefaultArg = NTTP->hasDefaultArgument();
2340 } else {
2341 assert(isa<TemplateTemplateParmDecl>(*P));
2342 TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P);
2343
2344 // Since putting the template argument list into the placeholder would
2345 // be very, very long, we just use an abbreviation.
2346 PlaceholderStr = "template<...> class";
2347 if (TTP->getIdentifier()) {
2348 PlaceholderStr += ' ';
2349 PlaceholderStr += TTP->getIdentifier()->getName();
2350 }
2351
2352 HasDefaultArg = TTP->hasDefaultArgument();
2353 }
2354
Douglas Gregor218937c2011-02-01 19:23:04 +00002355 if (HasDefaultArg && !InDefaultArg) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002356 // When we see an optional default argument, put that argument and
2357 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002358 CodeCompletionBuilder Opt(Result.getAllocator(),
2359 Result.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00002360 if (!FirstParameter)
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002361 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor8987b232011-09-27 23:30:47 +00002362 AddTemplateParameterChunks(Context, Policy, Template, Opt, MaxParameters,
Douglas Gregor218937c2011-02-01 19:23:04 +00002363 P - Params->begin(), true);
2364 Result.AddOptionalChunk(Opt.TakeString());
2365 break;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002366 }
2367
Douglas Gregor218937c2011-02-01 19:23:04 +00002368 InDefaultArg = false;
2369
Douglas Gregor86d9a522009-09-21 16:56:56 +00002370 if (FirstParameter)
2371 FirstParameter = false;
2372 else
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002373 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002374
2375 // Add the placeholder string.
Douglas Gregordae68752011-02-01 22:57:45 +00002376 Result.AddPlaceholderChunk(
2377 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor86d9a522009-09-21 16:56:56 +00002378 }
2379}
2380
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002381/// \brief Add a qualifier to the given code-completion string, if the
2382/// provided nested-name-specifier is non-NULL.
Douglas Gregora61a8792009-12-11 18:44:16 +00002383static void
Douglas Gregor218937c2011-02-01 19:23:04 +00002384AddQualifierToCompletionString(CodeCompletionBuilder &Result,
Douglas Gregora61a8792009-12-11 18:44:16 +00002385 NestedNameSpecifier *Qualifier,
2386 bool QualifierIsInformative,
Douglas Gregor8987b232011-09-27 23:30:47 +00002387 ASTContext &Context,
2388 const PrintingPolicy &Policy) {
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002389 if (!Qualifier)
2390 return;
2391
2392 std::string PrintedNNS;
2393 {
2394 llvm::raw_string_ostream OS(PrintedNNS);
Douglas Gregor8987b232011-09-27 23:30:47 +00002395 Qualifier->print(OS, Policy);
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002396 }
Douglas Gregor0563c262009-09-22 23:15:58 +00002397 if (QualifierIsInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002398 Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregor0563c262009-09-22 23:15:58 +00002399 else
Douglas Gregordae68752011-02-01 22:57:45 +00002400 Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002401}
2402
Douglas Gregor218937c2011-02-01 19:23:04 +00002403static void
2404AddFunctionTypeQualsToCompletionString(CodeCompletionBuilder &Result,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002405 const FunctionDecl *Function) {
Douglas Gregora61a8792009-12-11 18:44:16 +00002406 const FunctionProtoType *Proto
2407 = Function->getType()->getAs<FunctionProtoType>();
2408 if (!Proto || !Proto->getTypeQuals())
2409 return;
2410
Douglas Gregora63f6de2011-02-01 21:15:40 +00002411 // FIXME: Add ref-qualifier!
2412
2413 // Handle single qualifiers without copying
2414 if (Proto->getTypeQuals() == Qualifiers::Const) {
2415 Result.AddInformativeChunk(" const");
2416 return;
2417 }
2418
2419 if (Proto->getTypeQuals() == Qualifiers::Volatile) {
2420 Result.AddInformativeChunk(" volatile");
2421 return;
2422 }
2423
2424 if (Proto->getTypeQuals() == Qualifiers::Restrict) {
2425 Result.AddInformativeChunk(" restrict");
2426 return;
2427 }
2428
2429 // Handle multiple qualifiers.
Douglas Gregora61a8792009-12-11 18:44:16 +00002430 std::string QualsStr;
David Blaikie4ef832f2012-08-10 00:55:35 +00002431 if (Proto->isConst())
Douglas Gregora61a8792009-12-11 18:44:16 +00002432 QualsStr += " const";
David Blaikie4ef832f2012-08-10 00:55:35 +00002433 if (Proto->isVolatile())
Douglas Gregora61a8792009-12-11 18:44:16 +00002434 QualsStr += " volatile";
David Blaikie4ef832f2012-08-10 00:55:35 +00002435 if (Proto->isRestrict())
Douglas Gregora61a8792009-12-11 18:44:16 +00002436 QualsStr += " restrict";
Douglas Gregordae68752011-02-01 22:57:45 +00002437 Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr));
Douglas Gregora61a8792009-12-11 18:44:16 +00002438}
2439
Douglas Gregor6f942b22010-09-21 16:06:22 +00002440/// \brief Add the name of the given declaration
Douglas Gregor8987b232011-09-27 23:30:47 +00002441static void AddTypedNameChunk(ASTContext &Context, const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002442 const NamedDecl *ND,
2443 CodeCompletionBuilder &Result) {
Douglas Gregor6f942b22010-09-21 16:06:22 +00002444 DeclarationName Name = ND->getDeclName();
2445 if (!Name)
2446 return;
2447
2448 switch (Name.getNameKind()) {
Douglas Gregora63f6de2011-02-01 21:15:40 +00002449 case DeclarationName::CXXOperatorName: {
2450 const char *OperatorName = 0;
2451 switch (Name.getCXXOverloadedOperator()) {
2452 case OO_None:
2453 case OO_Conditional:
2454 case NUM_OVERLOADED_OPERATORS:
2455 OperatorName = "operator";
2456 break;
2457
2458#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
2459 case OO_##Name: OperatorName = "operator" Spelling; break;
2460#define OVERLOADED_OPERATOR_MULTI(Name,Spelling,Unary,Binary,MemberOnly)
2461#include "clang/Basic/OperatorKinds.def"
2462
2463 case OO_New: OperatorName = "operator new"; break;
2464 case OO_Delete: OperatorName = "operator delete"; break;
2465 case OO_Array_New: OperatorName = "operator new[]"; break;
2466 case OO_Array_Delete: OperatorName = "operator delete[]"; break;
2467 case OO_Call: OperatorName = "operator()"; break;
2468 case OO_Subscript: OperatorName = "operator[]"; break;
2469 }
2470 Result.AddTypedTextChunk(OperatorName);
2471 break;
2472 }
2473
Douglas Gregor6f942b22010-09-21 16:06:22 +00002474 case DeclarationName::Identifier:
2475 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor6f942b22010-09-21 16:06:22 +00002476 case DeclarationName::CXXDestructorName:
2477 case DeclarationName::CXXLiteralOperatorName:
Douglas Gregordae68752011-02-01 22:57:45 +00002478 Result.AddTypedTextChunk(
2479 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002480 break;
2481
2482 case DeclarationName::CXXUsingDirective:
2483 case DeclarationName::ObjCZeroArgSelector:
2484 case DeclarationName::ObjCOneArgSelector:
2485 case DeclarationName::ObjCMultiArgSelector:
2486 break;
2487
2488 case DeclarationName::CXXConstructorName: {
2489 CXXRecordDecl *Record = 0;
2490 QualType Ty = Name.getCXXNameType();
2491 if (const RecordType *RecordTy = Ty->getAs<RecordType>())
2492 Record = cast<CXXRecordDecl>(RecordTy->getDecl());
2493 else if (const InjectedClassNameType *InjectedTy
2494 = Ty->getAs<InjectedClassNameType>())
2495 Record = InjectedTy->getDecl();
2496 else {
Douglas Gregordae68752011-02-01 22:57:45 +00002497 Result.AddTypedTextChunk(
2498 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002499 break;
2500 }
2501
Douglas Gregordae68752011-02-01 22:57:45 +00002502 Result.AddTypedTextChunk(
2503 Result.getAllocator().CopyString(Record->getNameAsString()));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002504 if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) {
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002505 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Douglas Gregor8987b232011-09-27 23:30:47 +00002506 AddTemplateParameterChunks(Context, Policy, Template, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002507 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor6f942b22010-09-21 16:06:22 +00002508 }
2509 break;
2510 }
2511 }
2512}
2513
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002514CodeCompletionString *CodeCompletionResult::CreateCodeCompletionString(Sema &S,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002515 CodeCompletionAllocator &Allocator,
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002516 CodeCompletionTUInfo &CCTUInfo,
2517 bool IncludeBriefComments) {
2518 return CreateCodeCompletionString(S.Context, S.PP, Allocator, CCTUInfo,
2519 IncludeBriefComments);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002520}
2521
Douglas Gregor86d9a522009-09-21 16:56:56 +00002522/// \brief If possible, create a new code completion string for the given
2523/// result.
2524///
2525/// \returns Either a new, heap-allocated code completion string describing
2526/// how to use this result, or NULL to indicate that the string or name of the
2527/// result is all that is needed.
2528CodeCompletionString *
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002529CodeCompletionResult::CreateCodeCompletionString(ASTContext &Ctx,
2530 Preprocessor &PP,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002531 CodeCompletionAllocator &Allocator,
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002532 CodeCompletionTUInfo &CCTUInfo,
2533 bool IncludeBriefComments) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002534 CodeCompletionBuilder Result(Allocator, CCTUInfo, Priority, Availability);
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002535
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002536 PrintingPolicy Policy = getCompletionPrintingPolicy(Ctx, PP);
Douglas Gregor218937c2011-02-01 19:23:04 +00002537 if (Kind == RK_Pattern) {
2538 Pattern->Priority = Priority;
2539 Pattern->Availability = Availability;
Douglas Gregorba103062012-03-27 23:34:16 +00002540
2541 if (Declaration) {
2542 Result.addParentContext(Declaration->getDeclContext());
Douglas Gregorba103062012-03-27 23:34:16 +00002543 Pattern->ParentName = Result.getParentName();
Fariborz Jahanianc02ddb22013-03-22 17:55:27 +00002544 // Provide code completion comment for self.GetterName where
2545 // GetterName is the getter method for a property with name
2546 // different from the property name (declared via a property
2547 // getter attribute.
2548 const NamedDecl *ND = Declaration;
2549 if (const ObjCMethodDecl *M = dyn_cast<ObjCMethodDecl>(ND))
2550 if (M->isPropertyAccessor())
2551 if (const ObjCPropertyDecl *PDecl = M->findPropertyDecl())
2552 if (PDecl->getGetterName() == M->getSelector() &&
Fariborz Jahanian16861372013-03-23 01:10:45 +00002553 PDecl->getIdentifier() != M->getIdentifier()) {
2554 if (const RawComment *RC =
2555 Ctx.getRawCommentForAnyRedecl(M)) {
Fariborz Jahanianc02ddb22013-03-22 17:55:27 +00002556 Result.addBriefComment(RC->getBriefText(Ctx));
2557 Pattern->BriefComment = Result.getBriefComment();
2558 }
Fariborz Jahanian16861372013-03-23 01:10:45 +00002559 else if (const RawComment *RC =
2560 Ctx.getRawCommentForAnyRedecl(PDecl)) {
2561 Result.addBriefComment(RC->getBriefText(Ctx));
2562 Pattern->BriefComment = Result.getBriefComment();
2563 }
2564 }
Douglas Gregorba103062012-03-27 23:34:16 +00002565 }
2566
Douglas Gregor218937c2011-02-01 19:23:04 +00002567 return Pattern;
2568 }
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002569
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002570 if (Kind == RK_Keyword) {
Douglas Gregor218937c2011-02-01 19:23:04 +00002571 Result.AddTypedTextChunk(Keyword);
2572 return Result.TakeString();
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002573 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00002574
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002575 if (Kind == RK_Macro) {
Argyrios Kyrtzidis9818a1d2013-02-20 00:54:57 +00002576 const MacroDirective *MD = PP.getMacroDirectiveHistory(Macro);
2577 assert(MD && "Not a macro?");
2578 const MacroInfo *MI = MD->getInfo();
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002579
Douglas Gregordae68752011-02-01 22:57:45 +00002580 Result.AddTypedTextChunk(
2581 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002582
2583 if (!MI->isFunctionLike())
Douglas Gregor218937c2011-02-01 19:23:04 +00002584 return Result.TakeString();
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002585
2586 // Format a function-like macro with placeholders for the arguments.
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002587 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregore4244702011-07-30 08:17:44 +00002588 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002589
2590 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
2591 if (MI->isC99Varargs()) {
2592 --AEnd;
2593
2594 if (A == AEnd) {
2595 Result.AddPlaceholderChunk("...");
2596 }
Douglas Gregore4244702011-07-30 08:17:44 +00002597 }
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002598
Douglas Gregore4244702011-07-30 08:17:44 +00002599 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002600 if (A != MI->arg_begin())
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002601 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002602
2603 if (MI->isVariadic() && (A+1) == AEnd) {
Dylan Noblesmithf7ccbad2012-02-05 02:13:05 +00002604 SmallString<32> Arg = (*A)->getName();
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002605 if (MI->isC99Varargs())
2606 Arg += ", ...";
2607 else
2608 Arg += "...";
Douglas Gregordae68752011-02-01 22:57:45 +00002609 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002610 break;
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002611 }
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002612
2613 // Non-variadic macros are simple.
2614 Result.AddPlaceholderChunk(
2615 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregore4244702011-07-30 08:17:44 +00002616 }
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002617 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor218937c2011-02-01 19:23:04 +00002618 return Result.TakeString();
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002619 }
2620
Douglas Gregord8e8a582010-05-25 21:41:55 +00002621 assert(Kind == RK_Declaration && "Missed a result kind?");
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002622 const NamedDecl *ND = Declaration;
Douglas Gregorba103062012-03-27 23:34:16 +00002623 Result.addParentContext(ND->getDeclContext());
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002624
2625 if (IncludeBriefComments) {
2626 // Add documentation comment, if it exists.
Dmitri Gribenkof50555e2012-08-11 00:51:43 +00002627 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(ND)) {
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002628 Result.addBriefComment(RC->getBriefText(Ctx));
Fariborz Jahanianb98f7af2013-02-28 17:47:14 +00002629 }
2630 else if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2631 if (OMD->isPropertyAccessor())
2632 if (const ObjCPropertyDecl *PDecl = OMD->findPropertyDecl())
2633 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(PDecl))
2634 Result.addBriefComment(RC->getBriefText(Ctx));
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002635 }
2636
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002637 if (StartsNestedNameSpecifier) {
Douglas Gregordae68752011-02-01 22:57:45 +00002638 Result.AddTypedTextChunk(
2639 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002640 Result.AddTextChunk("::");
2641 return Result.TakeString();
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002642 }
Erik Verbruggen6164ea12011-10-14 15:31:08 +00002643
2644 for (Decl::attr_iterator i = ND->attr_begin(); i != ND->attr_end(); ++i) {
2645 if (AnnotateAttr *Attr = dyn_cast_or_null<AnnotateAttr>(*i)) {
2646 Result.AddAnnotation(Result.getAllocator().CopyString(Attr->getAnnotation()));
2647 }
2648 }
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002649
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002650 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002651
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002652 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002653 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002654 Ctx, Policy);
2655 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002656 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002657 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002658 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregora61a8792009-12-11 18:44:16 +00002659 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregor218937c2011-02-01 19:23:04 +00002660 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002661 }
2662
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002663 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002664 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002665 Ctx, Policy);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002666 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002667 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor6f942b22010-09-21 16:06:22 +00002668
Douglas Gregor86d9a522009-09-21 16:56:56 +00002669 // Figure out which template parameters are deduced (or have default
2670 // arguments).
Benjamin Kramer013b3662012-01-30 16:17:39 +00002671 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002672 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002673 unsigned LastDeducibleArgument;
2674 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2675 --LastDeducibleArgument) {
2676 if (!Deduced[LastDeducibleArgument - 1]) {
2677 // C++0x: Figure out if the template argument has a default. If so,
2678 // the user doesn't need to type this argument.
2679 // FIXME: We need to abstract template parameters better!
2680 bool HasDefaultArg = false;
2681 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregor218937c2011-02-01 19:23:04 +00002682 LastDeducibleArgument - 1);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002683 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2684 HasDefaultArg = TTP->hasDefaultArgument();
2685 else if (NonTypeTemplateParmDecl *NTTP
2686 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2687 HasDefaultArg = NTTP->hasDefaultArgument();
2688 else {
2689 assert(isa<TemplateTemplateParmDecl>(Param));
2690 HasDefaultArg
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002691 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002692 }
2693
2694 if (!HasDefaultArg)
2695 break;
2696 }
2697 }
2698
2699 if (LastDeducibleArgument) {
2700 // Some of the function template arguments cannot be deduced from a
2701 // function call, so we introduce an explicit template argument list
2702 // containing all of the arguments up to the first deducible argument.
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002703 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002704 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor86d9a522009-09-21 16:56:56 +00002705 LastDeducibleArgument);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002706 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002707 }
2708
2709 // Add the function parameters
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002710 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002711 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002712 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregora61a8792009-12-11 18:44:16 +00002713 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregor218937c2011-02-01 19:23:04 +00002714 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002715 }
2716
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002717 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002718 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002719 Ctx, Policy);
Douglas Gregordae68752011-02-01 22:57:45 +00002720 Result.AddTypedTextChunk(
2721 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002722 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002723 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002724 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor218937c2011-02-01 19:23:04 +00002725 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002726 }
2727
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002728 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregor9630eb62009-11-17 16:44:22 +00002729 Selector Sel = Method->getSelector();
2730 if (Sel.isUnarySelector()) {
Douglas Gregordae68752011-02-01 22:57:45 +00002731 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00002732 Sel.getNameForSlot(0)));
Douglas Gregor218937c2011-02-01 19:23:04 +00002733 return Result.TakeString();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002734 }
2735
Douglas Gregor813d8342011-02-18 22:29:55 +00002736 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregord3c68542009-11-19 01:08:35 +00002737 SelName += ':';
2738 if (StartParameter == 0)
Douglas Gregordae68752011-02-01 22:57:45 +00002739 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregord3c68542009-11-19 01:08:35 +00002740 else {
Douglas Gregordae68752011-02-01 22:57:45 +00002741 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregord3c68542009-11-19 01:08:35 +00002742
2743 // If there is only one parameter, and we're past it, add an empty
2744 // typed-text chunk since there is nothing to type.
2745 if (Method->param_size() == 1)
Douglas Gregor218937c2011-02-01 19:23:04 +00002746 Result.AddTypedTextChunk("");
Douglas Gregord3c68542009-11-19 01:08:35 +00002747 }
Douglas Gregor9630eb62009-11-17 16:44:22 +00002748 unsigned Idx = 0;
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002749 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2750 PEnd = Method->param_end();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002751 P != PEnd; (void)++P, ++Idx) {
2752 if (Idx > 0) {
Douglas Gregord3c68542009-11-19 01:08:35 +00002753 std::string Keyword;
2754 if (Idx > StartParameter)
Douglas Gregor218937c2011-02-01 19:23:04 +00002755 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor9630eb62009-11-17 16:44:22 +00002756 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramera0651c52011-07-26 16:59:25 +00002757 Keyword += II->getName();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002758 Keyword += ":";
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002759 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002760 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002761 else
Douglas Gregordae68752011-02-01 22:57:45 +00002762 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor9630eb62009-11-17 16:44:22 +00002763 }
Douglas Gregord3c68542009-11-19 01:08:35 +00002764
2765 // If we're before the starting parameter, skip the placeholder.
2766 if (Idx < StartParameter)
2767 continue;
Douglas Gregor9630eb62009-11-17 16:44:22 +00002768
2769 std::string Arg;
Douglas Gregor83482d12010-08-24 16:15:59 +00002770
2771 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002772 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregor83482d12010-08-24 16:15:59 +00002773 else {
John McCallf85e1932011-06-15 23:02:42 +00002774 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002775 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2776 + Arg + ")";
Douglas Gregor83482d12010-08-24 16:15:59 +00002777 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregoraba48082010-08-29 19:47:46 +00002778 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramera0651c52011-07-26 16:59:25 +00002779 Arg += II->getName();
Douglas Gregor83482d12010-08-24 16:15:59 +00002780 }
2781
Douglas Gregore17794f2010-08-31 05:13:43 +00002782 if (Method->isVariadic() && (P + 1) == PEnd)
2783 Arg += ", ...";
2784
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002785 if (DeclaringEntity)
Douglas Gregordae68752011-02-01 22:57:45 +00002786 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002787 else if (AllParametersAreInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002788 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor4ad96852009-11-19 07:41:15 +00002789 else
Douglas Gregordae68752011-02-01 22:57:45 +00002790 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor9630eb62009-11-17 16:44:22 +00002791 }
2792
Douglas Gregor2a17af02009-12-23 00:21:46 +00002793 if (Method->isVariadic()) {
Douglas Gregore17794f2010-08-31 05:13:43 +00002794 if (Method->param_size() == 0) {
2795 if (DeclaringEntity)
Douglas Gregor218937c2011-02-01 19:23:04 +00002796 Result.AddTextChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002797 else if (AllParametersAreInformative)
Douglas Gregor218937c2011-02-01 19:23:04 +00002798 Result.AddInformativeChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002799 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002800 Result.AddPlaceholderChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002801 }
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002802
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002803 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor2a17af02009-12-23 00:21:46 +00002804 }
2805
Douglas Gregor218937c2011-02-01 19:23:04 +00002806 return Result.TakeString();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002807 }
2808
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002809 if (Qualifier)
Douglas Gregor0563c262009-09-22 23:15:58 +00002810 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002811 Ctx, Policy);
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002812
Douglas Gregordae68752011-02-01 22:57:45 +00002813 Result.AddTypedTextChunk(
2814 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002815 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002816}
2817
Douglas Gregor86d802e2009-09-23 00:34:09 +00002818CodeCompletionString *
2819CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2820 unsigned CurrentArg,
Douglas Gregor32be4a52010-10-11 21:37:58 +00002821 Sema &S,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002822 CodeCompletionAllocator &Allocator,
2823 CodeCompletionTUInfo &CCTUInfo) const {
Douglas Gregor8987b232011-09-27 23:30:47 +00002824 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCallf85e1932011-06-15 23:02:42 +00002825
Douglas Gregor218937c2011-02-01 19:23:04 +00002826 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002827 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002828 FunctionDecl *FDecl = getFunction();
Douglas Gregor8987b232011-09-27 23:30:47 +00002829 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002830 const FunctionProtoType *Proto
2831 = dyn_cast<FunctionProtoType>(getFunctionType());
2832 if (!FDecl && !Proto) {
2833 // Function without a prototype. Just give the return type and a
2834 // highlighted ellipsis.
2835 const FunctionType *FT = getFunctionType();
Douglas Gregora63f6de2011-02-01 21:15:40 +00002836 Result.AddTextChunk(GetCompletionTypeString(FT->getResultType(),
Douglas Gregor8987b232011-09-27 23:30:47 +00002837 S.Context, Policy,
Douglas Gregora63f6de2011-02-01 21:15:40 +00002838 Result.getAllocator()));
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002839 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2840 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2841 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor218937c2011-02-01 19:23:04 +00002842 return Result.TakeString();
Douglas Gregor86d802e2009-09-23 00:34:09 +00002843 }
2844
2845 if (FDecl)
Douglas Gregordae68752011-02-01 22:57:45 +00002846 Result.AddTextChunk(
2847 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002848 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002849 Result.AddTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00002850 Result.getAllocator().CopyString(
John McCallf85e1932011-06-15 23:02:42 +00002851 Proto->getResultType().getAsString(Policy)));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002852
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002853 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002854 unsigned NumParams = FDecl? FDecl->getNumParams() : Proto->getNumArgs();
2855 for (unsigned I = 0; I != NumParams; ++I) {
2856 if (I)
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002857 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002858
2859 std::string ArgString;
2860 QualType ArgType;
2861
2862 if (FDecl) {
2863 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2864 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2865 } else {
2866 ArgType = Proto->getArgType(I);
2867 }
2868
John McCallf85e1932011-06-15 23:02:42 +00002869 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002870
2871 if (I == CurrentArg)
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002872 Result.AddChunk(CodeCompletionString::CK_CurrentParameter,
2873 Result.getAllocator().CopyString(ArgString));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002874 else
Douglas Gregordae68752011-02-01 22:57:45 +00002875 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002876 }
2877
2878 if (Proto && Proto->isVariadic()) {
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002879 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002880 if (CurrentArg < NumParams)
Douglas Gregor218937c2011-02-01 19:23:04 +00002881 Result.AddTextChunk("...");
Douglas Gregor86d802e2009-09-23 00:34:09 +00002882 else
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002883 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
Douglas Gregor86d802e2009-09-23 00:34:09 +00002884 }
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002885 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002886
Douglas Gregor218937c2011-02-01 19:23:04 +00002887 return Result.TakeString();
Douglas Gregor86d802e2009-09-23 00:34:09 +00002888}
2889
Chris Lattner5f9e2722011-07-23 10:55:15 +00002890unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregorb05496d2010-09-20 21:11:48 +00002891 const LangOptions &LangOpts,
Douglas Gregor1827e102010-08-16 16:18:59 +00002892 bool PreferredTypeIsPointer) {
2893 unsigned Priority = CCP_Macro;
2894
Douglas Gregorb05496d2010-09-20 21:11:48 +00002895 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2896 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2897 MacroName.equals("Nil")) {
Douglas Gregor1827e102010-08-16 16:18:59 +00002898 Priority = CCP_Constant;
2899 if (PreferredTypeIsPointer)
2900 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregorb05496d2010-09-20 21:11:48 +00002901 }
2902 // Treat "YES", "NO", "true", and "false" as constants.
2903 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2904 MacroName.equals("true") || MacroName.equals("false"))
2905 Priority = CCP_Constant;
2906 // Treat "bool" as a type.
2907 else if (MacroName.equals("bool"))
2908 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2909
Douglas Gregor1827e102010-08-16 16:18:59 +00002910
2911 return Priority;
2912}
2913
Dmitri Gribenko06d8c602013-01-11 20:32:41 +00002914CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002915 if (!D)
2916 return CXCursor_UnexposedDecl;
2917
2918 switch (D->getKind()) {
2919 case Decl::Enum: return CXCursor_EnumDecl;
2920 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2921 case Decl::Field: return CXCursor_FieldDecl;
2922 case Decl::Function:
2923 return CXCursor_FunctionDecl;
2924 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2925 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002926 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregor375bb142011-12-27 22:43:10 +00002927
Argyrios Kyrtzidisc15707d2012-01-24 21:39:26 +00002928 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002929 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2930 case Decl::ObjCMethod:
2931 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2932 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2933 case Decl::CXXMethod: return CXCursor_CXXMethod;
2934 case Decl::CXXConstructor: return CXCursor_Constructor;
2935 case Decl::CXXDestructor: return CXCursor_Destructor;
2936 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2937 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidisc15707d2012-01-24 21:39:26 +00002938 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002939 case Decl::ParmVar: return CXCursor_ParmDecl;
2940 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smith162e1c12011-04-15 14:24:37 +00002941 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002942 case Decl::Var: return CXCursor_VarDecl;
2943 case Decl::Namespace: return CXCursor_Namespace;
2944 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2945 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2946 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2947 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2948 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2949 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis2dfdb942011-09-30 17:58:23 +00002950 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002951 case Decl::ClassTemplatePartialSpecialization:
2952 return CXCursor_ClassTemplatePartialSpecialization;
2953 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor8e5900c2012-04-30 23:41:16 +00002954 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002955
2956 case Decl::Using:
2957 case Decl::UnresolvedUsingValue:
2958 case Decl::UnresolvedUsingTypename:
2959 return CXCursor_UsingDeclaration;
2960
Douglas Gregor352697a2011-06-03 23:08:58 +00002961 case Decl::ObjCPropertyImpl:
2962 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2963 case ObjCPropertyImplDecl::Dynamic:
2964 return CXCursor_ObjCDynamicDecl;
2965
2966 case ObjCPropertyImplDecl::Synthesize:
2967 return CXCursor_ObjCSynthesizeDecl;
2968 }
Argyrios Kyrtzidis6a010122012-10-05 00:22:24 +00002969
2970 case Decl::Import:
2971 return CXCursor_ModuleImportDecl;
Douglas Gregor352697a2011-06-03 23:08:58 +00002972
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002973 default:
Dmitri Gribenko06d8c602013-01-11 20:32:41 +00002974 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002975 switch (TD->getTagKind()) {
Joao Matos6666ed42012-08-31 18:45:21 +00002976 case TTK_Interface: // fall through
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002977 case TTK_Struct: return CXCursor_StructDecl;
2978 case TTK_Class: return CXCursor_ClassDecl;
2979 case TTK_Union: return CXCursor_UnionDecl;
2980 case TTK_Enum: return CXCursor_EnumDecl;
2981 }
2982 }
2983 }
2984
2985 return CXCursor_UnexposedDecl;
2986}
2987
Douglas Gregor590c7d52010-07-08 20:55:51 +00002988static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor3644d972012-10-09 16:01:50 +00002989 bool IncludeUndefined,
Douglas Gregor590c7d52010-07-08 20:55:51 +00002990 bool TargetTypeIsPointer = false) {
John McCall0a2c5e22010-08-25 06:19:51 +00002991 typedef CodeCompletionResult Result;
Douglas Gregor590c7d52010-07-08 20:55:51 +00002992
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002993 Results.EnterNewScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002994
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002995 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2996 MEnd = PP.macro_end();
Douglas Gregor590c7d52010-07-08 20:55:51 +00002997 M != MEnd; ++M) {
Douglas Gregor3644d972012-10-09 16:01:50 +00002998 if (IncludeUndefined || M->first->hasMacroDefinition())
2999 Results.AddResult(Result(M->first,
Douglas Gregor1827e102010-08-16 16:18:59 +00003000 getMacroUsagePriority(M->first->getName(),
David Blaikie4e4d0842012-03-11 07:00:24 +00003001 PP.getLangOpts(),
Douglas Gregor1827e102010-08-16 16:18:59 +00003002 TargetTypeIsPointer)));
Douglas Gregor590c7d52010-07-08 20:55:51 +00003003 }
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00003004
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00003005 Results.ExitScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00003006
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00003007}
3008
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003009static void AddPrettyFunctionResults(const LangOptions &LangOpts,
3010 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00003011 typedef CodeCompletionResult Result;
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003012
3013 Results.EnterNewScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00003014
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003015 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
3016 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith80ad52f2013-01-02 11:42:31 +00003017 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003018 Results.AddResult(Result("__func__", CCP_Constant));
3019 Results.ExitScope();
3020}
3021
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003022static void HandleCodeCompleteResults(Sema *S,
3023 CodeCompleteConsumer *CodeCompleter,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003024 CodeCompletionContext Context,
John McCall0a2c5e22010-08-25 06:19:51 +00003025 CodeCompletionResult *Results,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003026 unsigned NumResults) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00003027 if (CodeCompleter)
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003028 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor86d9a522009-09-21 16:56:56 +00003029}
3030
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003031static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3032 Sema::ParserCompletionContext PCC) {
3033 switch (PCC) {
John McCallf312b1e2010-08-26 23:41:50 +00003034 case Sema::PCC_Namespace:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003035 return CodeCompletionContext::CCC_TopLevel;
3036
John McCallf312b1e2010-08-26 23:41:50 +00003037 case Sema::PCC_Class:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003038 return CodeCompletionContext::CCC_ClassStructUnion;
3039
John McCallf312b1e2010-08-26 23:41:50 +00003040 case Sema::PCC_ObjCInterface:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003041 return CodeCompletionContext::CCC_ObjCInterface;
3042
John McCallf312b1e2010-08-26 23:41:50 +00003043 case Sema::PCC_ObjCImplementation:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003044 return CodeCompletionContext::CCC_ObjCImplementation;
3045
John McCallf312b1e2010-08-26 23:41:50 +00003046 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003047 return CodeCompletionContext::CCC_ObjCIvarList;
3048
John McCallf312b1e2010-08-26 23:41:50 +00003049 case Sema::PCC_Template:
3050 case Sema::PCC_MemberTemplate:
Douglas Gregor52779fb2010-09-23 23:01:17 +00003051 if (S.CurContext->isFileContext())
3052 return CodeCompletionContext::CCC_TopLevel;
David Blaikie7530c032012-01-17 06:56:22 +00003053 if (S.CurContext->isRecord())
Douglas Gregor52779fb2010-09-23 23:01:17 +00003054 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie7530c032012-01-17 06:56:22 +00003055 return CodeCompletionContext::CCC_Other;
Douglas Gregor52779fb2010-09-23 23:01:17 +00003056
John McCallf312b1e2010-08-26 23:41:50 +00003057 case Sema::PCC_RecoveryInFunction:
Douglas Gregor52779fb2010-09-23 23:01:17 +00003058 return CodeCompletionContext::CCC_Recovery;
Douglas Gregora5450a02010-10-18 22:01:46 +00003059
John McCallf312b1e2010-08-26 23:41:50 +00003060 case Sema::PCC_ForInit:
David Blaikie4e4d0842012-03-11 07:00:24 +00003061 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3062 S.getLangOpts().ObjC1)
Douglas Gregora5450a02010-10-18 22:01:46 +00003063 return CodeCompletionContext::CCC_ParenthesizedExpression;
3064 else
3065 return CodeCompletionContext::CCC_Expression;
3066
3067 case Sema::PCC_Expression:
John McCallf312b1e2010-08-26 23:41:50 +00003068 case Sema::PCC_Condition:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003069 return CodeCompletionContext::CCC_Expression;
3070
John McCallf312b1e2010-08-26 23:41:50 +00003071 case Sema::PCC_Statement:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003072 return CodeCompletionContext::CCC_Statement;
Douglas Gregor72db1082010-08-24 01:11:00 +00003073
John McCallf312b1e2010-08-26 23:41:50 +00003074 case Sema::PCC_Type:
Douglas Gregor72db1082010-08-24 01:11:00 +00003075 return CodeCompletionContext::CCC_Type;
Douglas Gregor02688102010-09-14 23:59:36 +00003076
3077 case Sema::PCC_ParenthesizedExpression:
3078 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003079
3080 case Sema::PCC_LocalDeclarationSpecifiers:
3081 return CodeCompletionContext::CCC_Type;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003082 }
David Blaikie7530c032012-01-17 06:56:22 +00003083
3084 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003085}
3086
Douglas Gregorf6961522010-08-27 21:18:54 +00003087/// \brief If we're in a C++ virtual member function, add completion results
3088/// that invoke the functions we override, since it's common to invoke the
3089/// overridden function as well as adding new functionality.
3090///
3091/// \param S The semantic analysis object for which we are generating results.
3092///
3093/// \param InContext This context in which the nested-name-specifier preceding
3094/// the code-completion point
3095static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3096 ResultBuilder &Results) {
3097 // Look through blocks.
3098 DeclContext *CurContext = S.CurContext;
3099 while (isa<BlockDecl>(CurContext))
3100 CurContext = CurContext->getParent();
3101
3102
3103 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3104 if (!Method || !Method->isVirtual())
3105 return;
3106
3107 // We need to have names for all of the parameters, if we're going to
3108 // generate a forwarding call.
3109 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3110 PEnd = Method->param_end();
3111 P != PEnd;
3112 ++P) {
3113 if (!(*P)->getDeclName())
3114 return;
3115 }
3116
Douglas Gregor8987b232011-09-27 23:30:47 +00003117 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorf6961522010-08-27 21:18:54 +00003118 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3119 MEnd = Method->end_overridden_methods();
3120 M != MEnd; ++M) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003121 CodeCompletionBuilder Builder(Results.getAllocator(),
3122 Results.getCodeCompletionTUInfo());
Dmitri Gribenko68a932d2013-02-14 13:53:30 +00003123 const CXXMethodDecl *Overridden = *M;
Douglas Gregorf6961522010-08-27 21:18:54 +00003124 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3125 continue;
3126
3127 // If we need a nested-name-specifier, add one now.
3128 if (!InContext) {
3129 NestedNameSpecifier *NNS
3130 = getRequiredQualification(S.Context, CurContext,
3131 Overridden->getDeclContext());
3132 if (NNS) {
3133 std::string Str;
3134 llvm::raw_string_ostream OS(Str);
Douglas Gregor8987b232011-09-27 23:30:47 +00003135 NNS->print(OS, Policy);
Douglas Gregordae68752011-02-01 22:57:45 +00003136 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorf6961522010-08-27 21:18:54 +00003137 }
3138 } else if (!InContext->Equals(Overridden->getDeclContext()))
3139 continue;
3140
Douglas Gregordae68752011-02-01 22:57:45 +00003141 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00003142 Overridden->getNameAsString()));
3143 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf6961522010-08-27 21:18:54 +00003144 bool FirstParam = true;
3145 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3146 PEnd = Method->param_end();
3147 P != PEnd; ++P) {
3148 if (FirstParam)
3149 FirstParam = false;
3150 else
Douglas Gregor218937c2011-02-01 19:23:04 +00003151 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf6961522010-08-27 21:18:54 +00003152
Douglas Gregordae68752011-02-01 22:57:45 +00003153 Builder.AddPlaceholderChunk(Results.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00003154 (*P)->getIdentifier()->getName()));
Douglas Gregorf6961522010-08-27 21:18:54 +00003155 }
Douglas Gregor218937c2011-02-01 19:23:04 +00003156 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3157 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorf6961522010-08-27 21:18:54 +00003158 CCP_SuperCompletion,
Douglas Gregorba103062012-03-27 23:34:16 +00003159 CXCursor_CXXMethod,
3160 CXAvailability_Available,
3161 Overridden));
Douglas Gregorf6961522010-08-27 21:18:54 +00003162 Results.Ignore(Overridden);
3163 }
3164}
3165
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003166void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3167 ModuleIdPath Path) {
3168 typedef CodeCompletionResult Result;
3169 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003170 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003171 CodeCompletionContext::CCC_Other);
3172 Results.EnterNewScope();
3173
3174 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003175 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003176 typedef CodeCompletionResult Result;
3177 if (Path.empty()) {
3178 // Enumerate all top-level modules.
Dmitri Gribenkocfa88f82013-01-12 19:30:44 +00003179 SmallVector<Module *, 8> Modules;
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003180 PP.getHeaderSearchInfo().collectAllModules(Modules);
3181 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3182 Builder.AddTypedTextChunk(
3183 Builder.getAllocator().CopyString(Modules[I]->Name));
3184 Results.AddResult(Result(Builder.TakeString(),
3185 CCP_Declaration,
3186 CXCursor_NotImplemented,
3187 Modules[I]->isAvailable()
3188 ? CXAvailability_Available
3189 : CXAvailability_NotAvailable));
3190 }
3191 } else {
3192 // Load the named module.
3193 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3194 Module::AllVisible,
3195 /*IsInclusionDirective=*/false);
3196 // Enumerate submodules.
3197 if (Mod) {
3198 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3199 SubEnd = Mod->submodule_end();
3200 Sub != SubEnd; ++Sub) {
3201
3202 Builder.AddTypedTextChunk(
3203 Builder.getAllocator().CopyString((*Sub)->Name));
3204 Results.AddResult(Result(Builder.TakeString(),
3205 CCP_Declaration,
3206 CXCursor_NotImplemented,
3207 (*Sub)->isAvailable()
3208 ? CXAvailability_Available
3209 : CXAvailability_NotAvailable));
3210 }
3211 }
3212 }
3213 Results.ExitScope();
3214 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3215 Results.data(),Results.size());
3216}
3217
Douglas Gregor01dfea02010-01-10 23:08:15 +00003218void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003219 ParserCompletionContext CompletionContext) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003220 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003221 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003222 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorf6961522010-08-27 21:18:54 +00003223 Results.EnterNewScope();
Douglas Gregorcee9ff12010-09-20 22:39:41 +00003224
Douglas Gregor01dfea02010-01-10 23:08:15 +00003225 // Determine how to filter results, e.g., so that the names of
3226 // values (functions, enumerators, function templates, etc.) are
3227 // only allowed where we can have an expression.
3228 switch (CompletionContext) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003229 case PCC_Namespace:
3230 case PCC_Class:
3231 case PCC_ObjCInterface:
3232 case PCC_ObjCImplementation:
3233 case PCC_ObjCInstanceVariableList:
3234 case PCC_Template:
3235 case PCC_MemberTemplate:
Douglas Gregor72db1082010-08-24 01:11:00 +00003236 case PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003237 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor01dfea02010-01-10 23:08:15 +00003238 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3239 break;
3240
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003241 case PCC_Statement:
Douglas Gregor02688102010-09-14 23:59:36 +00003242 case PCC_ParenthesizedExpression:
Douglas Gregoreb0d0142010-08-24 23:58:17 +00003243 case PCC_Expression:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003244 case PCC_ForInit:
3245 case PCC_Condition:
David Blaikie4e4d0842012-03-11 07:00:24 +00003246 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor4710e5b2010-05-28 00:49:12 +00003247 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3248 else
3249 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorf6961522010-08-27 21:18:54 +00003250
David Blaikie4e4d0842012-03-11 07:00:24 +00003251 if (getLangOpts().CPlusPlus)
Douglas Gregorf6961522010-08-27 21:18:54 +00003252 MaybeAddOverrideCalls(*this, /*InContext=*/0, Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00003253 break;
Douglas Gregordc845342010-05-25 05:58:43 +00003254
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003255 case PCC_RecoveryInFunction:
Douglas Gregordc845342010-05-25 05:58:43 +00003256 // Unfiltered
3257 break;
Douglas Gregor01dfea02010-01-10 23:08:15 +00003258 }
3259
Douglas Gregor3cdee122010-08-26 16:36:48 +00003260 // If we are in a C++ non-static member function, check the qualifiers on
3261 // the member function to filter/prioritize the results list.
3262 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3263 if (CurMethod->isInstance())
3264 Results.setObjectTypeQualifiers(
3265 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3266
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00003267 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003268 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3269 CodeCompleter->includeGlobals());
Douglas Gregor2a7925c2009-12-07 09:54:55 +00003270
Douglas Gregorbca403c2010-01-13 23:51:12 +00003271 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor2a7925c2009-12-07 09:54:55 +00003272 Results.ExitScope();
3273
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003274 switch (CompletionContext) {
Douglas Gregor02688102010-09-14 23:59:36 +00003275 case PCC_ParenthesizedExpression:
Douglas Gregor72db1082010-08-24 01:11:00 +00003276 case PCC_Expression:
3277 case PCC_Statement:
3278 case PCC_RecoveryInFunction:
3279 if (S->getFnParent())
David Blaikie4e4d0842012-03-11 07:00:24 +00003280 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor72db1082010-08-24 01:11:00 +00003281 break;
3282
3283 case PCC_Namespace:
3284 case PCC_Class:
3285 case PCC_ObjCInterface:
3286 case PCC_ObjCImplementation:
3287 case PCC_ObjCInstanceVariableList:
3288 case PCC_Template:
3289 case PCC_MemberTemplate:
3290 case PCC_ForInit:
3291 case PCC_Condition:
3292 case PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003293 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor72db1082010-08-24 01:11:00 +00003294 break;
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003295 }
3296
Douglas Gregor0c8296d2009-11-07 00:00:49 +00003297 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00003298 AddMacroResults(PP, Results, false);
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003299
Douglas Gregorcee9ff12010-09-20 22:39:41 +00003300 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003301 Results.data(),Results.size());
Douglas Gregor791215b2009-09-21 20:51:25 +00003302}
3303
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003304static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3305 ParsedType Receiver,
3306 IdentifierInfo **SelIdents,
3307 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00003308 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003309 bool IsSuper,
3310 ResultBuilder &Results);
3311
3312void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3313 bool AllowNonIdentifiers,
3314 bool AllowNestedNameSpecifiers) {
John McCall0a2c5e22010-08-25 06:19:51 +00003315 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00003316 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003317 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003318 AllowNestedNameSpecifiers
3319 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3320 : CodeCompletionContext::CCC_Name);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003321 Results.EnterNewScope();
3322
3323 // Type qualifiers can come after names.
3324 Results.AddResult(Result("const"));
3325 Results.AddResult(Result("volatile"));
David Blaikie4e4d0842012-03-11 07:00:24 +00003326 if (getLangOpts().C99)
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003327 Results.AddResult(Result("restrict"));
3328
David Blaikie4e4d0842012-03-11 07:00:24 +00003329 if (getLangOpts().CPlusPlus) {
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003330 if (AllowNonIdentifiers) {
3331 Results.AddResult(Result("operator"));
3332 }
3333
3334 // Add nested-name-specifiers.
3335 if (AllowNestedNameSpecifiers) {
3336 Results.allowNestedNameSpecifiers();
Douglas Gregor52779fb2010-09-23 23:01:17 +00003337 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003338 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3339 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3340 CodeCompleter->includeGlobals());
Douglas Gregor52779fb2010-09-23 23:01:17 +00003341 Results.setFilter(0);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003342 }
3343 }
3344 Results.ExitScope();
3345
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003346 // If we're in a context where we might have an expression (rather than a
3347 // declaration), and what we've seen so far is an Objective-C type that could
3348 // be a receiver of a class message, this may be a class message send with
3349 // the initial opening bracket '[' missing. Add appropriate completions.
3350 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
3351 DS.getTypeSpecType() == DeclSpec::TST_typename &&
3352 DS.getStorageClassSpecAsWritten() == DeclSpec::SCS_unspecified &&
3353 !DS.isThreadSpecified() && !DS.isExternInLinkageSpec() &&
3354 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3355 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
3356 DS.getTypeQualifiers() == 0 &&
3357 S &&
3358 (S->getFlags() & Scope::DeclScope) != 0 &&
3359 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3360 Scope::FunctionPrototypeScope |
3361 Scope::AtCatchScope)) == 0) {
3362 ParsedType T = DS.getRepAsType();
3363 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Douglas Gregor70c5ac72010-09-20 23:34:21 +00003364 AddClassMessageCompletions(*this, S, T, 0, 0, false, false, Results);
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003365 }
3366
Douglas Gregor4497dd42010-08-24 04:59:56 +00003367 // Note that we intentionally suppress macro results here, since we do not
3368 // encourage using macros to produce the names of entities.
3369
Douglas Gregor52779fb2010-09-23 23:01:17 +00003370 HandleCodeCompleteResults(this, CodeCompleter,
3371 Results.getCompletionContext(),
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003372 Results.data(), Results.size());
3373}
3374
Douglas Gregorfb629412010-08-23 21:17:50 +00003375struct Sema::CodeCompleteExpressionData {
3376 CodeCompleteExpressionData(QualType PreferredType = QualType())
3377 : PreferredType(PreferredType), IntegralConstantExpression(false),
3378 ObjCCollection(false) { }
3379
3380 QualType PreferredType;
3381 bool IntegralConstantExpression;
3382 bool ObjCCollection;
Chris Lattner5f9e2722011-07-23 10:55:15 +00003383 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregorfb629412010-08-23 21:17:50 +00003384};
3385
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003386/// \brief Perform code-completion in an expression context when we know what
3387/// type we're looking for.
Douglas Gregorfb629412010-08-23 21:17:50 +00003388void Sema::CodeCompleteExpression(Scope *S,
3389 const CodeCompleteExpressionData &Data) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003390 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003391 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00003392 CodeCompletionContext::CCC_Expression);
Douglas Gregorfb629412010-08-23 21:17:50 +00003393 if (Data.ObjCCollection)
3394 Results.setFilter(&ResultBuilder::IsObjCCollection);
3395 else if (Data.IntegralConstantExpression)
Douglas Gregorf9578432010-07-28 21:50:18 +00003396 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikie4e4d0842012-03-11 07:00:24 +00003397 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003398 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3399 else
3400 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorfb629412010-08-23 21:17:50 +00003401
3402 if (!Data.PreferredType.isNull())
3403 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3404
3405 // Ignore any declarations that we were told that we don't care about.
3406 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3407 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003408
3409 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003410 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3411 CodeCompleter->includeGlobals());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003412
3413 Results.EnterNewScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003414 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003415 Results.ExitScope();
3416
Douglas Gregor590c7d52010-07-08 20:55:51 +00003417 bool PreferredTypeIsPointer = false;
Douglas Gregorfb629412010-08-23 21:17:50 +00003418 if (!Data.PreferredType.isNull())
3419 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3420 || Data.PreferredType->isMemberPointerType()
3421 || Data.PreferredType->isBlockPointerType();
Douglas Gregor590c7d52010-07-08 20:55:51 +00003422
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003423 if (S->getFnParent() &&
3424 !Data.ObjCCollection &&
3425 !Data.IntegralConstantExpression)
David Blaikie4e4d0842012-03-11 07:00:24 +00003426 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003427
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003428 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00003429 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003430 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorfb629412010-08-23 21:17:50 +00003431 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3432 Data.PreferredType),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003433 Results.data(),Results.size());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003434}
3435
Douglas Gregorac5fd842010-09-18 01:28:11 +00003436void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3437 if (E.isInvalid())
3438 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikie4e4d0842012-03-11 07:00:24 +00003439 else if (getLangOpts().ObjC1)
Douglas Gregorac5fd842010-09-18 01:28:11 +00003440 CodeCompleteObjCInstanceMessage(S, E.take(), 0, 0, false);
Douglas Gregor78edf512010-09-15 16:23:04 +00003441}
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003442
Douglas Gregor73449212010-12-09 23:01:55 +00003443/// \brief The set of properties that have already been added, referenced by
3444/// property name.
3445typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3446
Douglas Gregorb92a4082012-06-12 13:44:08 +00003447/// \brief Retrieve the container definition, if any?
3448static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3449 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3450 if (Interface->hasDefinition())
3451 return Interface->getDefinition();
3452
3453 return Interface;
3454 }
3455
3456 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3457 if (Protocol->hasDefinition())
3458 return Protocol->getDefinition();
3459
3460 return Protocol;
3461 }
3462 return Container;
3463}
3464
3465static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor322328b2009-11-18 22:32:06 +00003466 bool AllowCategories,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003467 bool AllowNullaryMethods,
Douglas Gregor95ac6552009-11-18 01:29:26 +00003468 DeclContext *CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003469 AddedPropertiesSet &AddedProperties,
Douglas Gregor95ac6552009-11-18 01:29:26 +00003470 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00003471 typedef CodeCompletionResult Result;
Douglas Gregor95ac6552009-11-18 01:29:26 +00003472
Douglas Gregorb92a4082012-06-12 13:44:08 +00003473 // Retrieve the definition.
3474 Container = getContainerDef(Container);
3475
Douglas Gregor95ac6552009-11-18 01:29:26 +00003476 // Add properties in this container.
3477 for (ObjCContainerDecl::prop_iterator P = Container->prop_begin(),
3478 PEnd = Container->prop_end();
3479 P != PEnd;
Douglas Gregor73449212010-12-09 23:01:55 +00003480 ++P) {
3481 if (AddedProperties.insert(P->getIdentifier()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00003482 Results.MaybeAddResult(Result(*P, Results.getBasePriority(*P), 0),
3483 CurContext);
Douglas Gregor73449212010-12-09 23:01:55 +00003484 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003485
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003486 // Add nullary methods
3487 if (AllowNullaryMethods) {
3488 ASTContext &Context = Container->getASTContext();
Douglas Gregor8987b232011-09-27 23:30:47 +00003489 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003490 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
3491 MEnd = Container->meth_end();
3492 M != MEnd; ++M) {
3493 if (M->getSelector().isUnarySelector())
3494 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3495 if (AddedProperties.insert(Name)) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003496 CodeCompletionBuilder Builder(Results.getAllocator(),
3497 Results.getCodeCompletionTUInfo());
David Blaikie581deb32012-06-06 20:45:41 +00003498 AddResultTypeChunk(Context, Policy, *M, Builder);
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003499 Builder.AddTypedTextChunk(
3500 Results.getAllocator().CopyString(Name->getName()));
3501
David Blaikie581deb32012-06-06 20:45:41 +00003502 Results.MaybeAddResult(Result(Builder.TakeString(), *M,
Douglas Gregorba103062012-03-27 23:34:16 +00003503 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003504 CurContext);
3505 }
3506 }
3507 }
3508
3509
Douglas Gregor95ac6552009-11-18 01:29:26 +00003510 // Add properties in referenced protocols.
3511 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3512 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
3513 PEnd = Protocol->protocol_end();
3514 P != PEnd; ++P)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003515 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3516 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003517 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor322328b2009-11-18 22:32:06 +00003518 if (AllowCategories) {
3519 // Look through categories.
Douglas Gregord3297242013-01-16 23:00:23 +00003520 for (ObjCInterfaceDecl::known_categories_iterator
3521 Cat = IFace->known_categories_begin(),
3522 CatEnd = IFace->known_categories_end();
3523 Cat != CatEnd; ++Cat)
3524 AddObjCProperties(*Cat, AllowCategories, AllowNullaryMethods,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003525 CurContext, AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00003526 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003527
3528 // Look through protocols.
Ted Kremenek53b94412010-09-01 01:21:15 +00003529 for (ObjCInterfaceDecl::all_protocol_iterator
3530 I = IFace->all_referenced_protocol_begin(),
3531 E = IFace->all_referenced_protocol_end(); I != E; ++I)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003532 AddObjCProperties(*I, AllowCategories, AllowNullaryMethods, CurContext,
3533 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003534
3535 // Look in the superclass.
3536 if (IFace->getSuperClass())
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003537 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3538 AllowNullaryMethods, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003539 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003540 } else if (const ObjCCategoryDecl *Category
3541 = dyn_cast<ObjCCategoryDecl>(Container)) {
3542 // Look through protocols.
Ted Kremenek53b94412010-09-01 01:21:15 +00003543 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
3544 PEnd = Category->protocol_end();
Douglas Gregor95ac6552009-11-18 01:29:26 +00003545 P != PEnd; ++P)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003546 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3547 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003548 }
3549}
3550
Douglas Gregorf5cd27d2012-01-23 15:59:30 +00003551void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor81b747b2009-09-17 21:32:03 +00003552 SourceLocation OpLoc,
3553 bool IsArrow) {
Douglas Gregorf5cd27d2012-01-23 15:59:30 +00003554 if (!Base || !CodeCompleter)
Douglas Gregor81b747b2009-09-17 21:32:03 +00003555 return;
3556
Douglas Gregorf5cd27d2012-01-23 15:59:30 +00003557 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3558 if (ConvertedBase.isInvalid())
3559 return;
3560 Base = ConvertedBase.get();
3561
John McCall0a2c5e22010-08-25 06:19:51 +00003562 typedef CodeCompletionResult Result;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003563
Douglas Gregor81b747b2009-09-17 21:32:03 +00003564 QualType BaseType = Base->getType();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003565
3566 if (IsArrow) {
3567 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3568 BaseType = Ptr->getPointeeType();
3569 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor3cdee122010-08-26 16:36:48 +00003570 /*Do nothing*/ ;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003571 else
3572 return;
3573 }
3574
Douglas Gregor3da626b2011-07-07 16:03:39 +00003575 enum CodeCompletionContext::Kind contextKind;
3576
3577 if (IsArrow) {
3578 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3579 }
3580 else {
3581 if (BaseType->isObjCObjectPointerType() ||
3582 BaseType->isObjCObjectOrInterfaceType()) {
3583 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3584 }
3585 else {
3586 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3587 }
3588 }
3589
Douglas Gregor218937c2011-02-01 19:23:04 +00003590 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003591 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00003592 CodeCompletionContext(contextKind,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003593 BaseType),
3594 &ResultBuilder::IsMember);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003595 Results.EnterNewScope();
3596 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor3cdee122010-08-26 16:36:48 +00003597 // Indicate that we are performing a member access, and the cv-qualifiers
3598 // for the base object type.
3599 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3600
Douglas Gregor95ac6552009-11-18 01:29:26 +00003601 // Access to a C/C++ class, struct, or union.
Douglas Gregor45bcd432010-01-14 03:21:49 +00003602 Results.allowNestedNameSpecifiers();
Douglas Gregor0cc84042010-01-14 15:47:35 +00003603 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003604 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3605 CodeCompleter->includeGlobals());
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003606
David Blaikie4e4d0842012-03-11 07:00:24 +00003607 if (getLangOpts().CPlusPlus) {
Douglas Gregor95ac6552009-11-18 01:29:26 +00003608 if (!Results.empty()) {
3609 // The "template" keyword can follow "->" or "." in the grammar.
3610 // However, we only want to suggest the template keyword if something
3611 // is dependent.
3612 bool IsDependent = BaseType->isDependentType();
3613 if (!IsDependent) {
3614 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
3615 if (DeclContext *Ctx = (DeclContext *)DepScope->getEntity()) {
3616 IsDependent = Ctx->isDependentContext();
3617 break;
3618 }
3619 }
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003620
Douglas Gregor95ac6552009-11-18 01:29:26 +00003621 if (IsDependent)
Douglas Gregora4477812010-01-14 16:01:26 +00003622 Results.AddResult(Result("template"));
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003623 }
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003624 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003625 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3626 // Objective-C property reference.
Douglas Gregor73449212010-12-09 23:01:55 +00003627 AddedPropertiesSet AddedProperties;
Douglas Gregor95ac6552009-11-18 01:29:26 +00003628
3629 // Add property results based on our interface.
3630 const ObjCObjectPointerType *ObjCPtr
3631 = BaseType->getAsObjCInterfacePointerType();
3632 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003633 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3634 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003635 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003636
3637 // Add properties from the protocols in a qualified interface.
3638 for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(),
3639 E = ObjCPtr->qual_end();
3640 I != E; ++I)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003641 AddObjCProperties(*I, true, /*AllowNullaryMethods=*/true, CurContext,
3642 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003643 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCallc12c5bb2010-05-15 11:32:37 +00003644 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor95ac6552009-11-18 01:29:26 +00003645 // Objective-C instance variable access.
3646 ObjCInterfaceDecl *Class = 0;
3647 if (const ObjCObjectPointerType *ObjCPtr
3648 = BaseType->getAs<ObjCObjectPointerType>())
3649 Class = ObjCPtr->getInterfaceDecl();
3650 else
John McCallc12c5bb2010-05-15 11:32:37 +00003651 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor95ac6552009-11-18 01:29:26 +00003652
3653 // Add all ivars from this class and its superclasses.
Douglas Gregor80f4f4c2010-01-14 16:08:12 +00003654 if (Class) {
3655 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3656 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor8071e422010-08-15 06:18:01 +00003657 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3658 CodeCompleter->includeGlobals());
Douglas Gregor95ac6552009-11-18 01:29:26 +00003659 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003660 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003661
3662 // FIXME: How do we cope with isa?
3663
3664 Results.ExitScope();
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003665
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003666 // Hand off the results found for code completion.
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003667 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003668 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003669 Results.data(),Results.size());
Douglas Gregor81b747b2009-09-17 21:32:03 +00003670}
3671
Douglas Gregor374929f2009-09-18 15:37:17 +00003672void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3673 if (!CodeCompleter)
3674 return;
3675
Douglas Gregor86d9a522009-09-21 16:56:56 +00003676 ResultBuilder::LookupFilter Filter = 0;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003677 enum CodeCompletionContext::Kind ContextKind
3678 = CodeCompletionContext::CCC_Other;
Douglas Gregor374929f2009-09-18 15:37:17 +00003679 switch ((DeclSpec::TST)TagSpec) {
3680 case DeclSpec::TST_enum:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003681 Filter = &ResultBuilder::IsEnum;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003682 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003683 break;
3684
3685 case DeclSpec::TST_union:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003686 Filter = &ResultBuilder::IsUnion;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003687 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003688 break;
3689
3690 case DeclSpec::TST_struct:
Douglas Gregor374929f2009-09-18 15:37:17 +00003691 case DeclSpec::TST_class:
Joao Matos6666ed42012-08-31 18:45:21 +00003692 case DeclSpec::TST_interface:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003693 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003694 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003695 break;
3696
3697 default:
David Blaikieb219cfc2011-09-23 05:06:16 +00003698 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregor374929f2009-09-18 15:37:17 +00003699 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003700
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003701 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3702 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregor5d2fc402010-01-14 03:27:13 +00003703 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCall0d6b1642010-04-23 18:46:30 +00003704
3705 // First pass: look for tags.
3706 Results.setFilter(Filter);
Douglas Gregor8071e422010-08-15 06:18:01 +00003707 LookupVisibleDecls(S, LookupTagName, Consumer,
3708 CodeCompleter->includeGlobals());
John McCall0d6b1642010-04-23 18:46:30 +00003709
Douglas Gregor8071e422010-08-15 06:18:01 +00003710 if (CodeCompleter->includeGlobals()) {
3711 // Second pass: look for nested name specifiers.
3712 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3713 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3714 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003715
Douglas Gregor52779fb2010-09-23 23:01:17 +00003716 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003717 Results.data(),Results.size());
Douglas Gregor374929f2009-09-18 15:37:17 +00003718}
3719
Douglas Gregor1a480c42010-08-27 17:35:51 +00003720void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003721 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003722 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00003723 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor1a480c42010-08-27 17:35:51 +00003724 Results.EnterNewScope();
3725 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3726 Results.AddResult("const");
3727 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3728 Results.AddResult("volatile");
David Blaikie4e4d0842012-03-11 07:00:24 +00003729 if (getLangOpts().C99 &&
Douglas Gregor1a480c42010-08-27 17:35:51 +00003730 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3731 Results.AddResult("restrict");
3732 Results.ExitScope();
3733 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003734 Results.getCompletionContext(),
Douglas Gregor1a480c42010-08-27 17:35:51 +00003735 Results.data(), Results.size());
3736}
3737
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003738void Sema::CodeCompleteCase(Scope *S) {
John McCall781472f2010-08-25 08:40:02 +00003739 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003740 return;
John McCalla8e0cd82011-08-06 07:30:58 +00003741
John McCall781472f2010-08-25 08:40:02 +00003742 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCalla8e0cd82011-08-06 07:30:58 +00003743 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3744 if (!type->isEnumeralType()) {
3745 CodeCompleteExpressionData Data(type);
Douglas Gregorfb629412010-08-23 21:17:50 +00003746 Data.IntegralConstantExpression = true;
3747 CodeCompleteExpression(S, Data);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003748 return;
Douglas Gregorf9578432010-07-28 21:50:18 +00003749 }
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003750
3751 // Code-complete the cases of a switch statement over an enumeration type
3752 // by providing the list of
John McCalla8e0cd82011-08-06 07:30:58 +00003753 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregorb92a4082012-06-12 13:44:08 +00003754 if (EnumDecl *Def = Enum->getDefinition())
3755 Enum = Def;
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003756
3757 // Determine which enumerators we have already seen in the switch statement.
3758 // FIXME: Ideally, we would also be able to look *past* the code-completion
3759 // token, in case we are code-completing in the middle of the switch and not
3760 // at the end. However, we aren't able to do so at the moment.
3761 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003762 NestedNameSpecifier *Qualifier = 0;
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003763 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3764 SC = SC->getNextSwitchCase()) {
3765 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3766 if (!Case)
3767 continue;
3768
3769 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3770 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3771 if (EnumConstantDecl *Enumerator
3772 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3773 // We look into the AST of the case statement to determine which
3774 // enumerator was named. Alternatively, we could compute the value of
3775 // the integral constant expression, then compare it against the
3776 // values of each enumerator. However, value-based approach would not
3777 // work as well with C++ templates where enumerators declared within a
3778 // template are type- and value-dependent.
3779 EnumeratorsSeen.insert(Enumerator);
3780
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003781 // If this is a qualified-id, keep track of the nested-name-specifier
3782 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003783 //
3784 // switch (TagD.getKind()) {
3785 // case TagDecl::TK_enum:
3786 // break;
3787 // case XXX
3788 //
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003789 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003790 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3791 // TK_struct, and TK_class.
Douglas Gregora2813ce2009-10-23 18:54:35 +00003792 Qualifier = DRE->getQualifier();
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003793 }
3794 }
3795
David Blaikie4e4d0842012-03-11 07:00:24 +00003796 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003797 // If there are no prior enumerators in C++, check whether we have to
3798 // qualify the names of the enumerators that we suggest, because they
3799 // may not be visible in this scope.
Douglas Gregorb223d8c2012-02-01 05:02:47 +00003800 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003801 }
3802
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003803 // Add any enumerators that have not yet been mentioned.
Douglas Gregor218937c2011-02-01 19:23:04 +00003804 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003805 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00003806 CodeCompletionContext::CCC_Expression);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003807 Results.EnterNewScope();
3808 for (EnumDecl::enumerator_iterator E = Enum->enumerator_begin(),
3809 EEnd = Enum->enumerator_end();
3810 E != EEnd; ++E) {
David Blaikie581deb32012-06-06 20:45:41 +00003811 if (EnumeratorsSeen.count(*E))
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003812 continue;
3813
Douglas Gregord1f09b42013-01-31 04:52:16 +00003814 CodeCompletionResult R(*E, CCP_EnumInCase, Qualifier);
Douglas Gregor5c722c702011-02-18 23:30:37 +00003815 Results.AddResult(R, CurContext, 0, false);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003816 }
3817 Results.ExitScope();
Douglas Gregor2f880e42010-04-06 20:02:15 +00003818
Douglas Gregor3da626b2011-07-07 16:03:39 +00003819 //We need to make sure we're setting the right context,
3820 //so only say we include macros if the code completer says we do
3821 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3822 if (CodeCompleter->includeMacros()) {
Douglas Gregor3644d972012-10-09 16:01:50 +00003823 AddMacroResults(PP, Results, false);
Douglas Gregor3da626b2011-07-07 16:03:39 +00003824 kind = CodeCompletionContext::CCC_OtherWithMacros;
3825 }
3826
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003827 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00003828 kind,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003829 Results.data(),Results.size());
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003830}
3831
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003832namespace {
3833 struct IsBetterOverloadCandidate {
3834 Sema &S;
John McCall5769d612010-02-08 23:07:23 +00003835 SourceLocation Loc;
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003836
3837 public:
John McCall5769d612010-02-08 23:07:23 +00003838 explicit IsBetterOverloadCandidate(Sema &S, SourceLocation Loc)
3839 : S(S), Loc(Loc) { }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003840
3841 bool
3842 operator()(const OverloadCandidate &X, const OverloadCandidate &Y) const {
John McCall120d63c2010-08-24 20:38:10 +00003843 return isBetterOverloadCandidate(S, X, Y, Loc);
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003844 }
3845 };
3846}
3847
Ahmed Charles13a140c2012-02-25 11:00:22 +00003848static bool anyNullArguments(llvm::ArrayRef<Expr*> Args) {
3849 if (Args.size() && !Args.data())
Douglas Gregord28dcd72010-05-30 06:10:08 +00003850 return true;
Ahmed Charles13a140c2012-02-25 11:00:22 +00003851
3852 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregord28dcd72010-05-30 06:10:08 +00003853 if (!Args[I])
3854 return true;
Ahmed Charles13a140c2012-02-25 11:00:22 +00003855
Douglas Gregord28dcd72010-05-30 06:10:08 +00003856 return false;
3857}
3858
Richard Trieuf81e5a92011-09-09 02:00:50 +00003859void Sema::CodeCompleteCall(Scope *S, Expr *FnIn,
Ahmed Charles13a140c2012-02-25 11:00:22 +00003860 llvm::ArrayRef<Expr *> Args) {
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003861 if (!CodeCompleter)
3862 return;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003863
3864 // When we're code-completing for a call, we fall back to ordinary
3865 // name code-completion whenever we can't produce specific
3866 // results. We may want to revisit this strategy in the future,
3867 // e.g., by merging the two kinds of results.
3868
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003869 Expr *Fn = (Expr *)FnIn;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003870
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003871 // Ignore type-dependent call expressions entirely.
Ahmed Charles13a140c2012-02-25 11:00:22 +00003872 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3873 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003874 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003875 return;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003876 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003877
John McCall3b4294e2009-12-16 12:17:52 +00003878 // Build an overload candidate set based on the functions we find.
John McCall5769d612010-02-08 23:07:23 +00003879 SourceLocation Loc = Fn->getExprLoc();
3880 OverloadCandidateSet CandidateSet(Loc);
John McCall3b4294e2009-12-16 12:17:52 +00003881
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003882 // FIXME: What if we're calling something that isn't a function declaration?
3883 // FIXME: What if we're calling a pseudo-destructor?
3884 // FIXME: What if we're calling a member function?
3885
Douglas Gregorc0265402010-01-21 15:46:19 +00003886 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner5f9e2722011-07-23 10:55:15 +00003887 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorc0265402010-01-21 15:46:19 +00003888
John McCall3b4294e2009-12-16 12:17:52 +00003889 Expr *NakedFn = Fn->IgnoreParenCasts();
3890 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charles13a140c2012-02-25 11:00:22 +00003891 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
John McCall3b4294e2009-12-16 12:17:52 +00003892 /*PartialOverloading=*/ true);
3893 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3894 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorc0265402010-01-21 15:46:19 +00003895 if (FDecl) {
David Blaikie4e4d0842012-03-11 07:00:24 +00003896 if (!getLangOpts().CPlusPlus ||
Douglas Gregord28dcd72010-05-30 06:10:08 +00003897 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorc0265402010-01-21 15:46:19 +00003898 Results.push_back(ResultCandidate(FDecl));
3899 else
John McCall86820f52010-01-26 01:37:31 +00003900 // FIXME: access?
Ahmed Charles13a140c2012-02-25 11:00:22 +00003901 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), Args,
3902 CandidateSet, false, /*PartialOverloading*/true);
Douglas Gregorc0265402010-01-21 15:46:19 +00003903 }
John McCall3b4294e2009-12-16 12:17:52 +00003904 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003905
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003906 QualType ParamType;
3907
Douglas Gregorc0265402010-01-21 15:46:19 +00003908 if (!CandidateSet.empty()) {
3909 // Sort the overload candidate set by placing the best overloads first.
3910 std::stable_sort(CandidateSet.begin(), CandidateSet.end(),
John McCall5769d612010-02-08 23:07:23 +00003911 IsBetterOverloadCandidate(*this, Loc));
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003912
Douglas Gregorc0265402010-01-21 15:46:19 +00003913 // Add the remaining viable overload candidates as code-completion reslults.
3914 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3915 CandEnd = CandidateSet.end();
3916 Cand != CandEnd; ++Cand) {
3917 if (Cand->Viable)
3918 Results.push_back(ResultCandidate(Cand->Function));
3919 }
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003920
3921 // From the viable candidates, try to determine the type of this parameter.
3922 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3923 if (const FunctionType *FType = Results[I].getFunctionType())
3924 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
Ahmed Charles13a140c2012-02-25 11:00:22 +00003925 if (Args.size() < Proto->getNumArgs()) {
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003926 if (ParamType.isNull())
Ahmed Charles13a140c2012-02-25 11:00:22 +00003927 ParamType = Proto->getArgType(Args.size());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003928 else if (!Context.hasSameUnqualifiedType(
3929 ParamType.getNonReferenceType(),
Ahmed Charles13a140c2012-02-25 11:00:22 +00003930 Proto->getArgType(Args.size()).getNonReferenceType())) {
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003931 ParamType = QualType();
3932 break;
3933 }
3934 }
3935 }
3936 } else {
3937 // Try to determine the parameter type from the type of the expression
3938 // being called.
3939 QualType FunctionType = Fn->getType();
3940 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3941 FunctionType = Ptr->getPointeeType();
3942 else if (const BlockPointerType *BlockPtr
3943 = FunctionType->getAs<BlockPointerType>())
3944 FunctionType = BlockPtr->getPointeeType();
3945 else if (const MemberPointerType *MemPtr
3946 = FunctionType->getAs<MemberPointerType>())
3947 FunctionType = MemPtr->getPointeeType();
3948
3949 if (const FunctionProtoType *Proto
3950 = FunctionType->getAs<FunctionProtoType>()) {
Ahmed Charles13a140c2012-02-25 11:00:22 +00003951 if (Args.size() < Proto->getNumArgs())
3952 ParamType = Proto->getArgType(Args.size());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003953 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003954 }
Douglas Gregoref96eac2009-12-11 19:06:04 +00003955
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003956 if (ParamType.isNull())
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003957 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003958 else
3959 CodeCompleteExpression(S, ParamType);
3960
Douglas Gregor2e4c7a52010-04-06 20:19:47 +00003961 if (!Results.empty())
Ahmed Charles13a140c2012-02-25 11:00:22 +00003962 CodeCompleter->ProcessOverloadCandidates(*this, Args.size(), Results.data(),
Douglas Gregoref96eac2009-12-11 19:06:04 +00003963 Results.size());
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003964}
3965
John McCalld226f652010-08-21 09:40:31 +00003966void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3967 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003968 if (!VD) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003969 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003970 return;
3971 }
3972
3973 CodeCompleteExpression(S, VD->getType());
3974}
3975
3976void Sema::CodeCompleteReturn(Scope *S) {
3977 QualType ResultType;
3978 if (isa<BlockDecl>(CurContext)) {
3979 if (BlockScopeInfo *BSI = getCurBlock())
3980 ResultType = BSI->ReturnType;
3981 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
3982 ResultType = Function->getResultType();
3983 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
3984 ResultType = Method->getResultType();
3985
3986 if (ResultType.isNull())
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003987 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003988 else
3989 CodeCompleteExpression(S, ResultType);
3990}
3991
Douglas Gregord2d8be62011-07-30 08:36:53 +00003992void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregord2d8be62011-07-30 08:36:53 +00003993 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003994 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord2d8be62011-07-30 08:36:53 +00003995 mapCodeCompletionContext(*this, PCC_Statement));
3996 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3997 Results.EnterNewScope();
3998
3999 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4000 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4001 CodeCompleter->includeGlobals());
4002
4003 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
4004
4005 // "else" block
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004006 CodeCompletionBuilder Builder(Results.getAllocator(),
4007 Results.getCodeCompletionTUInfo());
Douglas Gregord2d8be62011-07-30 08:36:53 +00004008 Builder.AddTypedTextChunk("else");
Douglas Gregorf11641a2012-02-16 17:49:04 +00004009 if (Results.includeCodePatterns()) {
4010 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4011 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4012 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4013 Builder.AddPlaceholderChunk("statements");
4014 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4015 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4016 }
Douglas Gregord2d8be62011-07-30 08:36:53 +00004017 Results.AddResult(Builder.TakeString());
4018
4019 // "else if" block
4020 Builder.AddTypedTextChunk("else");
4021 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4022 Builder.AddTextChunk("if");
4023 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4024 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00004025 if (getLangOpts().CPlusPlus)
Douglas Gregord2d8be62011-07-30 08:36:53 +00004026 Builder.AddPlaceholderChunk("condition");
4027 else
4028 Builder.AddPlaceholderChunk("expression");
4029 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf11641a2012-02-16 17:49:04 +00004030 if (Results.includeCodePatterns()) {
4031 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4032 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4033 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4034 Builder.AddPlaceholderChunk("statements");
4035 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4036 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4037 }
Douglas Gregord2d8be62011-07-30 08:36:53 +00004038 Results.AddResult(Builder.TakeString());
4039
4040 Results.ExitScope();
4041
4042 if (S->getFnParent())
David Blaikie4e4d0842012-03-11 07:00:24 +00004043 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregord2d8be62011-07-30 08:36:53 +00004044
4045 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00004046 AddMacroResults(PP, Results, false);
Douglas Gregord2d8be62011-07-30 08:36:53 +00004047
4048 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4049 Results.data(),Results.size());
4050}
4051
Richard Trieuf81e5a92011-09-09 02:00:50 +00004052void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004053 if (LHS)
4054 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4055 else
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004056 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004057}
4058
Jeffrey Yasskin9ab14542010-04-08 16:38:48 +00004059void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor81b747b2009-09-17 21:32:03 +00004060 bool EnteringContext) {
4061 if (!SS.getScopeRep() || !CodeCompleter)
4062 return;
4063
Douglas Gregor86d9a522009-09-21 16:56:56 +00004064 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4065 if (!Ctx)
4066 return;
Douglas Gregord1cd31a2009-12-11 18:28:39 +00004067
4068 // Try to instantiate any non-dependent declaration contexts before
4069 // we look in them.
John McCall77bb1aa2010-05-01 00:40:08 +00004070 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregord1cd31a2009-12-11 18:28:39 +00004071 return;
4072
Douglas Gregor218937c2011-02-01 19:23:04 +00004073 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004074 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004075 CodeCompletionContext::CCC_Name);
Douglas Gregorf6961522010-08-27 21:18:54 +00004076 Results.EnterNewScope();
Douglas Gregor52779fb2010-09-23 23:01:17 +00004077
Douglas Gregor86d9a522009-09-21 16:56:56 +00004078 // The "template" keyword can follow "::" in the grammar, but only
4079 // put it into the grammar if the nested-name-specifier is dependent.
4080 NestedNameSpecifier *NNS = (NestedNameSpecifier *)SS.getScopeRep();
4081 if (!Results.empty() && NNS->isDependent())
Douglas Gregora4477812010-01-14 16:01:26 +00004082 Results.AddResult("template");
Douglas Gregorf6961522010-08-27 21:18:54 +00004083
4084 // Add calls to overridden virtual functions, if there are any.
4085 //
4086 // FIXME: This isn't wonderful, because we don't know whether we're actually
4087 // in a context that permits expressions. This is a general issue with
4088 // qualified-id completions.
4089 if (!EnteringContext)
4090 MaybeAddOverrideCalls(*this, Ctx, Results);
4091 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004092
Douglas Gregorf6961522010-08-27 21:18:54 +00004093 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4094 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4095
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004096 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor430d7a12011-07-25 17:48:11 +00004097 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004098 Results.data(),Results.size());
Douglas Gregor81b747b2009-09-17 21:32:03 +00004099}
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004100
4101void Sema::CodeCompleteUsing(Scope *S) {
4102 if (!CodeCompleter)
4103 return;
4104
Douglas Gregor218937c2011-02-01 19:23:04 +00004105 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004106 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00004107 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4108 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004109 Results.EnterNewScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004110
4111 // If we aren't in class scope, we could see the "namespace" keyword.
4112 if (!S->isClassScope())
John McCall0a2c5e22010-08-25 06:19:51 +00004113 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor86d9a522009-09-21 16:56:56 +00004114
4115 // After "using", we can see anything that would start a
4116 // nested-name-specifier.
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004117 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004118 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4119 CodeCompleter->includeGlobals());
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004120 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004121
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004122 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004123 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004124 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004125}
4126
4127void Sema::CodeCompleteUsingDirective(Scope *S) {
4128 if (!CodeCompleter)
4129 return;
4130
Douglas Gregor86d9a522009-09-21 16:56:56 +00004131 // After "using namespace", we expect to see a namespace name or namespace
4132 // alias.
Douglas Gregor218937c2011-02-01 19:23:04 +00004133 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004134 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004135 CodeCompletionContext::CCC_Namespace,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004136 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004137 Results.EnterNewScope();
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004138 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004139 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4140 CodeCompleter->includeGlobals());
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004141 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004142 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor8071e422010-08-15 06:18:01 +00004143 CodeCompletionContext::CCC_Namespace,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004144 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004145}
4146
4147void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4148 if (!CodeCompleter)
4149 return;
4150
Douglas Gregor86d9a522009-09-21 16:56:56 +00004151 DeclContext *Ctx = (DeclContext *)S->getEntity();
4152 if (!S->getParent())
4153 Ctx = Context.getTranslationUnitDecl();
4154
Douglas Gregor52779fb2010-09-23 23:01:17 +00004155 bool SuppressedGlobalResults
4156 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4157
Douglas Gregor218937c2011-02-01 19:23:04 +00004158 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004159 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00004160 SuppressedGlobalResults
4161 ? CodeCompletionContext::CCC_Namespace
4162 : CodeCompletionContext::CCC_Other,
4163 &ResultBuilder::IsNamespace);
4164
4165 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00004166 // We only want to see those namespaces that have already been defined
4167 // within this scope, because its likely that the user is creating an
4168 // extended namespace declaration. Keep track of the most recent
4169 // definition of each namespace.
4170 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4171 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4172 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4173 NS != NSEnd; ++NS)
David Blaikie581deb32012-06-06 20:45:41 +00004174 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor86d9a522009-09-21 16:56:56 +00004175
4176 // Add the most recent definition (or extended definition) of each
4177 // namespace to the list of results.
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004178 Results.EnterNewScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004179 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregorba103062012-03-27 23:34:16 +00004180 NS = OrigToLatest.begin(),
4181 NSEnd = OrigToLatest.end();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004182 NS != NSEnd; ++NS)
Douglas Gregord1f09b42013-01-31 04:52:16 +00004183 Results.AddResult(CodeCompletionResult(
4184 NS->second, Results.getBasePriority(NS->second), 0),
Douglas Gregor608300b2010-01-14 16:14:35 +00004185 CurContext, 0, false);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004186 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004187 }
4188
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004189 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004190 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004191 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004192}
4193
4194void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4195 if (!CodeCompleter)
4196 return;
4197
Douglas Gregor86d9a522009-09-21 16:56:56 +00004198 // After "namespace", we expect to see a namespace or alias.
Douglas Gregor218937c2011-02-01 19:23:04 +00004199 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004200 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004201 CodeCompletionContext::CCC_Namespace,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004202 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004203 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004204 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4205 CodeCompleter->includeGlobals());
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004206 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004207 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004208 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004209}
4210
Douglas Gregored8d3222009-09-18 20:05:18 +00004211void Sema::CodeCompleteOperatorName(Scope *S) {
4212 if (!CodeCompleter)
4213 return;
Douglas Gregor86d9a522009-09-21 16:56:56 +00004214
John McCall0a2c5e22010-08-25 06:19:51 +00004215 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004216 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004217 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004218 CodeCompletionContext::CCC_Type,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004219 &ResultBuilder::IsType);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004220 Results.EnterNewScope();
Douglas Gregored8d3222009-09-18 20:05:18 +00004221
Douglas Gregor86d9a522009-09-21 16:56:56 +00004222 // Add the names of overloadable operators.
4223#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4224 if (std::strcmp(Spelling, "?")) \
Douglas Gregora4477812010-01-14 16:01:26 +00004225 Results.AddResult(Result(Spelling));
Douglas Gregor86d9a522009-09-21 16:56:56 +00004226#include "clang/Basic/OperatorKinds.def"
4227
4228 // Add any type names visible from the current scope
Douglas Gregor45bcd432010-01-14 03:21:49 +00004229 Results.allowNestedNameSpecifiers();
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004230 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004231 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4232 CodeCompleter->includeGlobals());
Douglas Gregor86d9a522009-09-21 16:56:56 +00004233
4234 // Add any type specifiers
David Blaikie4e4d0842012-03-11 07:00:24 +00004235 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004236 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004237
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004238 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor8071e422010-08-15 06:18:01 +00004239 CodeCompletionContext::CCC_Type,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004240 Results.data(),Results.size());
Douglas Gregored8d3222009-09-18 20:05:18 +00004241}
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004242
Douglas Gregor0133f522010-08-28 00:00:50 +00004243void Sema::CodeCompleteConstructorInitializer(Decl *ConstructorD,
Sean Huntcbb67482011-01-08 20:30:50 +00004244 CXXCtorInitializer** Initializers,
Douglas Gregor0133f522010-08-28 00:00:50 +00004245 unsigned NumInitializers) {
Douglas Gregor8987b232011-09-27 23:30:47 +00004246 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor0133f522010-08-28 00:00:50 +00004247 CXXConstructorDecl *Constructor
4248 = static_cast<CXXConstructorDecl *>(ConstructorD);
4249 if (!Constructor)
4250 return;
4251
Douglas Gregor218937c2011-02-01 19:23:04 +00004252 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004253 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00004254 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregor0133f522010-08-28 00:00:50 +00004255 Results.EnterNewScope();
4256
4257 // Fill in any already-initialized fields or base classes.
4258 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4259 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
4260 for (unsigned I = 0; I != NumInitializers; ++I) {
4261 if (Initializers[I]->isBaseInitializer())
4262 InitializedBases.insert(
4263 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4264 else
Francois Pichet00eb3f92010-12-04 09:14:42 +00004265 InitializedFields.insert(cast<FieldDecl>(
4266 Initializers[I]->getAnyMember()));
Douglas Gregor0133f522010-08-28 00:00:50 +00004267 }
4268
4269 // Add completions for base classes.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004270 CodeCompletionBuilder Builder(Results.getAllocator(),
4271 Results.getCodeCompletionTUInfo());
Douglas Gregor0c431c82010-08-29 19:27:27 +00004272 bool SawLastInitializer = (NumInitializers == 0);
Douglas Gregor0133f522010-08-28 00:00:50 +00004273 CXXRecordDecl *ClassDecl = Constructor->getParent();
4274 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->bases_begin(),
4275 BaseEnd = ClassDecl->bases_end();
4276 Base != BaseEnd; ++Base) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004277 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4278 SawLastInitializer
4279 = NumInitializers > 0 &&
4280 Initializers[NumInitializers - 1]->isBaseInitializer() &&
4281 Context.hasSameUnqualifiedType(Base->getType(),
4282 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregor0133f522010-08-28 00:00:50 +00004283 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004284 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004285
Douglas Gregor218937c2011-02-01 19:23:04 +00004286 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004287 Results.getAllocator().CopyString(
John McCallf85e1932011-06-15 23:02:42 +00004288 Base->getType().getAsString(Policy)));
Douglas Gregor218937c2011-02-01 19:23:04 +00004289 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4290 Builder.AddPlaceholderChunk("args");
4291 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4292 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004293 SawLastInitializer? CCP_NextInitializer
4294 : CCP_MemberDeclaration));
4295 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004296 }
4297
4298 // Add completions for virtual base classes.
4299 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->vbases_begin(),
4300 BaseEnd = ClassDecl->vbases_end();
4301 Base != BaseEnd; ++Base) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004302 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4303 SawLastInitializer
4304 = NumInitializers > 0 &&
4305 Initializers[NumInitializers - 1]->isBaseInitializer() &&
4306 Context.hasSameUnqualifiedType(Base->getType(),
4307 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregor0133f522010-08-28 00:00:50 +00004308 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004309 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004310
Douglas Gregor218937c2011-02-01 19:23:04 +00004311 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004312 Builder.getAllocator().CopyString(
John McCallf85e1932011-06-15 23:02:42 +00004313 Base->getType().getAsString(Policy)));
Douglas Gregor218937c2011-02-01 19:23:04 +00004314 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4315 Builder.AddPlaceholderChunk("args");
4316 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4317 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004318 SawLastInitializer? CCP_NextInitializer
4319 : CCP_MemberDeclaration));
4320 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004321 }
4322
4323 // Add completions for members.
4324 for (CXXRecordDecl::field_iterator Field = ClassDecl->field_begin(),
4325 FieldEnd = ClassDecl->field_end();
4326 Field != FieldEnd; ++Field) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004327 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4328 SawLastInitializer
4329 = NumInitializers > 0 &&
Francois Pichet00eb3f92010-12-04 09:14:42 +00004330 Initializers[NumInitializers - 1]->isAnyMemberInitializer() &&
David Blaikie581deb32012-06-06 20:45:41 +00004331 Initializers[NumInitializers - 1]->getAnyMember() == *Field;
Douglas Gregor0133f522010-08-28 00:00:50 +00004332 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004333 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004334
4335 if (!Field->getDeclName())
4336 continue;
4337
Douglas Gregordae68752011-02-01 22:57:45 +00004338 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00004339 Field->getIdentifier()->getName()));
4340 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4341 Builder.AddPlaceholderChunk("args");
4342 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4343 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004344 SawLastInitializer? CCP_NextInitializer
Douglas Gregora67e03f2010-09-09 21:42:20 +00004345 : CCP_MemberDeclaration,
Douglas Gregorba103062012-03-27 23:34:16 +00004346 CXCursor_MemberRef,
4347 CXAvailability_Available,
David Blaikie581deb32012-06-06 20:45:41 +00004348 *Field));
Douglas Gregor0c431c82010-08-29 19:27:27 +00004349 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004350 }
4351 Results.ExitScope();
4352
Douglas Gregor52779fb2010-09-23 23:01:17 +00004353 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor0133f522010-08-28 00:00:50 +00004354 Results.data(), Results.size());
4355}
4356
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004357/// \brief Determine whether this scope denotes a namespace.
4358static bool isNamespaceScope(Scope *S) {
4359 DeclContext *DC = static_cast<DeclContext *>(S->getEntity());
4360 if (!DC)
4361 return false;
4362
4363 return DC->isFileContext();
4364}
4365
4366void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4367 bool AfterAmpersand) {
4368 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004369 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004370 CodeCompletionContext::CCC_Other);
4371 Results.EnterNewScope();
4372
4373 // Note what has already been captured.
4374 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4375 bool IncludedThis = false;
4376 for (SmallVectorImpl<LambdaCapture>::iterator C = Intro.Captures.begin(),
4377 CEnd = Intro.Captures.end();
4378 C != CEnd; ++C) {
4379 if (C->Kind == LCK_This) {
4380 IncludedThis = true;
4381 continue;
4382 }
4383
4384 Known.insert(C->Id);
4385 }
4386
4387 // Look for other capturable variables.
4388 for (; S && !isNamespaceScope(S); S = S->getParent()) {
4389 for (Scope::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
4390 D != DEnd; ++D) {
4391 VarDecl *Var = dyn_cast<VarDecl>(*D);
4392 if (!Var ||
4393 !Var->hasLocalStorage() ||
4394 Var->hasAttr<BlocksAttr>())
4395 continue;
4396
4397 if (Known.insert(Var->getIdentifier()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00004398 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
4399 CurContext, 0, false);
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004400 }
4401 }
4402
4403 // Add 'this', if it would be valid.
4404 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4405 addThisCompletion(*this, Results);
4406
4407 Results.ExitScope();
4408
4409 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4410 Results.data(), Results.size());
4411}
4412
James Dennetta40f7922012-06-14 03:11:41 +00004413/// Macro that optionally prepends an "@" to the string literal passed in via
4414/// Keyword, depending on whether NeedAt is true or false.
4415#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4416
Douglas Gregorbca403c2010-01-13 23:51:12 +00004417static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004418 ResultBuilder &Results,
4419 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004420 typedef CodeCompletionResult Result;
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004421 // Since we have an implementation, we can end it.
James Dennetta40f7922012-06-14 03:11:41 +00004422 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004423
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004424 CodeCompletionBuilder Builder(Results.getAllocator(),
4425 Results.getCodeCompletionTUInfo());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004426 if (LangOpts.ObjC2) {
4427 // @dynamic
James Dennetta40f7922012-06-14 03:11:41 +00004428 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004429 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4430 Builder.AddPlaceholderChunk("property");
4431 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004432
4433 // @synthesize
James Dennetta40f7922012-06-14 03:11:41 +00004434 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004435 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4436 Builder.AddPlaceholderChunk("property");
4437 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004438 }
4439}
4440
Douglas Gregorbca403c2010-01-13 23:51:12 +00004441static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004442 ResultBuilder &Results,
4443 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004444 typedef CodeCompletionResult Result;
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004445
4446 // Since we have an interface or protocol, we can end it.
James Dennetta40f7922012-06-14 03:11:41 +00004447 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004448
4449 if (LangOpts.ObjC2) {
4450 // @property
James Dennetta40f7922012-06-14 03:11:41 +00004451 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004452
4453 // @required
James Dennetta40f7922012-06-14 03:11:41 +00004454 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004455
4456 // @optional
James Dennetta40f7922012-06-14 03:11:41 +00004457 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004458 }
4459}
4460
Douglas Gregorbca403c2010-01-13 23:51:12 +00004461static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004462 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004463 CodeCompletionBuilder Builder(Results.getAllocator(),
4464 Results.getCodeCompletionTUInfo());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004465
4466 // @class name ;
James Dennetta40f7922012-06-14 03:11:41 +00004467 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004468 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4469 Builder.AddPlaceholderChunk("name");
4470 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004471
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004472 if (Results.includeCodePatterns()) {
4473 // @interface name
4474 // FIXME: Could introduce the whole pattern, including superclasses and
4475 // such.
James Dennetta40f7922012-06-14 03:11:41 +00004476 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004477 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4478 Builder.AddPlaceholderChunk("class");
4479 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004480
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004481 // @protocol name
James Dennetta40f7922012-06-14 03:11:41 +00004482 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004483 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4484 Builder.AddPlaceholderChunk("protocol");
4485 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004486
4487 // @implementation name
James Dennetta40f7922012-06-14 03:11:41 +00004488 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004489 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4490 Builder.AddPlaceholderChunk("class");
4491 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004492 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004493
4494 // @compatibility_alias name
James Dennetta40f7922012-06-14 03:11:41 +00004495 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004496 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4497 Builder.AddPlaceholderChunk("alias");
4498 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4499 Builder.AddPlaceholderChunk("class");
4500 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor06898632013-03-07 23:26:24 +00004501
4502 if (Results.getSema().getLangOpts().Modules) {
4503 // @import name
4504 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "import"));
4505 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4506 Builder.AddPlaceholderChunk("module");
4507 Results.AddResult(Result(Builder.TakeString()));
4508 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004509}
4510
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004511void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004512 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004513 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004514 CodeCompletionContext::CCC_Other);
Douglas Gregorc464ae82009-12-07 09:27:33 +00004515 Results.EnterNewScope();
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004516 if (isa<ObjCImplDecl>(CurContext))
David Blaikie4e4d0842012-03-11 07:00:24 +00004517 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004518 else if (CurContext->isObjCContainer())
David Blaikie4e4d0842012-03-11 07:00:24 +00004519 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004520 else
Douglas Gregorbca403c2010-01-13 23:51:12 +00004521 AddObjCTopLevelResults(Results, false);
Douglas Gregorc464ae82009-12-07 09:27:33 +00004522 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004523 HandleCodeCompleteResults(this, CodeCompleter,
4524 CodeCompletionContext::CCC_Other,
4525 Results.data(),Results.size());
Douglas Gregorc464ae82009-12-07 09:27:33 +00004526}
4527
Douglas Gregorbca403c2010-01-13 23:51:12 +00004528static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004529 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004530 CodeCompletionBuilder Builder(Results.getAllocator(),
4531 Results.getCodeCompletionTUInfo());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004532
4533 // @encode ( type-name )
Douglas Gregor8ca72082011-10-18 21:20:17 +00004534 const char *EncodeType = "char[]";
David Blaikie4e4d0842012-03-11 07:00:24 +00004535 if (Results.getSema().getLangOpts().CPlusPlus ||
4536 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004537 EncodeType = "const char[]";
Douglas Gregor8ca72082011-10-18 21:20:17 +00004538 Builder.AddResultTypeChunk(EncodeType);
James Dennetta40f7922012-06-14 03:11:41 +00004539 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004540 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4541 Builder.AddPlaceholderChunk("type-name");
4542 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4543 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004544
4545 // @protocol ( protocol-name )
Douglas Gregor8ca72082011-10-18 21:20:17 +00004546 Builder.AddResultTypeChunk("Protocol *");
James Dennetta40f7922012-06-14 03:11:41 +00004547 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004548 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4549 Builder.AddPlaceholderChunk("protocol-name");
4550 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4551 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004552
4553 // @selector ( selector )
Douglas Gregor8ca72082011-10-18 21:20:17 +00004554 Builder.AddResultTypeChunk("SEL");
James Dennetta40f7922012-06-14 03:11:41 +00004555 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004556 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4557 Builder.AddPlaceholderChunk("selector");
4558 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4559 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004560
4561 // @"string"
4562 Builder.AddResultTypeChunk("NSString *");
4563 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4564 Builder.AddPlaceholderChunk("string");
4565 Builder.AddTextChunk("\"");
4566 Results.AddResult(Result(Builder.TakeString()));
4567
Douglas Gregor79615892012-07-17 23:24:47 +00004568 // @[objects, ...]
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004569 Builder.AddResultTypeChunk("NSArray *");
James Dennetta40f7922012-06-14 03:11:41 +00004570 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004571 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004572 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4573 Results.AddResult(Result(Builder.TakeString()));
4574
Douglas Gregor79615892012-07-17 23:24:47 +00004575 // @{key : object, ...}
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004576 Builder.AddResultTypeChunk("NSDictionary *");
James Dennetta40f7922012-06-14 03:11:41 +00004577 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"{"));
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004578 Builder.AddPlaceholderChunk("key");
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004579 Builder.AddChunk(CodeCompletionString::CK_Colon);
4580 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4581 Builder.AddPlaceholderChunk("object, ...");
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004582 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4583 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004584
Douglas Gregor79615892012-07-17 23:24:47 +00004585 // @(expression)
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004586 Builder.AddResultTypeChunk("id");
4587 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004588 Builder.AddPlaceholderChunk("expression");
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004589 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4590 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004591}
4592
Douglas Gregorbca403c2010-01-13 23:51:12 +00004593static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004594 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004595 CodeCompletionBuilder Builder(Results.getAllocator(),
4596 Results.getCodeCompletionTUInfo());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004597
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004598 if (Results.includeCodePatterns()) {
4599 // @try { statements } @catch ( declaration ) { statements } @finally
4600 // { statements }
James Dennetta40f7922012-06-14 03:11:41 +00004601 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004602 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4603 Builder.AddPlaceholderChunk("statements");
4604 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4605 Builder.AddTextChunk("@catch");
4606 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4607 Builder.AddPlaceholderChunk("parameter");
4608 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4609 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4610 Builder.AddPlaceholderChunk("statements");
4611 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4612 Builder.AddTextChunk("@finally");
4613 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4614 Builder.AddPlaceholderChunk("statements");
4615 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4616 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004617 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004618
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004619 // @throw
James Dennetta40f7922012-06-14 03:11:41 +00004620 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004621 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4622 Builder.AddPlaceholderChunk("expression");
4623 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004624
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004625 if (Results.includeCodePatterns()) {
4626 // @synchronized ( expression ) { statements }
James Dennetta40f7922012-06-14 03:11:41 +00004627 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004628 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4629 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4630 Builder.AddPlaceholderChunk("expression");
4631 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4632 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4633 Builder.AddPlaceholderChunk("statements");
4634 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4635 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004636 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004637}
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004638
Douglas Gregorbca403c2010-01-13 23:51:12 +00004639static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004640 ResultBuilder &Results,
4641 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004642 typedef CodeCompletionResult Result;
James Dennetta40f7922012-06-14 03:11:41 +00004643 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4644 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4645 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004646 if (LangOpts.ObjC2)
James Dennetta40f7922012-06-14 03:11:41 +00004647 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004648}
4649
4650void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004651 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004652 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004653 CodeCompletionContext::CCC_Other);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004654 Results.EnterNewScope();
David Blaikie4e4d0842012-03-11 07:00:24 +00004655 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004656 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004657 HandleCodeCompleteResults(this, CodeCompleter,
4658 CodeCompletionContext::CCC_Other,
4659 Results.data(),Results.size());
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004660}
4661
4662void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004663 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004664 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004665 CodeCompletionContext::CCC_Other);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004666 Results.EnterNewScope();
Douglas Gregorbca403c2010-01-13 23:51:12 +00004667 AddObjCStatementResults(Results, false);
4668 AddObjCExpressionResults(Results, false);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004669 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004670 HandleCodeCompleteResults(this, CodeCompleter,
4671 CodeCompletionContext::CCC_Other,
4672 Results.data(),Results.size());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004673}
4674
4675void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004676 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004677 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004678 CodeCompletionContext::CCC_Other);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004679 Results.EnterNewScope();
Douglas Gregorbca403c2010-01-13 23:51:12 +00004680 AddObjCExpressionResults(Results, false);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004681 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004682 HandleCodeCompleteResults(this, CodeCompleter,
4683 CodeCompletionContext::CCC_Other,
4684 Results.data(),Results.size());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004685}
4686
Douglas Gregor988358f2009-11-19 00:14:45 +00004687/// \brief Determine whether the addition of the given flag to an Objective-C
4688/// property's attributes will cause a conflict.
Bill Wendlingad017fa2012-12-20 19:22:21 +00004689static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregor988358f2009-11-19 00:14:45 +00004690 // Check if we've already added this flag.
Bill Wendlingad017fa2012-12-20 19:22:21 +00004691 if (Attributes & NewFlag)
Douglas Gregor988358f2009-11-19 00:14:45 +00004692 return true;
4693
Bill Wendlingad017fa2012-12-20 19:22:21 +00004694 Attributes |= NewFlag;
Douglas Gregor988358f2009-11-19 00:14:45 +00004695
4696 // Check for collisions with "readonly".
Bill Wendlingad017fa2012-12-20 19:22:21 +00004697 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4698 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregor988358f2009-11-19 00:14:45 +00004699 return true;
4700
Jordan Rosed7403a72012-08-20 20:01:13 +00004701 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendlingad017fa2012-12-20 19:22:21 +00004702 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCallf85e1932011-06-15 23:02:42 +00004703 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregor988358f2009-11-19 00:14:45 +00004704 ObjCDeclSpec::DQ_PR_copy |
Jordan Rosed7403a72012-08-20 20:01:13 +00004705 ObjCDeclSpec::DQ_PR_retain |
4706 ObjCDeclSpec::DQ_PR_strong |
4707 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregor988358f2009-11-19 00:14:45 +00004708 if (AssignCopyRetMask &&
4709 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCallf85e1932011-06-15 23:02:42 +00004710 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregor988358f2009-11-19 00:14:45 +00004711 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCallf85e1932011-06-15 23:02:42 +00004712 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rosed7403a72012-08-20 20:01:13 +00004713 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4714 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregor988358f2009-11-19 00:14:45 +00004715 return true;
4716
4717 return false;
4718}
4719
Douglas Gregora93b1082009-11-18 23:08:07 +00004720void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroffece8e712009-10-08 21:55:05 +00004721 if (!CodeCompleter)
4722 return;
Douglas Gregord3c68542009-11-19 01:08:35 +00004723
Bill Wendlingad017fa2012-12-20 19:22:21 +00004724 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroffece8e712009-10-08 21:55:05 +00004725
Douglas Gregor218937c2011-02-01 19:23:04 +00004726 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004727 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004728 CodeCompletionContext::CCC_Other);
Steve Naroffece8e712009-10-08 21:55:05 +00004729 Results.EnterNewScope();
Bill Wendlingad017fa2012-12-20 19:22:21 +00004730 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall0a2c5e22010-08-25 06:19:51 +00004731 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004732 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall0a2c5e22010-08-25 06:19:51 +00004733 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004734 if (!ObjCPropertyFlagConflicts(Attributes,
John McCallf85e1932011-06-15 23:02:42 +00004735 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4736 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004737 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall0a2c5e22010-08-25 06:19:51 +00004738 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004739 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall0a2c5e22010-08-25 06:19:51 +00004740 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004741 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCallf85e1932011-06-15 23:02:42 +00004742 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004743 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall0a2c5e22010-08-25 06:19:51 +00004744 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004745 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall0a2c5e22010-08-25 06:19:51 +00004746 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004747 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian27f45232011-06-11 17:14:27 +00004748 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rosed7403a72012-08-20 20:01:13 +00004749
4750 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall0a7dd782012-08-21 02:47:43 +00004751 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendlingad017fa2012-12-20 19:22:21 +00004752 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rosed7403a72012-08-20 20:01:13 +00004753 Results.AddResult(CodeCompletionResult("weak"));
4754
Bill Wendlingad017fa2012-12-20 19:22:21 +00004755 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004756 CodeCompletionBuilder Setter(Results.getAllocator(),
4757 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00004758 Setter.AddTypedTextChunk("setter");
4759 Setter.AddTextChunk(" = ");
4760 Setter.AddPlaceholderChunk("method");
4761 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor54f01612009-11-19 00:01:57 +00004762 }
Bill Wendlingad017fa2012-12-20 19:22:21 +00004763 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004764 CodeCompletionBuilder Getter(Results.getAllocator(),
4765 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00004766 Getter.AddTypedTextChunk("getter");
4767 Getter.AddTextChunk(" = ");
4768 Getter.AddPlaceholderChunk("method");
4769 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor54f01612009-11-19 00:01:57 +00004770 }
Steve Naroffece8e712009-10-08 21:55:05 +00004771 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004772 HandleCodeCompleteResults(this, CodeCompleter,
4773 CodeCompletionContext::CCC_Other,
4774 Results.data(),Results.size());
Steve Naroffece8e712009-10-08 21:55:05 +00004775}
Steve Naroffc4df6d22009-11-07 02:08:14 +00004776
James Dennettde23c7e2012-06-17 05:33:25 +00004777/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregor4ad96852009-11-19 07:41:15 +00004778/// via code completion.
4779enum ObjCMethodKind {
Dmitri Gribenko49fdccb2012-06-08 23:13:42 +00004780 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4781 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4782 MK_OneArgSelector ///< One-argument selector.
Douglas Gregor4ad96852009-11-19 07:41:15 +00004783};
4784
Douglas Gregor458433d2010-08-26 15:07:07 +00004785static bool isAcceptableObjCSelector(Selector Sel,
4786 ObjCMethodKind WantKind,
4787 IdentifierInfo **SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004788 unsigned NumSelIdents,
4789 bool AllowSameLength = true) {
Douglas Gregor458433d2010-08-26 15:07:07 +00004790 if (NumSelIdents > Sel.getNumArgs())
4791 return false;
4792
4793 switch (WantKind) {
4794 case MK_Any: break;
4795 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4796 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4797 }
4798
Douglas Gregorcf544262010-11-17 21:36:08 +00004799 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4800 return false;
4801
Douglas Gregor458433d2010-08-26 15:07:07 +00004802 for (unsigned I = 0; I != NumSelIdents; ++I)
4803 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4804 return false;
4805
4806 return true;
4807}
4808
Douglas Gregor4ad96852009-11-19 07:41:15 +00004809static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4810 ObjCMethodKind WantKind,
4811 IdentifierInfo **SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004812 unsigned NumSelIdents,
4813 bool AllowSameLength = true) {
Douglas Gregor458433d2010-08-26 15:07:07 +00004814 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004815 NumSelIdents, AllowSameLength);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004816}
Douglas Gregord36adf52010-09-16 16:06:31 +00004817
4818namespace {
4819 /// \brief A set of selectors, which is used to avoid introducing multiple
4820 /// completions with the same selector into the result set.
4821 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4822}
4823
Douglas Gregor36ecb042009-11-17 23:22:23 +00004824/// \brief Add all of the Objective-C methods in the given Objective-C
4825/// container to the set of results.
4826///
4827/// The container will be a class, protocol, category, or implementation of
4828/// any of the above. This mether will recurse to include methods from
4829/// the superclasses of classes along with their categories, protocols, and
4830/// implementations.
4831///
4832/// \param Container the container in which we'll look to find methods.
4833///
James Dennetta40f7922012-06-14 03:11:41 +00004834/// \param WantInstanceMethods Whether to add instance methods (only); if
4835/// false, this routine will add factory methods (only).
Douglas Gregor36ecb042009-11-17 23:22:23 +00004836///
4837/// \param CurContext the context in which we're performing the lookup that
4838/// finds methods.
4839///
Douglas Gregorcf544262010-11-17 21:36:08 +00004840/// \param AllowSameLength Whether we allow a method to be added to the list
4841/// when it has the same number of parameters as we have selector identifiers.
4842///
Douglas Gregor36ecb042009-11-17 23:22:23 +00004843/// \param Results the structure into which we'll add results.
4844static void AddObjCMethods(ObjCContainerDecl *Container,
4845 bool WantInstanceMethods,
Douglas Gregor4ad96852009-11-19 07:41:15 +00004846 ObjCMethodKind WantKind,
Douglas Gregord3c68542009-11-19 01:08:35 +00004847 IdentifierInfo **SelIdents,
4848 unsigned NumSelIdents,
Douglas Gregor36ecb042009-11-17 23:22:23 +00004849 DeclContext *CurContext,
Douglas Gregord36adf52010-09-16 16:06:31 +00004850 VisitedSelectorSet &Selectors,
Douglas Gregorcf544262010-11-17 21:36:08 +00004851 bool AllowSameLength,
Douglas Gregor408be5a2010-08-25 01:08:01 +00004852 ResultBuilder &Results,
4853 bool InOriginalClass = true) {
John McCall0a2c5e22010-08-25 06:19:51 +00004854 typedef CodeCompletionResult Result;
Douglas Gregorb92a4082012-06-12 13:44:08 +00004855 Container = getContainerDef(Container);
Douglas Gregor5824b802013-01-30 06:58:39 +00004856 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4857 bool isRootClass = IFace && !IFace->getSuperClass();
Douglas Gregor36ecb042009-11-17 23:22:23 +00004858 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
4859 MEnd = Container->meth_end();
4860 M != MEnd; ++M) {
Douglas Gregor5824b802013-01-30 06:58:39 +00004861 // The instance methods on the root class can be messaged via the
4862 // metaclass.
4863 if (M->isInstanceMethod() == WantInstanceMethods ||
4864 (isRootClass && !WantInstanceMethods)) {
Douglas Gregord3c68542009-11-19 01:08:35 +00004865 // Check whether the selector identifiers we've been given are a
4866 // subset of the identifiers for this particular method.
David Blaikie581deb32012-06-06 20:45:41 +00004867 if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004868 AllowSameLength))
Douglas Gregord3c68542009-11-19 01:08:35 +00004869 continue;
Douglas Gregor4ad96852009-11-19 07:41:15 +00004870
David Blaikie262bc182012-04-30 02:36:29 +00004871 if (!Selectors.insert(M->getSelector()))
Douglas Gregord36adf52010-09-16 16:06:31 +00004872 continue;
4873
Douglas Gregord1f09b42013-01-31 04:52:16 +00004874 Result R = Result(*M, Results.getBasePriority(*M), 0);
Douglas Gregord3c68542009-11-19 01:08:35 +00004875 R.StartParameter = NumSelIdents;
Douglas Gregor4ad96852009-11-19 07:41:15 +00004876 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor408be5a2010-08-25 01:08:01 +00004877 if (!InOriginalClass)
4878 R.Priority += CCD_InBaseClass;
Douglas Gregord3c68542009-11-19 01:08:35 +00004879 Results.MaybeAddResult(R, CurContext);
4880 }
Douglas Gregor36ecb042009-11-17 23:22:23 +00004881 }
4882
Douglas Gregore396c7b2010-09-16 15:34:59 +00004883 // Visit the protocols of protocols.
4884 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +00004885 if (Protocol->hasDefinition()) {
4886 const ObjCList<ObjCProtocolDecl> &Protocols
4887 = Protocol->getReferencedProtocols();
4888 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4889 E = Protocols.end();
4890 I != E; ++I)
4891 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
4892 NumSelIdents, CurContext, Selectors, AllowSameLength,
4893 Results, false);
4894 }
Douglas Gregore396c7b2010-09-16 15:34:59 +00004895 }
4896
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00004897 if (!IFace || !IFace->hasDefinition())
Douglas Gregor36ecb042009-11-17 23:22:23 +00004898 return;
4899
4900 // Add methods in protocols.
Argyrios Kyrtzidisa5f44412012-03-13 01:09:41 +00004901 for (ObjCInterfaceDecl::protocol_iterator I = IFace->protocol_begin(),
4902 E = IFace->protocol_end();
Douglas Gregor36ecb042009-11-17 23:22:23 +00004903 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00004904 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004905 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004906
4907 // Add methods in categories.
Douglas Gregord3297242013-01-16 23:00:23 +00004908 for (ObjCInterfaceDecl::known_categories_iterator
4909 Cat = IFace->known_categories_begin(),
4910 CatEnd = IFace->known_categories_end();
4911 Cat != CatEnd; ++Cat) {
4912 ObjCCategoryDecl *CatDecl = *Cat;
4913
4914 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004915 NumSelIdents, CurContext, Selectors, AllowSameLength,
4916 Results, InOriginalClass);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004917
4918 // Add a categories protocol methods.
4919 const ObjCList<ObjCProtocolDecl> &Protocols
4920 = CatDecl->getReferencedProtocols();
4921 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4922 E = Protocols.end();
4923 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00004924 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004925 NumSelIdents, CurContext, Selectors, AllowSameLength,
4926 Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004927
4928 // Add methods in category implementations.
4929 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregor4ad96852009-11-19 07:41:15 +00004930 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004931 NumSelIdents, CurContext, Selectors, AllowSameLength,
4932 Results, InOriginalClass);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004933 }
4934
4935 // Add methods in superclass.
4936 if (IFace->getSuperClass())
Douglas Gregor4ad96852009-11-19 07:41:15 +00004937 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Douglas Gregorcf544262010-11-17 21:36:08 +00004938 SelIdents, NumSelIdents, CurContext, Selectors,
4939 AllowSameLength, Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004940
4941 // Add methods in our implementation, if any.
4942 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregor4ad96852009-11-19 07:41:15 +00004943 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004944 NumSelIdents, CurContext, Selectors, AllowSameLength,
4945 Results, InOriginalClass);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004946}
4947
4948
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004949void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregor4ad96852009-11-19 07:41:15 +00004950 // Try to find the interface where getters might live.
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004951 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004952 if (!Class) {
4953 if (ObjCCategoryDecl *Category
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004954 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregor4ad96852009-11-19 07:41:15 +00004955 Class = Category->getClassInterface();
4956
4957 if (!Class)
4958 return;
4959 }
4960
4961 // Find all of the potential getters.
Douglas Gregor218937c2011-02-01 19:23:04 +00004962 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004963 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004964 CodeCompletionContext::CCC_Other);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004965 Results.EnterNewScope();
4966
Douglas Gregord36adf52010-09-16 16:06:31 +00004967 VisitedSelectorSet Selectors;
4968 AddObjCMethods(Class, true, MK_ZeroArgSelector, 0, 0, CurContext, Selectors,
Douglas Gregorcf544262010-11-17 21:36:08 +00004969 /*AllowSameLength=*/true, Results);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004970 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004971 HandleCodeCompleteResults(this, CodeCompleter,
4972 CodeCompletionContext::CCC_Other,
4973 Results.data(),Results.size());
Douglas Gregor4ad96852009-11-19 07:41:15 +00004974}
4975
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004976void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregor4ad96852009-11-19 07:41:15 +00004977 // Try to find the interface where setters might live.
4978 ObjCInterfaceDecl *Class
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004979 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004980 if (!Class) {
4981 if (ObjCCategoryDecl *Category
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004982 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregor4ad96852009-11-19 07:41:15 +00004983 Class = Category->getClassInterface();
4984
4985 if (!Class)
4986 return;
4987 }
4988
4989 // Find all of the potential getters.
Douglas Gregor218937c2011-02-01 19:23:04 +00004990 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004991 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004992 CodeCompletionContext::CCC_Other);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004993 Results.EnterNewScope();
4994
Douglas Gregord36adf52010-09-16 16:06:31 +00004995 VisitedSelectorSet Selectors;
4996 AddObjCMethods(Class, true, MK_OneArgSelector, 0, 0, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00004997 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004998
4999 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005000 HandleCodeCompleteResults(this, CodeCompleter,
5001 CodeCompletionContext::CCC_Other,
5002 Results.data(),Results.size());
Douglas Gregor36ecb042009-11-17 23:22:23 +00005003}
5004
Douglas Gregorafc45782011-02-15 22:19:42 +00005005void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
5006 bool IsParameter) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005007 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005008 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005009 CodeCompletionContext::CCC_Type);
Douglas Gregord32b0222010-08-24 01:06:58 +00005010 Results.EnterNewScope();
5011
5012 // Add context-sensitive, Objective-C parameter-passing keywords.
5013 bool AddedInOut = false;
5014 if ((DS.getObjCDeclQualifier() &
5015 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
5016 Results.AddResult("in");
5017 Results.AddResult("inout");
5018 AddedInOut = true;
5019 }
5020 if ((DS.getObjCDeclQualifier() &
5021 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
5022 Results.AddResult("out");
5023 if (!AddedInOut)
5024 Results.AddResult("inout");
5025 }
5026 if ((DS.getObjCDeclQualifier() &
5027 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
5028 ObjCDeclSpec::DQ_Oneway)) == 0) {
5029 Results.AddResult("bycopy");
5030 Results.AddResult("byref");
5031 Results.AddResult("oneway");
5032 }
5033
Douglas Gregorafc45782011-02-15 22:19:42 +00005034 // If we're completing the return type of an Objective-C method and the
5035 // identifier IBAction refers to a macro, provide a completion item for
5036 // an action, e.g.,
5037 // IBAction)<#selector#>:(id)sender
5038 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
5039 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005040 CodeCompletionBuilder Builder(Results.getAllocator(),
5041 Results.getCodeCompletionTUInfo(),
5042 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorafc45782011-02-15 22:19:42 +00005043 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00005044 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorafc45782011-02-15 22:19:42 +00005045 Builder.AddPlaceholderChunk("selector");
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00005046 Builder.AddChunk(CodeCompletionString::CK_Colon);
5047 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorafc45782011-02-15 22:19:42 +00005048 Builder.AddTextChunk("id");
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00005049 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorafc45782011-02-15 22:19:42 +00005050 Builder.AddTextChunk("sender");
5051 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
5052 }
Douglas Gregor31aa5772013-01-30 07:11:43 +00005053
5054 // If we're completing the return type, provide 'instancetype'.
5055 if (!IsParameter) {
5056 Results.AddResult(CodeCompletionResult("instancetype"));
5057 }
Douglas Gregorafc45782011-02-15 22:19:42 +00005058
Douglas Gregord32b0222010-08-24 01:06:58 +00005059 // Add various builtin type names and specifiers.
5060 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5061 Results.ExitScope();
5062
5063 // Add the various type names
5064 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5065 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5066 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5067 CodeCompleter->includeGlobals());
5068
5069 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00005070 AddMacroResults(PP, Results, false);
Douglas Gregord32b0222010-08-24 01:06:58 +00005071
5072 HandleCodeCompleteResults(this, CodeCompleter,
5073 CodeCompletionContext::CCC_Type,
5074 Results.data(), Results.size());
5075}
5076
Douglas Gregor22f56992010-04-06 19:22:33 +00005077/// \brief When we have an expression with type "id", we may assume
5078/// that it has some more-specific class type based on knowledge of
5079/// common uses of Objective-C. This routine returns that class type,
5080/// or NULL if no better result could be determined.
5081static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregor78edf512010-09-15 16:23:04 +00005082 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor22f56992010-04-06 19:22:33 +00005083 if (!Msg)
5084 return 0;
5085
5086 Selector Sel = Msg->getSelector();
5087 if (Sel.isNull())
5088 return 0;
5089
5090 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5091 if (!Id)
5092 return 0;
5093
5094 ObjCMethodDecl *Method = Msg->getMethodDecl();
5095 if (!Method)
5096 return 0;
5097
5098 // Determine the class that we're sending the message to.
Douglas Gregor04badcf2010-04-21 00:45:42 +00005099 ObjCInterfaceDecl *IFace = 0;
5100 switch (Msg->getReceiverKind()) {
5101 case ObjCMessageExpr::Class:
John McCallc12c5bb2010-05-15 11:32:37 +00005102 if (const ObjCObjectType *ObjType
5103 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5104 IFace = ObjType->getInterface();
Douglas Gregor04badcf2010-04-21 00:45:42 +00005105 break;
5106
5107 case ObjCMessageExpr::Instance: {
5108 QualType T = Msg->getInstanceReceiver()->getType();
5109 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5110 IFace = Ptr->getInterfaceDecl();
5111 break;
5112 }
5113
5114 case ObjCMessageExpr::SuperInstance:
5115 case ObjCMessageExpr::SuperClass:
5116 break;
Douglas Gregor22f56992010-04-06 19:22:33 +00005117 }
5118
5119 if (!IFace)
5120 return 0;
5121
5122 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5123 if (Method->isInstanceMethod())
5124 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5125 .Case("retain", IFace)
John McCallf85e1932011-06-15 23:02:42 +00005126 .Case("strong", IFace)
Douglas Gregor22f56992010-04-06 19:22:33 +00005127 .Case("autorelease", IFace)
5128 .Case("copy", IFace)
5129 .Case("copyWithZone", IFace)
5130 .Case("mutableCopy", IFace)
5131 .Case("mutableCopyWithZone", IFace)
5132 .Case("awakeFromCoder", IFace)
5133 .Case("replacementObjectFromCoder", IFace)
5134 .Case("class", IFace)
5135 .Case("classForCoder", IFace)
5136 .Case("superclass", Super)
5137 .Default(0);
5138
5139 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5140 .Case("new", IFace)
5141 .Case("alloc", IFace)
5142 .Case("allocWithZone", IFace)
5143 .Case("class", IFace)
5144 .Case("superclass", Super)
5145 .Default(0);
5146}
5147
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005148// Add a special completion for a message send to "super", which fills in the
5149// most likely case of forwarding all of our arguments to the superclass
5150// function.
5151///
5152/// \param S The semantic analysis object.
5153///
Dmitri Gribenko70517ca2012-08-23 17:58:28 +00005154/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005155/// the "super" keyword. Otherwise, we just need to provide the arguments.
5156///
5157/// \param SelIdents The identifiers in the selector that have already been
5158/// provided as arguments for a send to "super".
5159///
5160/// \param NumSelIdents The number of identifiers in \p SelIdents.
5161///
5162/// \param Results The set of results to augment.
5163///
5164/// \returns the Objective-C method declaration that would be invoked by
5165/// this "super" completion. If NULL, no completion was added.
5166static ObjCMethodDecl *AddSuperSendCompletion(Sema &S, bool NeedSuperKeyword,
5167 IdentifierInfo **SelIdents,
5168 unsigned NumSelIdents,
5169 ResultBuilder &Results) {
5170 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5171 if (!CurMethod)
5172 return 0;
5173
5174 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5175 if (!Class)
5176 return 0;
5177
5178 // Try to find a superclass method with the same selector.
5179 ObjCMethodDecl *SuperMethod = 0;
Douglas Gregor78bcd912011-02-16 00:51:18 +00005180 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5181 // Check in the class
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005182 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5183 CurMethod->isInstanceMethod());
5184
Douglas Gregor78bcd912011-02-16 00:51:18 +00005185 // Check in categories or class extensions.
5186 if (!SuperMethod) {
Douglas Gregord3297242013-01-16 23:00:23 +00005187 for (ObjCInterfaceDecl::known_categories_iterator
5188 Cat = Class->known_categories_begin(),
5189 CatEnd = Class->known_categories_end();
5190 Cat != CatEnd; ++Cat) {
5191 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregor78bcd912011-02-16 00:51:18 +00005192 CurMethod->isInstanceMethod())))
5193 break;
Douglas Gregord3297242013-01-16 23:00:23 +00005194 }
Douglas Gregor78bcd912011-02-16 00:51:18 +00005195 }
5196 }
5197
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005198 if (!SuperMethod)
5199 return 0;
5200
5201 // Check whether the superclass method has the same signature.
5202 if (CurMethod->param_size() != SuperMethod->param_size() ||
5203 CurMethod->isVariadic() != SuperMethod->isVariadic())
5204 return 0;
5205
5206 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5207 CurPEnd = CurMethod->param_end(),
5208 SuperP = SuperMethod->param_begin();
5209 CurP != CurPEnd; ++CurP, ++SuperP) {
5210 // Make sure the parameter types are compatible.
5211 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5212 (*SuperP)->getType()))
5213 return 0;
5214
5215 // Make sure we have a parameter name to forward!
5216 if (!(*CurP)->getIdentifier())
5217 return 0;
5218 }
5219
5220 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005221 CodeCompletionBuilder Builder(Results.getAllocator(),
5222 Results.getCodeCompletionTUInfo());
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005223
5224 // Give this completion a return type.
Douglas Gregor8987b232011-09-27 23:30:47 +00005225 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5226 Builder);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005227
5228 // If we need the "super" keyword, add it (plus some spacing).
5229 if (NeedSuperKeyword) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005230 Builder.AddTypedTextChunk("super");
5231 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005232 }
5233
5234 Selector Sel = CurMethod->getSelector();
5235 if (Sel.isUnarySelector()) {
5236 if (NeedSuperKeyword)
Douglas Gregordae68752011-02-01 22:57:45 +00005237 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005238 Sel.getNameForSlot(0)));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005239 else
Douglas Gregordae68752011-02-01 22:57:45 +00005240 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005241 Sel.getNameForSlot(0)));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005242 } else {
5243 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5244 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
5245 if (I > NumSelIdents)
Douglas Gregor218937c2011-02-01 19:23:04 +00005246 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005247
5248 if (I < NumSelIdents)
Douglas Gregor218937c2011-02-01 19:23:04 +00005249 Builder.AddInformativeChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00005250 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005251 Sel.getNameForSlot(I) + ":"));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005252 else if (NeedSuperKeyword || I > NumSelIdents) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005253 Builder.AddTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00005254 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005255 Sel.getNameForSlot(I) + ":"));
Douglas Gregordae68752011-02-01 22:57:45 +00005256 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005257 (*CurP)->getIdentifier()->getName()));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005258 } else {
Douglas Gregor218937c2011-02-01 19:23:04 +00005259 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00005260 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005261 Sel.getNameForSlot(I) + ":"));
Douglas Gregordae68752011-02-01 22:57:45 +00005262 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005263 (*CurP)->getIdentifier()->getName()));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005264 }
5265 }
5266 }
5267
Douglas Gregorba103062012-03-27 23:34:16 +00005268 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5269 CCP_SuperCompletion));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005270 return SuperMethod;
5271}
5272
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005273void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall0a2c5e22010-08-25 06:19:51 +00005274 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00005275 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005276 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005277 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith80ad52f2013-01-02 11:42:31 +00005278 getLangOpts().CPlusPlus11
Douglas Gregor81f3bff2012-02-15 15:34:24 +00005279 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5280 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005281
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005282 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5283 Results.EnterNewScope();
Douglas Gregor8071e422010-08-15 06:18:01 +00005284 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5285 CodeCompleter->includeGlobals());
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005286
5287 // If we are in an Objective-C method inside a class that has a superclass,
5288 // add "super" as an option.
5289 if (ObjCMethodDecl *Method = getCurMethodDecl())
5290 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005291 if (Iface->getSuperClass()) {
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005292 Results.AddResult(Result("super"));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005293
5294 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, 0, 0, Results);
5295 }
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005296
Richard Smith80ad52f2013-01-02 11:42:31 +00005297 if (getLangOpts().CPlusPlus11)
Douglas Gregor81f3bff2012-02-15 15:34:24 +00005298 addThisCompletion(*this, Results);
5299
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005300 Results.ExitScope();
5301
5302 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00005303 AddMacroResults(PP, Results, false);
Douglas Gregorcee9ff12010-09-20 22:39:41 +00005304 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005305 Results.data(), Results.size());
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005306
5307}
5308
Douglas Gregor2725ca82010-04-21 19:57:20 +00005309void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
5310 IdentifierInfo **SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005311 unsigned NumSelIdents,
5312 bool AtArgumentExpression) {
Douglas Gregor2725ca82010-04-21 19:57:20 +00005313 ObjCInterfaceDecl *CDecl = 0;
5314 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5315 // Figure out which interface we're in.
5316 CDecl = CurMethod->getClassInterface();
5317 if (!CDecl)
5318 return;
5319
5320 // Find the superclass of this class.
5321 CDecl = CDecl->getSuperClass();
5322 if (!CDecl)
5323 return;
5324
5325 if (CurMethod->isInstanceMethod()) {
5326 // We are inside an instance method, which means that the message
5327 // send [super ...] is actually calling an instance method on the
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005328 // current object.
5329 return CodeCompleteObjCInstanceMessage(S, 0,
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005330 SelIdents, NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005331 AtArgumentExpression,
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005332 CDecl);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005333 }
5334
5335 // Fall through to send to the superclass in CDecl.
5336 } else {
5337 // "super" may be the name of a type or variable. Figure out which
5338 // it is.
Argyrios Kyrtzidis57f8da52013-03-14 22:56:43 +00005339 IdentifierInfo *Super = getSuperIdentifier();
Douglas Gregor2725ca82010-04-21 19:57:20 +00005340 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5341 LookupOrdinaryName);
5342 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5343 // "super" names an interface. Use it.
5344 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCallc12c5bb2010-05-15 11:32:37 +00005345 if (const ObjCObjectType *Iface
5346 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5347 CDecl = Iface->getInterface();
Douglas Gregor2725ca82010-04-21 19:57:20 +00005348 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5349 // "super" names an unresolved type; we can't be more specific.
5350 } else {
5351 // Assume that "super" names some kind of value and parse that way.
5352 CXXScopeSpec SS;
Abramo Bagnarae4b92762012-01-27 09:46:47 +00005353 SourceLocation TemplateKWLoc;
Douglas Gregor2725ca82010-04-21 19:57:20 +00005354 UnqualifiedId id;
5355 id.setIdentifier(Super, SuperLoc);
Abramo Bagnarae4b92762012-01-27 09:46:47 +00005356 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5357 false, false);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005358 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005359 SelIdents, NumSelIdents,
5360 AtArgumentExpression);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005361 }
5362
5363 // Fall through
5364 }
5365
John McCallb3d87482010-08-24 05:47:05 +00005366 ParsedType Receiver;
Douglas Gregor2725ca82010-04-21 19:57:20 +00005367 if (CDecl)
John McCallb3d87482010-08-24 05:47:05 +00005368 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor2725ca82010-04-21 19:57:20 +00005369 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005370 NumSelIdents, AtArgumentExpression,
5371 /*IsSuper=*/true);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005372}
5373
Douglas Gregorb9d77572010-09-21 00:03:25 +00005374/// \brief Given a set of code-completion results for the argument of a message
5375/// send, determine the preferred type (if any) for that argument expression.
5376static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5377 unsigned NumSelIdents) {
5378 typedef CodeCompletionResult Result;
5379 ASTContext &Context = Results.getSema().Context;
5380
5381 QualType PreferredType;
5382 unsigned BestPriority = CCP_Unlikely * 2;
5383 Result *ResultsData = Results.data();
5384 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5385 Result &R = ResultsData[I];
5386 if (R.Kind == Result::RK_Declaration &&
5387 isa<ObjCMethodDecl>(R.Declaration)) {
5388 if (R.Priority <= BestPriority) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00005389 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005390 if (NumSelIdents <= Method->param_size()) {
5391 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
5392 ->getType();
5393 if (R.Priority < BestPriority || PreferredType.isNull()) {
5394 BestPriority = R.Priority;
5395 PreferredType = MyPreferredType;
5396 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5397 MyPreferredType)) {
5398 PreferredType = QualType();
5399 }
5400 }
5401 }
5402 }
5403 }
5404
5405 return PreferredType;
5406}
5407
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005408static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5409 ParsedType Receiver,
5410 IdentifierInfo **SelIdents,
5411 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005412 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005413 bool IsSuper,
5414 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005415 typedef CodeCompletionResult Result;
Douglas Gregor24a069f2009-11-17 17:59:40 +00005416 ObjCInterfaceDecl *CDecl = 0;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005417
Douglas Gregor24a069f2009-11-17 17:59:40 +00005418 // If the given name refers to an interface type, retrieve the
5419 // corresponding declaration.
Douglas Gregor2725ca82010-04-21 19:57:20 +00005420 if (Receiver) {
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005421 QualType T = SemaRef.GetTypeFromParser(Receiver, 0);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005422 if (!T.isNull())
John McCallc12c5bb2010-05-15 11:32:37 +00005423 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5424 CDecl = Interface->getInterface();
Douglas Gregor24a069f2009-11-17 17:59:40 +00005425 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005426
Douglas Gregor36ecb042009-11-17 23:22:23 +00005427 // Add all of the factory methods in this Objective-C class, its protocols,
5428 // superclasses, categories, implementation, etc.
Steve Naroffc4df6d22009-11-07 02:08:14 +00005429 Results.EnterNewScope();
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005430
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005431 // If this is a send-to-super, try to add the special "super" send
5432 // completion.
5433 if (IsSuper) {
5434 if (ObjCMethodDecl *SuperMethod
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005435 = AddSuperSendCompletion(SemaRef, false, SelIdents, NumSelIdents,
5436 Results))
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005437 Results.Ignore(SuperMethod);
5438 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005439
Douglas Gregor265f7492010-08-27 15:29:55 +00005440 // If we're inside an Objective-C method definition, prefer its selector to
5441 // others.
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005442 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregor265f7492010-08-27 15:29:55 +00005443 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005444
Douglas Gregord36adf52010-09-16 16:06:31 +00005445 VisitedSelectorSet Selectors;
Douglas Gregor13438f92010-04-06 16:40:00 +00005446 if (CDecl)
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005447 AddObjCMethods(CDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005448 SemaRef.CurContext, Selectors, AtArgumentExpression,
5449 Results);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005450 else {
Douglas Gregor13438f92010-04-06 16:40:00 +00005451 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005452
Douglas Gregor719770d2010-04-06 17:30:22 +00005453 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00005454 // pool from the AST file.
Axel Naumann0ec56b72012-10-18 19:05:02 +00005455 if (SemaRef.getExternalSource()) {
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005456 for (uint32_t I = 0,
Axel Naumann0ec56b72012-10-18 19:05:02 +00005457 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall76bd1f32010-06-01 09:23:16 +00005458 I != N; ++I) {
Axel Naumann0ec56b72012-10-18 19:05:02 +00005459 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005460 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregor719770d2010-04-06 17:30:22 +00005461 continue;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005462
5463 SemaRef.ReadMethodPool(Sel);
Douglas Gregor719770d2010-04-06 17:30:22 +00005464 }
5465 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005466
5467 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5468 MEnd = SemaRef.MethodPool.end();
Sebastian Redldb9d2142010-08-02 23:18:59 +00005469 M != MEnd; ++M) {
5470 for (ObjCMethodList *MethList = &M->second.second;
5471 MethList && MethList->Method;
Douglas Gregor13438f92010-04-06 16:40:00 +00005472 MethList = MethList->Next) {
5473 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5474 NumSelIdents))
5475 continue;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005476
Douglas Gregord1f09b42013-01-31 04:52:16 +00005477 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Douglas Gregor13438f92010-04-06 16:40:00 +00005478 R.StartParameter = NumSelIdents;
5479 R.AllParametersAreInformative = false;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005480 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor13438f92010-04-06 16:40:00 +00005481 }
5482 }
5483 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005484
5485 Results.ExitScope();
5486}
Douglas Gregor13438f92010-04-06 16:40:00 +00005487
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005488void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
5489 IdentifierInfo **SelIdents,
5490 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005491 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005492 bool IsSuper) {
Douglas Gregore081a612011-07-21 01:05:26 +00005493
5494 QualType T = this->GetTypeFromParser(Receiver);
5495
Douglas Gregor218937c2011-02-01 19:23:04 +00005496 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005497 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregore081a612011-07-21 01:05:26 +00005498 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Douglas Gregor0a47d692011-07-26 15:24:30 +00005499 T, SelIdents, NumSelIdents));
Douglas Gregore081a612011-07-21 01:05:26 +00005500
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005501 AddClassMessageCompletions(*this, S, Receiver, SelIdents, NumSelIdents,
5502 AtArgumentExpression, IsSuper, Results);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005503
5504 // If we're actually at the argument expression (rather than prior to the
5505 // selector), we're actually performing code completion for an expression.
5506 // Determine whether we have a single, best method. If so, we can
5507 // code-complete the expression using the corresponding parameter type as
5508 // our preferred type, improving completion results.
5509 if (AtArgumentExpression) {
5510 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Douglas Gregore081a612011-07-21 01:05:26 +00005511 NumSelIdents);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005512 if (PreferredType.isNull())
5513 CodeCompleteOrdinaryName(S, PCC_Expression);
5514 else
5515 CodeCompleteExpression(S, PreferredType);
5516 return;
5517 }
5518
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005519 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregore081a612011-07-21 01:05:26 +00005520 Results.getCompletionContext(),
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005521 Results.data(), Results.size());
Steve Naroffc4df6d22009-11-07 02:08:14 +00005522}
5523
Richard Trieuf81e5a92011-09-09 02:00:50 +00005524void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Douglas Gregord3c68542009-11-19 01:08:35 +00005525 IdentifierInfo **SelIdents,
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005526 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005527 bool AtArgumentExpression,
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005528 ObjCInterfaceDecl *Super) {
John McCall0a2c5e22010-08-25 06:19:51 +00005529 typedef CodeCompletionResult Result;
Steve Naroffc4df6d22009-11-07 02:08:14 +00005530
5531 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffc4df6d22009-11-07 02:08:14 +00005532
Douglas Gregor36ecb042009-11-17 23:22:23 +00005533 // If necessary, apply function/array conversion to the receiver.
5534 // C99 6.7.5.3p[7,8].
John Wiegley429bb272011-04-08 18:41:53 +00005535 if (RecExpr) {
5536 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5537 if (Conv.isInvalid()) // conversion failed. bail.
5538 return;
5539 RecExpr = Conv.take();
5540 }
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005541 QualType ReceiverType = RecExpr? RecExpr->getType()
5542 : Super? Context.getObjCObjectPointerType(
5543 Context.getObjCInterfaceType(Super))
5544 : Context.getObjCIdType();
Steve Naroffc4df6d22009-11-07 02:08:14 +00005545
Douglas Gregorda892642010-11-08 21:12:30 +00005546 // If we're messaging an expression with type "id" or "Class", check
5547 // whether we know something special about the receiver that allows
5548 // us to assume a more-specific receiver type.
5549 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType())
5550 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5551 if (ReceiverType->isObjCClassType())
5552 return CodeCompleteObjCClassMessage(S,
5553 ParsedType::make(Context.getObjCInterfaceType(IFace)),
5554 SelIdents, NumSelIdents,
5555 AtArgumentExpression, Super);
5556
5557 ReceiverType = Context.getObjCObjectPointerType(
5558 Context.getObjCInterfaceType(IFace));
5559 }
5560
Douglas Gregor36ecb042009-11-17 23:22:23 +00005561 // Build the set of methods we can see.
Douglas Gregor218937c2011-02-01 19:23:04 +00005562 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005563 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregore081a612011-07-21 01:05:26 +00005564 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Douglas Gregor0a47d692011-07-26 15:24:30 +00005565 ReceiverType, SelIdents, NumSelIdents));
Douglas Gregore081a612011-07-21 01:05:26 +00005566
Douglas Gregor36ecb042009-11-17 23:22:23 +00005567 Results.EnterNewScope();
Douglas Gregor22f56992010-04-06 19:22:33 +00005568
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005569 // If this is a send-to-super, try to add the special "super" send
5570 // completion.
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005571 if (Super) {
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005572 if (ObjCMethodDecl *SuperMethod
5573 = AddSuperSendCompletion(*this, false, SelIdents, NumSelIdents,
5574 Results))
5575 Results.Ignore(SuperMethod);
5576 }
5577
Douglas Gregor265f7492010-08-27 15:29:55 +00005578 // If we're inside an Objective-C method definition, prefer its selector to
5579 // others.
5580 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5581 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregor36ecb042009-11-17 23:22:23 +00005582
Douglas Gregord36adf52010-09-16 16:06:31 +00005583 // Keep track of the selectors we've already added.
5584 VisitedSelectorSet Selectors;
5585
Douglas Gregorf74a4192009-11-18 00:06:18 +00005586 // Handle messages to Class. This really isn't a message to an instance
5587 // method, so we treat it the same way we would treat a message send to a
5588 // class method.
5589 if (ReceiverType->isObjCClassType() ||
5590 ReceiverType->isObjCQualifiedClassType()) {
5591 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5592 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Douglas Gregor4ad96852009-11-19 07:41:15 +00005593 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005594 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005595 }
5596 }
5597 // Handle messages to a qualified ID ("id<foo>").
5598 else if (const ObjCObjectPointerType *QualID
5599 = ReceiverType->getAsObjCQualifiedIdType()) {
5600 // Search protocols for instance methods.
5601 for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(),
5602 E = QualID->qual_end();
5603 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00005604 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00005605 Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005606 }
5607 // Handle messages to a pointer to interface type.
5608 else if (const ObjCObjectPointerType *IFacePtr
5609 = ReceiverType->getAsObjCInterfacePointerType()) {
5610 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregor4ad96852009-11-19 07:41:15 +00005611 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005612 NumSelIdents, CurContext, Selectors, AtArgumentExpression,
5613 Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005614
5615 // Search protocols for instance methods.
5616 for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(),
5617 E = IFacePtr->qual_end();
5618 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00005619 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00005620 Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005621 }
Douglas Gregor13438f92010-04-06 16:40:00 +00005622 // Handle messages to "id".
5623 else if (ReceiverType->isObjCIdType()) {
Douglas Gregor719770d2010-04-06 17:30:22 +00005624 // We're messaging "id", so provide all instance methods we know
5625 // about as code-completion results.
5626
5627 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00005628 // pool from the AST file.
Douglas Gregor719770d2010-04-06 17:30:22 +00005629 if (ExternalSource) {
John McCall76bd1f32010-06-01 09:23:16 +00005630 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5631 I != N; ++I) {
5632 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redldb9d2142010-08-02 23:18:59 +00005633 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor719770d2010-04-06 17:30:22 +00005634 continue;
5635
Sebastian Redldb9d2142010-08-02 23:18:59 +00005636 ReadMethodPool(Sel);
Douglas Gregor719770d2010-04-06 17:30:22 +00005637 }
5638 }
5639
Sebastian Redldb9d2142010-08-02 23:18:59 +00005640 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5641 MEnd = MethodPool.end();
5642 M != MEnd; ++M) {
5643 for (ObjCMethodList *MethList = &M->second.first;
5644 MethList && MethList->Method;
Douglas Gregor13438f92010-04-06 16:40:00 +00005645 MethList = MethList->Next) {
5646 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5647 NumSelIdents))
5648 continue;
Douglas Gregord36adf52010-09-16 16:06:31 +00005649
5650 if (!Selectors.insert(MethList->Method->getSelector()))
5651 continue;
5652
Douglas Gregord1f09b42013-01-31 04:52:16 +00005653 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Douglas Gregor13438f92010-04-06 16:40:00 +00005654 R.StartParameter = NumSelIdents;
5655 R.AllParametersAreInformative = false;
5656 Results.MaybeAddResult(R, CurContext);
5657 }
5658 }
5659 }
Steve Naroffc4df6d22009-11-07 02:08:14 +00005660 Results.ExitScope();
Douglas Gregorb9d77572010-09-21 00:03:25 +00005661
5662
5663 // If we're actually at the argument expression (rather than prior to the
5664 // selector), we're actually performing code completion for an expression.
5665 // Determine whether we have a single, best method. If so, we can
5666 // code-complete the expression using the corresponding parameter type as
5667 // our preferred type, improving completion results.
5668 if (AtArgumentExpression) {
5669 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
5670 NumSelIdents);
5671 if (PreferredType.isNull())
5672 CodeCompleteOrdinaryName(S, PCC_Expression);
5673 else
5674 CodeCompleteExpression(S, PreferredType);
5675 return;
5676 }
5677
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005678 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregore081a612011-07-21 01:05:26 +00005679 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005680 Results.data(),Results.size());
Steve Naroffc4df6d22009-11-07 02:08:14 +00005681}
Douglas Gregor55385fe2009-11-18 04:19:12 +00005682
Douglas Gregorfb629412010-08-23 21:17:50 +00005683void Sema::CodeCompleteObjCForCollection(Scope *S,
5684 DeclGroupPtrTy IterationVar) {
5685 CodeCompleteExpressionData Data;
5686 Data.ObjCCollection = true;
5687
5688 if (IterationVar.getAsOpaquePtr()) {
5689 DeclGroupRef DG = IterationVar.getAsVal<DeclGroupRef>();
5690 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5691 if (*I)
5692 Data.IgnoreDecls.push_back(*I);
5693 }
5694 }
5695
5696 CodeCompleteExpression(S, Data);
5697}
5698
Douglas Gregor458433d2010-08-26 15:07:07 +00005699void Sema::CodeCompleteObjCSelector(Scope *S, IdentifierInfo **SelIdents,
5700 unsigned NumSelIdents) {
5701 // If we have an external source, load the entire class method
5702 // pool from the AST file.
5703 if (ExternalSource) {
5704 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5705 I != N; ++I) {
5706 Selector Sel = ExternalSource->GetExternalSelector(I);
5707 if (Sel.isNull() || MethodPool.count(Sel))
5708 continue;
5709
5710 ReadMethodPool(Sel);
5711 }
5712 }
5713
Douglas Gregor218937c2011-02-01 19:23:04 +00005714 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005715 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005716 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor458433d2010-08-26 15:07:07 +00005717 Results.EnterNewScope();
5718 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5719 MEnd = MethodPool.end();
5720 M != MEnd; ++M) {
5721
5722 Selector Sel = M->first;
5723 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents, NumSelIdents))
5724 continue;
5725
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005726 CodeCompletionBuilder Builder(Results.getAllocator(),
5727 Results.getCodeCompletionTUInfo());
Douglas Gregor458433d2010-08-26 15:07:07 +00005728 if (Sel.isUnarySelector()) {
Douglas Gregordae68752011-02-01 22:57:45 +00005729 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005730 Sel.getNameForSlot(0)));
Douglas Gregor218937c2011-02-01 19:23:04 +00005731 Results.AddResult(Builder.TakeString());
Douglas Gregor458433d2010-08-26 15:07:07 +00005732 continue;
5733 }
5734
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005735 std::string Accumulator;
Douglas Gregor458433d2010-08-26 15:07:07 +00005736 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005737 if (I == NumSelIdents) {
5738 if (!Accumulator.empty()) {
Douglas Gregordae68752011-02-01 22:57:45 +00005739 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005740 Accumulator));
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005741 Accumulator.clear();
5742 }
5743 }
5744
Benjamin Kramera0651c52011-07-26 16:59:25 +00005745 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005746 Accumulator += ':';
Douglas Gregor458433d2010-08-26 15:07:07 +00005747 }
Douglas Gregordae68752011-02-01 22:57:45 +00005748 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregor218937c2011-02-01 19:23:04 +00005749 Results.AddResult(Builder.TakeString());
Douglas Gregor458433d2010-08-26 15:07:07 +00005750 }
5751 Results.ExitScope();
5752
5753 HandleCodeCompleteResults(this, CodeCompleter,
5754 CodeCompletionContext::CCC_SelectorName,
5755 Results.data(), Results.size());
5756}
5757
Douglas Gregor55385fe2009-11-18 04:19:12 +00005758/// \brief Add all of the protocol declarations that we find in the given
5759/// (translation unit) context.
5760static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor083128f2009-11-18 04:49:41 +00005761 bool OnlyForwardDeclarations,
Douglas Gregor55385fe2009-11-18 04:19:12 +00005762 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005763 typedef CodeCompletionResult Result;
Douglas Gregor55385fe2009-11-18 04:19:12 +00005764
5765 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5766 DEnd = Ctx->decls_end();
5767 D != DEnd; ++D) {
5768 // Record any protocols we find.
5769 if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*D))
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +00005770 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Douglas Gregord1f09b42013-01-31 04:52:16 +00005771 Results.AddResult(Result(Proto, Results.getBasePriority(Proto), 0),
5772 CurContext, 0, false);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005773 }
5774}
5775
5776void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5777 unsigned NumProtocols) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005778 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005779 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005780 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005781
Douglas Gregor70c23352010-12-09 21:44:02 +00005782 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5783 Results.EnterNewScope();
5784
5785 // Tell the result set to ignore all of the protocols we have
5786 // already seen.
5787 // FIXME: This doesn't work when caching code-completion results.
5788 for (unsigned I = 0; I != NumProtocols; ++I)
5789 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5790 Protocols[I].second))
5791 Results.Ignore(Protocol);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005792
Douglas Gregor70c23352010-12-09 21:44:02 +00005793 // Add all protocols.
5794 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5795 Results);
Douglas Gregor083128f2009-11-18 04:49:41 +00005796
Douglas Gregor70c23352010-12-09 21:44:02 +00005797 Results.ExitScope();
5798 }
5799
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005800 HandleCodeCompleteResults(this, CodeCompleter,
5801 CodeCompletionContext::CCC_ObjCProtocolName,
5802 Results.data(),Results.size());
Douglas Gregor083128f2009-11-18 04:49:41 +00005803}
5804
5805void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005806 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005807 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005808 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor083128f2009-11-18 04:49:41 +00005809
Douglas Gregor70c23352010-12-09 21:44:02 +00005810 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5811 Results.EnterNewScope();
5812
5813 // Add all protocols.
5814 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5815 Results);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005816
Douglas Gregor70c23352010-12-09 21:44:02 +00005817 Results.ExitScope();
5818 }
5819
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005820 HandleCodeCompleteResults(this, CodeCompleter,
5821 CodeCompletionContext::CCC_ObjCProtocolName,
5822 Results.data(),Results.size());
Douglas Gregor55385fe2009-11-18 04:19:12 +00005823}
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005824
5825/// \brief Add all of the Objective-C interface declarations that we find in
5826/// the given (translation unit) context.
5827static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5828 bool OnlyForwardDeclarations,
5829 bool OnlyUnimplemented,
5830 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005831 typedef CodeCompletionResult Result;
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005832
5833 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5834 DEnd = Ctx->decls_end();
5835 D != DEnd; ++D) {
Douglas Gregordeacbdc2010-08-11 12:19:30 +00005836 // Record any interfaces we find.
5837 if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*D))
Douglas Gregor7723fec2011-12-15 20:29:51 +00005838 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregordeacbdc2010-08-11 12:19:30 +00005839 (!OnlyUnimplemented || !Class->getImplementation()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00005840 Results.AddResult(Result(Class, Results.getBasePriority(Class), 0),
5841 CurContext, 0, false);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005842 }
5843}
5844
5845void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005846 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005847 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005848 CodeCompletionContext::CCC_Other);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005849 Results.EnterNewScope();
5850
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005851 if (CodeCompleter->includeGlobals()) {
5852 // Add all classes.
5853 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5854 false, Results);
5855 }
5856
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005857 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005858
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005859 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005860 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005861 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005862}
5863
Douglas Gregorc83c6872010-04-15 22:33:43 +00005864void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5865 SourceLocation ClassNameLoc) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005866 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005867 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005868 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005869 Results.EnterNewScope();
5870
5871 // Make sure that we ignore the class we're currently defining.
5872 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005873 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005874 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005875 Results.Ignore(CurClass);
5876
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005877 if (CodeCompleter->includeGlobals()) {
5878 // Add all classes.
5879 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5880 false, Results);
5881 }
5882
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005883 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005884
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005885 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005886 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005887 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005888}
5889
5890void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005891 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005892 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005893 CodeCompletionContext::CCC_Other);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005894 Results.EnterNewScope();
5895
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005896 if (CodeCompleter->includeGlobals()) {
5897 // Add all unimplemented classes.
5898 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5899 true, Results);
5900 }
5901
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005902 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005903
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005904 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005905 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005906 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005907}
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005908
5909void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorc83c6872010-04-15 22:33:43 +00005910 IdentifierInfo *ClassName,
5911 SourceLocation ClassNameLoc) {
John McCall0a2c5e22010-08-25 06:19:51 +00005912 typedef CodeCompletionResult Result;
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005913
Douglas Gregor218937c2011-02-01 19:23:04 +00005914 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005915 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00005916 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005917
5918 // Ignore any categories we find that have already been implemented by this
5919 // interface.
5920 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5921 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005922 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregord3297242013-01-16 23:00:23 +00005923 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
5924 for (ObjCInterfaceDecl::visible_categories_iterator
5925 Cat = Class->visible_categories_begin(),
5926 CatEnd = Class->visible_categories_end();
5927 Cat != CatEnd; ++Cat) {
5928 CategoryNames.insert(Cat->getIdentifier());
5929 }
5930 }
5931
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005932 // Add all of the categories we know about.
5933 Results.EnterNewScope();
5934 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
5935 for (DeclContext::decl_iterator D = TU->decls_begin(),
5936 DEnd = TU->decls_end();
5937 D != DEnd; ++D)
5938 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(*D))
5939 if (CategoryNames.insert(Category->getIdentifier()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00005940 Results.AddResult(Result(Category, Results.getBasePriority(Category),0),
5941 CurContext, 0, false);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005942 Results.ExitScope();
5943
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005944 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00005945 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005946 Results.data(),Results.size());
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005947}
5948
5949void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorc83c6872010-04-15 22:33:43 +00005950 IdentifierInfo *ClassName,
5951 SourceLocation ClassNameLoc) {
John McCall0a2c5e22010-08-25 06:19:51 +00005952 typedef CodeCompletionResult Result;
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005953
5954 // Find the corresponding interface. If we couldn't find the interface, the
5955 // program itself is ill-formed. However, we'll try to be helpful still by
5956 // providing the list of all of the categories we know about.
5957 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005958 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005959 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5960 if (!Class)
Douglas Gregorc83c6872010-04-15 22:33:43 +00005961 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005962
Douglas Gregor218937c2011-02-01 19:23:04 +00005963 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005964 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00005965 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005966
5967 // Add all of the categories that have have corresponding interface
5968 // declarations in this class and any of its superclasses, except for
5969 // already-implemented categories in the class itself.
5970 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5971 Results.EnterNewScope();
5972 bool IgnoreImplemented = true;
5973 while (Class) {
Douglas Gregord3297242013-01-16 23:00:23 +00005974 for (ObjCInterfaceDecl::visible_categories_iterator
5975 Cat = Class->visible_categories_begin(),
5976 CatEnd = Class->visible_categories_end();
5977 Cat != CatEnd; ++Cat) {
5978 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
5979 CategoryNames.insert(Cat->getIdentifier()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00005980 Results.AddResult(Result(*Cat, Results.getBasePriority(*Cat), 0),
5981 CurContext, 0, false);
Douglas Gregord3297242013-01-16 23:00:23 +00005982 }
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005983
5984 Class = Class->getSuperClass();
5985 IgnoreImplemented = false;
5986 }
5987 Results.ExitScope();
5988
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005989 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00005990 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005991 Results.data(),Results.size());
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005992}
Douglas Gregor322328b2009-11-18 22:32:06 +00005993
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005994void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005995 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005996 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005997 CodeCompletionContext::CCC_Other);
Douglas Gregor322328b2009-11-18 22:32:06 +00005998
5999 // Figure out where this @synthesize lives.
6000 ObjCContainerDecl *Container
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006001 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor322328b2009-11-18 22:32:06 +00006002 if (!Container ||
6003 (!isa<ObjCImplementationDecl>(Container) &&
6004 !isa<ObjCCategoryImplDecl>(Container)))
6005 return;
6006
6007 // Ignore any properties that have already been implemented.
Douglas Gregorb92a4082012-06-12 13:44:08 +00006008 Container = getContainerDef(Container);
6009 for (DeclContext::decl_iterator D = Container->decls_begin(),
Douglas Gregor322328b2009-11-18 22:32:06 +00006010 DEnd = Container->decls_end();
6011 D != DEnd; ++D)
6012 if (ObjCPropertyImplDecl *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(*D))
6013 Results.Ignore(PropertyImpl->getPropertyDecl());
6014
6015 // Add any properties that we find.
Douglas Gregor73449212010-12-09 23:01:55 +00006016 AddedPropertiesSet AddedProperties;
Douglas Gregor322328b2009-11-18 22:32:06 +00006017 Results.EnterNewScope();
6018 if (ObjCImplementationDecl *ClassImpl
6019 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor4b81cde2011-05-05 15:50:42 +00006020 AddObjCProperties(ClassImpl->getClassInterface(), false,
6021 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00006022 AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00006023 else
6024 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor4b81cde2011-05-05 15:50:42 +00006025 false, /*AllowNullaryMethods=*/false, CurContext,
6026 AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00006027 Results.ExitScope();
6028
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006029 HandleCodeCompleteResults(this, CodeCompleter,
6030 CodeCompletionContext::CCC_Other,
6031 Results.data(),Results.size());
Douglas Gregor322328b2009-11-18 22:32:06 +00006032}
6033
6034void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006035 IdentifierInfo *PropertyName) {
John McCall0a2c5e22010-08-25 06:19:51 +00006036 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00006037 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006038 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00006039 CodeCompletionContext::CCC_Other);
Douglas Gregor322328b2009-11-18 22:32:06 +00006040
6041 // Figure out where this @synthesize lives.
6042 ObjCContainerDecl *Container
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006043 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor322328b2009-11-18 22:32:06 +00006044 if (!Container ||
6045 (!isa<ObjCImplementationDecl>(Container) &&
6046 !isa<ObjCCategoryImplDecl>(Container)))
6047 return;
6048
6049 // Figure out which interface we're looking into.
6050 ObjCInterfaceDecl *Class = 0;
6051 if (ObjCImplementationDecl *ClassImpl
6052 = dyn_cast<ObjCImplementationDecl>(Container))
6053 Class = ClassImpl->getClassInterface();
6054 else
6055 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
6056 ->getClassInterface();
6057
Douglas Gregore8426052011-04-18 14:40:46 +00006058 // Determine the type of the property we're synthesizing.
6059 QualType PropertyType = Context.getObjCIdType();
6060 if (Class) {
6061 if (ObjCPropertyDecl *Property
6062 = Class->FindPropertyDeclaration(PropertyName)) {
6063 PropertyType
6064 = Property->getType().getNonReferenceType().getUnqualifiedType();
6065
6066 // Give preference to ivars
6067 Results.setPreferredType(PropertyType);
6068 }
6069 }
6070
Douglas Gregor322328b2009-11-18 22:32:06 +00006071 // Add all of the instance variables in this class and its superclasses.
6072 Results.EnterNewScope();
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006073 bool SawSimilarlyNamedIvar = false;
6074 std::string NameWithPrefix;
6075 NameWithPrefix += '_';
Benjamin Kramera0651c52011-07-26 16:59:25 +00006076 NameWithPrefix += PropertyName->getName();
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006077 std::string NameWithSuffix = PropertyName->getName().str();
6078 NameWithSuffix += '_';
Douglas Gregor322328b2009-11-18 22:32:06 +00006079 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006080 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
6081 Ivar = Ivar->getNextIvar()) {
Douglas Gregord1f09b42013-01-31 04:52:16 +00006082 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), 0),
6083 CurContext, 0, false);
Douglas Gregore8426052011-04-18 14:40:46 +00006084
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006085 // Determine whether we've seen an ivar with a name similar to the
6086 // property.
Douglas Gregore8426052011-04-18 14:40:46 +00006087 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006088 NameWithPrefix == Ivar->getName() ||
Douglas Gregore8426052011-04-18 14:40:46 +00006089 NameWithSuffix == Ivar->getName())) {
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006090 SawSimilarlyNamedIvar = true;
Douglas Gregore8426052011-04-18 14:40:46 +00006091
6092 // Reduce the priority of this result by one, to give it a slight
6093 // advantage over other results whose names don't match so closely.
6094 if (Results.size() &&
6095 Results.data()[Results.size() - 1].Kind
6096 == CodeCompletionResult::RK_Declaration &&
6097 Results.data()[Results.size() - 1].Declaration == Ivar)
6098 Results.data()[Results.size() - 1].Priority--;
6099 }
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006100 }
Douglas Gregor322328b2009-11-18 22:32:06 +00006101 }
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006102
6103 if (!SawSimilarlyNamedIvar) {
6104 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregore8426052011-04-18 14:40:46 +00006105 // an ivar of the appropriate type.
6106 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006107 typedef CodeCompletionResult Result;
6108 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006109 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6110 Priority,CXAvailability_Available);
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006111
Douglas Gregor8987b232011-09-27 23:30:47 +00006112 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregore8426052011-04-18 14:40:46 +00006113 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006114 Policy, Allocator));
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006115 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6116 Results.AddResult(Result(Builder.TakeString(), Priority,
6117 CXCursor_ObjCIvarDecl));
6118 }
6119
Douglas Gregor322328b2009-11-18 22:32:06 +00006120 Results.ExitScope();
6121
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006122 HandleCodeCompleteResults(this, CodeCompleter,
6123 CodeCompletionContext::CCC_Other,
6124 Results.data(),Results.size());
Douglas Gregor322328b2009-11-18 22:32:06 +00006125}
Douglas Gregore8f5a172010-04-07 00:21:17 +00006126
Douglas Gregor408be5a2010-08-25 01:08:01 +00006127// Mapping from selectors to the methods that implement that selector, along
6128// with the "in original class" flag.
6129typedef llvm::DenseMap<Selector, std::pair<ObjCMethodDecl *, bool> >
6130 KnownMethodsMap;
Douglas Gregore8f5a172010-04-07 00:21:17 +00006131
6132/// \brief Find all of the methods that reside in the given container
6133/// (and its superclasses, protocols, etc.) that meet the given
6134/// criteria. Insert those methods into the map of known methods,
6135/// indexed by selector so they can be easily found.
6136static void FindImplementableMethods(ASTContext &Context,
6137 ObjCContainerDecl *Container,
6138 bool WantInstanceMethods,
6139 QualType ReturnType,
Douglas Gregor408be5a2010-08-25 01:08:01 +00006140 KnownMethodsMap &KnownMethods,
6141 bool InOriginalClass = true) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006142 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregorb92a4082012-06-12 13:44:08 +00006143 // Make sure we have a definition; that's what we'll walk.
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00006144 if (!IFace->hasDefinition())
6145 return;
Douglas Gregorb92a4082012-06-12 13:44:08 +00006146
6147 IFace = IFace->getDefinition();
6148 Container = IFace;
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00006149
Douglas Gregore8f5a172010-04-07 00:21:17 +00006150 const ObjCList<ObjCProtocolDecl> &Protocols
6151 = IFace->getReferencedProtocols();
6152 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregorea766182010-10-18 18:21:28 +00006153 E = Protocols.end();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006154 I != E; ++I)
6155 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00006156 KnownMethods, InOriginalClass);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006157
Douglas Gregorea766182010-10-18 18:21:28 +00006158 // Add methods from any class extensions and categories.
Douglas Gregord3297242013-01-16 23:00:23 +00006159 for (ObjCInterfaceDecl::visible_categories_iterator
6160 Cat = IFace->visible_categories_begin(),
6161 CatEnd = IFace->visible_categories_end();
6162 Cat != CatEnd; ++Cat) {
6163 FindImplementableMethods(Context, *Cat, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00006164 KnownMethods, false);
Douglas Gregord3297242013-01-16 23:00:23 +00006165 }
6166
Douglas Gregorea766182010-10-18 18:21:28 +00006167 // Visit the superclass.
6168 if (IFace->getSuperClass())
6169 FindImplementableMethods(Context, IFace->getSuperClass(),
6170 WantInstanceMethods, ReturnType,
6171 KnownMethods, false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006172 }
6173
6174 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6175 // Recurse into protocols.
6176 const ObjCList<ObjCProtocolDecl> &Protocols
6177 = Category->getReferencedProtocols();
6178 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregorea766182010-10-18 18:21:28 +00006179 E = Protocols.end();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006180 I != E; ++I)
6181 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00006182 KnownMethods, InOriginalClass);
6183
6184 // If this category is the original class, jump to the interface.
6185 if (InOriginalClass && Category->getClassInterface())
6186 FindImplementableMethods(Context, Category->getClassInterface(),
6187 WantInstanceMethods, ReturnType, KnownMethods,
6188 false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006189 }
6190
6191 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregorb92a4082012-06-12 13:44:08 +00006192 // Make sure we have a definition; that's what we'll walk.
6193 if (!Protocol->hasDefinition())
6194 return;
6195 Protocol = Protocol->getDefinition();
6196 Container = Protocol;
6197
6198 // Recurse into protocols.
6199 const ObjCList<ObjCProtocolDecl> &Protocols
6200 = Protocol->getReferencedProtocols();
6201 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6202 E = Protocols.end();
6203 I != E; ++I)
6204 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6205 KnownMethods, false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006206 }
6207
6208 // Add methods in this container. This operation occurs last because
6209 // we want the methods from this container to override any methods
6210 // we've previously seen with the same selector.
6211 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
6212 MEnd = Container->meth_end();
6213 M != MEnd; ++M) {
David Blaikie262bc182012-04-30 02:36:29 +00006214 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006215 if (!ReturnType.isNull() &&
David Blaikie262bc182012-04-30 02:36:29 +00006216 !Context.hasSameUnqualifiedType(ReturnType, M->getResultType()))
Douglas Gregore8f5a172010-04-07 00:21:17 +00006217 continue;
6218
David Blaikie581deb32012-06-06 20:45:41 +00006219 KnownMethods[M->getSelector()] = std::make_pair(*M, InOriginalClass);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006220 }
6221 }
6222}
6223
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006224/// \brief Add the parenthesized return or parameter type chunk to a code
6225/// completion string.
6226static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor90f5f472012-04-10 18:35:07 +00006227 unsigned ObjCDeclQuals,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006228 ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006229 const PrintingPolicy &Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006230 CodeCompletionBuilder &Builder) {
6231 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor90f5f472012-04-10 18:35:07 +00006232 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6233 if (!Quals.empty())
6234 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor8987b232011-09-27 23:30:47 +00006235 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006236 Builder.getAllocator()));
6237 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6238}
6239
6240/// \brief Determine whether the given class is or inherits from a class by
6241/// the given name.
6242static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner5f9e2722011-07-23 10:55:15 +00006243 StringRef Name) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006244 if (!Class)
6245 return false;
6246
6247 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6248 return true;
6249
6250 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6251}
6252
6253/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6254/// Key-Value Observing (KVO).
6255static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6256 bool IsInstanceMethod,
6257 QualType ReturnType,
6258 ASTContext &Context,
Douglas Gregore74c25c2011-05-04 23:50:46 +00006259 VisitedSelectorSet &KnownSelectors,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006260 ResultBuilder &Results) {
6261 IdentifierInfo *PropName = Property->getIdentifier();
6262 if (!PropName || PropName->getLength() == 0)
6263 return;
6264
Douglas Gregor8987b232011-09-27 23:30:47 +00006265 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6266
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006267 // Builder that will create each code completion.
6268 typedef CodeCompletionResult Result;
6269 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006270 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006271
6272 // The selector table.
6273 SelectorTable &Selectors = Context.Selectors;
6274
6275 // The property name, copied into the code completion allocation region
6276 // on demand.
6277 struct KeyHolder {
6278 CodeCompletionAllocator &Allocator;
Chris Lattner5f9e2722011-07-23 10:55:15 +00006279 StringRef Key;
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006280 const char *CopiedKey;
6281
Chris Lattner5f9e2722011-07-23 10:55:15 +00006282 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006283 : Allocator(Allocator), Key(Key), CopiedKey(0) { }
6284
6285 operator const char *() {
6286 if (CopiedKey)
6287 return CopiedKey;
6288
6289 return CopiedKey = Allocator.CopyString(Key);
6290 }
6291 } Key(Allocator, PropName->getName());
6292
6293 // The uppercased name of the property name.
6294 std::string UpperKey = PropName->getName();
6295 if (!UpperKey.empty())
Jordan Rose223f0ff2013-02-09 10:09:43 +00006296 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006297
6298 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6299 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6300 Property->getType());
6301 bool ReturnTypeMatchesVoid
6302 = ReturnType.isNull() || ReturnType->isVoidType();
6303
6304 // Add the normal accessor -(type)key.
6305 if (IsInstanceMethod &&
Douglas Gregore74c25c2011-05-04 23:50:46 +00006306 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006307 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6308 if (ReturnType.isNull())
Douglas Gregor90f5f472012-04-10 18:35:07 +00006309 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6310 Context, Policy, Builder);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006311
6312 Builder.AddTypedTextChunk(Key);
6313 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6314 CXCursor_ObjCInstanceMethodDecl));
6315 }
6316
6317 // If we have an integral or boolean property (or the user has provided
6318 // an integral or boolean return type), add the accessor -(type)isKey.
6319 if (IsInstanceMethod &&
6320 ((!ReturnType.isNull() &&
6321 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6322 (ReturnType.isNull() &&
6323 (Property->getType()->isIntegerType() ||
6324 Property->getType()->isBooleanType())))) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006325 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006326 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006327 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006328 if (ReturnType.isNull()) {
6329 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6330 Builder.AddTextChunk("BOOL");
6331 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6332 }
6333
6334 Builder.AddTypedTextChunk(
6335 Allocator.CopyString(SelectorId->getName()));
6336 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6337 CXCursor_ObjCInstanceMethodDecl));
6338 }
6339 }
6340
6341 // Add the normal mutator.
6342 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6343 !Property->getSetterMethodDecl()) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006344 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006345 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006346 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006347 if (ReturnType.isNull()) {
6348 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6349 Builder.AddTextChunk("void");
6350 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6351 }
6352
6353 Builder.AddTypedTextChunk(
6354 Allocator.CopyString(SelectorId->getName()));
6355 Builder.AddTypedTextChunk(":");
Douglas Gregor90f5f472012-04-10 18:35:07 +00006356 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6357 Context, Policy, Builder);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006358 Builder.AddTextChunk(Key);
6359 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6360 CXCursor_ObjCInstanceMethodDecl));
6361 }
6362 }
6363
6364 // Indexed and unordered accessors
6365 unsigned IndexedGetterPriority = CCP_CodePattern;
6366 unsigned IndexedSetterPriority = CCP_CodePattern;
6367 unsigned UnorderedGetterPriority = CCP_CodePattern;
6368 unsigned UnorderedSetterPriority = CCP_CodePattern;
6369 if (const ObjCObjectPointerType *ObjCPointer
6370 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6371 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6372 // If this interface type is not provably derived from a known
6373 // collection, penalize the corresponding completions.
6374 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6375 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6376 if (!InheritsFromClassNamed(IFace, "NSArray"))
6377 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6378 }
6379
6380 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6381 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6382 if (!InheritsFromClassNamed(IFace, "NSSet"))
6383 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6384 }
6385 }
6386 } else {
6387 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6388 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6389 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6390 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6391 }
6392
6393 // Add -(NSUInteger)countOf<key>
6394 if (IsInstanceMethod &&
6395 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006396 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006397 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006398 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006399 if (ReturnType.isNull()) {
6400 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6401 Builder.AddTextChunk("NSUInteger");
6402 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6403 }
6404
6405 Builder.AddTypedTextChunk(
6406 Allocator.CopyString(SelectorId->getName()));
6407 Results.AddResult(Result(Builder.TakeString(),
6408 std::min(IndexedGetterPriority,
6409 UnorderedGetterPriority),
6410 CXCursor_ObjCInstanceMethodDecl));
6411 }
6412 }
6413
6414 // Indexed getters
6415 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6416 if (IsInstanceMethod &&
6417 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor62041592011-02-17 03:19:26 +00006418 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006419 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006420 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006421 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006422 if (ReturnType.isNull()) {
6423 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6424 Builder.AddTextChunk("id");
6425 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6426 }
6427
6428 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6429 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6430 Builder.AddTextChunk("NSUInteger");
6431 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6432 Builder.AddTextChunk("index");
6433 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6434 CXCursor_ObjCInstanceMethodDecl));
6435 }
6436 }
6437
6438 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6439 if (IsInstanceMethod &&
6440 (ReturnType.isNull() ||
6441 (ReturnType->isObjCObjectPointerType() &&
6442 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6443 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6444 ->getName() == "NSArray"))) {
Douglas Gregor62041592011-02-17 03:19:26 +00006445 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006446 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006447 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006448 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006449 if (ReturnType.isNull()) {
6450 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6451 Builder.AddTextChunk("NSArray *");
6452 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6453 }
6454
6455 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6456 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6457 Builder.AddTextChunk("NSIndexSet *");
6458 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6459 Builder.AddTextChunk("indexes");
6460 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6461 CXCursor_ObjCInstanceMethodDecl));
6462 }
6463 }
6464
6465 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6466 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006467 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006468 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006469 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006470 &Context.Idents.get("range")
6471 };
6472
Douglas Gregore74c25c2011-05-04 23:50:46 +00006473 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006474 if (ReturnType.isNull()) {
6475 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6476 Builder.AddTextChunk("void");
6477 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6478 }
6479
6480 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6481 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6482 Builder.AddPlaceholderChunk("object-type");
6483 Builder.AddTextChunk(" **");
6484 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6485 Builder.AddTextChunk("buffer");
6486 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6487 Builder.AddTypedTextChunk("range:");
6488 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6489 Builder.AddTextChunk("NSRange");
6490 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6491 Builder.AddTextChunk("inRange");
6492 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6493 CXCursor_ObjCInstanceMethodDecl));
6494 }
6495 }
6496
6497 // Mutable indexed accessors
6498
6499 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6500 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006501 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006502 IdentifierInfo *SelectorIds[2] = {
6503 &Context.Idents.get("insertObject"),
Douglas Gregor62041592011-02-17 03:19:26 +00006504 &Context.Idents.get(SelectorName)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006505 };
6506
Douglas Gregore74c25c2011-05-04 23:50:46 +00006507 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006508 if (ReturnType.isNull()) {
6509 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6510 Builder.AddTextChunk("void");
6511 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6512 }
6513
6514 Builder.AddTypedTextChunk("insertObject:");
6515 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6516 Builder.AddPlaceholderChunk("object-type");
6517 Builder.AddTextChunk(" *");
6518 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6519 Builder.AddTextChunk("object");
6520 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6521 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6522 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6523 Builder.AddPlaceholderChunk("NSUInteger");
6524 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6525 Builder.AddTextChunk("index");
6526 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6527 CXCursor_ObjCInstanceMethodDecl));
6528 }
6529 }
6530
6531 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6532 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006533 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006534 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006535 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006536 &Context.Idents.get("atIndexes")
6537 };
6538
Douglas Gregore74c25c2011-05-04 23:50:46 +00006539 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006540 if (ReturnType.isNull()) {
6541 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6542 Builder.AddTextChunk("void");
6543 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6544 }
6545
6546 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6547 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6548 Builder.AddTextChunk("NSArray *");
6549 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6550 Builder.AddTextChunk("array");
6551 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6552 Builder.AddTypedTextChunk("atIndexes:");
6553 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6554 Builder.AddPlaceholderChunk("NSIndexSet *");
6555 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6556 Builder.AddTextChunk("indexes");
6557 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6558 CXCursor_ObjCInstanceMethodDecl));
6559 }
6560 }
6561
6562 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6563 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006564 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006565 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006566 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006567 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006568 if (ReturnType.isNull()) {
6569 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6570 Builder.AddTextChunk("void");
6571 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6572 }
6573
6574 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6575 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6576 Builder.AddTextChunk("NSUInteger");
6577 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6578 Builder.AddTextChunk("index");
6579 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6580 CXCursor_ObjCInstanceMethodDecl));
6581 }
6582 }
6583
6584 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6585 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006586 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006587 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006588 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006589 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006590 if (ReturnType.isNull()) {
6591 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6592 Builder.AddTextChunk("void");
6593 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6594 }
6595
6596 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6597 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6598 Builder.AddTextChunk("NSIndexSet *");
6599 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6600 Builder.AddTextChunk("indexes");
6601 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6602 CXCursor_ObjCInstanceMethodDecl));
6603 }
6604 }
6605
6606 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6607 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006608 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006609 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006610 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006611 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006612 &Context.Idents.get("withObject")
6613 };
6614
Douglas Gregore74c25c2011-05-04 23:50:46 +00006615 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006616 if (ReturnType.isNull()) {
6617 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6618 Builder.AddTextChunk("void");
6619 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6620 }
6621
6622 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6623 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6624 Builder.AddPlaceholderChunk("NSUInteger");
6625 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6626 Builder.AddTextChunk("index");
6627 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6628 Builder.AddTypedTextChunk("withObject:");
6629 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6630 Builder.AddTextChunk("id");
6631 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6632 Builder.AddTextChunk("object");
6633 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6634 CXCursor_ObjCInstanceMethodDecl));
6635 }
6636 }
6637
6638 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6639 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006640 std::string SelectorName1
Chris Lattner5f9e2722011-07-23 10:55:15 +00006641 = (Twine("replace") + UpperKey + "AtIndexes").str();
6642 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006643 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006644 &Context.Idents.get(SelectorName1),
6645 &Context.Idents.get(SelectorName2)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006646 };
6647
Douglas Gregore74c25c2011-05-04 23:50:46 +00006648 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006649 if (ReturnType.isNull()) {
6650 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6651 Builder.AddTextChunk("void");
6652 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6653 }
6654
6655 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6656 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6657 Builder.AddPlaceholderChunk("NSIndexSet *");
6658 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6659 Builder.AddTextChunk("indexes");
6660 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6661 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6662 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6663 Builder.AddTextChunk("NSArray *");
6664 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6665 Builder.AddTextChunk("array");
6666 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6667 CXCursor_ObjCInstanceMethodDecl));
6668 }
6669 }
6670
6671 // Unordered getters
6672 // - (NSEnumerator *)enumeratorOfKey
6673 if (IsInstanceMethod &&
6674 (ReturnType.isNull() ||
6675 (ReturnType->isObjCObjectPointerType() &&
6676 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6677 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6678 ->getName() == "NSEnumerator"))) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006679 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006680 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006681 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006682 if (ReturnType.isNull()) {
6683 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6684 Builder.AddTextChunk("NSEnumerator *");
6685 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6686 }
6687
6688 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6689 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6690 CXCursor_ObjCInstanceMethodDecl));
6691 }
6692 }
6693
6694 // - (type *)memberOfKey:(type *)object
6695 if (IsInstanceMethod &&
6696 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006697 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006698 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006699 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006700 if (ReturnType.isNull()) {
6701 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6702 Builder.AddPlaceholderChunk("object-type");
6703 Builder.AddTextChunk(" *");
6704 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6705 }
6706
6707 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6708 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6709 if (ReturnType.isNull()) {
6710 Builder.AddPlaceholderChunk("object-type");
6711 Builder.AddTextChunk(" *");
6712 } else {
6713 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006714 Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006715 Builder.getAllocator()));
6716 }
6717 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6718 Builder.AddTextChunk("object");
6719 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6720 CXCursor_ObjCInstanceMethodDecl));
6721 }
6722 }
6723
6724 // Mutable unordered accessors
6725 // - (void)addKeyObject:(type *)object
6726 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006727 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006728 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006729 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006730 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006731 if (ReturnType.isNull()) {
6732 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6733 Builder.AddTextChunk("void");
6734 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6735 }
6736
6737 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6738 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6739 Builder.AddPlaceholderChunk("object-type");
6740 Builder.AddTextChunk(" *");
6741 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6742 Builder.AddTextChunk("object");
6743 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6744 CXCursor_ObjCInstanceMethodDecl));
6745 }
6746 }
6747
6748 // - (void)addKey:(NSSet *)objects
6749 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006750 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006751 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006752 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006753 if (ReturnType.isNull()) {
6754 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6755 Builder.AddTextChunk("void");
6756 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6757 }
6758
6759 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6760 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6761 Builder.AddTextChunk("NSSet *");
6762 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6763 Builder.AddTextChunk("objects");
6764 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6765 CXCursor_ObjCInstanceMethodDecl));
6766 }
6767 }
6768
6769 // - (void)removeKeyObject:(type *)object
6770 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006771 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006772 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006773 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006774 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006775 if (ReturnType.isNull()) {
6776 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6777 Builder.AddTextChunk("void");
6778 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6779 }
6780
6781 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6782 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6783 Builder.AddPlaceholderChunk("object-type");
6784 Builder.AddTextChunk(" *");
6785 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6786 Builder.AddTextChunk("object");
6787 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6788 CXCursor_ObjCInstanceMethodDecl));
6789 }
6790 }
6791
6792 // - (void)removeKey:(NSSet *)objects
6793 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006794 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006795 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006796 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006797 if (ReturnType.isNull()) {
6798 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6799 Builder.AddTextChunk("void");
6800 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6801 }
6802
6803 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6804 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6805 Builder.AddTextChunk("NSSet *");
6806 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6807 Builder.AddTextChunk("objects");
6808 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6809 CXCursor_ObjCInstanceMethodDecl));
6810 }
6811 }
6812
6813 // - (void)intersectKey:(NSSet *)objects
6814 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006815 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006816 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006817 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006818 if (ReturnType.isNull()) {
6819 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6820 Builder.AddTextChunk("void");
6821 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6822 }
6823
6824 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6825 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6826 Builder.AddTextChunk("NSSet *");
6827 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6828 Builder.AddTextChunk("objects");
6829 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6830 CXCursor_ObjCInstanceMethodDecl));
6831 }
6832 }
6833
6834 // Key-Value Observing
6835 // + (NSSet *)keyPathsForValuesAffectingKey
6836 if (!IsInstanceMethod &&
6837 (ReturnType.isNull() ||
6838 (ReturnType->isObjCObjectPointerType() &&
6839 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6840 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6841 ->getName() == "NSSet"))) {
Douglas Gregor62041592011-02-17 03:19:26 +00006842 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006843 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006844 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006845 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006846 if (ReturnType.isNull()) {
6847 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6848 Builder.AddTextChunk("NSSet *");
6849 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6850 }
6851
6852 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6853 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor3f828d12011-06-02 04:02:27 +00006854 CXCursor_ObjCClassMethodDecl));
6855 }
6856 }
6857
6858 // + (BOOL)automaticallyNotifiesObserversForKey
6859 if (!IsInstanceMethod &&
6860 (ReturnType.isNull() ||
6861 ReturnType->isIntegerType() ||
6862 ReturnType->isBooleanType())) {
6863 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006864 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor3f828d12011-06-02 04:02:27 +00006865 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6866 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6867 if (ReturnType.isNull()) {
6868 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6869 Builder.AddTextChunk("BOOL");
6870 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6871 }
6872
6873 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6874 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6875 CXCursor_ObjCClassMethodDecl));
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006876 }
6877 }
6878}
6879
Douglas Gregore8f5a172010-04-07 00:21:17 +00006880void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6881 bool IsInstanceMethod,
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006882 ParsedType ReturnTy) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006883 // Determine the return type of the method we're declaring, if
6884 // provided.
6885 QualType ReturnType = GetTypeFromParser(ReturnTy);
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006886 Decl *IDecl = 0;
6887 if (CurContext->isObjCContainer()) {
6888 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6889 IDecl = cast<Decl>(OCD);
6890 }
Douglas Gregorea766182010-10-18 18:21:28 +00006891 // Determine where we should start searching for methods.
6892 ObjCContainerDecl *SearchDecl = 0;
Douglas Gregore8f5a172010-04-07 00:21:17 +00006893 bool IsInImplementation = false;
John McCalld226f652010-08-21 09:40:31 +00006894 if (Decl *D = IDecl) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006895 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6896 SearchDecl = Impl->getClassInterface();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006897 IsInImplementation = true;
6898 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregorea766182010-10-18 18:21:28 +00006899 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006900 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006901 IsInImplementation = true;
Douglas Gregorea766182010-10-18 18:21:28 +00006902 } else
Douglas Gregore8f5a172010-04-07 00:21:17 +00006903 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006904 }
6905
6906 if (!SearchDecl && S) {
Douglas Gregorea766182010-10-18 18:21:28 +00006907 if (DeclContext *DC = static_cast<DeclContext *>(S->getEntity()))
Douglas Gregore8f5a172010-04-07 00:21:17 +00006908 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006909 }
6910
Douglas Gregorea766182010-10-18 18:21:28 +00006911 if (!SearchDecl) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006912 HandleCodeCompleteResults(this, CodeCompleter,
6913 CodeCompletionContext::CCC_Other,
6914 0, 0);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006915 return;
6916 }
6917
6918 // Find all of the methods that we could declare/implement here.
6919 KnownMethodsMap KnownMethods;
6920 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregorea766182010-10-18 18:21:28 +00006921 ReturnType, KnownMethods);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006922
Douglas Gregore8f5a172010-04-07 00:21:17 +00006923 // Add declarations or definitions for each of the known methods.
John McCall0a2c5e22010-08-25 06:19:51 +00006924 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00006925 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006926 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00006927 CodeCompletionContext::CCC_Other);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006928 Results.EnterNewScope();
Douglas Gregor8987b232011-09-27 23:30:47 +00006929 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006930 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6931 MEnd = KnownMethods.end();
6932 M != MEnd; ++M) {
Douglas Gregor408be5a2010-08-25 01:08:01 +00006933 ObjCMethodDecl *Method = M->second.first;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006934 CodeCompletionBuilder Builder(Results.getAllocator(),
6935 Results.getCodeCompletionTUInfo());
Douglas Gregore8f5a172010-04-07 00:21:17 +00006936
6937 // If the result type was not already provided, add it to the
6938 // pattern as (type).
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006939 if (ReturnType.isNull())
Douglas Gregor90f5f472012-04-10 18:35:07 +00006940 AddObjCPassingTypeChunk(Method->getResultType(),
6941 Method->getObjCDeclQualifier(),
6942 Context, Policy,
6943 Builder);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006944
6945 Selector Sel = Method->getSelector();
6946
6947 // Add the first part of the selector to the pattern.
Douglas Gregordae68752011-02-01 22:57:45 +00006948 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00006949 Sel.getNameForSlot(0)));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006950
6951 // Add parameters to the pattern.
6952 unsigned I = 0;
6953 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6954 PEnd = Method->param_end();
6955 P != PEnd; (void)++P, ++I) {
6956 // Add the part of the selector name.
6957 if (I == 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00006958 Builder.AddTypedTextChunk(":");
Douglas Gregore8f5a172010-04-07 00:21:17 +00006959 else if (I < Sel.getNumArgs()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00006960 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6961 Builder.AddTypedTextChunk(
Douglas Gregor813d8342011-02-18 22:29:55 +00006962 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006963 } else
6964 break;
6965
6966 // Add the parameter type.
Douglas Gregor90f5f472012-04-10 18:35:07 +00006967 AddObjCPassingTypeChunk((*P)->getOriginalType(),
6968 (*P)->getObjCDeclQualifier(),
6969 Context, Policy,
Douglas Gregor8987b232011-09-27 23:30:47 +00006970 Builder);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006971
6972 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregordae68752011-02-01 22:57:45 +00006973 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006974 }
6975
6976 if (Method->isVariadic()) {
6977 if (Method->param_size() > 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00006978 Builder.AddChunk(CodeCompletionString::CK_Comma);
6979 Builder.AddTextChunk("...");
Douglas Gregore17794f2010-08-31 05:13:43 +00006980 }
Douglas Gregore8f5a172010-04-07 00:21:17 +00006981
Douglas Gregor447107d2010-05-28 00:57:46 +00006982 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006983 // We will be defining the method here, so add a compound statement.
Douglas Gregor218937c2011-02-01 19:23:04 +00006984 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6985 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6986 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006987 if (!Method->getResultType()->isVoidType()) {
6988 // If the result type is not void, add a return clause.
Douglas Gregor218937c2011-02-01 19:23:04 +00006989 Builder.AddTextChunk("return");
6990 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6991 Builder.AddPlaceholderChunk("expression");
6992 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006993 } else
Douglas Gregor218937c2011-02-01 19:23:04 +00006994 Builder.AddPlaceholderChunk("statements");
Douglas Gregore8f5a172010-04-07 00:21:17 +00006995
Douglas Gregor218937c2011-02-01 19:23:04 +00006996 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6997 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006998 }
6999
Douglas Gregor408be5a2010-08-25 01:08:01 +00007000 unsigned Priority = CCP_CodePattern;
7001 if (!M->second.second)
7002 Priority += CCD_InBaseClass;
7003
Douglas Gregorba103062012-03-27 23:34:16 +00007004 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregore8f5a172010-04-07 00:21:17 +00007005 }
7006
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007007 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
7008 // the properties in this class and its categories.
David Blaikie4e4d0842012-03-11 07:00:24 +00007009 if (Context.getLangOpts().ObjC2) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00007010 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007011 Containers.push_back(SearchDecl);
7012
Douglas Gregore74c25c2011-05-04 23:50:46 +00007013 VisitedSelectorSet KnownSelectors;
7014 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
7015 MEnd = KnownMethods.end();
7016 M != MEnd; ++M)
7017 KnownSelectors.insert(M->first);
7018
7019
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007020 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
7021 if (!IFace)
7022 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
7023 IFace = Category->getClassInterface();
7024
7025 if (IFace) {
Douglas Gregord3297242013-01-16 23:00:23 +00007026 for (ObjCInterfaceDecl::visible_categories_iterator
7027 Cat = IFace->visible_categories_begin(),
7028 CatEnd = IFace->visible_categories_end();
7029 Cat != CatEnd; ++Cat) {
7030 Containers.push_back(*Cat);
7031 }
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007032 }
7033
7034 for (unsigned I = 0, N = Containers.size(); I != N; ++I) {
7035 for (ObjCContainerDecl::prop_iterator P = Containers[I]->prop_begin(),
7036 PEnd = Containers[I]->prop_end();
7037 P != PEnd; ++P) {
David Blaikie581deb32012-06-06 20:45:41 +00007038 AddObjCKeyValueCompletions(*P, IsInstanceMethod, ReturnType, Context,
Douglas Gregore74c25c2011-05-04 23:50:46 +00007039 KnownSelectors, Results);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007040 }
7041 }
7042 }
7043
Douglas Gregore8f5a172010-04-07 00:21:17 +00007044 Results.ExitScope();
7045
Douglas Gregore6b1bb62010-08-11 21:23:17 +00007046 HandleCodeCompleteResults(this, CodeCompleter,
7047 CodeCompletionContext::CCC_Other,
7048 Results.data(),Results.size());
Douglas Gregore8f5a172010-04-07 00:21:17 +00007049}
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007050
7051void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
7052 bool IsInstanceMethod,
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007053 bool AtParameterName,
John McCallb3d87482010-08-24 05:47:05 +00007054 ParsedType ReturnTy,
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007055 IdentifierInfo **SelIdents,
7056 unsigned NumSelIdents) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007057 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00007058 // pool from the AST file.
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007059 if (ExternalSource) {
7060 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
7061 I != N; ++I) {
7062 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redldb9d2142010-08-02 23:18:59 +00007063 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007064 continue;
Sebastian Redldb9d2142010-08-02 23:18:59 +00007065
7066 ReadMethodPool(Sel);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007067 }
7068 }
7069
7070 // Build the set of methods we can see.
John McCall0a2c5e22010-08-25 06:19:51 +00007071 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00007072 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007073 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00007074 CodeCompletionContext::CCC_Other);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007075
7076 if (ReturnTy)
7077 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redldb9d2142010-08-02 23:18:59 +00007078
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007079 Results.EnterNewScope();
Sebastian Redldb9d2142010-08-02 23:18:59 +00007080 for (GlobalMethodPool::iterator M = MethodPool.begin(),
7081 MEnd = MethodPool.end();
7082 M != MEnd; ++M) {
7083 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
7084 &M->second.second;
7085 MethList && MethList->Method;
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007086 MethList = MethList->Next) {
7087 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
7088 NumSelIdents))
7089 continue;
7090
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007091 if (AtParameterName) {
7092 // Suggest parameter names we've seen before.
7093 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
7094 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
7095 if (Param->getIdentifier()) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007096 CodeCompletionBuilder Builder(Results.getAllocator(),
7097 Results.getCodeCompletionTUInfo());
Douglas Gregordae68752011-02-01 22:57:45 +00007098 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00007099 Param->getIdentifier()->getName()));
7100 Results.AddResult(Builder.TakeString());
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007101 }
7102 }
7103
7104 continue;
7105 }
7106
Douglas Gregord1f09b42013-01-31 04:52:16 +00007107 Result R(MethList->Method, Results.getBasePriority(MethList->Method), 0);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007108 R.StartParameter = NumSelIdents;
7109 R.AllParametersAreInformative = false;
7110 R.DeclaringEntity = true;
7111 Results.MaybeAddResult(R, CurContext);
7112 }
7113 }
7114
7115 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00007116 HandleCodeCompleteResults(this, CodeCompleter,
7117 CodeCompletionContext::CCC_Other,
7118 Results.data(),Results.size());
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007119}
Douglas Gregor87c08a52010-08-13 22:48:40 +00007120
Douglas Gregorf29c5232010-08-24 22:20:20 +00007121void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregor218937c2011-02-01 19:23:04 +00007122 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007123 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00007124 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregorf44e8542010-08-24 19:08:16 +00007125 Results.EnterNewScope();
7126
7127 // #if <condition>
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007128 CodeCompletionBuilder Builder(Results.getAllocator(),
7129 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00007130 Builder.AddTypedTextChunk("if");
7131 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7132 Builder.AddPlaceholderChunk("condition");
7133 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007134
7135 // #ifdef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007136 Builder.AddTypedTextChunk("ifdef");
7137 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7138 Builder.AddPlaceholderChunk("macro");
7139 Results.AddResult(Builder.TakeString());
7140
Douglas Gregorf44e8542010-08-24 19:08:16 +00007141 // #ifndef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007142 Builder.AddTypedTextChunk("ifndef");
7143 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7144 Builder.AddPlaceholderChunk("macro");
7145 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007146
7147 if (InConditional) {
7148 // #elif <condition>
Douglas Gregor218937c2011-02-01 19:23:04 +00007149 Builder.AddTypedTextChunk("elif");
7150 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7151 Builder.AddPlaceholderChunk("condition");
7152 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007153
7154 // #else
Douglas Gregor218937c2011-02-01 19:23:04 +00007155 Builder.AddTypedTextChunk("else");
7156 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007157
7158 // #endif
Douglas Gregor218937c2011-02-01 19:23:04 +00007159 Builder.AddTypedTextChunk("endif");
7160 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007161 }
7162
7163 // #include "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00007164 Builder.AddTypedTextChunk("include");
7165 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7166 Builder.AddTextChunk("\"");
7167 Builder.AddPlaceholderChunk("header");
7168 Builder.AddTextChunk("\"");
7169 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007170
7171 // #include <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00007172 Builder.AddTypedTextChunk("include");
7173 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7174 Builder.AddTextChunk("<");
7175 Builder.AddPlaceholderChunk("header");
7176 Builder.AddTextChunk(">");
7177 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007178
7179 // #define <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007180 Builder.AddTypedTextChunk("define");
7181 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7182 Builder.AddPlaceholderChunk("macro");
7183 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007184
7185 // #define <macro>(<args>)
Douglas Gregor218937c2011-02-01 19:23:04 +00007186 Builder.AddTypedTextChunk("define");
7187 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7188 Builder.AddPlaceholderChunk("macro");
7189 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7190 Builder.AddPlaceholderChunk("args");
7191 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7192 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007193
7194 // #undef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007195 Builder.AddTypedTextChunk("undef");
7196 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7197 Builder.AddPlaceholderChunk("macro");
7198 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007199
7200 // #line <number>
Douglas Gregor218937c2011-02-01 19:23:04 +00007201 Builder.AddTypedTextChunk("line");
7202 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7203 Builder.AddPlaceholderChunk("number");
7204 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007205
7206 // #line <number> "filename"
Douglas Gregor218937c2011-02-01 19:23:04 +00007207 Builder.AddTypedTextChunk("line");
7208 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7209 Builder.AddPlaceholderChunk("number");
7210 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7211 Builder.AddTextChunk("\"");
7212 Builder.AddPlaceholderChunk("filename");
7213 Builder.AddTextChunk("\"");
7214 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007215
7216 // #error <message>
Douglas Gregor218937c2011-02-01 19:23:04 +00007217 Builder.AddTypedTextChunk("error");
7218 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7219 Builder.AddPlaceholderChunk("message");
7220 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007221
7222 // #pragma <arguments>
Douglas Gregor218937c2011-02-01 19:23:04 +00007223 Builder.AddTypedTextChunk("pragma");
7224 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7225 Builder.AddPlaceholderChunk("arguments");
7226 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007227
David Blaikie4e4d0842012-03-11 07:00:24 +00007228 if (getLangOpts().ObjC1) {
Douglas Gregorf44e8542010-08-24 19:08:16 +00007229 // #import "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00007230 Builder.AddTypedTextChunk("import");
7231 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7232 Builder.AddTextChunk("\"");
7233 Builder.AddPlaceholderChunk("header");
7234 Builder.AddTextChunk("\"");
7235 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007236
7237 // #import <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00007238 Builder.AddTypedTextChunk("import");
7239 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7240 Builder.AddTextChunk("<");
7241 Builder.AddPlaceholderChunk("header");
7242 Builder.AddTextChunk(">");
7243 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007244 }
7245
7246 // #include_next "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00007247 Builder.AddTypedTextChunk("include_next");
7248 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7249 Builder.AddTextChunk("\"");
7250 Builder.AddPlaceholderChunk("header");
7251 Builder.AddTextChunk("\"");
7252 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007253
7254 // #include_next <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00007255 Builder.AddTypedTextChunk("include_next");
7256 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7257 Builder.AddTextChunk("<");
7258 Builder.AddPlaceholderChunk("header");
7259 Builder.AddTextChunk(">");
7260 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007261
7262 // #warning <message>
Douglas Gregor218937c2011-02-01 19:23:04 +00007263 Builder.AddTypedTextChunk("warning");
7264 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7265 Builder.AddPlaceholderChunk("message");
7266 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007267
7268 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7269 // completions for them. And __include_macros is a Clang-internal extension
7270 // that we don't want to encourage anyone to use.
7271
7272 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7273 Results.ExitScope();
7274
Douglas Gregorf44e8542010-08-24 19:08:16 +00007275 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor721f3592010-08-25 18:41:16 +00007276 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregorf44e8542010-08-24 19:08:16 +00007277 Results.data(), Results.size());
7278}
7279
7280void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorf29c5232010-08-24 22:20:20 +00007281 CodeCompleteOrdinaryName(S,
John McCallf312b1e2010-08-26 23:41:50 +00007282 S->getFnParent()? Sema::PCC_RecoveryInFunction
7283 : Sema::PCC_Namespace);
Douglas Gregorf44e8542010-08-24 19:08:16 +00007284}
7285
Douglas Gregorf29c5232010-08-24 22:20:20 +00007286void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregor218937c2011-02-01 19:23:04 +00007287 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007288 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00007289 IsDefinition? CodeCompletionContext::CCC_MacroName
7290 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007291 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7292 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007293 CodeCompletionBuilder Builder(Results.getAllocator(),
7294 Results.getCodeCompletionTUInfo());
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007295 Results.EnterNewScope();
7296 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7297 MEnd = PP.macro_end();
7298 M != MEnd; ++M) {
Douglas Gregordae68752011-02-01 22:57:45 +00007299 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00007300 M->first->getName()));
Argyrios Kyrtzidisc04bb922012-09-27 00:24:09 +00007301 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7302 CCP_CodePattern,
7303 CXCursor_MacroDefinition));
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007304 }
7305 Results.ExitScope();
7306 } else if (IsDefinition) {
7307 // FIXME: Can we detect when the user just wrote an include guard above?
7308 }
7309
Douglas Gregor52779fb2010-09-23 23:01:17 +00007310 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007311 Results.data(), Results.size());
7312}
7313
Douglas Gregorf29c5232010-08-24 22:20:20 +00007314void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregor218937c2011-02-01 19:23:04 +00007315 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007316 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00007317 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorf29c5232010-08-24 22:20:20 +00007318
7319 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00007320 AddMacroResults(PP, Results, true);
Douglas Gregorf29c5232010-08-24 22:20:20 +00007321
7322 // defined (<macro>)
7323 Results.EnterNewScope();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007324 CodeCompletionBuilder Builder(Results.getAllocator(),
7325 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00007326 Builder.AddTypedTextChunk("defined");
7327 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7328 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7329 Builder.AddPlaceholderChunk("macro");
7330 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7331 Results.AddResult(Builder.TakeString());
Douglas Gregorf29c5232010-08-24 22:20:20 +00007332 Results.ExitScope();
7333
7334 HandleCodeCompleteResults(this, CodeCompleter,
7335 CodeCompletionContext::CCC_PreprocessorExpression,
7336 Results.data(), Results.size());
7337}
7338
7339void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7340 IdentifierInfo *Macro,
7341 MacroInfo *MacroInfo,
7342 unsigned Argument) {
7343 // FIXME: In the future, we could provide "overload" results, much like we
7344 // do for function calls.
7345
Argyrios Kyrtzidis5c5f03e2011-08-18 19:41:28 +00007346 // Now just ignore this. There will be another code-completion callback
7347 // for the expanded tokens.
Douglas Gregorf29c5232010-08-24 22:20:20 +00007348}
7349
Douglas Gregor55817af2010-08-25 17:04:25 +00007350void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor55817af2010-08-25 17:04:25 +00007351 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregoraf1c6b52010-08-25 17:10:00 +00007352 CodeCompletionContext::CCC_NaturalLanguage,
Douglas Gregor55817af2010-08-25 17:04:25 +00007353 0, 0);
7354}
7355
Douglas Gregordae68752011-02-01 22:57:45 +00007356void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007357 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner5f9e2722011-07-23 10:55:15 +00007358 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007359 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7360 CodeCompletionContext::CCC_Recovery);
Douglas Gregor8071e422010-08-15 06:18:01 +00007361 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7362 CodeCompletionDeclConsumer Consumer(Builder,
7363 Context.getTranslationUnitDecl());
7364 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7365 Consumer);
7366 }
Douglas Gregor87c08a52010-08-13 22:48:40 +00007367
7368 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00007369 AddMacroResults(PP, Builder, true);
Douglas Gregor87c08a52010-08-13 22:48:40 +00007370
7371 Results.clear();
7372 Results.insert(Results.end(),
7373 Builder.data(), Builder.data() + Builder.size());
7374}