blob: 1cb264265aa6a1ad11e928cfc43cb6fdeb6756ae [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() &&
2553 PDecl->getIdentifier() != M->getIdentifier())
2554 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(PDecl)) {
2555 Result.addBriefComment(RC->getBriefText(Ctx));
2556 Pattern->BriefComment = Result.getBriefComment();
2557 }
Douglas Gregorba103062012-03-27 23:34:16 +00002558 }
2559
Douglas Gregor218937c2011-02-01 19:23:04 +00002560 return Pattern;
2561 }
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002562
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002563 if (Kind == RK_Keyword) {
Douglas Gregor218937c2011-02-01 19:23:04 +00002564 Result.AddTypedTextChunk(Keyword);
2565 return Result.TakeString();
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002566 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00002567
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002568 if (Kind == RK_Macro) {
Argyrios Kyrtzidis9818a1d2013-02-20 00:54:57 +00002569 const MacroDirective *MD = PP.getMacroDirectiveHistory(Macro);
2570 assert(MD && "Not a macro?");
2571 const MacroInfo *MI = MD->getInfo();
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002572
Douglas Gregordae68752011-02-01 22:57:45 +00002573 Result.AddTypedTextChunk(
2574 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002575
2576 if (!MI->isFunctionLike())
Douglas Gregor218937c2011-02-01 19:23:04 +00002577 return Result.TakeString();
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002578
2579 // Format a function-like macro with placeholders for the arguments.
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002580 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregore4244702011-07-30 08:17:44 +00002581 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002582
2583 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
2584 if (MI->isC99Varargs()) {
2585 --AEnd;
2586
2587 if (A == AEnd) {
2588 Result.AddPlaceholderChunk("...");
2589 }
Douglas Gregore4244702011-07-30 08:17:44 +00002590 }
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002591
Douglas Gregore4244702011-07-30 08:17:44 +00002592 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002593 if (A != MI->arg_begin())
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002594 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002595
2596 if (MI->isVariadic() && (A+1) == AEnd) {
Dylan Noblesmithf7ccbad2012-02-05 02:13:05 +00002597 SmallString<32> Arg = (*A)->getName();
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002598 if (MI->isC99Varargs())
2599 Arg += ", ...";
2600 else
2601 Arg += "...";
Douglas Gregordae68752011-02-01 22:57:45 +00002602 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002603 break;
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002604 }
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002605
2606 // Non-variadic macros are simple.
2607 Result.AddPlaceholderChunk(
2608 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregore4244702011-07-30 08:17:44 +00002609 }
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002610 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor218937c2011-02-01 19:23:04 +00002611 return Result.TakeString();
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002612 }
2613
Douglas Gregord8e8a582010-05-25 21:41:55 +00002614 assert(Kind == RK_Declaration && "Missed a result kind?");
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002615 const NamedDecl *ND = Declaration;
Douglas Gregorba103062012-03-27 23:34:16 +00002616 Result.addParentContext(ND->getDeclContext());
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002617
2618 if (IncludeBriefComments) {
2619 // Add documentation comment, if it exists.
Dmitri Gribenkof50555e2012-08-11 00:51:43 +00002620 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(ND)) {
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002621 Result.addBriefComment(RC->getBriefText(Ctx));
Fariborz Jahanianb98f7af2013-02-28 17:47:14 +00002622 }
2623 else if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2624 if (OMD->isPropertyAccessor())
2625 if (const ObjCPropertyDecl *PDecl = OMD->findPropertyDecl())
2626 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(PDecl))
2627 Result.addBriefComment(RC->getBriefText(Ctx));
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002628 }
2629
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002630 if (StartsNestedNameSpecifier) {
Douglas Gregordae68752011-02-01 22:57:45 +00002631 Result.AddTypedTextChunk(
2632 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002633 Result.AddTextChunk("::");
2634 return Result.TakeString();
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002635 }
Erik Verbruggen6164ea12011-10-14 15:31:08 +00002636
2637 for (Decl::attr_iterator i = ND->attr_begin(); i != ND->attr_end(); ++i) {
2638 if (AnnotateAttr *Attr = dyn_cast_or_null<AnnotateAttr>(*i)) {
2639 Result.AddAnnotation(Result.getAllocator().CopyString(Attr->getAnnotation()));
2640 }
2641 }
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002642
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002643 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002644
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002645 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002646 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002647 Ctx, Policy);
2648 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002649 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002650 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002651 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregora61a8792009-12-11 18:44:16 +00002652 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregor218937c2011-02-01 19:23:04 +00002653 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002654 }
2655
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002656 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002657 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002658 Ctx, Policy);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002659 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002660 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor6f942b22010-09-21 16:06:22 +00002661
Douglas Gregor86d9a522009-09-21 16:56:56 +00002662 // Figure out which template parameters are deduced (or have default
2663 // arguments).
Benjamin Kramer013b3662012-01-30 16:17:39 +00002664 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002665 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002666 unsigned LastDeducibleArgument;
2667 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2668 --LastDeducibleArgument) {
2669 if (!Deduced[LastDeducibleArgument - 1]) {
2670 // C++0x: Figure out if the template argument has a default. If so,
2671 // the user doesn't need to type this argument.
2672 // FIXME: We need to abstract template parameters better!
2673 bool HasDefaultArg = false;
2674 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregor218937c2011-02-01 19:23:04 +00002675 LastDeducibleArgument - 1);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002676 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2677 HasDefaultArg = TTP->hasDefaultArgument();
2678 else if (NonTypeTemplateParmDecl *NTTP
2679 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2680 HasDefaultArg = NTTP->hasDefaultArgument();
2681 else {
2682 assert(isa<TemplateTemplateParmDecl>(Param));
2683 HasDefaultArg
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002684 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002685 }
2686
2687 if (!HasDefaultArg)
2688 break;
2689 }
2690 }
2691
2692 if (LastDeducibleArgument) {
2693 // Some of the function template arguments cannot be deduced from a
2694 // function call, so we introduce an explicit template argument list
2695 // containing all of the arguments up to the first deducible argument.
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002696 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002697 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor86d9a522009-09-21 16:56:56 +00002698 LastDeducibleArgument);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002699 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002700 }
2701
2702 // Add the function parameters
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002703 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002704 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002705 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregora61a8792009-12-11 18:44:16 +00002706 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregor218937c2011-02-01 19:23:04 +00002707 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002708 }
2709
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002710 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002711 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002712 Ctx, Policy);
Douglas Gregordae68752011-02-01 22:57:45 +00002713 Result.AddTypedTextChunk(
2714 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002715 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002716 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002717 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor218937c2011-02-01 19:23:04 +00002718 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002719 }
2720
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002721 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregor9630eb62009-11-17 16:44:22 +00002722 Selector Sel = Method->getSelector();
2723 if (Sel.isUnarySelector()) {
Douglas Gregordae68752011-02-01 22:57:45 +00002724 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00002725 Sel.getNameForSlot(0)));
Douglas Gregor218937c2011-02-01 19:23:04 +00002726 return Result.TakeString();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002727 }
2728
Douglas Gregor813d8342011-02-18 22:29:55 +00002729 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregord3c68542009-11-19 01:08:35 +00002730 SelName += ':';
2731 if (StartParameter == 0)
Douglas Gregordae68752011-02-01 22:57:45 +00002732 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregord3c68542009-11-19 01:08:35 +00002733 else {
Douglas Gregordae68752011-02-01 22:57:45 +00002734 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregord3c68542009-11-19 01:08:35 +00002735
2736 // If there is only one parameter, and we're past it, add an empty
2737 // typed-text chunk since there is nothing to type.
2738 if (Method->param_size() == 1)
Douglas Gregor218937c2011-02-01 19:23:04 +00002739 Result.AddTypedTextChunk("");
Douglas Gregord3c68542009-11-19 01:08:35 +00002740 }
Douglas Gregor9630eb62009-11-17 16:44:22 +00002741 unsigned Idx = 0;
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002742 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2743 PEnd = Method->param_end();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002744 P != PEnd; (void)++P, ++Idx) {
2745 if (Idx > 0) {
Douglas Gregord3c68542009-11-19 01:08:35 +00002746 std::string Keyword;
2747 if (Idx > StartParameter)
Douglas Gregor218937c2011-02-01 19:23:04 +00002748 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor9630eb62009-11-17 16:44:22 +00002749 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramera0651c52011-07-26 16:59:25 +00002750 Keyword += II->getName();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002751 Keyword += ":";
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002752 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002753 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002754 else
Douglas Gregordae68752011-02-01 22:57:45 +00002755 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor9630eb62009-11-17 16:44:22 +00002756 }
Douglas Gregord3c68542009-11-19 01:08:35 +00002757
2758 // If we're before the starting parameter, skip the placeholder.
2759 if (Idx < StartParameter)
2760 continue;
Douglas Gregor9630eb62009-11-17 16:44:22 +00002761
2762 std::string Arg;
Douglas Gregor83482d12010-08-24 16:15:59 +00002763
2764 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002765 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregor83482d12010-08-24 16:15:59 +00002766 else {
John McCallf85e1932011-06-15 23:02:42 +00002767 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002768 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2769 + Arg + ")";
Douglas Gregor83482d12010-08-24 16:15:59 +00002770 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregoraba48082010-08-29 19:47:46 +00002771 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramera0651c52011-07-26 16:59:25 +00002772 Arg += II->getName();
Douglas Gregor83482d12010-08-24 16:15:59 +00002773 }
2774
Douglas Gregore17794f2010-08-31 05:13:43 +00002775 if (Method->isVariadic() && (P + 1) == PEnd)
2776 Arg += ", ...";
2777
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002778 if (DeclaringEntity)
Douglas Gregordae68752011-02-01 22:57:45 +00002779 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002780 else if (AllParametersAreInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002781 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor4ad96852009-11-19 07:41:15 +00002782 else
Douglas Gregordae68752011-02-01 22:57:45 +00002783 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor9630eb62009-11-17 16:44:22 +00002784 }
2785
Douglas Gregor2a17af02009-12-23 00:21:46 +00002786 if (Method->isVariadic()) {
Douglas Gregore17794f2010-08-31 05:13:43 +00002787 if (Method->param_size() == 0) {
2788 if (DeclaringEntity)
Douglas Gregor218937c2011-02-01 19:23:04 +00002789 Result.AddTextChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002790 else if (AllParametersAreInformative)
Douglas Gregor218937c2011-02-01 19:23:04 +00002791 Result.AddInformativeChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002792 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002793 Result.AddPlaceholderChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002794 }
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002795
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002796 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor2a17af02009-12-23 00:21:46 +00002797 }
2798
Douglas Gregor218937c2011-02-01 19:23:04 +00002799 return Result.TakeString();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002800 }
2801
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002802 if (Qualifier)
Douglas Gregor0563c262009-09-22 23:15:58 +00002803 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002804 Ctx, Policy);
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002805
Douglas Gregordae68752011-02-01 22:57:45 +00002806 Result.AddTypedTextChunk(
2807 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002808 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002809}
2810
Douglas Gregor86d802e2009-09-23 00:34:09 +00002811CodeCompletionString *
2812CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
2813 unsigned CurrentArg,
Douglas Gregor32be4a52010-10-11 21:37:58 +00002814 Sema &S,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002815 CodeCompletionAllocator &Allocator,
2816 CodeCompletionTUInfo &CCTUInfo) const {
Douglas Gregor8987b232011-09-27 23:30:47 +00002817 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCallf85e1932011-06-15 23:02:42 +00002818
Douglas Gregor218937c2011-02-01 19:23:04 +00002819 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002820 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002821 FunctionDecl *FDecl = getFunction();
Douglas Gregor8987b232011-09-27 23:30:47 +00002822 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002823 const FunctionProtoType *Proto
2824 = dyn_cast<FunctionProtoType>(getFunctionType());
2825 if (!FDecl && !Proto) {
2826 // Function without a prototype. Just give the return type and a
2827 // highlighted ellipsis.
2828 const FunctionType *FT = getFunctionType();
Douglas Gregora63f6de2011-02-01 21:15:40 +00002829 Result.AddTextChunk(GetCompletionTypeString(FT->getResultType(),
Douglas Gregor8987b232011-09-27 23:30:47 +00002830 S.Context, Policy,
Douglas Gregora63f6de2011-02-01 21:15:40 +00002831 Result.getAllocator()));
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002832 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2833 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2834 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor218937c2011-02-01 19:23:04 +00002835 return Result.TakeString();
Douglas Gregor86d802e2009-09-23 00:34:09 +00002836 }
2837
2838 if (FDecl)
Douglas Gregordae68752011-02-01 22:57:45 +00002839 Result.AddTextChunk(
2840 Result.getAllocator().CopyString(FDecl->getNameAsString()));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002841 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002842 Result.AddTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00002843 Result.getAllocator().CopyString(
John McCallf85e1932011-06-15 23:02:42 +00002844 Proto->getResultType().getAsString(Policy)));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002845
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002846 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002847 unsigned NumParams = FDecl? FDecl->getNumParams() : Proto->getNumArgs();
2848 for (unsigned I = 0; I != NumParams; ++I) {
2849 if (I)
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002850 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002851
2852 std::string ArgString;
2853 QualType ArgType;
2854
2855 if (FDecl) {
2856 ArgString = FDecl->getParamDecl(I)->getNameAsString();
2857 ArgType = FDecl->getParamDecl(I)->getOriginalType();
2858 } else {
2859 ArgType = Proto->getArgType(I);
2860 }
2861
John McCallf85e1932011-06-15 23:02:42 +00002862 ArgType.getAsStringInternal(ArgString, Policy);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002863
2864 if (I == CurrentArg)
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002865 Result.AddChunk(CodeCompletionString::CK_CurrentParameter,
2866 Result.getAllocator().CopyString(ArgString));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002867 else
Douglas Gregordae68752011-02-01 22:57:45 +00002868 Result.AddTextChunk(Result.getAllocator().CopyString(ArgString));
Douglas Gregor86d802e2009-09-23 00:34:09 +00002869 }
2870
2871 if (Proto && Proto->isVariadic()) {
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002872 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002873 if (CurrentArg < NumParams)
Douglas Gregor218937c2011-02-01 19:23:04 +00002874 Result.AddTextChunk("...");
Douglas Gregor86d802e2009-09-23 00:34:09 +00002875 else
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002876 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
Douglas Gregor86d802e2009-09-23 00:34:09 +00002877 }
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002878 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002879
Douglas Gregor218937c2011-02-01 19:23:04 +00002880 return Result.TakeString();
Douglas Gregor86d802e2009-09-23 00:34:09 +00002881}
2882
Chris Lattner5f9e2722011-07-23 10:55:15 +00002883unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregorb05496d2010-09-20 21:11:48 +00002884 const LangOptions &LangOpts,
Douglas Gregor1827e102010-08-16 16:18:59 +00002885 bool PreferredTypeIsPointer) {
2886 unsigned Priority = CCP_Macro;
2887
Douglas Gregorb05496d2010-09-20 21:11:48 +00002888 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2889 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2890 MacroName.equals("Nil")) {
Douglas Gregor1827e102010-08-16 16:18:59 +00002891 Priority = CCP_Constant;
2892 if (PreferredTypeIsPointer)
2893 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregorb05496d2010-09-20 21:11:48 +00002894 }
2895 // Treat "YES", "NO", "true", and "false" as constants.
2896 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2897 MacroName.equals("true") || MacroName.equals("false"))
2898 Priority = CCP_Constant;
2899 // Treat "bool" as a type.
2900 else if (MacroName.equals("bool"))
2901 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2902
Douglas Gregor1827e102010-08-16 16:18:59 +00002903
2904 return Priority;
2905}
2906
Dmitri Gribenko06d8c602013-01-11 20:32:41 +00002907CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002908 if (!D)
2909 return CXCursor_UnexposedDecl;
2910
2911 switch (D->getKind()) {
2912 case Decl::Enum: return CXCursor_EnumDecl;
2913 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2914 case Decl::Field: return CXCursor_FieldDecl;
2915 case Decl::Function:
2916 return CXCursor_FunctionDecl;
2917 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2918 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002919 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregor375bb142011-12-27 22:43:10 +00002920
Argyrios Kyrtzidisc15707d2012-01-24 21:39:26 +00002921 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002922 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2923 case Decl::ObjCMethod:
2924 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2925 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2926 case Decl::CXXMethod: return CXCursor_CXXMethod;
2927 case Decl::CXXConstructor: return CXCursor_Constructor;
2928 case Decl::CXXDestructor: return CXCursor_Destructor;
2929 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2930 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidisc15707d2012-01-24 21:39:26 +00002931 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002932 case Decl::ParmVar: return CXCursor_ParmDecl;
2933 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smith162e1c12011-04-15 14:24:37 +00002934 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002935 case Decl::Var: return CXCursor_VarDecl;
2936 case Decl::Namespace: return CXCursor_Namespace;
2937 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2938 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2939 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2940 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2941 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2942 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis2dfdb942011-09-30 17:58:23 +00002943 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002944 case Decl::ClassTemplatePartialSpecialization:
2945 return CXCursor_ClassTemplatePartialSpecialization;
2946 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor8e5900c2012-04-30 23:41:16 +00002947 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002948
2949 case Decl::Using:
2950 case Decl::UnresolvedUsingValue:
2951 case Decl::UnresolvedUsingTypename:
2952 return CXCursor_UsingDeclaration;
2953
Douglas Gregor352697a2011-06-03 23:08:58 +00002954 case Decl::ObjCPropertyImpl:
2955 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
2956 case ObjCPropertyImplDecl::Dynamic:
2957 return CXCursor_ObjCDynamicDecl;
2958
2959 case ObjCPropertyImplDecl::Synthesize:
2960 return CXCursor_ObjCSynthesizeDecl;
2961 }
Argyrios Kyrtzidis6a010122012-10-05 00:22:24 +00002962
2963 case Decl::Import:
2964 return CXCursor_ModuleImportDecl;
Douglas Gregor352697a2011-06-03 23:08:58 +00002965
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002966 default:
Dmitri Gribenko06d8c602013-01-11 20:32:41 +00002967 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002968 switch (TD->getTagKind()) {
Joao Matos6666ed42012-08-31 18:45:21 +00002969 case TTK_Interface: // fall through
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002970 case TTK_Struct: return CXCursor_StructDecl;
2971 case TTK_Class: return CXCursor_ClassDecl;
2972 case TTK_Union: return CXCursor_UnionDecl;
2973 case TTK_Enum: return CXCursor_EnumDecl;
2974 }
2975 }
2976 }
2977
2978 return CXCursor_UnexposedDecl;
2979}
2980
Douglas Gregor590c7d52010-07-08 20:55:51 +00002981static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor3644d972012-10-09 16:01:50 +00002982 bool IncludeUndefined,
Douglas Gregor590c7d52010-07-08 20:55:51 +00002983 bool TargetTypeIsPointer = false) {
John McCall0a2c5e22010-08-25 06:19:51 +00002984 typedef CodeCompletionResult Result;
Douglas Gregor590c7d52010-07-08 20:55:51 +00002985
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002986 Results.EnterNewScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002987
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002988 for (Preprocessor::macro_iterator M = PP.macro_begin(),
2989 MEnd = PP.macro_end();
Douglas Gregor590c7d52010-07-08 20:55:51 +00002990 M != MEnd; ++M) {
Douglas Gregor3644d972012-10-09 16:01:50 +00002991 if (IncludeUndefined || M->first->hasMacroDefinition())
2992 Results.AddResult(Result(M->first,
Douglas Gregor1827e102010-08-16 16:18:59 +00002993 getMacroUsagePriority(M->first->getName(),
David Blaikie4e4d0842012-03-11 07:00:24 +00002994 PP.getLangOpts(),
Douglas Gregor1827e102010-08-16 16:18:59 +00002995 TargetTypeIsPointer)));
Douglas Gregor590c7d52010-07-08 20:55:51 +00002996 }
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002997
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002998 Results.ExitScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002999
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00003000}
3001
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003002static void AddPrettyFunctionResults(const LangOptions &LangOpts,
3003 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00003004 typedef CodeCompletionResult Result;
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003005
3006 Results.EnterNewScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00003007
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003008 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
3009 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith80ad52f2013-01-02 11:42:31 +00003010 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003011 Results.AddResult(Result("__func__", CCP_Constant));
3012 Results.ExitScope();
3013}
3014
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003015static void HandleCodeCompleteResults(Sema *S,
3016 CodeCompleteConsumer *CodeCompleter,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003017 CodeCompletionContext Context,
John McCall0a2c5e22010-08-25 06:19:51 +00003018 CodeCompletionResult *Results,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003019 unsigned NumResults) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00003020 if (CodeCompleter)
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003021 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor86d9a522009-09-21 16:56:56 +00003022}
3023
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003024static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3025 Sema::ParserCompletionContext PCC) {
3026 switch (PCC) {
John McCallf312b1e2010-08-26 23:41:50 +00003027 case Sema::PCC_Namespace:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003028 return CodeCompletionContext::CCC_TopLevel;
3029
John McCallf312b1e2010-08-26 23:41:50 +00003030 case Sema::PCC_Class:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003031 return CodeCompletionContext::CCC_ClassStructUnion;
3032
John McCallf312b1e2010-08-26 23:41:50 +00003033 case Sema::PCC_ObjCInterface:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003034 return CodeCompletionContext::CCC_ObjCInterface;
3035
John McCallf312b1e2010-08-26 23:41:50 +00003036 case Sema::PCC_ObjCImplementation:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003037 return CodeCompletionContext::CCC_ObjCImplementation;
3038
John McCallf312b1e2010-08-26 23:41:50 +00003039 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003040 return CodeCompletionContext::CCC_ObjCIvarList;
3041
John McCallf312b1e2010-08-26 23:41:50 +00003042 case Sema::PCC_Template:
3043 case Sema::PCC_MemberTemplate:
Douglas Gregor52779fb2010-09-23 23:01:17 +00003044 if (S.CurContext->isFileContext())
3045 return CodeCompletionContext::CCC_TopLevel;
David Blaikie7530c032012-01-17 06:56:22 +00003046 if (S.CurContext->isRecord())
Douglas Gregor52779fb2010-09-23 23:01:17 +00003047 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie7530c032012-01-17 06:56:22 +00003048 return CodeCompletionContext::CCC_Other;
Douglas Gregor52779fb2010-09-23 23:01:17 +00003049
John McCallf312b1e2010-08-26 23:41:50 +00003050 case Sema::PCC_RecoveryInFunction:
Douglas Gregor52779fb2010-09-23 23:01:17 +00003051 return CodeCompletionContext::CCC_Recovery;
Douglas Gregora5450a02010-10-18 22:01:46 +00003052
John McCallf312b1e2010-08-26 23:41:50 +00003053 case Sema::PCC_ForInit:
David Blaikie4e4d0842012-03-11 07:00:24 +00003054 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3055 S.getLangOpts().ObjC1)
Douglas Gregora5450a02010-10-18 22:01:46 +00003056 return CodeCompletionContext::CCC_ParenthesizedExpression;
3057 else
3058 return CodeCompletionContext::CCC_Expression;
3059
3060 case Sema::PCC_Expression:
John McCallf312b1e2010-08-26 23:41:50 +00003061 case Sema::PCC_Condition:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003062 return CodeCompletionContext::CCC_Expression;
3063
John McCallf312b1e2010-08-26 23:41:50 +00003064 case Sema::PCC_Statement:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003065 return CodeCompletionContext::CCC_Statement;
Douglas Gregor72db1082010-08-24 01:11:00 +00003066
John McCallf312b1e2010-08-26 23:41:50 +00003067 case Sema::PCC_Type:
Douglas Gregor72db1082010-08-24 01:11:00 +00003068 return CodeCompletionContext::CCC_Type;
Douglas Gregor02688102010-09-14 23:59:36 +00003069
3070 case Sema::PCC_ParenthesizedExpression:
3071 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003072
3073 case Sema::PCC_LocalDeclarationSpecifiers:
3074 return CodeCompletionContext::CCC_Type;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003075 }
David Blaikie7530c032012-01-17 06:56:22 +00003076
3077 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003078}
3079
Douglas Gregorf6961522010-08-27 21:18:54 +00003080/// \brief If we're in a C++ virtual member function, add completion results
3081/// that invoke the functions we override, since it's common to invoke the
3082/// overridden function as well as adding new functionality.
3083///
3084/// \param S The semantic analysis object for which we are generating results.
3085///
3086/// \param InContext This context in which the nested-name-specifier preceding
3087/// the code-completion point
3088static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3089 ResultBuilder &Results) {
3090 // Look through blocks.
3091 DeclContext *CurContext = S.CurContext;
3092 while (isa<BlockDecl>(CurContext))
3093 CurContext = CurContext->getParent();
3094
3095
3096 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3097 if (!Method || !Method->isVirtual())
3098 return;
3099
3100 // We need to have names for all of the parameters, if we're going to
3101 // generate a forwarding call.
3102 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3103 PEnd = Method->param_end();
3104 P != PEnd;
3105 ++P) {
3106 if (!(*P)->getDeclName())
3107 return;
3108 }
3109
Douglas Gregor8987b232011-09-27 23:30:47 +00003110 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorf6961522010-08-27 21:18:54 +00003111 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3112 MEnd = Method->end_overridden_methods();
3113 M != MEnd; ++M) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003114 CodeCompletionBuilder Builder(Results.getAllocator(),
3115 Results.getCodeCompletionTUInfo());
Dmitri Gribenko68a932d2013-02-14 13:53:30 +00003116 const CXXMethodDecl *Overridden = *M;
Douglas Gregorf6961522010-08-27 21:18:54 +00003117 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3118 continue;
3119
3120 // If we need a nested-name-specifier, add one now.
3121 if (!InContext) {
3122 NestedNameSpecifier *NNS
3123 = getRequiredQualification(S.Context, CurContext,
3124 Overridden->getDeclContext());
3125 if (NNS) {
3126 std::string Str;
3127 llvm::raw_string_ostream OS(Str);
Douglas Gregor8987b232011-09-27 23:30:47 +00003128 NNS->print(OS, Policy);
Douglas Gregordae68752011-02-01 22:57:45 +00003129 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorf6961522010-08-27 21:18:54 +00003130 }
3131 } else if (!InContext->Equals(Overridden->getDeclContext()))
3132 continue;
3133
Douglas Gregordae68752011-02-01 22:57:45 +00003134 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00003135 Overridden->getNameAsString()));
3136 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf6961522010-08-27 21:18:54 +00003137 bool FirstParam = true;
3138 for (CXXMethodDecl::param_iterator P = Method->param_begin(),
3139 PEnd = Method->param_end();
3140 P != PEnd; ++P) {
3141 if (FirstParam)
3142 FirstParam = false;
3143 else
Douglas Gregor218937c2011-02-01 19:23:04 +00003144 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf6961522010-08-27 21:18:54 +00003145
Douglas Gregordae68752011-02-01 22:57:45 +00003146 Builder.AddPlaceholderChunk(Results.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00003147 (*P)->getIdentifier()->getName()));
Douglas Gregorf6961522010-08-27 21:18:54 +00003148 }
Douglas Gregor218937c2011-02-01 19:23:04 +00003149 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3150 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorf6961522010-08-27 21:18:54 +00003151 CCP_SuperCompletion,
Douglas Gregorba103062012-03-27 23:34:16 +00003152 CXCursor_CXXMethod,
3153 CXAvailability_Available,
3154 Overridden));
Douglas Gregorf6961522010-08-27 21:18:54 +00003155 Results.Ignore(Overridden);
3156 }
3157}
3158
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003159void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3160 ModuleIdPath Path) {
3161 typedef CodeCompletionResult Result;
3162 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003163 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003164 CodeCompletionContext::CCC_Other);
3165 Results.EnterNewScope();
3166
3167 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003168 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003169 typedef CodeCompletionResult Result;
3170 if (Path.empty()) {
3171 // Enumerate all top-level modules.
Dmitri Gribenkocfa88f82013-01-12 19:30:44 +00003172 SmallVector<Module *, 8> Modules;
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003173 PP.getHeaderSearchInfo().collectAllModules(Modules);
3174 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3175 Builder.AddTypedTextChunk(
3176 Builder.getAllocator().CopyString(Modules[I]->Name));
3177 Results.AddResult(Result(Builder.TakeString(),
3178 CCP_Declaration,
3179 CXCursor_NotImplemented,
3180 Modules[I]->isAvailable()
3181 ? CXAvailability_Available
3182 : CXAvailability_NotAvailable));
3183 }
3184 } else {
3185 // Load the named module.
3186 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3187 Module::AllVisible,
3188 /*IsInclusionDirective=*/false);
3189 // Enumerate submodules.
3190 if (Mod) {
3191 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3192 SubEnd = Mod->submodule_end();
3193 Sub != SubEnd; ++Sub) {
3194
3195 Builder.AddTypedTextChunk(
3196 Builder.getAllocator().CopyString((*Sub)->Name));
3197 Results.AddResult(Result(Builder.TakeString(),
3198 CCP_Declaration,
3199 CXCursor_NotImplemented,
3200 (*Sub)->isAvailable()
3201 ? CXAvailability_Available
3202 : CXAvailability_NotAvailable));
3203 }
3204 }
3205 }
3206 Results.ExitScope();
3207 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3208 Results.data(),Results.size());
3209}
3210
Douglas Gregor01dfea02010-01-10 23:08:15 +00003211void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003212 ParserCompletionContext CompletionContext) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003213 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003214 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003215 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorf6961522010-08-27 21:18:54 +00003216 Results.EnterNewScope();
Douglas Gregorcee9ff12010-09-20 22:39:41 +00003217
Douglas Gregor01dfea02010-01-10 23:08:15 +00003218 // Determine how to filter results, e.g., so that the names of
3219 // values (functions, enumerators, function templates, etc.) are
3220 // only allowed where we can have an expression.
3221 switch (CompletionContext) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003222 case PCC_Namespace:
3223 case PCC_Class:
3224 case PCC_ObjCInterface:
3225 case PCC_ObjCImplementation:
3226 case PCC_ObjCInstanceVariableList:
3227 case PCC_Template:
3228 case PCC_MemberTemplate:
Douglas Gregor72db1082010-08-24 01:11:00 +00003229 case PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003230 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor01dfea02010-01-10 23:08:15 +00003231 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3232 break;
3233
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003234 case PCC_Statement:
Douglas Gregor02688102010-09-14 23:59:36 +00003235 case PCC_ParenthesizedExpression:
Douglas Gregoreb0d0142010-08-24 23:58:17 +00003236 case PCC_Expression:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003237 case PCC_ForInit:
3238 case PCC_Condition:
David Blaikie4e4d0842012-03-11 07:00:24 +00003239 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor4710e5b2010-05-28 00:49:12 +00003240 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3241 else
3242 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorf6961522010-08-27 21:18:54 +00003243
David Blaikie4e4d0842012-03-11 07:00:24 +00003244 if (getLangOpts().CPlusPlus)
Douglas Gregorf6961522010-08-27 21:18:54 +00003245 MaybeAddOverrideCalls(*this, /*InContext=*/0, Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00003246 break;
Douglas Gregordc845342010-05-25 05:58:43 +00003247
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003248 case PCC_RecoveryInFunction:
Douglas Gregordc845342010-05-25 05:58:43 +00003249 // Unfiltered
3250 break;
Douglas Gregor01dfea02010-01-10 23:08:15 +00003251 }
3252
Douglas Gregor3cdee122010-08-26 16:36:48 +00003253 // If we are in a C++ non-static member function, check the qualifiers on
3254 // the member function to filter/prioritize the results list.
3255 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3256 if (CurMethod->isInstance())
3257 Results.setObjectTypeQualifiers(
3258 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3259
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00003260 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003261 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3262 CodeCompleter->includeGlobals());
Douglas Gregor2a7925c2009-12-07 09:54:55 +00003263
Douglas Gregorbca403c2010-01-13 23:51:12 +00003264 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor2a7925c2009-12-07 09:54:55 +00003265 Results.ExitScope();
3266
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003267 switch (CompletionContext) {
Douglas Gregor02688102010-09-14 23:59:36 +00003268 case PCC_ParenthesizedExpression:
Douglas Gregor72db1082010-08-24 01:11:00 +00003269 case PCC_Expression:
3270 case PCC_Statement:
3271 case PCC_RecoveryInFunction:
3272 if (S->getFnParent())
David Blaikie4e4d0842012-03-11 07:00:24 +00003273 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor72db1082010-08-24 01:11:00 +00003274 break;
3275
3276 case PCC_Namespace:
3277 case PCC_Class:
3278 case PCC_ObjCInterface:
3279 case PCC_ObjCImplementation:
3280 case PCC_ObjCInstanceVariableList:
3281 case PCC_Template:
3282 case PCC_MemberTemplate:
3283 case PCC_ForInit:
3284 case PCC_Condition:
3285 case PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003286 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor72db1082010-08-24 01:11:00 +00003287 break;
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003288 }
3289
Douglas Gregor0c8296d2009-11-07 00:00:49 +00003290 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00003291 AddMacroResults(PP, Results, false);
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003292
Douglas Gregorcee9ff12010-09-20 22:39:41 +00003293 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003294 Results.data(),Results.size());
Douglas Gregor791215b2009-09-21 20:51:25 +00003295}
3296
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003297static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3298 ParsedType Receiver,
3299 IdentifierInfo **SelIdents,
3300 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00003301 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003302 bool IsSuper,
3303 ResultBuilder &Results);
3304
3305void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3306 bool AllowNonIdentifiers,
3307 bool AllowNestedNameSpecifiers) {
John McCall0a2c5e22010-08-25 06:19:51 +00003308 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00003309 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003310 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003311 AllowNestedNameSpecifiers
3312 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3313 : CodeCompletionContext::CCC_Name);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003314 Results.EnterNewScope();
3315
3316 // Type qualifiers can come after names.
3317 Results.AddResult(Result("const"));
3318 Results.AddResult(Result("volatile"));
David Blaikie4e4d0842012-03-11 07:00:24 +00003319 if (getLangOpts().C99)
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003320 Results.AddResult(Result("restrict"));
3321
David Blaikie4e4d0842012-03-11 07:00:24 +00003322 if (getLangOpts().CPlusPlus) {
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003323 if (AllowNonIdentifiers) {
3324 Results.AddResult(Result("operator"));
3325 }
3326
3327 // Add nested-name-specifiers.
3328 if (AllowNestedNameSpecifiers) {
3329 Results.allowNestedNameSpecifiers();
Douglas Gregor52779fb2010-09-23 23:01:17 +00003330 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003331 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3332 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3333 CodeCompleter->includeGlobals());
Douglas Gregor52779fb2010-09-23 23:01:17 +00003334 Results.setFilter(0);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003335 }
3336 }
3337 Results.ExitScope();
3338
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003339 // If we're in a context where we might have an expression (rather than a
3340 // declaration), and what we've seen so far is an Objective-C type that could
3341 // be a receiver of a class message, this may be a class message send with
3342 // the initial opening bracket '[' missing. Add appropriate completions.
3343 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
3344 DS.getTypeSpecType() == DeclSpec::TST_typename &&
3345 DS.getStorageClassSpecAsWritten() == DeclSpec::SCS_unspecified &&
3346 !DS.isThreadSpecified() && !DS.isExternInLinkageSpec() &&
3347 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3348 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
3349 DS.getTypeQualifiers() == 0 &&
3350 S &&
3351 (S->getFlags() & Scope::DeclScope) != 0 &&
3352 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3353 Scope::FunctionPrototypeScope |
3354 Scope::AtCatchScope)) == 0) {
3355 ParsedType T = DS.getRepAsType();
3356 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Douglas Gregor70c5ac72010-09-20 23:34:21 +00003357 AddClassMessageCompletions(*this, S, T, 0, 0, false, false, Results);
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003358 }
3359
Douglas Gregor4497dd42010-08-24 04:59:56 +00003360 // Note that we intentionally suppress macro results here, since we do not
3361 // encourage using macros to produce the names of entities.
3362
Douglas Gregor52779fb2010-09-23 23:01:17 +00003363 HandleCodeCompleteResults(this, CodeCompleter,
3364 Results.getCompletionContext(),
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003365 Results.data(), Results.size());
3366}
3367
Douglas Gregorfb629412010-08-23 21:17:50 +00003368struct Sema::CodeCompleteExpressionData {
3369 CodeCompleteExpressionData(QualType PreferredType = QualType())
3370 : PreferredType(PreferredType), IntegralConstantExpression(false),
3371 ObjCCollection(false) { }
3372
3373 QualType PreferredType;
3374 bool IntegralConstantExpression;
3375 bool ObjCCollection;
Chris Lattner5f9e2722011-07-23 10:55:15 +00003376 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregorfb629412010-08-23 21:17:50 +00003377};
3378
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003379/// \brief Perform code-completion in an expression context when we know what
3380/// type we're looking for.
Douglas Gregorfb629412010-08-23 21:17:50 +00003381void Sema::CodeCompleteExpression(Scope *S,
3382 const CodeCompleteExpressionData &Data) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003383 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003384 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00003385 CodeCompletionContext::CCC_Expression);
Douglas Gregorfb629412010-08-23 21:17:50 +00003386 if (Data.ObjCCollection)
3387 Results.setFilter(&ResultBuilder::IsObjCCollection);
3388 else if (Data.IntegralConstantExpression)
Douglas Gregorf9578432010-07-28 21:50:18 +00003389 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikie4e4d0842012-03-11 07:00:24 +00003390 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003391 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3392 else
3393 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorfb629412010-08-23 21:17:50 +00003394
3395 if (!Data.PreferredType.isNull())
3396 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3397
3398 // Ignore any declarations that we were told that we don't care about.
3399 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3400 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003401
3402 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003403 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3404 CodeCompleter->includeGlobals());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003405
3406 Results.EnterNewScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003407 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003408 Results.ExitScope();
3409
Douglas Gregor590c7d52010-07-08 20:55:51 +00003410 bool PreferredTypeIsPointer = false;
Douglas Gregorfb629412010-08-23 21:17:50 +00003411 if (!Data.PreferredType.isNull())
3412 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3413 || Data.PreferredType->isMemberPointerType()
3414 || Data.PreferredType->isBlockPointerType();
Douglas Gregor590c7d52010-07-08 20:55:51 +00003415
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003416 if (S->getFnParent() &&
3417 !Data.ObjCCollection &&
3418 !Data.IntegralConstantExpression)
David Blaikie4e4d0842012-03-11 07:00:24 +00003419 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003420
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003421 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00003422 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003423 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorfb629412010-08-23 21:17:50 +00003424 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3425 Data.PreferredType),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003426 Results.data(),Results.size());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003427}
3428
Douglas Gregorac5fd842010-09-18 01:28:11 +00003429void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3430 if (E.isInvalid())
3431 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikie4e4d0842012-03-11 07:00:24 +00003432 else if (getLangOpts().ObjC1)
Douglas Gregorac5fd842010-09-18 01:28:11 +00003433 CodeCompleteObjCInstanceMessage(S, E.take(), 0, 0, false);
Douglas Gregor78edf512010-09-15 16:23:04 +00003434}
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003435
Douglas Gregor73449212010-12-09 23:01:55 +00003436/// \brief The set of properties that have already been added, referenced by
3437/// property name.
3438typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3439
Douglas Gregorb92a4082012-06-12 13:44:08 +00003440/// \brief Retrieve the container definition, if any?
3441static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3442 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3443 if (Interface->hasDefinition())
3444 return Interface->getDefinition();
3445
3446 return Interface;
3447 }
3448
3449 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3450 if (Protocol->hasDefinition())
3451 return Protocol->getDefinition();
3452
3453 return Protocol;
3454 }
3455 return Container;
3456}
3457
3458static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor322328b2009-11-18 22:32:06 +00003459 bool AllowCategories,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003460 bool AllowNullaryMethods,
Douglas Gregor95ac6552009-11-18 01:29:26 +00003461 DeclContext *CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003462 AddedPropertiesSet &AddedProperties,
Douglas Gregor95ac6552009-11-18 01:29:26 +00003463 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00003464 typedef CodeCompletionResult Result;
Douglas Gregor95ac6552009-11-18 01:29:26 +00003465
Douglas Gregorb92a4082012-06-12 13:44:08 +00003466 // Retrieve the definition.
3467 Container = getContainerDef(Container);
3468
Douglas Gregor95ac6552009-11-18 01:29:26 +00003469 // Add properties in this container.
3470 for (ObjCContainerDecl::prop_iterator P = Container->prop_begin(),
3471 PEnd = Container->prop_end();
3472 P != PEnd;
Douglas Gregor73449212010-12-09 23:01:55 +00003473 ++P) {
3474 if (AddedProperties.insert(P->getIdentifier()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00003475 Results.MaybeAddResult(Result(*P, Results.getBasePriority(*P), 0),
3476 CurContext);
Douglas Gregor73449212010-12-09 23:01:55 +00003477 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003478
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003479 // Add nullary methods
3480 if (AllowNullaryMethods) {
3481 ASTContext &Context = Container->getASTContext();
Douglas Gregor8987b232011-09-27 23:30:47 +00003482 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003483 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
3484 MEnd = Container->meth_end();
3485 M != MEnd; ++M) {
3486 if (M->getSelector().isUnarySelector())
3487 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
3488 if (AddedProperties.insert(Name)) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003489 CodeCompletionBuilder Builder(Results.getAllocator(),
3490 Results.getCodeCompletionTUInfo());
David Blaikie581deb32012-06-06 20:45:41 +00003491 AddResultTypeChunk(Context, Policy, *M, Builder);
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003492 Builder.AddTypedTextChunk(
3493 Results.getAllocator().CopyString(Name->getName()));
3494
David Blaikie581deb32012-06-06 20:45:41 +00003495 Results.MaybeAddResult(Result(Builder.TakeString(), *M,
Douglas Gregorba103062012-03-27 23:34:16 +00003496 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003497 CurContext);
3498 }
3499 }
3500 }
3501
3502
Douglas Gregor95ac6552009-11-18 01:29:26 +00003503 // Add properties in referenced protocols.
3504 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3505 for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
3506 PEnd = Protocol->protocol_end();
3507 P != PEnd; ++P)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003508 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3509 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003510 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor322328b2009-11-18 22:32:06 +00003511 if (AllowCategories) {
3512 // Look through categories.
Douglas Gregord3297242013-01-16 23:00:23 +00003513 for (ObjCInterfaceDecl::known_categories_iterator
3514 Cat = IFace->known_categories_begin(),
3515 CatEnd = IFace->known_categories_end();
3516 Cat != CatEnd; ++Cat)
3517 AddObjCProperties(*Cat, AllowCategories, AllowNullaryMethods,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003518 CurContext, AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00003519 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003520
3521 // Look through protocols.
Ted Kremenek53b94412010-09-01 01:21:15 +00003522 for (ObjCInterfaceDecl::all_protocol_iterator
3523 I = IFace->all_referenced_protocol_begin(),
3524 E = IFace->all_referenced_protocol_end(); I != E; ++I)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003525 AddObjCProperties(*I, AllowCategories, AllowNullaryMethods, CurContext,
3526 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003527
3528 // Look in the superclass.
3529 if (IFace->getSuperClass())
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003530 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3531 AllowNullaryMethods, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003532 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003533 } else if (const ObjCCategoryDecl *Category
3534 = dyn_cast<ObjCCategoryDecl>(Container)) {
3535 // Look through protocols.
Ted Kremenek53b94412010-09-01 01:21:15 +00003536 for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
3537 PEnd = Category->protocol_end();
Douglas Gregor95ac6552009-11-18 01:29:26 +00003538 P != PEnd; ++P)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003539 AddObjCProperties(*P, AllowCategories, AllowNullaryMethods, CurContext,
3540 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003541 }
3542}
3543
Douglas Gregorf5cd27d2012-01-23 15:59:30 +00003544void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor81b747b2009-09-17 21:32:03 +00003545 SourceLocation OpLoc,
3546 bool IsArrow) {
Douglas Gregorf5cd27d2012-01-23 15:59:30 +00003547 if (!Base || !CodeCompleter)
Douglas Gregor81b747b2009-09-17 21:32:03 +00003548 return;
3549
Douglas Gregorf5cd27d2012-01-23 15:59:30 +00003550 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3551 if (ConvertedBase.isInvalid())
3552 return;
3553 Base = ConvertedBase.get();
3554
John McCall0a2c5e22010-08-25 06:19:51 +00003555 typedef CodeCompletionResult Result;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003556
Douglas Gregor81b747b2009-09-17 21:32:03 +00003557 QualType BaseType = Base->getType();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003558
3559 if (IsArrow) {
3560 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3561 BaseType = Ptr->getPointeeType();
3562 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor3cdee122010-08-26 16:36:48 +00003563 /*Do nothing*/ ;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003564 else
3565 return;
3566 }
3567
Douglas Gregor3da626b2011-07-07 16:03:39 +00003568 enum CodeCompletionContext::Kind contextKind;
3569
3570 if (IsArrow) {
3571 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3572 }
3573 else {
3574 if (BaseType->isObjCObjectPointerType() ||
3575 BaseType->isObjCObjectOrInterfaceType()) {
3576 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3577 }
3578 else {
3579 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3580 }
3581 }
3582
Douglas Gregor218937c2011-02-01 19:23:04 +00003583 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003584 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00003585 CodeCompletionContext(contextKind,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003586 BaseType),
3587 &ResultBuilder::IsMember);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003588 Results.EnterNewScope();
3589 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor3cdee122010-08-26 16:36:48 +00003590 // Indicate that we are performing a member access, and the cv-qualifiers
3591 // for the base object type.
3592 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3593
Douglas Gregor95ac6552009-11-18 01:29:26 +00003594 // Access to a C/C++ class, struct, or union.
Douglas Gregor45bcd432010-01-14 03:21:49 +00003595 Results.allowNestedNameSpecifiers();
Douglas Gregor0cc84042010-01-14 15:47:35 +00003596 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003597 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3598 CodeCompleter->includeGlobals());
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003599
David Blaikie4e4d0842012-03-11 07:00:24 +00003600 if (getLangOpts().CPlusPlus) {
Douglas Gregor95ac6552009-11-18 01:29:26 +00003601 if (!Results.empty()) {
3602 // The "template" keyword can follow "->" or "." in the grammar.
3603 // However, we only want to suggest the template keyword if something
3604 // is dependent.
3605 bool IsDependent = BaseType->isDependentType();
3606 if (!IsDependent) {
3607 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
3608 if (DeclContext *Ctx = (DeclContext *)DepScope->getEntity()) {
3609 IsDependent = Ctx->isDependentContext();
3610 break;
3611 }
3612 }
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003613
Douglas Gregor95ac6552009-11-18 01:29:26 +00003614 if (IsDependent)
Douglas Gregora4477812010-01-14 16:01:26 +00003615 Results.AddResult(Result("template"));
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003616 }
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003617 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003618 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3619 // Objective-C property reference.
Douglas Gregor73449212010-12-09 23:01:55 +00003620 AddedPropertiesSet AddedProperties;
Douglas Gregor95ac6552009-11-18 01:29:26 +00003621
3622 // Add property results based on our interface.
3623 const ObjCObjectPointerType *ObjCPtr
3624 = BaseType->getAsObjCInterfacePointerType();
3625 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003626 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3627 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003628 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003629
3630 // Add properties from the protocols in a qualified interface.
3631 for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(),
3632 E = ObjCPtr->qual_end();
3633 I != E; ++I)
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003634 AddObjCProperties(*I, true, /*AllowNullaryMethods=*/true, CurContext,
3635 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003636 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCallc12c5bb2010-05-15 11:32:37 +00003637 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor95ac6552009-11-18 01:29:26 +00003638 // Objective-C instance variable access.
3639 ObjCInterfaceDecl *Class = 0;
3640 if (const ObjCObjectPointerType *ObjCPtr
3641 = BaseType->getAs<ObjCObjectPointerType>())
3642 Class = ObjCPtr->getInterfaceDecl();
3643 else
John McCallc12c5bb2010-05-15 11:32:37 +00003644 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor95ac6552009-11-18 01:29:26 +00003645
3646 // Add all ivars from this class and its superclasses.
Douglas Gregor80f4f4c2010-01-14 16:08:12 +00003647 if (Class) {
3648 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3649 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor8071e422010-08-15 06:18:01 +00003650 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3651 CodeCompleter->includeGlobals());
Douglas Gregor95ac6552009-11-18 01:29:26 +00003652 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003653 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003654
3655 // FIXME: How do we cope with isa?
3656
3657 Results.ExitScope();
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003658
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003659 // Hand off the results found for code completion.
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003660 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003661 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003662 Results.data(),Results.size());
Douglas Gregor81b747b2009-09-17 21:32:03 +00003663}
3664
Douglas Gregor374929f2009-09-18 15:37:17 +00003665void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3666 if (!CodeCompleter)
3667 return;
3668
Douglas Gregor86d9a522009-09-21 16:56:56 +00003669 ResultBuilder::LookupFilter Filter = 0;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003670 enum CodeCompletionContext::Kind ContextKind
3671 = CodeCompletionContext::CCC_Other;
Douglas Gregor374929f2009-09-18 15:37:17 +00003672 switch ((DeclSpec::TST)TagSpec) {
3673 case DeclSpec::TST_enum:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003674 Filter = &ResultBuilder::IsEnum;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003675 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003676 break;
3677
3678 case DeclSpec::TST_union:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003679 Filter = &ResultBuilder::IsUnion;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003680 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003681 break;
3682
3683 case DeclSpec::TST_struct:
Douglas Gregor374929f2009-09-18 15:37:17 +00003684 case DeclSpec::TST_class:
Joao Matos6666ed42012-08-31 18:45:21 +00003685 case DeclSpec::TST_interface:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003686 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003687 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003688 break;
3689
3690 default:
David Blaikieb219cfc2011-09-23 05:06:16 +00003691 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregor374929f2009-09-18 15:37:17 +00003692 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003693
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003694 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3695 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregor5d2fc402010-01-14 03:27:13 +00003696 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCall0d6b1642010-04-23 18:46:30 +00003697
3698 // First pass: look for tags.
3699 Results.setFilter(Filter);
Douglas Gregor8071e422010-08-15 06:18:01 +00003700 LookupVisibleDecls(S, LookupTagName, Consumer,
3701 CodeCompleter->includeGlobals());
John McCall0d6b1642010-04-23 18:46:30 +00003702
Douglas Gregor8071e422010-08-15 06:18:01 +00003703 if (CodeCompleter->includeGlobals()) {
3704 // Second pass: look for nested name specifiers.
3705 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3706 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3707 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003708
Douglas Gregor52779fb2010-09-23 23:01:17 +00003709 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003710 Results.data(),Results.size());
Douglas Gregor374929f2009-09-18 15:37:17 +00003711}
3712
Douglas Gregor1a480c42010-08-27 17:35:51 +00003713void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003714 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003715 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00003716 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor1a480c42010-08-27 17:35:51 +00003717 Results.EnterNewScope();
3718 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3719 Results.AddResult("const");
3720 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3721 Results.AddResult("volatile");
David Blaikie4e4d0842012-03-11 07:00:24 +00003722 if (getLangOpts().C99 &&
Douglas Gregor1a480c42010-08-27 17:35:51 +00003723 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3724 Results.AddResult("restrict");
3725 Results.ExitScope();
3726 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003727 Results.getCompletionContext(),
Douglas Gregor1a480c42010-08-27 17:35:51 +00003728 Results.data(), Results.size());
3729}
3730
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003731void Sema::CodeCompleteCase(Scope *S) {
John McCall781472f2010-08-25 08:40:02 +00003732 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003733 return;
John McCalla8e0cd82011-08-06 07:30:58 +00003734
John McCall781472f2010-08-25 08:40:02 +00003735 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCalla8e0cd82011-08-06 07:30:58 +00003736 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3737 if (!type->isEnumeralType()) {
3738 CodeCompleteExpressionData Data(type);
Douglas Gregorfb629412010-08-23 21:17:50 +00003739 Data.IntegralConstantExpression = true;
3740 CodeCompleteExpression(S, Data);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003741 return;
Douglas Gregorf9578432010-07-28 21:50:18 +00003742 }
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003743
3744 // Code-complete the cases of a switch statement over an enumeration type
3745 // by providing the list of
John McCalla8e0cd82011-08-06 07:30:58 +00003746 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregorb92a4082012-06-12 13:44:08 +00003747 if (EnumDecl *Def = Enum->getDefinition())
3748 Enum = Def;
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003749
3750 // Determine which enumerators we have already seen in the switch statement.
3751 // FIXME: Ideally, we would also be able to look *past* the code-completion
3752 // token, in case we are code-completing in the middle of the switch and not
3753 // at the end. However, we aren't able to do so at the moment.
3754 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003755 NestedNameSpecifier *Qualifier = 0;
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003756 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3757 SC = SC->getNextSwitchCase()) {
3758 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3759 if (!Case)
3760 continue;
3761
3762 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3763 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3764 if (EnumConstantDecl *Enumerator
3765 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3766 // We look into the AST of the case statement to determine which
3767 // enumerator was named. Alternatively, we could compute the value of
3768 // the integral constant expression, then compare it against the
3769 // values of each enumerator. However, value-based approach would not
3770 // work as well with C++ templates where enumerators declared within a
3771 // template are type- and value-dependent.
3772 EnumeratorsSeen.insert(Enumerator);
3773
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003774 // If this is a qualified-id, keep track of the nested-name-specifier
3775 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003776 //
3777 // switch (TagD.getKind()) {
3778 // case TagDecl::TK_enum:
3779 // break;
3780 // case XXX
3781 //
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003782 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003783 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3784 // TK_struct, and TK_class.
Douglas Gregora2813ce2009-10-23 18:54:35 +00003785 Qualifier = DRE->getQualifier();
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003786 }
3787 }
3788
David Blaikie4e4d0842012-03-11 07:00:24 +00003789 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003790 // If there are no prior enumerators in C++, check whether we have to
3791 // qualify the names of the enumerators that we suggest, because they
3792 // may not be visible in this scope.
Douglas Gregorb223d8c2012-02-01 05:02:47 +00003793 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003794 }
3795
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003796 // Add any enumerators that have not yet been mentioned.
Douglas Gregor218937c2011-02-01 19:23:04 +00003797 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003798 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00003799 CodeCompletionContext::CCC_Expression);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003800 Results.EnterNewScope();
3801 for (EnumDecl::enumerator_iterator E = Enum->enumerator_begin(),
3802 EEnd = Enum->enumerator_end();
3803 E != EEnd; ++E) {
David Blaikie581deb32012-06-06 20:45:41 +00003804 if (EnumeratorsSeen.count(*E))
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003805 continue;
3806
Douglas Gregord1f09b42013-01-31 04:52:16 +00003807 CodeCompletionResult R(*E, CCP_EnumInCase, Qualifier);
Douglas Gregor5c722c702011-02-18 23:30:37 +00003808 Results.AddResult(R, CurContext, 0, false);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003809 }
3810 Results.ExitScope();
Douglas Gregor2f880e42010-04-06 20:02:15 +00003811
Douglas Gregor3da626b2011-07-07 16:03:39 +00003812 //We need to make sure we're setting the right context,
3813 //so only say we include macros if the code completer says we do
3814 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3815 if (CodeCompleter->includeMacros()) {
Douglas Gregor3644d972012-10-09 16:01:50 +00003816 AddMacroResults(PP, Results, false);
Douglas Gregor3da626b2011-07-07 16:03:39 +00003817 kind = CodeCompletionContext::CCC_OtherWithMacros;
3818 }
3819
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003820 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00003821 kind,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003822 Results.data(),Results.size());
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003823}
3824
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003825namespace {
3826 struct IsBetterOverloadCandidate {
3827 Sema &S;
John McCall5769d612010-02-08 23:07:23 +00003828 SourceLocation Loc;
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003829
3830 public:
John McCall5769d612010-02-08 23:07:23 +00003831 explicit IsBetterOverloadCandidate(Sema &S, SourceLocation Loc)
3832 : S(S), Loc(Loc) { }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003833
3834 bool
3835 operator()(const OverloadCandidate &X, const OverloadCandidate &Y) const {
John McCall120d63c2010-08-24 20:38:10 +00003836 return isBetterOverloadCandidate(S, X, Y, Loc);
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003837 }
3838 };
3839}
3840
Ahmed Charles13a140c2012-02-25 11:00:22 +00003841static bool anyNullArguments(llvm::ArrayRef<Expr*> Args) {
3842 if (Args.size() && !Args.data())
Douglas Gregord28dcd72010-05-30 06:10:08 +00003843 return true;
Ahmed Charles13a140c2012-02-25 11:00:22 +00003844
3845 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregord28dcd72010-05-30 06:10:08 +00003846 if (!Args[I])
3847 return true;
Ahmed Charles13a140c2012-02-25 11:00:22 +00003848
Douglas Gregord28dcd72010-05-30 06:10:08 +00003849 return false;
3850}
3851
Richard Trieuf81e5a92011-09-09 02:00:50 +00003852void Sema::CodeCompleteCall(Scope *S, Expr *FnIn,
Ahmed Charles13a140c2012-02-25 11:00:22 +00003853 llvm::ArrayRef<Expr *> Args) {
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003854 if (!CodeCompleter)
3855 return;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003856
3857 // When we're code-completing for a call, we fall back to ordinary
3858 // name code-completion whenever we can't produce specific
3859 // results. We may want to revisit this strategy in the future,
3860 // e.g., by merging the two kinds of results.
3861
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003862 Expr *Fn = (Expr *)FnIn;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003863
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003864 // Ignore type-dependent call expressions entirely.
Ahmed Charles13a140c2012-02-25 11:00:22 +00003865 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3866 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003867 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003868 return;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003869 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003870
John McCall3b4294e2009-12-16 12:17:52 +00003871 // Build an overload candidate set based on the functions we find.
John McCall5769d612010-02-08 23:07:23 +00003872 SourceLocation Loc = Fn->getExprLoc();
3873 OverloadCandidateSet CandidateSet(Loc);
John McCall3b4294e2009-12-16 12:17:52 +00003874
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003875 // FIXME: What if we're calling something that isn't a function declaration?
3876 // FIXME: What if we're calling a pseudo-destructor?
3877 // FIXME: What if we're calling a member function?
3878
Douglas Gregorc0265402010-01-21 15:46:19 +00003879 typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
Chris Lattner5f9e2722011-07-23 10:55:15 +00003880 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorc0265402010-01-21 15:46:19 +00003881
John McCall3b4294e2009-12-16 12:17:52 +00003882 Expr *NakedFn = Fn->IgnoreParenCasts();
3883 if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charles13a140c2012-02-25 11:00:22 +00003884 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
John McCall3b4294e2009-12-16 12:17:52 +00003885 /*PartialOverloading=*/ true);
3886 else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) {
3887 FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl());
Douglas Gregorc0265402010-01-21 15:46:19 +00003888 if (FDecl) {
David Blaikie4e4d0842012-03-11 07:00:24 +00003889 if (!getLangOpts().CPlusPlus ||
Douglas Gregord28dcd72010-05-30 06:10:08 +00003890 !FDecl->getType()->getAs<FunctionProtoType>())
Douglas Gregorc0265402010-01-21 15:46:19 +00003891 Results.push_back(ResultCandidate(FDecl));
3892 else
John McCall86820f52010-01-26 01:37:31 +00003893 // FIXME: access?
Ahmed Charles13a140c2012-02-25 11:00:22 +00003894 AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), Args,
3895 CandidateSet, false, /*PartialOverloading*/true);
Douglas Gregorc0265402010-01-21 15:46:19 +00003896 }
John McCall3b4294e2009-12-16 12:17:52 +00003897 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003898
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003899 QualType ParamType;
3900
Douglas Gregorc0265402010-01-21 15:46:19 +00003901 if (!CandidateSet.empty()) {
3902 // Sort the overload candidate set by placing the best overloads first.
3903 std::stable_sort(CandidateSet.begin(), CandidateSet.end(),
John McCall5769d612010-02-08 23:07:23 +00003904 IsBetterOverloadCandidate(*this, Loc));
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003905
Douglas Gregorc0265402010-01-21 15:46:19 +00003906 // Add the remaining viable overload candidates as code-completion reslults.
3907 for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(),
3908 CandEnd = CandidateSet.end();
3909 Cand != CandEnd; ++Cand) {
3910 if (Cand->Viable)
3911 Results.push_back(ResultCandidate(Cand->Function));
3912 }
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003913
3914 // From the viable candidates, try to determine the type of this parameter.
3915 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
3916 if (const FunctionType *FType = Results[I].getFunctionType())
3917 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType))
Ahmed Charles13a140c2012-02-25 11:00:22 +00003918 if (Args.size() < Proto->getNumArgs()) {
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003919 if (ParamType.isNull())
Ahmed Charles13a140c2012-02-25 11:00:22 +00003920 ParamType = Proto->getArgType(Args.size());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003921 else if (!Context.hasSameUnqualifiedType(
3922 ParamType.getNonReferenceType(),
Ahmed Charles13a140c2012-02-25 11:00:22 +00003923 Proto->getArgType(Args.size()).getNonReferenceType())) {
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003924 ParamType = QualType();
3925 break;
3926 }
3927 }
3928 }
3929 } else {
3930 // Try to determine the parameter type from the type of the expression
3931 // being called.
3932 QualType FunctionType = Fn->getType();
3933 if (const PointerType *Ptr = FunctionType->getAs<PointerType>())
3934 FunctionType = Ptr->getPointeeType();
3935 else if (const BlockPointerType *BlockPtr
3936 = FunctionType->getAs<BlockPointerType>())
3937 FunctionType = BlockPtr->getPointeeType();
3938 else if (const MemberPointerType *MemPtr
3939 = FunctionType->getAs<MemberPointerType>())
3940 FunctionType = MemPtr->getPointeeType();
3941
3942 if (const FunctionProtoType *Proto
3943 = FunctionType->getAs<FunctionProtoType>()) {
Ahmed Charles13a140c2012-02-25 11:00:22 +00003944 if (Args.size() < Proto->getNumArgs())
3945 ParamType = Proto->getArgType(Args.size());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003946 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003947 }
Douglas Gregoref96eac2009-12-11 19:06:04 +00003948
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003949 if (ParamType.isNull())
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003950 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003951 else
3952 CodeCompleteExpression(S, ParamType);
3953
Douglas Gregor2e4c7a52010-04-06 20:19:47 +00003954 if (!Results.empty())
Ahmed Charles13a140c2012-02-25 11:00:22 +00003955 CodeCompleter->ProcessOverloadCandidates(*this, Args.size(), Results.data(),
Douglas Gregoref96eac2009-12-11 19:06:04 +00003956 Results.size());
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003957}
3958
John McCalld226f652010-08-21 09:40:31 +00003959void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
3960 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003961 if (!VD) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003962 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003963 return;
3964 }
3965
3966 CodeCompleteExpression(S, VD->getType());
3967}
3968
3969void Sema::CodeCompleteReturn(Scope *S) {
3970 QualType ResultType;
3971 if (isa<BlockDecl>(CurContext)) {
3972 if (BlockScopeInfo *BSI = getCurBlock())
3973 ResultType = BSI->ReturnType;
3974 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
3975 ResultType = Function->getResultType();
3976 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
3977 ResultType = Method->getResultType();
3978
3979 if (ResultType.isNull())
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003980 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003981 else
3982 CodeCompleteExpression(S, ResultType);
3983}
3984
Douglas Gregord2d8be62011-07-30 08:36:53 +00003985void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregord2d8be62011-07-30 08:36:53 +00003986 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003987 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord2d8be62011-07-30 08:36:53 +00003988 mapCodeCompletionContext(*this, PCC_Statement));
3989 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3990 Results.EnterNewScope();
3991
3992 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3993 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3994 CodeCompleter->includeGlobals());
3995
3996 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
3997
3998 // "else" block
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003999 CodeCompletionBuilder Builder(Results.getAllocator(),
4000 Results.getCodeCompletionTUInfo());
Douglas Gregord2d8be62011-07-30 08:36:53 +00004001 Builder.AddTypedTextChunk("else");
Douglas Gregorf11641a2012-02-16 17:49:04 +00004002 if (Results.includeCodePatterns()) {
4003 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4004 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4005 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4006 Builder.AddPlaceholderChunk("statements");
4007 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4008 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4009 }
Douglas Gregord2d8be62011-07-30 08:36:53 +00004010 Results.AddResult(Builder.TakeString());
4011
4012 // "else if" block
4013 Builder.AddTypedTextChunk("else");
4014 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4015 Builder.AddTextChunk("if");
4016 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4017 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00004018 if (getLangOpts().CPlusPlus)
Douglas Gregord2d8be62011-07-30 08:36:53 +00004019 Builder.AddPlaceholderChunk("condition");
4020 else
4021 Builder.AddPlaceholderChunk("expression");
4022 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf11641a2012-02-16 17:49:04 +00004023 if (Results.includeCodePatterns()) {
4024 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4025 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4026 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4027 Builder.AddPlaceholderChunk("statements");
4028 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4029 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4030 }
Douglas Gregord2d8be62011-07-30 08:36:53 +00004031 Results.AddResult(Builder.TakeString());
4032
4033 Results.ExitScope();
4034
4035 if (S->getFnParent())
David Blaikie4e4d0842012-03-11 07:00:24 +00004036 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregord2d8be62011-07-30 08:36:53 +00004037
4038 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00004039 AddMacroResults(PP, Results, false);
Douglas Gregord2d8be62011-07-30 08:36:53 +00004040
4041 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4042 Results.data(),Results.size());
4043}
4044
Richard Trieuf81e5a92011-09-09 02:00:50 +00004045void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004046 if (LHS)
4047 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4048 else
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004049 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004050}
4051
Jeffrey Yasskin9ab14542010-04-08 16:38:48 +00004052void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor81b747b2009-09-17 21:32:03 +00004053 bool EnteringContext) {
4054 if (!SS.getScopeRep() || !CodeCompleter)
4055 return;
4056
Douglas Gregor86d9a522009-09-21 16:56:56 +00004057 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4058 if (!Ctx)
4059 return;
Douglas Gregord1cd31a2009-12-11 18:28:39 +00004060
4061 // Try to instantiate any non-dependent declaration contexts before
4062 // we look in them.
John McCall77bb1aa2010-05-01 00:40:08 +00004063 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregord1cd31a2009-12-11 18:28:39 +00004064 return;
4065
Douglas Gregor218937c2011-02-01 19:23:04 +00004066 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004067 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004068 CodeCompletionContext::CCC_Name);
Douglas Gregorf6961522010-08-27 21:18:54 +00004069 Results.EnterNewScope();
Douglas Gregor52779fb2010-09-23 23:01:17 +00004070
Douglas Gregor86d9a522009-09-21 16:56:56 +00004071 // The "template" keyword can follow "::" in the grammar, but only
4072 // put it into the grammar if the nested-name-specifier is dependent.
4073 NestedNameSpecifier *NNS = (NestedNameSpecifier *)SS.getScopeRep();
4074 if (!Results.empty() && NNS->isDependent())
Douglas Gregora4477812010-01-14 16:01:26 +00004075 Results.AddResult("template");
Douglas Gregorf6961522010-08-27 21:18:54 +00004076
4077 // Add calls to overridden virtual functions, if there are any.
4078 //
4079 // FIXME: This isn't wonderful, because we don't know whether we're actually
4080 // in a context that permits expressions. This is a general issue with
4081 // qualified-id completions.
4082 if (!EnteringContext)
4083 MaybeAddOverrideCalls(*this, Ctx, Results);
4084 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004085
Douglas Gregorf6961522010-08-27 21:18:54 +00004086 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4087 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4088
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004089 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor430d7a12011-07-25 17:48:11 +00004090 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004091 Results.data(),Results.size());
Douglas Gregor81b747b2009-09-17 21:32:03 +00004092}
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004093
4094void Sema::CodeCompleteUsing(Scope *S) {
4095 if (!CodeCompleter)
4096 return;
4097
Douglas Gregor218937c2011-02-01 19:23:04 +00004098 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004099 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00004100 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4101 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004102 Results.EnterNewScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004103
4104 // If we aren't in class scope, we could see the "namespace" keyword.
4105 if (!S->isClassScope())
John McCall0a2c5e22010-08-25 06:19:51 +00004106 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor86d9a522009-09-21 16:56:56 +00004107
4108 // After "using", we can see anything that would start a
4109 // nested-name-specifier.
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004110 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004111 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4112 CodeCompleter->includeGlobals());
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004113 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004114
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004115 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004116 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004117 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004118}
4119
4120void Sema::CodeCompleteUsingDirective(Scope *S) {
4121 if (!CodeCompleter)
4122 return;
4123
Douglas Gregor86d9a522009-09-21 16:56:56 +00004124 // After "using namespace", we expect to see a namespace name or namespace
4125 // alias.
Douglas Gregor218937c2011-02-01 19:23:04 +00004126 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004127 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004128 CodeCompletionContext::CCC_Namespace,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004129 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004130 Results.EnterNewScope();
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004131 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004132 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4133 CodeCompleter->includeGlobals());
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004134 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004135 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor8071e422010-08-15 06:18:01 +00004136 CodeCompletionContext::CCC_Namespace,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004137 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004138}
4139
4140void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4141 if (!CodeCompleter)
4142 return;
4143
Douglas Gregor86d9a522009-09-21 16:56:56 +00004144 DeclContext *Ctx = (DeclContext *)S->getEntity();
4145 if (!S->getParent())
4146 Ctx = Context.getTranslationUnitDecl();
4147
Douglas Gregor52779fb2010-09-23 23:01:17 +00004148 bool SuppressedGlobalResults
4149 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4150
Douglas Gregor218937c2011-02-01 19:23:04 +00004151 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004152 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00004153 SuppressedGlobalResults
4154 ? CodeCompletionContext::CCC_Namespace
4155 : CodeCompletionContext::CCC_Other,
4156 &ResultBuilder::IsNamespace);
4157
4158 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00004159 // We only want to see those namespaces that have already been defined
4160 // within this scope, because its likely that the user is creating an
4161 // extended namespace declaration. Keep track of the most recent
4162 // definition of each namespace.
4163 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4164 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4165 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4166 NS != NSEnd; ++NS)
David Blaikie581deb32012-06-06 20:45:41 +00004167 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor86d9a522009-09-21 16:56:56 +00004168
4169 // Add the most recent definition (or extended definition) of each
4170 // namespace to the list of results.
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004171 Results.EnterNewScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004172 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregorba103062012-03-27 23:34:16 +00004173 NS = OrigToLatest.begin(),
4174 NSEnd = OrigToLatest.end();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004175 NS != NSEnd; ++NS)
Douglas Gregord1f09b42013-01-31 04:52:16 +00004176 Results.AddResult(CodeCompletionResult(
4177 NS->second, Results.getBasePriority(NS->second), 0),
Douglas Gregor608300b2010-01-14 16:14:35 +00004178 CurContext, 0, false);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004179 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004180 }
4181
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004182 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004183 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004184 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004185}
4186
4187void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4188 if (!CodeCompleter)
4189 return;
4190
Douglas Gregor86d9a522009-09-21 16:56:56 +00004191 // After "namespace", we expect to see a namespace or alias.
Douglas Gregor218937c2011-02-01 19:23:04 +00004192 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004193 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004194 CodeCompletionContext::CCC_Namespace,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004195 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004196 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004197 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4198 CodeCompleter->includeGlobals());
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004199 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004200 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004201 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004202}
4203
Douglas Gregored8d3222009-09-18 20:05:18 +00004204void Sema::CodeCompleteOperatorName(Scope *S) {
4205 if (!CodeCompleter)
4206 return;
Douglas Gregor86d9a522009-09-21 16:56:56 +00004207
John McCall0a2c5e22010-08-25 06:19:51 +00004208 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004209 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004210 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004211 CodeCompletionContext::CCC_Type,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004212 &ResultBuilder::IsType);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004213 Results.EnterNewScope();
Douglas Gregored8d3222009-09-18 20:05:18 +00004214
Douglas Gregor86d9a522009-09-21 16:56:56 +00004215 // Add the names of overloadable operators.
4216#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4217 if (std::strcmp(Spelling, "?")) \
Douglas Gregora4477812010-01-14 16:01:26 +00004218 Results.AddResult(Result(Spelling));
Douglas Gregor86d9a522009-09-21 16:56:56 +00004219#include "clang/Basic/OperatorKinds.def"
4220
4221 // Add any type names visible from the current scope
Douglas Gregor45bcd432010-01-14 03:21:49 +00004222 Results.allowNestedNameSpecifiers();
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004223 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004224 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4225 CodeCompleter->includeGlobals());
Douglas Gregor86d9a522009-09-21 16:56:56 +00004226
4227 // Add any type specifiers
David Blaikie4e4d0842012-03-11 07:00:24 +00004228 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004229 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004230
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004231 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor8071e422010-08-15 06:18:01 +00004232 CodeCompletionContext::CCC_Type,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004233 Results.data(),Results.size());
Douglas Gregored8d3222009-09-18 20:05:18 +00004234}
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004235
Douglas Gregor0133f522010-08-28 00:00:50 +00004236void Sema::CodeCompleteConstructorInitializer(Decl *ConstructorD,
Sean Huntcbb67482011-01-08 20:30:50 +00004237 CXXCtorInitializer** Initializers,
Douglas Gregor0133f522010-08-28 00:00:50 +00004238 unsigned NumInitializers) {
Douglas Gregor8987b232011-09-27 23:30:47 +00004239 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor0133f522010-08-28 00:00:50 +00004240 CXXConstructorDecl *Constructor
4241 = static_cast<CXXConstructorDecl *>(ConstructorD);
4242 if (!Constructor)
4243 return;
4244
Douglas Gregor218937c2011-02-01 19:23:04 +00004245 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004246 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00004247 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregor0133f522010-08-28 00:00:50 +00004248 Results.EnterNewScope();
4249
4250 // Fill in any already-initialized fields or base classes.
4251 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4252 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
4253 for (unsigned I = 0; I != NumInitializers; ++I) {
4254 if (Initializers[I]->isBaseInitializer())
4255 InitializedBases.insert(
4256 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4257 else
Francois Pichet00eb3f92010-12-04 09:14:42 +00004258 InitializedFields.insert(cast<FieldDecl>(
4259 Initializers[I]->getAnyMember()));
Douglas Gregor0133f522010-08-28 00:00:50 +00004260 }
4261
4262 // Add completions for base classes.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004263 CodeCompletionBuilder Builder(Results.getAllocator(),
4264 Results.getCodeCompletionTUInfo());
Douglas Gregor0c431c82010-08-29 19:27:27 +00004265 bool SawLastInitializer = (NumInitializers == 0);
Douglas Gregor0133f522010-08-28 00:00:50 +00004266 CXXRecordDecl *ClassDecl = Constructor->getParent();
4267 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->bases_begin(),
4268 BaseEnd = ClassDecl->bases_end();
4269 Base != BaseEnd; ++Base) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004270 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4271 SawLastInitializer
4272 = NumInitializers > 0 &&
4273 Initializers[NumInitializers - 1]->isBaseInitializer() &&
4274 Context.hasSameUnqualifiedType(Base->getType(),
4275 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregor0133f522010-08-28 00:00:50 +00004276 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004277 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004278
Douglas Gregor218937c2011-02-01 19:23:04 +00004279 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004280 Results.getAllocator().CopyString(
John McCallf85e1932011-06-15 23:02:42 +00004281 Base->getType().getAsString(Policy)));
Douglas Gregor218937c2011-02-01 19:23:04 +00004282 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4283 Builder.AddPlaceholderChunk("args");
4284 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4285 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004286 SawLastInitializer? CCP_NextInitializer
4287 : CCP_MemberDeclaration));
4288 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004289 }
4290
4291 // Add completions for virtual base classes.
4292 for (CXXRecordDecl::base_class_iterator Base = ClassDecl->vbases_begin(),
4293 BaseEnd = ClassDecl->vbases_end();
4294 Base != BaseEnd; ++Base) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004295 if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) {
4296 SawLastInitializer
4297 = NumInitializers > 0 &&
4298 Initializers[NumInitializers - 1]->isBaseInitializer() &&
4299 Context.hasSameUnqualifiedType(Base->getType(),
4300 QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0));
Douglas Gregor0133f522010-08-28 00:00:50 +00004301 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004302 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004303
Douglas Gregor218937c2011-02-01 19:23:04 +00004304 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004305 Builder.getAllocator().CopyString(
John McCallf85e1932011-06-15 23:02:42 +00004306 Base->getType().getAsString(Policy)));
Douglas Gregor218937c2011-02-01 19:23:04 +00004307 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4308 Builder.AddPlaceholderChunk("args");
4309 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4310 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004311 SawLastInitializer? CCP_NextInitializer
4312 : CCP_MemberDeclaration));
4313 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004314 }
4315
4316 // Add completions for members.
4317 for (CXXRecordDecl::field_iterator Field = ClassDecl->field_begin(),
4318 FieldEnd = ClassDecl->field_end();
4319 Field != FieldEnd; ++Field) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004320 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) {
4321 SawLastInitializer
4322 = NumInitializers > 0 &&
Francois Pichet00eb3f92010-12-04 09:14:42 +00004323 Initializers[NumInitializers - 1]->isAnyMemberInitializer() &&
David Blaikie581deb32012-06-06 20:45:41 +00004324 Initializers[NumInitializers - 1]->getAnyMember() == *Field;
Douglas Gregor0133f522010-08-28 00:00:50 +00004325 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004326 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004327
4328 if (!Field->getDeclName())
4329 continue;
4330
Douglas Gregordae68752011-02-01 22:57:45 +00004331 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00004332 Field->getIdentifier()->getName()));
4333 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4334 Builder.AddPlaceholderChunk("args");
4335 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4336 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004337 SawLastInitializer? CCP_NextInitializer
Douglas Gregora67e03f2010-09-09 21:42:20 +00004338 : CCP_MemberDeclaration,
Douglas Gregorba103062012-03-27 23:34:16 +00004339 CXCursor_MemberRef,
4340 CXAvailability_Available,
David Blaikie581deb32012-06-06 20:45:41 +00004341 *Field));
Douglas Gregor0c431c82010-08-29 19:27:27 +00004342 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004343 }
4344 Results.ExitScope();
4345
Douglas Gregor52779fb2010-09-23 23:01:17 +00004346 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor0133f522010-08-28 00:00:50 +00004347 Results.data(), Results.size());
4348}
4349
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004350/// \brief Determine whether this scope denotes a namespace.
4351static bool isNamespaceScope(Scope *S) {
4352 DeclContext *DC = static_cast<DeclContext *>(S->getEntity());
4353 if (!DC)
4354 return false;
4355
4356 return DC->isFileContext();
4357}
4358
4359void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4360 bool AfterAmpersand) {
4361 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004362 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004363 CodeCompletionContext::CCC_Other);
4364 Results.EnterNewScope();
4365
4366 // Note what has already been captured.
4367 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4368 bool IncludedThis = false;
4369 for (SmallVectorImpl<LambdaCapture>::iterator C = Intro.Captures.begin(),
4370 CEnd = Intro.Captures.end();
4371 C != CEnd; ++C) {
4372 if (C->Kind == LCK_This) {
4373 IncludedThis = true;
4374 continue;
4375 }
4376
4377 Known.insert(C->Id);
4378 }
4379
4380 // Look for other capturable variables.
4381 for (; S && !isNamespaceScope(S); S = S->getParent()) {
4382 for (Scope::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
4383 D != DEnd; ++D) {
4384 VarDecl *Var = dyn_cast<VarDecl>(*D);
4385 if (!Var ||
4386 !Var->hasLocalStorage() ||
4387 Var->hasAttr<BlocksAttr>())
4388 continue;
4389
4390 if (Known.insert(Var->getIdentifier()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00004391 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
4392 CurContext, 0, false);
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004393 }
4394 }
4395
4396 // Add 'this', if it would be valid.
4397 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4398 addThisCompletion(*this, Results);
4399
4400 Results.ExitScope();
4401
4402 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4403 Results.data(), Results.size());
4404}
4405
James Dennetta40f7922012-06-14 03:11:41 +00004406/// Macro that optionally prepends an "@" to the string literal passed in via
4407/// Keyword, depending on whether NeedAt is true or false.
4408#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4409
Douglas Gregorbca403c2010-01-13 23:51:12 +00004410static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004411 ResultBuilder &Results,
4412 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004413 typedef CodeCompletionResult Result;
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004414 // Since we have an implementation, we can end it.
James Dennetta40f7922012-06-14 03:11:41 +00004415 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004416
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004417 CodeCompletionBuilder Builder(Results.getAllocator(),
4418 Results.getCodeCompletionTUInfo());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004419 if (LangOpts.ObjC2) {
4420 // @dynamic
James Dennetta40f7922012-06-14 03:11:41 +00004421 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004422 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4423 Builder.AddPlaceholderChunk("property");
4424 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004425
4426 // @synthesize
James Dennetta40f7922012-06-14 03:11:41 +00004427 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004428 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4429 Builder.AddPlaceholderChunk("property");
4430 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004431 }
4432}
4433
Douglas Gregorbca403c2010-01-13 23:51:12 +00004434static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004435 ResultBuilder &Results,
4436 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004437 typedef CodeCompletionResult Result;
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004438
4439 // Since we have an interface or protocol, we can end it.
James Dennetta40f7922012-06-14 03:11:41 +00004440 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004441
4442 if (LangOpts.ObjC2) {
4443 // @property
James Dennetta40f7922012-06-14 03:11:41 +00004444 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004445
4446 // @required
James Dennetta40f7922012-06-14 03:11:41 +00004447 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004448
4449 // @optional
James Dennetta40f7922012-06-14 03:11:41 +00004450 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004451 }
4452}
4453
Douglas Gregorbca403c2010-01-13 23:51:12 +00004454static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004455 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004456 CodeCompletionBuilder Builder(Results.getAllocator(),
4457 Results.getCodeCompletionTUInfo());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004458
4459 // @class name ;
James Dennetta40f7922012-06-14 03:11:41 +00004460 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004461 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4462 Builder.AddPlaceholderChunk("name");
4463 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004464
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004465 if (Results.includeCodePatterns()) {
4466 // @interface name
4467 // FIXME: Could introduce the whole pattern, including superclasses and
4468 // such.
James Dennetta40f7922012-06-14 03:11:41 +00004469 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004470 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4471 Builder.AddPlaceholderChunk("class");
4472 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004473
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004474 // @protocol name
James Dennetta40f7922012-06-14 03:11:41 +00004475 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004476 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4477 Builder.AddPlaceholderChunk("protocol");
4478 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004479
4480 // @implementation name
James Dennetta40f7922012-06-14 03:11:41 +00004481 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004482 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4483 Builder.AddPlaceholderChunk("class");
4484 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004485 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004486
4487 // @compatibility_alias name
James Dennetta40f7922012-06-14 03:11:41 +00004488 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004489 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4490 Builder.AddPlaceholderChunk("alias");
4491 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4492 Builder.AddPlaceholderChunk("class");
4493 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor06898632013-03-07 23:26:24 +00004494
4495 if (Results.getSema().getLangOpts().Modules) {
4496 // @import name
4497 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "import"));
4498 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4499 Builder.AddPlaceholderChunk("module");
4500 Results.AddResult(Result(Builder.TakeString()));
4501 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004502}
4503
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004504void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004505 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004506 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004507 CodeCompletionContext::CCC_Other);
Douglas Gregorc464ae82009-12-07 09:27:33 +00004508 Results.EnterNewScope();
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004509 if (isa<ObjCImplDecl>(CurContext))
David Blaikie4e4d0842012-03-11 07:00:24 +00004510 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004511 else if (CurContext->isObjCContainer())
David Blaikie4e4d0842012-03-11 07:00:24 +00004512 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004513 else
Douglas Gregorbca403c2010-01-13 23:51:12 +00004514 AddObjCTopLevelResults(Results, false);
Douglas Gregorc464ae82009-12-07 09:27:33 +00004515 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004516 HandleCodeCompleteResults(this, CodeCompleter,
4517 CodeCompletionContext::CCC_Other,
4518 Results.data(),Results.size());
Douglas Gregorc464ae82009-12-07 09:27:33 +00004519}
4520
Douglas Gregorbca403c2010-01-13 23:51:12 +00004521static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004522 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004523 CodeCompletionBuilder Builder(Results.getAllocator(),
4524 Results.getCodeCompletionTUInfo());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004525
4526 // @encode ( type-name )
Douglas Gregor8ca72082011-10-18 21:20:17 +00004527 const char *EncodeType = "char[]";
David Blaikie4e4d0842012-03-11 07:00:24 +00004528 if (Results.getSema().getLangOpts().CPlusPlus ||
4529 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004530 EncodeType = "const char[]";
Douglas Gregor8ca72082011-10-18 21:20:17 +00004531 Builder.AddResultTypeChunk(EncodeType);
James Dennetta40f7922012-06-14 03:11:41 +00004532 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004533 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4534 Builder.AddPlaceholderChunk("type-name");
4535 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4536 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004537
4538 // @protocol ( protocol-name )
Douglas Gregor8ca72082011-10-18 21:20:17 +00004539 Builder.AddResultTypeChunk("Protocol *");
James Dennetta40f7922012-06-14 03:11:41 +00004540 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004541 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4542 Builder.AddPlaceholderChunk("protocol-name");
4543 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4544 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004545
4546 // @selector ( selector )
Douglas Gregor8ca72082011-10-18 21:20:17 +00004547 Builder.AddResultTypeChunk("SEL");
James Dennetta40f7922012-06-14 03:11:41 +00004548 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004549 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4550 Builder.AddPlaceholderChunk("selector");
4551 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4552 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004553
4554 // @"string"
4555 Builder.AddResultTypeChunk("NSString *");
4556 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4557 Builder.AddPlaceholderChunk("string");
4558 Builder.AddTextChunk("\"");
4559 Results.AddResult(Result(Builder.TakeString()));
4560
Douglas Gregor79615892012-07-17 23:24:47 +00004561 // @[objects, ...]
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004562 Builder.AddResultTypeChunk("NSArray *");
James Dennetta40f7922012-06-14 03:11:41 +00004563 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004564 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004565 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4566 Results.AddResult(Result(Builder.TakeString()));
4567
Douglas Gregor79615892012-07-17 23:24:47 +00004568 // @{key : object, ...}
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004569 Builder.AddResultTypeChunk("NSDictionary *");
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("key");
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004572 Builder.AddChunk(CodeCompletionString::CK_Colon);
4573 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4574 Builder.AddPlaceholderChunk("object, ...");
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004575 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4576 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004577
Douglas Gregor79615892012-07-17 23:24:47 +00004578 // @(expression)
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004579 Builder.AddResultTypeChunk("id");
4580 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004581 Builder.AddPlaceholderChunk("expression");
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004582 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4583 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004584}
4585
Douglas Gregorbca403c2010-01-13 23:51:12 +00004586static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004587 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004588 CodeCompletionBuilder Builder(Results.getAllocator(),
4589 Results.getCodeCompletionTUInfo());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004590
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004591 if (Results.includeCodePatterns()) {
4592 // @try { statements } @catch ( declaration ) { statements } @finally
4593 // { statements }
James Dennetta40f7922012-06-14 03:11:41 +00004594 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004595 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4596 Builder.AddPlaceholderChunk("statements");
4597 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4598 Builder.AddTextChunk("@catch");
4599 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4600 Builder.AddPlaceholderChunk("parameter");
4601 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4602 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4603 Builder.AddPlaceholderChunk("statements");
4604 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4605 Builder.AddTextChunk("@finally");
4606 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4607 Builder.AddPlaceholderChunk("statements");
4608 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4609 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004610 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004611
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004612 // @throw
James Dennetta40f7922012-06-14 03:11:41 +00004613 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004614 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4615 Builder.AddPlaceholderChunk("expression");
4616 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004617
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004618 if (Results.includeCodePatterns()) {
4619 // @synchronized ( expression ) { statements }
James Dennetta40f7922012-06-14 03:11:41 +00004620 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004621 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4622 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4623 Builder.AddPlaceholderChunk("expression");
4624 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4625 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4626 Builder.AddPlaceholderChunk("statements");
4627 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4628 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004629 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004630}
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004631
Douglas Gregorbca403c2010-01-13 23:51:12 +00004632static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004633 ResultBuilder &Results,
4634 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004635 typedef CodeCompletionResult Result;
James Dennetta40f7922012-06-14 03:11:41 +00004636 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4637 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4638 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004639 if (LangOpts.ObjC2)
James Dennetta40f7922012-06-14 03:11:41 +00004640 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004641}
4642
4643void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004644 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004645 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004646 CodeCompletionContext::CCC_Other);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004647 Results.EnterNewScope();
David Blaikie4e4d0842012-03-11 07:00:24 +00004648 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004649 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004650 HandleCodeCompleteResults(this, CodeCompleter,
4651 CodeCompletionContext::CCC_Other,
4652 Results.data(),Results.size());
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004653}
4654
4655void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004656 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004657 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004658 CodeCompletionContext::CCC_Other);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004659 Results.EnterNewScope();
Douglas Gregorbca403c2010-01-13 23:51:12 +00004660 AddObjCStatementResults(Results, false);
4661 AddObjCExpressionResults(Results, false);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004662 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004663 HandleCodeCompleteResults(this, CodeCompleter,
4664 CodeCompletionContext::CCC_Other,
4665 Results.data(),Results.size());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004666}
4667
4668void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004669 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004670 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004671 CodeCompletionContext::CCC_Other);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004672 Results.EnterNewScope();
Douglas Gregorbca403c2010-01-13 23:51:12 +00004673 AddObjCExpressionResults(Results, false);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004674 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004675 HandleCodeCompleteResults(this, CodeCompleter,
4676 CodeCompletionContext::CCC_Other,
4677 Results.data(),Results.size());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004678}
4679
Douglas Gregor988358f2009-11-19 00:14:45 +00004680/// \brief Determine whether the addition of the given flag to an Objective-C
4681/// property's attributes will cause a conflict.
Bill Wendlingad017fa2012-12-20 19:22:21 +00004682static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregor988358f2009-11-19 00:14:45 +00004683 // Check if we've already added this flag.
Bill Wendlingad017fa2012-12-20 19:22:21 +00004684 if (Attributes & NewFlag)
Douglas Gregor988358f2009-11-19 00:14:45 +00004685 return true;
4686
Bill Wendlingad017fa2012-12-20 19:22:21 +00004687 Attributes |= NewFlag;
Douglas Gregor988358f2009-11-19 00:14:45 +00004688
4689 // Check for collisions with "readonly".
Bill Wendlingad017fa2012-12-20 19:22:21 +00004690 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4691 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregor988358f2009-11-19 00:14:45 +00004692 return true;
4693
Jordan Rosed7403a72012-08-20 20:01:13 +00004694 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendlingad017fa2012-12-20 19:22:21 +00004695 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCallf85e1932011-06-15 23:02:42 +00004696 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregor988358f2009-11-19 00:14:45 +00004697 ObjCDeclSpec::DQ_PR_copy |
Jordan Rosed7403a72012-08-20 20:01:13 +00004698 ObjCDeclSpec::DQ_PR_retain |
4699 ObjCDeclSpec::DQ_PR_strong |
4700 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregor988358f2009-11-19 00:14:45 +00004701 if (AssignCopyRetMask &&
4702 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCallf85e1932011-06-15 23:02:42 +00004703 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregor988358f2009-11-19 00:14:45 +00004704 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCallf85e1932011-06-15 23:02:42 +00004705 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rosed7403a72012-08-20 20:01:13 +00004706 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4707 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregor988358f2009-11-19 00:14:45 +00004708 return true;
4709
4710 return false;
4711}
4712
Douglas Gregora93b1082009-11-18 23:08:07 +00004713void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroffece8e712009-10-08 21:55:05 +00004714 if (!CodeCompleter)
4715 return;
Douglas Gregord3c68542009-11-19 01:08:35 +00004716
Bill Wendlingad017fa2012-12-20 19:22:21 +00004717 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroffece8e712009-10-08 21:55:05 +00004718
Douglas Gregor218937c2011-02-01 19:23:04 +00004719 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004720 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004721 CodeCompletionContext::CCC_Other);
Steve Naroffece8e712009-10-08 21:55:05 +00004722 Results.EnterNewScope();
Bill Wendlingad017fa2012-12-20 19:22:21 +00004723 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall0a2c5e22010-08-25 06:19:51 +00004724 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004725 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall0a2c5e22010-08-25 06:19:51 +00004726 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004727 if (!ObjCPropertyFlagConflicts(Attributes,
John McCallf85e1932011-06-15 23:02:42 +00004728 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4729 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004730 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall0a2c5e22010-08-25 06:19:51 +00004731 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004732 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall0a2c5e22010-08-25 06:19:51 +00004733 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004734 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCallf85e1932011-06-15 23:02:42 +00004735 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004736 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall0a2c5e22010-08-25 06:19:51 +00004737 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004738 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall0a2c5e22010-08-25 06:19:51 +00004739 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004740 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian27f45232011-06-11 17:14:27 +00004741 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rosed7403a72012-08-20 20:01:13 +00004742
4743 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall0a7dd782012-08-21 02:47:43 +00004744 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendlingad017fa2012-12-20 19:22:21 +00004745 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rosed7403a72012-08-20 20:01:13 +00004746 Results.AddResult(CodeCompletionResult("weak"));
4747
Bill Wendlingad017fa2012-12-20 19:22:21 +00004748 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004749 CodeCompletionBuilder Setter(Results.getAllocator(),
4750 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00004751 Setter.AddTypedTextChunk("setter");
4752 Setter.AddTextChunk(" = ");
4753 Setter.AddPlaceholderChunk("method");
4754 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor54f01612009-11-19 00:01:57 +00004755 }
Bill Wendlingad017fa2012-12-20 19:22:21 +00004756 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004757 CodeCompletionBuilder Getter(Results.getAllocator(),
4758 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00004759 Getter.AddTypedTextChunk("getter");
4760 Getter.AddTextChunk(" = ");
4761 Getter.AddPlaceholderChunk("method");
4762 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor54f01612009-11-19 00:01:57 +00004763 }
Steve Naroffece8e712009-10-08 21:55:05 +00004764 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004765 HandleCodeCompleteResults(this, CodeCompleter,
4766 CodeCompletionContext::CCC_Other,
4767 Results.data(),Results.size());
Steve Naroffece8e712009-10-08 21:55:05 +00004768}
Steve Naroffc4df6d22009-11-07 02:08:14 +00004769
James Dennettde23c7e2012-06-17 05:33:25 +00004770/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregor4ad96852009-11-19 07:41:15 +00004771/// via code completion.
4772enum ObjCMethodKind {
Dmitri Gribenko49fdccb2012-06-08 23:13:42 +00004773 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4774 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4775 MK_OneArgSelector ///< One-argument selector.
Douglas Gregor4ad96852009-11-19 07:41:15 +00004776};
4777
Douglas Gregor458433d2010-08-26 15:07:07 +00004778static bool isAcceptableObjCSelector(Selector Sel,
4779 ObjCMethodKind WantKind,
4780 IdentifierInfo **SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004781 unsigned NumSelIdents,
4782 bool AllowSameLength = true) {
Douglas Gregor458433d2010-08-26 15:07:07 +00004783 if (NumSelIdents > Sel.getNumArgs())
4784 return false;
4785
4786 switch (WantKind) {
4787 case MK_Any: break;
4788 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4789 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4790 }
4791
Douglas Gregorcf544262010-11-17 21:36:08 +00004792 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4793 return false;
4794
Douglas Gregor458433d2010-08-26 15:07:07 +00004795 for (unsigned I = 0; I != NumSelIdents; ++I)
4796 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4797 return false;
4798
4799 return true;
4800}
4801
Douglas Gregor4ad96852009-11-19 07:41:15 +00004802static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4803 ObjCMethodKind WantKind,
4804 IdentifierInfo **SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004805 unsigned NumSelIdents,
4806 bool AllowSameLength = true) {
Douglas Gregor458433d2010-08-26 15:07:07 +00004807 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004808 NumSelIdents, AllowSameLength);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004809}
Douglas Gregord36adf52010-09-16 16:06:31 +00004810
4811namespace {
4812 /// \brief A set of selectors, which is used to avoid introducing multiple
4813 /// completions with the same selector into the result set.
4814 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4815}
4816
Douglas Gregor36ecb042009-11-17 23:22:23 +00004817/// \brief Add all of the Objective-C methods in the given Objective-C
4818/// container to the set of results.
4819///
4820/// The container will be a class, protocol, category, or implementation of
4821/// any of the above. This mether will recurse to include methods from
4822/// the superclasses of classes along with their categories, protocols, and
4823/// implementations.
4824///
4825/// \param Container the container in which we'll look to find methods.
4826///
James Dennetta40f7922012-06-14 03:11:41 +00004827/// \param WantInstanceMethods Whether to add instance methods (only); if
4828/// false, this routine will add factory methods (only).
Douglas Gregor36ecb042009-11-17 23:22:23 +00004829///
4830/// \param CurContext the context in which we're performing the lookup that
4831/// finds methods.
4832///
Douglas Gregorcf544262010-11-17 21:36:08 +00004833/// \param AllowSameLength Whether we allow a method to be added to the list
4834/// when it has the same number of parameters as we have selector identifiers.
4835///
Douglas Gregor36ecb042009-11-17 23:22:23 +00004836/// \param Results the structure into which we'll add results.
4837static void AddObjCMethods(ObjCContainerDecl *Container,
4838 bool WantInstanceMethods,
Douglas Gregor4ad96852009-11-19 07:41:15 +00004839 ObjCMethodKind WantKind,
Douglas Gregord3c68542009-11-19 01:08:35 +00004840 IdentifierInfo **SelIdents,
4841 unsigned NumSelIdents,
Douglas Gregor36ecb042009-11-17 23:22:23 +00004842 DeclContext *CurContext,
Douglas Gregord36adf52010-09-16 16:06:31 +00004843 VisitedSelectorSet &Selectors,
Douglas Gregorcf544262010-11-17 21:36:08 +00004844 bool AllowSameLength,
Douglas Gregor408be5a2010-08-25 01:08:01 +00004845 ResultBuilder &Results,
4846 bool InOriginalClass = true) {
John McCall0a2c5e22010-08-25 06:19:51 +00004847 typedef CodeCompletionResult Result;
Douglas Gregorb92a4082012-06-12 13:44:08 +00004848 Container = getContainerDef(Container);
Douglas Gregor5824b802013-01-30 06:58:39 +00004849 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4850 bool isRootClass = IFace && !IFace->getSuperClass();
Douglas Gregor36ecb042009-11-17 23:22:23 +00004851 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
4852 MEnd = Container->meth_end();
4853 M != MEnd; ++M) {
Douglas Gregor5824b802013-01-30 06:58:39 +00004854 // The instance methods on the root class can be messaged via the
4855 // metaclass.
4856 if (M->isInstanceMethod() == WantInstanceMethods ||
4857 (isRootClass && !WantInstanceMethods)) {
Douglas Gregord3c68542009-11-19 01:08:35 +00004858 // Check whether the selector identifiers we've been given are a
4859 // subset of the identifiers for this particular method.
David Blaikie581deb32012-06-06 20:45:41 +00004860 if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004861 AllowSameLength))
Douglas Gregord3c68542009-11-19 01:08:35 +00004862 continue;
Douglas Gregor4ad96852009-11-19 07:41:15 +00004863
David Blaikie262bc182012-04-30 02:36:29 +00004864 if (!Selectors.insert(M->getSelector()))
Douglas Gregord36adf52010-09-16 16:06:31 +00004865 continue;
4866
Douglas Gregord1f09b42013-01-31 04:52:16 +00004867 Result R = Result(*M, Results.getBasePriority(*M), 0);
Douglas Gregord3c68542009-11-19 01:08:35 +00004868 R.StartParameter = NumSelIdents;
Douglas Gregor4ad96852009-11-19 07:41:15 +00004869 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor408be5a2010-08-25 01:08:01 +00004870 if (!InOriginalClass)
4871 R.Priority += CCD_InBaseClass;
Douglas Gregord3c68542009-11-19 01:08:35 +00004872 Results.MaybeAddResult(R, CurContext);
4873 }
Douglas Gregor36ecb042009-11-17 23:22:23 +00004874 }
4875
Douglas Gregore396c7b2010-09-16 15:34:59 +00004876 // Visit the protocols of protocols.
4877 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +00004878 if (Protocol->hasDefinition()) {
4879 const ObjCList<ObjCProtocolDecl> &Protocols
4880 = Protocol->getReferencedProtocols();
4881 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4882 E = Protocols.end();
4883 I != E; ++I)
4884 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
4885 NumSelIdents, CurContext, Selectors, AllowSameLength,
4886 Results, false);
4887 }
Douglas Gregore396c7b2010-09-16 15:34:59 +00004888 }
4889
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00004890 if (!IFace || !IFace->hasDefinition())
Douglas Gregor36ecb042009-11-17 23:22:23 +00004891 return;
4892
4893 // Add methods in protocols.
Argyrios Kyrtzidisa5f44412012-03-13 01:09:41 +00004894 for (ObjCInterfaceDecl::protocol_iterator I = IFace->protocol_begin(),
4895 E = IFace->protocol_end();
Douglas Gregor36ecb042009-11-17 23:22:23 +00004896 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00004897 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004898 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004899
4900 // Add methods in categories.
Douglas Gregord3297242013-01-16 23:00:23 +00004901 for (ObjCInterfaceDecl::known_categories_iterator
4902 Cat = IFace->known_categories_begin(),
4903 CatEnd = IFace->known_categories_end();
4904 Cat != CatEnd; ++Cat) {
4905 ObjCCategoryDecl *CatDecl = *Cat;
4906
4907 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004908 NumSelIdents, CurContext, Selectors, AllowSameLength,
4909 Results, InOriginalClass);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004910
4911 // Add a categories protocol methods.
4912 const ObjCList<ObjCProtocolDecl> &Protocols
4913 = CatDecl->getReferencedProtocols();
4914 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4915 E = Protocols.end();
4916 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00004917 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004918 NumSelIdents, CurContext, Selectors, AllowSameLength,
4919 Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004920
4921 // Add methods in category implementations.
4922 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregor4ad96852009-11-19 07:41:15 +00004923 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004924 NumSelIdents, CurContext, Selectors, AllowSameLength,
4925 Results, InOriginalClass);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004926 }
4927
4928 // Add methods in superclass.
4929 if (IFace->getSuperClass())
Douglas Gregor4ad96852009-11-19 07:41:15 +00004930 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Douglas Gregorcf544262010-11-17 21:36:08 +00004931 SelIdents, NumSelIdents, CurContext, Selectors,
4932 AllowSameLength, Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004933
4934 // Add methods in our implementation, if any.
4935 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregor4ad96852009-11-19 07:41:15 +00004936 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004937 NumSelIdents, CurContext, Selectors, AllowSameLength,
4938 Results, InOriginalClass);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004939}
4940
4941
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004942void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregor4ad96852009-11-19 07:41:15 +00004943 // Try to find the interface where getters might live.
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004944 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004945 if (!Class) {
4946 if (ObjCCategoryDecl *Category
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004947 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregor4ad96852009-11-19 07:41:15 +00004948 Class = Category->getClassInterface();
4949
4950 if (!Class)
4951 return;
4952 }
4953
4954 // Find all of the potential getters.
Douglas Gregor218937c2011-02-01 19:23:04 +00004955 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004956 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004957 CodeCompletionContext::CCC_Other);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004958 Results.EnterNewScope();
4959
Douglas Gregord36adf52010-09-16 16:06:31 +00004960 VisitedSelectorSet Selectors;
4961 AddObjCMethods(Class, true, MK_ZeroArgSelector, 0, 0, CurContext, Selectors,
Douglas Gregorcf544262010-11-17 21:36:08 +00004962 /*AllowSameLength=*/true, Results);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004963 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004964 HandleCodeCompleteResults(this, CodeCompleter,
4965 CodeCompletionContext::CCC_Other,
4966 Results.data(),Results.size());
Douglas Gregor4ad96852009-11-19 07:41:15 +00004967}
4968
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004969void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregor4ad96852009-11-19 07:41:15 +00004970 // Try to find the interface where setters might live.
4971 ObjCInterfaceDecl *Class
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004972 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004973 if (!Class) {
4974 if (ObjCCategoryDecl *Category
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004975 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregor4ad96852009-11-19 07:41:15 +00004976 Class = Category->getClassInterface();
4977
4978 if (!Class)
4979 return;
4980 }
4981
4982 // Find all of the potential getters.
Douglas Gregor218937c2011-02-01 19:23:04 +00004983 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004984 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004985 CodeCompletionContext::CCC_Other);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004986 Results.EnterNewScope();
4987
Douglas Gregord36adf52010-09-16 16:06:31 +00004988 VisitedSelectorSet Selectors;
4989 AddObjCMethods(Class, true, MK_OneArgSelector, 0, 0, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00004990 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004991
4992 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004993 HandleCodeCompleteResults(this, CodeCompleter,
4994 CodeCompletionContext::CCC_Other,
4995 Results.data(),Results.size());
Douglas Gregor36ecb042009-11-17 23:22:23 +00004996}
4997
Douglas Gregorafc45782011-02-15 22:19:42 +00004998void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
4999 bool IsParameter) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005000 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005001 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005002 CodeCompletionContext::CCC_Type);
Douglas Gregord32b0222010-08-24 01:06:58 +00005003 Results.EnterNewScope();
5004
5005 // Add context-sensitive, Objective-C parameter-passing keywords.
5006 bool AddedInOut = false;
5007 if ((DS.getObjCDeclQualifier() &
5008 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
5009 Results.AddResult("in");
5010 Results.AddResult("inout");
5011 AddedInOut = true;
5012 }
5013 if ((DS.getObjCDeclQualifier() &
5014 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
5015 Results.AddResult("out");
5016 if (!AddedInOut)
5017 Results.AddResult("inout");
5018 }
5019 if ((DS.getObjCDeclQualifier() &
5020 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
5021 ObjCDeclSpec::DQ_Oneway)) == 0) {
5022 Results.AddResult("bycopy");
5023 Results.AddResult("byref");
5024 Results.AddResult("oneway");
5025 }
5026
Douglas Gregorafc45782011-02-15 22:19:42 +00005027 // If we're completing the return type of an Objective-C method and the
5028 // identifier IBAction refers to a macro, provide a completion item for
5029 // an action, e.g.,
5030 // IBAction)<#selector#>:(id)sender
5031 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
5032 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005033 CodeCompletionBuilder Builder(Results.getAllocator(),
5034 Results.getCodeCompletionTUInfo(),
5035 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorafc45782011-02-15 22:19:42 +00005036 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00005037 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorafc45782011-02-15 22:19:42 +00005038 Builder.AddPlaceholderChunk("selector");
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00005039 Builder.AddChunk(CodeCompletionString::CK_Colon);
5040 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorafc45782011-02-15 22:19:42 +00005041 Builder.AddTextChunk("id");
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00005042 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorafc45782011-02-15 22:19:42 +00005043 Builder.AddTextChunk("sender");
5044 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
5045 }
Douglas Gregor31aa5772013-01-30 07:11:43 +00005046
5047 // If we're completing the return type, provide 'instancetype'.
5048 if (!IsParameter) {
5049 Results.AddResult(CodeCompletionResult("instancetype"));
5050 }
Douglas Gregorafc45782011-02-15 22:19:42 +00005051
Douglas Gregord32b0222010-08-24 01:06:58 +00005052 // Add various builtin type names and specifiers.
5053 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5054 Results.ExitScope();
5055
5056 // Add the various type names
5057 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5058 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5059 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5060 CodeCompleter->includeGlobals());
5061
5062 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00005063 AddMacroResults(PP, Results, false);
Douglas Gregord32b0222010-08-24 01:06:58 +00005064
5065 HandleCodeCompleteResults(this, CodeCompleter,
5066 CodeCompletionContext::CCC_Type,
5067 Results.data(), Results.size());
5068}
5069
Douglas Gregor22f56992010-04-06 19:22:33 +00005070/// \brief When we have an expression with type "id", we may assume
5071/// that it has some more-specific class type based on knowledge of
5072/// common uses of Objective-C. This routine returns that class type,
5073/// or NULL if no better result could be determined.
5074static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregor78edf512010-09-15 16:23:04 +00005075 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor22f56992010-04-06 19:22:33 +00005076 if (!Msg)
5077 return 0;
5078
5079 Selector Sel = Msg->getSelector();
5080 if (Sel.isNull())
5081 return 0;
5082
5083 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5084 if (!Id)
5085 return 0;
5086
5087 ObjCMethodDecl *Method = Msg->getMethodDecl();
5088 if (!Method)
5089 return 0;
5090
5091 // Determine the class that we're sending the message to.
Douglas Gregor04badcf2010-04-21 00:45:42 +00005092 ObjCInterfaceDecl *IFace = 0;
5093 switch (Msg->getReceiverKind()) {
5094 case ObjCMessageExpr::Class:
John McCallc12c5bb2010-05-15 11:32:37 +00005095 if (const ObjCObjectType *ObjType
5096 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5097 IFace = ObjType->getInterface();
Douglas Gregor04badcf2010-04-21 00:45:42 +00005098 break;
5099
5100 case ObjCMessageExpr::Instance: {
5101 QualType T = Msg->getInstanceReceiver()->getType();
5102 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5103 IFace = Ptr->getInterfaceDecl();
5104 break;
5105 }
5106
5107 case ObjCMessageExpr::SuperInstance:
5108 case ObjCMessageExpr::SuperClass:
5109 break;
Douglas Gregor22f56992010-04-06 19:22:33 +00005110 }
5111
5112 if (!IFace)
5113 return 0;
5114
5115 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5116 if (Method->isInstanceMethod())
5117 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5118 .Case("retain", IFace)
John McCallf85e1932011-06-15 23:02:42 +00005119 .Case("strong", IFace)
Douglas Gregor22f56992010-04-06 19:22:33 +00005120 .Case("autorelease", IFace)
5121 .Case("copy", IFace)
5122 .Case("copyWithZone", IFace)
5123 .Case("mutableCopy", IFace)
5124 .Case("mutableCopyWithZone", IFace)
5125 .Case("awakeFromCoder", IFace)
5126 .Case("replacementObjectFromCoder", IFace)
5127 .Case("class", IFace)
5128 .Case("classForCoder", IFace)
5129 .Case("superclass", Super)
5130 .Default(0);
5131
5132 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5133 .Case("new", IFace)
5134 .Case("alloc", IFace)
5135 .Case("allocWithZone", IFace)
5136 .Case("class", IFace)
5137 .Case("superclass", Super)
5138 .Default(0);
5139}
5140
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005141// Add a special completion for a message send to "super", which fills in the
5142// most likely case of forwarding all of our arguments to the superclass
5143// function.
5144///
5145/// \param S The semantic analysis object.
5146///
Dmitri Gribenko70517ca2012-08-23 17:58:28 +00005147/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005148/// the "super" keyword. Otherwise, we just need to provide the arguments.
5149///
5150/// \param SelIdents The identifiers in the selector that have already been
5151/// provided as arguments for a send to "super".
5152///
5153/// \param NumSelIdents The number of identifiers in \p SelIdents.
5154///
5155/// \param Results The set of results to augment.
5156///
5157/// \returns the Objective-C method declaration that would be invoked by
5158/// this "super" completion. If NULL, no completion was added.
5159static ObjCMethodDecl *AddSuperSendCompletion(Sema &S, bool NeedSuperKeyword,
5160 IdentifierInfo **SelIdents,
5161 unsigned NumSelIdents,
5162 ResultBuilder &Results) {
5163 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5164 if (!CurMethod)
5165 return 0;
5166
5167 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5168 if (!Class)
5169 return 0;
5170
5171 // Try to find a superclass method with the same selector.
5172 ObjCMethodDecl *SuperMethod = 0;
Douglas Gregor78bcd912011-02-16 00:51:18 +00005173 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5174 // Check in the class
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005175 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5176 CurMethod->isInstanceMethod());
5177
Douglas Gregor78bcd912011-02-16 00:51:18 +00005178 // Check in categories or class extensions.
5179 if (!SuperMethod) {
Douglas Gregord3297242013-01-16 23:00:23 +00005180 for (ObjCInterfaceDecl::known_categories_iterator
5181 Cat = Class->known_categories_begin(),
5182 CatEnd = Class->known_categories_end();
5183 Cat != CatEnd; ++Cat) {
5184 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregor78bcd912011-02-16 00:51:18 +00005185 CurMethod->isInstanceMethod())))
5186 break;
Douglas Gregord3297242013-01-16 23:00:23 +00005187 }
Douglas Gregor78bcd912011-02-16 00:51:18 +00005188 }
5189 }
5190
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005191 if (!SuperMethod)
5192 return 0;
5193
5194 // Check whether the superclass method has the same signature.
5195 if (CurMethod->param_size() != SuperMethod->param_size() ||
5196 CurMethod->isVariadic() != SuperMethod->isVariadic())
5197 return 0;
5198
5199 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5200 CurPEnd = CurMethod->param_end(),
5201 SuperP = SuperMethod->param_begin();
5202 CurP != CurPEnd; ++CurP, ++SuperP) {
5203 // Make sure the parameter types are compatible.
5204 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5205 (*SuperP)->getType()))
5206 return 0;
5207
5208 // Make sure we have a parameter name to forward!
5209 if (!(*CurP)->getIdentifier())
5210 return 0;
5211 }
5212
5213 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005214 CodeCompletionBuilder Builder(Results.getAllocator(),
5215 Results.getCodeCompletionTUInfo());
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005216
5217 // Give this completion a return type.
Douglas Gregor8987b232011-09-27 23:30:47 +00005218 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5219 Builder);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005220
5221 // If we need the "super" keyword, add it (plus some spacing).
5222 if (NeedSuperKeyword) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005223 Builder.AddTypedTextChunk("super");
5224 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005225 }
5226
5227 Selector Sel = CurMethod->getSelector();
5228 if (Sel.isUnarySelector()) {
5229 if (NeedSuperKeyword)
Douglas Gregordae68752011-02-01 22:57:45 +00005230 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005231 Sel.getNameForSlot(0)));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005232 else
Douglas Gregordae68752011-02-01 22:57:45 +00005233 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005234 Sel.getNameForSlot(0)));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005235 } else {
5236 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5237 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
5238 if (I > NumSelIdents)
Douglas Gregor218937c2011-02-01 19:23:04 +00005239 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005240
5241 if (I < NumSelIdents)
Douglas Gregor218937c2011-02-01 19:23:04 +00005242 Builder.AddInformativeChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00005243 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005244 Sel.getNameForSlot(I) + ":"));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005245 else if (NeedSuperKeyword || I > NumSelIdents) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005246 Builder.AddTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00005247 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005248 Sel.getNameForSlot(I) + ":"));
Douglas Gregordae68752011-02-01 22:57:45 +00005249 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005250 (*CurP)->getIdentifier()->getName()));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005251 } else {
Douglas Gregor218937c2011-02-01 19:23:04 +00005252 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00005253 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005254 Sel.getNameForSlot(I) + ":"));
Douglas Gregordae68752011-02-01 22:57:45 +00005255 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005256 (*CurP)->getIdentifier()->getName()));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005257 }
5258 }
5259 }
5260
Douglas Gregorba103062012-03-27 23:34:16 +00005261 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5262 CCP_SuperCompletion));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005263 return SuperMethod;
5264}
5265
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005266void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall0a2c5e22010-08-25 06:19:51 +00005267 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00005268 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005269 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005270 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith80ad52f2013-01-02 11:42:31 +00005271 getLangOpts().CPlusPlus11
Douglas Gregor81f3bff2012-02-15 15:34:24 +00005272 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5273 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005274
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005275 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5276 Results.EnterNewScope();
Douglas Gregor8071e422010-08-15 06:18:01 +00005277 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5278 CodeCompleter->includeGlobals());
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005279
5280 // If we are in an Objective-C method inside a class that has a superclass,
5281 // add "super" as an option.
5282 if (ObjCMethodDecl *Method = getCurMethodDecl())
5283 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005284 if (Iface->getSuperClass()) {
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005285 Results.AddResult(Result("super"));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005286
5287 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, 0, 0, Results);
5288 }
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005289
Richard Smith80ad52f2013-01-02 11:42:31 +00005290 if (getLangOpts().CPlusPlus11)
Douglas Gregor81f3bff2012-02-15 15:34:24 +00005291 addThisCompletion(*this, Results);
5292
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005293 Results.ExitScope();
5294
5295 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00005296 AddMacroResults(PP, Results, false);
Douglas Gregorcee9ff12010-09-20 22:39:41 +00005297 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005298 Results.data(), Results.size());
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005299
5300}
5301
Douglas Gregor2725ca82010-04-21 19:57:20 +00005302void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
5303 IdentifierInfo **SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005304 unsigned NumSelIdents,
5305 bool AtArgumentExpression) {
Douglas Gregor2725ca82010-04-21 19:57:20 +00005306 ObjCInterfaceDecl *CDecl = 0;
5307 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5308 // Figure out which interface we're in.
5309 CDecl = CurMethod->getClassInterface();
5310 if (!CDecl)
5311 return;
5312
5313 // Find the superclass of this class.
5314 CDecl = CDecl->getSuperClass();
5315 if (!CDecl)
5316 return;
5317
5318 if (CurMethod->isInstanceMethod()) {
5319 // We are inside an instance method, which means that the message
5320 // send [super ...] is actually calling an instance method on the
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005321 // current object.
5322 return CodeCompleteObjCInstanceMessage(S, 0,
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005323 SelIdents, NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005324 AtArgumentExpression,
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005325 CDecl);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005326 }
5327
5328 // Fall through to send to the superclass in CDecl.
5329 } else {
5330 // "super" may be the name of a type or variable. Figure out which
5331 // it is.
Argyrios Kyrtzidis57f8da52013-03-14 22:56:43 +00005332 IdentifierInfo *Super = getSuperIdentifier();
Douglas Gregor2725ca82010-04-21 19:57:20 +00005333 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5334 LookupOrdinaryName);
5335 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5336 // "super" names an interface. Use it.
5337 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCallc12c5bb2010-05-15 11:32:37 +00005338 if (const ObjCObjectType *Iface
5339 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5340 CDecl = Iface->getInterface();
Douglas Gregor2725ca82010-04-21 19:57:20 +00005341 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5342 // "super" names an unresolved type; we can't be more specific.
5343 } else {
5344 // Assume that "super" names some kind of value and parse that way.
5345 CXXScopeSpec SS;
Abramo Bagnarae4b92762012-01-27 09:46:47 +00005346 SourceLocation TemplateKWLoc;
Douglas Gregor2725ca82010-04-21 19:57:20 +00005347 UnqualifiedId id;
5348 id.setIdentifier(Super, SuperLoc);
Abramo Bagnarae4b92762012-01-27 09:46:47 +00005349 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5350 false, false);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005351 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005352 SelIdents, NumSelIdents,
5353 AtArgumentExpression);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005354 }
5355
5356 // Fall through
5357 }
5358
John McCallb3d87482010-08-24 05:47:05 +00005359 ParsedType Receiver;
Douglas Gregor2725ca82010-04-21 19:57:20 +00005360 if (CDecl)
John McCallb3d87482010-08-24 05:47:05 +00005361 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor2725ca82010-04-21 19:57:20 +00005362 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005363 NumSelIdents, AtArgumentExpression,
5364 /*IsSuper=*/true);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005365}
5366
Douglas Gregorb9d77572010-09-21 00:03:25 +00005367/// \brief Given a set of code-completion results for the argument of a message
5368/// send, determine the preferred type (if any) for that argument expression.
5369static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5370 unsigned NumSelIdents) {
5371 typedef CodeCompletionResult Result;
5372 ASTContext &Context = Results.getSema().Context;
5373
5374 QualType PreferredType;
5375 unsigned BestPriority = CCP_Unlikely * 2;
5376 Result *ResultsData = Results.data();
5377 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5378 Result &R = ResultsData[I];
5379 if (R.Kind == Result::RK_Declaration &&
5380 isa<ObjCMethodDecl>(R.Declaration)) {
5381 if (R.Priority <= BestPriority) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00005382 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005383 if (NumSelIdents <= Method->param_size()) {
5384 QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1]
5385 ->getType();
5386 if (R.Priority < BestPriority || PreferredType.isNull()) {
5387 BestPriority = R.Priority;
5388 PreferredType = MyPreferredType;
5389 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5390 MyPreferredType)) {
5391 PreferredType = QualType();
5392 }
5393 }
5394 }
5395 }
5396 }
5397
5398 return PreferredType;
5399}
5400
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005401static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5402 ParsedType Receiver,
5403 IdentifierInfo **SelIdents,
5404 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005405 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005406 bool IsSuper,
5407 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005408 typedef CodeCompletionResult Result;
Douglas Gregor24a069f2009-11-17 17:59:40 +00005409 ObjCInterfaceDecl *CDecl = 0;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005410
Douglas Gregor24a069f2009-11-17 17:59:40 +00005411 // If the given name refers to an interface type, retrieve the
5412 // corresponding declaration.
Douglas Gregor2725ca82010-04-21 19:57:20 +00005413 if (Receiver) {
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005414 QualType T = SemaRef.GetTypeFromParser(Receiver, 0);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005415 if (!T.isNull())
John McCallc12c5bb2010-05-15 11:32:37 +00005416 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5417 CDecl = Interface->getInterface();
Douglas Gregor24a069f2009-11-17 17:59:40 +00005418 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005419
Douglas Gregor36ecb042009-11-17 23:22:23 +00005420 // Add all of the factory methods in this Objective-C class, its protocols,
5421 // superclasses, categories, implementation, etc.
Steve Naroffc4df6d22009-11-07 02:08:14 +00005422 Results.EnterNewScope();
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005423
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005424 // If this is a send-to-super, try to add the special "super" send
5425 // completion.
5426 if (IsSuper) {
5427 if (ObjCMethodDecl *SuperMethod
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005428 = AddSuperSendCompletion(SemaRef, false, SelIdents, NumSelIdents,
5429 Results))
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005430 Results.Ignore(SuperMethod);
5431 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005432
Douglas Gregor265f7492010-08-27 15:29:55 +00005433 // If we're inside an Objective-C method definition, prefer its selector to
5434 // others.
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005435 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregor265f7492010-08-27 15:29:55 +00005436 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005437
Douglas Gregord36adf52010-09-16 16:06:31 +00005438 VisitedSelectorSet Selectors;
Douglas Gregor13438f92010-04-06 16:40:00 +00005439 if (CDecl)
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005440 AddObjCMethods(CDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005441 SemaRef.CurContext, Selectors, AtArgumentExpression,
5442 Results);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005443 else {
Douglas Gregor13438f92010-04-06 16:40:00 +00005444 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005445
Douglas Gregor719770d2010-04-06 17:30:22 +00005446 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00005447 // pool from the AST file.
Axel Naumann0ec56b72012-10-18 19:05:02 +00005448 if (SemaRef.getExternalSource()) {
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005449 for (uint32_t I = 0,
Axel Naumann0ec56b72012-10-18 19:05:02 +00005450 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall76bd1f32010-06-01 09:23:16 +00005451 I != N; ++I) {
Axel Naumann0ec56b72012-10-18 19:05:02 +00005452 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005453 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregor719770d2010-04-06 17:30:22 +00005454 continue;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005455
5456 SemaRef.ReadMethodPool(Sel);
Douglas Gregor719770d2010-04-06 17:30:22 +00005457 }
5458 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005459
5460 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5461 MEnd = SemaRef.MethodPool.end();
Sebastian Redldb9d2142010-08-02 23:18:59 +00005462 M != MEnd; ++M) {
5463 for (ObjCMethodList *MethList = &M->second.second;
5464 MethList && MethList->Method;
Douglas Gregor13438f92010-04-06 16:40:00 +00005465 MethList = MethList->Next) {
5466 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5467 NumSelIdents))
5468 continue;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005469
Douglas Gregord1f09b42013-01-31 04:52:16 +00005470 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Douglas Gregor13438f92010-04-06 16:40:00 +00005471 R.StartParameter = NumSelIdents;
5472 R.AllParametersAreInformative = false;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005473 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor13438f92010-04-06 16:40:00 +00005474 }
5475 }
5476 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005477
5478 Results.ExitScope();
5479}
Douglas Gregor13438f92010-04-06 16:40:00 +00005480
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005481void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
5482 IdentifierInfo **SelIdents,
5483 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005484 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005485 bool IsSuper) {
Douglas Gregore081a612011-07-21 01:05:26 +00005486
5487 QualType T = this->GetTypeFromParser(Receiver);
5488
Douglas Gregor218937c2011-02-01 19:23:04 +00005489 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005490 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregore081a612011-07-21 01:05:26 +00005491 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Douglas Gregor0a47d692011-07-26 15:24:30 +00005492 T, SelIdents, NumSelIdents));
Douglas Gregore081a612011-07-21 01:05:26 +00005493
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005494 AddClassMessageCompletions(*this, S, Receiver, SelIdents, NumSelIdents,
5495 AtArgumentExpression, IsSuper, Results);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005496
5497 // If we're actually at the argument expression (rather than prior to the
5498 // selector), we're actually performing code completion for an expression.
5499 // Determine whether we have a single, best method. If so, we can
5500 // code-complete the expression using the corresponding parameter type as
5501 // our preferred type, improving completion results.
5502 if (AtArgumentExpression) {
5503 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Douglas Gregore081a612011-07-21 01:05:26 +00005504 NumSelIdents);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005505 if (PreferredType.isNull())
5506 CodeCompleteOrdinaryName(S, PCC_Expression);
5507 else
5508 CodeCompleteExpression(S, PreferredType);
5509 return;
5510 }
5511
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005512 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregore081a612011-07-21 01:05:26 +00005513 Results.getCompletionContext(),
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005514 Results.data(), Results.size());
Steve Naroffc4df6d22009-11-07 02:08:14 +00005515}
5516
Richard Trieuf81e5a92011-09-09 02:00:50 +00005517void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Douglas Gregord3c68542009-11-19 01:08:35 +00005518 IdentifierInfo **SelIdents,
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005519 unsigned NumSelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005520 bool AtArgumentExpression,
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005521 ObjCInterfaceDecl *Super) {
John McCall0a2c5e22010-08-25 06:19:51 +00005522 typedef CodeCompletionResult Result;
Steve Naroffc4df6d22009-11-07 02:08:14 +00005523
5524 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffc4df6d22009-11-07 02:08:14 +00005525
Douglas Gregor36ecb042009-11-17 23:22:23 +00005526 // If necessary, apply function/array conversion to the receiver.
5527 // C99 6.7.5.3p[7,8].
John Wiegley429bb272011-04-08 18:41:53 +00005528 if (RecExpr) {
5529 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5530 if (Conv.isInvalid()) // conversion failed. bail.
5531 return;
5532 RecExpr = Conv.take();
5533 }
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005534 QualType ReceiverType = RecExpr? RecExpr->getType()
5535 : Super? Context.getObjCObjectPointerType(
5536 Context.getObjCInterfaceType(Super))
5537 : Context.getObjCIdType();
Steve Naroffc4df6d22009-11-07 02:08:14 +00005538
Douglas Gregorda892642010-11-08 21:12:30 +00005539 // If we're messaging an expression with type "id" or "Class", check
5540 // whether we know something special about the receiver that allows
5541 // us to assume a more-specific receiver type.
5542 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType())
5543 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5544 if (ReceiverType->isObjCClassType())
5545 return CodeCompleteObjCClassMessage(S,
5546 ParsedType::make(Context.getObjCInterfaceType(IFace)),
5547 SelIdents, NumSelIdents,
5548 AtArgumentExpression, Super);
5549
5550 ReceiverType = Context.getObjCObjectPointerType(
5551 Context.getObjCInterfaceType(IFace));
5552 }
5553
Douglas Gregor36ecb042009-11-17 23:22:23 +00005554 // Build the set of methods we can see.
Douglas Gregor218937c2011-02-01 19:23:04 +00005555 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005556 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregore081a612011-07-21 01:05:26 +00005557 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Douglas Gregor0a47d692011-07-26 15:24:30 +00005558 ReceiverType, SelIdents, NumSelIdents));
Douglas Gregore081a612011-07-21 01:05:26 +00005559
Douglas Gregor36ecb042009-11-17 23:22:23 +00005560 Results.EnterNewScope();
Douglas Gregor22f56992010-04-06 19:22:33 +00005561
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005562 // If this is a send-to-super, try to add the special "super" send
5563 // completion.
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005564 if (Super) {
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005565 if (ObjCMethodDecl *SuperMethod
5566 = AddSuperSendCompletion(*this, false, SelIdents, NumSelIdents,
5567 Results))
5568 Results.Ignore(SuperMethod);
5569 }
5570
Douglas Gregor265f7492010-08-27 15:29:55 +00005571 // If we're inside an Objective-C method definition, prefer its selector to
5572 // others.
5573 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5574 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregor36ecb042009-11-17 23:22:23 +00005575
Douglas Gregord36adf52010-09-16 16:06:31 +00005576 // Keep track of the selectors we've already added.
5577 VisitedSelectorSet Selectors;
5578
Douglas Gregorf74a4192009-11-18 00:06:18 +00005579 // Handle messages to Class. This really isn't a message to an instance
5580 // method, so we treat it the same way we would treat a message send to a
5581 // class method.
5582 if (ReceiverType->isObjCClassType() ||
5583 ReceiverType->isObjCQualifiedClassType()) {
5584 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5585 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Douglas Gregor4ad96852009-11-19 07:41:15 +00005586 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents, NumSelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005587 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005588 }
5589 }
5590 // Handle messages to a qualified ID ("id<foo>").
5591 else if (const ObjCObjectPointerType *QualID
5592 = ReceiverType->getAsObjCQualifiedIdType()) {
5593 // Search protocols for instance methods.
5594 for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(),
5595 E = QualID->qual_end();
5596 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00005597 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00005598 Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005599 }
5600 // Handle messages to a pointer to interface type.
5601 else if (const ObjCObjectPointerType *IFacePtr
5602 = ReceiverType->getAsObjCInterfacePointerType()) {
5603 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregor4ad96852009-11-19 07:41:15 +00005604 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005605 NumSelIdents, CurContext, Selectors, AtArgumentExpression,
5606 Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005607
5608 // Search protocols for instance methods.
5609 for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(),
5610 E = IFacePtr->qual_end();
5611 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00005612 AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00005613 Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005614 }
Douglas Gregor13438f92010-04-06 16:40:00 +00005615 // Handle messages to "id".
5616 else if (ReceiverType->isObjCIdType()) {
Douglas Gregor719770d2010-04-06 17:30:22 +00005617 // We're messaging "id", so provide all instance methods we know
5618 // about as code-completion results.
5619
5620 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00005621 // pool from the AST file.
Douglas Gregor719770d2010-04-06 17:30:22 +00005622 if (ExternalSource) {
John McCall76bd1f32010-06-01 09:23:16 +00005623 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5624 I != N; ++I) {
5625 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redldb9d2142010-08-02 23:18:59 +00005626 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor719770d2010-04-06 17:30:22 +00005627 continue;
5628
Sebastian Redldb9d2142010-08-02 23:18:59 +00005629 ReadMethodPool(Sel);
Douglas Gregor719770d2010-04-06 17:30:22 +00005630 }
5631 }
5632
Sebastian Redldb9d2142010-08-02 23:18:59 +00005633 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5634 MEnd = MethodPool.end();
5635 M != MEnd; ++M) {
5636 for (ObjCMethodList *MethList = &M->second.first;
5637 MethList && MethList->Method;
Douglas Gregor13438f92010-04-06 16:40:00 +00005638 MethList = MethList->Next) {
5639 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
5640 NumSelIdents))
5641 continue;
Douglas Gregord36adf52010-09-16 16:06:31 +00005642
5643 if (!Selectors.insert(MethList->Method->getSelector()))
5644 continue;
5645
Douglas Gregord1f09b42013-01-31 04:52:16 +00005646 Result R(MethList->Method, Results.getBasePriority(MethList->Method),0);
Douglas Gregor13438f92010-04-06 16:40:00 +00005647 R.StartParameter = NumSelIdents;
5648 R.AllParametersAreInformative = false;
5649 Results.MaybeAddResult(R, CurContext);
5650 }
5651 }
5652 }
Steve Naroffc4df6d22009-11-07 02:08:14 +00005653 Results.ExitScope();
Douglas Gregorb9d77572010-09-21 00:03:25 +00005654
5655
5656 // If we're actually at the argument expression (rather than prior to the
5657 // selector), we're actually performing code completion for an expression.
5658 // Determine whether we have a single, best method. If so, we can
5659 // code-complete the expression using the corresponding parameter type as
5660 // our preferred type, improving completion results.
5661 if (AtArgumentExpression) {
5662 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
5663 NumSelIdents);
5664 if (PreferredType.isNull())
5665 CodeCompleteOrdinaryName(S, PCC_Expression);
5666 else
5667 CodeCompleteExpression(S, PreferredType);
5668 return;
5669 }
5670
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005671 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregore081a612011-07-21 01:05:26 +00005672 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005673 Results.data(),Results.size());
Steve Naroffc4df6d22009-11-07 02:08:14 +00005674}
Douglas Gregor55385fe2009-11-18 04:19:12 +00005675
Douglas Gregorfb629412010-08-23 21:17:50 +00005676void Sema::CodeCompleteObjCForCollection(Scope *S,
5677 DeclGroupPtrTy IterationVar) {
5678 CodeCompleteExpressionData Data;
5679 Data.ObjCCollection = true;
5680
5681 if (IterationVar.getAsOpaquePtr()) {
5682 DeclGroupRef DG = IterationVar.getAsVal<DeclGroupRef>();
5683 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5684 if (*I)
5685 Data.IgnoreDecls.push_back(*I);
5686 }
5687 }
5688
5689 CodeCompleteExpression(S, Data);
5690}
5691
Douglas Gregor458433d2010-08-26 15:07:07 +00005692void Sema::CodeCompleteObjCSelector(Scope *S, IdentifierInfo **SelIdents,
5693 unsigned NumSelIdents) {
5694 // If we have an external source, load the entire class method
5695 // pool from the AST file.
5696 if (ExternalSource) {
5697 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5698 I != N; ++I) {
5699 Selector Sel = ExternalSource->GetExternalSelector(I);
5700 if (Sel.isNull() || MethodPool.count(Sel))
5701 continue;
5702
5703 ReadMethodPool(Sel);
5704 }
5705 }
5706
Douglas Gregor218937c2011-02-01 19:23:04 +00005707 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005708 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005709 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor458433d2010-08-26 15:07:07 +00005710 Results.EnterNewScope();
5711 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5712 MEnd = MethodPool.end();
5713 M != MEnd; ++M) {
5714
5715 Selector Sel = M->first;
5716 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents, NumSelIdents))
5717 continue;
5718
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005719 CodeCompletionBuilder Builder(Results.getAllocator(),
5720 Results.getCodeCompletionTUInfo());
Douglas Gregor458433d2010-08-26 15:07:07 +00005721 if (Sel.isUnarySelector()) {
Douglas Gregordae68752011-02-01 22:57:45 +00005722 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005723 Sel.getNameForSlot(0)));
Douglas Gregor218937c2011-02-01 19:23:04 +00005724 Results.AddResult(Builder.TakeString());
Douglas Gregor458433d2010-08-26 15:07:07 +00005725 continue;
5726 }
5727
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005728 std::string Accumulator;
Douglas Gregor458433d2010-08-26 15:07:07 +00005729 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005730 if (I == NumSelIdents) {
5731 if (!Accumulator.empty()) {
Douglas Gregordae68752011-02-01 22:57:45 +00005732 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005733 Accumulator));
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005734 Accumulator.clear();
5735 }
5736 }
5737
Benjamin Kramera0651c52011-07-26 16:59:25 +00005738 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005739 Accumulator += ':';
Douglas Gregor458433d2010-08-26 15:07:07 +00005740 }
Douglas Gregordae68752011-02-01 22:57:45 +00005741 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregor218937c2011-02-01 19:23:04 +00005742 Results.AddResult(Builder.TakeString());
Douglas Gregor458433d2010-08-26 15:07:07 +00005743 }
5744 Results.ExitScope();
5745
5746 HandleCodeCompleteResults(this, CodeCompleter,
5747 CodeCompletionContext::CCC_SelectorName,
5748 Results.data(), Results.size());
5749}
5750
Douglas Gregor55385fe2009-11-18 04:19:12 +00005751/// \brief Add all of the protocol declarations that we find in the given
5752/// (translation unit) context.
5753static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor083128f2009-11-18 04:49:41 +00005754 bool OnlyForwardDeclarations,
Douglas Gregor55385fe2009-11-18 04:19:12 +00005755 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005756 typedef CodeCompletionResult Result;
Douglas Gregor55385fe2009-11-18 04:19:12 +00005757
5758 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5759 DEnd = Ctx->decls_end();
5760 D != DEnd; ++D) {
5761 // Record any protocols we find.
5762 if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*D))
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +00005763 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Douglas Gregord1f09b42013-01-31 04:52:16 +00005764 Results.AddResult(Result(Proto, Results.getBasePriority(Proto), 0),
5765 CurContext, 0, false);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005766 }
5767}
5768
5769void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5770 unsigned NumProtocols) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005771 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005772 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005773 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005774
Douglas Gregor70c23352010-12-09 21:44:02 +00005775 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5776 Results.EnterNewScope();
5777
5778 // Tell the result set to ignore all of the protocols we have
5779 // already seen.
5780 // FIXME: This doesn't work when caching code-completion results.
5781 for (unsigned I = 0; I != NumProtocols; ++I)
5782 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5783 Protocols[I].second))
5784 Results.Ignore(Protocol);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005785
Douglas Gregor70c23352010-12-09 21:44:02 +00005786 // Add all protocols.
5787 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5788 Results);
Douglas Gregor083128f2009-11-18 04:49:41 +00005789
Douglas Gregor70c23352010-12-09 21:44:02 +00005790 Results.ExitScope();
5791 }
5792
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005793 HandleCodeCompleteResults(this, CodeCompleter,
5794 CodeCompletionContext::CCC_ObjCProtocolName,
5795 Results.data(),Results.size());
Douglas Gregor083128f2009-11-18 04:49:41 +00005796}
5797
5798void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005799 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005800 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005801 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor083128f2009-11-18 04:49:41 +00005802
Douglas Gregor70c23352010-12-09 21:44:02 +00005803 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5804 Results.EnterNewScope();
5805
5806 // Add all protocols.
5807 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5808 Results);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005809
Douglas Gregor70c23352010-12-09 21:44:02 +00005810 Results.ExitScope();
5811 }
5812
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005813 HandleCodeCompleteResults(this, CodeCompleter,
5814 CodeCompletionContext::CCC_ObjCProtocolName,
5815 Results.data(),Results.size());
Douglas Gregor55385fe2009-11-18 04:19:12 +00005816}
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005817
5818/// \brief Add all of the Objective-C interface declarations that we find in
5819/// the given (translation unit) context.
5820static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5821 bool OnlyForwardDeclarations,
5822 bool OnlyUnimplemented,
5823 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005824 typedef CodeCompletionResult Result;
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005825
5826 for (DeclContext::decl_iterator D = Ctx->decls_begin(),
5827 DEnd = Ctx->decls_end();
5828 D != DEnd; ++D) {
Douglas Gregordeacbdc2010-08-11 12:19:30 +00005829 // Record any interfaces we find.
5830 if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*D))
Douglas Gregor7723fec2011-12-15 20:29:51 +00005831 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregordeacbdc2010-08-11 12:19:30 +00005832 (!OnlyUnimplemented || !Class->getImplementation()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00005833 Results.AddResult(Result(Class, Results.getBasePriority(Class), 0),
5834 CurContext, 0, false);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005835 }
5836}
5837
5838void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005839 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005840 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005841 CodeCompletionContext::CCC_Other);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005842 Results.EnterNewScope();
5843
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005844 if (CodeCompleter->includeGlobals()) {
5845 // Add all classes.
5846 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5847 false, Results);
5848 }
5849
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005850 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005851
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005852 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005853 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005854 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005855}
5856
Douglas Gregorc83c6872010-04-15 22:33:43 +00005857void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5858 SourceLocation ClassNameLoc) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005859 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005860 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005861 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005862 Results.EnterNewScope();
5863
5864 // Make sure that we ignore the class we're currently defining.
5865 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005866 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005867 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005868 Results.Ignore(CurClass);
5869
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005870 if (CodeCompleter->includeGlobals()) {
5871 // Add all classes.
5872 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5873 false, Results);
5874 }
5875
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005876 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005877
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005878 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005879 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005880 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005881}
5882
5883void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005884 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005885 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005886 CodeCompletionContext::CCC_Other);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005887 Results.EnterNewScope();
5888
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005889 if (CodeCompleter->includeGlobals()) {
5890 // Add all unimplemented classes.
5891 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5892 true, Results);
5893 }
5894
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005895 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005896
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005897 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005898 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005899 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005900}
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005901
5902void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorc83c6872010-04-15 22:33:43 +00005903 IdentifierInfo *ClassName,
5904 SourceLocation ClassNameLoc) {
John McCall0a2c5e22010-08-25 06:19:51 +00005905 typedef CodeCompletionResult Result;
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005906
Douglas Gregor218937c2011-02-01 19:23:04 +00005907 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005908 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00005909 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005910
5911 // Ignore any categories we find that have already been implemented by this
5912 // interface.
5913 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5914 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005915 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregord3297242013-01-16 23:00:23 +00005916 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
5917 for (ObjCInterfaceDecl::visible_categories_iterator
5918 Cat = Class->visible_categories_begin(),
5919 CatEnd = Class->visible_categories_end();
5920 Cat != CatEnd; ++Cat) {
5921 CategoryNames.insert(Cat->getIdentifier());
5922 }
5923 }
5924
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005925 // Add all of the categories we know about.
5926 Results.EnterNewScope();
5927 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
5928 for (DeclContext::decl_iterator D = TU->decls_begin(),
5929 DEnd = TU->decls_end();
5930 D != DEnd; ++D)
5931 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(*D))
5932 if (CategoryNames.insert(Category->getIdentifier()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00005933 Results.AddResult(Result(Category, Results.getBasePriority(Category),0),
5934 CurContext, 0, false);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005935 Results.ExitScope();
5936
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005937 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00005938 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005939 Results.data(),Results.size());
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005940}
5941
5942void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorc83c6872010-04-15 22:33:43 +00005943 IdentifierInfo *ClassName,
5944 SourceLocation ClassNameLoc) {
John McCall0a2c5e22010-08-25 06:19:51 +00005945 typedef CodeCompletionResult Result;
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005946
5947 // Find the corresponding interface. If we couldn't find the interface, the
5948 // program itself is ill-formed. However, we'll try to be helpful still by
5949 // providing the list of all of the categories we know about.
5950 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005951 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005952 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
5953 if (!Class)
Douglas Gregorc83c6872010-04-15 22:33:43 +00005954 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005955
Douglas Gregor218937c2011-02-01 19:23:04 +00005956 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005957 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00005958 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005959
5960 // Add all of the categories that have have corresponding interface
5961 // declarations in this class and any of its superclasses, except for
5962 // already-implemented categories in the class itself.
5963 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5964 Results.EnterNewScope();
5965 bool IgnoreImplemented = true;
5966 while (Class) {
Douglas Gregord3297242013-01-16 23:00:23 +00005967 for (ObjCInterfaceDecl::visible_categories_iterator
5968 Cat = Class->visible_categories_begin(),
5969 CatEnd = Class->visible_categories_end();
5970 Cat != CatEnd; ++Cat) {
5971 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
5972 CategoryNames.insert(Cat->getIdentifier()))
Douglas Gregord1f09b42013-01-31 04:52:16 +00005973 Results.AddResult(Result(*Cat, Results.getBasePriority(*Cat), 0),
5974 CurContext, 0, false);
Douglas Gregord3297242013-01-16 23:00:23 +00005975 }
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005976
5977 Class = Class->getSuperClass();
5978 IgnoreImplemented = false;
5979 }
5980 Results.ExitScope();
5981
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005982 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00005983 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005984 Results.data(),Results.size());
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005985}
Douglas Gregor322328b2009-11-18 22:32:06 +00005986
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005987void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005988 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005989 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005990 CodeCompletionContext::CCC_Other);
Douglas Gregor322328b2009-11-18 22:32:06 +00005991
5992 // Figure out where this @synthesize lives.
5993 ObjCContainerDecl *Container
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005994 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor322328b2009-11-18 22:32:06 +00005995 if (!Container ||
5996 (!isa<ObjCImplementationDecl>(Container) &&
5997 !isa<ObjCCategoryImplDecl>(Container)))
5998 return;
5999
6000 // Ignore any properties that have already been implemented.
Douglas Gregorb92a4082012-06-12 13:44:08 +00006001 Container = getContainerDef(Container);
6002 for (DeclContext::decl_iterator D = Container->decls_begin(),
Douglas Gregor322328b2009-11-18 22:32:06 +00006003 DEnd = Container->decls_end();
6004 D != DEnd; ++D)
6005 if (ObjCPropertyImplDecl *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(*D))
6006 Results.Ignore(PropertyImpl->getPropertyDecl());
6007
6008 // Add any properties that we find.
Douglas Gregor73449212010-12-09 23:01:55 +00006009 AddedPropertiesSet AddedProperties;
Douglas Gregor322328b2009-11-18 22:32:06 +00006010 Results.EnterNewScope();
6011 if (ObjCImplementationDecl *ClassImpl
6012 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor4b81cde2011-05-05 15:50:42 +00006013 AddObjCProperties(ClassImpl->getClassInterface(), false,
6014 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00006015 AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00006016 else
6017 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor4b81cde2011-05-05 15:50:42 +00006018 false, /*AllowNullaryMethods=*/false, CurContext,
6019 AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00006020 Results.ExitScope();
6021
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006022 HandleCodeCompleteResults(this, CodeCompleter,
6023 CodeCompletionContext::CCC_Other,
6024 Results.data(),Results.size());
Douglas Gregor322328b2009-11-18 22:32:06 +00006025}
6026
6027void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006028 IdentifierInfo *PropertyName) {
John McCall0a2c5e22010-08-25 06:19:51 +00006029 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00006030 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006031 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00006032 CodeCompletionContext::CCC_Other);
Douglas Gregor322328b2009-11-18 22:32:06 +00006033
6034 // Figure out where this @synthesize lives.
6035 ObjCContainerDecl *Container
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006036 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor322328b2009-11-18 22:32:06 +00006037 if (!Container ||
6038 (!isa<ObjCImplementationDecl>(Container) &&
6039 !isa<ObjCCategoryImplDecl>(Container)))
6040 return;
6041
6042 // Figure out which interface we're looking into.
6043 ObjCInterfaceDecl *Class = 0;
6044 if (ObjCImplementationDecl *ClassImpl
6045 = dyn_cast<ObjCImplementationDecl>(Container))
6046 Class = ClassImpl->getClassInterface();
6047 else
6048 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
6049 ->getClassInterface();
6050
Douglas Gregore8426052011-04-18 14:40:46 +00006051 // Determine the type of the property we're synthesizing.
6052 QualType PropertyType = Context.getObjCIdType();
6053 if (Class) {
6054 if (ObjCPropertyDecl *Property
6055 = Class->FindPropertyDeclaration(PropertyName)) {
6056 PropertyType
6057 = Property->getType().getNonReferenceType().getUnqualifiedType();
6058
6059 // Give preference to ivars
6060 Results.setPreferredType(PropertyType);
6061 }
6062 }
6063
Douglas Gregor322328b2009-11-18 22:32:06 +00006064 // Add all of the instance variables in this class and its superclasses.
6065 Results.EnterNewScope();
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006066 bool SawSimilarlyNamedIvar = false;
6067 std::string NameWithPrefix;
6068 NameWithPrefix += '_';
Benjamin Kramera0651c52011-07-26 16:59:25 +00006069 NameWithPrefix += PropertyName->getName();
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006070 std::string NameWithSuffix = PropertyName->getName().str();
6071 NameWithSuffix += '_';
Douglas Gregor322328b2009-11-18 22:32:06 +00006072 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006073 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
6074 Ivar = Ivar->getNextIvar()) {
Douglas Gregord1f09b42013-01-31 04:52:16 +00006075 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), 0),
6076 CurContext, 0, false);
Douglas Gregore8426052011-04-18 14:40:46 +00006077
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006078 // Determine whether we've seen an ivar with a name similar to the
6079 // property.
Douglas Gregore8426052011-04-18 14:40:46 +00006080 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006081 NameWithPrefix == Ivar->getName() ||
Douglas Gregore8426052011-04-18 14:40:46 +00006082 NameWithSuffix == Ivar->getName())) {
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006083 SawSimilarlyNamedIvar = true;
Douglas Gregore8426052011-04-18 14:40:46 +00006084
6085 // Reduce the priority of this result by one, to give it a slight
6086 // advantage over other results whose names don't match so closely.
6087 if (Results.size() &&
6088 Results.data()[Results.size() - 1].Kind
6089 == CodeCompletionResult::RK_Declaration &&
6090 Results.data()[Results.size() - 1].Declaration == Ivar)
6091 Results.data()[Results.size() - 1].Priority--;
6092 }
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006093 }
Douglas Gregor322328b2009-11-18 22:32:06 +00006094 }
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006095
6096 if (!SawSimilarlyNamedIvar) {
6097 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregore8426052011-04-18 14:40:46 +00006098 // an ivar of the appropriate type.
6099 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006100 typedef CodeCompletionResult Result;
6101 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006102 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6103 Priority,CXAvailability_Available);
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006104
Douglas Gregor8987b232011-09-27 23:30:47 +00006105 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregore8426052011-04-18 14:40:46 +00006106 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006107 Policy, Allocator));
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006108 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6109 Results.AddResult(Result(Builder.TakeString(), Priority,
6110 CXCursor_ObjCIvarDecl));
6111 }
6112
Douglas Gregor322328b2009-11-18 22:32:06 +00006113 Results.ExitScope();
6114
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006115 HandleCodeCompleteResults(this, CodeCompleter,
6116 CodeCompletionContext::CCC_Other,
6117 Results.data(),Results.size());
Douglas Gregor322328b2009-11-18 22:32:06 +00006118}
Douglas Gregore8f5a172010-04-07 00:21:17 +00006119
Douglas Gregor408be5a2010-08-25 01:08:01 +00006120// Mapping from selectors to the methods that implement that selector, along
6121// with the "in original class" flag.
6122typedef llvm::DenseMap<Selector, std::pair<ObjCMethodDecl *, bool> >
6123 KnownMethodsMap;
Douglas Gregore8f5a172010-04-07 00:21:17 +00006124
6125/// \brief Find all of the methods that reside in the given container
6126/// (and its superclasses, protocols, etc.) that meet the given
6127/// criteria. Insert those methods into the map of known methods,
6128/// indexed by selector so they can be easily found.
6129static void FindImplementableMethods(ASTContext &Context,
6130 ObjCContainerDecl *Container,
6131 bool WantInstanceMethods,
6132 QualType ReturnType,
Douglas Gregor408be5a2010-08-25 01:08:01 +00006133 KnownMethodsMap &KnownMethods,
6134 bool InOriginalClass = true) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006135 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregorb92a4082012-06-12 13:44:08 +00006136 // Make sure we have a definition; that's what we'll walk.
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00006137 if (!IFace->hasDefinition())
6138 return;
Douglas Gregorb92a4082012-06-12 13:44:08 +00006139
6140 IFace = IFace->getDefinition();
6141 Container = IFace;
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00006142
Douglas Gregore8f5a172010-04-07 00:21:17 +00006143 const ObjCList<ObjCProtocolDecl> &Protocols
6144 = IFace->getReferencedProtocols();
6145 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregorea766182010-10-18 18:21:28 +00006146 E = Protocols.end();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006147 I != E; ++I)
6148 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00006149 KnownMethods, InOriginalClass);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006150
Douglas Gregorea766182010-10-18 18:21:28 +00006151 // Add methods from any class extensions and categories.
Douglas Gregord3297242013-01-16 23:00:23 +00006152 for (ObjCInterfaceDecl::visible_categories_iterator
6153 Cat = IFace->visible_categories_begin(),
6154 CatEnd = IFace->visible_categories_end();
6155 Cat != CatEnd; ++Cat) {
6156 FindImplementableMethods(Context, *Cat, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00006157 KnownMethods, false);
Douglas Gregord3297242013-01-16 23:00:23 +00006158 }
6159
Douglas Gregorea766182010-10-18 18:21:28 +00006160 // Visit the superclass.
6161 if (IFace->getSuperClass())
6162 FindImplementableMethods(Context, IFace->getSuperClass(),
6163 WantInstanceMethods, ReturnType,
6164 KnownMethods, false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006165 }
6166
6167 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6168 // Recurse into protocols.
6169 const ObjCList<ObjCProtocolDecl> &Protocols
6170 = Category->getReferencedProtocols();
6171 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregorea766182010-10-18 18:21:28 +00006172 E = Protocols.end();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006173 I != E; ++I)
6174 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00006175 KnownMethods, InOriginalClass);
6176
6177 // If this category is the original class, jump to the interface.
6178 if (InOriginalClass && Category->getClassInterface())
6179 FindImplementableMethods(Context, Category->getClassInterface(),
6180 WantInstanceMethods, ReturnType, KnownMethods,
6181 false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006182 }
6183
6184 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregorb92a4082012-06-12 13:44:08 +00006185 // Make sure we have a definition; that's what we'll walk.
6186 if (!Protocol->hasDefinition())
6187 return;
6188 Protocol = Protocol->getDefinition();
6189 Container = Protocol;
6190
6191 // Recurse into protocols.
6192 const ObjCList<ObjCProtocolDecl> &Protocols
6193 = Protocol->getReferencedProtocols();
6194 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6195 E = Protocols.end();
6196 I != E; ++I)
6197 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6198 KnownMethods, false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006199 }
6200
6201 // Add methods in this container. This operation occurs last because
6202 // we want the methods from this container to override any methods
6203 // we've previously seen with the same selector.
6204 for (ObjCContainerDecl::method_iterator M = Container->meth_begin(),
6205 MEnd = Container->meth_end();
6206 M != MEnd; ++M) {
David Blaikie262bc182012-04-30 02:36:29 +00006207 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006208 if (!ReturnType.isNull() &&
David Blaikie262bc182012-04-30 02:36:29 +00006209 !Context.hasSameUnqualifiedType(ReturnType, M->getResultType()))
Douglas Gregore8f5a172010-04-07 00:21:17 +00006210 continue;
6211
David Blaikie581deb32012-06-06 20:45:41 +00006212 KnownMethods[M->getSelector()] = std::make_pair(*M, InOriginalClass);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006213 }
6214 }
6215}
6216
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006217/// \brief Add the parenthesized return or parameter type chunk to a code
6218/// completion string.
6219static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor90f5f472012-04-10 18:35:07 +00006220 unsigned ObjCDeclQuals,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006221 ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006222 const PrintingPolicy &Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006223 CodeCompletionBuilder &Builder) {
6224 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor90f5f472012-04-10 18:35:07 +00006225 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6226 if (!Quals.empty())
6227 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor8987b232011-09-27 23:30:47 +00006228 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006229 Builder.getAllocator()));
6230 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6231}
6232
6233/// \brief Determine whether the given class is or inherits from a class by
6234/// the given name.
6235static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner5f9e2722011-07-23 10:55:15 +00006236 StringRef Name) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006237 if (!Class)
6238 return false;
6239
6240 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6241 return true;
6242
6243 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6244}
6245
6246/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6247/// Key-Value Observing (KVO).
6248static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6249 bool IsInstanceMethod,
6250 QualType ReturnType,
6251 ASTContext &Context,
Douglas Gregore74c25c2011-05-04 23:50:46 +00006252 VisitedSelectorSet &KnownSelectors,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006253 ResultBuilder &Results) {
6254 IdentifierInfo *PropName = Property->getIdentifier();
6255 if (!PropName || PropName->getLength() == 0)
6256 return;
6257
Douglas Gregor8987b232011-09-27 23:30:47 +00006258 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6259
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006260 // Builder that will create each code completion.
6261 typedef CodeCompletionResult Result;
6262 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006263 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006264
6265 // The selector table.
6266 SelectorTable &Selectors = Context.Selectors;
6267
6268 // The property name, copied into the code completion allocation region
6269 // on demand.
6270 struct KeyHolder {
6271 CodeCompletionAllocator &Allocator;
Chris Lattner5f9e2722011-07-23 10:55:15 +00006272 StringRef Key;
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006273 const char *CopiedKey;
6274
Chris Lattner5f9e2722011-07-23 10:55:15 +00006275 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006276 : Allocator(Allocator), Key(Key), CopiedKey(0) { }
6277
6278 operator const char *() {
6279 if (CopiedKey)
6280 return CopiedKey;
6281
6282 return CopiedKey = Allocator.CopyString(Key);
6283 }
6284 } Key(Allocator, PropName->getName());
6285
6286 // The uppercased name of the property name.
6287 std::string UpperKey = PropName->getName();
6288 if (!UpperKey.empty())
Jordan Rose223f0ff2013-02-09 10:09:43 +00006289 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006290
6291 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6292 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6293 Property->getType());
6294 bool ReturnTypeMatchesVoid
6295 = ReturnType.isNull() || ReturnType->isVoidType();
6296
6297 // Add the normal accessor -(type)key.
6298 if (IsInstanceMethod &&
Douglas Gregore74c25c2011-05-04 23:50:46 +00006299 KnownSelectors.insert(Selectors.getNullarySelector(PropName)) &&
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006300 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6301 if (ReturnType.isNull())
Douglas Gregor90f5f472012-04-10 18:35:07 +00006302 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6303 Context, Policy, Builder);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006304
6305 Builder.AddTypedTextChunk(Key);
6306 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6307 CXCursor_ObjCInstanceMethodDecl));
6308 }
6309
6310 // If we have an integral or boolean property (or the user has provided
6311 // an integral or boolean return type), add the accessor -(type)isKey.
6312 if (IsInstanceMethod &&
6313 ((!ReturnType.isNull() &&
6314 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6315 (ReturnType.isNull() &&
6316 (Property->getType()->isIntegerType() ||
6317 Property->getType()->isBooleanType())))) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006318 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006319 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006320 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006321 if (ReturnType.isNull()) {
6322 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6323 Builder.AddTextChunk("BOOL");
6324 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6325 }
6326
6327 Builder.AddTypedTextChunk(
6328 Allocator.CopyString(SelectorId->getName()));
6329 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6330 CXCursor_ObjCInstanceMethodDecl));
6331 }
6332 }
6333
6334 // Add the normal mutator.
6335 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6336 !Property->getSetterMethodDecl()) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006337 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006338 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006339 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006340 if (ReturnType.isNull()) {
6341 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6342 Builder.AddTextChunk("void");
6343 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6344 }
6345
6346 Builder.AddTypedTextChunk(
6347 Allocator.CopyString(SelectorId->getName()));
6348 Builder.AddTypedTextChunk(":");
Douglas Gregor90f5f472012-04-10 18:35:07 +00006349 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6350 Context, Policy, Builder);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006351 Builder.AddTextChunk(Key);
6352 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6353 CXCursor_ObjCInstanceMethodDecl));
6354 }
6355 }
6356
6357 // Indexed and unordered accessors
6358 unsigned IndexedGetterPriority = CCP_CodePattern;
6359 unsigned IndexedSetterPriority = CCP_CodePattern;
6360 unsigned UnorderedGetterPriority = CCP_CodePattern;
6361 unsigned UnorderedSetterPriority = CCP_CodePattern;
6362 if (const ObjCObjectPointerType *ObjCPointer
6363 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6364 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6365 // If this interface type is not provably derived from a known
6366 // collection, penalize the corresponding completions.
6367 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6368 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6369 if (!InheritsFromClassNamed(IFace, "NSArray"))
6370 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6371 }
6372
6373 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6374 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6375 if (!InheritsFromClassNamed(IFace, "NSSet"))
6376 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6377 }
6378 }
6379 } else {
6380 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6381 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6382 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6383 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6384 }
6385
6386 // Add -(NSUInteger)countOf<key>
6387 if (IsInstanceMethod &&
6388 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006389 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006390 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006391 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006392 if (ReturnType.isNull()) {
6393 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6394 Builder.AddTextChunk("NSUInteger");
6395 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6396 }
6397
6398 Builder.AddTypedTextChunk(
6399 Allocator.CopyString(SelectorId->getName()));
6400 Results.AddResult(Result(Builder.TakeString(),
6401 std::min(IndexedGetterPriority,
6402 UnorderedGetterPriority),
6403 CXCursor_ObjCInstanceMethodDecl));
6404 }
6405 }
6406
6407 // Indexed getters
6408 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6409 if (IsInstanceMethod &&
6410 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor62041592011-02-17 03:19:26 +00006411 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006412 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006413 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006414 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006415 if (ReturnType.isNull()) {
6416 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6417 Builder.AddTextChunk("id");
6418 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6419 }
6420
6421 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6422 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6423 Builder.AddTextChunk("NSUInteger");
6424 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6425 Builder.AddTextChunk("index");
6426 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6427 CXCursor_ObjCInstanceMethodDecl));
6428 }
6429 }
6430
6431 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6432 if (IsInstanceMethod &&
6433 (ReturnType.isNull() ||
6434 (ReturnType->isObjCObjectPointerType() &&
6435 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6436 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6437 ->getName() == "NSArray"))) {
Douglas Gregor62041592011-02-17 03:19:26 +00006438 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006439 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006440 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006441 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006442 if (ReturnType.isNull()) {
6443 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6444 Builder.AddTextChunk("NSArray *");
6445 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6446 }
6447
6448 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6449 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6450 Builder.AddTextChunk("NSIndexSet *");
6451 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6452 Builder.AddTextChunk("indexes");
6453 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6454 CXCursor_ObjCInstanceMethodDecl));
6455 }
6456 }
6457
6458 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6459 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006460 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006461 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006462 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006463 &Context.Idents.get("range")
6464 };
6465
Douglas Gregore74c25c2011-05-04 23:50:46 +00006466 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006467 if (ReturnType.isNull()) {
6468 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6469 Builder.AddTextChunk("void");
6470 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6471 }
6472
6473 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6474 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6475 Builder.AddPlaceholderChunk("object-type");
6476 Builder.AddTextChunk(" **");
6477 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6478 Builder.AddTextChunk("buffer");
6479 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6480 Builder.AddTypedTextChunk("range:");
6481 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6482 Builder.AddTextChunk("NSRange");
6483 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6484 Builder.AddTextChunk("inRange");
6485 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6486 CXCursor_ObjCInstanceMethodDecl));
6487 }
6488 }
6489
6490 // Mutable indexed accessors
6491
6492 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6493 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006494 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006495 IdentifierInfo *SelectorIds[2] = {
6496 &Context.Idents.get("insertObject"),
Douglas Gregor62041592011-02-17 03:19:26 +00006497 &Context.Idents.get(SelectorName)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006498 };
6499
Douglas Gregore74c25c2011-05-04 23:50:46 +00006500 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006501 if (ReturnType.isNull()) {
6502 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6503 Builder.AddTextChunk("void");
6504 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6505 }
6506
6507 Builder.AddTypedTextChunk("insertObject:");
6508 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6509 Builder.AddPlaceholderChunk("object-type");
6510 Builder.AddTextChunk(" *");
6511 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6512 Builder.AddTextChunk("object");
6513 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6514 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6515 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6516 Builder.AddPlaceholderChunk("NSUInteger");
6517 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6518 Builder.AddTextChunk("index");
6519 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6520 CXCursor_ObjCInstanceMethodDecl));
6521 }
6522 }
6523
6524 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6525 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006526 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006527 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006528 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006529 &Context.Idents.get("atIndexes")
6530 };
6531
Douglas Gregore74c25c2011-05-04 23:50:46 +00006532 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006533 if (ReturnType.isNull()) {
6534 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6535 Builder.AddTextChunk("void");
6536 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6537 }
6538
6539 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6540 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6541 Builder.AddTextChunk("NSArray *");
6542 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6543 Builder.AddTextChunk("array");
6544 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6545 Builder.AddTypedTextChunk("atIndexes:");
6546 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6547 Builder.AddPlaceholderChunk("NSIndexSet *");
6548 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6549 Builder.AddTextChunk("indexes");
6550 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6551 CXCursor_ObjCInstanceMethodDecl));
6552 }
6553 }
6554
6555 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6556 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006557 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006558 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006559 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006560 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006561 if (ReturnType.isNull()) {
6562 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6563 Builder.AddTextChunk("void");
6564 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6565 }
6566
6567 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6568 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6569 Builder.AddTextChunk("NSUInteger");
6570 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6571 Builder.AddTextChunk("index");
6572 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6573 CXCursor_ObjCInstanceMethodDecl));
6574 }
6575 }
6576
6577 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6578 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006579 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006580 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006581 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006582 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006583 if (ReturnType.isNull()) {
6584 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6585 Builder.AddTextChunk("void");
6586 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6587 }
6588
6589 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6590 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6591 Builder.AddTextChunk("NSIndexSet *");
6592 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6593 Builder.AddTextChunk("indexes");
6594 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6595 CXCursor_ObjCInstanceMethodDecl));
6596 }
6597 }
6598
6599 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6600 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006601 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006602 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006603 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006604 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006605 &Context.Idents.get("withObject")
6606 };
6607
Douglas Gregore74c25c2011-05-04 23:50:46 +00006608 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006609 if (ReturnType.isNull()) {
6610 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6611 Builder.AddTextChunk("void");
6612 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6613 }
6614
6615 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6616 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6617 Builder.AddPlaceholderChunk("NSUInteger");
6618 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6619 Builder.AddTextChunk("index");
6620 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6621 Builder.AddTypedTextChunk("withObject:");
6622 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6623 Builder.AddTextChunk("id");
6624 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6625 Builder.AddTextChunk("object");
6626 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6627 CXCursor_ObjCInstanceMethodDecl));
6628 }
6629 }
6630
6631 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6632 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006633 std::string SelectorName1
Chris Lattner5f9e2722011-07-23 10:55:15 +00006634 = (Twine("replace") + UpperKey + "AtIndexes").str();
6635 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006636 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006637 &Context.Idents.get(SelectorName1),
6638 &Context.Idents.get(SelectorName2)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006639 };
6640
Douglas Gregore74c25c2011-05-04 23:50:46 +00006641 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006642 if (ReturnType.isNull()) {
6643 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6644 Builder.AddTextChunk("void");
6645 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6646 }
6647
6648 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6649 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6650 Builder.AddPlaceholderChunk("NSIndexSet *");
6651 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6652 Builder.AddTextChunk("indexes");
6653 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6654 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6655 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6656 Builder.AddTextChunk("NSArray *");
6657 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6658 Builder.AddTextChunk("array");
6659 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6660 CXCursor_ObjCInstanceMethodDecl));
6661 }
6662 }
6663
6664 // Unordered getters
6665 // - (NSEnumerator *)enumeratorOfKey
6666 if (IsInstanceMethod &&
6667 (ReturnType.isNull() ||
6668 (ReturnType->isObjCObjectPointerType() &&
6669 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6670 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6671 ->getName() == "NSEnumerator"))) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006672 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006673 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006674 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006675 if (ReturnType.isNull()) {
6676 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6677 Builder.AddTextChunk("NSEnumerator *");
6678 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6679 }
6680
6681 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6682 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6683 CXCursor_ObjCInstanceMethodDecl));
6684 }
6685 }
6686
6687 // - (type *)memberOfKey:(type *)object
6688 if (IsInstanceMethod &&
6689 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006690 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006691 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006692 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006693 if (ReturnType.isNull()) {
6694 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6695 Builder.AddPlaceholderChunk("object-type");
6696 Builder.AddTextChunk(" *");
6697 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6698 }
6699
6700 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6701 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6702 if (ReturnType.isNull()) {
6703 Builder.AddPlaceholderChunk("object-type");
6704 Builder.AddTextChunk(" *");
6705 } else {
6706 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006707 Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006708 Builder.getAllocator()));
6709 }
6710 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6711 Builder.AddTextChunk("object");
6712 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6713 CXCursor_ObjCInstanceMethodDecl));
6714 }
6715 }
6716
6717 // Mutable unordered accessors
6718 // - (void)addKeyObject:(type *)object
6719 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006720 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006721 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006722 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006723 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006724 if (ReturnType.isNull()) {
6725 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6726 Builder.AddTextChunk("void");
6727 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6728 }
6729
6730 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6731 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6732 Builder.AddPlaceholderChunk("object-type");
6733 Builder.AddTextChunk(" *");
6734 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6735 Builder.AddTextChunk("object");
6736 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6737 CXCursor_ObjCInstanceMethodDecl));
6738 }
6739 }
6740
6741 // - (void)addKey:(NSSet *)objects
6742 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006743 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006744 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006745 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006746 if (ReturnType.isNull()) {
6747 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6748 Builder.AddTextChunk("void");
6749 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6750 }
6751
6752 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6753 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6754 Builder.AddTextChunk("NSSet *");
6755 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6756 Builder.AddTextChunk("objects");
6757 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6758 CXCursor_ObjCInstanceMethodDecl));
6759 }
6760 }
6761
6762 // - (void)removeKeyObject:(type *)object
6763 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006764 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006765 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006766 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006767 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006768 if (ReturnType.isNull()) {
6769 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6770 Builder.AddTextChunk("void");
6771 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6772 }
6773
6774 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6775 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6776 Builder.AddPlaceholderChunk("object-type");
6777 Builder.AddTextChunk(" *");
6778 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6779 Builder.AddTextChunk("object");
6780 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6781 CXCursor_ObjCInstanceMethodDecl));
6782 }
6783 }
6784
6785 // - (void)removeKey:(NSSet *)objects
6786 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006787 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006788 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006789 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006790 if (ReturnType.isNull()) {
6791 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6792 Builder.AddTextChunk("void");
6793 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6794 }
6795
6796 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6797 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6798 Builder.AddTextChunk("NSSet *");
6799 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6800 Builder.AddTextChunk("objects");
6801 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6802 CXCursor_ObjCInstanceMethodDecl));
6803 }
6804 }
6805
6806 // - (void)intersectKey:(NSSet *)objects
6807 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006808 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006809 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006810 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006811 if (ReturnType.isNull()) {
6812 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6813 Builder.AddTextChunk("void");
6814 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6815 }
6816
6817 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6818 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6819 Builder.AddTextChunk("NSSet *");
6820 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6821 Builder.AddTextChunk("objects");
6822 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6823 CXCursor_ObjCInstanceMethodDecl));
6824 }
6825 }
6826
6827 // Key-Value Observing
6828 // + (NSSet *)keyPathsForValuesAffectingKey
6829 if (!IsInstanceMethod &&
6830 (ReturnType.isNull() ||
6831 (ReturnType->isObjCObjectPointerType() &&
6832 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6833 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6834 ->getName() == "NSSet"))) {
Douglas Gregor62041592011-02-17 03:19:26 +00006835 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006836 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006837 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Douglas Gregore74c25c2011-05-04 23:50:46 +00006838 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006839 if (ReturnType.isNull()) {
6840 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6841 Builder.AddTextChunk("NSSet *");
6842 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6843 }
6844
6845 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6846 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor3f828d12011-06-02 04:02:27 +00006847 CXCursor_ObjCClassMethodDecl));
6848 }
6849 }
6850
6851 // + (BOOL)automaticallyNotifiesObserversForKey
6852 if (!IsInstanceMethod &&
6853 (ReturnType.isNull() ||
6854 ReturnType->isIntegerType() ||
6855 ReturnType->isBooleanType())) {
6856 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006857 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor3f828d12011-06-02 04:02:27 +00006858 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
6859 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))) {
6860 if (ReturnType.isNull()) {
6861 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6862 Builder.AddTextChunk("BOOL");
6863 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6864 }
6865
6866 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6867 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6868 CXCursor_ObjCClassMethodDecl));
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006869 }
6870 }
6871}
6872
Douglas Gregore8f5a172010-04-07 00:21:17 +00006873void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6874 bool IsInstanceMethod,
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006875 ParsedType ReturnTy) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006876 // Determine the return type of the method we're declaring, if
6877 // provided.
6878 QualType ReturnType = GetTypeFromParser(ReturnTy);
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006879 Decl *IDecl = 0;
6880 if (CurContext->isObjCContainer()) {
6881 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6882 IDecl = cast<Decl>(OCD);
6883 }
Douglas Gregorea766182010-10-18 18:21:28 +00006884 // Determine where we should start searching for methods.
6885 ObjCContainerDecl *SearchDecl = 0;
Douglas Gregore8f5a172010-04-07 00:21:17 +00006886 bool IsInImplementation = false;
John McCalld226f652010-08-21 09:40:31 +00006887 if (Decl *D = IDecl) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006888 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6889 SearchDecl = Impl->getClassInterface();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006890 IsInImplementation = true;
6891 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregorea766182010-10-18 18:21:28 +00006892 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006893 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006894 IsInImplementation = true;
Douglas Gregorea766182010-10-18 18:21:28 +00006895 } else
Douglas Gregore8f5a172010-04-07 00:21:17 +00006896 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006897 }
6898
6899 if (!SearchDecl && S) {
Douglas Gregorea766182010-10-18 18:21:28 +00006900 if (DeclContext *DC = static_cast<DeclContext *>(S->getEntity()))
Douglas Gregore8f5a172010-04-07 00:21:17 +00006901 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006902 }
6903
Douglas Gregorea766182010-10-18 18:21:28 +00006904 if (!SearchDecl) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006905 HandleCodeCompleteResults(this, CodeCompleter,
6906 CodeCompletionContext::CCC_Other,
6907 0, 0);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006908 return;
6909 }
6910
6911 // Find all of the methods that we could declare/implement here.
6912 KnownMethodsMap KnownMethods;
6913 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregorea766182010-10-18 18:21:28 +00006914 ReturnType, KnownMethods);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006915
Douglas Gregore8f5a172010-04-07 00:21:17 +00006916 // Add declarations or definitions for each of the known methods.
John McCall0a2c5e22010-08-25 06:19:51 +00006917 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00006918 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006919 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00006920 CodeCompletionContext::CCC_Other);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006921 Results.EnterNewScope();
Douglas Gregor8987b232011-09-27 23:30:47 +00006922 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006923 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6924 MEnd = KnownMethods.end();
6925 M != MEnd; ++M) {
Douglas Gregor408be5a2010-08-25 01:08:01 +00006926 ObjCMethodDecl *Method = M->second.first;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006927 CodeCompletionBuilder Builder(Results.getAllocator(),
6928 Results.getCodeCompletionTUInfo());
Douglas Gregore8f5a172010-04-07 00:21:17 +00006929
6930 // If the result type was not already provided, add it to the
6931 // pattern as (type).
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006932 if (ReturnType.isNull())
Douglas Gregor90f5f472012-04-10 18:35:07 +00006933 AddObjCPassingTypeChunk(Method->getResultType(),
6934 Method->getObjCDeclQualifier(),
6935 Context, Policy,
6936 Builder);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006937
6938 Selector Sel = Method->getSelector();
6939
6940 // Add the first part of the selector to the pattern.
Douglas Gregordae68752011-02-01 22:57:45 +00006941 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00006942 Sel.getNameForSlot(0)));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006943
6944 // Add parameters to the pattern.
6945 unsigned I = 0;
6946 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
6947 PEnd = Method->param_end();
6948 P != PEnd; (void)++P, ++I) {
6949 // Add the part of the selector name.
6950 if (I == 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00006951 Builder.AddTypedTextChunk(":");
Douglas Gregore8f5a172010-04-07 00:21:17 +00006952 else if (I < Sel.getNumArgs()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00006953 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6954 Builder.AddTypedTextChunk(
Douglas Gregor813d8342011-02-18 22:29:55 +00006955 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006956 } else
6957 break;
6958
6959 // Add the parameter type.
Douglas Gregor90f5f472012-04-10 18:35:07 +00006960 AddObjCPassingTypeChunk((*P)->getOriginalType(),
6961 (*P)->getObjCDeclQualifier(),
6962 Context, Policy,
Douglas Gregor8987b232011-09-27 23:30:47 +00006963 Builder);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006964
6965 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregordae68752011-02-01 22:57:45 +00006966 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006967 }
6968
6969 if (Method->isVariadic()) {
6970 if (Method->param_size() > 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00006971 Builder.AddChunk(CodeCompletionString::CK_Comma);
6972 Builder.AddTextChunk("...");
Douglas Gregore17794f2010-08-31 05:13:43 +00006973 }
Douglas Gregore8f5a172010-04-07 00:21:17 +00006974
Douglas Gregor447107d2010-05-28 00:57:46 +00006975 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006976 // We will be defining the method here, so add a compound statement.
Douglas Gregor218937c2011-02-01 19:23:04 +00006977 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6978 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
6979 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006980 if (!Method->getResultType()->isVoidType()) {
6981 // If the result type is not void, add a return clause.
Douglas Gregor218937c2011-02-01 19:23:04 +00006982 Builder.AddTextChunk("return");
6983 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6984 Builder.AddPlaceholderChunk("expression");
6985 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006986 } else
Douglas Gregor218937c2011-02-01 19:23:04 +00006987 Builder.AddPlaceholderChunk("statements");
Douglas Gregore8f5a172010-04-07 00:21:17 +00006988
Douglas Gregor218937c2011-02-01 19:23:04 +00006989 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
6990 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006991 }
6992
Douglas Gregor408be5a2010-08-25 01:08:01 +00006993 unsigned Priority = CCP_CodePattern;
6994 if (!M->second.second)
6995 Priority += CCD_InBaseClass;
6996
Douglas Gregorba103062012-03-27 23:34:16 +00006997 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregore8f5a172010-04-07 00:21:17 +00006998 }
6999
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007000 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
7001 // the properties in this class and its categories.
David Blaikie4e4d0842012-03-11 07:00:24 +00007002 if (Context.getLangOpts().ObjC2) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00007003 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007004 Containers.push_back(SearchDecl);
7005
Douglas Gregore74c25c2011-05-04 23:50:46 +00007006 VisitedSelectorSet KnownSelectors;
7007 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
7008 MEnd = KnownMethods.end();
7009 M != MEnd; ++M)
7010 KnownSelectors.insert(M->first);
7011
7012
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007013 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
7014 if (!IFace)
7015 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
7016 IFace = Category->getClassInterface();
7017
7018 if (IFace) {
Douglas Gregord3297242013-01-16 23:00:23 +00007019 for (ObjCInterfaceDecl::visible_categories_iterator
7020 Cat = IFace->visible_categories_begin(),
7021 CatEnd = IFace->visible_categories_end();
7022 Cat != CatEnd; ++Cat) {
7023 Containers.push_back(*Cat);
7024 }
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007025 }
7026
7027 for (unsigned I = 0, N = Containers.size(); I != N; ++I) {
7028 for (ObjCContainerDecl::prop_iterator P = Containers[I]->prop_begin(),
7029 PEnd = Containers[I]->prop_end();
7030 P != PEnd; ++P) {
David Blaikie581deb32012-06-06 20:45:41 +00007031 AddObjCKeyValueCompletions(*P, IsInstanceMethod, ReturnType, Context,
Douglas Gregore74c25c2011-05-04 23:50:46 +00007032 KnownSelectors, Results);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007033 }
7034 }
7035 }
7036
Douglas Gregore8f5a172010-04-07 00:21:17 +00007037 Results.ExitScope();
7038
Douglas Gregore6b1bb62010-08-11 21:23:17 +00007039 HandleCodeCompleteResults(this, CodeCompleter,
7040 CodeCompletionContext::CCC_Other,
7041 Results.data(),Results.size());
Douglas Gregore8f5a172010-04-07 00:21:17 +00007042}
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007043
7044void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
7045 bool IsInstanceMethod,
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007046 bool AtParameterName,
John McCallb3d87482010-08-24 05:47:05 +00007047 ParsedType ReturnTy,
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007048 IdentifierInfo **SelIdents,
7049 unsigned NumSelIdents) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007050 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00007051 // pool from the AST file.
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007052 if (ExternalSource) {
7053 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
7054 I != N; ++I) {
7055 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redldb9d2142010-08-02 23:18:59 +00007056 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007057 continue;
Sebastian Redldb9d2142010-08-02 23:18:59 +00007058
7059 ReadMethodPool(Sel);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007060 }
7061 }
7062
7063 // Build the set of methods we can see.
John McCall0a2c5e22010-08-25 06:19:51 +00007064 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00007065 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007066 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00007067 CodeCompletionContext::CCC_Other);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007068
7069 if (ReturnTy)
7070 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redldb9d2142010-08-02 23:18:59 +00007071
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007072 Results.EnterNewScope();
Sebastian Redldb9d2142010-08-02 23:18:59 +00007073 for (GlobalMethodPool::iterator M = MethodPool.begin(),
7074 MEnd = MethodPool.end();
7075 M != MEnd; ++M) {
7076 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
7077 &M->second.second;
7078 MethList && MethList->Method;
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007079 MethList = MethList->Next) {
7080 if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents,
7081 NumSelIdents))
7082 continue;
7083
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007084 if (AtParameterName) {
7085 // Suggest parameter names we've seen before.
7086 if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) {
7087 ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1];
7088 if (Param->getIdentifier()) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007089 CodeCompletionBuilder Builder(Results.getAllocator(),
7090 Results.getCodeCompletionTUInfo());
Douglas Gregordae68752011-02-01 22:57:45 +00007091 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00007092 Param->getIdentifier()->getName()));
7093 Results.AddResult(Builder.TakeString());
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007094 }
7095 }
7096
7097 continue;
7098 }
7099
Douglas Gregord1f09b42013-01-31 04:52:16 +00007100 Result R(MethList->Method, Results.getBasePriority(MethList->Method), 0);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007101 R.StartParameter = NumSelIdents;
7102 R.AllParametersAreInformative = false;
7103 R.DeclaringEntity = true;
7104 Results.MaybeAddResult(R, CurContext);
7105 }
7106 }
7107
7108 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00007109 HandleCodeCompleteResults(this, CodeCompleter,
7110 CodeCompletionContext::CCC_Other,
7111 Results.data(),Results.size());
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007112}
Douglas Gregor87c08a52010-08-13 22:48:40 +00007113
Douglas Gregorf29c5232010-08-24 22:20:20 +00007114void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregor218937c2011-02-01 19:23:04 +00007115 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007116 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00007117 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregorf44e8542010-08-24 19:08:16 +00007118 Results.EnterNewScope();
7119
7120 // #if <condition>
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007121 CodeCompletionBuilder Builder(Results.getAllocator(),
7122 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00007123 Builder.AddTypedTextChunk("if");
7124 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7125 Builder.AddPlaceholderChunk("condition");
7126 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007127
7128 // #ifdef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007129 Builder.AddTypedTextChunk("ifdef");
7130 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7131 Builder.AddPlaceholderChunk("macro");
7132 Results.AddResult(Builder.TakeString());
7133
Douglas Gregorf44e8542010-08-24 19:08:16 +00007134 // #ifndef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007135 Builder.AddTypedTextChunk("ifndef");
7136 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7137 Builder.AddPlaceholderChunk("macro");
7138 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007139
7140 if (InConditional) {
7141 // #elif <condition>
Douglas Gregor218937c2011-02-01 19:23:04 +00007142 Builder.AddTypedTextChunk("elif");
7143 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7144 Builder.AddPlaceholderChunk("condition");
7145 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007146
7147 // #else
Douglas Gregor218937c2011-02-01 19:23:04 +00007148 Builder.AddTypedTextChunk("else");
7149 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007150
7151 // #endif
Douglas Gregor218937c2011-02-01 19:23:04 +00007152 Builder.AddTypedTextChunk("endif");
7153 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007154 }
7155
7156 // #include "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00007157 Builder.AddTypedTextChunk("include");
7158 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7159 Builder.AddTextChunk("\"");
7160 Builder.AddPlaceholderChunk("header");
7161 Builder.AddTextChunk("\"");
7162 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007163
7164 // #include <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00007165 Builder.AddTypedTextChunk("include");
7166 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7167 Builder.AddTextChunk("<");
7168 Builder.AddPlaceholderChunk("header");
7169 Builder.AddTextChunk(">");
7170 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007171
7172 // #define <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007173 Builder.AddTypedTextChunk("define");
7174 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7175 Builder.AddPlaceholderChunk("macro");
7176 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007177
7178 // #define <macro>(<args>)
Douglas Gregor218937c2011-02-01 19:23:04 +00007179 Builder.AddTypedTextChunk("define");
7180 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7181 Builder.AddPlaceholderChunk("macro");
7182 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7183 Builder.AddPlaceholderChunk("args");
7184 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7185 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007186
7187 // #undef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007188 Builder.AddTypedTextChunk("undef");
7189 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7190 Builder.AddPlaceholderChunk("macro");
7191 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007192
7193 // #line <number>
Douglas Gregor218937c2011-02-01 19:23:04 +00007194 Builder.AddTypedTextChunk("line");
7195 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7196 Builder.AddPlaceholderChunk("number");
7197 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007198
7199 // #line <number> "filename"
Douglas Gregor218937c2011-02-01 19:23:04 +00007200 Builder.AddTypedTextChunk("line");
7201 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7202 Builder.AddPlaceholderChunk("number");
7203 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7204 Builder.AddTextChunk("\"");
7205 Builder.AddPlaceholderChunk("filename");
7206 Builder.AddTextChunk("\"");
7207 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007208
7209 // #error <message>
Douglas Gregor218937c2011-02-01 19:23:04 +00007210 Builder.AddTypedTextChunk("error");
7211 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7212 Builder.AddPlaceholderChunk("message");
7213 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007214
7215 // #pragma <arguments>
Douglas Gregor218937c2011-02-01 19:23:04 +00007216 Builder.AddTypedTextChunk("pragma");
7217 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7218 Builder.AddPlaceholderChunk("arguments");
7219 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007220
David Blaikie4e4d0842012-03-11 07:00:24 +00007221 if (getLangOpts().ObjC1) {
Douglas Gregorf44e8542010-08-24 19:08:16 +00007222 // #import "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00007223 Builder.AddTypedTextChunk("import");
7224 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7225 Builder.AddTextChunk("\"");
7226 Builder.AddPlaceholderChunk("header");
7227 Builder.AddTextChunk("\"");
7228 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007229
7230 // #import <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00007231 Builder.AddTypedTextChunk("import");
7232 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7233 Builder.AddTextChunk("<");
7234 Builder.AddPlaceholderChunk("header");
7235 Builder.AddTextChunk(">");
7236 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007237 }
7238
7239 // #include_next "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00007240 Builder.AddTypedTextChunk("include_next");
7241 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7242 Builder.AddTextChunk("\"");
7243 Builder.AddPlaceholderChunk("header");
7244 Builder.AddTextChunk("\"");
7245 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007246
7247 // #include_next <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00007248 Builder.AddTypedTextChunk("include_next");
7249 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7250 Builder.AddTextChunk("<");
7251 Builder.AddPlaceholderChunk("header");
7252 Builder.AddTextChunk(">");
7253 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007254
7255 // #warning <message>
Douglas Gregor218937c2011-02-01 19:23:04 +00007256 Builder.AddTypedTextChunk("warning");
7257 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7258 Builder.AddPlaceholderChunk("message");
7259 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007260
7261 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7262 // completions for them. And __include_macros is a Clang-internal extension
7263 // that we don't want to encourage anyone to use.
7264
7265 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7266 Results.ExitScope();
7267
Douglas Gregorf44e8542010-08-24 19:08:16 +00007268 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor721f3592010-08-25 18:41:16 +00007269 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregorf44e8542010-08-24 19:08:16 +00007270 Results.data(), Results.size());
7271}
7272
7273void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorf29c5232010-08-24 22:20:20 +00007274 CodeCompleteOrdinaryName(S,
John McCallf312b1e2010-08-26 23:41:50 +00007275 S->getFnParent()? Sema::PCC_RecoveryInFunction
7276 : Sema::PCC_Namespace);
Douglas Gregorf44e8542010-08-24 19:08:16 +00007277}
7278
Douglas Gregorf29c5232010-08-24 22:20:20 +00007279void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregor218937c2011-02-01 19:23:04 +00007280 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007281 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00007282 IsDefinition? CodeCompletionContext::CCC_MacroName
7283 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007284 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7285 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007286 CodeCompletionBuilder Builder(Results.getAllocator(),
7287 Results.getCodeCompletionTUInfo());
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007288 Results.EnterNewScope();
7289 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7290 MEnd = PP.macro_end();
7291 M != MEnd; ++M) {
Douglas Gregordae68752011-02-01 22:57:45 +00007292 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00007293 M->first->getName()));
Argyrios Kyrtzidisc04bb922012-09-27 00:24:09 +00007294 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7295 CCP_CodePattern,
7296 CXCursor_MacroDefinition));
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007297 }
7298 Results.ExitScope();
7299 } else if (IsDefinition) {
7300 // FIXME: Can we detect when the user just wrote an include guard above?
7301 }
7302
Douglas Gregor52779fb2010-09-23 23:01:17 +00007303 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007304 Results.data(), Results.size());
7305}
7306
Douglas Gregorf29c5232010-08-24 22:20:20 +00007307void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregor218937c2011-02-01 19:23:04 +00007308 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007309 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00007310 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorf29c5232010-08-24 22:20:20 +00007311
7312 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00007313 AddMacroResults(PP, Results, true);
Douglas Gregorf29c5232010-08-24 22:20:20 +00007314
7315 // defined (<macro>)
7316 Results.EnterNewScope();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007317 CodeCompletionBuilder Builder(Results.getAllocator(),
7318 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00007319 Builder.AddTypedTextChunk("defined");
7320 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7321 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7322 Builder.AddPlaceholderChunk("macro");
7323 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7324 Results.AddResult(Builder.TakeString());
Douglas Gregorf29c5232010-08-24 22:20:20 +00007325 Results.ExitScope();
7326
7327 HandleCodeCompleteResults(this, CodeCompleter,
7328 CodeCompletionContext::CCC_PreprocessorExpression,
7329 Results.data(), Results.size());
7330}
7331
7332void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7333 IdentifierInfo *Macro,
7334 MacroInfo *MacroInfo,
7335 unsigned Argument) {
7336 // FIXME: In the future, we could provide "overload" results, much like we
7337 // do for function calls.
7338
Argyrios Kyrtzidis5c5f03e2011-08-18 19:41:28 +00007339 // Now just ignore this. There will be another code-completion callback
7340 // for the expanded tokens.
Douglas Gregorf29c5232010-08-24 22:20:20 +00007341}
7342
Douglas Gregor55817af2010-08-25 17:04:25 +00007343void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor55817af2010-08-25 17:04:25 +00007344 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregoraf1c6b52010-08-25 17:10:00 +00007345 CodeCompletionContext::CCC_NaturalLanguage,
Douglas Gregor55817af2010-08-25 17:04:25 +00007346 0, 0);
7347}
7348
Douglas Gregordae68752011-02-01 22:57:45 +00007349void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007350 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner5f9e2722011-07-23 10:55:15 +00007351 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007352 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7353 CodeCompletionContext::CCC_Recovery);
Douglas Gregor8071e422010-08-15 06:18:01 +00007354 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7355 CodeCompletionDeclConsumer Consumer(Builder,
7356 Context.getTranslationUnitDecl());
7357 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7358 Consumer);
7359 }
Douglas Gregor87c08a52010-08-13 22:48:40 +00007360
7361 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00007362 AddMacroResults(PP, Builder, true);
Douglas Gregor87c08a52010-08-13 22:48:40 +00007363
7364 Results.clear();
7365 Results.insert(Results.end(),
7366 Builder.data(), Builder.data() + Builder.size());
7367}